/* 	程序名：singlefeature.c
功能：分总程序：读入图像文件，得出单个文件的特征值
*/
#pragma once
#include <stdlib.h>
#include <stdio.h>
#include <math.h>
#include <cv.h>
#include <highgui.h>
#include "Point.h"
#include "Cword.h"
//#include "FreeImage.h"     //用于读gif的图像
#include<io.h>             //下面的5个用于读取文件夹下的所有文件名
#include<vector>
#include<iostream>
using namespace std;
#include <string.h>
#include <direct.h>
#include"Thinner.h"

/*各种声明*/

void		getFiles(string path, vector<string>& files );//读取文件名下所有文件                             
char*		getType(char fileName[], char type[]); //获取图像类型       
int*		binary(IplImage*  img,int bithro);  //二值化图像      
int			outlinefeature(IplImage* imglk,int feature[ ][50]);//计算图像的轮廓特征值                         
IplImage*	Cjbsb(IplImage* img,IplImage* imgjbsb,int jbwhite,int jbblack);//图像角标识别
IplImage*	gif2ipl(const char* filename);     //读取gif的外部函数

IplImage* singlefeature(char* path,int feature[ ][50]){
	//定义变量
  	IplImage* img = 0;          //原图
    IplImage* imglk = 0;        //轮廓图
	IplImage* imggj = 0;        //骨架图
	IplImage* imgjbsb = 0;      //角标识别图
	IplImage* imgbj = 0;        //只提取笔记部分的图像
	IplImage* imgbjhf = 0;      //为文字区域画上方格
    IplImage* imgwzbj = 0;      //为文字区域标出是否为文字（文字标记）

	char imgtype[10];           //判断图像类型
  	int height,width,step,channels;
  	uchar *data;

    int i,j;
	int *black;                 //用于返回图像每行黑像素的个数
	//int feature[50][50]={0};    //特征值初始化
	
	img=cvLoadImage(path,0);

  	/* 获取图像信息*/
  	height    = img->height;  
  	width     = img->width;	
  	step      = img->widthStep;	
  	channels  = img->nChannels;
  	data      = (uchar *)img->imageData;

/*开始处理*/

    /*图像放大*/
    IplImage* imgbig = 0;          //原图的放大图
    CvSize dst_cvsize;             //目标图像的大小
	float scale=1;
	if(width<840){
        scale=(float)840/width;
		dst_cvsize.width=880;
		dst_cvsize.height=(int)(height*scale);
	}
	else
	{
		dst_cvsize.width=width;
		dst_cvsize.height=height;
	}

	imgbig=cvCreateImage(dst_cvsize,img->depth,img->nChannels);
    cvResize(img,imgbig,CV_INTER_LINEAR);	// CV_INTER_NN - 最近邻插值,
											//CV_INTER_LINEAR - 双线性插值 (缺省使用),
											//CV_INTER_AREA - 使用象素关系重采样。当图像缩小时候，该方法可以避免波纹出现。
											//CV_INTER_CUBIC - 立方插值.

	/*二值化*/
	int bithro=230;                              //输入二值化的阈值   (0--255)
    black=binary(imgbig,bithro);                    //二值化，并统计黑像素的个数，返回每行黑像素的个数(black)
    //cvNamedWindow("二值化结果图",CV_WINDOW_AUTOSIZE);                   //显示图像
    //cvShowImage("二值化结果图",img); 
	//printf("二值化求解完成！！\n");

	
	
    /*角标识别*/
	int jbwhite=5,jbblack=4;
	imgjbsb = cvCreateImage(cvGetSize(imgbig),imgbig->depth,imgbig->nChannels);
    imgbj=Cjbsb(imgbig,imgjbsb,jbwhite,jbblack);                              //返回文字的笔迹部分
 

	/*计算骨架图*/
    imggj = cvCreateImage(cvGetSize(imgbj),imgbj->depth,imgbj->nChannels);   //复制
	cvCopy(imgbj,imggj,NULL);
	uchar *gjdata= (uchar *)imggj->imageData;
	
	beforethin(gjdata,gjdata,imggj->width, imggj->height);

	for(j=0;j<imggj->height;j++)
	{ //取值范围转到0--1  
	      
		for(i=0;i<imggj->width;i++)
		{
             if(gjdata[j*imggj->widthStep+i]==255)
				 gjdata[j*imggj->widthStep+i]=1;
		}
	}
	ThinnerRosenfeld(imggj->imageData, imggj->height, imggj->width);

	for(j=0;j<imggj->height;j++)
	{//取值范围转到0--255,反转过来
		for(i=0;i<imggj->width;i++)
		{
		     if(gjdata[j*imggj->widthStep+i]==1)
				 gjdata[j*imggj->widthStep+i]=0;
			 else
				 gjdata[j*imggj->widthStep+i]=255;
		}
	}

	//保存图像 应先生成图像文件名
	/*
	char processPic[100]="E:/imggj/";
	char *namePic=new char[20];
	bool flag=false;
	string xuehao=path,kaoshihao=path;
	int num_iter=sizeof(path);
	for(int iter=0;iter<num_iter;iter++)
	{
		if(path[iter]=='x')
		{
			flag=true;
			break;
		}
	}
	if(flag)
	{
		xuehao=xuehao.substr(27,13);
		kaoshihao=kaoshihao.substr(40,5);
	}else
	{
		xuehao=xuehao.substr(27,12);
		kaoshihao=kaoshihao.substr(39,5);
	}
	strcat(processPic,xuehao.c_str());
	_mkdir(processPic);  
	strcat(processPic,kaoshihao.c_str()); 
	strcat(processPic,".jpg");
	cvSaveImage(processPic,imggj);
*/
	/*计算骨架特征徝*/
    outlinefeature(imggj,feature);   //特征值占48*48的右上三角形，feature调用返回
    //cvWaitKey(0);

	/*释放内存*/
	cvReleaseImage(&imgbig);
  	cvReleaseImage(&img );  
	cvReleaseImage(&imgbj ); 
	cvReleaseImage(&imglk );  
	cvReleaseImage(&imgjbsb );  
	cvReleaseImage(&imgbjhf );  
	cvReleaseImage(&imgwzbj );
    cvDestroyAllWindows();  

  	return imggj;
}