external-repos/squoosh
1
0
Fork 1
mirror of https://github.com/GoogleChromeLabs/squoosh.git synced 2025-11-14 09:39:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reuse rotate instance and calculate pages correctly

Browse Source
This commit is contained in:
Surma
2019-02-11 16:14:37 +00:00
parent b6a8f7eeba
commit ef3faa58bc
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/codecs/rotate/processor.ts
View File

@@ -1,23 +1,30 @@
import wasmUrl from '../../../codecs/rotate/rotate.wasm'; import wasmUrl from '../../../codecs/rotate/rotate.wasm';
import { RotateOptions, RotateModuleInstance } from './processor-meta'; import { RotateOptions, RotateModuleInstance } from './processor-meta';
const instancePromise = (WebAssembly as any).instantiateStreaming(fetch(wasmUrl));
export async function rotate( export async function rotate(
data: ImageData, data: ImageData,
opts: RotateOptions, opts: RotateOptions,
): Promise<ImageData> { ): Promise<ImageData> {
const flipDimensions = opts.rotate % 180 !== 0; const { instance } = (await instancePromise) as {instance: RotateModuleInstance};
// Number of wasm memory pages (á 64KiB) needed to store the image twice. // Number of wasm memory pages (á 64KiB) needed to store the image twice.
const bytesPerImage = data.width * data.height * 4; const bytesPerImage = data.width * data.height * 4;
const numPagesNeeded = Math.ceil(bytesPerImage * 2 / (64 * 1024)); const numPagesNeeded = Math.ceil(bytesPerImage * 2 / (64 * 1024) + 4);
const { instance } = (await (WebAssembly as any).instantiateStreaming( // Only count full pages, just to be safe.
fetch(wasmUrl), const numPagesAvailable = Math.floor(instance.exports.memory.buffer.byteLength / (64 * 1024));
)) as { instance: RotateModuleInstance }; const additionalPagesToAllocate = numPagesNeeded - numPagesAvailable;
instance.exports.memory.grow(numPagesNeeded); if (additionalPagesToAllocate > 0) {
instance.exports.memory.grow(additionalPagesToAllocate);
}
const view = new Uint8ClampedArray(instance.exports.memory.buffer); const view = new Uint8ClampedArray(instance.exports.memory.buffer);
view.set(data.data); view.set(data.data);
instance.exports.rotate(data.width, data.height, opts.rotate); instance.exports.rotate(data.width, data.height, opts.rotate);
const flipDimensions = opts.rotate % 180 !== 0;
return new ImageData( return new ImageData(
view.slice(bytesPerImage, bytesPerImage * 2), view.slice(bytesPerImage, bytesPerImage * 2),
flipDimensions ? data.height : data.width, flipDimensions ? data.height : data.width,