package train;

import java.awt.Image;

import java.awt.image.BufferedImage;

import java.io.BufferedReader;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileReader;

import java.io.IOException;

import java.util.ArrayList;

import java.util.HashMap;

import javax.imageio.ImageIO;

import Base.Base;

public class Identy{

private HashMap<Integer, String> labelMap = null;

public Identy(){

loadLabelMap();

}

private void loadLabelMap(){

labelMap = new HashMap<Integer, String>();

BufferedReader reader = null;

try {

reader = new BufferedReader(new FileReader(new File("svm/label.txt")));

String buff = null;

while((buff = reader.readLine()) != null){

String[] arr = buff.split(" ");

labelMap.put(Integer.parseInt(arr[1]), arr[0]);

}

System.out.println("load image label finish!");

} catch (FileNotFoundException e) {

e.printStackTrace();

} catch (IOException e) {

e.printStackTrace();

} finally{

if (reader != null) {

try {

reader.close();

} catch (IOException e) {

e.printStackTrace();

}

}

}

}

private String getClassName(int label){

return labelMap.get(label);

}

/**

* 具体的预测，返回识别的文字

* @param file

* @return

* @throws IOException

*/

public String predict(File file) throws IOException{

BufferedImage sourceImage = ImageIO.read(file);

Preprocess preprocess = new Preprocess();

BufferedImage binaryImage = preprocess.getBinaryImage(sourceImage);

SegCfg segCfg = new SegCfg();

ArrayList<BufferedImage> interList = segCfg.cfs(binaryImage);

ArrayList<BufferedImage> imageList = new ArrayList<BufferedImage>();

SegWaterDrop segWaterDrop = new SegWaterDrop();

for (BufferedImage img : interList) {

ArrayList<BufferedImage> tmpList = segWaterDrop.drop(img);

for (BufferedImage sumImg : tmpList) {

imageList.add(ImageUtil.scaleImage(sumImg));

}

}

for (int i = 0; i < imageList.size(); i++) {

ImageIO.write(imageList.get(i), "JPG", new File("tmp/" + i + ".jpg"));

}

Predict.run(imageList);

String result = "";

BufferedReader reader = null;

try {

reader = new BufferedReader(new FileReader(new File("svm/result.txt")));

String buff = "";

while((buff = reader.readLine()) != null){

int label = (int)Double.parseDouble(buff);

String className = getClassName(label);

result += className + " ";

}

System.out.println(result);

} catch (Exception e) {

e.printStackTrace();

} finally{

if (reader != null) {

reader.close();

}

}

return result;

}

public static void main(String[] args) throws IOException{

Identy index = new Identy();

// index.predict(new File("download/1.jpg"));

index.predict(new File("tmp/test.jpg"));

}

}