adicionado exemplo de detecção de bordas

Piemontez · May 2, 2024 · eec5218 · eec5218
1 parent 5536b4f
commit eec5218
Show file tree

Hide file tree

Showing 5 changed files with 102 additions and 4 deletions.
diff --git a/src/plugins/opencv/edge/CVSobelComponent.ts → src/plugins/opencv/edge/SobelComponent.ts b/src/plugins/opencv/edge/CVSobelComponent.ts → src/plugins/opencv/edge/SobelComponent.ts
@@ -10,7 +10,7 @@ import { edgeTabName } from './tabname';
  * Sobel component and node
  */
 
-export class CVSobelComponent extends CVFIOComponent {
+export class SobelComponent extends CVFIOComponent {
   static menu = { tabTitle: edgeTabName, title: 'Sobel' };
 
   static processor = class SobelProcessor extends CVFNodeProcessor {

diff --git a/src/plugins/opencv/edge/index.ts b/src/plugins/opencv/edge/index.ts
@@ -1,4 +1,4 @@
-export * from './CVSobelComponent';
+export * from './SobelComponent';
 export * from './CannyComponent';
 export * from './LaplacianComponent';
 export * from './ScharrComponent';

diff --git a/src/plugins/opencvTemplates/edge.ts b/src/plugins/opencvTemplates/edge.ts
@@ -0,0 +1,98 @@
+import { CVVideoCaptureComponent } from '../opencv/inputs/CVVideoCaptureComponent';
+import { CvtColorComponent } from '../opencv/conversors/CvtColorComponent';
+import { CVFComponent } from '../../ide/components/NodeComponent';
+import { useNodeStore } from '../../core/contexts/NodeStore';
+import { ProjectTemplate } from '../../core/types/project-template';
+import { makeXYPosition } from '../../core/utils/makeXYPosition';
+import cv from 'opencv-ts';
+import { EqualizeHistComponent } from '../opencv/others';
+import { SobelComponent, CannyComponent, LaplacianComponent } from '../opencv/edge';
+import { GaussianBlurComponent } from '../opencv/smoothing';
+
+const group = 'OpenCV';
+
+const EdgeSamplesAction: ProjectTemplate = {
+  group,
+  title: 'Edges Samples',
+  action: () => {
+    let comp: typeof CVFComponent | null;
+    let pos;
+
+    // Add Video
+    comp = useNodeStore.getState().getNodeType(CVVideoCaptureComponent.name);
+    pos = makeXYPosition(-3, 0);
+    const videoId = useNodeStore.getState().addNodeFromComponent(comp!, pos).id;
+
+    // Add Convert
+    comp = useNodeStore.getState().getNodeType(CvtColorComponent.name);
+    pos = makeXYPosition(-1, -1);
+    const cvtColorId = useNodeStore.getState().addNodeFromComponent(comp!, pos, { code: cv.COLOR_BGR2GRAY }).id;
+    // Add Convert
+    comp = useNodeStore.getState().getNodeType(GaussianBlurComponent.name);
+    pos = makeXYPosition(-1, 0);
+    const gaussianBlurId = useNodeStore.getState().addNodeFromComponent(comp!, pos, { code: cv.COLOR_BGR2GRAY }).id;
+    // Add Equalize
+    comp = useNodeStore.getState().getNodeType(EqualizeHistComponent.name);
+    pos = makeXYPosition(-1, 1);
+    const eqId = useNodeStore.getState().addNodeFromComponent(comp!, pos, { dsize: new cv.Size(128, 128) }).id;
+
+    // Add SobelComponent
+    comp = useNodeStore.getState().getNodeType(SobelComponent.name);
+    pos = makeXYPosition(1, -2);
+    const sobel01Id = useNodeStore.getState().addNodeFromComponent(comp!, pos).id;
+    // Add SobelComponent
+    comp = useNodeStore.getState().getNodeType(SobelComponent.name);
+    pos = makeXYPosition(2, -2);
+    const sobel02Id = useNodeStore.getState().addNodeFromComponent(comp!, pos).id;
+    // Add SobelComponent
+    comp = useNodeStore.getState().getNodeType(SobelComponent.name);
+    pos = makeXYPosition(3, -2);
+    const sobel03Id = useNodeStore.getState().addNodeFromComponent(comp!, pos).id;
+
+    // Add HistogramCalcComponent
+    comp = useNodeStore.getState().getNodeType(CannyComponent.name);
+    pos = makeXYPosition(1, 0);
+    const canny01Id = useNodeStore.getState().addNodeFromComponent(comp!, pos).id;
+    // Add HistogramCalcComponent
+    comp = useNodeStore.getState().getNodeType(CannyComponent.name);
+    pos = makeXYPosition(2, 0);
+    const canny02Id = useNodeStore.getState().addNodeFromComponent(comp!, pos).id;
+    // Add HistogramCalcComponent
+    comp = useNodeStore.getState().getNodeType(CannyComponent.name);
+    pos = makeXYPosition(3, 0);
+    const canny03Id = useNodeStore.getState().addNodeFromComponent(comp!, pos).id;
+
+    // Add HistogramCalcComponent
+    comp = useNodeStore.getState().getNodeType(LaplacianComponent.name);
+    pos = makeXYPosition(1, 2);
+    const laplacian01Id = useNodeStore.getState().addNodeFromComponent(comp!, pos).id;
+    // Add HistogramCalcComponent
+    comp = useNodeStore.getState().getNodeType(LaplacianComponent.name);
+    pos = makeXYPosition(2, 2);
+    const laplacian02Id = useNodeStore.getState().addNodeFromComponent(comp!, pos).id;
+    // Add HistogramCalcComponent
+    comp = useNodeStore.getState().getNodeType(LaplacianComponent.name);
+    pos = makeXYPosition(3, 2);
+    const laplacian03Id = useNodeStore.getState().addNodeFromComponent(comp!, pos).id;
+
+    useNodeStore.getState().addEdge(videoId, cvtColorId, 'out', 'src1');
+    useNodeStore.getState().addEdge(cvtColorId, gaussianBlurId, 'out', 'src1');
+    useNodeStore.getState().addEdge(cvtColorId, eqId, 'out', 'src1');
+
+    useNodeStore.getState().addEdge(cvtColorId, sobel01Id, 'out', 'src1');
+    useNodeStore.getState().addEdge(gaussianBlurId, sobel02Id, 'out', 'src1');
+    useNodeStore.getState().addEdge(eqId, sobel03Id, 'out', 'src1');
+
+    useNodeStore.getState().addEdge(cvtColorId, canny01Id, 'out', 'src1');
+    useNodeStore.getState().addEdge(gaussianBlurId, canny02Id, 'out', 'src1');
+    useNodeStore.getState().addEdge(eqId, canny03Id, 'out', 'src1');
+
+    useNodeStore.getState().addEdge(cvtColorId, laplacian01Id, 'out', 'src1');
+    useNodeStore.getState().addEdge(gaussianBlurId, laplacian02Id, 'out', 'src1');
+    useNodeStore.getState().addEdge(eqId, laplacian03Id, 'out', 'src1');
+
+    setTimeout(useNodeStore.getState().fitView, 100);
+  },
+};
+
+export default EdgeSamplesAction;
diff --git a/src/plugins/opencvTemplates/histogram-cal.ts b/src/plugins/opencvTemplates/histogram-cal.ts
@@ -1,10 +1,8 @@
 import { CVVideoCaptureComponent } from '../opencv/inputs/CVVideoCaptureComponent';
 import { CvtColorComponent } from '../opencv/conversors/CvtColorComponent';
 import { CVFComponent } from '../../ide/components/NodeComponent';
-import { ThresholdComponent } from '../opencv/segmentation/ThresholdComponent';
 import { useNodeStore } from '../../core/contexts/NodeStore';
 import { ProjectTemplate } from '../../core/types/project-template';
-import { CVResizeComponent } from '../opencv/geometricTransformations';
 import { makeXYPosition } from '../../core/utils/makeXYPosition';
 import cv from 'opencv-ts';
 import { EqualizeHistComponent } from '../opencv/others';

diff --git a/src/plugins/opencvTemplates/index.tsx b/src/plugins/opencvTemplates/index.tsx
@@ -2,12 +2,14 @@ import { PluginType } from '../../core/types/plugin';
 import ThresholdSamplesAction from './threshold-samples';
 import MorphologySamplesAction from './morphology-samples';
 import HistogramCalcSamplesAction from './histogram-cal';
+import EdgeSamplesAction from './edge';
 
 const OpenCVTemplatesPlugin: PluginType = {
   name: 'OpenCV Templates Plugin',
   components: [],
   templates: [
     //
+    EdgeSamplesAction,
     ThresholdSamplesAction,
     MorphologySamplesAction,
     HistogramCalcSamplesAction,