lootforge 0.3.0

package/bin/lootforge.js

@@ -0,0 +1,28 @@
+#!/usr/bin/env node
+
+import { existsSync } from "node:fs";
+import { dirname, resolve } from "node:path";
+import process from "node:process";
+import { fileURLToPath, pathToFileURL } from "node:url";
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+const distEntry = resolve(__dirname, "../dist/cli/index.js");
+
+if (!existsSync(distEntry)) {
+  process.stderr.write("lootforge: CLI is not built yet. Run `npm run build` first.\n");
+  process.exit(1);
+}
+
+try {
+  const moduleUrl = pathToFileURL(distEntry).href;
+  const module = await import(moduleUrl);
+  const exitCode = await module.main(process.argv.slice(2));
+
+  if (typeof exitCode === "number") {
+    process.exitCode = exitCode;
+  }
+} catch (error) {
+  const message = error instanceof Error ? error.message : String(error);
+  process.stderr.write(`lootforge: ${message}\n`);
+  process.exitCode = 1;
+}

package/dist/benchmarks/coarseToFineCost.d.ts

@@ -0,0 +1,21 @@
+import { type CandidateScoreRecord, type ProviderRunResult } from "../providers/types.js";
+export interface CandidateStageCostModel {
+    full: number;
+    draft: number;
+    refine: number;
+    autocorrect: number;
+}
+export declare const DEFAULT_CANDIDATE_STAGE_COST_MODEL: CandidateStageCostModel;
+export interface RunCostSummary {
+    targetCount: number;
+    approvedTargets: number;
+    providerRunCount: number;
+    scoredCandidateCount: number;
+    totalCostUnits: number;
+    costPerApprovedTarget: number;
+}
+type JobCostInput = Pick<ProviderRunResult, "candidateScores">;
+export declare function summarizeGenerateRunCost(jobs: JobCostInput[], model?: CandidateStageCostModel): RunCostSummary;
+export declare function estimateJobCostUnits(job: JobCostInput, model?: CandidateStageCostModel): number;
+export declare function estimateCandidateScoreCostUnits(score: CandidateScoreRecord, model?: CandidateStageCostModel): number;
+export {};

package/dist/benchmarks/coarseToFineCost.js

@@ -0,0 +1,49 @@
+export const DEFAULT_CANDIDATE_STAGE_COST_MODEL = {
+    full: 3,
+    draft: 1,
+    refine: 3,
+    autocorrect: 3,
+};
+export function summarizeGenerateRunCost(jobs, model = DEFAULT_CANDIDATE_STAGE_COST_MODEL) {
+    const approvedTargets = countSelectedAcceptedTargets(jobs);
+    const scoredCandidateCount = jobs.reduce((count, job) => count + (job.candidateScores?.length ?? 0), 0);
+    const totalCostUnits = jobs.reduce((sum, job) => sum + estimateJobCostUnits(job, model), 0);
+    return {
+        targetCount: jobs.length,
+        approvedTargets,
+        providerRunCount: jobs.length,
+        scoredCandidateCount,
+        totalCostUnits,
+        costPerApprovedTarget: approvedTargets > 0 ? totalCostUnits / approvedTargets : Infinity,
+    };
+}
+export function estimateJobCostUnits(job, model = DEFAULT_CANDIDATE_STAGE_COST_MODEL) {
+    const scores = job.candidateScores ?? [];
+    if (scores.length === 0) {
+        return model.full;
+    }
+    return scores.reduce((sum, score) => sum + estimateCandidateScoreCostUnits(score, model), 0);
+}
+export function estimateCandidateScoreCostUnits(score, model = DEFAULT_CANDIDATE_STAGE_COST_MODEL) {
+    if (score.stage === "draft") {
+        return model.draft;
+    }
+    if (score.stage === "refine") {
+        return model.refine;
+    }
+    if (score.stage === "autocorrect") {
+        return model.autocorrect;
+    }
+    return model.full;
+}
+function countSelectedAcceptedTargets(jobs) {
+    let approved = 0;
+    for (const job of jobs) {
+        const selected = job.candidateScores?.find((score) => score.selected) ?? job.candidateScores?.[0];
+        if (selected?.passedAcceptance) {
+            approved += 1;
+        }
+    }
+    return approved;
+}
+//# sourceMappingURL=coarseToFineCost.js.map

