ITEP-66305 - Prepare to move ritm tool (#640)

Author: jpggvilaca

web_ui/packages/smart-tools/index.ts

@@ -25,3 +25,15 @@ export {
     getPointsFromMat,
     getMatFromPoints,
 } from './src/utils/tool-utils';
+
+export { sessionParams, type SessionParameters } from './src/utils/wasm-utils';
+
+export const RITMModels = {
+    main: new URL('./src/ritm/models/main.onnx', import.meta.url).toString(),
+    preprocess: new URL('./src/ritm/models/preprocess.onnx', import.meta.url).toString(),
+};
+
+export const SegmentAnythingModels = {
+    encoder: new URL('./src/segment-anything/models/mobile_sam.encoder.onnx', import.meta.url).toString(),
+    decoder: new URL('./src/segment-anything/models/sam_vit_h_4b8939.decoder.onnx', import.meta.url).toString(),
+};

web_ui/src/webworkers/ritm/main.onnx …es/smart-tools/src/ritm/models/main.onnxweb_ui/src/webworkers/ritm/main.onnx renamed to web_ui/packages/smart-tools/src/ritm/models/main.onnx web_ui/src/webworkers/ritm/main.onnx renamed to web_ui/packages/smart-tools/src/ritm/models/main.onnx

@@ -1,12 +1,10 @@
 // Copyright (C) 2022-2025 Intel Corporation
 // LIMITED EDGE SOFTWARE DISTRIBUTION LICENSE
 
-import { loadSource } from '@geti/smart-tools';
+import { loadSource, SessionParameters, sessionParams } from '@geti/smart-tools';
 import type { InferenceSession } from 'onnxruntime-common';
 import * as ort from 'onnxruntime-web';
 
-import { SessionParameters, sessionParams } from '../../wasm-utils';
-
 const loadModel = async (modelPath: string) => {
     return await (await loadSource(modelPath))?.arrayBuffer();
 };

…b_ui/src/webworkers/ritm/preprocess.onnx …rt-tools/src/ritm/models/preprocess.onnxweb_ui/src/webworkers/ritm/preprocess.onnx renamed to web_ui/packages/smart-tools/src/ritm/models/preprocess.onnx web_ui/src/webworkers/ritm/preprocess.onnx renamed to web_ui/packages/smart-tools/src/ritm/models/preprocess.onnx

@@ -144,3 +144,11 @@ jest.mock('@geti/core/src/services/use-deployment-config-query.hook', () => ({
 }));
 
 process.env.REACT_APP_VALIDATION_COMPONENT_TESTS = 'true';
+
+jest.mock('@geti/smart-tools', () => ({
+    sessionParams: {
+        numThreads: 0,
+        executionProviders: ['cpu'],
+        wasmRoot: {},
+    },
+}));

…segment-anything/mobile_sam.encoder.onnx …-anything/models/mobile_sam.encoder.onnxweb_ui/src/webworkers/segment-anything/mobile_sam.encoder.onnx renamed to web_ui/packages/smart-tools/src/segment-anything/models/mobile_sam.encoder.onnx web_ui/src/webworkers/segment-anything/mobile_sam.encoder.onnx renamed to web_ui/packages/smart-tools/src/segment-anything/models/mobile_sam.encoder.onnx

@@ -9,6 +9,8 @@ import {
     isPolygonValid,
     loadSource,
     OpenCVLoader,
+    RITMModels,
+    sessionParams,
     stackPlanes,
 } from '@geti/smart-tools';
 import { Point, Polygon, Shape, ShapeType } from '@geti/smart-tools/src/shared/interfaces';
@@ -20,7 +22,6 @@ import type OpenCVTypes from 'OpenCVTypes';
 
 import { RegionOfInterest } from '../core/annotations/annotation.interface';
 import { RITMContour, RITMPoint, TEMPLATE_SIZE } from '../pages/annotator/tools/ritm-tool/ritm-tool.interface';
-import { sessionParams } from '../pages/annotator/tools/wasm-utils';
 
 declare const self: DedicatedWorkerGlobalScope;
 
@@ -48,9 +49,10 @@ class RITM {
 
     async load() {
         ort.env.wasm.wasmPaths = sessionParams.wasmRoot;
+
         this.models = {
-            main: await this.loadModel(new URL('./ritm/main.onnx', import.meta.url).toString()),
-            preprocess: await this.loadModel(new URL('./ritm/preprocess.onnx', import.meta.url).toString()),
+            main: await this.loadModel(RITMModels.main),
+            preprocess: await this.loadModel(RITMModels.preprocess),
         };
     }
 

…t-anything/sam_vit_h_4b8939.decoder.onnx …ing/models/sam_vit_h_4b8939.decoder.onnxweb_ui/src/webworkers/segment-anything/sam_vit_h_4b8939.decoder.onnx renamed to web_ui/packages/smart-tools/src/segment-anything/models/sam_vit_h_4b8939.decoder.onnx web_ui/src/webworkers/segment-anything/sam_vit_h_4b8939.decoder.onnx renamed to web_ui/packages/smart-tools/src/segment-anything/models/sam_vit_h_4b8939.decoder.onnx

@@ -3,7 +3,7 @@
 
 //  Dependencies get bundled into the worker
 
-import { OpenCVLoader } from '@geti/smart-tools';
+import { OpenCVLoader, SegmentAnythingModels } from '@geti/smart-tools';
 import { expose } from 'comlink';
 import type OpenCVTypes from 'OpenCVTypes';
 
@@ -50,8 +50,8 @@ class SegmentAnythingModelWrapper {
                 padSize: 1024,
             },
             modelPaths: new Map([
-                ['encoder', new URL('./segment-anything/mobile_sam.encoder.onnx', import.meta.url).toString()],
-                ['decoder', new URL('./segment-anything/sam_vit_h_4b8939.decoder.onnx', import.meta.url).toString()],
+                ['encoder', SegmentAnythingModels.encoder],
+                ['decoder', SegmentAnythingModels.decoder],
             ]),
         };