Mirrors/Stirling-PDF
Mirrors/Stirling-PDF
1
0
Fork 0
mirror of https://github.com/Frooodle/Stirling-PDF.git synced 2025-01-10 00:06:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

More Types, Fixed isImageBlank

Browse Source
This commit is contained in:
Felix Kaspar 2023-11-16 01:56:17 +01:00
parent 8f1f64f6c0
commit 921a8c431c
4 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server-node/declarations/PdfcpuWrapper.d.ts vendored Normal file
View File

@ -0,0 +1,3 @@
declare module '@stirling-pdf/shared-operations/src/wasm/pdfcpu/pdfcpu-wrapper-node.js' {
export function oneToOne(wasmArray: string[], snapshot: Uint8Array): Promise<Uint8Array>;
}

9
shared-operations/src/functions/common/detectEmptyPages.ts
View File

@ -1,9 +1,9 @@
import { PdfFile } from '../../wrappers/PdfFile'; import { PdfFile } from '../../wrappers/PdfFile';
import { PDFPageProxy } from "pdfjs-dist/types/src/display/api.js"; import { PDFPageProxy } from "pdfjs-dist/types/src/display/api.js";
import { Image } from 'image-js'; import { Image, ImageKind } from 'image-js';
import { getImagesOnPage } from "./getImagesOnPage.js"; import { getImagesOnPage, PDFJSImage } from "./getImagesOnPage.js";
export async function detectEmptyPages(file: PdfFile, whiteThreashold: number): Promise<number[]> { export async function detectEmptyPages(file: PdfFile, whiteThreashold: number): Promise<number[]> {
const pdfDoc = await file.pdfJsDocument; const pdfDoc = await file.pdfJsDocument;
@ -43,8 +43,9 @@ async function areImagesBlank(page: PDFPageProxy, threshold: number): Promise<bo
return true; return true;
} }
async function isImageBlank(image: string | Uint8Array | ArrayBuffer, threshold: number): Promise<boolean> { // TODO: Fix this function
var img = await Image.load(image); async function isImageBlank(image: PDFJSImage, threshold: number): Promise<boolean> {
var img = new Image(image.width, image.height, image.data, { kind: "RGB" as ImageKind }); // TODO: Maybe respect image.kind and convert accordingly, needs to be tested with a pdf with alpha-image
var grey = img.grey(); var grey = img.grey();
var mean = grey.getMean(); var mean = grey.getMean();
return mean[0] <= threshold; return mean[0] <= threshold;

2
shared-operations/src/functions/common/getImagesOnPage.ts
View File

@ -8,7 +8,7 @@ export type PDFJSImage = {
height: number; height: number;
interpolate?: any; interpolate?: any;
kind: number; kind: number;
data: Uint8Array; data: Uint8ClampedArray;
}; };
export async function getImagesOnPage(page: PDFPageProxy): Promise<PDFJSImage[]> { export async function getImagesOnPage(page: PDFPageProxy): Promise<PDFJSImage[]> {

8
shared-operations/src/wasm/pdfcpu/pdfcpu-wrapper-node.js
View File

@ -74,12 +74,12 @@ export async function oneToOne(wasmArray, snapshot) {
} }
console.log("Nuping Done"); console.log("Nuping Done");
/* TODO:
* Make this more elegant, this waits for the write to finish.
* Maybe replace wasmfs with https://github.com/streamich/memfs
*/
await checkExistsWithTimeout("/output.pdf", 1000); await checkExistsWithTimeout("/output.pdf", 1000);
console.log("Write started..."); console.log("Write started...");
// TODO: Make this more elegant, this waits for the write to finish.
// Maybe replace wasmfs with https://github.com/streamich/memfs
let fileSize; let fileSize;
while (true) { while (true) {
fileSize = fs.statSync("/output.pdf").size; fileSize = fs.statSync("/output.pdf").size;