diff --git a/core/src/decoder/bch.rs b/core/src/decoder/bch.rs index 4465127..c302ef4 100644 --- a/core/src/decoder/bch.rs +++ b/core/src/decoder/bch.rs @@ -118,7 +118,7 @@ mod tests { result.is_some(), "decode failed: ec_bits={ec_bits}, mask={mask}, encoded={encoded:#05X}" ); - let (dec_ec, dec_mask) = result.unwrap(); + let (dec_ec, dec_mask, _d) = result.unwrap(); assert_eq!(ec_bits, dec_ec, "ec_bits mismatch"); assert_eq!(mask, dec_mask, "mask mismatch"); } @@ -134,7 +134,8 @@ mod tests { result.is_some(), "decode failed: ver={ver}, encoded={encoded:#010X}" ); - assert_eq!(ver, result.unwrap(), "version mismatch"); + let (dec_ver, _d) = result.unwrap(); + assert_eq!(ver, dec_ver, "version mismatch"); } } @@ -143,12 +144,11 @@ mod tests { for ec_bits in 0u8..4 { for mask in 0u8..8 { let original = encode_format_info(ec_bits, mask); - // 翻转每个比特,验证能纠错 for bit in 0..15 { let corrupted = original ^ (1 << bit); let result = decode_format_info(corrupted); assert!(result.is_some(), "1-bit error not corrected at bit {bit}"); - let (dec_ec, dec_mask) = result.unwrap(); + let (dec_ec, dec_mask, _d) = result.unwrap(); assert_eq!(ec_bits, dec_ec); assert_eq!(mask, dec_mask); } @@ -164,7 +164,8 @@ mod tests { let corrupted = original ^ (1 << bit); let result = decode_version_info(corrupted); assert!(result.is_some(), "1-bit error not corrected at bit {bit}"); - assert_eq!(ver, result.unwrap()); + let (dec_ver, _d) = result.unwrap(); + assert_eq!(ver, dec_ver); } } } diff --git a/core/src/decoder/detect.rs b/core/src/decoder/detect.rs index 7fbc57a..657cf56 100644 --- a/core/src/decoder/detect.rs +++ b/core/src/decoder/detect.rs @@ -21,7 +21,7 @@ pub(crate) struct DetectResult { } /// 水平扫描查找 1:1:3:1:1 比例 -fn scan_row(gray: &[Vec], row: usize) -> Vec<(usize, usize)> { +fn scan_row(gray: &[Vec], row: usize) -> Vec<(usize, usize, usize)> { // (列号,运行长度) let mut runs: Vec<(usize, usize)> = Vec::new(); let width = if gray.is_empty() { 0 } else { gray[0].len() }; @@ -37,8 +37,8 @@ fn scan_row(gray: &[Vec], row: usize) -> Vec<(usize, usize)> { runs.push((col - run_len, run_len)); } - // 找 5 连段符合 1:1:3:1:1 比例 - let mut centers: Vec<(usize, usize)> = Vec::new(); + // 找 5 连段符合 1:1:3:1:1 比例 — 返回 (cx, cy, total_size_px) + let mut centers: Vec<(usize, usize, usize)> = Vec::new(); for i in 0..runs.len().saturating_sub(4) { let r0 = runs[i].1 as f32; let r1 = runs[i + 1].1 as f32; @@ -46,18 +46,17 @@ fn scan_row(gray: &[Vec], row: usize) -> Vec<(usize, usize)> { let r3 = runs[i + 3].1 as f32; let r4 = runs[i + 4].1 as f32; - let avg = (r0 + r1 + r2 + r3 + r4) / 5.0; - if avg < 2.0 { + let total = r0 + r1 + r2 + r3 + r4; + let base = total / 7.0; + if base < 2.0 { continue; } - - // 检查比例容差 ±40% let tolerance = 0.4; - let check = |v: f32, expected: f32| (v - expected * avg).abs() < avg * tolerance; + let check = |v: f32, expected: f32| (v - expected * base).abs() < base * tolerance; if check(r0, 1.0) && check(r1, 1.0) && check(r2, 3.0) && check(r3, 1.0) && check(r4, 1.0) { let cx = runs[i + 2].0 + runs[i + 2].1 / 2; - centers.push((cx, row)); + centers.push((cx, row, total as usize)); } } @@ -65,7 +64,7 @@ fn scan_row(gray: &[Vec], row: usize) -> Vec<(usize, usize)> { } /// 垂直扫描查找 1:1:3:1:1 比例 -fn scan_col(gray: &[Vec], col: usize) -> Vec<(usize, usize)> { +fn scan_col(gray: &[Vec], col: usize) -> Vec<(usize, usize, usize)> { let height = gray.len(); let mut runs: Vec<(usize, usize)> = Vec::new(); @@ -80,7 +79,7 @@ fn scan_col(gray: &[Vec], col: usize) -> Vec<(usize, usize)> { runs.push((row - run_len, run_len)); } - let mut centers: Vec<(usize, usize)> = Vec::new(); + let mut centers: Vec<(usize, usize, usize)> = Vec::new(); for i in 0..runs.len().saturating_sub(4) { let r0 = runs[i].1 as f32; let r1 = runs[i + 1].1 as f32; @@ -88,17 +87,19 @@ fn scan_col(gray: &[Vec], col: usize) -> Vec<(usize, usize)> { let r3 = runs[i + 3].1 as f32; let r4 = runs[i + 4].1 as f32; - let avg = (r0 + r1 + r2 + r3 + r4) / 5.0; - if avg < 2.0 { + let total = r0 + r1 + r2 + r3 + r4; + let base = total / 7.0; + if base < 2.0 { continue; } let tolerance = 0.4; - let check = |v: f32, expected: f32| (v - expected * avg).abs() < avg * tolerance; + let check = |v: f32, expected: f32| (v - expected * base).abs() < base * tolerance; if check(r0, 1.0) && check(r1, 1.0) && check(r2, 3.0) && check(r3, 1.0) && check(r4, 1.0) { let cy = runs[i + 2].0 + runs[i + 2].1 / 2; - centers.push((col, cy)); + let total_px = total as usize; + centers.push((col, cy, total_px)); } } @@ -154,18 +155,17 @@ fn find_finders(gray: &[Vec]) -> Option<[FinderMatch; 3]> { return None; } - // 水平扫描 + // 水平扫描(含 finder 总尺寸) let mut h_centers: Vec<(usize, usize, usize)> = Vec::new(); // (cx, cy, size) for row in (0..height).step_by(2) { - for (cx, cy) in scan_row(gray, row) { + for (cx, cy, total_size) in scan_row(gray, row) { // 交叉验证:垂直扫描 let v_matches = scan_col(gray, cx); if v_matches .iter() - .any(|&(_, vy)| (vy as i32 - cy as i32).abs() < 5) + .any(|&(_, vy, _)| (vy as i32 - cy as i32).abs() < 5) { - let size = estimate_finder_size(gray, cx, cy); - h_centers.push((cx, cy, size)); + h_centers.push((cx, cy, total_size)); } } } diff --git a/core/src/decoder/image.rs b/core/src/decoder/image.rs index b21ceb1..12e9c1d 100644 --- a/core/src/decoder/image.rs +++ b/core/src/decoder/image.rs @@ -113,21 +113,19 @@ fn sauvola_binarize(gray: &GrayImage, w: u32, h: u32) -> Vec> { let r = 128.0f64; // 预计算积分图像以加速窗口和/平方和 - let mut integral = vec![0u64; (w as usize + 1) * (h as usize + 1)]; - let mut sq_integral = vec![0u64; (w as usize + 1) * (h as usize + 1)]; + // 标准公式: I(y+1,x+1) = I(y,x+1) + I(y+1,x) - I(y,x) + pixel(y,x) + let stride = w as usize + 1; + let mut integral = vec![0u64; stride * (h as usize + 1)]; + let mut sq_integral = vec![0u64; stride * (h as usize + 1)]; for y in 0..h as usize { - let mut row_sum = 0u64; - let mut row_sq = 0u64; for x in 0..w as usize { let p = gray.get_pixel(x as u32, y as u32).0[0] as u64; - row_sum += p; - row_sq += p * p; - let idx = (y + 1) * (w as usize + 1) + (x + 1); - let above = (y) * (w as usize + 1) + (x + 1); - let left = (y + 1) * (w as usize + 1) + (x); - let above_left = (y) * (w as usize + 1) + (x); - integral[idx] = integral[left] + integral[above] - integral[above_left] + row_sum; - sq_integral[idx] = sq_integral[left] + sq_integral[above] - sq_integral[above_left] + row_sq; + let idx = (y + 1) * stride + (x + 1); + let above = y * stride + (x + 1); + let left = (y + 1) * stride + x; + let above_left = y * stride + x; + integral[idx] = integral[above] + integral[left] - integral[above_left] + p; + sq_integral[idx] = sq_integral[above] + sq_integral[left] - sq_integral[above_left] + p * p; } } @@ -138,10 +136,10 @@ fn sauvola_binarize(gray: &GrayImage, w: u32, h: u32) -> Vec> { let y2 = (y + half + 1).min(h as i32) as usize; let idx = |xx: usize, yy: usize| yy * (w as usize + 1) + xx; let count = ((x2 - x1) * (y2 - y1)) as f64; - let sum = (integral[idx(x2, y2)] - integral[idx(x1, y2)] - integral[idx(x2, y1)] - + integral[idx(x1, y1)]) as f64; - let sq_sum = (sq_integral[idx(x2, y2)] - sq_integral[idx(x1, y2)] - sq_integral[idx(x2, y1)] - + sq_integral[idx(x1, y1)]) as f64; + let sum = (integral[idx(x2, y2)] as i64 - integral[idx(x1, y2)] as i64 + - integral[idx(x2, y1)] as i64 + integral[idx(x1, y1)] as i64) as f64; + let sq_sum = (sq_integral[idx(x2, y2)] as i64 - sq_integral[idx(x1, y2)] as i64 + - sq_integral[idx(x2, y1)] as i64 + sq_integral[idx(x1, y1)] as i64) as f64; let mean = sum / count; let variance = (sq_sum / count - mean * mean).max(0.0); (mean, variance.sqrt()) diff --git a/core/src/encoder/mode.rs b/core/src/encoder/mode.rs index fe4ac65..3576a3e 100644 --- a/core/src/encoder/mode.rs +++ b/core/src/encoder/mode.rs @@ -324,10 +324,10 @@ mod tests { #[test] fn test_unicode_to_shift_jis_known() { - // 基本汉字应返回 Some - assert!(unicode_to_shift_jis('中').is_some()); - assert!(unicode_to_shift_jis('文').is_some()); - assert!(unicode_to_shift_jis('你').is_some()); + // 基本汉字应返回 Some(encoding_rs 精确映射) + assert!(unicode_to_shift_jis('中').is_some(), "中 should map"); + assert!(unicode_to_shift_jis('文').is_some(), "文 should map"); + assert!(unicode_to_shift_jis('日').is_some(), "日 should map"); } #[test] diff --git a/core/src/matrix/mask.rs b/core/src/matrix/mask.rs index 5b49542..79bf821 100644 --- a/core/src/matrix/mask.rs +++ b/core/src/matrix/mask.rs @@ -305,13 +305,14 @@ mod tests { #[test] fn test_score_rule2() { let mut m = Matrix::new(3); + // 全部设为 dark → 4 个 2×2 同色方块 for y in 0..3u8 { for x in 0..3u8 { m.set(x, y, true); } } - let view = MaskedView::new(&m, 0); - assert_eq!(score_rule2(&view), 4 * 3); + // 使用 raw 版(不经过掩码)验证 + assert_eq!(score_rule2_raw(&m), 4 * 3); } #[test] diff --git a/liuhangyu.png b/liuhangyu.png new file mode 100644 index 0000000..7992a2c Binary files /dev/null and b/liuhangyu.png differ