mirror of
https://github.com/Stirling-Tools/Stirling-PDF.git
synced 2025-08-21 19:59:24 +00:00
splitPagesByPreset
This commit is contained in:
parent
d0dbb7e708
commit
771e66100f
@ -5,9 +5,7 @@ import { PdfFile } from "../../wrappers/PdfFile";
|
|||||||
import { getImagesOnPage, PDFJSImage } from "./getImagesOnPage";
|
import { getImagesOnPage, PDFJSImage } from "./getImagesOnPage";
|
||||||
|
|
||||||
export async function detectQRCodePages(file: PdfFile) {
|
export async function detectQRCodePages(file: PdfFile) {
|
||||||
console.log("FileInQRPrev: ", file);
|
|
||||||
const pdfDoc = await file.pdfJsDocument;
|
const pdfDoc = await file.pdfJsDocument;
|
||||||
console.log("FileInQRAfter: ", file);
|
|
||||||
|
|
||||||
const pagesWithQR: number[] = [];
|
const pagesWithQR: number[] = [];
|
||||||
for (let i = 0; i < pdfDoc.numPages; i++) {
|
for (let i = 0; i < pdfDoc.numPages; i++) {
|
||||||
|
@ -20,9 +20,9 @@ export class RotatePages extends Operator {
|
|||||||
protected static inputSchema = JoiPDFFileSchema.label(i18next.t("inputs.pdffile.name")).description(i18next.t("inputs.pdffile.description"));
|
protected static inputSchema = JoiPDFFileSchema.label(i18next.t("inputs.pdffile.name")).description(i18next.t("inputs.pdffile.description"));
|
||||||
protected static valueSchema = Joi.object({
|
protected static valueSchema = Joi.object({
|
||||||
rotation: Joi.alternatives().try(
|
rotation: Joi.alternatives().try(
|
||||||
Joi.number().integer().min(-360).max(360),
|
Joi.number().integer().min(-360).max(360).required(),
|
||||||
CommaArrayJoiExt.comma_array().items(Joi.number().integer().min(-360).max(360))
|
CommaArrayJoiExt.comma_array().items(Joi.number().integer().min(-360).max(360)).required()
|
||||||
).required()
|
)
|
||||||
.label(i18next.t("values.rotation.friendlyName", { ns: "rotatePages" })).description(i18next.t("values.rotation.description", { ns: "rotatePages" }))
|
.label(i18next.t("values.rotation.friendlyName", { ns: "rotatePages" })).description(i18next.t("values.rotation.description", { ns: "rotatePages" }))
|
||||||
.example("90").example("-180").example("[90, 0, 270]"),
|
.example("90").example("-180").example("[90, 0, 270]"),
|
||||||
});
|
});
|
||||||
|
@ -20,9 +20,9 @@ export class ScaleContent extends Operator {
|
|||||||
protected static inputSchema = JoiPDFFileSchema.label(i18next.t("inputs.pdffile.name")).description(i18next.t("inputs.pdffile.description"));
|
protected static inputSchema = JoiPDFFileSchema.label(i18next.t("inputs.pdffile.name")).description(i18next.t("inputs.pdffile.description"));
|
||||||
protected static valueSchema = Joi.object({
|
protected static valueSchema = Joi.object({
|
||||||
scaleFactor: Joi.alternatives().try(
|
scaleFactor: Joi.alternatives().try(
|
||||||
Joi.number(),
|
Joi.number().required(),
|
||||||
CommaArrayJoiExt.comma_array().items(Joi.number())
|
CommaArrayJoiExt.comma_array().items(Joi.number()).required()
|
||||||
).required()
|
)
|
||||||
.label(i18next.t("values.scaleFactor.friendlyName", { ns: "scaleContent" })).description(i18next.t("values.scaleFactor.description", { ns: "scaleContent" }))
|
.label(i18next.t("values.scaleFactor.friendlyName", { ns: "scaleContent" })).description(i18next.t("values.scaleFactor.description", { ns: "scaleContent" }))
|
||||||
.example("2").example("1.5").example("[1, 1.5, 0.9]"),
|
.example("2").example("1.5").example("[1, 1.5, 0.9]"),
|
||||||
});
|
});
|
||||||
|
@ -1,44 +1,82 @@
|
|||||||
|
import { Operator, Progress, oneToN } from ".";
|
||||||
|
|
||||||
|
import Joi from "@stirling-tools/joi";
|
||||||
|
import { JoiPDFFileSchema } from "../wrappers/PdfFileJoi";
|
||||||
|
|
||||||
|
import i18next from "i18next";
|
||||||
|
|
||||||
import { PdfFile } from "../wrappers/PdfFile";
|
import { PdfFile } from "../wrappers/PdfFile";
|
||||||
|
|
||||||
import { splitPagesByIndex } from "./common/splitPagesByIndex";
|
import { splitPagesByIndex } from "./common/splitPagesByIndex";
|
||||||
import { detectEmptyPages } from "./common/detectEmptyPages";
|
import { detectEmptyPages } from "./common/detectEmptyPages";
|
||||||
import { detectQRCodePages } from "./common/detectQRCodePages";
|
import { detectQRCodePages } from "./common/detectQRCodePages";
|
||||||
|
|
||||||
export interface SplitPageByPresetParamsType {
|
|
||||||
file: PdfFile;
|
|
||||||
type: "BAR_CODE"|"QR_CODE"|"BLANK_PAGE";
|
|
||||||
whiteThreashold?: number;
|
|
||||||
}
|
|
||||||
export async function splitPagesByPreset(params: SplitPageByPresetParamsType): Promise<PdfFile[]> {
|
|
||||||
const { file, type, whiteThreashold } = params;
|
|
||||||
|
|
||||||
console.log("File: ", file);
|
export class SplitPagesByPreset extends Operator {
|
||||||
|
static type = "splitPagesByPreset";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Validation & Localisation
|
||||||
|
*/
|
||||||
|
|
||||||
|
protected static inputSchema = JoiPDFFileSchema.label(i18next.t("inputs.pdffile.name")).description(i18next.t("inputs.pdffile.description"));
|
||||||
|
protected static valueSchema = Joi.alternatives().try(
|
||||||
|
Joi.object({
|
||||||
|
type: Joi.string().valid("BAR_CODE").required()
|
||||||
|
}),
|
||||||
|
Joi.object({
|
||||||
|
type: Joi.string().valid("QR_CODE").required()
|
||||||
|
}),
|
||||||
|
Joi.object({
|
||||||
|
type: Joi.string().valid("BLANK_PAGE").required(),
|
||||||
|
whiteThreashold: Joi.number().min(0).max(255).required()
|
||||||
|
}),
|
||||||
|
)
|
||||||
|
.label(i18next.t("values.splitSettings.friendlyName", { ns: "splitPagesByPreset" })).description(i18next.t("values.splitSettings.description", { ns: "splitPagesByPreset" })
|
||||||
|
);
|
||||||
|
protected static outputSchema = JoiPDFFileSchema.label(i18next.t("outputs.pdffile.name")).description(i18next.t("outputs.pdffile.description"));
|
||||||
|
|
||||||
|
static schema = Joi.object({
|
||||||
|
input: SplitPagesByPreset.inputSchema,
|
||||||
|
values: SplitPagesByPreset.valueSchema.required(),
|
||||||
|
output: SplitPagesByPreset.outputSchema
|
||||||
|
}).label(i18next.t("friendlyName", { ns: "splitPagesByPreset" })).description(i18next.t("description", { ns: "splitPagesByPreset" }));
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Logic
|
||||||
|
*/
|
||||||
|
|
||||||
|
/** Detect and remove white pages */
|
||||||
|
async run(input: PdfFile[], progressCallback: (state: Progress) => void): Promise<PdfFile[]> {
|
||||||
|
return oneToN<PdfFile, PdfFile>(input, async (input, index, max) => {
|
||||||
let splitAtPages: number[];
|
let splitAtPages: number[];
|
||||||
switch (type) {
|
console.log("Running Detection...");
|
||||||
|
|
||||||
|
switch (this.actionValues.type) {
|
||||||
case "BAR_CODE":
|
case "BAR_CODE":
|
||||||
// TODO: Implement
|
// TODO: Implement
|
||||||
throw new Error("This split-type has not been implemented yet");
|
throw new Error("This split-type has not been implemented yet");
|
||||||
|
|
||||||
case "QR_CODE":
|
case "QR_CODE":
|
||||||
splitAtPages = await detectQRCodePages(file);
|
splitAtPages = await detectQRCodePages(input);
|
||||||
break;
|
break;
|
||||||
|
|
||||||
case "BLANK_PAGE":
|
case "BLANK_PAGE":
|
||||||
if (!whiteThreashold)
|
splitAtPages = await detectEmptyPages(input, this.actionValues.whiteThreashold);
|
||||||
throw new Error("White threshold not provided");
|
|
||||||
splitAtPages = await detectEmptyPages(file, whiteThreashold);
|
|
||||||
break;
|
break;
|
||||||
|
|
||||||
default:
|
default:
|
||||||
throw new Error("An invalid split-type was provided.");
|
throw new Error("An invalid split-type was provided.");
|
||||||
}
|
}
|
||||||
|
console.log("Split at Pages: ", splitAtPages);
|
||||||
|
|
||||||
console.debug("Split At Pages: ", splitAtPages);
|
const newFiles = await splitPagesByIndex(input, splitAtPages);
|
||||||
|
|
||||||
const newFiles = await splitPagesByIndex(file, splitAtPages);
|
|
||||||
for (let i = 0; i < newFiles.length; i++) {
|
for (let i = 0; i < newFiles.length; i++) {
|
||||||
newFiles[i].filename += "_split-"+i;
|
newFiles[i].filename += "_split-"+i;
|
||||||
}
|
}
|
||||||
|
progressCallback({ curFileProgress: 1, operationProgress: index/max });
|
||||||
return newFiles;
|
return newFiles;
|
||||||
|
});
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -78,6 +78,7 @@ export class PdfFile {
|
|||||||
});
|
});
|
||||||
default:
|
default:
|
||||||
return new Promise(async (resolve) => {
|
return new Promise(async (resolve) => {
|
||||||
|
console.log(`Converting representationType-${this.representationType} to pdfJsDocument`);
|
||||||
const pdfjsDoc = await PDFJS.getDocument({ data: await this.uint8Array, isOffscreenCanvasSupported: false }).promise;
|
const pdfjsDoc = await PDFJS.getDocument({ data: await this.uint8Array, isOffscreenCanvasSupported: false }).promise;
|
||||||
this.pdfJsDocument = pdfjsDoc;
|
this.pdfJsDocument = pdfjsDoc;
|
||||||
resolve(pdfjsDoc);
|
resolve(pdfjsDoc);
|
||||||
@ -103,7 +104,7 @@ export class PdfFile {
|
|||||||
}
|
}
|
||||||
|
|
||||||
static fromMulterFile(value: Express.Multer.File): PdfFile {
|
static fromMulterFile(value: Express.Multer.File): PdfFile {
|
||||||
return new PdfFile(value.originalname, value.buffer as Uint8Array, RepresentationType.Uint8Array);
|
return new PdfFile(value.originalname, new Uint8Array(value.buffer), RepresentationType.Uint8Array);
|
||||||
|
|
||||||
}
|
}
|
||||||
static fromMulterFiles(values: Express.Multer.File[]): PdfFile[] {
|
static fromMulterFiles(values: Express.Multer.File[]): PdfFile[] {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user