#include<iostream>
#include<fstream>
#include<opencv2/opencv.hpp>
#include<opencv2/highgui.hpp>
#include<opencv2/core.hpp>
#include<io.h>                          //api和結(jié)構(gòu)體
#include<string.h>
#include<string>
using namespace std;
using namespace cv;
void ergodicTest(string filename, string name);    //遍歷函數(shù)
string Image_Compression(string imgpath);          //壓縮圖片并返回字符串
int distance(string str1, string str2);            //對(duì)比函數(shù)不一樣的位數(shù)
void compare();
int turn(char a);
void main()
{
	const char* filepath = "E:\\learn\\vsfile\\c++project\\pictureData\\train-images";     
	ergodicTest(filepath,"train_num.txt");         //處理訓(xùn)練集
	const char* test_path= "E:\\learn\\vsfile\\c++project\\pictureData\\test-images";
	ergodicTest(test_path, "test_num.txt");
	compare();   
}
 
void ergodicTest(string filename,string name)       //遍歷并把路徑存到files
{
	string firstfilename = filename + "\\*.bmp";
	struct _finddata_t fileinfo;
	intptr_t handle;            //不能用long，因?yàn)榫葐栴}會(huì)導(dǎo)致訪問沖突,longlong也可
	string rout = "E:\\learn\\vsfile\\c++project\\pictureData\\" + name;
	ofstream file;
	file.open(rout, ios::out);
	handle = _findfirst(firstfilename.c_str(), &fileinfo);
	if ( _findfirst(firstfilename.c_str(), &fileinfo) != -1)
	{
		do
		{
			file << fileinfo.name << ":" << Image_Compression(filename + "\\" + fileinfo.name) << endl;
		} while (!_findnext(handle, &fileinfo));
		file.close();
		_findclose(handle);
	}
}
string Image_Compression(string imgpath)   //輸入圖片地址返回圖片二值像素字符
{
	Mat Image = imread(imgpath);               //輸入的圖片
	cvtColor(Image, Image, COLOR_BGR2GRAY);
	int Matrix[28][28];                        //將digitization轉(zhuǎn)化為字符串類型
	for (int row = 0; row < Image.rows; row++)  //把圖片的像素點(diǎn)傳給數(shù)組
		for (int col = 0; col < Image.cols; col++)
		{
			Matrix[row][col] = Image.at<uchar>(row, col);
		}
	string img_str = "";                   //用來(lái)存儲(chǔ)結(jié)果字符串
	int x = 0, y = 0;
	for (int k = 1; k < 50; k++)
	{
		int total = 0;
		for (int q = 0; q < 4; q++)
			for (int p = 0; p < 4; p++)
				if (Matrix[x + q][y + p] > 127) total += 1;
		y = (y + 4) % 28;
		if (total >= 6) img_str += '1';    //將28*28的圖片轉(zhuǎn)化為7*7即壓縮
		else img_str += '0';
		if (k % 7 == 0)
		{
			x += 4;
			y = 0;
		}
	}
	return img_str;
}
 
int distance(string str1, string str2)  //比對(duì)兩個(gè)字符串有多少個(gè)不一樣
{
	int counts=0;
	for (int i = 0; i < str1.length(); i++)
	{
		if (str1[i] == str2[i]) continue;
		else counts++; 
	}
	return counts;
}
int turn(char a)
{
	stringstream str;
	int f = 1;
	str << a;
	str >> f;
	str.clear();
	return f;
}
 
void compare()
{
	ifstream train_data;//建立讀文件流
	ifstream test_data;
	string tmp1 = "";         //從train中取數(shù)據(jù)存在tmp1
	string tmp11 = "";
	string tmp2 = "";         //從test中取
	string tmp22 = "";
	bool refused = false; //拒絕識(shí)別標(biāo)志
	int tr_num = 0;       //用來(lái)存儲(chǔ)最小值的文件名(訓(xùn)練集)
	int num_refused = 0;   //拒絕識(shí)別個(gè)數(shù)
	int num_false = 0;     //識(shí)別錯(cuò)誤個(gè)數(shù)
	int num_true = 0;      //正確識(shí)別個(gè)數(shù)
	test_data.open("E:\\learn\\vsfile\\c++project\\pictureData\\test_num.txt");
	for (int p = 0; p < 200; p++)
	{
		int total = 49;      //方便比大小，設(shè)置初值為49
		getline(test_data, tmp2);
		tmp22 = tmp2;    //在切割字符串之前保留，以便后面知曉該字符串是哪個(gè)數(shù)字的
		if(tmp2.length()==57) tmp2.erase(0,8); //erase函數(shù)是用來(lái)切割字符串的，這里是切割第0位的后面8位，存剩余的其他位
		else tmp2.erase(0,9);
		train_data.open("E:\\learn\\vsfile\\c++project\\pictureData\\train_num.txt");
		for (int j = 0; j < 1000; j++)         //一個(gè)測(cè)試樣本和所有訓(xùn)練樣本對(duì)比
		{
			getline(train_data, tmp1);
			tmp11 = tmp1;
			if (tmp1.length() == 57) tmp1.erase(0, 8);
			else tmp1.erase(0, 9);
			if (distance(tmp1, tmp2) < total)  //取最相近的
			{
				refused = false;   //拒絕識(shí)別被設(shè)置為否，即識(shí)別沒有被拒絕
				total = distance(tmp1, tmp2);
				tr_num = turn(tmp11[0]);          //記錄數(shù)字
			}
			else if(distance(tmp1, tmp2) == total && tr_num!= turn(tmp11[0]))  //發(fā)現(xiàn)相同相似度，且兩者歸屬的數(shù)字不同
			{
				refused = true;   //拒絕識(shí)別
				continue;          //循環(huán)繼續(xù)
			}	
		}
		train_data.close();
 
		if (!refused)
		{
			if (tr_num == turn(tmp22[0]))
			{
				//cout << tmp2[0] << endl;
				num_true++;
				cout << "識(shí)別為：" << tr_num << endl;
			}
			else
			{
				num_false++;
				cout << "識(shí)別錯(cuò)誤！" << endl;
			}
		}
		else
		{
			num_refused++;
			cout << "拒絕識(shí)別！" << endl;
		}
	}
	test_data.close();
	double t = num_true / 200.0, f = num_false / 200.0, r = num_refused / 200.0;
	cout << "正確率為：" << t << endl;
	cout << "錯(cuò)誤率為：" << f << endl;
	cout << "拒絕識(shí)別率為：" << r << endl;
}