rxing/src/qrcode/encoder/byte_matrix.rs

/*
 * Copyright 2008 ZXing authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

use std::fmt;

use crate::common::BitMatrix;

/**
 * JAVAPORT: The original code was a 2D array of ints, but since it only ever gets assigned
 * -1, 0, and 1, I'm going to use less memory and go with bytes.
 *
 * @author dswitkin@google.com (Daniel Switkin)
 */
#[derive(Debug, PartialEq, Eq, Clone)]
pub struct ByteMatrix {
    bytes: Vec<Vec<u8>>,
    width: u32,
    height: u32,
}

impl ByteMatrix {
    pub fn new(width: u32, height: u32) -> Self {
        Self {
            bytes: vec![vec![0u8; width as usize]; height as usize],
            width,
            height,
        }
        // bytes = new byte[height][width];
    }

    pub fn getHeight(&self) -> u32 {
        self.height
    }

    pub fn getWidth(&self) -> u32 {
        self.width
    }

    pub fn get(&self, x: u32, y: u32) -> u8 {
        self.bytes[y as usize][x as usize]
    }

    /**
     * @return an internal representation as bytes, in row-major order. array[y][x] represents point (x,y)
     */
    pub fn getArray(&self) -> &Vec<Vec<u8>> {
        &self.bytes
    }

    pub fn set(&mut self, x: u32, y: u32, value: u8) {
        self.bytes[y as usize][x as usize] = value;
    }

    // pub fn set(int x, int y, int value) {
    //   bytes[y][x] = (byte) value;
    // }

    pub fn set_bool(&mut self, x: u32, y: u32, value: bool) {
        self.bytes[y as usize][x as usize] = u8::from(value); //if value { 1 } else { 0 };
    }

    pub fn clear(&mut self, value: u8) {
        for row in self.bytes.iter_mut() {
            *row = vec![value; row.len()];
        }
        // for (byte[] aByte : bytes) {
        //   Arrays.fill(aByte, value);
        // }
    }
}

impl fmt::Display for ByteMatrix {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        let mut result = String::with_capacity(2 * self.width as usize * self.height as usize + 2);
        for y in 0..self.height as usize {
            // for (int y = 0; y < height; ++y) {
            let bytesY = &self.bytes[y];
            for byte in bytesY.iter().take(self.width as usize) {
                // for x in 0..self.width as usize {
                // for (int x = 0; x < width; ++x) {
                match *byte {
                    0 => result.push_str(" 0"),
                    1 => result.push_str(" 1"),
                    _ => result.push_str("  "),
                };
                // switch (bytesY[x]) {
                //   case 0:
                //     result.append(" 0");
                //     break;
                //   case 1:
                //     result.append(" 1");
                //     break;
                //   default:
                //     result.append("  ");
                //     break;
                // }
            }
            result.push('\n');
        }
        write!(f, "{}", result)
    }
}

impl From<ByteMatrix> for BitMatrix {
    fn from(bm: ByteMatrix) -> Self {
        let mut bit_matrix = BitMatrix::new(bm.getWidth(), bm.getHeight()).unwrap();
        for y in 0..bm.getHeight() {
            for x in 0..bm.getWidth() {
                if bm.get(x, y) > 0 {
                    bit_matrix.set(x, y);
                }
            }
        }
        bit_matrix
    }
}