/**
 *  Hunter-Prey simulator version 1.1
 *  Copyright 2011 Kenneth Mackin
 */
public class Field {
    Agent[] agent;
    FieldView view;

    int width = 50;
    int height = 50;
    int agents = 5;
    int MAXSTEPS=1000;
    int TRAININGCYCLE=500;
    int AVERAGECYCLE=10;

    int RANGE=2;

    double[][] param; 
    double[][] bestparam; 
    double bestfitness;

    public static void main(String[] args){
	boolean voption = false;
	if(args.length>0){
	    if(args[0].equals("-v")){
		System.err.println("Visible mode");
		voption = true;
	    }
	}
	new Field(voption);
    }

    public Field(boolean visible){
	agent = new Agent[agents];

	if(visible){
	    view = new FieldView(this);
	}

	double bestfitness=MAXSTEPS+1;

	for(int i=0; i<TRAININGCYCLE; i++){
	    System.err.println("Training:"+i);
	    double averagefitness=0;
	    //adjustable Hunter parameters
	    //param = new double[][]{{0.4,-5,-4},{0.4,5,-4},{0.7,3,3},{0.7,-3,3}};
	    param = new double[agents-1][3];
	    for(int a=0; a<agents-1; a++){
		param[a][0]=(Math.random()*1.0);
		param[a][1]=(int)((Math.random()*11)-5);
		param[a][2]=(int)((Math.random()*11)-5);
	    }

	    for(int j=0; j<AVERAGECYCLE; j++){
		init(param);
		int fitness = this.start(false);
		System.out.println(fitness);
		averagefitness+=fitness;
	    }
	    averagefitness/=AVERAGECYCLE;
	    System.out.println(averagefitness);
	    if(averagefitness < bestfitness){
		bestfitness = averagefitness;
		bestparam = param;
		System.err.println("Improved Best Fitness!");
	    }
	}

	System.out.print(bestfitness);
	for(int i=0; i<agents-1; i++){
	    System.out.print(param[i][0]+",");
	    System.out.print(param[i][1]+",");
	    System.out.println(param[i][2]+",");
	}
	System.out.println();

	init(bestparam);
	start(visible);

    }

    public void init(double[][] param){
	agent[0] = new PreyAgent(this, width/2, height/2, 0);

	for(int i=1; i<agents; i++){ //initialize Hunters
	    agent[i] = new HunterAgent(this, (int)(Math.random()*width), height-1, i, param[i-1][0], param[i-1][1], param[i-1][2]);
	}

    }

    public int start(boolean visible){
	int counter=0;
	boolean end=false;
	int fitness=0;

	while(!end){
	    counter++;

	    for(int i=1; i< agent.length; i++){
		agent[i].move();
	    }

	    agent[0].move();

	    if(visible){
		view.update();
	    }

	    for(int i=1; i< agent.length; i++){
		if((Math.abs(agent[i].getX() - agent[0].getX()) <= RANGE) &&
		   (Math.abs(agent[i].getY() - agent[0].getY()) <= RANGE) ){
		    System.err.println("Capture!");
		    fitness=counter;
		    end=true;
		}
	    }

	    if(agent[0].getX()<0 || agent[0].getX()>=width || 
	       agent[0].getY()<0 || agent[0].getY()>=height){
		System.err.println("Escape!");
		fitness=MAXSTEPS;
		end=true;
	    }

	    if(counter>=MAXSTEPS){
		System.err.println("Time up!");
		fitness=MAXSTEPS;
		end=true;
	    }
	}
	
	if(visible){
	    view.showAgents();
	}

	System.err.println("counter="+counter);

	return fitness;

    }

    public int getWidth(){
	return width;
    }

    public int getHeight(){
	return height;
    }

    public int getX(int index){
	return agent[index].getX();
    }

    public int getY(int index){
	return agent[index].getY();
    }

    public int getAgents(){
	return agents;
    }

}