package train;

import java.awt.Label;
import java.awt.image.BufferedImage;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.FilenameFilter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;

import javax.imageio.ImageIO;

import svmHelper.svm_scale;
import svmHelper.svm_train;

public class Train {
	
	/**
	 * 类标号map，如：a=>1 b=>2
	 */
	private HashMap<String, Integer> labelMap = null;
	
	/**
	 * 所有图像分类的map，key为当前类标号， value为对应的图片，图片以二维数组的形式保存
	 */
	private HashMap<String, ArrayList<Integer[][]>> imageMap = null;
	
	public Train(){
		init();
	}
	
	private void init(){
		labelMap = new HashMap<String, Integer>();
		imageMap = new HashMap<String, ArrayList<Integer[][]>>();
		loadImageLabel();
		loadImage();
		svmFormat();
	}
	
	
	private void loadImageLabel(){
		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(arr[0], Integer.parseInt(arr[1]));
			 }
			 
			 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 void loadImage(){
		File dir = new File("4_scale/");
		//只列出jpg
		File[] files = dir.listFiles(new FilenameFilter() {
			
			public boolean isJpg(String file){   
			    if (file.toLowerCase().endsWith(".jpg")){   
			      return true;   
			    }else{   
			      return false;   
			    }   
			}
			
			@Override
			public boolean accept(File dir, String name) {
				// TODO Auto-generated method stub
				return isJpg(name);
			}
		});
		
		for (File file : files) {
			try {
				transferToMap(file);
				
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
		
		System.out.println("load mage end");
		
	}
	
	/**
	 * 将image 转换到 map中
	 * @param file
	 * @throws IOException
	 */
	private void transferToMap(File file) throws IOException{
		BufferedImage image = ImageIO.read(file);
		int width = image.getWidth();
		int height = image.getHeight();
		Integer[][] imgArr = new Integer[height][width];
		
		for (int y = 0; y < height; y++) {
			for (int x = 0; x < width; x++) {
				//黑色点标记为1
				int value = ImageUtil.isBlack(image.getRGB(x, y)) ? 1 : 0;
				imgArr[y][x] = value;
			}
		}
		
		ArrayList<Integer[][]> imgList = null;
		String className = ImageUtil.getImgClass(file.getName());
		if (imageMap.containsKey(className)) {
			imgList = imageMap.get(className);
			imgList.add(imgArr);
		}else{
			imgList = new ArrayList<Integer[][]>();
			imgList.add(imgArr);
			imageMap.put(className, imgList);
		}
	}
	
	
	/**
	 * 转换成svm预料的格式
	 */
	public void svmFormat(){
		PrintWriter writer = null;
		try {
			writer = new PrintWriter(new File("svm/svm.train"));
			Iterator<String> iterator = this.imageMap.keySet().iterator();
			while(iterator.hasNext()){
				String className = iterator.next();
				int classLabel = labelMap.get(className);
				ArrayList<Integer[][]> list = imageMap.get(className);
				System.out.println(className);
				for (Integer[][] img : list) {
					String tmpLine = classLabel + " ";
					int index = 1;
					for (int i = 0; i < img.length; i++) {
						for (int j = 0; j < img[i].length; j++) {
							tmpLine += index + ":" + img[i][j] + " ";
							index ++;
						}
					}
//					System.out.println(tmpLine);
					writer.write(tmpLine + "\r\n");
					writer.flush();
				}
			}
		} catch (Exception e) {
			e.printStackTrace();
		} finally{
			if (writer != null) {
				writer.close();
			}
		}
	}
	
	
	/**
	 * 生成模型
	 * @throws IOException
	 */
	public static void run() throws IOException{
		//train参数
		String[] arg = {"-t","0","svm/svm.train","svm/svm.model"};
		
		//predict参数
		String[] parg = {"svm/svmscale.test","svm/svm.model","svm/result.txt"};
		
		System.out.println("训练开始");
		svm_train.main(arg);
		System.out.println("训练结束");
	}
	
	private static void produceLabel(){
		FileWriter writer = null;
		try {
			
			writer = new FileWriter(new File("svm/label.txt"));
			String charactor = "1234567890abcdefghijklmnopqrstuvwxyz";
			 
			for (int i = 0; i < charactor.length(); i++) {
				char c = charactor.charAt(i);
				String str = c + " " + (i+1) + "\r\n";
				writer.write(str);
			}
			writer.flush();
			
		} catch (IOException e) {
			e.printStackTrace();
		} finally{
			if (writer != null) {
				try {
					writer.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
		
	}
	
	public static void main(String[] args) throws IOException{
		System.out.println("begin");
//		Train model = new Train();
		
		run();
		System.out.println("end");
	}

	
	
	
	
	
	
	
	
	
	
	
	
	
	
	
	
}