feat: improve qr detection and add jpg support (#21)

* feat: change image from mono to gray-scale

* feat: use rayon to process pages in parallel

* feat: increase resolution from 300dpi to 400dpi

* feat: add support for jpg/jpeg images

* feat: code formatting

---------

Co-authored-by: mike <[email protected]>

Cargo.toml

@@ -22,6 +22,7 @@ rqrr = "0.7.1"
 image = "0.25.1"
 tempfile = "3.10.1"
 serde = { version = "1.0.203", features = ["derive"] }
+rayon = "1.10.0"
 
 [dev-dependencies]
 assert_cmd = "2.0.14"

src/lib.rs

@@ -10,6 +10,7 @@ mod qr_parser;
 
 use crate::models::qr_data::QRData;
 use image;
+use rayon::prelude::*;
 use rqrr::PreparedImage;
 use tempfile::tempdir;
 
@@ -19,21 +20,24 @@ pub fn get_qr_bill_data(file_path: String, fail_on_error: bool) -> Vec<QRData> {
         input if input.ends_with(".pdf") => {
             pdf_converter::convert_to_png(&file_path, &tmp_dir.path())
         }
-        input if input.ends_with(".png") => {
+        input if input.ends_with(".png") || input.ends_with(".jpg") || input.ends_with(".jpeg") => {
             vec![image::open(&file_path).expect("Error loading image")]
         }
         _ => panic!("Unsupported file format"),
     };
 
-    let mut all_qr_codes = Vec::new();
-    for img in images {
-        let mut img = PreparedImage::prepare(img.to_luma8());
-        img.detect_grids()
-            .into_iter()
-            .filter_map(|result| result.decode().ok())
-            .map(|(_, content)| qr_parser::get_qr_code_data(&content))
-            .for_each(|qr_data| all_qr_codes.push(qr_data));
-    }
+    let all_qr_codes: Vec<_> = images
+        .into_par_iter()
+        .map(|img| {
+            let mut img = PreparedImage::prepare(img.to_luma8());
+            img.detect_grids()
+                .into_par_iter()
+                .filter_map(|result| result.decode().ok())
+                .map(|(_, content)| qr_parser::get_qr_code_data(&content))
+                .collect::<Vec<_>>()
+        })
+        .flatten()
+        .collect();
 
     // check if there were any errors
     if fail_on_error && all_qr_codes.iter().any(|result| result.is_err()) {

src/pdf_converter.rs

@@ -27,8 +27,8 @@ where
             "-dSAFER",
             "-dNOPAUSE",
             "-dFILTERTEXT",
-            "-r300",
-            "-sDEVICE=pngmono",
+            "-r400",
+            "-sDEVICE=pnggray",
             &format!(
                 "-sOutputFile={}/%03d.png",
                 tmp_path.as_ref().to_string_lossy()