feat: add compression tools

This commit is contained in:
2026-08-26 18:06:03 +05:00
parent 501d2dda4e
commit 83d3ce1198
8 changed files with 189 additions and 8 deletions
+39
View File
@@ -0,0 +1,39 @@
import { describe, expect, it } from 'vitest';
import { COMPRESSION_LEVELS, findMaxColorsWithin } from './compress';
describe('COMPRESSION_LEVELS', () => {
it('пресеты упорядочены по убыванию цветов', () => {
const values = Object.values(COMPRESSION_LEVELS);
for (let i = 1; i < values.length; i++) expect(values[i - 1]).toBeGreaterThan(values[i]);
});
});
describe('findMaxColorsWithin', () => {
const sizeOf = (k: number) => 1000 + k * 100; // размер растёт с k
it('подбирает максимум k, укладывающийся в цель', async () => {
// target 5200 → подходит k≤42 → ожидаем 42 при maxK≥42
const k = await findMaxColorsWithin(5200, 256, async (kk) => sizeOf(kk));
expect(k).toBeGreaterThanOrEqual(40);
expect(sizeOf(k)).toBeLessThanOrEqual(5200);
});
it('цель достигается даже на минимуме — возвращает 2', async () => {
const k = await findMaxColorsWithin(50, 256, async (kk) => sizeOf(kk));
expect(k).toBe(2);
});
it('encodeSize null трактуется как провал', async () => {
const k = await findMaxColorsWithin(999999, 16, async () => null);
expect(k).toBe(2);
});
it('бинарный поиск сходится быстрее полного перебора', async () => {
let calls = 0;
await findMaxColorsWithin(30000, 256, async (kk) => {
calls++;
return sizeOf(kk);
});
expect(calls).toBeLessThanOrEqual(12); // log2(254)+probe(2)
});
});
+46
View File
@@ -0,0 +1,46 @@
/** Пресеты сжатия: уровень → число цветов квантования. */
export const COMPRESSION_LEVELS = {
light: 192,
balanced: 96,
strong: 44,
extreme: 16
} as const;
export type CompressionLevel = keyof typeof COMPRESSION_LEVELS;
/**
* Бинарный поиск наибольшего k ∈ [2..maxK], при котором закодированный размер
* укладывается в targetBytes. Если даже k=2 не влезает — возвращается 2 (best effort).
* encodeSize может вернуть null (ошибка кодирования) — трактуется как «не влезло».
*/
export async function findMaxColorsWithin(
targetBytes: number,
maxK: number,
encodeSize: (k: number) => Promise<number | null>
): Promise<number> {
const hi = Math.max(2, Math.round(maxK));
let ok = 2;
let fitsAtTwo = false;
const probe = async (k: number): Promise<boolean> => {
const size = await encodeSize(Math.max(2, Math.min(hi, k)));
if (size === null) return false;
if (size <= targetBytes) {
ok = k;
fitsAtTwo = fitsAtTwo || k === 2;
return true;
}
return false;
};
if (!(await probe(2))) return 2;
let lo = 2;
let hiK = hi;
while (hiK - lo > 1) {
const mid = Math.floor((lo + hiK) / 2);
if (await probe(mid)) lo = mid;
else hiK = mid;
}
await probe(lo);
void fitsAtTwo;
return ok;
}