123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144 |
- #include<cmath>
- #include<algorithm>
- #ifdef HAS_OPENCV3
- #include <opencv2/core.hpp> //Any OPENCV3 code
- #else
- #include <opencv2/core/core.hpp> //Any Opencv2 code
- #endif
- #include"lines.h"
- #include "geometry.h"
- #include "cvutils.h"
- using namespace cv;
- #define likely(x) __builtin_expect(!!(x), 1)
- #define unlikely(x) __builtin_expect(!!(x), 0)
- // all those colors are still visible on b/w (the first component is not so low)
- auto WHITE = cv::Scalar(255,255,255);
- auto BLUE = cv::Scalar(200,50,50);
- auto MAGENTA = cv::Scalar(110,10,200);
- auto BROWN = cv::Scalar(90,100,60);
- auto BLACK = cv::Scalar(0,0,0);
- auto YELLOW = cv::Scalar(20,200,200);
- int main(int argc, char *argv[])
- {
- char const *fname = "files/masckera.png";
- if( 1<argc )
- fname = argv[1];
- if( 2<argc) {
- std::cerr << "Too many arguments" << std::endl;
- return EXIT_FAILURE;
- }
- cv::Mat img;
- img=cv::imread(fname,CV_LOAD_IMAGE_GRAYSCALE);
- if( img.empty() ) {
- std::cerr << "Error opening image, aborting" << std::endl;
- return EXIT_FAILURE;
- }
- std::vector< std::vector<cv::Point> > contours;
- std::vector<cv::Vec4i> hierarchy;
- cv::findContours(img,contours,hierarchy,CV_RETR_EXTERNAL,CV_CHAIN_APPROX_SIMPLE);
- unsigned short dotwidth = img.cols >> 6; // divide by 64, so efficient
- cv::drawContours(img,contours,-1,BROWN,dotwidth>>4);
- if( 0==contours.size() ) {
- std::cerr << "No contours found" << std::endl;
- return EXIT_FAILURE;
- }
- auto contour = contours[0];
- if( 1!=contours.size() )
- {
- // choosing the biggest contour in the image
- // you can test it with files/2contours.png
- for(auto cont: contours) {
- if(cont.size() > contour.size())
- contour = cont;
- }
- }
- std::vector<cv::Point> hull;
- cv::convexHull(cv::Mat(contour),hull,false);
- //these are the "horizontal" lines
- unsigned *maxdistances = max2_distance(hull);
- cv::Point corn_1, corn_2, corn_3, corn_4;
- corn_1 = hull[maxdistances[0]];
- corn_2 = hull[(maxdistances[0]+1)%hull.size()];
- corn_3 = hull[maxdistances[1]];
- corn_4 = hull[(maxdistances[1]+1)%hull.size()];
- #ifdef _DEBUG
- img = cv::imread(fname,CV_LOAD_IMAGE_COLOR); // uncomment to this to display image with colors
- std::cout << "Maxdist1:" << maxdistances[0] << " " << corn_1 << corn_2 << std::endl;
- std::cout << "Maxdist2:" << maxdistances[1] << " " << corn_3 << corn_4 << std::endl;
- cv::namedWindow("win", CV_GUI_NORMAL);
- // cv::imshow("4 corners",img);
- // cv::waitKey(0);
- #endif
- std::vector<std::vector<cv::Point>> verticals;
- verticals.push_back(find_longest_line(hull, (maxdistances[0]+1)%hull.size(), maxdistances[1]));
- std::cout << maxdistances[1] << std::endl;
- std::cout << maxdistances[1]+1 << std::endl;
- std::cout << (maxdistances[1]+1)%hull.size() << std::endl;
- verticals.push_back(find_longest_line(hull, (maxdistances[1]+1)%hull.size(), maxdistances[0]));
- free(maxdistances);
- // theta is the angle of the line connecting point 1 and 2; it will be the
- // rotation of our new coordinate system
- auto cs = CoordinateSystem(corn_1, corn_2);
- assert(cs.map(corn_1).x == 0);
- assert(cs.map(corn_1).y == 0);
- assert(cs.map(corn_2).x > 0);
- assert(cs.map(corn_2).y == 0);
- cv::Vec<double,3> diag1, diag2;
- diag1 = get_line(cs.map(corn_1), cs.map(corn_3));
- diag2 = get_line(cs.map(corn_4), cs.map(corn_2));
- std::cout << "mapped diag1: " << diag1 << std::endl;
- std::cout << "mapped diag2: " << diag2 << std::endl;
- std::vector<cv::Point> points1, points2;
- for(cv::Point p: contour)
- { // the point is interesting where it is above both lines or below both lines
- cv::Point mapped = cs.map(p);
- if(is_above_line(diag1, mapped)) {
- if(is_above_line(diag2, mapped)) {
- points1.push_back(p);
- }
- } else {
- if(!is_above_line(diag2, mapped)) {
- points2.push_back(p);
- }
- }
- }
- #ifdef _DEBUG
- cv::Point middle1 = further_point_from_line(get_line(corn_1, corn_2), points2);
- cv::Point middle2 = further_point_from_line(get_line(corn_3, corn_4), points1);
- cv::line(img, corn_1, middle1, MAGENTA, dotwidth>>3);
- cv::line(img, corn_2, middle1, MAGENTA, dotwidth>>3);
- cv::line(img, middle1, middle2, BLACK, dotwidth>>4);
- cv::line(img, corn_3, middle2, MAGENTA, dotwidth>>3);
- cv::line(img, corn_4, middle2, MAGENTA, dotwidth>>3);
- cv::line(img, corn_2, corn_3, BLUE, dotwidth>>4);
- cv::line(img, corn_1, corn_4, BLUE, dotwidth>>4);
- std::cout << middle1 << middle2 << std::endl;
- cv::imshow("win",img);
- while(1)
- {
- if( (char)cv::waitKey(0) == 113 )
- break;
- }
- img.release();
- #endif
- return EXIT_SUCCESS;
- }
- // vim: set noet ts=4 sw=4:
|