Peter M. Groen / openbr

Commit e214a85a7017d14f186e81fe3738aa11d9cef2af

Authored by Josh Klontz
1 parent faf5f230

remove delaunay

Inline Side-by-side
Showing 1 changed file with 0 additions and 172 deletions
openbr/plugins/metadata/delaunay.cpp deleted
View file @faf5f23
1   -/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
2   - * Copyright 2012 The MITRE Corporation *
3   - * *
4   - * Licensed under the Apache License, Version 2.0 (the "License"); *
5   - * you may not use this file except in compliance with the License. *
6   - * You may obtain a copy of the License at *
7   - * *
8   - * http://www.apache.org/licenses/LICENSE-2.0 *
9   - * *
10   - * Unless required by applicable law or agreed to in writing, software *
11   - * distributed under the License is distributed on an "AS IS" BASIS, *
12   - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *
13   - * See the License for the specific language governing permissions and *
14   - * limitations under the License. *
15   - * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
16   -
17   -#include <opencv2/imgproc/imgproc.hpp>
18   -#include <Eigen/Dense>
19   -
20   -#include <openbr/plugins/openbr_internal.h>
21   -#include <openbr/core/opencvutils.h>
22   -
23   -using namespace cv;
24   -
25   -namespace br
26   -{
27   -
28   -/*!
29   - * \ingroup transforms
30   - * \brief Creates a Delaunay triangulation based on a set of points
31   - * \author Scott Klum \cite sklum
32   - */
33   -class DelaunayTransform : public UntrainableTransform
34   -{
35   - Q_OBJECT
36   -
37   - Q_PROPERTY(float scaleFactor READ get_scaleFactor WRITE set_scaleFactor RESET reset_scaleFactor STORED false)
38   - Q_PROPERTY(bool warp READ get_warp WRITE set_warp RESET reset_warp STORED false)
39   - BR_PROPERTY(float, scaleFactor, 1)
40   - BR_PROPERTY(bool, warp, true)
41   -
42   - void project(const Template &src, Template &dst) const
43   - {
44   - QList<QPointF> points = src.file.points();
45   - QList<QRectF> rects = src.file.rects();
46   -
47   - if (points.empty() || rects.empty()) {
48   - dst = src;
49   - if (Globals->verbose) qWarning("Delauney triangulation failed because points or rects are empty.");
50   - return;
51   - }
52   -
53   - int cols = src.m().cols;
54   - int rows = src.m().rows;
55   -
56   - // Assume rect appended last was bounding box
57   - points.append(rects.last().topLeft());
58   - points.append(rects.last().topRight());
59   - points.append(rects.last().bottomLeft());
60   - points.append(rects.last().bottomRight());
61   -
62   - Subdiv2D subdiv(Rect(0,0,cols,rows));
63   - // Make sure points are valid for Subdiv2D
64   - // TODO: Modify points to make them valid
65   - for (int i = 0; i < points.size(); i++) {
66   - if (points[i].x() < 0 || points[i].y() < 0 || points[i].y() >= rows || points[i].x() >= cols) {
67   - dst = src;
68   - if (Globals->verbose) qWarning("Delauney triangulation failed because points lie on boundary.");
69   - return;
70   - }
71   - subdiv.insert(OpenCVUtils::toPoint(points[i]));
72   - }
73   -
74   - std::vector<Vec6f> triangleList;
75   - subdiv.getTriangleList(triangleList);
76   -
77   - QList<QPointF> validTriangles;
78   -
79   - for (size_t i = 0; i < triangleList.size(); i++) {
80   - // Check the triangle to make sure it's falls within the matrix
81   - bool valid = true;
82   -
83   - QList<QPointF> vertices;
84   - vertices.append(QPointF(triangleList[i][0],triangleList[i][1]));
85   - vertices.append(QPointF(triangleList[i][2],triangleList[i][3]));
86   - vertices.append(QPointF(triangleList[i][4],triangleList[i][5]));
87   - for (int j = 0; j < 3; j++) if (vertices[j].x() > cols || vertices[j].y() > rows || vertices[j].x() < 0 || vertices[j].y() < 0) valid = false;
88   -
89   - if (valid) validTriangles.append(vertices);
90   - }
91   -
92   - if (warp) {
93   - dst.m() = Mat::zeros(rows,cols,src.m().type());
94   -
95   - QList<float> procrustesStats = src.file.getList<float>("ProcrustesStats");
96   -
97   - Eigen::MatrixXf R(2,2);
98   - R(0,0) = procrustesStats.at(0);
99   - R(1,0) = procrustesStats.at(1);
100   - R(1,1) = procrustesStats.at(2);
101   - R(0,1) = procrustesStats.at(3);
102   -
103   - cv::Scalar mean(2);
104   - mean[0] = procrustesStats.at(4);
105   - mean[1] = procrustesStats.at(5);
106   -
107   - float norm = procrustesStats.at(6);
108   -
109   - QList<Point2f> mappedPoints;
110   -
111   - for (int i = 0; i < validTriangles.size(); i+=3) {
112   - // Matrix to store original (pre-transformed) triangle vertices
113   - Eigen::MatrixXf srcMat(3, 2);
114   -
115   - for (int j = 0; j < 3; j++) {
116   - srcMat(j,0) = (validTriangles[i+j].x()-mean[0])/norm;
117   - srcMat(j,1) = (validTriangles[i+j].y()-mean[1])/norm;
118   - }
119   -
120   - Eigen::MatrixXf dstMat = srcMat*R;
121   -
122   - Point2f srcPoints[3];
123   - for (int j = 0; j < 3; j++) srcPoints[j] = OpenCVUtils::toPoint(validTriangles[i+j]);
124   -
125   - Point2f dstPoints[3];
126   - for (int j = 0; j < 3; j++) {
127   - // Scale and shift destination points
128   - Point2f warpedPoint = Point2f(dstMat(j,0)*scaleFactor+cols/2,dstMat(j,1)*scaleFactor+rows/2);
129   - dstPoints[j] = warpedPoint;
130   - mappedPoints.append(warpedPoint);
131   - }
132   -
133   - Mat buffer(rows,cols,src.m().type());
134   -
135   - warpAffine(src.m(), buffer, getAffineTransform(srcPoints, dstPoints), Size(cols,rows));
136   -
137   - Mat mask = Mat::zeros(rows, cols, CV_8UC1);
138   - Point maskPoints[1][3];
139   - maskPoints[0][0] = dstPoints[0];
140   - maskPoints[0][1] = dstPoints[1];
141   - maskPoints[0][2] = dstPoints[2];
142   - const Point* ppt = { maskPoints[0] };
143   -
144   - fillConvexPoly(mask, ppt, 3, Scalar(255,255,255), 8);
145   -
146   - Mat output(rows,cols,src.m().type());
147   -
148   - if (i > 0) {
149   - Mat overlap;
150   - bitwise_and(dst.m(),mask,overlap);
151   - mask.setTo(0, overlap!=0);
152   - }
153   -
154   - bitwise_and(buffer,mask,output);
155   -
156   - dst.m() += output;
157   - }
158   -
159   - // Overwrite any rects
160   - Rect boundingBox = boundingRect(mappedPoints.toVector().toStdVector());
161   - dst.file.setRects(QList<QRectF>() << OpenCVUtils::fromRect(boundingBox));
162   - } else dst = src;
163   -
164   - dst.file.setList<QPointF>("DelaunayTriangles", validTriangles);
165   - }
166   -};
167   -
168   -BR_REGISTER(Transform, DelaunayTransform)
169   -
170   -} // namespace br
171   -
172   -#include "metadata/delaunay.moc"