Stirling-PDF/shared-operations/src/functions/common/getImagesOnPage.ts

import { PDFPageProxy } from "pdfjs-dist/types/src/display/api";

import * as PDFJS from "pdfjs-dist";

import { createCanvas } from "canvas";

export interface PDFJSImage {
    width: number;
    height: number;
    interpolate?: any;
    kind: number; // TODO: Document what this is, maybe hasAlpha?
    data: Uint8ClampedArray;
}

export async function getImagesOnPage(page: PDFPageProxy): Promise<PDFJSImage[]> {
    const ops = await page.getOperatorList();
    const images: PDFJSImage[] = [];
    for (let j=0; j < ops.fnArray.length; j++) {
        if (ops.fnArray[j] == PDFJS.OPS.paintImageXObject) {
            const image: PDFJSImage = page.objs.get(ops.argsArray[j][0])

            console.log("Image: ", image);
            images.push(image);
        }
    }
    return images;
}