use crate::{
Binarizer, BinaryBitmap, DecodeHints, Exceptions, Point, RXingResult, Reader,
common::{Quadrilateral, Result},
point,
};
use super::MultipleBarcodeReader;
#[derive(Default)]
pub struct GenericMultipleBarcodeReader<T: Reader>(T);
impl<T: Reader> MultipleBarcodeReader for GenericMultipleBarcodeReader<T> {
fn decode_multiple<B: Binarizer>(
&mut self,
image: &mut BinaryBitmap<B>,
) -> Result<Vec<RXingResult>> {
self.decode_multiple_with_hints(image, &DecodeHints::default())
}
fn decode_multiple_with_hints<B: Binarizer>(
&mut self,
image: &mut BinaryBitmap<B>,
hints: &DecodeHints,
) -> Result<Vec<RXingResult>> {
let mut results = Vec::new();
self.do_decode_multiple(image, hints, &mut results, 0, 0, 0);
let unique_results: Vec<RXingResult> = results
.iter()
.enumerate()
.filter(|(i, r)| {
let already_found = if r.getPoints().len() >= 4 {
let q1 = Quadrilateral::new(
r.getPoints()[0],
r.getPoints()[1],
r.getPoints()[2],
r.getPoints()[3],
);
results.iter().skip(*i + 1).any(|e| {
if e.getPoints().len() >= 4 {
let q2 = Quadrilateral::new(
e.getPoints()[0],
e.getPoints()[1],
e.getPoints()[2],
e.getPoints()[3],
);
Quadrilateral::have_intersecting_bounding_boxes(&q1, &q2)
} else {
e.getPoints().iter().any(|p| q1.is_inside(*p))
}
})
} else {
results.iter().skip(*i + 1).any(|e| {
if e.getPoints().len() >= 4 {
let q2 = Quadrilateral::new(
e.getPoints()[0],
e.getPoints()[1],
e.getPoints()[2],
e.getPoints()[3],
);
e.getPoints().iter().any(|p| q2.is_inside(*p))
} else {
e.getText() == r.getText()
&& e.getBarcodeFormat() == r.getBarcodeFormat()
}
})
};
!already_found
})
.map(|(_, r)| r)
.cloned()
.collect();
if unique_results.is_empty() {
return Err(Exceptions::NOT_FOUND);
}
Ok(unique_results)
}
}
impl<T: Reader> GenericMultipleBarcodeReader<T> {
const MIN_DIMENSION_TO_RECUR: f32 = 100.0;
const MAX_DEPTH: u32 = 4;
pub fn new(delegate: T) -> Self {
Self(delegate)
}
fn do_decode_multiple<B: Binarizer>(
&mut self,
image: &mut BinaryBitmap<B>,
hints: &DecodeHints,
results: &mut Vec<RXingResult>,
xOffset: u32,
yOffset: u32,
currentDepth: u32,
) {
if currentDepth > Self::MAX_DEPTH {
return;
}
let Ok(result) = self.0.decode_with_hints(image, hints) else {
return;
};
let resultPoints = result.getPoints().to_vec();
let possible_new_result = Self::translatePoints(result, xOffset, yOffset);
results.push(possible_new_result);
if resultPoints.is_empty() {
return;
}
let width = image.get_width();
let height = image.get_height();
let mut minX: f32 = width as f32;
let mut minY: f32 = height as f32;
let mut maxX: f32 = 0.0;
let mut maxY: f32 = 0.0;
for point in resultPoints.into_iter() {
let x = point.x;
let y = point.y;
minX = f32::min(x, minX);
minY = f32::min(y, minY);
maxX = f32::max(x, maxX);
maxY = f32::max(y, maxY);
}
if minX > Self::MIN_DIMENSION_TO_RECUR {
self.do_decode_multiple(
&mut image.crop(0, 0, minX as usize, height),
hints,
results,
xOffset,
yOffset,
currentDepth + 1,
);
}
if minY > Self::MIN_DIMENSION_TO_RECUR {
self.do_decode_multiple(
&mut image.crop(0, 0, width, minY as usize),
hints,
results,
xOffset,
yOffset,
currentDepth + 1,
);
}
if maxX < (width as f32) - Self::MIN_DIMENSION_TO_RECUR {
self.do_decode_multiple(
&mut image.crop(maxX as usize, 0, width - maxX as usize, height),
hints,
results,
xOffset + maxX as u32,
yOffset,
currentDepth + 1,
);
}
if maxY < (height as f32) - Self::MIN_DIMENSION_TO_RECUR {
self.do_decode_multiple(
&mut image.crop(0, maxY as usize, width, height - maxY as usize),
hints,
results,
xOffset,
yOffset + maxY as u32,
currentDepth + 1,
);
}
}
fn translatePoints(result: RXingResult, xOffset: u32, yOffset: u32) -> RXingResult {
let oldPoints = result.getPoints();
if oldPoints.is_empty() {
return result;
}
let newPoints: Vec<Point> = oldPoints
.iter()
.map(|oldPoint| point(oldPoint.x + xOffset as f32, oldPoint.y + yOffset as f32))
.collect();
let mut newRXingResult = RXingResult::new_complex(
result.getText(),
result.getRawBytes().to_vec(),
result.getNumBits(),
newPoints,
*result.getBarcodeFormat(),
result.getTimestamp(),
);
newRXingResult.putAllMetadata(result.getRXingResultMetadata().clone());
newRXingResult
}
}