Source Release

webapp/src/canvas.rs

@@ -0,0 +1,60 @@
+use wasm_bindgen::{JsCast};
+use web_sys::{console, CanvasRenderingContext2d, HtmlCanvasElement};
+use visioncortex::{ColorImage};
+
+use super::common::document;
+
+pub struct Canvas {
+    html_canvas: HtmlCanvasElement,
+    cctx: CanvasRenderingContext2d,
+}
+
+impl Canvas {
+    pub fn new_from_id(canvas_id: &str) -> Canvas {
+        let html_canvas = document().get_element_by_id(canvas_id).unwrap();
+        let html_canvas: HtmlCanvasElement = html_canvas
+            .dyn_into::<HtmlCanvasElement>()
+            .map_err(|_| ())
+            .unwrap();
+
+        let cctx = html_canvas
+            .get_context("2d")
+            .unwrap()
+            .unwrap()
+            .dyn_into::<CanvasRenderingContext2d>()
+            .unwrap();
+
+        Canvas {
+            html_canvas,
+            cctx,
+        }
+    }
+
+    pub fn width(&self) -> usize {
+        self.html_canvas.width() as usize
+    }
+
+    pub fn height(&self) -> usize {
+        self.html_canvas.height() as usize
+    }
+
+    pub fn get_image_data(&self, x: u32, y: u32, width: u32, height: u32) -> Vec<u8> {
+        let image = self
+            .cctx
+            .get_image_data(x as f64, y as f64, width as f64, height as f64)
+            .unwrap();
+        image.data().to_vec()
+    }
+
+    pub fn get_image_data_as_image(&self, x: u32, y: u32, width: u32, height: u32) -> ColorImage {
+        ColorImage {
+            pixels: self.get_image_data(x, y, width, height),
+            width: width as usize,
+            height: height as usize,
+        }
+    }
+
+    pub fn log(&self, string: &str) {
+        console::log_1(&wasm_bindgen::JsValue::from_str(string));
+    }
+}