rm-webgpu-compute-tasks 0.0.5 → 0.0.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "rm-webgpu-compute-tasks",
-  "version": "0.0.5",
+  "version": "0.0.7",
   "description": "",
   "main": "./src/index.js",
   "types": "./src/index.d.ts",

package/src/index.d.ts

@@ -283,6 +283,29 @@ declare interface IData {
     }[];
 }
 
+declare interface IItemsFile {
+    cameraMat: number[];
+    extrinsicMat: number[];
+    filename: string;
+    imageSize: [number, number];
+    position: {
+        x: number;
+        y: number;
+        z: number;
+    };
+    quaternion: {
+        x: number;
+        y: number;
+        z: number;
+        w: number;
+    };
+    sam3Objects: {
+        category: string;
+        objId: string;
+        coordinates: [number, number][];
+    }[];
+}
+
 declare type InitOption = {
     featureLevel?: string;
     powerPreference?: GPUPowerPreference;
@@ -487,7 +510,11 @@ export declare class Renderer {
 }
 
 export declare const segmentObjectPointClouds: typeof sopc_ & {
-    getResult(opt: Option_2): Promise<number[][][]>;
+    getResult(opt: Option_2): Promise<{
+        results: number[][][];
+        objIds: string[][];
+        itemsFiles: IItemsFile[];
+    }>;
 };
 
 export declare const shaderLocationMap: Map<string, number>;
@@ -592,6 +619,8 @@ declare function sopc_(opt: Option_2): Promise<{
     camera: PerspectiveCamera;
     results: number[][][];
     pointsArray: Float32Array<ArrayBufferLike>;
+    objIds: string[][];
+    itemsFiles: IItemsFile[];
 }>;
 
 export declare class Texture {

package/src/index.js

@@ -3274,7 +3274,7 @@ async function sopc_(opt) {
       const isTranscendingCreation = transcendingCreation.includes(type);
       pointarr = sort(pointarr, isTranscendingCreation, itemsFile.position)[0];
       if (pointarr.length <= 6 * 3) return;
-      if (!results[fileIndex]) results[fileIndex] = new Array(itemsFiles[fileIndex].sam3Objects.length);
+      if (!results[fileIndex]) results[fileIndex] = new Array(itemsFiles[fileIndex].sam3Objects.length).fill([]);
       results[fileIndex][itemIndex] = pointarr;
     });
   });
@@ -3285,15 +3285,17 @@ async function sopc_(opt) {
     renderer,
     camera,
     results,
-    pointsArray
+    pointsArray,
+    objIds: itemsFiles.map((item) => item.sam3Objects.map((sam3Object) => sam3Object.objId)),
+    itemsFiles
   };
 }
 const segmentObjectPointClouds = Object.assign(sopc_, {
   async getResult(opt) {
     await initWebgpu();
-    const { results } = await sopc_(opt);
+    const { results, objIds, itemsFiles } = await sopc_(opt);
     await destroyWebgpu();
-    return results;
+    return { results, objIds, itemsFiles };
   }
 });
 export {