package/dist/benchmarks/coarseToFineCost.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"coarseToFineCost.js","sourceRoot":"","sources":["../../src/benchmarks/coarseToFineCost.ts"],"names":[],"mappings":"AASA,MAAM,CAAC,MAAM,kCAAkC,GAA4B;IACzE,IAAI,EAAE,CAAC;IACP,KAAK,EAAE,CAAC;IACR,MAAM,EAAE,CAAC;IACT,WAAW,EAAE,CAAC;CACf,CAAC;AAaF,MAAM,UAAU,wBAAwB,CACtC,IAAoB,EACpB,QAAiC,kCAAkC;IAEnE,MAAM,eAAe,GAAG,4BAA4B,CAAC,IAAI,CAAC,CAAC;IAC3D,MAAM,oBAAoB,GAAG,IAAI,CAAC,MAAM,CACtC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,GAAG,CAAC,eAAe,EAAE,MAAM,IAAI,CAAC,CAAC,EAC1D,CAAC,CACF,CAAC;IACF,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,oBAAoB,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5F,OAAO;QACL,WAAW,EAAE,IAAI,CAAC,MAAM;QACxB,eAAe;QACf,gBAAgB,EAAE,IAAI,CAAC,MAAM;QAC7B,oBAAoB;QACpB,cAAc;QACd,qBAAqB,EAAE,eAAe,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,eAAe,CAAC,CAAC,CAAC,QAAQ;KACzF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,GAAiB,EACjB,QAAiC,kCAAkC;IAEnE,MAAM,MAAM,GAAG,GAAG,CAAC,eAAe,IAAI,EAAE,CAAC;IACzC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,KAAK,CAAC,IAAI,CAAC;IACpB,CAAC;IAED,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,+BAA+B,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;AAC/F,CAAC;AAED,MAAM,UAAU,+BAA+B,CAC7C,KAA2B,EAC3B,QAAiC,kCAAkC;IAEnE,IAAI,KAAK,CAAC,KAAK,KAAK,OAAO,EAAE,CAAC;QAC5B,OAAO,KAAK,CAAC,KAAK,CAAC;IACrB,CAAC;IACD,IAAI,KAAK,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC7B,OAAO,KAAK,CAAC,MAAM,CAAC;IACtB,CAAC;IACD,IAAI,KAAK,CAAC,KAAK,KAAK,aAAa,EAAE,CAAC;QAClC,OAAO,KAAK,CAAC,WAAW,CAAC;IAC3B,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC;AACpB,CAAC;AAED,SAAS,4BAA4B,CAAC,IAAoB;IACxD,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,QAAQ,GACZ,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,CAAC;QACnF,IAAI,QAAQ,EAAE,gBAAgB,EAAE,CAAC;YAC/B,QAAQ,IAAI,CAAC,CAAC;QAChB,CAAC;IACH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC"}

package/dist/checks/boundaryMetrics.d.ts

@@ -0,0 +1,12 @@
+export interface BoundaryQualityMetrics {
+    edgePixelCount: number;
+    haloRisk: number;
+    strayNoiseRatio: number;
+    edgeSharpness: number;
+}
+export declare function computeBoundaryQualityMetrics(params: {
+    raw: Buffer;
+    channels: number;
+    width: number;
+    height: number;
+}): BoundaryQualityMetrics | undefined;

package/dist/checks/boundaryMetrics.js

