From f4b5c89c4197f2342e3182cc290b6d37a005f855 Mon Sep 17 00:00:00 2001
From: Josh Klontz <josh@rankone.io>
Date: Thu, 21 Oct 2021 14:28:07 -0600
Subject: [PATCH] Revert "remove check"

---
 openbr/plugins/metadata/check.cpp       | 59 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 openbr/plugins/metadata/checkpoints.cpp | 41 +++++++++++++++++++++++++++++++++++++++++
 openbr/plugins/metadata/checkrects.cpp  | 73 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 173 insertions(+), 0 deletions(-)
 create mode 100644 openbr/plugins/metadata/check.cpp
 create mode 100644 openbr/plugins/metadata/checkpoints.cpp
 create mode 100644 openbr/plugins/metadata/checkrects.cpp

diff --git a/openbr/plugins/metadata/check.cpp b/openbr/plugins/metadata/check.cpp
new file mode 100644
index 0000000..3e62234
--- /dev/null
+++ b/openbr/plugins/metadata/check.cpp
@@ -0,0 +1,59 @@
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
+ * Copyright 2012 The MITRE Corporation                                      *
+ *                                                                           *
+ * Licensed under the Apache License, Version 2.0 (the "License");           *
+ * you may not use this file except in compliance with the License.          *
+ * You may obtain a copy of the License at                                   *
+ *                                                                           *
+ *     http://www.apache.org/licenses/LICENSE-2.0                            *
+ *                                                                           *
+ * Unless required by applicable law or agreed to in writing, software       *
+ * distributed under the License is distributed on an "AS IS" BASIS,         *
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  *
+ * See the License for the specific language governing permissions and       *
+ * limitations under the License.                                            *
+ * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+#include <openbr/plugins/openbr_internal.h>
+
+using namespace cv;
+
+namespace br
+{
+
+/*!
+ * \ingroup transforms
+ * \brief Checks the Template for NaN values.
+ * \author Josh Klontz \cite jklontz
+ */
+class CheckTransform : public UntrainableMetaTransform
+{
+    Q_OBJECT
+    static int count;
+    int index;
+
+ public:
+    CheckTransform() : index(count++) {}
+
+    void project(const Template &src, Template &dst) const
+    {
+        dst = src;
+        foreach (const Mat &m, src) {
+            Mat fm;
+            m.convertTo(fm, CV_32F);
+            const int elements = fm.rows * fm.cols * fm.channels();
+            const float *data = (const float*)fm.data;
+            for (int i=0; i<elements; i++)
+                if (data[i] != data[i])
+                    qFatal("%s NaN check %d failed!", qPrintable(src.file.flat()), index);
+        }
+    }
+};
+
+int CheckTransform::count = 0;
+
+BR_REGISTER(Transform, CheckTransform)
+
+} // namespace br
+
+#include "metadata/check.moc"
diff --git a/openbr/plugins/metadata/checkpoints.cpp b/openbr/plugins/metadata/checkpoints.cpp
new file mode 100644
index 0000000..3f03f61
--- /dev/null
+++ b/openbr/plugins/metadata/checkpoints.cpp
@@ -0,0 +1,41 @@
+#include <openbr/plugins/openbr_internal.h>
+
+namespace br
+{
+
+/*!
+ * \ingroup transforms
+ * \brief Checks the points in a template for missing (-1,-1) values
+ * \author Scott Klum \cite sklum
+ * \br_property QList<int> indices Indices of points to check.
+ */
+class CheckPointsTransform : public UntrainableMetadataTransform
+{
+    Q_OBJECT
+
+    Q_PROPERTY(QList<int> indices READ get_indices WRITE set_indices RESET reset_indices STORED false)
+    Q_PROPERTY(int count READ get_count WRITE set_count RESET reset_count STORED false)
+    BR_PROPERTY(QList<int>, indices, QList<int>())
+    BR_PROPERTY(int, count, 0)
+
+    void projectMetadata(const File &src, File &dst) const
+    {
+        dst = src;
+
+        const QList<QPointF> points = src.points();
+        if (count && points.size() < count)
+            dst.fte = true;
+
+        for (int i=0; i<indices.size(); i++)
+            if (src.points()[indices[i]] == QPointF(-1,-1)) {
+                dst.fte = true;
+                break;
+            }
+    }
+};
+
+BR_REGISTER(Transform, CheckPointsTransform)
+
+} // namespace br
+
+#include "metadata/checkpoints.moc"
diff --git a/openbr/plugins/metadata/checkrects.cpp b/openbr/plugins/metadata/checkrects.cpp
new file mode 100644
index 0000000..1471759
--- /dev/null
+++ b/openbr/plugins/metadata/checkrects.cpp
@@ -0,0 +1,73 @@
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
+ * Copyright 2015 Rank One Computing Corporation                             *
+ *                                                                           *
+ * Licensed under the Apache License, Version 2.0 (the "License");           *
+ * you may not use this file except in compliance with the License.          *
+ * You may obtain a copy of the License at                                   *
+ *                                                                           *
+ *     http://www.apache.org/licenses/LICENSE-2.0                            *
+ *                                                                           *
+ * Unless required by applicable law or agreed to in writing, software       *
+ * distributed under the License is distributed on an "AS IS" BASIS,         *
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  *
+ * See the License for the specific language governing permissions and       *
+ * limitations under the License.                                            *
+ * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+#include <openbr/plugins/openbr_internal.h>
+
+namespace br
+{
+
+/*!
+ * \ingroup transforms
+ * \brief Checks the rects in a template for invalid values. The user can specify to fix the rects or remove them.
+ * \author Keyur Patel \cite kpatel
+ */
+
+class CheckRectsTransform : public UntrainableTransform
+{
+    Q_OBJECT
+    Q_PROPERTY(bool fixRects READ get_fixRects WRITE set_fixRects RESET reset_fixRects STORED false)
+    Q_PROPERTY(bool removeBadRect READ get_removeBadRect WRITE set_removeBadRect RESET reset_removeBadRect STORED false)
+    BR_PROPERTY(bool, fixRects, false)
+    BR_PROPERTY(bool, removeBadRect, true)
+
+    void project(const Template &src, Template &dst) const
+    {
+        dst = src;
+        dst.file.clearRects();
+        QList<QRectF> rects = src.file.rects();
+
+        if (fixRects) {
+            foreach (QRectF r, rects)
+                if ((r.height() <= src.m().rows) && (r.width() <= src.m().cols)) /* can't fix rects that don't fit the image */ {
+                    if (r.left() < 0)                   r.moveLeft(0);
+                    if (r.right() > src.m().cols-1)     r.moveRight(src.m().cols-1);
+                    if (r.top() < 0)                    r.moveTop(0);
+                    if (r.bottom() > src.m().rows-1)    r.moveBottom(src.m().rows-1);
+                    dst.file.appendRect(r);
+                }
+        } else {
+            foreach (QRectF r, rects){
+                if ((r.left() < 0) || (r.right() > src.m().cols-1) || (r.top() < 0) || (r.bottom() > src.m().rows-1)){
+                    if (removeBadRect){
+                        rects.removeOne(r);
+                    }
+                    else {
+                        dst.file.fte = true;
+                        break;
+                    }
+                }
+            }
+            dst.file.setRects(rects);
+        }
+    }
+};
+
+BR_REGISTER(Transform, CheckRectsTransform)
+
+} // namespace br
+
+#include "metadata/checkrects.moc"
+
--
libgit2 0.21.4