use crate::Exceptions;
use crate::common::Result;
use super::SymbolInfo;
const FACTOR_SIZE: usize = 16;
const FACTOR_SETS: [u32; FACTOR_SIZE] =
[5, 7, 10, 11, 12, 14, 18, 20, 24, 28, 36, 42, 48, 56, 62, 68];
const FACTORS: [&[u32]; FACTOR_SIZE] = [
&[228, 48, 15, 111, 62],
&[23, 68, 144, 134, 240, 92, 254],
&[28, 24, 185, 166, 223, 248, 116, 255, 110, 61],
&[175, 138, 205, 12, 194, 168, 39, 245, 60, 97, 120],
&[41, 153, 158, 91, 61, 42, 142, 213, 97, 178, 100, 242],
&[
156, 97, 192, 252, 95, 9, 157, 119, 138, 45, 18, 186, 83, 185,
],
&[
83, 195, 100, 39, 188, 75, 66, 61, 241, 213, 109, 129, 94, 254, 225, 48, 90, 188,
],
&[
15, 195, 244, 9, 233, 71, 168, 2, 188, 160, 153, 145, 253, 79, 108, 82, 27, 174, 186, 172,
],
&[
52, 190, 88, 205, 109, 39, 176, 21, 155, 197, 251, 223, 155, 21, 5, 172, 254, 124, 12, 181,
184, 96, 50, 193,
],
&[
211, 231, 43, 97, 71, 96, 103, 174, 37, 151, 170, 53, 75, 34, 249, 121, 17, 138, 110, 213,
141, 136, 120, 151, 233, 168, 93, 255,
],
&[
245, 127, 242, 218, 130, 250, 162, 181, 102, 120, 84, 179, 220, 251, 80, 182, 229, 18, 2,
4, 68, 33, 101, 137, 95, 119, 115, 44, 175, 184, 59, 25, 225, 98, 81, 112,
],
&[
77, 193, 137, 31, 19, 38, 22, 153, 247, 105, 122, 2, 245, 133, 242, 8, 175, 95, 100, 9,
167, 105, 214, 111, 57, 121, 21, 1, 253, 57, 54, 101, 248, 202, 69, 50, 150, 177, 226, 5,
9, 5,
],
&[
245, 132, 172, 223, 96, 32, 117, 22, 238, 133, 238, 231, 205, 188, 237, 87, 191, 106, 16,
147, 118, 23, 37, 90, 170, 205, 131, 88, 120, 100, 66, 138, 186, 240, 82, 44, 176, 87, 187,
147, 160, 175, 69, 213, 92, 253, 225, 19,
],
&[
175, 9, 223, 238, 12, 17, 220, 208, 100, 29, 175, 170, 230, 192, 215, 235, 150, 159, 36,
223, 38, 200, 132, 54, 228, 146, 218, 234, 117, 203, 29, 232, 144, 238, 22, 150, 201, 117,
62, 207, 164, 13, 137, 245, 127, 67, 247, 28, 155, 43, 203, 107, 233, 53, 143, 46,
],
&[
242, 93, 169, 50, 144, 210, 39, 118, 202, 188, 201, 189, 143, 108, 196, 37, 185, 112, 134,
230, 245, 63, 197, 190, 250, 106, 185, 221, 175, 64, 114, 71, 161, 44, 147, 6, 27, 218, 51,
63, 87, 10, 40, 130, 188, 17, 163, 31, 176, 170, 4, 107, 232, 7, 94, 166, 224, 124, 86, 47,
11, 204,
],
&[
220, 228, 173, 89, 251, 149, 159, 56, 89, 33, 147, 244, 154, 36, 73, 127, 213, 136, 248,
180, 234, 197, 158, 177, 68, 122, 93, 213, 15, 160, 227, 236, 66, 139, 153, 185, 202, 167,
179, 25, 220, 232, 96, 210, 231, 136, 223, 239, 181, 241, 59, 52, 172, 25, 49, 232, 211,
189, 64, 54, 108, 153, 132, 63, 96, 103, 82, 186,
],
];
const MODULO_VALUE: usize = 0x12D;
const LOG: [u32; 256] = {
let mut log_array = [0u32; 256];
let mut p = 1;
let mut i = 0;
while i < 255 {
log_array[p] = i;
p *= 2;
if p >= 256 {
p ^= MODULO_VALUE;
}
i += 1;
}
log_array
};
const ALOG: [u32; 255] = {
let mut alog_array = [0u32; 255];
let mut p = 1_usize;
let mut i = 0;
while i < 255 {
alog_array[i] = p as u32;
p *= 2;
if p >= 256 {
p ^= MODULO_VALUE;
}
i += 1;
}
alog_array
};
pub fn encodeECC200(codewords: &str, symbolInfo: &SymbolInfo) -> Result<String> {
let codewords: Vec<u8> = codewords.chars().map(|c| c as u8).collect();
if codewords.len() != symbolInfo.getDataCapacity() as usize {
return Err(Exceptions::illegal_argument_with(
"The number of codewords does not match the selected symbol",
));
}
let mut sb = vec![0; (symbolInfo.getDataCapacity() + symbolInfo.getErrorCodewords()) as usize];
sb[..codewords.len()].copy_from_slice(&codewords);
let blockCount = symbolInfo.getInterleavedBlockCount() as usize;
if blockCount == 1 {
let ecc = createECCBlock(&codewords, symbolInfo.getErrorCodewords() as usize)?;
sb[codewords.len()..ecc.len() + codewords.len()].copy_from_slice(ecc.as_slice());
} else {
let mut dataSizes = vec![0u32; blockCount];
let mut errorSizes = vec![0u32; blockCount];
for i in 0..blockCount {
dataSizes[i] = symbolInfo.getDataLengthForInterleavedBlock(i as u32 + 1) as u32;
errorSizes[i] = symbolInfo.getErrorLengthForInterleavedBlock(i as u32 + 1);
}
for block in 0..blockCount {
let mut temp = Vec::with_capacity(dataSizes[block] as usize);
let mut d = block;
while d < symbolInfo.getDataCapacity() as usize {
temp.push(codewords[d]);
d += blockCount;
}
let ecc = createECCBlock(&temp, errorSizes[block] as usize)?;
let mut pos = 0;
let mut e = block;
while e < errorSizes[block] as usize * blockCount {
sb[symbolInfo.getDataCapacity() as usize + e] = ecc[pos];
pos += 1;
e += blockCount;
}
}
}
Ok(sb.into_iter().map(|c| c as char).collect())
}
fn createECCBlock(codewords: &[u8], numECWords: usize) -> Result<Vec<u8>> {
let table = FACTOR_SETS
.iter()
.position(|&set| set == numECWords as u32)
.ok_or_else(|| {
Exceptions::illegal_argument_with(format!(
"Illegal number of error correction codewords specified: {numECWords}"
))
})?;
let poly = FACTORS[table];
let mut ecc = vec![0u8; numECWords];
for codeword in codewords {
let m = ecc[numECWords - 1] as usize ^ *codeword as usize;
for k in (1..numECWords).rev() {
if m != 0 && poly[k] != 0 {
ecc[k] = ecc[k - 1] ^ ALOG[(LOG[m] + LOG[poly[k] as usize]) as usize % 255] as u8;
} else {
ecc[k] = ecc[k - 1];
}
}
if m != 0 && poly[0] != 0 {
ecc[0] = ALOG[(LOG[m] + LOG[poly[0] as usize]) as usize % 255] as u8;
} else {
ecc[0] = 0;
}
}
Ok(ecc.into_iter().rev().collect())
}
#[cfg(test)]
mod test_case {
use crate::datamatrix::encoder::{SymbolInfoLookup, high_level_encode_test_case::visualize};
use super::encodeECC200;
#[test]
fn testRS() {
let cw: [char; 3] = [142 as char, 164 as char, 186 as char];
let symbolInfo = SymbolInfoLookup::new()
.lookup(3)
.expect("find lookup")
.unwrap(); let s = encodeECC200(&String::from_iter(cw), symbolInfo).expect("encode");
assert_eq!("142 164 186 114 25 5 88 102", visualize(&s));
let cw: [char; 3] = [66 as char, 129 as char, 70 as char];
let s = encodeECC200(&String::from_iter(cw), symbolInfo).expect("encode");
assert_eq!("66 129 70 138 234 82 82 95", visualize(&s));
}
}