@@ -0,0 +1,102 @@
+const CARDINAL_NEIGHBORS = [
+    [0, -1],
+    [-1, 0],
+    [1, 0],
+    [0, 1],
+];
+const ALL_NEIGHBORS = [
+    [-1, -1],
+    [0, -1],
+    [1, -1],
+    [-1, 0],
+    [1, 0],
+    [-1, 1],
+    [0, 1],
+    [1, 1],
+];
+export function computeBoundaryQualityMetrics(params) {
+    if (params.channels < 4 || params.width <= 0 || params.height <= 0) {
+        return undefined;
+    }
+    const alphaAt = (x, y) => {
+        return params.raw[(y * params.width + x) * params.channels + 3] ?? 0;
+    };
+    const rgbAt = (x, y) => {
+        const index = (y * params.width + x) * params.channels;
+        return {
+            r: params.raw[index] ?? 0,
+            g: params.raw[index + 1] ?? 0,
+            b: params.raw[index + 2] ?? 0,
+        };
+    };
+    let opaquePixels = 0;
+    let isolatedOpaquePixels = 0;
+    let edgePixels = 0;
+    let haloPixels = 0;
+    let edgeSharpnessTotal = 0;
+    let edgeSharpnessSamples = 0;
+    forEachPixel({ width: params.width, height: params.height }, (x, y) => {
+        const alpha = alphaAt(x, y);
+        if (alpha <= 0) {
+            return;
+        }
+        opaquePixels += 1;
+        let transparentNeighborCount = 0;
+        for (const [dx, dy] of ALL_NEIGHBORS) {
+            const nx = x + dx;
+            const ny = y + dy;
+            if (!isWithinBounds({ width: params.width, height: params.height }, nx, ny)) {
+                transparentNeighborCount += 1;
+                continue;
+            }
+            if (alphaAt(nx, ny) === 0) {
+                transparentNeighborCount += 1;
+            }
+        }
+        if (transparentNeighborCount === ALL_NEIGHBORS.length) {
+            isolatedOpaquePixels += 1;
+        }
+        let isBoundary = false;
+        for (const [dx, dy] of CARDINAL_NEIGHBORS) {
+            const nx = x + dx;
+            const ny = y + dy;
+            if (!isWithinBounds({ width: params.width, height: params.height }, nx, ny)) {
+                isBoundary = true;
+                edgeSharpnessTotal += alpha / 255;
+                edgeSharpnessSamples += 1;
+                continue;
+            }
+            if (alphaAt(nx, ny) === 0) {
+                isBoundary = true;
+                edgeSharpnessTotal += alpha / 255;
+                edgeSharpnessSamples += 1;
+            }
+        }
+        if (!isBoundary) {
+            return;
+        }
+        edgePixels += 1;
+        const { r, g, b } = rgbAt(x, y);
+        const luma = 0.2126 * r + 0.7152 * g + 0.0722 * b;
+        if (alpha < 250 && luma > 180) {
+            haloPixels += 1;
+        }
+    });
+    return {
+        edgePixelCount: edgePixels,
+        haloRisk: edgePixels > 0 ? haloPixels / edgePixels : 0,
+        strayNoiseRatio: opaquePixels > 0 ? isolatedOpaquePixels / opaquePixels : 0,
+        edgeSharpness: edgeSharpnessSamples > 0 ? edgeSharpnessTotal / edgeSharpnessSamples : 0,
+    };
+}
+function forEachPixel(bounds, fn) {
+    for (let y = 0; y < bounds.height; y += 1) {
+        for (let x = 0; x < bounds.width; x += 1) {
+            fn(x, y);
+        }
+    }
+}
+function isWithinBounds(bounds, x, y) {
+    return x >= 0 && y >= 0 && x < bounds.width && y < bounds.height;
+}
+//# sourceMappingURL=boundaryMetrics.js.map

