@ar-js-org/ar.js-threejs/src/Worker.ts

AR.js (THREEx, ARjs) modular package - Typescript version

/* eslint-env worker */
import jsartoolkit from "@ar-js-org/artoolkit5-js";
import type ARController from "@ar-js-org/artoolkit5-js/types/ARController";
const ctx: Worker = self as any;

let ar: ARController | null = null

ctx.onmessage = function (e) {
  var msg = e.data;
  switch (msg.type) {
    case "init": {
      load(msg);
      return;
    }
    case "process": {
      next = msg.imagedata;
      process();
      return;
    }
  }
};

var next: any = null;
var markerResult = null;

function load(msg: any) {
  var camUrl, nftMarkerUrl: string;
  var basePath = self.origin;
  console.log("base path:", basePath);
  // test if the msg.param (the incoming url) is an http or https path
  var regexC =
    /https?:\/\/(www\.)?[-a-zA-Z0-9@:%._\+~#=]{2,256}\.[a-z]{2,6}\b([-a-zA-Z0-9@:%_\+.~#()?&//=]*)/gim;
  var reC = regexC.test(msg.param);
  if (reC == true) {
    camUrl = msg.param;
  } else if (reC == false) {
    camUrl = basePath + "/" + msg.param;
  }
  var onLoad = function (arController:  ARController) {
    ar = arController;
    var cameraMatrix = ar.getCameraMatrix();

    // after the ARController is set up, we load the NFT Marker
    var regexM =
      /https?:\/\/(www\.)?[-a-zA-Z0-9@:%._\+~#=]{2,256}\.[a-z]{2,6}\b([-a-zA-Z0-9@:%_\+.~#()?&//=]*)/gim;
    var reM = regexM.test(msg.marker);
    if (reM == true) {
      nftMarkerUrl = msg.marker;
    } else if (reM == false) {
      nftMarkerUrl = basePath + "/" + msg.marker;
    }
    ar.loadNFTMarker(nftMarkerUrl)
      .then(function (markerId: number) {
        ar.trackNFTMarkerId(markerId, 80);
        let marker = ar.getNFTData(markerId, 0);
        console.log("nftMarker data: ", marker);
        postMessage({ type: "markerInfos", marker: JSON.stringify(marker) });
        postMessage({ type: "endLoading" });
      })
      .catch(function (err: any) {
        console.log("Error in loading marker on Worker", err);
      });

    // ...and we listen for event when marker has been found from camera
    ar.addEventListener("getNFTMarker", function (ev: any) {
      // let AR.js know that a NFT marker has been found, with its matrix for positioning
      markerResult = {
        type: "found",
        matrix: JSON.stringify(ev.data.matrix),
      };
    });

    postMessage({ type: "loaded", proj: JSON.stringify(cameraMatrix) });
  };

  var onError = function (error: any) {
    console.error("Error while intizalizing arController", error);
  };

  jsartoolkit.ARController.initWithDimensions(msg.pw, msg.ph, camUrl, {})
    .then(onLoad)
    .catch(onError);
}

function process() {
  markerResult = null;

  if (ar && ar.process) {
    ar.process(next);
  }

  if (markerResult) {
    postMessage(markerResult);
  } else {
    postMessage({
      type: "not found",
    });
  }
  next = null;
}