feat(footer): Support PointCloud PageNumber

open-mmlab · Feb 20, 2023 · 794847c · 794847c
1 parent e9c22fb
commit 794847c
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/packages/lb-components/src/components/pointCloudView/PointCloudListener.tsx b/packages/lb-components/src/components/pointCloudView/PointCloudListener.tsx
@@ -20,7 +20,7 @@ import { usePolygon } from './hooks/usePolygon';
 const { EPolygonPattern } = cTool;
 
 interface IProps extends IA2MapStateProps {
-  checkMode: boolean;
+  checkMode?: boolean;
 }
 
 const PointCloudListener: React.FC<IProps> = ({ currentData, config, checkMode }) => {

diff --git a/packages/lb-components/src/components/pointCloudView/index.tsx b/packages/lb-components/src/components/pointCloudView/index.tsx
@@ -38,7 +38,7 @@ const PointCloudView: React.FC<IProps> = ({ imgList, drawLayerSlot, checkMode })
 
   return (
     <>
-      <PointCloudListener  checkMode={checkMode}/>
+      <PointCloudListener checkMode={checkMode}/>
       <div className={getClassName('point-cloud-layout')} onContextMenu={(e) => e.preventDefault()}>
         <div className={getClassName('point-cloud-wrapper')}>
           <div className={getClassName('point-cloud-container', 'left')}>

diff --git a/packages/lb-components/src/views/MainView/toolFooter/PageNumber/index.tsx b/packages/lb-components/src/views/MainView/toolFooter/PageNumber/index.tsx
@@ -1,10 +1,11 @@
-import React, { useEffect, useState } from 'react';
+import React, { useContext, useEffect, useState } from 'react';
 import { AppState } from '@/store';
 import { connect } from 'react-redux';
 import { GraphToolInstance } from '@/store/annotation/types';
 import { Divider } from 'antd/es';
 import { useTranslation } from 'react-i18next';
 import { LabelBeeContext } from '@/store/ctx';
+import { PointCloudContext } from '@/components/pointCloudView/PointCloudContext';
 
 interface IProps {
   toolInstance: GraphToolInstance;
@@ -13,6 +14,7 @@ interface IProps {
 const PageNumber = (props: IProps) => {
   const { toolInstance } = props;
   const [_, forceRender] = useState(0);
+  const { pointCloudBoxList } = useContext(PointCloudContext);
   const { t } = useTranslation();
   useEffect(() => {
     if (toolInstance) {
@@ -26,7 +28,8 @@ const PageNumber = (props: IProps) => {
     return null;
   }
 
-  const count = toolInstance?.currentPageResult?.length;
+  const count = toolInstance?.currentPageResult?.length ?? pointCloudBoxList.length;
+
   if (count >= 0) {
     return (
       <span>