1/**
2 * Compression measurements off the main thread, so the scrolling view never
3 * stutters while zstd -19 or the LPC fit runs. One message in (the model),
4 * one message out (the nine codec results).
5 */
6import { LatentSource } from '../model/latent'
7import {
8 initCodecs,
9 compressAll,
10 entropyBounds,
11 type BoundResult,
12 ZLIB,
13 ZSTD,
14 ANS,
15 DELTA_ZLIB,
16 DELTA_ZSTD,
17 DELTA_ANS,
18 lpcCodecs,
19 type CodecResult,
20} from '../compress/codecs'
22export interface CompressRequest {
23 id: number
24 kernel: Float64Array
25 sigma: number
26 dither: boolean
27 blockSize: number
28 lpcOrder: number
29 seed: number
30}
32export interface CompressResponse {
33 id: number
34 results: CodecResult[]
35 /** Order-0 entropy limit for each prefilter group. */
36 bounds: BoundResult[]
37 /** Empirical std of the quantized block, for display sanity. */
38 empiricalStd: number
39 error?: string
40}
42const PLAIN_CODECS = [ZLIB, ZSTD, ANS, DELTA_ZLIB, DELTA_ZSTD, DELTA_ANS]
44const post = self.postMessage as (message: CompressResponse) => void
46self.onmessage = async (e: MessageEvent<CompressRequest>) => {
47 const { id, kernel, sigma, dither, blockSize, lpcOrder, seed } = e.data
48 try {
49 await initCodecs()
50 const samples = new LatentSource(seed).window(0, blockSize, kernel, sigma, dither)
51 let sum = 0
52 let sumSq = 0
53 for (let i = 0; i < samples.length; i++) {
54 sum += samples[i]
55 sumSq += samples[i] * samples[i]
56 }
57 const mean = sum / samples.length
58 const empiricalStd = Math.sqrt(Math.max(0, sumSq / samples.length - mean * mean))
59 post({
60 id,
61 results: compressAll(samples, [...PLAIN_CODECS, ...lpcCodecs(lpcOrder)]),
62 bounds: entropyBounds(samples, lpcOrder),
63 empiricalStd,
64 })
65 } catch (err) {
66 post({ id, results: [], bounds: [], empiricalStd: 0, error: String(err) })
67 }
68}