AzuTear
66 lines
2.7 KiB
TypeScript
66 lines
2.7 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { detectLockState, isLocked, lockIconCropRect, lockSignalRatio } from "./lockDetection";
|
|
import type { Bitmap } from "./ocrPreprocess";
|
|
import { profileDetailRect } from "./layoutProfile";
|
|
|
|
// Build a BGRA bitmap where `goldPixels` of the pixels are lock-gold and the rest dark.
|
|
function bitmap(goldPixels: number, total: number): Bitmap {
|
|
const data = Buffer.alloc(total * 4);
|
|
for (let pixel = 0; pixel < total; pixel++) {
|
|
const index = pixel * 4;
|
|
if (pixel < goldPixels) {
|
|
data[index] = 40; // B
|
|
data[index + 1] = 170; // G
|
|
data[index + 2] = 230; // R -> gold
|
|
}
|
|
data[index + 3] = 255;
|
|
}
|
|
return { data, width: total, height: 1 };
|
|
}
|
|
|
|
function solidPixels(pixels: Array<{ b: number; g: number; r: number }>): Bitmap {
|
|
const data = Buffer.alloc(pixels.length * 4);
|
|
pixels.forEach((pixel, index) => {
|
|
const offset = index * 4;
|
|
data[offset] = pixel.b;
|
|
data[offset + 1] = pixel.g;
|
|
data[offset + 2] = pixel.r;
|
|
data[offset + 3] = 255;
|
|
});
|
|
return { data, width: pixels.length, height: 1 };
|
|
}
|
|
|
|
describe("lockDetection", () => {
|
|
it("places the lock crop on the lock button in the substat panel", () => {
|
|
const size = { width: 2560, height: 1440 };
|
|
const detail = profileDetailRect(size);
|
|
const rect = lockIconCropRect(detail, size);
|
|
expect(rect.x).toBeGreaterThan(detail.x + detail.width * 0.5);
|
|
expect(rect.x + rect.width).toBeLessThanOrEqual(size.width);
|
|
expect(rect.y).toBeGreaterThan(detail.y + detail.height * 0.3);
|
|
expect(rect.y).toBeLessThan(detail.y + detail.height * 0.45);
|
|
});
|
|
|
|
it("measures the gold-pixel ratio", () => {
|
|
expect(lockSignalRatio(bitmap(0, 100))).toBe(0);
|
|
expect(lockSignalRatio(bitmap(50, 100))).toBeCloseTo(0.5, 5);
|
|
expect(lockSignalRatio(bitmap(100, 100))).toBe(1);
|
|
});
|
|
|
|
it("thresholds the ratio into a locked flag", () => {
|
|
expect(isLocked(0.02)).toBe(false);
|
|
expect(isLocked(0.2)).toBe(true);
|
|
expect(detectLockState(bitmap(20, 100))).toBe(true);
|
|
expect(detectLockState(bitmap(1, 100))).toBe(false);
|
|
});
|
|
|
|
it("counts the current red lock glyph but ignores grey unlocked button pixels", () => {
|
|
expect(lockSignalRatio(solidPixels([{ b: 90, g: 92, r: 235 }]))).toBe(1);
|
|
expect(lockSignalRatio({ data: Buffer.from([235, 92, 90, 255]), width: 1, height: 1 })).toBe(1);
|
|
expect(lockSignalRatio({ data: Buffer.from([255, 235, 92, 90]), width: 1, height: 1 })).toBe(1);
|
|
expect(lockSignalRatio(solidPixels([{ b: 235, g: 235, r: 235 }]))).toBe(0);
|
|
expect(lockSignalRatio({ data: Buffer.from([255, 235, 235, 235]), width: 1, height: 1 })).toBe(0);
|
|
expect(lockSignalRatio(solidPixels([{ b: 120, g: 122, r: 128 }]))).toBe(0);
|
|
});
|
|
});
|