vtracer/webapp/src/conversion/binary_image.rs

use wasm_bindgen::prelude::*;
use visioncortex::{clusters::Clusters, Color, ColorName, PointI32, PathSimplifyMode};

use crate::{canvas::*};
use crate::svg::*;

use serde::Deserialize;
use super::util;

#[derive(Debug, Deserialize)]
pub struct BinaryImageConverterParams {
    pub canvas_id: String,
    pub svg_id: String,
    pub mode: String,
    pub corner_threshold: f64,
    pub length_threshold: f64,
    pub max_iterations: usize,
    pub splice_threshold: f64,
    pub filter_speckle: usize,
}

#[wasm_bindgen]
pub struct BinaryImageConverter {
    canvas: Canvas,
    svg: Svg,
    clusters: Clusters,
    counter: usize,
    mode: PathSimplifyMode,
    params: BinaryImageConverterParams,
}

impl BinaryImageConverter {
    pub fn new(params: BinaryImageConverterParams) -> Self {
        let canvas = Canvas::new_from_id(&params.canvas_id);
        let svg = Svg::new_from_id(&params.svg_id);
        Self {
            canvas,
            svg,
            clusters: Clusters::default(),
            counter: 0,
            mode: util::path_simplify_mode(&params.mode),
            params,
        }
    }
}

#[wasm_bindgen]
impl BinaryImageConverter {
    pub fn new_with_string(params: String) -> Self {
        let params: BinaryImageConverterParams = serde_json::from_str(params.as_str()).unwrap();
        Self::new(params)
    }

    pub fn init(&mut self) {
        let width = self.canvas.width() as u32;
        let height = self.canvas.height() as u32;
        let image = self.canvas.get_image_data_as_image(0, 0, width, height);
        let binary_image = image.to_binary_image(|x| x.r < 128);
        self.clusters = binary_image.to_clusters(false);
        self.canvas.log(&format!(
            "clusters.len() = {}, self.clusters.rect.left = {}",
            self.clusters.len(),
            self.clusters.rect.left
        ));
    }

    pub fn tick(&mut self) -> bool {
        if self.counter < self.clusters.len() {
            self.canvas.log(&format!("tick {}", self.counter));
            let cluster = self.clusters.get_cluster(self.counter);
            if cluster.size() >= self.params.filter_speckle {
                let svg_path = cluster.to_svg_path(
                    self.mode,
                    self.params.corner_threshold,
                    self.params.length_threshold,
                    self.params.max_iterations,
                    self.params.splice_threshold
                );
                let color = Color::color(&ColorName::White);
                self.svg.prepend_path_with_fill(
                    &svg_path,
                    &PointI32::default(),
                    &color,
                );
            }
            self.counter += 1;
            false
        } else {
            self.canvas.log("done");
            true
        }
    }

    pub fn progress(&self) -> u32 {
        100 * self.counter as u32 / self.clusters.len() as u32
    }
}