sixel-image 0.2.1

use sixel_tokenizer::SixelEvent;
use std::collections::BTreeMap;
use std::iter;

use crate::{Pixel, SixelColor, SixelImage, DCS, RA};

#[derive(Debug, Clone)]
pub struct SixelDeserializer {
    dcs: DCS,
    ra: Option<RA>,
    color_registers: BTreeMap<u16, SixelColor>,
    current_color: u16,
    sixel_cursor_y: usize,
    sixel_cursor_x: usize,
    pixels: Vec<Vec<Pixel>>,
    max_height: Option<usize>,
    stop_parsing: bool,
    got_dcs: bool,
}

impl SixelDeserializer {
    pub fn new() -> Self {
        SixelDeserializer {
            dcs: DCS {
                macro_parameter: 0,
                transparent_bg: false,
            },
            ra: None,
            color_registers: BTreeMap::new(),
            current_color: 0, // this is totally undefined behaviour and seems like a free for all in general
            sixel_cursor_y: 0,
            sixel_cursor_x: 0,
            pixels: vec![vec![]], // start with one empty line
            max_height: None,
            stop_parsing: false,
            got_dcs: false,
        }
    }
    /// Provide a `max_height` value in pixels, all pixels beyond this max height will not be
    /// parsed
    pub fn max_height(mut self, max_height: usize) -> Self {
        self.max_height = Some(max_height);
        self
    }
    /// Create a new [`SixelImage`] out of the existing state and consume it.
    pub fn create_image(&mut self) -> Result<SixelImage, &'static str> {
        if !self.got_dcs {
            return Err("Corrupted image sequence");
        }
        let dcs = std::mem::take(&mut self.dcs);
        let ra = std::mem::take(&mut self.ra);
        let pixels = std::mem::take(&mut self.pixels);
        let color_registers = std::mem::take(&mut self.color_registers);
        Ok(SixelImage {
            dcs,
            ra,
            pixels,
            color_registers,
        })
    }
    /// Handle a [`SixelEvent`], changing the internal state to match
    pub fn handle_event(&mut self, event: SixelEvent) -> Result<(), &'static str> {
        if !self.got_dcs && !matches!(event, SixelEvent::Dcs { .. }) {
            return Err("Corrupted image sequence");
        }
        if self.stop_parsing {
            return Ok(());
        }
        match event {
            SixelEvent::ColorIntroducer {
                color_coordinate_system,
                color_number,
            } => {
                match color_coordinate_system {
                    Some(color_coordinate_system) => {
                        // define a color in a register
                        let color = SixelColor::from(color_coordinate_system);
                        self.color_registers.insert(color_number, color);
                    }
                    None => {
                        // switch to register number
                        self.current_color = color_number;
                    }
                }
            }
            SixelEvent::RasterAttribute { pan, pad, ph, pv } => {
                if !self.dcs.transparent_bg {
                    if let Some(pv) = pv {
                        self.pad_lines_vertically(pv);
                    }
                    if let Some(ph) = ph {
                        self.pad_lines_horizontally(ph);
                    }
                }
                self.ra = Some(RA { pan, pad, ph, pv });
            }
            SixelEvent::Data { byte } => {
                self.make_sure_six_lines_exist_after_cursor();
                self.add_sixel_byte(byte, 1);
                self.sixel_cursor_x += 1;
            }
            SixelEvent::Repeat {
                repeat_count,
                byte_to_repeat,
            } => {
                self.make_sure_six_lines_exist_after_cursor();
                self.add_sixel_byte(byte_to_repeat, repeat_count);
                self.sixel_cursor_x += repeat_count;
            }
            SixelEvent::Dcs {
                macro_parameter,
                transparent_background,
                horizontal_pixel_distance: _,
            } => {
                self.got_dcs = true;
                if let Some(mp) = macro_parameter {
                    self.dcs.macro_parameter = mp;
                }
                if transparent_background == Some(1) {
                    self.dcs.transparent_bg = true;
                }
            }
            SixelEvent::GotoBeginningOfLine => {
                self.sixel_cursor_x = 0;
            }
            SixelEvent::GotoNextLine => {
                if let Some(max_height) = self.max_height {
                    if self.sixel_cursor_y + 12 > max_height {
                        // 12 because we move the cursor to the top of the sixel column and need to count 6 more down to make sure we don't exceed
                        self.stop_parsing = true;
                        return Ok(());
                    }
                }
                self.sixel_cursor_y += 6;
                self.sixel_cursor_x = 0;
            }
            SixelEvent::UnknownSequence(_) => {
                return Err("Corrupted Sixel sequence");
            }
            SixelEvent::End => {}
        }
        Ok(())
    }
    fn make_sure_six_lines_exist_after_cursor(&mut self) {
        let lines_to_add = (self.sixel_cursor_y + 6).saturating_sub(self.pixels.len());
        for _ in 0..lines_to_add {
            self.pixels.push(vec![]);
        }
    }
    fn add_sixel_byte(&mut self, byte: u8, repeat_count: usize) {
        let mut pixel_line_index_in_sixel = 0;
        for bit in SixelPixelIterator::new(byte.saturating_sub(63)) {
            let current_line = self
                .pixels
                .get_mut(self.sixel_cursor_y + pixel_line_index_in_sixel)
                .unwrap();
            let new_pixel = Pixel {
                on: bit,
                color: self.current_color,
            };
            for i in 0..repeat_count {
                match current_line.get_mut(self.sixel_cursor_x + i) {
                    Some(pixel_in_current_position) if bit => {
                        let _ = std::mem::replace(pixel_in_current_position, new_pixel);
                    }
                    None => {
                        current_line.push(new_pixel);
                    }
                    _ => {} // bit is off and pixel already exists, so noop
                }
            }
            pixel_line_index_in_sixel += 1;
        }
    }
    fn pad_lines_vertically(&mut self, pad_until: usize) {
        let empty_pixel = Pixel {
            on: true,
            color: self.current_color,
        };
        if self.pixels.len() < pad_until {
            let empty_line = vec![empty_pixel; pad_until];
            let lines_to_pad = pad_until - self.pixels.len();
            let line_padding = iter::repeat(empty_line).take(lines_to_pad);
            self.pixels.extend(line_padding);
        }
    }
    fn pad_lines_horizontally(&mut self, pad_until: usize) {
        let empty_pixel = Pixel {
            on: true,
            color: self.current_color,
        };
        for pixel_line in self.pixels.iter_mut() {
            if pixel_line.len() < pad_until {
                let pixel_count_to_pad = pad_until - pixel_line.len();
                let pixel_padding = iter::repeat(empty_pixel).take(pixel_count_to_pad);
                pixel_line.extend(pixel_padding);
            }
        }
    }
}

#[derive(Debug, Clone, Copy)]
struct SixelPixelIterator {
    sixel_byte: u8,
    current_mask: u8,
}
impl SixelPixelIterator {
    pub fn new(sixel_byte: u8) -> Self {
        SixelPixelIterator {
            sixel_byte,
            current_mask: 1,
        }
    }
}
impl Iterator for SixelPixelIterator {
    type Item = bool;
    fn next(&mut self) -> Option<Self::Item> {
        // iterate through the bits in a byte from right (least significant) to left (most
        // significant), eg. 89 => 1011001 => true, false, false, true, true, false, true
        let bit = self.sixel_byte & self.current_mask == self.current_mask;
        self.current_mask <<= 1;
        if self.current_mask == 128 {
            None
        } else {
            Some(bit)
        }
    }
}