package uk.ac.leeds.sog.moses.agent;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.IOException;
import java.io.PrintStream;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.StringTokenizer;

import org.apache.log4j.Logger;

/**
 * @author BELINDA
 *This class builds up the model and outputs the model description
 */
public class ModelBuilder {
	
	public static final int NUMBER_OF_DATA_FIELDS = 18;
	
	private Model i_model;
	private Map i_modelMap;
	private List i_invalidData;
	private int i_numOfValidaData;
	private int i_numberOfHRPs;
	private int i_numberOfSpouses;
	private int i_numberOfChildren;
	private int i_numberOfElderlyDependents;
	private int i_numberOfAdultDependents;
	private List i_areas;
	
	private static Logger s_logger = Logger.getLogger(ModelBuilder.class);
	private static String s_lineSeparator = System.getProperty("line.separator");
	
	//the constructor
	public ModelBuilder() {
		i_model = new Model();
		i_modelMap = new HashMap();
		i_invalidData = new ArrayList();
		i_numOfValidaData = 0;
		i_numberOfHRPs = 0;
		i_numberOfSpouses = 0;
		i_numberOfChildren = 0;
		i_numberOfElderlyDependents = 0;
		i_numberOfAdultDependents = 0;
		i_areas = new ArrayList();
	}
	
	//method to get the model
	public Model getModel() {
		return i_model;
	}
	
