use crate::{
    draw::draw_line_dotted,
    point::Point,
    qualibration::{
        annalyse::{get_horizontal_segment, image_diff},
        compute_image::{image_treshold, image_warp},
        param::Param,
        Sequence,
    },
};

use opencv::{
    core::{Point as OcvPoint, VecN},
    highgui,
    imgproc::line,
    Result,
};

opencv::opencv_branch_4! {
    use opencv::imgproc::LINE_8;
}
opencv::not_opencv_branch_4! {
    use opencv::core::LINE_AA;
}

#[derive(Debug, Clone, Copy)]
pub struct InitIdcode {
    finished: bool,
    cnt: usize,
    beg: Point,
    end: Point,
}

impl InitIdcode {
    pub fn new(beg: Point, end: Point) -> InitIdcode {
        InitIdcode {
            finished: false,
            cnt: 0,
            beg,
            end,
        }
    }
}

impl Sequence for InitIdcode {
    fn draw(&self, mem: &Param) -> Option<Vec<Point>> {
        if self.finished {
            return None;
        }
        if self.cnt == 0 {
            return Some(vec![]);
        }
        let mut pl = vec![];
        pl.extend(draw_line_dotted(
            &self.beg,
            &self.end,
            mem.nb_all as usize,
            mem.nb_visible as usize,
            true,
        ));
        Some(pl)
    }
    fn compute_sequence(&mut self, mem: &mut Param) -> Result<(), Box<dyn std::error::Error>> {
        if self.cnt == 0 {
            self.cnt += 1;
            return Ok(())
        }
        let id = mem.seq_id;
        let mut id_code_1 = image_diff(&mem.imgs[id][1], &mem.imgs[id][0])?;

        id_code_1 = image_warp(&id_code_1, &mem.homography, mem.h_size)?;
        id_code_1 = image_treshold(&id_code_1, &mem.tresh)?;

        let code_seg_1 = get_horizontal_segment(&id_code_1)?;
        let code_seg_1 = code_seg_1[1..(code_seg_1.len() - 1)].to_owned();

        // on dessine
        let color_1: VecN<f64, 4> = VecN::new(255., 0., 0., 255.);
        for i in 0..code_seg_1.len() {
            let (((x0, y0), (x1, y1)), size) = code_seg_1[i];
            //line(&mut id_code_1, );
            let s = size as i32;
            let x = ((x0 + x1) / 2.) as i32;
            let y = ((y0 + y1) / 2.) as i32;
            let a = OcvPoint::from_vec2(VecN::from_array([x, y - s]));
            let b = OcvPoint::from_vec2(VecN::from_array([x, y + s]));
            line(&mut id_code_1, a, b, color_1, 1, LINE_8, 0)?;
            if i < (code_seg_1.len() - 1) {
                let (((x2, _), _), _) = code_seg_1[i + 1];
                let x = ((x1 + x2) / 2.) as i32;
                let y = ((y0 + y1) / 2.) as i32;
                let a = OcvPoint::from_vec2(VecN::from_array([x, y - s]));
                let b = OcvPoint::from_vec2(VecN::from_array([x, y + s]));
                line(&mut id_code_1, a, b, color_1, 1, LINE_8, 0)?;
            }
        }

        highgui::imshow("code 1", &id_code_1)?;
        self.finished = true;
        Ok(())
    }
    fn is_capture(&self) -> bool {
        true
    }
}