#include <opencv2/core.hpp>
#include <opencv2/imgcodecs.hpp>
#include "opencv2/imgproc.hpp"
#include <opencv2/highgui.hpp>
#include <iostream>

using namespace cv;
using namespace std;

#include <map>

class RectComp
{  
public:
	Rect rm;  
	RectComp(Rect rms)  
	{  
		rm = rms;  
	}  
	bool operator < (const RectComp& ti) const  
	{  
		return rm.x< ti.rm.x;  
	}  
};

int main()  
{  
	//裝載圖片  
	Mat srcImage1= imread("D:\\opencv_image\\images\\20150820103614491.png");   
	namedWindow("hello-1", 1);  
	imshow("hello-1", srcImage1 ); 
	Mat srcImage2,srcImage3,srcImage4,srcImage5;  

	//圖片變成灰度圖片  
	cvtColor(srcImage1,srcImage2,CV_BGR2GRAY);  
	//圖片二值化  
	threshold(srcImage2,srcImage3,200,255,THRESH_BINARY_INV);  

	//確定腐蝕和膨脹核的大小  
	Mat element = getStructuringElement(MORPH_RECT, Size(4, 4));  
	//腐蝕操作  
	erode(srcImage3,srcImage4,element);  
	//膨脹操作  
	dilate(srcImage4,srcImage5,element);  

	namedWindow("hello-5", 1);  
	imshow("hello-5", srcImage5 );  

	//確定每張答題卡的ROI區域  
	Mat imag_ch1 = srcImage5(Rect(0,30,100,60));  
	namedWindow("img1", 1);  
	imshow("img1",imag_ch1);  

	//提取已經塗好了的選項  
	std::vector<std::vector<cv::Point> > chapter1;  
	findContours(imag_ch1,chapter1,RETR_EXTERNAL,CHAIN_APPROX_SIMPLE);  
	Mat result(imag_ch1.size(), CV_8U , cv::Scalar(255)) ;  
	cv::drawContours(result,chapter1,-1,cv::Scalar(0),2);  
	namedWindow("resultImage", 1);  
	cv::imshow("resultImage" , result);  

	vector<RectComp>RectCompList;  
	for(int i = 0;i<chapter1.size();i++)  
	{  
		Rect rm= cv::boundingRect(cv::Mat(chapter1[i]));  
		RectComp *ti = new RectComp(rm);  
		RectCompList.push_back(*ti);  
		//         printf("Rect %d x = %d,y = %d \n",i,rm.x,rm.y);  
	}  
	sort(RectCompList.begin(),RectCompList.end());  
	map<int,string>listenAnswer;  
	//判斷這部分的答題卡是否都已塗上  
	for(int t = 0;t<RectCompList.size();t++)  
	{  
		if(RectCompList.at(t).rm.y<=22)  
		{  
			listenAnswer[t] = "A";  
		}  
		else if((RectCompList.at(t).rm.y>22)&&(RectCompList.at(t).rm.y<=34))  
		{  
			listenAnswer[t] = "B";  
		}  
		else if(RectCompList.at(t).rm.y>34)  
		{  
			listenAnswer[t] = "C";  
		}  
		printf("sorted %d x = %d,y = %d \n",t,RectCompList.at(t).rm.x,RectCompList.at(t).rm.y);  
	}  
	for(map<int,string>::iterator it = listenAnswer.begin();it!=listenAnswer.end();++it)  
	{  
		cout<<"num:"<<it->first+1<<","<<"answer:"<<it->second<<endl;  
	}  

	waitKey(0);  
	return 0;  
}