mirror of
https://github.com/Ku6epXBOCTuK/easy-png-tools.git
synced 2026-09-14 21:46:35 +00:00
feat: add compression tools
This commit is contained in:
@@ -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)
|
||||
});
|
||||
});
|
||||
@@ -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;
|
||||
}
|
||||
Reference in New Issue
Block a user