package/dist/checks/boundaryMetrics.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"boundaryMetrics.js","sourceRoot":"","sources":["../../src/checks/boundaryMetrics.ts"],"names":[],"mappings":"AAYA,MAAM,kBAAkB,GAAwC;IAC9D,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACP,CAAC,CAAC,EAAE,CAAC,CAAC;IACN,CAAC,CAAC,EAAE,CAAC,CAAC;CACP,CAAC;AAEF,MAAM,aAAa,GAAwC;IACzD,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACR,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACP,CAAC,CAAC,EAAE,CAAC,CAAC;IACN,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACP,CAAC,CAAC,EAAE,CAAC,CAAC;IACN,CAAC,CAAC,EAAE,CAAC,CAAC;CACP,CAAC;AAEF,MAAM,UAAU,6BAA6B,CAAC,MAK7C;IACC,IAAI,MAAM,CAAC,QAAQ,GAAG,CAAC,IAAI,MAAM,CAAC,KAAK,IAAI,CAAC,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;QACnE,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,CAAC,CAAS,EAAE,CAAS,EAAU,EAAE;QAC/C,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;IACvE,CAAC,CAAC;IACF,MAAM,KAAK,GAAG,CAAC,CAAS,EAAE,CAAS,EAAuC,EAAE;QAC1E,MAAM,KAAK,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC;QACvD,OAAO;YACL,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;YACzB,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC;YAC7B,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC;SAC9B,CAAC;IACJ,CAAC,CAAC;IAEF,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAC7B,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAC3B,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAE7B,YAAY,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACpE,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QAED,YAAY,IAAI,CAAC,CAAC;QAElB,IAAI,wBAAwB,GAAG,CAAC,CAAC;QACjC,KAAK,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,aAAa,EAAE,CAAC;YACrC,MAAM,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC;YAClB,MAAM,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC;YAClB,IAAI,CAAC,cAAc,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBAC5E,wBAAwB,IAAI,CAAC,CAAC;gBAC9B,SAAS;YACX,CAAC;YACD,IAAI,OAAO,CAAC,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,wBAAwB,IAAI,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;QACD,IAAI,wBAAwB,KAAK,aAAa,CAAC,MAAM,EAAE,CAAC;YACtD,oBAAoB,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,IAAI,UAAU,GAAG,KAAK,CAAC;QACvB,KAAK,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,kBAAkB,EAAE,CAAC;YAC1C,MAAM,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC;YAClB,MAAM,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC;YAClB,IAAI,CAAC,cAAc,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBAC5E,UAAU,GAAG,IAAI,CAAC;gBAClB,kBAAkB,IAAI,KAAK,GAAG,GAAG,CAAC;gBAClC,oBAAoB,IAAI,CAAC,CAAC;gBAC1B,SAAS;YACX,CAAC;YACD,IAAI,OAAO,CAAC,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,UAAU,GAAG,IAAI,CAAC;gBAClB,kBAAkB,IAAI,KAAK,GAAG,GAAG,CAAC;gBAClC,oBAAoB,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QAED,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,UAAU,IAAI,CAAC,CAAC;QAChB,MAAM,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAChC,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,GAAG,MAAM,GAAG,CAAC,GAAG,MAAM,GAAG,CAAC,CAAC;QAClD,IAAI,KAAK,GAAG,GAAG,IAAI,IAAI,GAAG,GAAG,EAAE,CAAC;YAC9B,UAAU,IAAI,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,cAAc,EAAE,UAAU;QAC1B,QAAQ,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;QACtD,eAAe,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC,oBAAoB,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;QAC3E,aAAa,EAAE,oBAAoB,GAAG,CAAC,CAAC,CAAC,CAAC,kBAAkB,GAAG,oBAAoB,CAAC,CAAC,CAAC,CAAC;KACxF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,MAAmB,EAAE,EAAkC;IAC3E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QAC1C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YACzC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACX,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,MAAmB,EAAE,CAAS,EAAE,CAAS;IAC/D,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;AACnE,CAAC"}

package/dist/checks/candidateScore.d.ts

@@ -0,0 +1,11 @@
+import { type CandidateScoreRecord, type PlannedTarget } from "../providers/types.js";
+export interface ScoreCandidateImagesOptions {
+    outDir?: string;
+    includeSoftAdapters?: boolean;
+    includeVlmGate?: boolean;
+    includePaletteCheck?: boolean;
+}
+export declare function scoreCandidateImages(target: PlannedTarget, outputPaths: string[], options?: ScoreCandidateImagesOptions): Promise<{
+    bestPath: string;
+    scores: CandidateScoreRecord[];
+}>;