Better thresholding

Scott Klum
1 parent eb124bd3
Showing 1 changed file with 13 additions and 4 deletions
openbr/plugins/imgproc/threshold.cpp
@@ -16,22 +16,31 @@ class ThresholdTransform : public UntrainableTransform
 {
     Q_OBJECT
-    Q_ENUMS(Method)
     Q_ENUMS(Type)
+    Q_PROPERTY(Type type READ get_type WRITE set_type RESET reset_type STORED false)
+    Q_PROPERTY(bool otsu READ get_otsu WRITE set_otsu RESET reset_otsu STORED false)
     Q_PROPERTY(int thresh READ get_thresh WRITE set_thresh RESET reset_thresh STORED false)
     Q_PROPERTY(int maxValue READ get_maxValue WRITE set_maxValue RESET reset_maxValue STORED false)
-    public:
+public:
+    enum Type { Binary = THRESH_BINARY,
+                BinaryInv = THRESH_BINARY_INV,
+                Trunc = THRESH_TRUNC,
+                ToZero = THRESH_TOZERO,
+                ToZeroInv = THRESH_TOZERO_INV};
+
+private:
+    BR_PROPERTY(Type, type, Binary)
+    BR_PROPERTY(bool, otsu, false)
     BR_PROPERTY(int, thresh, 0)
     BR_PROPERTY(int, maxValue, 255)
-
     void project(const Template &src, Template &dst) const
     {
         dst = src;
         Mat mask;
-        threshold(src, mask, thresh, maxValue, THRESH_BINARY+THRESH_OTSU);
+        threshold(src, mask, thresh, maxValue, otsu ? type+THRESH_OTSU : type);
         dst.file.set("Mask",QVariant::fromValue(mask));
     }