use crate::pdf417::pdf_417_common;
const RATIOS_TABLE: [[f32; pdf_417_common::BARS_IN_MODULE as usize]; 2787] = {
let mut table =
[[0.0; pdf_417_common::BARS_IN_MODULE as usize]; pdf_417_common::SYMBOL_TABLE.len()];
let mut i = 0_usize;
while i < pdf_417_common::SYMBOL_TABLE.len() {
let mut currentSymbol = pdf_417_common::SYMBOL_TABLE[i];
let mut currentBit = currentSymbol & 0x1;
let mut j = 0_usize;
while j < pdf_417_common::BARS_IN_MODULE as usize {
let mut size = 0.0;
while (currentSymbol & 0x1) == currentBit {
size += 1.0;
currentSymbol >>= 1;
}
currentBit = currentSymbol & 0x1;
table[i][pdf_417_common::BARS_IN_MODULE as usize - j - 1] =
size / pdf_417_common::MODULES_IN_CODEWORD as f32;
j += 1;
}
i += 1;
}
table
};
pub fn getDecodedValue(moduleBitCount: &[u32]) -> u32 {
let decodedValue = getDecodedCodewordValue(&sampleBitCounts(moduleBitCount));
if decodedValue != -1 {
return decodedValue as u32;
}
getClosestDecodedValue(moduleBitCount) as u32
}
fn sampleBitCounts(moduleBitCount: &[u32]) -> [u32; 8] {
let bitCountSum: u32 = moduleBitCount.iter().sum(); let mut result = [0; pdf_417_common::BARS_IN_MODULE as usize];
let mut bitCountIndex = 0;
let mut sumPreviousBits = 0;
for i in 0..pdf_417_common::MODULES_IN_CODEWORD {
let sampleIndex: f32 = bitCountSum as f32
/ (2.0 * pdf_417_common::MODULES_IN_CODEWORD as f32)
+ (i as f32 * bitCountSum as f32) / pdf_417_common::MODULES_IN_CODEWORD as f32;
if sumPreviousBits as f32 + moduleBitCount[bitCountIndex] as f32 <= sampleIndex {
sumPreviousBits += moduleBitCount[bitCountIndex];
bitCountIndex += 1;
}
result[bitCountIndex] += 1;
}
result
}
fn getDecodedCodewordValue(moduleBitCount: &[u32]) -> i32 {
let decodedValue = getBitValue(moduleBitCount);
if pdf_417_common::getCodeword(decodedValue as u32) == -1 {
-1
} else {
decodedValue
}
}
fn getBitValue(moduleBitCount: &[u32]) -> i32 {
let mut result: u64 = 0;
for (i, mbc) in moduleBitCount.iter().enumerate() {
for _bit in 0..(*mbc) {
result = (result << 1) | u64::from(i % 2 == 0); }
}
result as i32
}
fn getClosestDecodedValue(moduleBitCount: &[u32]) -> i32 {
let bitCountSum: u32 = moduleBitCount.iter().sum(); let mut bitCountRatios = [0.0; pdf_417_common::BARS_IN_MODULE as usize];
if bitCountSum > 1 {
for i in 0..bitCountRatios.len() {
bitCountRatios[i] = moduleBitCount[i] as f32 / bitCountSum as f32;
}
}
let mut bestMatchError = f32::MAX;
let mut bestMatch = -1_i32;
for (j, ratioTableRow) in RATIOS_TABLE.iter().enumerate() {
let mut error = 0.0;
for k in 0..pdf_417_common::BARS_IN_MODULE as usize {
let diff = ratioTableRow[k] - bitCountRatios[k];
error += diff * diff;
if error >= bestMatchError {
break;
}
}
if error < bestMatchError {
bestMatchError = error;
bestMatch = pdf_417_common::SYMBOL_TABLE[j] as i32;
}
}
bestMatch
}
#[cfg(test)]
mod test {
use crate::pdf417::decoder::pdf_417_codeword_decoder::getDecodedValue;
#[test]
fn test_cw_227() {
let sample_data = [2, 2, 3, 1, 6, 4, 3, 4];
assert_ne!(getDecodedValue(&sample_data), 110360);
assert_eq!(getDecodedValue(&sample_data), 93980);
}
#[test]
fn test_2() {
let sample = [2, 1, 4, 2, 5, 3, 7, 2];
assert_ne!(95134, getDecodedValue(&sample));
}
#[test]
fn test_128268() {
let sample = [7, 2, 1, 2, 2, 5, 3, 3];
assert_eq!(128268, getDecodedValue(&sample));
}
#[test]
fn test_125304() {
let sample = [6, 1, 2, 2, 2, 2, 6, 4];
assert_eq!(125304, getDecodedValue(&sample));
}
#[test]
fn test_125216() {
let sample = [6, 1, 2, 2, 2, 3, 2, 7];
assert_eq!(125216, getDecodedValue(&sample));
}
#[test]
fn test_83768() {
let sample = [1, 2, 1, 4, 5, 3, 5, 4];
assert_eq!(83768, getDecodedValue(&sample));
}
#[test]
fn test_96060() {
let sample = [2, 1, 4, 2, 5, 3, 7, 2];
assert_eq!(96060, getDecodedValue(&sample));
}
#[test]
fn test_97372() {
let sample = [3, 1, 7, 4, 2, 2, 4, 2];
assert_eq!(97372, getDecodedValue(&sample));
}
}