package RobustHMM;

import java.io.FileNotFoundException;
import java.io.FileReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Scanner;

import be.ac.ulg.montefiore.run.jahmm.Observation;
import be.ac.ulg.montefiore.run.jahmm.ObservationInteger;
import be.ac.ulg.montefiore.run.jahmm.ObservationReal;
import be.ac.ulg.montefiore.run.jahmm.ObservationVector;

public class TrainingReader {

	private String input;
	private List<List<?>> obs;
	private String method;
	
	public TrainingReader(String i) throws FileNotFoundException{
		input = i;
		read();
	}

	public List<List<?>> getObs(){
		return obs;
	}
	public String getMethod(){return method;}
	
	private void read() throws FileNotFoundException{
		Scanner inFile = new Scanner((Readable) new FileReader(input));
		obs = new ArrayList<List<?>>();
		while(inFile.hasNext()){
			String line = inFile.nextLine();
			//if (line.startsWith(">")){
				if (isInt(line)){
					HashMap<Integer,ArrayList<ObservationInteger>> map = readInt(inFile);
					for (int index : map.keySet()){
						List<?> tempList = map.get(index);
						obs.add(tempList);
					}
				}
				else if(isReal(line)){
					HashMap<Integer,ArrayList<ObservationReal>> map = readReal(inFile);
					for (int index : map.keySet()){
						List<?> tempList = map.get(index);
						obs.add(tempList);
					}
				}
				else if (isVector(line)){
					HashMap<Integer,ArrayList<ObservationVector>> map = readVector(inFile);
					for (int index : map.keySet()){
						List<?> tempList = map.get(index);
						obs.add(tempList);
					}
				}
			//}
			
		}
	}
	private HashMap<Integer,ArrayList<ObservationVector>> readVector(Scanner inFile){
		HashMap<Integer,ArrayList<ObservationVector>> map = new HashMap<Integer,ArrayList<ObservationVector>>();
		while (inFile.hasNext()){
			String line = inFile.nextLine();
			String[] temp = line.split(",");
			int index = Integer.parseInt(temp[0]);
			double[] tempvalue = new double[temp.length-1];
			for (int i = 1; i < temp.length;i++){
				tempvalue[i-1] = Double.parseDouble(temp[i]);
			}
			ObservationVector value = new ObservationVector(tempvalue);
			if (map.containsKey(index)){
				ArrayList<ObservationVector> list = map.get(index);
				list.add(value);
				map.put(index, list);
			}
			else if (!map.containsKey(index)){
				ArrayList<ObservationVector> list = new ArrayList<ObservationVector>();
				list.add(value);
				map.put(index,list);
			}
		}
		
		return map;
	}
	private HashMap<Integer,ArrayList<ObservationReal>> readReal(Scanner inFile){
		HashMap<Integer,ArrayList<ObservationReal>> map = new HashMap<Integer,ArrayList<ObservationReal>>();
		while (inFile.hasNext()){
			String line = inFile.nextLine();
			String[] temp = line.split(",");
			int index = Integer.parseInt(temp[0]);
			double tempvalue = Double.parseDouble(temp[1]);
			ObservationReal value = new ObservationReal(tempvalue);
			if (map.containsKey(index)){
				ArrayList<ObservationReal> list = map.get(index);
				list.add(value);
				map.put(index, list);
			}
			else if (!map.containsKey(index)){
				ArrayList<ObservationReal> list = new ArrayList<ObservationReal>();
				list.add(value);
				map.put(index,list);
			}
		}
		
		return map;
	}
	private HashMap<Integer,ArrayList<ObservationInteger>> readInt(Scanner inFile){
		
		HashMap<Integer,ArrayList<ObservationInteger>> map = new HashMap<Integer,ArrayList<ObservationInteger>>();
		while (inFile.hasNext()){
			String line = inFile.nextLine();
			String[] temp = line.split(",");
			int index = Integer.parseInt(temp[0]);
			int tempvalue = Integer.parseInt(temp[1]);
			ObservationInteger value = new ObservationInteger(tempvalue);
			if (map.containsKey(index)){
				ArrayList<ObservationInteger> list = map.get(index);
				list.add(value);
				map.put(index, list);
			}
			else if (!map.containsKey(index)){
				ArrayList<ObservationInteger> list = new ArrayList<ObservationInteger>();
				list.add(value);
				map.put(index,list);
			}
		}
		return map;
		
	}
	
	private boolean isInt(String line){
		if (line.contains("Integer") || line.contains("integer") || line.contains("int") || line.contains("Int")){
			method = "Integer";
			return true;
		}
		else{return false;}
	}
	private boolean isReal(String line){
		if (line.contains("Real") || line.contains("real")){
			method = "Real";
			return true;
		}
		else{return false;}
	}
	private boolean isVector(String line){
		if (line.contains("Vector") || line.contains("vector")){
			method = "Vector";
			return true;
		}
		else{return false;}
	}
}