QRCode Detection working.

public/functions/splitOn.js

@@ -1,4 +1,7 @@
-import { detectEmptyPages } from "./shared/detectEmptyPages";
+import { detectEmptyPages } from "./shared/detectEmptyPages.js";
+import { getImagesOnPage } from "./shared/getImagesOnPage.js";
+
+import jsQR from "jsQR";
 
 /**
  * @typedef {"BAR_CODE"|"QR_CODE"|"BLANK_PAGE"} SplitType
@@ -9,24 +12,22 @@ import { detectEmptyPages } from "./shared/detectEmptyPages";
  * @param {Uint16Array} snapshot
  * @param {SplitType} type
  * @param {} PDFJS
- * @param {} OpenCV
+ * @param {import('opencv-wasm')} OpenCV
  * @param {} PDFLib
- * @param {} QRCode
  * @returns 
  */
-export async function splitOn(snapshot, type, whiteThreashold, PDFJS, OpenCV, PDFLib, QRCode) {
+export async function splitOn(snapshot, type, whiteThreashold, PDFJS, OpenCV, PDFLib) {
     
     let splitAtPages = [];
 
     switch (type) {
         case "BAR_CODE":
             // TODO: Implement
-            throw new Error("This split-type has not been implemented yet")
+            throw new Error("This split-type has not been implemented yet");
             break;
 
         case "QR_CODE":
-            // TODO: Implement
-            throw new Error("This split-type has not been implemented yet")
+            splitAtPages = await getPagesWithQRCode(snapshot);
             break;
 
         case "BLANK_PAGE":
@@ -45,4 +46,49 @@ export async function splitOn(snapshot, type, whiteThreashold, PDFJS, OpenCV, PD
     // TODO: Remove detected Pages & Split
 
     return pdfDoc.save();
+
+    async function getPagesWithQRCode(snapshot) {
+        const pdfDoc = await PDFJS.getDocument(snapshot).promise;
+
+        const pagesWithQR = [];
+        for (let i = 1; i <= pdfDoc.numPages; i++) {
+            const page = await pdfDoc.getPage(i);
+            console.log("Checking page " + i);
+
+            const images = await getImagesOnPage(page, PDFJS);
+
+            for (const image of images) {
+                const data = await checkForQROnImage(image);
+                if(data == "https://github.com/Frooodle/Stirling-PDF") {
+                    pagesWithQR.push(i);
+                }
+            }
+        }
+        return pagesWithQR;
+    }
+
+    async function checkForQROnImage(image) {
+        console.log(image.data, image.width, image.height, image.width * image.height * 4);
+        
+        // TODO: There is an issue with the jsQR package, and the package seems to be stale, we could create a fork and fix the issue (The package expects rgba but sometimes we have rgb). In the meanwhile we just force rgba:
+        if(image.data.length == image.width * image.height * 3) {
+            const tmpArray = new Uint8ClampedArray(image.width * image.height * 4);
+
+            // Iterate through the original array and add an alpha channel
+            for (let i = 0, j = 0; i < image.data.length; i += 3, j += 4) {
+                tmpArray[j] = image.data[i];     // Red channel
+                tmpArray[j + 1] = image.data[i + 1]; // Green channel
+                tmpArray[j + 2] = image.data[i + 2]; // Blue channel
+                tmpArray[j + 3] = 255;               // Alpha channel (fully opaque)
+            }
+
+            image.data = tmpArray;
+        }
+
+        const code = jsQR(image.data, image.width, image.height);
+        if(code)
+            return code.data;
+        else
+            return null;
+    }
 };