zorya-classifier 0.2.0 → 0.2.2

package/dist/infrastructure/segmentation/grabcut/worker-inference.js

@@ -1,190 +1,7 @@
-import { createHash } from 'node:crypto';
-import { open } from 'node:fs/promises';
-import { createRequire } from 'node:module';
-import { dirname, resolve } from 'node:path';
-import { Script } from 'node:vm';
-import { z } from 'zod';
-const artifact = {
-    bytes: 10_964_323,
-    sha256: '63366510248adf3a7eddf3e793dd825404efb7df3749f4d6f8557c7fa4ca8aa0',
-};
-const cvSchema = z.object({
-    Mat: z.custom((value) => typeof value === 'function'),
-    Rect: z.custom((value) => typeof value === 'function'),
-    cvtColor: z.custom((value) => typeof value === 'function'),
-    grabCut: z.custom((value) => typeof value === 'function'),
-    COLOR_RGBA2RGB: z.number().int(),
-    CV_8UC4: z.number().int(),
-    CV_8UC3: z.number().int(),
-    CV_8UC1: z.number().int(),
-    GC_BGD: z.number().int(),
-    GC_FGD: z.number().int(),
-    GC_PR_BGD: z.number().int(),
-    GC_PR_FGD: z.number().int(),
-    GC_INIT_WITH_MASK: z.number().int(),
-    GC_INIT_WITH_RECT: z.number().int(),
-});
-async function readVerifiedAsset(path) {
-    const handle = await open(path, 'r');
-    try {
-        const stat = await handle.stat();
-        if (!stat.isFile() || stat.size !== artifact.bytes)
-            throw new Error('Pinned OpenCV size mismatch');
-        const source = Buffer.alloc(artifact.bytes);
-        let offset = 0;
-        while (offset < source.length) {
-            const read = await handle.read(source, offset, source.length - offset, null);
-            if (!read.bytesRead)
-                throw new Error('Pinned OpenCV asset read was incomplete');
-            offset += read.bytesRead;
-        }
-        if ((await handle.read(Buffer.alloc(1), 0, 1, null)).bytesRead !== 0)
-            throw new Error('Pinned OpenCV asset changed while being read');
-        if (createHash('sha256').update(source).digest('hex') !== artifact.sha256)
-            throw new Error('Pinned OpenCV SHA256 mismatch');
-        return source.toString('utf8');
-    }
-    finally {
-        await handle.close();
-    }
-}
-async function loadOpenCv(path) {
-    path = resolve(path);
-    const source = await readVerifiedAsset(path);
-    const wrapper = { exports: {} };
-    const factory = new Script('(function(exports,require,module,__filename,__dirname){' + source + '\n})', { filename: path }).runInThisContext();
-    if (typeof factory !== 'function')
-        throw new Error('Pinned OpenCV asset is not a module');
-    Reflect.apply(factory, undefined, [
-        wrapper.exports,
-        createRequire(path),
-        wrapper,
-        path,
-        dirname(path),
-    ]);
-    const exports = wrapper.exports;
-    if (typeof exports === 'object' && exports !== null && 'then' in exports) {
-        const then = Reflect.get(exports, 'then');
-        if (typeof then === 'function') {
-            return await new Promise((resolve, reject) => {
-                try {
-                    Reflect.apply(then, exports, [
-                        (value) => {
-                            const checked = cvSchema.safeParse(value);
-                            if (checked.success)
-                                resolve(checked.data);
-                            else
-                                reject(new Error('Pinned OpenCV namespace is invalid'));
-                        },
-                        reject,
-                    ]);
-                }
-                catch (error) {
-                    reject(error);
-                }
-            });
-        }
-    }
-    return cvSchema.parse(exports);
-}
-export async function probeGrabCut(assetPath) {
-    await loadOpenCv(assetPath);
-}
-const matShapeSchema = z
-    .object({ delete: z.custom((value) => typeof value === 'function') })
-    .passthrough();
-const matSchema = z.custom((value) => matShapeSchema.safeParse(value).success);
-function matrix(cv, args = []) {
-    return matSchema.parse(Reflect.construct(cv.Mat, args));
-}
-function getBytes(mat, length) {
-    const bytes = z.instanceof(Uint8Array).parse(mat.data);
-    if (bytes.length !== length)
-        throw new Error('OpenCV matrix dimensions do not match');
-    return bytes;
-}
-export function initializeGrabCutMask(input) {
-    if (input.flags.length !== input.width * input.height)
-        throw new Error('GrabCut mask dimensions do not match source');
-    input.flags.fill(input.labels.background);
-    if (input.prompt === 'whole-region') {
-        for (let y = input.wholeRegion.top; y < input.wholeRegion.top + input.wholeRegion.height; y++)
-            for (let x = input.wholeRegion.left; x < input.wholeRegion.left + input.wholeRegion.width; x++)
-                input.flags[y * input.width + x] = input.labels.probableForeground;
-    }
-    else {
-        for (let y = input.roi.top; y < input.roi.top + input.roi.height; y++)
-            for (let x = input.roi.left; x < input.roi.left + input.roi.width; x++)
-                input.flags[y * input.width + x] = input.labels.probableBackground;
-    }
-    const seedLeft = Math.max(input.seedRegion.left, input.roi.left);
-    const seedTop = Math.max(input.seedRegion.top, input.roi.top);
-    const seedRight = Math.min(input.seedRegion.left + input.seedRegion.width, input.roi.left + input.roi.width);
-    const seedBottom = Math.min(input.seedRegion.top + input.seedRegion.height, input.roi.top + input.roi.height);
-    if (seedLeft >= seedRight || seedTop >= seedBottom)
-        throw new Error('GrabCut seed region does not intersect hard ROI');
-    for (let y = seedTop; y < seedBottom; y++)
-        for (let x = seedLeft; x < seedRight; x++)
-            input.flags[y * input.width + x] = input.labels.foreground;
-}
-/** Run native GrabCut, then reacquire Mat.data before copying its post-call flags. */
-export function runGrabCutAndCopyMask(mat, width, height, roi, runNative, labels) {
-    runNative();
-    const flags = getBytes(mat, width * height);
-    const output = new Uint8Array(width * height);
-    for (let y = roi.top; y < roi.top + roi.height; y++) {
-        for (let x = roi.left; x < roi.left + roi.width; x++) {
-            const at = y * width + x;
-            const flag = flags[at];
-            if (flag === labels.foreground || flag === labels.probableForeground)
-                output[at] = 1;
-        }
-    }
-    return output;
-}
-function seededComponents(mask, width, height, input) {
-    const visited = new Uint8Array(mask.length);
-    const output = new Uint8Array(mask.length);
-    const stack = [];
-    for (let start = 0; start < mask.length; start++) {
-        if (mask[start] !== 1 || visited[start] === 1)
-            continue;
-        stack.push(start);
-        visited[start] = 1;
-        const component = [];
-        let hitsSeed = false;
-        while (stack.length) {
-            const at = stack.pop();
-            if (at === undefined)
-                continue;
-            component.push(at);
-            const x = at % width;
-            const y = Math.floor(at / width);
-            if (x >= input.seedRegion.left &&
-                y >= input.seedRegion.top &&
-                x < input.seedRegion.left + input.seedRegion.width &&
-                y < input.seedRegion.top + input.seedRegion.height)
-                hitsSeed = true;
-            const neighbors = [at - 1, at + 1, at - width, at + width];
-            for (const neighbor of neighbors) {
-                if (neighbor < 0 ||
-                    neighbor >= mask.length ||
-                    visited[neighbor] === 1 ||
-                    mask[neighbor] !== 1)
-                    continue;
-                const nx = neighbor % width;
-                if (Math.abs(nx - x) + Math.abs(Math.floor(neighbor / width) - y) !== 1)
-                    continue;
-                visited[neighbor] = 1;
-                stack.push(neighbor);
-            }
-        }
-        if (hitsSeed)
-            for (const index of component)
-                output[index] = 1;
-    }
-    return output;
-}
+import { getBytes, initializeGrabCutMask, runGrabCutAndCopyMask, seededComponents, } from './masks.js';
+import { loadOpenCv, matrix, probeGrabCut } from './runtime.js';
+export { initializeGrabCutMask, runGrabCutAndCopyMask };
+export { probeGrabCut };
 export async function runGrabCutTask(request) {
     const cv = await loadOpenCv(request.assetPath);
     const pixels = request.width * request.height;

package/dist/infrastructure/segmentation/mobile-sam/index.d.ts

@@ -1,58 +1,3 @@
-import { z } from 'zod';
-import type { SegmentationPort } from '../../../entities/segmentation.js';
-import type { ObjectProposalPort } from '../../../entities/object-proposals.js';
-import { NativeChildPool } from '../native-child.js';
-import { type NativeSegmentationRequest } from '../native-protocol.js';
-declare const optionsSchema: z.ZodObject<{
-    encoderPath: z.ZodString;
-    decoderPath: z.ZodString;
-    executionProvider: z.ZodDefault<z.ZodEnum<["cpu", "dml"]>>;
-    deviceId: z.ZodOptional<z.ZodNumber>;
-    threads: z.ZodDefault<z.ZodNumber>;
-}, "strict", z.ZodTypeAny, {
-    encoderPath: string;
-    decoderPath: string;
-    executionProvider: "cpu" | "dml";
-    threads: number;
-    deviceId?: number | undefined;
-}, {
-    encoderPath: string;
-    decoderPath: string;
-    executionProvider?: "cpu" | "dml" | undefined;
-    deviceId?: number | undefined;
-    threads?: number | undefined;
-}>;
-export type MobileSamSegmenterOptions = z.input<typeof optionsSchema>;
-declare const proposalOptionsSchema: z.ZodObject<{
-    encoderPath: z.ZodString;
-    decoderPath: z.ZodString;
-    executionProvider: z.ZodDefault<z.ZodLiteral<"cpu">>;
-    threads: z.ZodDefault<z.ZodNumber>;
-}, "strict", z.ZodTypeAny, {
-    encoderPath: string;
-    decoderPath: string;
-    executionProvider: "cpu";
-    threads: number;
-}, {
-    encoderPath: string;
-    decoderPath: string;
-    executionProvider?: "cpu" | undefined;
-    threads?: number | undefined;
-}>;
-export type MobileSamObjectProposalsOptions = z.input<typeof proposalOptionsSchema>;
-type SerializedTask = (Omit<Extract<NativeSegmentationRequest, {
-    kind: 'mobile-sam';
-}>, 'id' | 'rgba'> & {
-    rgba: string;
-}) | (Omit<Extract<NativeSegmentationRequest, {
-    kind: 'mobile-sam-expand';
-}>, 'id' | 'rgba'> & {
-    rgba: string;
-});
-/** Explicit pinned MobileSAM inference in a separately owned Node child process. */
-export declare function createMobileSamSegmenter(options: MobileSamSegmenterOptions): Promise<SegmentationPort>;
-/** Internal adapter seam: pool ownership stays with the caller. */
-export declare function createMobileSamPortForPool(pool: Pick<NativeChildPool<SerializedTask>, 'runLazy' | 'settle' | 'dispose'>, parsed: z.output<typeof optionsSchema>): SegmentationPort & Required<Pick<SegmentationPort, 'expand'>>;
-/** Explicit CPU-only, caller-owned one-shot 3x3-point / 36-mask proposal adapter. */
-export declare function createMobileSamObjectProposals(options: MobileSamObjectProposalsOptions): Promise<ObjectProposalPort>;
-export {};
+export type { MobileSamSegmenterOptions, MobileSamObjectProposalsOptions } from './options.js';
+export { createMobileSamSegmenter, createMobileSamPortForPool } from './segmenter.js';
+export { createMobileSamObjectProposals } from './object-proposals.js';