Merge pull request #531 from drbobbob/advanced-filter-contours

Add AdvancedFilterCountoursOperation

Author: ThomasJClark

core/src/main/java/edu/wpi/grip/core/operations/composite/FilterContoursOperation.java

@@ -45,6 +45,19 @@ public class FilterContoursOperation implements Operation {
     private final SocketHint<List> solidityHint =
             SocketHints.Inputs.createNumberListRangeSocketHint("Solidity", 0, 100);
 
+    private final SocketHint<Number> minVertexHint =
+            SocketHints.Inputs.createNumberSpinnerSocketHint("Min Vertices", 0, 0, Integer.MAX_VALUE);
+
+    private final SocketHint<Number> maxVertexHint =
+            SocketHints.Inputs.createNumberSpinnerSocketHint("Max Vertices", 0, 1000000, Integer.MAX_VALUE);
+
+    private final SocketHint<Number> minRatioHint =
+            SocketHints.Inputs.createNumberSpinnerSocketHint("Min Ratio", 0, 0, Integer.MAX_VALUE);
+
+    private final SocketHint<Number> maxRatioHint =
+            SocketHints.Inputs.createNumberSpinnerSocketHint("Max Ratio", 1000, 0, Integer.MAX_VALUE);
+
+
     @Override
     public String getName() {
         return "Filter Contours";
@@ -76,6 +89,10 @@ public InputSocket<?>[] createInputSockets(EventBus eventBus) {
                 new InputSocket<>(eventBus, minHeightHint),
                 new InputSocket<>(eventBus, maxHeightHint),
                 new InputSocket<>(eventBus, solidityHint),
+                new InputSocket<>(eventBus, minVertexHint),
+                new InputSocket<>(eventBus, maxVertexHint),
+                new InputSocket<>(eventBus, minRatioHint),
+                new InputSocket<>(eventBus, maxRatioHint),
         };
     }
 
@@ -96,6 +113,11 @@ public void perform(InputSocket<?>[] inputs, OutputSocket<?>[] outputs) {
         final double maxHeight = ((Number) inputs[6].getValue().get()).doubleValue();
         final double minSolidity = ((List<Number>) inputs[7].getValue().get()).get(0).doubleValue();
         final double maxSolidity = ((List<Number>) inputs[7].getValue().get()).get(1).doubleValue();
+        final double minVertexCount = ((Number) inputs[8].getValue().get()).doubleValue();
+        final double maxVertexCount = ((Number) inputs[9].getValue().get()).doubleValue();
+        final double minRatio = ((Number) inputs[10].getValue().get()).doubleValue();
+        final double maxRatio = ((Number) inputs[11].getValue().get()).doubleValue();
+
 
         final MatVector inputContours = inputSocket.getValue().get().getContours();
         final MatVector outputContours = new MatVector(inputContours.size());
@@ -119,6 +141,11 @@ public void perform(InputSocket<?>[] inputs, OutputSocket<?>[] outputs) {
             final double solidity = 100 * area / contourArea(hull);
             if (solidity < minSolidity || solidity > maxSolidity) continue;
 
+            if(contour.rows() < minVertexCount || contour.rows() > maxVertexCount) continue;
+
+            final double ratio = bb.width() / bb.height();
+            if (ratio < minRatio || ratio > maxRatio) continue;
+
             outputContours.put(filteredContourCount++, contour);
         }