@pilio/gemini-watermark-remover 1.0.45 → 1.0.46

package/src/core/pipelineMetrics.js

@@ -1,88 +1,88 @@
-import {
-    computeRegionGradientCorrelation,
-    computeRegionSpatialCorrelation
-} from './adaptiveDetector.js';
-import { calculateNearBlackRatio } from './candidateSelector.js';
-
-const FIRST_PASS_SIGN_FLIP_GRADIENT_THRESHOLD = 0.08;
-const FIRST_PASS_SIGN_FLIP_MIN_GRADIENT_DROP = 0.2;
-
-export function shouldStopAfterFirstPass({
-    originalSpatialScore,
-    originalGradientScore,
-    firstPassSpatialScore,
-    firstPassGradientScore
-}) {
-    if (Math.abs(firstPassSpatialScore) <= 0.25) {
-        return true;
-    }
-
-    return originalSpatialScore >= 0 &&
-        firstPassSpatialScore < 0 &&
-        firstPassGradientScore <= FIRST_PASS_SIGN_FLIP_GRADIENT_THRESHOLD &&
-        (originalGradientScore - firstPassGradientScore) >= FIRST_PASS_SIGN_FLIP_MIN_GRADIENT_DROP;
-}
-
-export function createRegionCorrelationMetrics({
-    imageData,
-    alphaMap,
-    position,
-    includeNearBlackRatio = false
-} = {}) {
-    const region = { x: position.x, y: position.y, size: position.width };
-    const metrics = {
-        spatialScore: computeRegionSpatialCorrelation({
-            imageData,
-            alphaMap,
-            region
-        }),
-        gradientScore: computeRegionGradientCorrelation({
-            imageData,
-            alphaMap,
-            region
-        })
-    };
-    if (includeNearBlackRatio) {
-        metrics.nearBlackRatio = calculateNearBlackRatio(imageData, position);
-    }
-    return metrics;
-}
-
-export function createFirstPassMetrics({
-    imageData,
-    alphaMap,
-    position,
-    originalSpatialScore,
-    originalGradientScore
-} = {}) {
-    const firstPassMetrics = createRegionCorrelationMetrics({
-        imageData,
-        alphaMap,
-        position,
-        includeNearBlackRatio: true
-    });
-    const clearedResidual = shouldStopAfterFirstPass({
-        originalSpatialScore,
-        originalGradientScore,
-        firstPassSpatialScore: firstPassMetrics.spatialScore,
-        firstPassGradientScore: firstPassMetrics.gradientScore
-    });
-
-    return {
-        spatialScore: firstPassMetrics.spatialScore,
-        gradientScore: firstPassMetrics.gradientScore,
-        nearBlackRatio: firstPassMetrics.nearBlackRatio,
-        clearedResidual,
-        passStopReason: clearedResidual ? 'residual-low' : 'single-pass',
-        passRecord: {
-            index: 1,
-            beforeSpatialScore: originalSpatialScore,
-            beforeGradientScore: originalGradientScore,
-            afterSpatialScore: firstPassMetrics.spatialScore,
-            afterGradientScore: firstPassMetrics.gradientScore,
-            improvement: Math.abs(originalSpatialScore) - Math.abs(firstPassMetrics.spatialScore),
-            gradientDelta: firstPassMetrics.gradientScore - originalGradientScore,
-            nearBlackRatio: firstPassMetrics.nearBlackRatio
-        }
-    };
-}
+import {
+    computeRegionGradientCorrelation,
+    computeRegionSpatialCorrelation
+} from './adaptiveDetector.js';
+import { calculateNearBlackRatio } from './candidateSelector.js';
+
+const FIRST_PASS_SIGN_FLIP_GRADIENT_THRESHOLD = 0.08;
+const FIRST_PASS_SIGN_FLIP_MIN_GRADIENT_DROP = 0.2;
+
+export function shouldStopAfterFirstPass({
+    originalSpatialScore,
+    originalGradientScore,
+    firstPassSpatialScore,
+    firstPassGradientScore
+}) {
+    if (Math.abs(firstPassSpatialScore) <= 0.25) {
+        return true;
+    }
+
+    return originalSpatialScore >= 0 &&
+        firstPassSpatialScore < 0 &&
+        firstPassGradientScore <= FIRST_PASS_SIGN_FLIP_GRADIENT_THRESHOLD &&
+        (originalGradientScore - firstPassGradientScore) >= FIRST_PASS_SIGN_FLIP_MIN_GRADIENT_DROP;
+}
+
+export function createRegionCorrelationMetrics({
+    imageData,
+    alphaMap,
+    position,
+    includeNearBlackRatio = false
+} = {}) {
+    const region = { x: position.x, y: position.y, size: position.width };
+    const metrics = {
+        spatialScore: computeRegionSpatialCorrelation({
+            imageData,
+            alphaMap,
+            region
+        }),
+        gradientScore: computeRegionGradientCorrelation({
+            imageData,
+            alphaMap,
+            region
+        })
+    };
+    if (includeNearBlackRatio) {
+        metrics.nearBlackRatio = calculateNearBlackRatio(imageData, position);
+    }
+    return metrics;
+}
+
+export function createFirstPassMetrics({
+    imageData,
+    alphaMap,
+    position,
+    originalSpatialScore,
+    originalGradientScore
+} = {}) {
+    const firstPassMetrics = createRegionCorrelationMetrics({
+        imageData,
+        alphaMap,
+        position,
+        includeNearBlackRatio: true
+    });
+    const clearedResidual = shouldStopAfterFirstPass({
+        originalSpatialScore,
+        originalGradientScore,
+        firstPassSpatialScore: firstPassMetrics.spatialScore,
+        firstPassGradientScore: firstPassMetrics.gradientScore
+    });
+
+    return {
+        spatialScore: firstPassMetrics.spatialScore,
+        gradientScore: firstPassMetrics.gradientScore,
+        nearBlackRatio: firstPassMetrics.nearBlackRatio,
+        clearedResidual,
+        passStopReason: clearedResidual ? 'residual-low' : 'single-pass',
+        passRecord: {
+            index: 1,
+            beforeSpatialScore: originalSpatialScore,
+            beforeGradientScore: originalGradientScore,
+            afterSpatialScore: firstPassMetrics.spatialScore,
+            afterGradientScore: firstPassMetrics.gradientScore,
+            improvement: Math.abs(originalSpatialScore) - Math.abs(firstPassMetrics.spatialScore),
+            gradientDelta: firstPassMetrics.gradientScore - originalGradientScore,
+            nearBlackRatio: firstPassMetrics.nearBlackRatio
+        }
+    };
+}

