Moved linesegment class to statedetector

src/statedetection/CMakeLists.txt

@@ -4,6 +4,7 @@
 set(statedetector_source_files
   state_detector.cpp
   featurematcher.cpp
+  line_segment.cpp
   state_detector_impl.cpp
 )
 

src/statedetection/featurematcher.h

@@ -24,11 +24,10 @@
 #include "opencv2/imgproc/imgproc.hpp"
 #include "opencv2/features2d/features2d.hpp"
 #include "opencv2/video/tracking.hpp"
+#include "opencv2/highgui/highgui.hpp"
 
+#include "line_segment.h"
 #include "support/filesystem.h"
-#include "constants.h"
-#include "utility.h"
-#include "config.h"
 
 namespace alpr
 {

src/statedetection/line_segment.cpp

@@ -0,0 +1,158 @@
+/*
+ * Copyright (c) 2015 OpenALPR Technology, Inc.
+ * Open source Automated License Plate Recognition [http://www.openalpr.com]
+ *
+ * This file is part of OpenALPR.
+ *
+ * OpenALPR is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License
+ * version 3 as published by the Free Software Foundation
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "line_segment.h"
+
+using namespace cv;
+
+LineSegment::LineSegment()
+{
+  init(0, 0, 0, 0);
+}
+
+LineSegment::LineSegment(Point p1, Point p2)
+{
+  init(p1.x, p1.y, p2.x, p2.y);
+}
+
+LineSegment::LineSegment(int x1, int y1, int x2, int y2)
+{
+  init(x1, y1, x2, y2);
+}
+
+void LineSegment::init(int x1, int y1, int x2, int y2)
+{
+  this->p1 = Point(x1, y1);
+  this->p2 = Point(x2, y2);
+
+  if (p2.x - p1.x == 0)
+    this->slope = 0.00000000001;
+  else
+    this->slope = (float) (p2.y - p1.y) / (float) (p2.x - p1.x);
+
+  this->length = distanceBetweenPoints(p1, p2);
+
+  this->angle = angleBetweenPoints(p1, p2);
+}
+
+bool LineSegment::isPointBelowLine( Point tp )
+{
+  return ((p2.x - p1.x)*(tp.y - p1.y) - (p2.y - p1.y)*(tp.x - p1.x)) > 0;
+}
+
+float LineSegment::getPointAt(float x)
+{
+  return slope * (x - p2.x) + p2.y;
+}
+
+Point LineSegment::closestPointOnSegmentTo(Point p)
+{
+  float top = (p.x - p1.x) * (p2.x - p1.x) + (p.y - p1.y)*(p2.y - p1.y);
+
+  float bottom = distanceBetweenPoints(p2, p1);
+  bottom = bottom * bottom;
+
+  float u = top / bottom;
+
+  float x = p1.x + u * (p2.x - p1.x);
+  float y = p1.y + u * (p2.y - p1.y);
+
+  return Point(x, y);
+}
+
+Point LineSegment::intersection(LineSegment line)
+{
+  float c1, c2;
+  float intersection_X = -1, intersection_Y= -1;
+
+  c1 = p1.y - slope * p1.x; // which is same as y2 - slope * x2
+
+  c2 = line.p2.y - line.slope * line.p2.x; // which is same as y2 - slope * x2
+
+  if( (slope - line.slope) == 0)
+  {
+    //std::cout << "No Intersection between the lines" << endl;
+  }
+  else if (p1.x == p2.x)
+  {
+    // Line1 is vertical
+    return Point(p1.x, line.getPointAt(p1.x));
+  }
+  else if (line.p1.x == line.p2.x)
+  {
+    // Line2 is vertical
+    return Point(line.p1.x, getPointAt(line.p1.x));
+  }
+  else
+  {
+    intersection_X = (c2 - c1) / (slope - line.slope);
+    intersection_Y = slope * intersection_X + c1;
+  }
+
+  return Point(intersection_X, intersection_Y);
+}
+
+Point LineSegment::midpoint()
+{
+  // Handle the case where the line is vertical
+  if (p1.x == p2.x)
+  {
+    float ydiff = p2.y-p1.y;
+    float y = p1.y + (ydiff/2);
+    return Point(p1.x, y);
+  }
+  float diff = p2.x - p1.x;
+  float midX = ((float) p1.x) + (diff / 2);
+  int midY = getPointAt(midX);
+
+  return Point(midX, midY);
+}
+
+LineSegment LineSegment::getParallelLine(float distance)
+{
+  float diff_x = p2.x - p1.x;
+  float diff_y = p2.y - p1.y;
+  float angle = atan2( diff_x, diff_y);
+  float dist_x = distance * cos(angle);
+  float dist_y = -distance * sin(angle);
+
+  int offsetX = (int)round(dist_x);
+  int offsetY = (int)round(dist_y);
+
+  LineSegment result(p1.x + offsetX, p1.y + offsetY,
+                     p2.x + offsetX, p2.y + offsetY);
+
+  return result;
+}
+
+double LineSegment::distanceBetweenPoints(Point p1, Point p2)
+{
+  float asquared = (p2.x - p1.x)*(p2.x - p1.x);
+  float bsquared = (p2.y - p1.y)*(p2.y - p1.y);
+
+  return sqrt(asquared + bsquared);
+}
+
+float LineSegment::angleBetweenPoints(Point p1, Point p2)
+{
+  int deltaY = p2.y - p1.y;
+  int deltaX = p2.x - p1.x;
+
+  return atan2((float) deltaY, (float) deltaX) * (180 / CV_PI);
+}

src/statedetection/line_segment.h

@@ -0,0 +1,67 @@
+/*
+ * Copyright (c) 2015 OpenALPR Technology, Inc.
+ * Open source Automated License Plate Recognition [http://www.openalpr.com]
+ *
+ * This file is part of OpenALPR.
+ *
+ * OpenALPR is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License
+ * version 3 as published by the Free Software Foundation
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef OPENALPR_LINE_SEGMENT_H
+#define OPENALPR_LINE_SEGMENT_H
+
+#include "opencv2/core/core.hpp"
+
+class LineSegment
+{
+
+public:
+  cv::Point p1, p2;
+  float slope;
+  float length;
+  float angle;
+
+  // LineSegment(Point point1, Point point2);
+  LineSegment();
+  LineSegment(int x1, int y1, int x2, int y2);
+  LineSegment(cv::Point p1, cv::Point p2);
+
+  void init(int x1, int y1, int x2, int y2);
+
+  bool isPointBelowLine(cv::Point tp);
+
+  float getPointAt(float x);
+
+  cv::Point closestPointOnSegmentTo(cv::Point p);
+
+  cv::Point intersection(LineSegment line);
+
+  LineSegment getParallelLine(float distance);
+
+  cv::Point midpoint();
+
+  inline std::string str()
+  {
+    std::stringstream ss;
+    ss << "(" << p1.x << ", " << p1.y << ") : (" << p2.x << ", " << p2.y << ")";
+    return ss.str() ;
+  }
+private:
+
+  double distanceBetweenPoints(cv::Point p1, cv::Point p2);
+  float angleBetweenPoints(cv::Point p1, cv::Point p2);
+  
+};
+
+
+#endif //OPENALPR_LINE_SEGMENT_H