@draco-rust/decoder 0.1.1 → 0.1.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,5 +1,37 @@
1
1
  /* @ts-self-types="./index.d.ts" */
2
2
 
3
+ /**
4
+ * Decodes a Draco stream and returns its attributes as typed arrays.
5
+ *
6
+ * `attributes` is `undefined` for every attribute in its own component type,
7
+ * or an array of requests `{ name, id, semantic, type }`:
8
+ *
9
+ * - `id`, a unique id, finds that attribute, as glTF's
10
+ * `KHR_draco_mesh_compression` names them; a missing one is an error.
11
+ * - `semantic` (`"POSITION"`, `"NORMAL"`, `"COLOR"`, `"TEX_COORD"` or
12
+ * `"GENERIC"`) finds the first attribute of that type instead, as a `.drc`
13
+ * file is read; a missing one is left out.
14
+ * - `type`, a typed array's name such as `"Float32Array"` or
15
+ * `"Uint16Array"`, converts the values the way upstream's
16
+ * `GetAttributeDataArrayForAllPoints` does; without it the values keep
17
+ * their own type. A value the type cannot hold is an error.
18
+ *
19
+ * The result is `{ success, error, warnings, geometry, index, attributes }`,
20
+ * where `geometry` is `"mesh"` or `"point_cloud"` as the stream's header says,
21
+ * `index` a `Uint32Array` of triangle corners for a mesh and `null` for a
22
+ * point cloud, and each attribute `{ name, uniqueId, semantic, array,
23
+ * itemSize, normalized }`.
24
+ * @param {Uint8Array} data
25
+ * @param {DracoAttributeRequest[] | undefined} attributes
26
+ * @returns {DracoDecodeResult}
27
+ */
28
+ export function decode_draco(data, attributes) {
29
+ const ptr0 = passArray8ToWasm0(data, wasm.__wbindgen_export);
30
+ const len0 = WASM_VECTOR_LEN;
31
+ const ret = wasm.decode_draco(ptr0, len0, addHeapObject(attributes));
32
+ return takeObject(ret);
33
+ }
34
+
3
35
  /**
4
36
  * Install the panic hook, so a panic reads as a message rather than a trap.
5
37
  */