package/src/core/pipelinePassState.js

@@ -1,36 +1,36 @@
-function finiteOrFallback(value, fallback) {
-    return typeof value === 'number' && Number.isFinite(value) ? value : fallback;
-}
-
-export function createEmptyPipelinePassState() {
-    return {
-        passCount: 0,
-        attemptedPassCount: 0,
-        passStopReason: null,
-        passes: null
-    };
-}
-
-export function createFirstPassPipelinePassState({
-    firstPassMetrics = null
-} = {}) {
-    return {
-        passCount: 1,
-        attemptedPassCount: 1,
-        passStopReason: firstPassMetrics?.passStopReason ?? null,
-        passes: [firstPassMetrics?.passRecord]
-    };
-}
-
-export function applyPipelinePassOutcome({
-    current,
-    outcome
-} = {}) {
-    const passIncrement = Math.max(0, finiteOrFallback(outcome?.passIncrement, 0));
-    return {
-        passCount: finiteOrFallback(current?.passCount, 0) + passIncrement,
-        attemptedPassCount: finiteOrFallback(current?.attemptedPassCount, 0) + passIncrement,
-        passStopReason: outcome?.passStopReason ?? current?.passStopReason ?? null,
-        passes: current?.passes ?? null
-    };
-}
+function finiteOrFallback(value, fallback) {
+    return typeof value === 'number' && Number.isFinite(value) ? value : fallback;
+}
+
+export function createEmptyPipelinePassState() {
+    return {
+        passCount: 0,
+        attemptedPassCount: 0,
+        passStopReason: null,
+        passes: null
+    };
+}
+
+export function createFirstPassPipelinePassState({
+    firstPassMetrics = null
+} = {}) {
+    return {
+        passCount: 1,
+        attemptedPassCount: 1,
+        passStopReason: firstPassMetrics?.passStopReason ?? null,
+        passes: [firstPassMetrics?.passRecord]
+    };
+}
+
+export function applyPipelinePassOutcome({
+    current,
+    outcome
+} = {}) {
+    const passIncrement = Math.max(0, finiteOrFallback(outcome?.passIncrement, 0));
+    return {
+        passCount: finiteOrFallback(current?.passCount, 0) + passIncrement,
+        attemptedPassCount: finiteOrFallback(current?.attemptedPassCount, 0) + passIncrement,
+        passStopReason: outcome?.passStopReason ?? current?.passStopReason ?? null,
+        passes: current?.passes ?? null
+    };
+}

