use crate::{
    layout::Layout,
    layout::{
        bend::LooseBendIndex, dot::LooseDotIndex, geometry::GeometryIndex, seg::SeqLooseSegIndex,
    },
    primitive::{GetInterior, GetJoints, GetOtherJoint, LooseBend, LooseDot},
};

#[derive(Debug, Clone, Copy)]
pub struct Segbend {
    pub seg: SeqLooseSegIndex,
    pub dot: LooseDotIndex,
    pub bend: LooseBendIndex,
}

impl Segbend {
    pub fn from_dot(dot: LooseDotIndex, layout: &Layout) -> Self {
        let bend = LooseDot::new(dot, layout).bend();
        let dot = LooseBend::new(bend, layout).other_joint(dot);
        let seg = LooseDot::new(dot, layout).seg().unwrap();
        Self { bend, dot, seg }
    }
}

impl GetInterior<GeometryIndex> for Segbend {
    fn interior(&self) -> Vec<GeometryIndex> {
        vec![self.bend.into(), self.dot.into(), self.seg.into()]
    }
}

impl GetJoints<SeqLooseSegIndex, LooseBendIndex> for Segbend {
    fn joints(&self) -> (SeqLooseSegIndex, LooseBendIndex) {
        (self.seg, self.bend)
    }
}