rm-archive/PuzzleSolver
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed errors if not enough corners were found!

Browse Source
This commit is contained in:
TabDragon 2017-12-29 15:00:43 +01:00
parent f76a5dd73d
commit ce6fc0db72
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
Source/functions/AbstractionLayers/Layer1/AbstractionLayer_1.cpp
View File

@ -17,7 +17,7 @@ bool AbstractionLayer_1::PreProcessing(coor mySize, const vector<Part*>* partAr
if(!analyse.getImages()) if(!analyse.getImages())
{ {
cerr << "Error: No pictures found!" << endl; cerr << "Error occured in getImages!" << endl;
return false; return false;
} }
else else
@ -373,6 +373,11 @@ bool analyseParts::getImages(){
mask.setContour(contours); mask.setContour(contours);
mask.setHierarchy(hierarchy); mask.setHierarchy(hierarchy);
corners = findCorners(contours[0],mask.getCenter()); corners = findCorners(contours[0],mask.getCenter());
if((corners.empty()) || (corners.size() < 4))
{
cerr << "Error occured in findCorners" << endl;
return false;
}
mask.setCorners(corners); mask.setCorners(corners);
mask.setTabs(analyseContour(corners,contours[0])); mask.setTabs(analyseContour(corners,contours[0]));
masks.push_back(mask); masks.push_back(mask);
@ -413,6 +418,13 @@ vector<Point> analyseParts::findCorners(vector<Point> contour, Point center){
buf3.push_back(contour[i]); buf3.push_back(contour[i]);
} }
} }
if((buf0.empty()) || (buf1.empty()) || (buf2.empty()) || (buf3.empty()))
{
// return an empty vector if not all corners were found!
return vector<Point>();
}
quad_contour.push_back(buf0); quad_contour.push_back(buf0);
quad_contour.push_back(buf1); quad_contour.push_back(buf1);
quad_contour.push_back(buf2); quad_contour.push_back(buf2);