import { describe, expect, it } from "vitest";
import {
	backgroundMaskPreview,
	backgroundRemovalMask,
	removeBackground,
} from "./background";
import { makeImage } from "./test-helpers";

const GREEN = [0, 255, 0, 255];
const RED = [255, 0, 0, 255];

describe("backgroundRemovalMask", () => {
	it("глобальный режим удаляет все совпадающие пиксели", () => {
		const mask = backgroundRemovalMask(makeImage(2, 1, [GREEN, RED]), {
			color: "#00ff00",
			tolerancePercent: 0,
			outerOnly: false,
			smoothPasses: 0,
		});
		expect([...mask]).toEqual([1, 0]);
	});

	describe("режим внешних областей", () => {
		const ringRedCenterGreen = [RED, RED, RED, RED, GREEN, RED, RED, RED, RED];

		it("заливка от краёв не достаёт до изолированного совпадающего острова", () => {
			const mask = backgroundRemovalMask(makeImage(3, 3, ringRedCenterGreen), {
				color: "#00ff00",
				tolerancePercent: 0,
				outerOnly: true,
				smoothPasses: 0,
			});
			expect(mask[4]).toBe(0);
		});

		it("глобальный режим удаляет и изолированный остров", () => {
			const mask = backgroundRemovalMask(makeImage(3, 3, ringRedCenterGreen), {
				color: "#00ff00",
				tolerancePercent: 0,
				outerOnly: false,
				smoothPasses: 0,
			});
			expect(mask[4]).toBe(1);
			expect(mask[0]).toBe(0);
		});
	});

	it("допуск расширяет захват по цветовому расстоянию", () => {
		const img = makeImage(2, 1, [
			[10, 10, 10, 255],
			[128, 128, 128, 255],
		]);
		const tight = backgroundRemovalMask(img, {
			color: "#000000",
			tolerancePercent: 40,
			outerOnly: false,
			smoothPasses: 0,
		});
		const wide = backgroundRemovalMask(img, {
			color: "#000000",
			tolerancePercent: 60,
			outerOnly: false,
			smoothPasses: 0,
		});
		expect(tight[1]).toBe(0);
		expect(wide[1]).toBe(1);
	});
});

describe("smoothMask-поведение через backgroundRemovalMask", () => {
	const ringRedCenterGreen = [RED, RED, RED, RED, GREEN, RED, RED, RED, RED];
	const opts = { color: "#00ff00", tolerancePercent: 0, outerOnly: false };

	const alphaAtCenter = (img: { data: Uint8ClampedArray }) => img.data[19];

	it("без сглаживания центр удалён", () => {
		const img = removeBackground(makeImage(3, 3, ringRedCenterGreen), {
			...opts,
			smoothPasses: 0,
		});
		expect(alphaAtCenter(img)).toBe(0);
	});

	it("два прохода мажоритарного фильтра возвращают изолированный пиксель", () => {
		const img = removeBackground(makeImage(3, 3, ringRedCenterGreen), {
			...opts,
			smoothPasses: 2,
		});
		expect(alphaAtCenter(img)).toBe(255);
	});
});

describe("removeBackground", () => {
	it("обнуляет альфу удалённых, сохраняет RGB остальных", () => {
		const out = removeBackground(makeImage(2, 1, [GREEN, [5, 6, 7, 200]]), {
			color: "#00ff00",
			tolerancePercent: 0,
			outerOnly: false,
			smoothPasses: 0,
		});
		expect(out.data[3]).toBe(0);
		expect([...out.data.slice(4, 8)]).toEqual([5, 6, 7, 200]);
	});
});

describe("backgroundMaskPreview", () => {
	it("белое там, где удаление, чёрное — где остаёмся, всё непрозрачно", () => {
		const preview = backgroundMaskPreview(
			makeImage(2, 1, [GREEN, [9, 9, 9, 60]]),
			{
				color: "#00ff00",
				tolerancePercent: 0,
				outerOnly: false,
				smoothPasses: 0,
			},
		);
		expect([...preview.data]).toEqual([255, 255, 255, 255, 0, 0, 0, 255]);
	});
});