package/src/core/pipelineRepairGates.js

@@ -1,154 +1,154 @@
-const DEFAULT_PREVIEW_EDGE_CLEANUP_MAX_SIZE = 32;
-const DEFAULT_KNOWN_48_EDGE_CLEANUP_MIN_SIZE = 40;
-const DEFAULT_KNOWN_48_EDGE_CLEANUP_MAX_SIZE = 56;
-const DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE = 36;
-const DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE_TOLERANCE = 2;
-
-export function shouldUsePreviewAnchorFastCleanup(
-    selectedTrial,
-    position,
-    { previewEdgeCleanupMaxSize = DEFAULT_PREVIEW_EDGE_CLEANUP_MAX_SIZE } = {}
-) {
-    return selectedTrial?.provenance?.previewAnchor === true &&
-        position?.width >= 24 &&
-        position?.width <= previewEdgeCleanupMaxSize;
-}
-
-export function isKnown48AnchorConfig(
-    config,
-    {
-        known48EdgeCleanupMinSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MIN_SIZE,
-        known48EdgeCleanupMaxSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MAX_SIZE
-    } = {}
-) {
-    if (!config || config.logoSize < known48EdgeCleanupMinSize || config.logoSize > known48EdgeCleanupMaxSize) {
-        return false;
-    }
-
-    const marginRight = Number(config.marginRight);
-    const marginBottom = Number(config.marginBottom);
-    if (!Number.isFinite(marginRight) || !Number.isFinite(marginBottom)) return false;
-
-    const isCurrentLargeMargin = Math.abs(marginRight - 96) <= 2 && Math.abs(marginBottom - 96) <= 2;
-    const isCurrentStandardMargin = marginRight >= 28 && marginRight <= 36 && marginBottom >= 28 && marginBottom <= 36;
-    return isCurrentLargeMargin || isCurrentStandardMargin;
-}
-
-export function shouldUseKnown48EdgeCleanup({
-    selectedTrial,
-    position,
-    source,
-    known48EdgeCleanupMinSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MIN_SIZE,
-    known48EdgeCleanupMaxSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MAX_SIZE
-} = {}) {
-    if (selectedTrial?.provenance?.previewAnchor === true) return false;
-    const sourceText = String(source || '');
-    const isStrongUndersizedAdaptive = shouldUseStrongUndersizedAdaptiveCleanup({
-        selectedTrial,
-        position,
-        source
-    });
-    if (
-        !isStrongUndersizedAdaptive &&
-        (position?.width < known48EdgeCleanupMinSize || position?.width > known48EdgeCleanupMaxSize)
-    ) return false;
-    if (!isStrongUndersizedAdaptive && !isKnown48AnchorConfig(selectedTrial?.config, {
-        known48EdgeCleanupMinSize,
-        known48EdgeCleanupMaxSize
-    })) return false;
-
-    if (isStrongUndersizedAdaptive) return true;
-    return sourceText === 'standard' ||
-        sourceText.startsWith('standard+gain') ||
-        sourceText.includes('catalog') ||
-        sourceText.includes('fixed-local');
-}
-
-export function shouldUseStrongUndersizedAdaptiveCleanup({
-    selectedTrial,
-    position,
-    source
-} = {}) {
-    if (selectedTrial?.provenance?.previewAnchor === true) return false;
-    const sourceText = String(source || '');
-    return (
-        selectedTrial?.provenance?.adaptive === true &&
-        selectedTrial?.provenance?.strongUndersizedMatch === true &&
-        position?.width >= 38 &&
-        position?.width <= 42 &&
-        sourceText.startsWith('adaptive')
-    );
-}
-
-export function isV2SmallAnchorConfig(
-    config,
-    { v2SmallEdgeCleanupSize = DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE } = {}
-) {
-    if (!config || config.logoSize !== v2SmallEdgeCleanupSize || config.alphaVariant !== 'v2') {
-        return false;
-    }
-
-    const marginRight = Number(config.marginRight);
-    const marginBottom = Number(config.marginBottom);
-    return Number.isFinite(marginRight) &&
-        Number.isFinite(marginBottom) &&
-        marginRight >= 48 &&
-        marginBottom >= 48;
-}
-
-export function shouldUseV2SmallEdgeCleanup({
-    selectedTrial,
-    position,
-    source,
-    v2SmallEdgeCleanupSize = DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE,
-    v2SmallEdgeCleanupSizeTolerance = DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE_TOLERANCE
-} = {}) {
-    if (selectedTrial?.provenance?.previewAnchor === true) return false;
-    if (
-        position?.width < v2SmallEdgeCleanupSize - v2SmallEdgeCleanupSizeTolerance ||
-        position?.width > v2SmallEdgeCleanupSize + v2SmallEdgeCleanupSizeTolerance
-    ) {
-        return false;
-    }
-    if (!isV2SmallAnchorConfig(selectedTrial?.config, { v2SmallEdgeCleanupSize })) return false;
-    if (selectedTrial?.provenance?.catalogFamily !== 'gemini-v2-small') return false;
-
-    const sourceText = String(source || '');
-    return sourceText.includes('catalog');
-}
-
-export function createRepairCleanupFlags({
-    selectedTrial,
-    position,
-    source,
-    previewEdgeCleanupMaxSize = DEFAULT_PREVIEW_EDGE_CLEANUP_MAX_SIZE,
-    known48EdgeCleanupMinSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MIN_SIZE,
-    known48EdgeCleanupMaxSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MAX_SIZE,
-    v2SmallEdgeCleanupSize = DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE,
-    v2SmallEdgeCleanupSizeTolerance = DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE_TOLERANCE
-} = {}) {
-    return {
-        usePreviewAnchorFastCleanup: shouldUsePreviewAnchorFastCleanup(selectedTrial, position, {
-            previewEdgeCleanupMaxSize
-        }),
-        useKnown48EdgeCleanup: shouldUseKnown48EdgeCleanup({
-            selectedTrial,
-            position,
-            source,
-            known48EdgeCleanupMinSize,
-            known48EdgeCleanupMaxSize
-        }),
-        useStrongUndersizedAdaptiveCleanup: shouldUseStrongUndersizedAdaptiveCleanup({
-            selectedTrial,
-            position,
-            source
-        }),
-        useV2SmallEdgeCleanup: shouldUseV2SmallEdgeCleanup({
-            selectedTrial,
-            position,
-            source,
-            v2SmallEdgeCleanupSize,
-            v2SmallEdgeCleanupSizeTolerance
-        })
-    };
-}
+const DEFAULT_PREVIEW_EDGE_CLEANUP_MAX_SIZE = 32;
+const DEFAULT_KNOWN_48_EDGE_CLEANUP_MIN_SIZE = 40;
+const DEFAULT_KNOWN_48_EDGE_CLEANUP_MAX_SIZE = 56;
+const DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE = 36;
+const DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE_TOLERANCE = 2;
+
+export function shouldUsePreviewAnchorFastCleanup(
+    selectedTrial,
+    position,
+    { previewEdgeCleanupMaxSize = DEFAULT_PREVIEW_EDGE_CLEANUP_MAX_SIZE } = {}
+) {
+    return selectedTrial?.provenance?.previewAnchor === true &&
+        position?.width >= 24 &&
+        position?.width <= previewEdgeCleanupMaxSize;
+}
+
+export function isKnown48AnchorConfig(
+    config,
+    {
+        known48EdgeCleanupMinSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MIN_SIZE,
+        known48EdgeCleanupMaxSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MAX_SIZE
+    } = {}
+) {
+    if (!config || config.logoSize < known48EdgeCleanupMinSize || config.logoSize > known48EdgeCleanupMaxSize) {
+        return false;
+    }
+
+    const marginRight = Number(config.marginRight);
+    const marginBottom = Number(config.marginBottom);
+    if (!Number.isFinite(marginRight) || !Number.isFinite(marginBottom)) return false;
+
+    const isCurrentLargeMargin = Math.abs(marginRight - 96) <= 2 && Math.abs(marginBottom - 96) <= 2;
+    const isCurrentStandardMargin = marginRight >= 28 && marginRight <= 36 && marginBottom >= 28 && marginBottom <= 36;
+    return isCurrentLargeMargin || isCurrentStandardMargin;
+}
+
+export function shouldUseKnown48EdgeCleanup({
+    selectedTrial,
+    position,
+    source,
+    known48EdgeCleanupMinSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MIN_SIZE,
+    known48EdgeCleanupMaxSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MAX_SIZE
+} = {}) {
+    if (selectedTrial?.provenance?.previewAnchor === true) return false;
+    const sourceText = String(source || '');
+    const isStrongUndersizedAdaptive = shouldUseStrongUndersizedAdaptiveCleanup({
+        selectedTrial,
+        position,
+        source
+    });
+    if (
+        !isStrongUndersizedAdaptive &&
+        (position?.width < known48EdgeCleanupMinSize || position?.width > known48EdgeCleanupMaxSize)
+    ) return false;
+    if (!isStrongUndersizedAdaptive && !isKnown48AnchorConfig(selectedTrial?.config, {
+        known48EdgeCleanupMinSize,
+        known48EdgeCleanupMaxSize
+    })) return false;
+
+    if (isStrongUndersizedAdaptive) return true;
+    return sourceText === 'standard' ||
+        sourceText.startsWith('standard+gain') ||
+        sourceText.includes('catalog') ||
+        sourceText.includes('fixed-local');
+}
+
+export function shouldUseStrongUndersizedAdaptiveCleanup({
+    selectedTrial,
+    position,
+    source
+} = {}) {
+    if (selectedTrial?.provenance?.previewAnchor === true) return false;
+    const sourceText = String(source || '');
+    return (
+        selectedTrial?.provenance?.adaptive === true &&
+        selectedTrial?.provenance?.strongUndersizedMatch === true &&
+        position?.width >= 38 &&
+        position?.width <= 42 &&
+        sourceText.startsWith('adaptive')
+    );
+}
+
+export function isV2SmallAnchorConfig(
+    config,
+    { v2SmallEdgeCleanupSize = DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE } = {}
+) {
+    if (!config || config.logoSize !== v2SmallEdgeCleanupSize || config.alphaVariant !== 'v2') {
+        return false;
+    }
+
+    const marginRight = Number(config.marginRight);
+    const marginBottom = Number(config.marginBottom);
+    return Number.isFinite(marginRight) &&
+        Number.isFinite(marginBottom) &&
+        marginRight >= 48 &&
+        marginBottom >= 48;
+}
+
+export function shouldUseV2SmallEdgeCleanup({
+    selectedTrial,
+    position,
+    source,
+    v2SmallEdgeCleanupSize = DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE,
+    v2SmallEdgeCleanupSizeTolerance = DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE_TOLERANCE
+} = {}) {
+    if (selectedTrial?.provenance?.previewAnchor === true) return false;
+    if (
+        position?.width < v2SmallEdgeCleanupSize - v2SmallEdgeCleanupSizeTolerance ||
+        position?.width > v2SmallEdgeCleanupSize + v2SmallEdgeCleanupSizeTolerance
+    ) {
+        return false;
+    }
+    if (!isV2SmallAnchorConfig(selectedTrial?.config, { v2SmallEdgeCleanupSize })) return false;
+    if (selectedTrial?.provenance?.catalogFamily !== 'gemini-v2-small') return false;
+
+    const sourceText = String(source || '');
+    return sourceText.includes('catalog');
+}
+
+export function createRepairCleanupFlags({
+    selectedTrial,
+    position,
+    source,
+    previewEdgeCleanupMaxSize = DEFAULT_PREVIEW_EDGE_CLEANUP_MAX_SIZE,
+    known48EdgeCleanupMinSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MIN_SIZE,
+    known48EdgeCleanupMaxSize = DEFAULT_KNOWN_48_EDGE_CLEANUP_MAX_SIZE,
+    v2SmallEdgeCleanupSize = DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE,
+    v2SmallEdgeCleanupSizeTolerance = DEFAULT_V2_SMALL_EDGE_CLEANUP_SIZE_TOLERANCE
+} = {}) {
+    return {
+        usePreviewAnchorFastCleanup: shouldUsePreviewAnchorFastCleanup(selectedTrial, position, {
+            previewEdgeCleanupMaxSize
+        }),
+        useKnown48EdgeCleanup: shouldUseKnown48EdgeCleanup({
+            selectedTrial,
+            position,
+            source,
+            known48EdgeCleanupMinSize,
+            known48EdgeCleanupMaxSize
+        }),
+        useStrongUndersizedAdaptiveCleanup: shouldUseStrongUndersizedAdaptiveCleanup({
+            selectedTrial,
+            position,
+            source
+        }),
+        useV2SmallEdgeCleanup: shouldUseV2SmallEdgeCleanup({
+            selectedTrial,
+            position,
+            source,
+            v2SmallEdgeCleanupSize,
+            v2SmallEdgeCleanupSizeTolerance
+        })
+    };
+}