summaryrefslogtreecommitdiff
path: root/src/label_formats/label_format.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/label_formats/label_format.rs')
-rw-r--r--src/label_formats/label_format.rs227
1 files changed, 0 insertions, 227 deletions
diff --git a/src/label_formats/label_format.rs b/src/label_formats/label_format.rs
deleted file mode 100644
index afbb2ad..0000000
--- a/src/label_formats/label_format.rs
+++ /dev/null
@@ -1,227 +0,0 @@
-pub struct LabelFormat {
- pub buffer: Vec<u16>,
- pub width: usize,
- pub height: usize,
-}
-
-impl LabelFormat {
- pub fn get_left(&self, x: usize, y: usize) -> Option<u16> {
- if x > 0 {
- return Some(self.buffer[(x-1) + y * self.width]);
- }
- return None;
- }
-
- pub fn get_right(&self, x: usize, y: usize) -> Option<u16> {
- if (x+1) < self.width {
- return Some(self.buffer[(x+1) + y * self.width]);
- }
- return None;
- }
-
- pub fn get_up(&self, x: usize, y: usize) -> Option<u16> {
- if y > 0 {
- return Some(self.buffer[x + (y-1) * self.width]);
- }
- return None;
- }
-
- pub fn get_down(&self, x: usize, y: usize) -> Option<u16> {
- if (y+1) < self.height {
- return Some(self.buffer[x + (y+1) * self.width]);
- }
- return None;
- }
-
- pub fn display(&self, include_space: bool) {
- for y in 0..self.height {
- for x in 0..self.width {
- print!("{:04X}", self.buffer[x + y*self.width]);
- if include_space {
- print!(" ");
- }
- }
- println!();
- }
- }
-
- pub fn idilate(&mut self) {
- let dilated = self.dilate();
- self.buffer = dilated.buffer;
- }
-
- pub fn dilate(&self) -> LabelFormat {
- let width = self.width;
- let height = self.height;
- let mut output_buffer = vec![0u16; width*height];
- for y in 0..height {
- for x in 0..width {
- let index = x + y * width;
- let current_color = self.buffer[index];
- if current_color != 0 {
- output_buffer[index] = current_color;
- continue;
- }
- if let Some(color) = self.get_up(x,y) {
- if color != 0 {
- output_buffer[index] = color;
- continue;
- }
- }
- if let Some(color) = self.get_right(x,y) {
- if color != 0 {
- output_buffer[index] = color;
- continue;
- }
- }
- if let Some(color) = self.get_down(x,y) {
- if color != 0 {
- output_buffer[index] = color;
- continue;
- }
- }
- if let Some(color) = self.get_left(x,y) {
- if color != 0 {
- output_buffer[index] = color;
- continue;
- }
- }
- }
- }
- LabelFormat {
- buffer: output_buffer,
- width,
- height,
- }
- }
-
- pub fn ierode(&mut self) {
- let eroded = self.erode();
- self.buffer = eroded.buffer;
- }
-
- pub fn erode(&self) -> LabelFormat {
- let width = self.width;
- let height = self.height;
- let mut output_buffer = vec![0u16; width*height];
- for y in 0..height {
- for x in 0..width {
- let index = x + y * width;
- let current_color = self.buffer[index];
- if current_color == 0 {
- continue;
- }
- if let Some(color) = self.get_up(x,y) {
- if color == 0 {
- continue;
- }
- }
- if let Some(color) = self.get_right(x,y) {
- if color == 0 {
- continue;
- }
- }
- if let Some(color) = self.get_down(x,y) {
- if color == 0 {
- continue;
- }
- }
- if let Some(color) = self.get_left(x,y) {
- if color == 0 {
- continue;
- }
- }
- output_buffer[index] = current_color;
- }
- }
- LabelFormat {
- buffer: output_buffer,
- width,
- height,
- }
- }
-
- pub fn icloseup(&mut self, n: usize) {
- for _ in 0..n {
- self.ierode();
- }
- for _ in 0..n {
- self.idilate();
- }
- }
-
- pub fn closeup(&self, n: usize) -> LabelFormat {
- let mut x = self.erode();
- x.icloseup(n-1);
- x.idilate();
- x
- }
-
- pub fn ifill(&mut self, n: usize) {
- for _ in 0..n {
- self.idilate();
- }
- for _ in 0..n {
- self.ierode();
- }
- }
-
- pub fn fill(&self, n: usize) -> LabelFormat {
- let mut x = self.dilate();
- x.icloseup(n-1);
- x.ierode();
- x
- }
-
- pub fn dump(&self, filename: &str) -> Result<(), std::io::Error> {
- crate::binfile::dump_u16_vec(filename, self.buffer.clone())
- }
-}
-
-#[cfg(test)]
-mod tests {
- use super::*;
-
- #[test]
- fn label_morphology_test() {
- const DIM: usize = 6;
- let mut test_data = LabelFormat {
- buffer: vec![0u16; DIM*DIM],
- width: DIM,
- height: DIM,
- };
-
- test_data.buffer[2+3*DIM] = 1;
- test_data.buffer[3+3*DIM] = 1;
- test_data.buffer[4+3*DIM] = 2;
- test_data.display(true);
- println!();
- let dilated = test_data.dilate();
- dilated.display(true);
- println!();
- let restored = dilated.erode();
- restored.display(true);
- }
-
- #[test]
- fn mutable_morphology_test() {
- const DIM: usize = 6;
- let mut test_data = LabelFormat {
- buffer: vec![0u16; DIM*DIM],
- width: DIM,
- height: DIM,
- };
-
- test_data.buffer[2+3*DIM] = 1;
- test_data.buffer[3+3*DIM] = 1;
- test_data.buffer[4+3*DIM] = 2;
- test_data.display(true);
- println!();
- test_data.idilate();
- test_data.display(true);
- println!();
- test_data.ierode();
- test_data.display(true);
- println!();
- }
-}