refactor: extract shared ContentMetrics for overlay position calculations (#26310)

This commit is contained in:
Min Idzelis
2026-03-02 22:49:56 -05:00
committed by GitHub
parent 6deb97d5bc
commit 20c639e52a
9 changed files with 415 additions and 162 deletions

View File

@@ -0,0 +1,116 @@
import type { OcrBoundingBox } from '$lib/stores/ocr.svelte';
import type { ContentMetrics } from '$lib/utils/container-utils';
import { getOcrBoundingBoxes } from '$lib/utils/ocr-utils';
describe('getOcrBoundingBoxes', () => {
it('should scale normalized coordinates by display dimensions', () => {
const ocrData: OcrBoundingBox[] = [
{
id: 'box1',
assetId: 'asset1',
x1: 0.1,
y1: 0.2,
x2: 0.9,
y2: 0.2,
x3: 0.9,
y3: 0.8,
x4: 0.1,
y4: 0.8,
boxScore: 0.95,
textScore: 0.9,
text: 'hello',
},
];
const metrics: ContentMetrics = { contentWidth: 1000, contentHeight: 500, offsetX: 0, offsetY: 0 };
const boxes = getOcrBoundingBoxes(ocrData, metrics);
expect(boxes).toHaveLength(1);
expect(boxes[0].id).toBe('box1');
expect(boxes[0].text).toBe('hello');
expect(boxes[0].confidence).toBe(0.9);
expect(boxes[0].points).toEqual([
{ x: 100, y: 100 },
{ x: 900, y: 100 },
{ x: 900, y: 400 },
{ x: 100, y: 400 },
]);
});
it('should apply offsets for letterboxed images', () => {
const ocrData: OcrBoundingBox[] = [
{
id: 'box1',
assetId: 'asset1',
x1: 0,
y1: 0,
x2: 1,
y2: 0,
x3: 1,
y3: 1,
x4: 0,
y4: 1,
boxScore: 0.9,
textScore: 0.8,
text: 'test',
},
];
const metrics: ContentMetrics = { contentWidth: 600, contentHeight: 400, offsetX: 100, offsetY: 50 };
const boxes = getOcrBoundingBoxes(ocrData, metrics);
expect(boxes[0].points).toEqual([
{ x: 100, y: 50 },
{ x: 700, y: 50 },
{ x: 700, y: 450 },
{ x: 100, y: 450 },
]);
});
it('should return empty array for empty input', () => {
const metrics: ContentMetrics = { contentWidth: 800, contentHeight: 600, offsetX: 0, offsetY: 0 };
expect(getOcrBoundingBoxes([], metrics)).toEqual([]);
});
it('should handle multiple boxes', () => {
const ocrData: OcrBoundingBox[] = [
{
id: 'a',
assetId: 'asset1',
x1: 0,
y1: 0,
x2: 0.5,
y2: 0,
x3: 0.5,
y3: 0.5,
x4: 0,
y4: 0.5,
boxScore: 0.9,
textScore: 0.8,
text: 'first',
},
{
id: 'b',
assetId: 'asset1',
x1: 0.5,
y1: 0.5,
x2: 1,
y2: 0.5,
x3: 1,
y3: 1,
x4: 0.5,
y4: 1,
boxScore: 0.9,
textScore: 0.7,
text: 'second',
},
];
const metrics: ContentMetrics = { contentWidth: 200, contentHeight: 200, offsetX: 0, offsetY: 0 };
const boxes = getOcrBoundingBoxes(ocrData, metrics);
expect(boxes).toHaveLength(2);
expect(boxes[0].text).toBe('first');
expect(boxes[1].text).toBe('second');
});
});