feat: Add loadPCDFile for render .pcd files

open-mmlab · Jun 14, 2022 · 2cea7a0 · 2cea7a0
1 parent c5a56e1
commit 2cea7a0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 8 deletions.
diff --git a/packages/lb-annotation/src/core/pointCloud/index.tsx b/packages/lb-annotation/src/core/pointCloud/index.tsx
@@ -101,7 +101,7 @@ export class PointCloud {
   }
 
   public renderCircle() {
-    const radius = 15;
+    const radius = 100;
     const curve = new THREE.EllipseCurve(
       15,
       15, // ax, aY
@@ -124,6 +124,14 @@ export class PointCloud {
     this.render();
   }
 
+  public loadPCDFile = (src: string) => {
+    this.pcdLoader.load(src, (points: any) => {
+      this.scene.add(points);
+      points.material.size = 0.3;
+      this.render();
+    });
+  };
+
   public render() {
     this.renderer.render(this.scene, this.camera);
   }

diff --git a/packages/lb-demo/src/components/PointCloud/index.jsx b/packages/lb-demo/src/components/PointCloud/index.jsx
@@ -7,7 +7,7 @@
 import { PointCloud } from '@labelbee/lb-annotation';
 import { useEffect, useRef } from 'react';
 
-const pointCloudID = "LABELBEE-POINTCLOUD"
+const pointCloudID = 'LABELBEE-POINTCLOUD';
 
 const PointCloudDom = () => {
   const ref = useRef();
@@ -16,13 +16,12 @@ const PointCloudDom = () => {
   useEffect(() => {
     console.log(ref.current);
     if (ref.current) {
-      pointCloudRef.current = new PointCloud({ container: ref.current});
+      pointCloudRef.current = new PointCloud({ container: ref.current });
+      pointCloudRef.current.loadPCDFile('http://10.53.25.142:8001/1/000001.pcd');
     }
   }, []);
 
-  return (
-    <div id={pointCloudID} ref={ref} />
-  )
-}
+  return <div id={pointCloudID} ref={ref} />;
+};
 
-export default PointCloudDom
+export default PointCloudDom;