	//method to build up the model
	public Model buildModel(String filename) {
		try {
			BufferedReader in = new BufferedReader(new FileReader(filename));
			
			String line = in.readLine();
			while(line != null) {
				
				// read file and create hrp/child/elderlyDependent/spouse/AdultDependent lists
				StringTokenizer st = new StringTokenizer(line, ",");
				// HRP number of tokens: NUMBER_OF_DATA_FIELDS; non-HRP number of tokens: NUMBER_OF_DATA_FIELDS - 1
				int numberOfTokens = st.countTokens();
				if(numberOfTokens == NUMBER_OF_DATA_FIELDS || numberOfTokens == (NUMBER_OF_DATA_FIELDS - 1)) {
					String pid = st.nextToken();
					if(pid.equalsIgnoreCase("PID")) {
						// not a data line in the data file
						// PID,IndividualSARRecordID,ID,Age,Social Class,Gender,Marital Status etc.
						s_logger.debug("not a data line: " + line);
					} else {
						st.nextToken();
						st.nextToken();
						st.nextToken();
						st.nextToken();
						st.nextToken();
						st.nextToken();
						st.nextToken();
						st.nextToken();
						
						String location = st.nextToken();
						//finish for all people in the area, then move to the next area
						if(i_modelMap.containsKey(location)) {
							AreaModel areaModel = (AreaModel) i_modelMap.get(location);
							areaModel.addPersonData(line);
						} else {
							AreaModel areaModel = new AreaModel(location);
							areaModel.addPersonData(line);
							i_modelMap.put(location, areaModel);
							i_areas.add(location);
						}
						
						i_numOfValidaData++;
					}
				} else {
					s_logger.error("data error(number of fields): " + st.countTokens());
					s_logger.error("data error(number of fields): " + line);
					i_invalidData.add(line);
				}
				
				// read next line
				line = in.readLine();
			}
			
		} catch(IOException e) {
			e.printStackTrace();
			s_logger.error("Unable to read data. System stops.");
			System.exit(1);
		}
		// get all areaModel information then output to model description
		for(int i=0; i<i_areas.size(); i++) {
			String location = (String) i_areas.get(i);
			
			AreaModel areaModel = (AreaModel) i_modelMap.get(location);
			
			areaModel.createAgents();
			
			i_numberOfHRPs += areaModel.getHRP().size();
			i_numberOfSpouses += areaModel.getSpouse().size();
			i_numberOfChildren += areaModel.getChild().size();
			i_numberOfElderlyDependents += areaModel.getElderly().size();
			i_numberOfAdultDependents += areaModel.getAdult().size();
			
			areaModel.buildModel();
			
			i_model.addAreaModel(areaModel);
		}
		
		outputModel();
		
		return i_model;
	}
	//This method prints out the model description
    private void outputModel() {
    	List models = i_model.getListOfAreaModels();
    	int size = models.size();
    	String outputFileName = System.getProperty("user.dir" ) + System.getProperty("file.separator") + "model.txt";
    	try {
    		PrintStream out = new PrintStream(new FileOutputStream(new File(outputFileName)));
    		out.println("--------------------- Generated Household Model -------------------------");
    		out.println("Number of valid data: " + i_numOfValidaData);
    		out.println("Number of invalid data: " + i_invalidData.size());
    		out.println("Number of area models(areas): " + i_model.getListOfAreaModels().size());
    		out.println("Number of HRPs: " + i_numberOfHRPs);
    		out.println("Number of spouses: " + i_numberOfSpouses);
    		out.println("Number of children: " + i_numberOfChildren);
    		out.println("Number of elderly dependents: " + i_numberOfElderlyDependents);
    		out.println("Number of adult dependents: " + i_numberOfAdultDependents);
    		
    		int numHouseholds = 0;
    		int numPersons = 0;
    		int numHRP = 0;
    		int numChild = 0;
    		int numElderly = 0;
    		int numAdult = 0;
    		int numSpouse = 0;
    		for(int i=0; i<size; i++) {
    			AreaModel areaModel = (AreaModel) models.get(i);
    			numPersons += areaModel.getPersons().size();
    			numHouseholds += areaModel.getListOfHouseholds().size();
    			numHRP += areaModel.getUnassignedHRP().size();
    			numChild += areaModel.getChild().size();
    			numElderly += areaModel.getElderly().size();
    			numAdult += areaModel.getAdult().size();
    			numSpouse += areaModel.getSpouse().size();
    		}
    		out.println("Number of persons : " + numPersons);
    		out.println("Number of households: " + numHouseholds);
    		out.println("Number of HRPs left: " + numHRP);
    		out.println("Number of children left: " + numChild);
    		out.println("Number of elderly dependents left: " + numElderly);
    		out.println("Number of adult dependents left: " + numAdult);
    		out.println("Number of spouse left: " + numSpouse);
    		out.println();
    		
    		if(i_invalidData.size() != 0) {
    			out.println("***** invalid data *****");
        		for(int i=0; i<i_invalidData.size(); i++) {
        			out.println((String) i_invalidData.get(i));
        		}
        		out.println("***** invalid data *****");
        		out.println();
    		}
        	for(int i=0; i<size; i++) {
        		AreaModel areaModel = (AreaModel) models.get(i);
        		out.println("--------------------- Area Model " + (i+1) + " -------------------------");
        		out.println("Model location: " + areaModel.getLocation());
        		out.println(areaModel.getModelDscription("before building model"));
        		out.println(areaModel.getModelDscription("after building model"));
        		out.println("Number of households = " + areaModel.getListOfHouseholds().size());
        		
        		int numberOfHRP = 0;
        		int i_numberOfSpouses = 0;
        		int i_numberOfChildren = 0;
        		int i_numberOfElderlyDependents = 0;
        		int i_numberOfAdultDependents = 0;
        		List households = areaModel.getListOfHouseholds();
        		for(int k=0; k<households.size(); k++) {
        			Household household = (Household) households.get(k);
        			if(household.getHrp() != null) {
        				numberOfHRP += 1;
        			}
        			
        			if(household.getSpouse() != null) {
        				i_numberOfSpouses += 1;
        			}
        			
        			i_numberOfChildren += household.getListOfChildren().size();
        			i_numberOfElderlyDependents += household.getListOfElderlyDependents().size();
        			i_numberOfAdultDependents += household.getListOfAdultDependents().size();
        		}
        		
        		out.println("Number of HRPS assigned = " + areaModel.getListOfHouseholds().size());
        		out.println("Number of spouse assigned = " + i_numberOfSpouses);
        		out.println("Number of children assigned = " + i_numberOfChildren);
        		out.println("Number of elderly dependents assigned = " + i_numberOfElderlyDependents);
        		out.println("Number of adult dependents assigned = " + i_numberOfAdultDependents);
        		out.println();
        		
        		out.println("Generated households:");
        		out.println("Household ID,Household Status,Household Size,HRP PID,HRP Age,Spouse PID,Spouse Age," +
        				"Child PID,Elderly Dependent PID,Adult Dependent PID,Location,Ward LOcation,OLocation");
        		out.flush();
        		
        		for(int k=0; k<households.size(); k++) {
        			Household household = (Household) households.get(k);
        			out.println(household.toString());
        			out.flush();
        		}
        		
        		out.println();
        		out.println("************************************************************************");
        		out.println();
        		
        		out.println("Persons unassigned: ");
        		out.println();
        		out.println("PID,IndividualSARRecordID,ID,Age,Social Class,Gender,Marital Status," +
        				"With long term illness, Health Status, Location (Output Area Zone_Code)," +
        				"HRP (Household Reference Person),Household ID,Carer,In formal care," +
        				"Household contains dependent children,Number of usual residents," +
        				"Number of Over 65 year olds in household,Fitness");
        		List unassignedPeople = new ArrayList();
        		List temp = areaModel.getUnassignedHRP();
        		for(int k=0; k<temp.size(); k++) {
        			Person person = (Person) temp.get(k);
        			unassignedPeople.add(person);
        		}
        		temp = areaModel.getChild();
        		for(int k=0; k<temp.size(); k++) {
        			Person person = (Person) temp.get(k);
        			unassignedPeople.add(person);
        		}
        		temp = areaModel.getAdult();
        		for(int k=0; k<temp.size(); k++) {
        			Person person = (Person) temp.get(k);
        			unassignedPeople.add(person);
        		}
        		temp = areaModel.getElderly();
        		for(int k=0; k<temp.size(); k++) {
        			Person person = (Person) temp.get(k);
        			unassignedPeople.add(person);
        		}
        		temp = areaModel.getSpouse();
        		for(int k=0; k<temp.size(); k++) {
        			Person person = (Person) temp.get(k);
        			unassignedPeople.add(person);
        		}
        		Collections.sort(unassignedPeople);
        		List peopleData = areaModel.getPeopleData();
        		for(int k=0; k<unassignedPeople.size(); k++) {
        			Person person = (Person) unassignedPeople.get(k);
        			String pidStr = Integer.toString(person.getPid()) + ",";
        			for(int j=0; j<peopleData.size(); j++) {
        				String personData = (String) peopleData.get(j);
        				if(personData.startsWith(pidStr)) {
        					out.println(personData);
        				}
        			}
        		}
        		out.println();
        		out.println("--------------------- Area Model " + (i+1) + " -------------------------");
        		out.println();
        	}
        	
    		out.close();
    	} catch(Exception e) {
    		e.printStackTrace();
    		s_logger.error("Unable to output model.");
    		System.exit(1);
    	}
    }
    
}