@@ -13,7 +45,7 @@ export function init() {
13
45
  * @returns {any}
14
46
  */
15
47
  export function parse_drc_bytes(data) {
16
- const ptr0 = passArray8ToWasm0(data, wasm.__wbindgen_export2);
48
+ const ptr0 = passArray8ToWasm0(data, wasm.__wbindgen_export);
17
49
  const len0 = WASM_VECTOR_LEN;
18
50
  const ret = wasm.parse_drc_bytes(ptr0, len0);
19
51
  return takeObject(ret);
@@ -30,7 +62,7 @@ export function supported_extensions() {
30
62
  var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
31
63
  var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
32
64
  var v1 = getArrayJsValueFromWasm0(r0, r1);
33
- wasm.__wbindgen_export3(r0, r1 * 4, 4);
65
+ wasm.__wbindgen_export4(r0, r1 * 4, 4);
34
66
  return v1;
35
67
  } finally {
36
68
  wasm.__wbindgen_add_to_stack_pointer(16);
@@ -54,15 +86,53 @@ export function version() {
54
86
  return getStringFromWasm0(r0, r1);
55
87
  } finally {
56
88
  wasm.__wbindgen_add_to_stack_pointer(16);
57
- wasm.__wbindgen_export3(deferred1_0, deferred1_1, 1);
89
+ wasm.__wbindgen_export4(deferred1_0, deferred1_1, 1);
58
90
  }
59
91
  }
60
92
  function __wbg_get_imports() {
61
93
  const import0 = {
62
94
  __proto__: null,
95
+ __wbg___wbindgen_is_null_e343b7d08827ba72: function(arg0) {
96
+ const ret = getObject(arg0) === null;
97
+ return ret;
98
+ },
99
+ __wbg___wbindgen_is_undefined_8865fb403f8fe9d8: function(arg0) {
100
+ const ret = getObject(arg0) === undefined;
101
+ return ret;
102
+ },
103
+ __wbg___wbindgen_number_get_2e0e7dee9f701a71: function(arg0, arg1) {
104
+ const obj = getObject(arg1);
105
+ const ret = typeof(obj) === 'number' ? obj : undefined;
106
+ getDataViewMemory0().setFloat64(arg0 + 8 * 1, isLikeNone(ret) ? 0 : ret, true);
107
+ getDataViewMemory0().setInt32(arg0 + 4 * 0, !isLikeNone(ret), true);
108
+ },
109
+ __wbg___wbindgen_string_get_0380ccaa2f57f0d9: function(arg0, arg1) {
110
+ const obj = getObject(arg1);
111
+ const ret = typeof(obj) === 'string' ? obj : undefined;
112
+ var ptr1 = isLikeNone(ret) ? 0 : passStringToWasm0(ret, wasm.__wbindgen_export, wasm.__wbindgen_export2);
113
+ var len1 = WASM_VECTOR_LEN;
114
+ getDataViewMemory0().setInt32(arg0 + 4 * 1, len1, true);
115
+ getDataViewMemory0().setInt32(arg0 + 4 * 0, ptr1, true);
116
+ },
63
117
  __wbg___wbindgen_throw_41e9ee4f547fc59a: function(arg0, arg1) {
64
118
  throw new Error(getStringFromWasm0(arg0, arg1));
65
119
  },
120
+ __wbg_get_31af05bd4842a84f: function() { return handleError(function (arg0, arg1) {
121
+ const ret = Reflect.get(getObject(arg0), getObject(arg1));
122
+ return addHeapObject(ret);
123
+ }, arguments); },
124
+ __wbg_get_unchecked_288889d017702237: function(arg0, arg1) {
125
+ const ret = getObject(arg0)[arg1 >>> 0];
126
+ return addHeapObject(ret);
127
+ },
128
+ __wbg_isArray_e15a2ff68ffdbef2: function(arg0) {
129
+ const ret = Array.isArray(getObject(arg0));
130
+ return ret;
131
+ },
132
+ __wbg_length_d4bdea10311bd9cf: function(arg0) {
133
+ const ret = getObject(arg0).length;
134
+ return ret;
135
+ },
66
136
  __wbg_new_617a8cdb8bb1130e: function() {
67
137
  const ret = new Object();
68
138
  return addHeapObject(ret);
@@ -71,14 +141,34 @@ function __wbg_get_imports() {
71
141
  const ret = new Array();
72
142
  return addHeapObject(ret);
73
143
  },
144
+ __wbg_new_from_slice_23f60f47cde8d664: function(arg0, arg1) {
145
+ const ret = new Uint16Array(getArrayU16FromWasm0(arg0, arg1));
146
+ return addHeapObject(ret);
147
+ },
148
+ __wbg_new_from_slice_2443ca2017fac0fd: function(arg0, arg1) {
149
+ const ret = new Int16Array(getArrayI16FromWasm0(arg0, arg1));
150
+ return addHeapObject(ret);
151
+ },
152
+ __wbg_new_from_slice_36f6b4e63daa7f59: function(arg0, arg1) {
153
+ const ret = new Int8Array(getArrayI8FromWasm0(arg0, arg1));
154
+ return addHeapObject(ret);
155
+ },
74
156
  __wbg_new_from_slice_37a3c3a194266cb8: function(arg0, arg1) {
75
157
  const ret = new Uint32Array(getArrayU32FromWasm0(arg0, arg1));
76
158
  return addHeapObject(ret);
77
159
  },
160
+ __wbg_new_from_slice_911f717853b222c7: function(arg0, arg1) {
161
+ const ret = new Int32Array(getArrayI32FromWasm0(arg0, arg1));
162
+ return addHeapObject(ret);
163
+ },
78
164
  __wbg_new_from_slice_9a868026ffa4208a: function(arg0, arg1) {
79
165
  const ret = new Uint8Array(getArrayU8FromWasm0(arg0, arg1));
80
166
  return addHeapObject(ret);
81
167
  },
168
+ __wbg_new_from_slice_aa8f37bd5e72dcc7: function(arg0, arg1) {
169
+ const ret = new BigInt64Array(getArrayI64FromWasm0(arg0, arg1));
170
+ return addHeapObject(ret);
171
+ },
82
172
  __wbg_new_from_slice_ca6ad97db1f4779a: function(arg0, arg1) {
83
173
  const ret = new Float32Array(getArrayF32FromWasm0(arg0, arg1));
84
174
  return addHeapObject(ret);
@@ -87,6 +177,10 @@ function __wbg_get_imports() {
87
177
  const ret = new Float64Array(getArrayF64FromWasm0(arg0, arg1));
88
178
  return addHeapObject(ret);
89
179
  },
180
+ __wbg_new_from_slice_f6cd96ab370a57b3: function(arg0, arg1) {
181
+ const ret = new BigUint64Array(getArrayU64FromWasm0(arg0, arg1));
182
+ return addHeapObject(ret);
183
+ },
90
184
  __wbg_push_2baf45db356cf468: function(arg0, arg1) {
91
185
  const ret = getObject(arg0).push(getObject(arg1));
92
186
  return ret;
@@ -140,6 +234,26 @@ function getArrayF64FromWasm0(ptr, len) {
140
234
  return getFloat64ArrayMemory0().subarray(ptr / 8, ptr / 8 + len);
141
235
  }
142
236
 
237
+ function getArrayI16FromWasm0(ptr, len) {
238
+ ptr = ptr >>> 0;
239
+ return getInt16ArrayMemory0().subarray(ptr / 2, ptr / 2 + len);
240
+ }
241
+
242
+ function getArrayI32FromWasm0(ptr, len) {
243
+ ptr = ptr >>> 0;
244
+ return getInt32ArrayMemory0().subarray(ptr / 4, ptr / 4 + len);
245
+ }
246
+
247
+ function getArrayI64FromWasm0(ptr, len) {
248
+ ptr = ptr >>> 0;
249
+ return getBigInt64ArrayMemory0().subarray(ptr / 8, ptr / 8 + len);
250
+ }
251
+
252
+ function getArrayI8FromWasm0(ptr, len) {
253
+ ptr = ptr >>> 0;
254
+ return getInt8ArrayMemory0().subarray(ptr / 1, ptr / 1 + len);
255
+ }
256
+
143
257
  function getArrayJsValueFromWasm0(ptr, len) {
144
258
  ptr = ptr >>> 0;
145
259
  const mem = getDataViewMemory0();
@@ -150,16 +264,42 @@ function getArrayJsValueFromWasm0(ptr, len) {
150
264
  return result;
151
265
  }
152
266
 
267
+ function getArrayU16FromWasm0(ptr, len) {
268
+ ptr = ptr >>> 0;
269
+ return getUint16ArrayMemory0().subarray(ptr / 2, ptr / 2 + len);
270
+ }
271
+
153
272
  function getArrayU32FromWasm0(ptr, len) {
154
273
  ptr = ptr >>> 0;
155
274
  return getUint32ArrayMemory0().subarray(ptr / 4, ptr / 4 + len);
156
275
  }
157
276
 
277
+ function getArrayU64FromWasm0(ptr, len) {
278
+ ptr = ptr >>> 0;
279
+ return getBigUint64ArrayMemory0().subarray(ptr / 8, ptr / 8 + len);
280
+ }
281
+
158
282
  function getArrayU8FromWasm0(ptr, len) {
159
283
  ptr = ptr >>> 0;
160
284
  return getUint8ArrayMemory0().subarray(ptr / 1, ptr / 1 + len);
161
285
  }
162
286
 
287
+ let cachedBigInt64ArrayMemory0 = null;
288
+ function getBigInt64ArrayMemory0() {
289
+ if (cachedBigInt64ArrayMemory0 === null || cachedBigInt64ArrayMemory0.byteLength === 0) {
290
+ cachedBigInt64ArrayMemory0 = new BigInt64Array(wasm.memory.buffer);
291
+ }
292
+ return cachedBigInt64ArrayMemory0;
293
+ }
294
+
295
+ let cachedBigUint64ArrayMemory0 = null;
296
+ function getBigUint64ArrayMemory0() {
297
+ if (cachedBigUint64ArrayMemory0 === null || cachedBigUint64ArrayMemory0.byteLength === 0) {
298
+ cachedBigUint64ArrayMemory0 = new BigUint64Array(wasm.memory.buffer);
299
+ }
300
+ return cachedBigUint64ArrayMemory0;
301
+ }
302
+
163
303
  let cachedDataViewMemory0 = null;
164
304
  function getDataViewMemory0() {
165
305
  if (cachedDataViewMemory0 === null || cachedDataViewMemory0.buffer.detached === true || (cachedDataViewMemory0.buffer.detached === undefined && cachedDataViewMemory0.buffer !== wasm.memory.buffer)) {
@@ -184,10 +324,42 @@ function getFloat64ArrayMemory0() {
184
324
  return cachedFloat64ArrayMemory0;
185
325
  }
186
326
 
327
+ let cachedInt16ArrayMemory0 = null;
328
+ function getInt16ArrayMemory0() {
329
+ if (cachedInt16ArrayMemory0 === null || cachedInt16ArrayMemory0.byteLength === 0) {
330
+ cachedInt16ArrayMemory0 = new Int16Array(wasm.memory.buffer);
331
+ }
332
+ return cachedInt16ArrayMemory0;
333
+ }
334
+
335
+ let cachedInt32ArrayMemory0 = null;
336
+ function getInt32ArrayMemory0() {
337
+ if (cachedInt32ArrayMemory0 === null || cachedInt32ArrayMemory0.byteLength === 0) {
338
+ cachedInt32ArrayMemory0 = new Int32Array(wasm.memory.buffer);
339
+ }
340
+ return cachedInt32ArrayMemory0;
341
+ }
342
+
343
+ let cachedInt8ArrayMemory0 = null;
344
+ function getInt8ArrayMemory0() {
345
+ if (cachedInt8ArrayMemory0 === null || cachedInt8ArrayMemory0.byteLength === 0) {
346
+ cachedInt8ArrayMemory0 = new Int8Array(wasm.memory.buffer);
347
+ }
348
+ return cachedInt8ArrayMemory0;
349
+ }
350
+
187
351
  function getStringFromWasm0(ptr, len) {
188
352
  return decodeText(ptr >>> 0, len);
189
353
  }
190
354
 
355
+ let cachedUint16ArrayMemory0 = null;
356
+ function getUint16ArrayMemory0() {
357
+ if (cachedUint16ArrayMemory0 === null || cachedUint16ArrayMemory0.byteLength === 0) {
358
+ cachedUint16ArrayMemory0 = new Uint16Array(wasm.memory.buffer);
359
+ }
360
+ return cachedUint16ArrayMemory0;
361
+ }
362
+
191
363
  let cachedUint32ArrayMemory0 = null;
192
364
  function getUint32ArrayMemory0() {
193
365
  if (cachedUint32ArrayMemory0 === null || cachedUint32ArrayMemory0.byteLength === 0) {
@@ -210,7 +382,7 @@ function handleError(f, args) {
210
382
  try {
211
383
  return f.apply(this, args);
212
384
  } catch (e) {
213
- wasm.__wbindgen_export(addHeapObject(e));
385
+ wasm.__wbindgen_export3(addHeapObject(e));
214
386
  }
215
387
  }
216
388
 
@@ -219,6 +391,10 @@ heap.push(undefined, null, true, false);
219
391
 
220
392
  let heap_next = heap.length;
221
393
 
394
+ function isLikeNone(x) {
395
+ return x === undefined || x === null;
396
+ }
397
+
222
398
  function passArray8ToWasm0(arg, malloc) {
223
399
  const ptr = malloc(arg.length * 1, 1) >>> 0;
224
400
  getUint8ArrayMemory0().set(arg, ptr / 1);
@@ -226,6 +402,43 @@ function passArray8ToWasm0(arg, malloc) {
226
402
  return ptr;
227
403
  }
228
404
 
405
+ function passStringToWasm0(arg, malloc, realloc) {
406
+ if (realloc === undefined) {
407
+ const buf = cachedTextEncoder.encode(arg);
408
+ const ptr = malloc(buf.length, 1) >>> 0;
409
+ getUint8ArrayMemory0().subarray(ptr, ptr + buf.length).set(buf);
410
+ WASM_VECTOR_LEN = buf.length;
411
+ return ptr;
412
+ }
413
+
414
+ let len = arg.length;
415
+ let ptr = malloc(len, 1) >>> 0;
416
+
417
+ const mem = getUint8ArrayMemory0();
418
+
419
+ let offset = 0;
420
+
421
+ for (; offset < len; offset++) {
422
+ const code = arg.charCodeAt(offset);
423
+ if (code > 0x7F) break;
424
+ mem[ptr + offset] = code;
425
+ }
426
+ if (offset !== len) {
427
+ if (offset !== 0) {
428
+ arg = arg.slice(offset);
429
+ }
430
+ ptr = realloc(ptr, len, len = offset + arg.length * 3, 1) >>> 0;
431
+ const view = getUint8ArrayMemory0().subarray(ptr + offset, ptr + len);
432
+ const ret = cachedTextEncoder.encodeInto(arg, view);
433
+
434
+ offset += ret.written;
435
+ ptr = realloc(ptr, len, offset, 1) >>> 0;
436
+ }
437
+
438
+ WASM_VECTOR_LEN = offset;
439
+ return ptr;
440
+ }
441
+
229
442
  function takeObject(idx) {
230
443
  const ret = getObject(idx);
231
444
  dropObject(idx);
@@ -246,6 +459,19 @@ function decodeText(ptr, len) {
246
459
  return cachedTextDecoder.decode(getUint8ArrayMemory0().subarray(ptr, ptr + len));
247
460
  }
248
461
 
462
+ const cachedTextEncoder = new TextEncoder();
463
+
464
+ if (!('encodeInto' in cachedTextEncoder)) {
465
+ cachedTextEncoder.encodeInto = function (arg, view) {
466
+ const buf = cachedTextEncoder.encode(arg);
467
+ view.set(buf);
468
+ return {
469
+ read: arg.length,
470
+ written: buf.length
471
+ };
472
+ };
473
+ }
474
+
249
475
  let WASM_VECTOR_LEN = 0;
250
476
 
251
477
  let wasmModule, wasmInstance, wasm;
@@ -253,9 +479,15 @@ function __wbg_finalize_init(instance, module) {
253
479
  wasmInstance = instance;
254
480
  wasm = instance.exports;
255
481
  wasmModule = module;
482
+ cachedBigInt64ArrayMemory0 = null;
483
+ cachedBigUint64ArrayMemory0 = null;
256
484
  cachedDataViewMemory0 = null;
257
485
  cachedFloat32ArrayMemory0 = null;
258
486
  cachedFloat64ArrayMemory0 = null;
487
+ cachedInt16ArrayMemory0 = null;
488
+ cachedInt32ArrayMemory0 = null;
489
+ cachedInt8ArrayMemory0 = null;
490
+ cachedUint16ArrayMemory0 = null;
259
491
  cachedUint32ArrayMemory0 = null;
260
492
  cachedUint8ArrayMemory0 = null;
261
493
  wasm.__wbindgen_start();
Binary file
@@ -1,12 +1,14 @@
1
1
  /* tslint:disable */
2
2
  /* eslint-disable */
3
3
  export const memory: WebAssembly.Memory;
4
+ export const decode_draco: (a: number, b: number, c: number) => number;
4
5
  export const init: () => void;
5
6
  export const parse_drc_bytes: (a: number, b: number) => number;
6
7
  export const supported_extensions: (a: number) => void;
7
8
  export const version: (a: number) => void;
8
- export const __wbindgen_export: (a: number) => void;
9
- export const __wbindgen_export2: (a: number, b: number) => number;
9
+ export const __wbindgen_export: (a: number, b: number) => number;
10
+ export const __wbindgen_export2: (a: number, b: number, c: number, d: number) => number;
11
+ export const __wbindgen_export3: (a: number) => void;
10
12
  export const __wbindgen_add_to_stack_pointer: (a: number) => number;
11
- export const __wbindgen_export3: (a: number, b: number, c: number) => void;
13
+ export const __wbindgen_export4: (a: number, b: number, c: number) => void;
12
14
  export const __wbindgen_start: () => void;
@@ -0,0 +1,45 @@
1
+ import type { DracoAttributeRequest, DracoDecodeResult } from './index.js';
2
+
3
+ export interface DecoderPoolOptions {
4
+ /**
5
+ * How many workers to run at most; they start as work arrives. Defaults to
6
+ * one less than the cores, between one and four. `0` decodes on the calling
7
+ * thread, as does an environment without workers.
8
+ */
9
+ workers?: number;
10
+ /**
11
+ * The entry's wasm, when it is not next to this module where the pool looks
12
+ * for it: a URL, a `Response`, the bytes, or a compiled module.
13
+ */
14
+ wasm?: string | URL | Response | BufferSource | WebAssembly.Module;
15
+ }
16
+
17
+ export interface DecodeOptions {
18
+ /**
19
+ * Hand the stream's buffer to the worker instead of copying it, leaving it
20
+ * detached here. For a buffer nothing else reads afterwards.
21
+ */
22
+ transfer?: boolean;
23
+ }
24
+
25
+ export interface DecoderPool {
26
+ /**
27
+ * `decode_draco` on a worker. A stream that does not decode resolves with
28
+ * `success: false`, as `decode_draco` returns it; a worker that fails rejects
29
+ * the tasks it held.
30
+ */
31
+ decode(
32
+ data: ArrayBuffer | ArrayBufferView,
33
+ attributes?: DracoAttributeRequest[],
34
+ options?: DecodeOptions,
35
+ ): Promise<DracoDecodeResult>;
36
+ /** Compiles the wasm and starts one worker ahead of the first decode. */
37
+ preload(): Promise<void>;
38
+ /** Changes the worker limit; idle workers past it stop now, busy ones when done. */
39
+ setWorkerLimit(workers: number): void;
40
+ readonly workerLimit: number;
41
+ /** Stops every worker and rejects what they held. */
42
+ dispose(): void;
43
+ }
44
+
45
+ export function createDecoderPool(options?: DecoderPoolOptions): DecoderPool;
@@ -0,0 +1,193 @@
1
+ // Decoding off the calling thread: a pool of workers, each running this entry's
2
+ // own module. The wasm is compiled once here and handed to every worker, the
3
+ // stream goes over by transfer and the arrays come back the same way, so a
4
+ // decode copies nothing it does not have to.
5
+ import init, { decode_draco } from './index.js';
6
+
7
+ /** Leaves a core for the page, and stops at four, past which a page rarely has
8
+ * that many streams in flight at once. */
9
+ const defaultWorkers = () =>
10
+ Math.max(1, Math.min(4, (globalThis.navigator?.hardwareConcurrency ?? 4) - 1));
11
+
12
+ const builtin = (name) =>
13
+ typeof process !== 'undefined' && typeof process.getBuiltinModule === 'function'
14
+ ? process.getBuiltinModule(name)
15
+ : undefined;
16
+
17
+ async function compile(wasm) {
18
+ if (wasm instanceof WebAssembly.Module) return wasm;
19
+ let source = wasm ?? new URL('./index_bg.wasm', import.meta.url);
20
+ if (typeof source === 'string' || source instanceof URL) {
21
+ const url = new URL(source, import.meta.url);
22
+ // Node's fetch does not read files.
23
+ const fs = url.protocol === 'file:' ? builtin('node:fs/promises') : undefined;
24
+ if (fs) return WebAssembly.compile(await fs.readFile(url));
25
+ source = await fetch(url);
26
+ }
27
+ if (typeof Response !== 'undefined' && source instanceof Response) {
28
+ if (!source.ok) throw new Error(`failed to fetch ${source.url}: ${source.status}`);
29
+ if (source.headers.get('content-type') === 'application/wasm') {
30
+ return WebAssembly.compileStreaming(source);
31
+ }
32
+ source = await source.arrayBuffer();
33
+ }
34
+ return WebAssembly.compile(source);
35
+ }
36
+
37
+ /** What the pool needs of a worker, the same over a Web Worker and a Node
38
+ * worker thread. */
39
+ function spawn(blobUrl) {
40
+ if (typeof Worker === 'function') {
41
+ const url = new URL('./worker.js', import.meta.url);
42
+ // A bundler recognises this exact form and emits the worker as a chunk.
43
+ const worker =
44
+ typeof location === 'undefined' || url.origin === location.origin
45
+ ? new Worker(new URL('./worker.js', import.meta.url), { type: 'module' })
46
+ : new Worker(blobUrl(url), { type: 'module' });
47
+ return {
48
+ post: (message, transfer) => worker.postMessage(message, transfer),
49
+ listen: (onMessage, onError) => {
50
+ worker.addEventListener('message', (event) => onMessage(event.data));
51
+ worker.addEventListener('error', (event) => {
52
+ event.preventDefault();
53
+ onError(new Error(event.message || 'the decoder worker failed'));
54
+ });
55
+ worker.addEventListener('messageerror', () => onError(new Error('a decode result could not be read')));
56
+ },
57
+ busy: () => {},
58
+ terminate: () => worker.terminate(),
59
+ };
60
+ }
61
+ const threads = builtin('node:worker_threads');
62
+ if (!threads) return null;
63
+ const worker = new threads.Worker(new URL('./worker.js', import.meta.url));
64
+ // An idle worker does not keep Node running; one with work does.
65
+ worker.unref();
66
+ return {
67
+ post: (message, transfer) => worker.postMessage(message, transfer),
68
+ listen: (onMessage, onError) => {
69
+ worker.on('message', onMessage);
70
+ worker.on('error', onError);
71
+ worker.on('exit', (code) => onError(new Error(`the decoder worker exited with code ${code}`)));
72
+ },
73
+ busy: (busy) => (busy ? worker.ref() : worker.unref()),
74
+ terminate: () => worker.terminate(),
75
+ };
76
+ }
77
+
78
+ const workersAvailable = () =>
79
+ typeof Worker === 'function' || builtin('node:worker_threads') !== undefined;
80
+
81
+ export function createDecoderPool(options = {}) {
82
+ let limit = options.workers ?? defaultWorkers();
83
+ let compiled = null;
84
+ let local = null;
85
+ let blob = null;
86
+ let disposed = false;
87
+ let nextId = 0;
88
+ const slots = [];
89
+
90
+ const load = () => (compiled ??= compile(options.wasm));
91
+ // A worker script must share the page's origin. From a CDN, a worker made
92
+ // here imports it instead, which a module worker may do across origins.
93
+ const blobUrl = (url) =>
94
+ (blob ??= URL.createObjectURL(
95
+ new Blob([`import ${JSON.stringify(url.href)};`], { type: 'text/javascript' }),
96
+ ));
97
+
98
+ function retire(slot, error) {
99
+ const at = slots.indexOf(slot);
100
+ if (at >= 0) slots.splice(at, 1);
101
+ slot.worker.terminate();
102
+ for (const { reject } of slot.pending.values()) reject(error);
103
+ slot.pending.clear();
104
+ }
105
+
106
+ function open(module) {
107
+ const worker = spawn(blobUrl);
108
+ if (!worker) return null;
109
+ const slot = { worker, pending: new Map() };
110
+ worker.listen(
111
+ (message) => {
112
+ const task = slot.pending.get(message.id);
113
+ if (!task) return;
114
+ slot.pending.delete(message.id);
115
+ if (slot.pending.size === 0) {
116
+ worker.busy(false);
117
+ if (slots.indexOf(slot) >= limit) retire(slot);
118
+ }
119
+ if ('error' in message) task.reject(new Error(message.error));
120
+ else task.resolve(message.result);
121
+ },
122
+ // A worker that fails takes only its own tasks down; the next decode
123
+ // starts another.
124
+ (error) => {
125
+ if (slots.includes(slot)) retire(slot, error);
126
+ },
127
+ );
128
+ worker.post({ type: 'init', module });
129
+ slots.push(slot);
130
+ return slot;
131
+ }
132
+
133
+ /** The least loaded worker, or a new one while every worker is busy and
134
+ * the limit allows another. */
135
+ function pick(module) {
136
+ let best = null;
137
+ for (const slot of slots.slice(0, limit)) {
138
+ if (!best || slot.pending.size < best.pending.size) best = slot;
139
+ }
140
+ if ((!best || best.pending.size > 0) && slots.length < limit) return open(module) ?? best;
141
+ return best;
142
+ }
143
+
144
+ const onThisThread = () =>
145
+ (local ??= load().then((module) => init({ module_or_path: module })));
146
+
147
+ async function decode(data, attributes, { transfer = false } = {}) {
148
+ if (disposed) throw new Error('the decoder pool was disposed');
149
+ const bytes = ArrayBuffer.isView(data)
150
+ ? new Uint8Array(data.buffer, data.byteOffset, data.byteLength)
151
+ : new Uint8Array(data);
152
+ if (limit > 0 && workersAvailable()) {
153
+ const module = await load();
154
+ if (disposed) throw new Error('the decoder pool was disposed');
155
+ const slot = pick(module);
156
+ if (slot) {
157
+ const id = nextId++;
158
+ const sent = transfer ? bytes : bytes.slice();
159
+ return new Promise((resolve, reject) => {
160
+ slot.pending.set(id, { resolve, reject });
161
+ slot.worker.busy(true);
162
+ slot.worker.post({ id, data: sent, attributes }, [sent.buffer]);
163
+ });
164
+ }
165
+ }
166
+ await onThisThread();
167
+ return decode_draco(bytes, attributes);
168
+ }
169
+
170
+ return {
171
+ decode,
172
+ async preload() {
173
+ if (limit > 0 && workersAvailable()) {
174
+ const module = await load();
175
+ if (!disposed && slots.length === 0) open(module);
176
+ } else {
177
+ await onThisThread();
178
+ }
179
+ },
180
+ setWorkerLimit(workers) {
181
+ limit = workers;
182
+ for (const slot of slots.slice(limit)) if (slot.pending.size === 0) retire(slot);
183
+ },
184
+ get workerLimit() {
185
+ return limit;
186
+ },
187
+ dispose() {
188
+ disposed = true;
189
+ for (const slot of [...slots]) retire(slot, new Error('the decoder pool was disposed'));
190
+ if (blob) URL.revokeObjectURL(blob);
191
+ },
192
+ };
193
+ }
@@ -0,0 +1,65 @@
1
+ import type { DracoArrayType } from './index.js';
2
+ import type { DecoderPool, DecoderPoolOptions } from './pool.js';
3
+
4
+ /**
5
+ * The three.js classes the loader builds with, passed in so the package uses
6
+ * the application's own copy of three: `import * as THREE from "three"` will
7
+ * do. With `InterleavedBuffer` and `InterleavedBufferAttribute`, an attribute
8
+ * whose items are not a multiple of four bytes is padded to one, as DRACOLoader
9
+ * pads it since r181; with `Color` and `ColorManagement`, a `.drc` file's sRGB
10
+ * vertex colours are made linear as DRACOLoader makes them.
11
+ */
12
+ export interface ThreeClasses<Geometry> {
13
+ BufferGeometry: new () => Geometry;
14
+ BufferAttribute: new (array: any, itemSize: number, normalized?: boolean) => any;
15
+ InterleavedBuffer?: new (array: any, stride: number) => any;
16
+ InterleavedBufferAttribute?: new (buffer: any, itemSize: number, offset: number) => any;
17
+ Color?: new () => any;
18
+ /** `colorSpaceToWorking` since three r177, `toWorkingColorSpace` before. */
19
+ ColorManagement?:
20
+ | { colorSpaceToWorking(color: any, colorSpace: string): any }
21
+ | { toWorkingColorSpace(color: any, colorSpace: string): any };
22
+ }
23
+
24
+ export interface DracoLoaderOptions extends DecoderPoolOptions {
25
+ /** A pool to share with other loaders, instead of one of the loader's own. */
26
+ pool?: DecoderPool;
27
+ }
28
+
29
+ export interface DecodeGeometryOptions {
30
+ /** three.js attribute name to Draco unique id, as glTF names them. Without
31
+ * it, `position`, `normal`, `color` and `uv` are read by type. */
32
+ attributeIDs?: Record<string, number>;
33
+ /** three.js attribute name to the typed array it comes back in. */
34
+ attributeTypes?: Record<string, DracoArrayType>;
35
+ /** `"srgb"` makes vertex colours linear; the default leaves them as stored. */
36
+ vertexColorSpace?: string;
37
+ }
38
+
39
+ /** The part of three.js's DRACOLoader that GLTFLoader and `.drc` users call. */
40
+ export interface DracoLoader<Geometry> {
41
+ readonly pool: DecoderPool;
42
+ decodeGeometry(buffer: ArrayBuffer | ArrayBufferView, options?: DecodeGeometryOptions): Promise<Geometry>;
43
+ decodeDracoFile(
44
+ buffer: ArrayBuffer,
45
+ callback: (geometry: Geometry) => void,
46
+ attributeIDs?: Record<string, number> | null,
47
+ attributeTypes?: Record<string, DracoArrayType> | null,
48
+ vertexColorSpace?: string,
49
+ onError?: (error: unknown) => void,
50
+ ): Promise<void>;
51
+ parse(buffer: ArrayBuffer, onLoad: (geometry: Geometry) => void, onError?: (error: unknown) => void): void;
52
+ parseAsync(buffer: ArrayBuffer | ArrayBufferView): Promise<Geometry>;
53
+ preload(): this;
54
+ setWorkerLimit(workers: number): this;
55
+ dispose(): this;
56
+ }
57
+
58
+ /**
59
+ * A stand-in for three.js's DRACOLoader on the decoder pool:
60
+ * `gltfLoader.setDRACOLoader(createDracoLoader(THREE))`.
61
+ */
62
+ export function createDracoLoader<Geometry>(
63
+ three: ThreeClasses<Geometry>,
64
+ options?: DracoLoaderOptions,
65
+ ): DracoLoader<Geometry>;