onnxruntime

1
// Copyright (c) Microsoft Corporation. All rights reserved.
2
// Licensed under the MIT License.
3

4
import { getInstance } from './wasm-factory';
5

6
export const allocWasmString = (data: string, allocs: number[]): number => {
7
  const wasm = getInstance();
8

9
  const dataLength = wasm.lengthBytesUTF8(data) + 1;
10
  const dataOffset = wasm._malloc(dataLength);
11
  wasm.stringToUTF8(data, dataOffset, dataLength);
12
  allocs.push(dataOffset);
13

14
  return dataOffset;
15
};
16

17
interface ExtraOptionsHandler {
18
  (name: string, value: string): void;
19
}
20

21
export const iterateExtraOptions = (
22
  options: Record<string, unknown>,
23
  prefix: string,
24
  seen: WeakSet<Record<string, unknown>>,
25
  handler: ExtraOptionsHandler,
26
): void => {
27
  if (typeof options == 'object' && options !== null) {
28
    if (seen.has(options)) {
29
      throw new Error('Circular reference in options');
30
    } else {
31
      seen.add(options);
32
    }
33
  }
34

35
  Object.entries(options).forEach(([key, value]) => {
36
    const name = prefix ? prefix + key : key;
37
    if (typeof value === 'object') {
38
      iterateExtraOptions(value as Record<string, unknown>, name + '.', seen, handler);
39
    } else if (typeof value === 'string' || typeof value === 'number') {
40
      handler(name, value.toString());
41
    } else if (typeof value === 'boolean') {
42
      handler(name, value ? '1' : '0');
43
    } else {
44
      throw new Error(`Can't handle extra config type: ${typeof value}`);
45
    }
46
  });
47
};
48

49
/**
50
 * check web assembly API's last error and throw error if any error occurred.
51
 * @param message a message used when an error occurred.
52
 */
53
export const checkLastError = (message: string): void => {
54
  const wasm = getInstance();
55

56
  const stack = wasm.stackSave();
57
  try {
58
    const paramsOffset = wasm.stackAlloc(8);
59
    wasm._OrtGetLastError(paramsOffset, paramsOffset + 4);
60
    const errorCode = wasm.HEAP32[paramsOffset / 4];
61
    const errorMessagePointer = wasm.HEAPU32[paramsOffset / 4 + 1];
62
    const errorMessage = errorMessagePointer ? wasm.UTF8ToString(errorMessagePointer) : '';
63
    throw new Error(`${message} ERROR_CODE: ${errorCode}, ERROR_MESSAGE: ${errorMessage}`);
64
  } finally {
65
    wasm.stackRestore(stack);
66
  }
67
};
68