const log = require('@vladmandic/pilogger');
const tf = require('@tensorflow/tfjs'); // wasm backend requires tfjs to be loaded first
const wasm = require('@tensorflow/tfjs-backend-wasm'); // wasm backend does not get auto-loaded in nodejs
const { Canvas, Image } = require('canvas');
const H = require('../dist/human.node-wasm.js');
const test = require('./test-main.js').test;

// @ts-ignore
H.env.Canvas = Canvas; // requires monkey-patch as wasm does not have tf.browser namespace
// @ts-ignore
H.env.Image = Image; // requires monkey-patch as wasm does not have tf.browser namespace

const config = {
  cacheSensitivity: 0,
  modelBasePath: 'https://vladmandic.github.io/human/models/',
  backend: 'wasm',
  // wasmPath: 'node_modules/@tensorflow/tfjs-backend-wasm/dist/',
  wasmPath: `https://cdn.jsdelivr.net/npm/@tensorflow/tfjs-backend-wasm@${tf.version_core}/dist/`,
  debug: false,
  async: false,
  face: {
    enabled: true,
    detector: { rotation: false },
    mesh: { enabled: true },
    iris: { enabled: true },
    description: { enabled: true },
    emotion: { enabled: true },
    antispoof: { enabled: true },
    liveness: { enabled: true },
  },
  hand: { enabled: true, rotation: false },
  body: { enabled: true },
  object: { enabled: true },
  segmentation: { enabled: true },
  filter: { enabled: false },
};

async function main() {
  wasm.setWasmPaths(config.wasmPath);
  await tf.setBackend('wasm');
  await tf.ready();
  H.env.updateBackend();
  log.info(H.env.wasm);
  test(H.Human, config);
}

main();