started include of ab1 opencv - segfault!
added pictures into git for easier debug, can remove again
@ -10,6 +10,22 @@
|
|||||||
|
|
||||||
void AbstractionLayer_1::PreProcessing(coor mySize, const vector<Part*>* partArray)
|
void AbstractionLayer_1::PreProcessing(coor mySize, const vector<Part*>* partArray)
|
||||||
{
|
{
|
||||||
|
|
||||||
|
analyseParts analyse(1008);
|
||||||
|
vector<Part> parts;
|
||||||
|
Part buf;
|
||||||
|
unsigned char tabs = 0;
|
||||||
|
for(int i = 0; i < 1008; i++){
|
||||||
|
tabs = analyse.getTabs(i);
|
||||||
|
buf.m_a1.m_connections=tabs;
|
||||||
|
parts.push_back(buf);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
//Zugriff auf den vector mit den einzelnen teilen: part[0].getConnenctions() entspricht pömpel von bild 0.jpg und liefert ein unsigned char, poempl Belegung wie ausgemacht
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
InitialiseConstraintMatrixSize(mySize.col+2, mySize.row+2);
|
InitialiseConstraintMatrixSize(mySize.col+2, mySize.row+2);
|
||||||
setEdgeZero();
|
setEdgeZero();
|
||||||
}
|
}
|
||||||
@ -166,3 +182,483 @@ void AbstractionLayer_1_Properties::print()
|
|||||||
std::cout << std::bitset<8>(this->m_connections);
|
std::cout << std::bitset<8>(this->m_connections);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
Mat analyseParts::makeBorder(Mat& im_bw) {
|
||||||
|
Mat im_part;
|
||||||
|
|
||||||
|
Scalar value = (0,0,0);
|
||||||
|
Point center = findCenter(im_bw);
|
||||||
|
copyMakeBorder(im_bw, im_part, abs((IMG_SIZE/2)-center.y),abs((IMG_SIZE/2)-(im_bw.rows-center.y)), abs((IMG_SIZE/2)-center.x),abs( (IMG_SIZE/2)-(im_bw.cols-center.x )), BORDER_CONSTANT, value);
|
||||||
|
|
||||||
|
return im_part;
|
||||||
|
}
|
||||||
|
|
||||||
|
Mat analyseParts::readImages(int count)
|
||||||
|
{
|
||||||
|
char name[100];
|
||||||
|
Mat corr;
|
||||||
|
Mat ref_gray;
|
||||||
|
|
||||||
|
sprintf(name, PATH, count);
|
||||||
|
cout << "path" << name << endl;
|
||||||
|
|
||||||
|
Mat src = imread(name, 1);
|
||||||
|
if (!src.data)
|
||||||
|
cerr << "Problem loading image!!!" << endl;
|
||||||
|
if(DISPLAY)imshow("src",src);
|
||||||
|
|
||||||
|
Mat im_gray, im_bw;
|
||||||
|
cvtColor(src, im_gray, CV_RGB2GRAY);
|
||||||
|
im_bw = (im_gray > 220);
|
||||||
|
im_bw = 255 - im_bw;
|
||||||
|
im_bw = makeBorder(im_bw);
|
||||||
|
return im_bw;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
Mat analyseParts::morphDilateErode(Mat &im_bw)
|
||||||
|
{
|
||||||
|
Mat dst = im_bw.clone();
|
||||||
|
int operation = 3;
|
||||||
|
int morph_size = 2.4;
|
||||||
|
int dilation_size = 2;
|
||||||
|
int dilation_size1 = 1.5;
|
||||||
|
int erosion_size = 2;
|
||||||
|
int erosion_type = MORPH_RECT;
|
||||||
|
int dilation_type = MORPH_RECT;
|
||||||
|
Mat element = getStructuringElement(0, Size(2 * morph_size + 1, 2 * morph_size + 1), Point(morph_size, morph_size));
|
||||||
|
|
||||||
|
Mat element1 = getStructuringElement(dilation_type, Size(2 * dilation_size + 1, 2 * dilation_size + 1), Point(dilation_size, dilation_size));
|
||||||
|
Mat element3 = getStructuringElement(dilation_type, Size(2 * dilation_size1 + 1, 2 * dilation_size1 + 1), Point(dilation_size1, dilation_size1));
|
||||||
|
Mat element2 = getStructuringElement(erosion_type, Size(2 * erosion_size, 2 * erosion_size), Point(erosion_size, erosion_size));
|
||||||
|
|
||||||
|
morphologyEx(im_bw, dst, operation, element);
|
||||||
|
dilate( dst, dst, element );
|
||||||
|
erode( dst, dst, element2 );
|
||||||
|
|
||||||
|
return dst;
|
||||||
|
}
|
||||||
|
|
||||||
|
vector<vector<Point>> analyseParts::findingContours(Mat& dst)
|
||||||
|
{
|
||||||
|
|
||||||
|
vector<vector<Point> > contours1;
|
||||||
|
vector<vector<Point> > contours;
|
||||||
|
vector<Point> poly;
|
||||||
|
vector<Vec4i> hierarchy1;
|
||||||
|
vector<Vec4i> hierarchy;
|
||||||
|
int dilation_size1 = 1.5;
|
||||||
|
int erosion_size = 2;
|
||||||
|
int erosion_type = MORPH_RECT;
|
||||||
|
int dilation_type = MORPH_RECT;
|
||||||
|
|
||||||
|
Mat element3 = getStructuringElement(dilation_type, Size(2 * dilation_size1 + 1, 2 * dilation_size1 + 1), Point(dilation_size1, dilation_size1));
|
||||||
|
Mat element2 = getStructuringElement(erosion_type, Size(2 * erosion_size, 2 * erosion_size), Point(erosion_size, erosion_size));
|
||||||
|
|
||||||
|
Mat dst1 = Mat::zeros(dst .size(), CV_8UC1);
|
||||||
|
findContours(dst, contours1, hierarchy1, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_TC89_L1, Point(0, 0));
|
||||||
|
for (int i = 0; i < contours1.size(); i++) {
|
||||||
|
drawContours(dst1, contours1, i, Scalar(255, 255, 255), 2, 8, hierarchy1, 0);
|
||||||
|
}
|
||||||
|
dilate( dst1, dst1, element3 );
|
||||||
|
erode( dst1, dst1, element2 );
|
||||||
|
findContours(dst1, contours, hierarchy, CV_RETR_CCOMP, CV_CHAIN_APPROX_SIMPLE, Point(0, 0));
|
||||||
|
return contours;
|
||||||
|
}
|
||||||
|
|
||||||
|
Mat analyseParts::polyApprox(vector<vector<Point>>& contours)
|
||||||
|
{
|
||||||
|
vector<vector<Point>> largestContour;
|
||||||
|
largestContour.resize(contours.size());
|
||||||
|
int erosion_size = 2;
|
||||||
|
int erosion_type = MORPH_RECT;
|
||||||
|
Mat element2 = getStructuringElement(erosion_type, Size(2 * erosion_size, 2 * erosion_size), Point(erosion_size, erosion_size));
|
||||||
|
int index = 0;
|
||||||
|
double area = 0.0;
|
||||||
|
int strength = 5;
|
||||||
|
vector<Vec4i> hierarchy;
|
||||||
|
Mat mask_gray;
|
||||||
|
|
||||||
|
|
||||||
|
for (int i = 0; i < contours.size(); i++) {
|
||||||
|
approxPolyDP(Mat(contours[i]), largestContour[index], strength, true);
|
||||||
|
index++;
|
||||||
|
}
|
||||||
|
|
||||||
|
Mat mask = Mat::zeros(Point(IMG_SIZE,IMG_SIZE), CV_8UC3);
|
||||||
|
for (int i = 0; i < largestContour.size(); i++) {
|
||||||
|
area = contourArea(largestContour[i]);
|
||||||
|
if(area > 500 )
|
||||||
|
drawContours(mask, largestContour, i, Scalar(255, 255, 255), CV_FILLED, 8, hierarchy, 0);
|
||||||
|
}
|
||||||
|
erode( mask, mask, element2 );
|
||||||
|
|
||||||
|
cvtColor(mask, mask_gray, CV_RGB2GRAY);
|
||||||
|
mask_gray = (mask_gray > 200);
|
||||||
|
|
||||||
|
return mask_gray;
|
||||||
|
}
|
||||||
|
|
||||||
|
Mat createEmpty(Point size, int color) {
|
||||||
|
if (color) {
|
||||||
|
Mat empty = Mat::zeros(size, CV_8UC3);
|
||||||
|
return empty;
|
||||||
|
} else {
|
||||||
|
Mat empty = Mat::zeros(size, CV_8UC1);
|
||||||
|
return empty;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
float analyseParts::lengthTwoPoints (Point one, Point two) {
|
||||||
|
float length = 0;
|
||||||
|
length = sqrt((one.x - two.x) * (one.x - two.x) + (one.y - two.y)*(one.y - two.y));
|
||||||
|
return length;
|
||||||
|
}
|
||||||
|
|
||||||
|
float analyseParts::angle(Point one, Point two, Point three) {
|
||||||
|
float angle = 0;
|
||||||
|
float disa, disb, disc;
|
||||||
|
|
||||||
|
disa = lengthTwoPoints(two, three);
|
||||||
|
disb = lengthTwoPoints(one, three);
|
||||||
|
disc = lengthTwoPoints(one, two);
|
||||||
|
|
||||||
|
angle = acos( ((disa*disa) + (disc *disc) - (disb * disb)) / (2 * disa * disc) );
|
||||||
|
angle = angle * 180 / CV_PI;
|
||||||
|
return angle;
|
||||||
|
}
|
||||||
|
|
||||||
|
void analyseParts::getImages(){
|
||||||
|
Details mask;
|
||||||
|
Mat src;
|
||||||
|
vector<vector<Point> > contours;
|
||||||
|
vector<Vec4i> hierarchy;
|
||||||
|
vector<Point> corners;
|
||||||
|
|
||||||
|
vector<Mat> puzzleimages;
|
||||||
|
vector<vector<Point> > contours1;
|
||||||
|
Mat mask_gray;
|
||||||
|
|
||||||
|
for (int i = 0; i < nr_parts; i++) {
|
||||||
|
if(DISPLAY) cout << "Bild " << i << endl;
|
||||||
|
Mat im_bw = readImages(i);
|
||||||
|
Mat dst = morphDilateErode(im_bw);
|
||||||
|
contours1 = findingContours(dst);
|
||||||
|
mask_gray = polyApprox(contours1);
|
||||||
|
mask.setImage(mask_gray);
|
||||||
|
mask.setCenter(findCenter(mask_gray));
|
||||||
|
findContours(mask_gray, contours, hierarchy, CV_RETR_CCOMP, CV_CHAIN_APPROX_SIMPLE, Point(0, 0));
|
||||||
|
mask.setContour(contours);
|
||||||
|
mask.setHierarchy(hierarchy);
|
||||||
|
corners = findCorners(contours[0],mask.getCenter());
|
||||||
|
mask.setCorners(corners);
|
||||||
|
mask.setTabs(analyseContour(corners,contours[0]));
|
||||||
|
masks.push_back(mask);
|
||||||
|
destroyAllWindows();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Point analyseParts::findCenter(Mat img){
|
||||||
|
Mat im = img.clone();
|
||||||
|
Canny(im, im, 800, 1000, 3);
|
||||||
|
vector<vector<Point> > contours;
|
||||||
|
vector<Vec4i> hierarchy;
|
||||||
|
findContours( im, contours, hierarchy, CV_RETR_TREE, CV_CHAIN_APPROX_SIMPLE, Point(0, 0) );
|
||||||
|
Moments m = moments(im, true);
|
||||||
|
Point center(m.m10/m.m00, m.m01/m.m00);
|
||||||
|
return center;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
vector<Point> analyseParts::findCorners(vector<Point> contour, Point center){
|
||||||
|
int minContourPoint = 5;
|
||||||
|
vector<vector<Point>> quad_contour;
|
||||||
|
vector<Point> buf0;
|
||||||
|
vector<Point> buf1;
|
||||||
|
vector<Point> buf2;
|
||||||
|
vector<Point> buf3;
|
||||||
|
|
||||||
|
for(int i = 0; i < contour.size();i++){
|
||||||
|
if(contour[i].x > center.x && contour[i].y > center.y) {
|
||||||
|
buf0.push_back(contour[i]);
|
||||||
|
} else if(contour[i].x < center.x && contour[i].y > center.y) {
|
||||||
|
buf1.push_back(contour[i]);
|
||||||
|
} else if(contour[i].x > center.x && contour[i].y < center.y) {
|
||||||
|
buf2.push_back(contour[i]);
|
||||||
|
} else if(contour[i].x < center.x && contour[i].y < center.y) {
|
||||||
|
buf3.push_back(contour[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
quad_contour.push_back(buf0);
|
||||||
|
quad_contour.push_back(buf1);
|
||||||
|
quad_contour.push_back(buf2);
|
||||||
|
quad_contour.push_back(buf3);
|
||||||
|
|
||||||
|
Mat drawing = createEmpty(Point(IMG_SIZE,IMG_SIZE),1);
|
||||||
|
for(int j = 0; j < 4;j++){
|
||||||
|
for(int i = 0; i < quad_contour[j].size(); i++) {
|
||||||
|
circle(drawing,quad_contour[j][i],2,Scalar(abs(j+2)*50,abs(j+2)*20,0),3,8);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
circle(drawing,center,2,Scalar(0,255,255),3,8,0);
|
||||||
|
|
||||||
|
/*finde ecke rechts unten*/
|
||||||
|
vector<Point> corners;
|
||||||
|
float dist = 0;
|
||||||
|
float max_dist = 0;
|
||||||
|
int max_idx = 0;
|
||||||
|
int num = 0;
|
||||||
|
float distToCenter = 0;
|
||||||
|
for(int i = 0; i < (IMG_SIZE/2); i++){
|
||||||
|
for(int j = 0; j < quad_contour[0].size(); j++){
|
||||||
|
if(quad_contour[0][j].x > center.x+i && quad_contour[0][j].y > center.y+i)
|
||||||
|
num++;
|
||||||
|
}
|
||||||
|
if(num < minContourPoint) {
|
||||||
|
dist = i;
|
||||||
|
i = (IMG_SIZE/2);
|
||||||
|
}
|
||||||
|
num = 0;
|
||||||
|
}
|
||||||
|
for(int j = 0; j < quad_contour[0].size(); j++){
|
||||||
|
if(quad_contour[0][j].x > center.x+dist && quad_contour[0][j].y > center.y+dist) {
|
||||||
|
distToCenter = lengthTwoPoints(center,quad_contour[0][j]);
|
||||||
|
if(distToCenter > max_dist) {
|
||||||
|
max_dist = distToCenter;
|
||||||
|
max_idx = j;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
corners.push_back(quad_contour[0][max_idx]);
|
||||||
|
line(drawing,Point(center.x+dist,(IMG_SIZE/2)),Point(center.x+dist,IMG_SIZE),Scalar(255,0,255),3,8);
|
||||||
|
line(drawing,Point((IMG_SIZE/2),center.y+dist),Point(IMG_SIZE,center.y+dist),Scalar(255,0,255),3,8);
|
||||||
|
circle(drawing,quad_contour[0][max_idx],5,Scalar(50,100,255),5,8);
|
||||||
|
|
||||||
|
/*finde ecke links unten*/
|
||||||
|
dist = 0;
|
||||||
|
max_dist = 0;
|
||||||
|
max_idx = 0;
|
||||||
|
num = 0;
|
||||||
|
distToCenter = 0;
|
||||||
|
for(int i = 0; i < (IMG_SIZE/2); i++){
|
||||||
|
for(int j = 0; j < quad_contour[1].size(); j++){
|
||||||
|
if(quad_contour[1][j].x < center.x-i && quad_contour[1][j].y > center.y+i)
|
||||||
|
num++;
|
||||||
|
}
|
||||||
|
if(num < minContourPoint) {
|
||||||
|
dist = i;
|
||||||
|
i = (IMG_SIZE/2);
|
||||||
|
}
|
||||||
|
num = 0;
|
||||||
|
}
|
||||||
|
for(int j = 0; j < quad_contour[1].size(); j++){
|
||||||
|
if(quad_contour[1][j].x < center.x-dist && quad_contour[1][j].y > center.y+dist) {
|
||||||
|
distToCenter = lengthTwoPoints(center,quad_contour[1][j]);
|
||||||
|
if(distToCenter > max_dist) {
|
||||||
|
max_dist = distToCenter;
|
||||||
|
max_idx = j;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
corners.push_back(quad_contour[1][max_idx]);
|
||||||
|
line(drawing,Point(center.x-dist,(IMG_SIZE/2)),Point(center.x-dist,IMG_SIZE),Scalar(255,0,255),3,8);
|
||||||
|
line(drawing,Point((IMG_SIZE/2),center.y+dist),Point(0,center.y+dist),Scalar(255,0,255),3,8);
|
||||||
|
circle(drawing,quad_contour[1][max_idx],5,Scalar(50,100,255),5,8);
|
||||||
|
|
||||||
|
/*finde ecke rechts oben*/
|
||||||
|
dist = 0;
|
||||||
|
max_dist = 0;
|
||||||
|
max_idx = 0;
|
||||||
|
num = 0;
|
||||||
|
distToCenter = 0;
|
||||||
|
for(int i = 0; i < (IMG_SIZE/2); i++){
|
||||||
|
for(int j = 0; j < quad_contour[2].size(); j++){
|
||||||
|
if(quad_contour[2][j].x > center.x+i && quad_contour[2][j].y < center.y-i)
|
||||||
|
num++;
|
||||||
|
}
|
||||||
|
if(num < minContourPoint) {
|
||||||
|
dist = i;
|
||||||
|
i = (IMG_SIZE/2);
|
||||||
|
}
|
||||||
|
num = 0;
|
||||||
|
}
|
||||||
|
for(int j = 0; j < quad_contour[2].size(); j++){
|
||||||
|
if(quad_contour[2][j].x > center.x+dist && quad_contour[2][j].y < center.y-dist) {
|
||||||
|
distToCenter = lengthTwoPoints(center,quad_contour[2][j]);
|
||||||
|
if(distToCenter > max_dist) {
|
||||||
|
max_dist = distToCenter;
|
||||||
|
max_idx = j;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
corners.push_back(quad_contour[2][max_idx]);
|
||||||
|
line(drawing,Point(center.x+dist,(IMG_SIZE/2)),Point(center.x+dist,0),Scalar(255,0,255),3,8);
|
||||||
|
line(drawing,Point((IMG_SIZE/2),center.y-dist),Point(IMG_SIZE,center.y-dist),Scalar(255,0,255),3,8);
|
||||||
|
circle(drawing,quad_contour[2][max_idx],5,Scalar(50,100,255),5,8);
|
||||||
|
|
||||||
|
/*finde ecke links oben*/
|
||||||
|
dist = 0;
|
||||||
|
max_dist = 0;
|
||||||
|
max_idx = 0;
|
||||||
|
num = 0;
|
||||||
|
distToCenter = 0;
|
||||||
|
for(int i = 0; i < (IMG_SIZE/2); i++){
|
||||||
|
for(int j = 0; j < quad_contour[3].size(); j++){
|
||||||
|
if(quad_contour[3][j].x < center.x-i && quad_contour[3][j].y < center.y-i)
|
||||||
|
num++;
|
||||||
|
}
|
||||||
|
if(num < minContourPoint) {
|
||||||
|
dist = i;
|
||||||
|
i = (IMG_SIZE/2);
|
||||||
|
}
|
||||||
|
num = 0;
|
||||||
|
}
|
||||||
|
for(int j = 0; j < quad_contour[3].size(); j++){
|
||||||
|
if(quad_contour[3][j].x < center.x-dist && quad_contour[3][j].y < center.y-dist) {
|
||||||
|
distToCenter = lengthTwoPoints(center,quad_contour[3][j]);
|
||||||
|
if(distToCenter > max_dist) {
|
||||||
|
max_dist = distToCenter;
|
||||||
|
max_idx = j;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
corners.push_back(quad_contour[3][max_idx]);
|
||||||
|
line(drawing,Point(center.x-dist,(IMG_SIZE/2)),Point(center.x-dist,0),Scalar(255,0,255),3,8);
|
||||||
|
line(drawing,Point(0,center.y-dist),Point((IMG_SIZE/2),center.y-dist),Scalar(255,0,255),3,8);
|
||||||
|
circle(drawing,quad_contour[3][max_idx],5,Scalar(50,100,255),5,8);
|
||||||
|
if(DISPLAY) imshow("draw",drawing);
|
||||||
|
return corners;
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned char analyseParts::analyseContour(vector<Point> corners, vector<Point> contour) {
|
||||||
|
vector<Point> contour_right;
|
||||||
|
vector<Point> contour_top;
|
||||||
|
vector<Point> contour_left;
|
||||||
|
vector<Point> contour_bottom;
|
||||||
|
Mat drawing = createEmpty(Point(IMG_SIZE,IMG_SIZE),1);
|
||||||
|
int count = 0;
|
||||||
|
int corner0 = 0, corner1 = 0, corner2 = 0, corner3 = 0;
|
||||||
|
for(int i = 0; i < contour.size(); i++){
|
||||||
|
if(contour[i] == corners[0])
|
||||||
|
corner0 = i;
|
||||||
|
if(contour[i] == corners[1])
|
||||||
|
corner1 = i;
|
||||||
|
if(contour[i] == corners[2])
|
||||||
|
corner2 = i;
|
||||||
|
if(contour[i] == corners[3])
|
||||||
|
corner3 = i;
|
||||||
|
}
|
||||||
|
count = corner0;
|
||||||
|
while(contour[count] != contour[corner2]){
|
||||||
|
count++;
|
||||||
|
count %= contour.size();
|
||||||
|
contour_right.push_back(contour[count]);
|
||||||
|
circle(drawing,contour[count],3,Scalar(255,0,0),2,8);
|
||||||
|
}
|
||||||
|
count = corner2;
|
||||||
|
while(contour[count] != contour[corner3]){
|
||||||
|
count++;
|
||||||
|
count %= contour.size();
|
||||||
|
contour_top.push_back(contour[count]);
|
||||||
|
circle(drawing,contour[count],3,Scalar(0,255,0),2,8);
|
||||||
|
}
|
||||||
|
count = corner3;
|
||||||
|
while(contour[count] != contour[corner1]){
|
||||||
|
count++;
|
||||||
|
count %= contour.size();
|
||||||
|
contour_left.push_back(contour[count]);
|
||||||
|
circle(drawing,contour[count],3,Scalar(0,0,255),2,8);
|
||||||
|
}
|
||||||
|
count = corner1;
|
||||||
|
while(contour[count] != contour[corner0]){
|
||||||
|
count++;
|
||||||
|
count %= contour.size();
|
||||||
|
contour_bottom.push_back(contour[count]);
|
||||||
|
circle(drawing,contour[count],3,Scalar(255,255,255),2,8);
|
||||||
|
}
|
||||||
|
float ref_right = (contour[corner0].x+contour[corner2].x)/2;
|
||||||
|
float ref_top = (contour[corner2].y+contour[corner3].y)/2;
|
||||||
|
float ref_left = (contour[corner3].x+contour[corner1].x)/2;
|
||||||
|
float ref_bottom = (contour[corner1].y+contour[corner0].y)/2;
|
||||||
|
|
||||||
|
float max_dist = 0;
|
||||||
|
float dist = 0;
|
||||||
|
int max_idx = 0;
|
||||||
|
for(int i = 0; i < contour_right.size(); i++){
|
||||||
|
dist = abs(ref_right-contour_right[i].x);
|
||||||
|
if(dist > max_dist) {
|
||||||
|
max_dist = dist;
|
||||||
|
max_idx = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned char tabs = 0;
|
||||||
|
circle(drawing,contour_right[max_idx],10,Scalar(255,0,255),2,8);
|
||||||
|
if (ref_right - contour_right[max_idx].x <= -20)
|
||||||
|
tabs |= (2 << RIGHT);
|
||||||
|
if (ref_right - contour_right[max_idx].x >= 20)
|
||||||
|
tabs |= (1 << RIGHT);
|
||||||
|
if (abs(ref_right - contour_right[max_idx].x) < 20)
|
||||||
|
tabs |= (0 << RIGHT);
|
||||||
|
|
||||||
|
max_dist = 0;
|
||||||
|
dist = 0;
|
||||||
|
max_idx = 0;
|
||||||
|
for(int i = 0; i < contour_top.size(); i++){
|
||||||
|
dist = abs(ref_top-contour_top[i].y);
|
||||||
|
if(dist > max_dist) {
|
||||||
|
max_dist = dist;
|
||||||
|
max_idx = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
circle(drawing,contour_top[max_idx],10,Scalar(255,0,255),2,8);
|
||||||
|
if (ref_top - contour_top[max_idx].y <= -20)
|
||||||
|
tabs |= (1 << TOP);
|
||||||
|
if (ref_top - contour_top[max_idx].y >= 20)
|
||||||
|
tabs |= (2 << TOP);
|
||||||
|
if (abs(ref_top - contour_top[max_idx].y) < 20)
|
||||||
|
tabs |= (0 << TOP);
|
||||||
|
|
||||||
|
max_dist = 0;
|
||||||
|
dist = 0;
|
||||||
|
max_idx = 0;
|
||||||
|
for(int i = 0; i < contour_left.size(); i++){
|
||||||
|
dist = abs(ref_left-contour_left[i].x);
|
||||||
|
if(dist > max_dist) {
|
||||||
|
max_dist = dist;
|
||||||
|
max_idx = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
circle(drawing,contour_left[max_idx],10,Scalar(255,0,255),2,8);
|
||||||
|
if (ref_left - contour_left[max_idx].x <= -20)
|
||||||
|
tabs |= (1 << LEFT);
|
||||||
|
if (ref_left - contour_left[max_idx].x >= 20)
|
||||||
|
tabs |= (2 << LEFT);
|
||||||
|
if (abs(ref_left - contour_left[max_idx].x) < 20)
|
||||||
|
tabs |= (0 << LEFT);
|
||||||
|
|
||||||
|
max_dist = 0;
|
||||||
|
dist = 0;
|
||||||
|
max_idx = 0;
|
||||||
|
for(int i = 0; i < contour_bottom.size(); i++){
|
||||||
|
dist = abs(ref_bottom-contour_bottom[i].y);
|
||||||
|
if(dist > max_dist) {
|
||||||
|
max_dist = dist;
|
||||||
|
max_idx = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
circle(drawing,contour_bottom[max_idx],10,Scalar(255,0,255),2,8);
|
||||||
|
if (ref_bottom - contour_bottom[max_idx].y <= -20)
|
||||||
|
tabs |= (2 << BOTTOM);
|
||||||
|
if (ref_bottom - contour_bottom[max_idx].y >= 20)
|
||||||
|
tabs |= (1 << BOTTOM);
|
||||||
|
if (abs(ref_bottom - contour_bottom[max_idx].y) < 20)
|
||||||
|
tabs |= (0 << BOTTOM);
|
||||||
|
|
||||||
|
//cout << bitset<sizeof(char) * CHAR_BIT> (tabs) << "b\n";
|
||||||
|
|
||||||
|
if(DISPLAY)imshow("corners",drawing);
|
||||||
|
if(DISPLAY)waitKey(0);
|
||||||
|
return tabs;
|
||||||
|
}
|
||||||
|
|
||||||
|
@ -8,18 +8,39 @@
|
|||||||
#include "AbstractionLayer_1_Properties.h"
|
#include "AbstractionLayer_1_Properties.h"
|
||||||
#include "../AbstraktionLayer_Base.h"
|
#include "../AbstraktionLayer_Base.h"
|
||||||
|
|
||||||
|
#include <utility>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <bitset>
|
#include <bitset>
|
||||||
#include <random>
|
#include <random>
|
||||||
|
#include <iostream>
|
||||||
|
#include "opencv2/highgui/highgui.hpp"
|
||||||
|
#include "opencv2/imgproc/imgproc.hpp"
|
||||||
|
#include <iostream>
|
||||||
|
#include <cstdio>
|
||||||
|
#include <cstdlib>
|
||||||
|
#include <cmath>
|
||||||
|
#include <vector>
|
||||||
|
#include <bitset>
|
||||||
|
|
||||||
|
#define DISPLAY false
|
||||||
|
#define PATH "..\\pieces\\%d.png"
|
||||||
|
#define IMG_SIZE 400
|
||||||
|
#define TOP 6
|
||||||
|
#define RIGHT 4
|
||||||
|
#define BOTTOM 2
|
||||||
|
#define LEFT 0
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
using namespace cv;
|
||||||
|
|
||||||
class AbstractionLayer_1 : public AbstractionLayer_Base<AbstractionLayer_1_Properties>
|
class AbstractionLayer_1 : public AbstractionLayer_Base<AbstractionLayer_1_Properties>
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
void PreProcessing(coor mySize, const vector<Part*>* partArray) final;
|
void PreProcessing(coor mySize, const vector<Part*>* partArray) override ;
|
||||||
bool EvaluateQuality (const coor constraintCoordinate, qualityVector& qVector);
|
bool EvaluateQuality ( coor constraintCoordinate, qualityVector& qVector)override;
|
||||||
bool SetConstraintOnPosition(const coor constraintCoordinate, const AbstractionLayer_1_Properties constraint);
|
bool SetConstraintOnPosition( coor constraintCoordinate, AbstractionLayer_1_Properties constraint)override;
|
||||||
bool RemoveConstraintOnPosition(const coor constraintCoordinate);
|
bool RemoveConstraintOnPosition( coor constraintCoordinate)override;
|
||||||
bool PlaceOfPartGood(coor myCoor, uint8_t& myPart);
|
bool PlaceOfPartGood(coor myCoor, uint8_t& myPart);
|
||||||
void shift(uint8_t& Part, int shifts);
|
void shift(uint8_t& Part, int shifts);
|
||||||
void setEdgeZero();
|
void setEdgeZero();
|
||||||
@ -32,4 +53,55 @@ public:
|
|||||||
private:
|
private:
|
||||||
};
|
};
|
||||||
|
|
||||||
|
class Details{
|
||||||
|
public:
|
||||||
|
vector<vector<Point>> getContour(){return contour;}
|
||||||
|
Mat getImage(){return image;}
|
||||||
|
Point getCenter(){return center;}
|
||||||
|
vector<Vec4i> getHierarchy(){return hierarchy;}
|
||||||
|
unsigned char getTabs(){return tabs;}
|
||||||
|
|
||||||
|
void setContour(vector<vector<Point>> cont){contour = std::move(cont);}
|
||||||
|
void setImage(Mat im){image = std::move(im);}
|
||||||
|
void setCenter(const Point &c){center = c;}
|
||||||
|
void setHierarchy(vector<Vec4i> hier){hierarchy = std::move(hier);}
|
||||||
|
void setCorners(vector<Point> cor){corners = std::move(cor);}
|
||||||
|
void setTabs(unsigned char t){tabs = t;}
|
||||||
|
|
||||||
|
vector<Point> getCorners(){return corners;}
|
||||||
|
|
||||||
|
private:
|
||||||
|
Mat image;
|
||||||
|
vector<Point> corners;
|
||||||
|
vector<vector<Point>> contour;
|
||||||
|
vector<Vec4i> hierarchy;
|
||||||
|
Point center;
|
||||||
|
unsigned char tabs;
|
||||||
|
};
|
||||||
|
|
||||||
|
class analyseParts{
|
||||||
|
public:
|
||||||
|
explicit analyseParts(int s = 1008): nr_parts(s){getImages();}
|
||||||
|
Mat getImage(int i){if(i>= nr_parts)return masks[nr_parts-1].getImage(); else return masks[i].getImage();}
|
||||||
|
vector<vector<Point>> getContour(int i){if(i>= nr_parts)return masks[nr_parts-1].getContour(); else return masks[i].getContour();}
|
||||||
|
Point getCenter(int i){if(i>= nr_parts)return masks[nr_parts-1].getCenter(); else return masks[i].getCenter();}
|
||||||
|
vector<Vec4i> getHierarchy(int i){if(i>= nr_parts)return masks[nr_parts-1].getHierarchy(); else return masks[i].getHierarchy();}
|
||||||
|
unsigned char getTabs(int i){if(i>= nr_parts)return masks[nr_parts-1].getTabs(); else return masks[i].getTabs();}
|
||||||
|
Point findCenter(Mat);
|
||||||
|
vector<Point> findCorners(vector<Point>,Point);
|
||||||
|
unsigned char analyseContour(vector<Point>, vector<Point>);
|
||||||
|
Mat makeBorder(Mat&);
|
||||||
|
Mat readImages(int);
|
||||||
|
Mat morphDilateErode(Mat&);
|
||||||
|
vector<vector<Point>> findingContours(Mat&);
|
||||||
|
Mat polyApprox(vector<vector<Point>> &);
|
||||||
|
private:
|
||||||
|
void getImages();
|
||||||
|
float lengthTwoPoints(Point, Point);
|
||||||
|
float angle(Point, Point, Point);
|
||||||
|
vector<Details> masks;
|
||||||
|
int nr_parts;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
#endif //SOURCE_ABSTRACTIONLAYER_1_H
|
#endif //SOURCE_ABSTRACTIONLAYER_1_H
|
||||||
|
BIN
Source/pieces/0.png
Normal file
After Width: | Height: | Size: 42 KiB |
BIN
Source/pieces/1.png
Normal file
After Width: | Height: | Size: 60 KiB |
BIN
Source/pieces/10.png
Normal file
After Width: | Height: | Size: 57 KiB |
BIN
Source/pieces/100.png
Normal file
After Width: | Height: | Size: 60 KiB |
BIN
Source/pieces/1000.png
Normal file
After Width: | Height: | Size: 62 KiB |
BIN
Source/pieces/1001.png
Normal file
After Width: | Height: | Size: 61 KiB |
BIN
Source/pieces/1002.png
Normal file
After Width: | Height: | Size: 46 KiB |
BIN
Source/pieces/1003.png
Normal file
After Width: | Height: | Size: 66 KiB |
BIN
Source/pieces/1004.png
Normal file
After Width: | Height: | Size: 39 KiB |
BIN
Source/pieces/1005.png
Normal file
After Width: | Height: | Size: 44 KiB |
BIN
Source/pieces/1006.png
Normal file
After Width: | Height: | Size: 44 KiB |
BIN
Source/pieces/1007.png
Normal file
After Width: | Height: | Size: 43 KiB |
BIN
Source/pieces/101.png
Normal file
After Width: | Height: | Size: 50 KiB |
BIN
Source/pieces/102.png
Normal file
After Width: | Height: | Size: 49 KiB |
BIN
Source/pieces/103.png
Normal file
After Width: | Height: | Size: 61 KiB |
BIN
Source/pieces/104.png
Normal file
After Width: | Height: | Size: 60 KiB |
BIN
Source/pieces/105.png
Normal file
After Width: | Height: | Size: 53 KiB |
BIN
Source/pieces/106.png
Normal file
After Width: | Height: | Size: 65 KiB |
BIN
Source/pieces/107.png
Normal file
After Width: | Height: | Size: 58 KiB |
BIN
Source/pieces/108.png
Normal file
After Width: | Height: | Size: 80 KiB |
BIN
Source/pieces/109.png
Normal file
After Width: | Height: | Size: 56 KiB |
BIN
Source/pieces/11.png
Normal file
After Width: | Height: | Size: 55 KiB |
BIN
Source/pieces/110.png
Normal file
After Width: | Height: | Size: 62 KiB |
BIN
Source/pieces/111.png
Normal file
After Width: | Height: | Size: 49 KiB |
BIN
Source/pieces/112.png
Normal file
After Width: | Height: | Size: 56 KiB |
BIN
Source/pieces/113.png
Normal file
After Width: | Height: | Size: 52 KiB |
BIN
Source/pieces/114.png
Normal file
After Width: | Height: | Size: 41 KiB |
BIN
Source/pieces/115.png
Normal file
After Width: | Height: | Size: 71 KiB |
BIN
Source/pieces/116.png
Normal file
After Width: | Height: | Size: 64 KiB |
BIN
Source/pieces/117.png
Normal file
After Width: | Height: | Size: 56 KiB |
BIN
Source/pieces/118.png
Normal file
After Width: | Height: | Size: 49 KiB |
BIN
Source/pieces/119.png
Normal file
After Width: | Height: | Size: 57 KiB |
BIN
Source/pieces/12.png
Normal file
After Width: | Height: | Size: 56 KiB |
BIN
Source/pieces/120.png
Normal file
After Width: | Height: | Size: 55 KiB |
BIN
Source/pieces/121.png
Normal file
After Width: | Height: | Size: 63 KiB |
BIN
Source/pieces/122.png
Normal file
After Width: | Height: | Size: 62 KiB |
BIN
Source/pieces/123.png
Normal file
After Width: | Height: | Size: 60 KiB |
BIN
Source/pieces/124.png
Normal file
After Width: | Height: | Size: 46 KiB |
BIN
Source/pieces/125.png
Normal file
After Width: | Height: | Size: 57 KiB |
BIN
Source/pieces/126.png
Normal file
After Width: | Height: | Size: 59 KiB |
BIN
Source/pieces/127.png
Normal file
After Width: | Height: | Size: 72 KiB |
BIN
Source/pieces/128.png
Normal file
After Width: | Height: | Size: 47 KiB |
BIN
Source/pieces/129.png
Normal file
After Width: | Height: | Size: 50 KiB |
BIN
Source/pieces/13.png
Normal file
After Width: | Height: | Size: 64 KiB |
BIN
Source/pieces/130.png
Normal file
After Width: | Height: | Size: 51 KiB |
BIN
Source/pieces/131.png
Normal file
After Width: | Height: | Size: 58 KiB |
BIN
Source/pieces/132.png
Normal file
After Width: | Height: | Size: 46 KiB |
BIN
Source/pieces/133.png
Normal file
After Width: | Height: | Size: 45 KiB |
BIN
Source/pieces/134.png
Normal file
After Width: | Height: | Size: 57 KiB |
BIN
Source/pieces/135.png
Normal file
After Width: | Height: | Size: 61 KiB |
BIN
Source/pieces/136.png
Normal file
After Width: | Height: | Size: 60 KiB |
BIN
Source/pieces/137.png
Normal file
After Width: | Height: | Size: 54 KiB |
BIN
Source/pieces/138.png
Normal file
After Width: | Height: | Size: 51 KiB |
BIN
Source/pieces/139.png
Normal file
After Width: | Height: | Size: 72 KiB |
BIN
Source/pieces/14.png
Normal file
After Width: | Height: | Size: 65 KiB |
BIN
Source/pieces/140.png
Normal file
After Width: | Height: | Size: 60 KiB |
BIN
Source/pieces/141.png
Normal file
After Width: | Height: | Size: 70 KiB |
BIN
Source/pieces/142.png
Normal file
After Width: | Height: | Size: 47 KiB |
BIN
Source/pieces/143.png
Normal file
After Width: | Height: | Size: 57 KiB |
BIN
Source/pieces/144.png
Normal file
After Width: | Height: | Size: 60 KiB |
BIN
Source/pieces/145.png
Normal file
After Width: | Height: | Size: 40 KiB |
BIN
Source/pieces/146.png
Normal file
After Width: | Height: | Size: 45 KiB |
BIN
Source/pieces/147.png
Normal file
After Width: | Height: | Size: 60 KiB |
BIN
Source/pieces/148.png
Normal file
After Width: | Height: | Size: 60 KiB |
BIN
Source/pieces/149.png
Normal file
After Width: | Height: | Size: 62 KiB |
BIN
Source/pieces/15.png
Normal file
After Width: | Height: | Size: 42 KiB |
BIN
Source/pieces/150.png
Normal file
After Width: | Height: | Size: 53 KiB |
BIN
Source/pieces/151.png
Normal file
After Width: | Height: | Size: 63 KiB |
BIN
Source/pieces/152.png
Normal file
After Width: | Height: | Size: 56 KiB |
BIN
Source/pieces/153.png
Normal file
After Width: | Height: | Size: 65 KiB |
BIN
Source/pieces/154.png
Normal file
After Width: | Height: | Size: 72 KiB |
BIN
Source/pieces/155.png
Normal file
After Width: | Height: | Size: 66 KiB |
BIN
Source/pieces/156.png
Normal file
After Width: | Height: | Size: 46 KiB |
BIN
Source/pieces/157.png
Normal file
After Width: | Height: | Size: 46 KiB |
BIN
Source/pieces/158.png
Normal file
After Width: | Height: | Size: 60 KiB |
BIN
Source/pieces/159.png
Normal file
After Width: | Height: | Size: 58 KiB |
BIN
Source/pieces/16.png
Normal file
After Width: | Height: | Size: 62 KiB |
BIN
Source/pieces/160.png
Normal file
After Width: | Height: | Size: 50 KiB |
BIN
Source/pieces/161.png
Normal file
After Width: | Height: | Size: 58 KiB |
BIN
Source/pieces/162.png
Normal file
After Width: | Height: | Size: 70 KiB |
BIN
Source/pieces/163.png
Normal file
After Width: | Height: | Size: 46 KiB |
BIN
Source/pieces/164.png
Normal file
After Width: | Height: | Size: 58 KiB |
BIN
Source/pieces/165.png
Normal file
After Width: | Height: | Size: 51 KiB |
BIN
Source/pieces/166.png
Normal file
After Width: | Height: | Size: 39 KiB |
BIN
Source/pieces/167.png
Normal file
After Width: | Height: | Size: 43 KiB |
BIN
Source/pieces/168.png
Normal file
After Width: | Height: | Size: 62 KiB |
BIN
Source/pieces/169.png
Normal file
After Width: | Height: | Size: 68 KiB |
BIN
Source/pieces/17.png
Normal file
After Width: | Height: | Size: 59 KiB |
BIN
Source/pieces/170.png
Normal file
After Width: | Height: | Size: 70 KiB |
BIN
Source/pieces/171.png
Normal file
After Width: | Height: | Size: 57 KiB |
BIN
Source/pieces/172.png
Normal file
After Width: | Height: | Size: 56 KiB |
BIN
Source/pieces/173.png
Normal file
After Width: | Height: | Size: 44 KiB |
BIN
Source/pieces/174.png
Normal file
After Width: | Height: | Size: 68 KiB |
BIN
Source/pieces/175.png
Normal file
After Width: | Height: | Size: 59 KiB |
BIN
Source/pieces/176.png
Normal file
After Width: | Height: | Size: 48 KiB |
BIN
Source/pieces/177.png
Normal file
After Width: | Height: | Size: 55 KiB |
BIN
Source/pieces/178.png
Normal file
After Width: | Height: | Size: 43 KiB |
BIN
Source/pieces/179.png
Normal file
After Width: | Height: | Size: 43 KiB |