//package Dasearch;

import java.awt.*;
import java.awt.event.*;
import java.awt.Graphics;
import javax.swing.*;
import java.awt.image.BufferedImage;
import javax.imageio.ImageIO;
import java.io.*;
import java.net.Socket;

public class pathfinder extends JFrame
{
    public static boolean showtrace = true; // does not erase image
    protected BufferedImage diamondgif, chargif;  // animated gif images
    protected Graphics display;
    protected Image dbuf; // double buffer
    protected static int gap = 20; // 24;  // side/radius of hexagon
    protected int yoff = 40; // 40;
    protected int delaytime = 250; // animation delay time
    protected astar PG;
    int rows, cols;
    int XDIM, YDIM; // window dimensions
    int gemx, gemy, profx, profy;
    protected BufferedImage[] imageof; // image vector for terrain.
    protected BufferedImage[] imagechar; //image vector for character based on terrain.
    Color[] colorof ={Color.green,Color.yellow,Color.white,Color.blue};
 // color corresponding to each terrain type.

    public static String fpath = "";  // file path prefix
    public static String savefile = "myastar.run";
    public static boolean downloadconfig = false;
    public static boolean sendpath = false;

    // graphical representation
    Hexagon[][] HX;
    int hpdist = Hexagon.calchpdist(gap);
    
    // access center graphical coordinates at cell i,j
    int getx(int i, int j) { return HX[i][j].x; }
    int gety(int i, int j) { return HX[i][j].y; }

    Graphics Gg;

    public void paint(Graphics g) {} // override automatic repaint

    public pathfinder(int r, int c, String[] argv) // constructor
    {
	rows = r; cols = c;
	if (argv.length>0 && !isint(argv[0])) {
	    astar.genmap=false; // load map from file or socket
	    if (argv.length==2&&argv[0].equals("download"))
		{downloadconfig=sendpath=true;}
	}
	PG = new myastar(rows,cols);  // note it's myastar, not astar
	if (downloadconfig) downloadrun(defaultserver,defaultsrvport,argv[1]);
	else if (!astar.genmap&&!argv[0].equals("relay")) loadrun(argv[0],argv.length==1);
	else if (!astar.genmap) loadrun(true); // relayconfig set externally
	if (astar.genmap) PG = new myastar(r,c); // if (down)loadrun failed
	//((myastar)PG).PF=this; 
	rows = PG.ROWS;  cols = PG.COLS;
	System.out.print("Costs: ");
	for(int tc:PG.costof) System.out.print(tc+" ");
	System.out.println();
	hpdist = Hexagon.calchpdist(gap);	
	HX = new Hexagon[rows][cols];  // graphical map
	for(int i=0;i<rows;i++)
	    { 
	      int odd = i%2;
	      for(int j=0;j<cols;j++)
		{
		    HX[i][j] = new Hexagon(yoff/2+(j*2+odd)*hpdist,yoff+gap+(3*gap/2*i),gap);
		}
	    }//for i,j
	XDIM = cols*hpdist*2+ yoff/2; 
	YDIM = ((rows+1)*gap*3)/2 - yoff/4;
	this.setBounds(0,0,XDIM+5,YDIM+yoff+5);
	this.setVisible(true); 
        this.setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE);
 try {   // draw to double buffer
	dbuf = createImage(XDIM,YDIM+yoff);
	display = dbuf.getGraphics();  // display draws to buffer
	display.fillRect(0,0,XDIM,YDIM+yoff);  // fill background	

	diamondgif = ImageIO.read(new File(fpath+"gem1.gif"));
	prepareImage(diamondgif,this);
	chargif = ImageIO.read(new File(fpath+"man15.gif"));
	prepareImage(chargif,this);
	
	imagechar = new BufferedImage[4]; // image of character on terrain type
	imagechar[astar.OPEN] = chargif;
	imagechar[astar.WATER] =
	    ImageIO.read(new File(fpath+"boat.gif"));
	imagechar[astar.FIRE] = 
	    ImageIO.read(new File(fpath+"dragon.gif"));
	imagechar[astar.DESERT] = chargif;
	prepareImage(imagechar[astar.WATER],this);
	imageof = new BufferedImage[4];  // background image for terrain type
	BufferedImage im = ImageIO.read(new File(fpath+"Water.gif"));
	int iw = im.getWidth(this), ih = im.getHeight(this);
	imageof[astar.WATER]= new BufferedImage(gap*2,gap*2,BufferedImage.TYPE_INT_RGB);
	Hexagon Hi = new Hexagon(hpdist,gap,gap);	
	Gg = imageof[astar.WATER].getGraphics();
	//	Gg.setClip(Hi);
	Gg.drawImage(im,0,0,null);
	prepareImage(imageof[astar.WATER],this);

	BufferedImage imm = ImageIO.read(new File(fpath+"grass1.gif"));	
	iw = imm.getWidth(null); ih = imm.getHeight(null);
	imageof[astar.OPEN] = new BufferedImage(gap*2,gap*2,BufferedImage.TYPE_INT_RGB);
	Gg = imageof[astar.OPEN].getGraphics();
	//	Gg.setClip(Hi);
	Gg.drawImage(imm,0,0,null);
	prepareImage(imageof[astar.OPEN],this);

	//	BufferedImage imf = ImageIO.read(new File(fpath+"volcano.gif"));
	BufferedImage imf = ImageIO.read(new File(fpath+"flames.jpeg"));	
	iw = imf.getWidth(null); ih = imf.getHeight(null);
	imageof[astar.FIRE] = new BufferedImage(gap*2,gap*2,BufferedImage.TYPE_INT_RGB);
	Gg = imageof[astar.FIRE].getGraphics();
	Gg.drawImage(imf,0,0,null);
	prepareImage(imageof[astar.FIRE],this);	

	try{Thread.sleep(500);} catch(Exception e) {} // Synch with system
	// draw static background as a green rectangle
	display.setColor(Color.green);
	display.fillRect(0,0,XDIM,YDIM+yoff);  // fill background

	// generate random starting positions.
	  // generate initial positions of professor and diamond
	if (astar.genmap)
	{
	  do 
	    {
		gemx = (int)(Math.random() * PG.COLS);  // diamond
		gemy = (int)(Math.random() * PG.ROWS);
	    }
	  while (PG.M[gemy][gemx]!=PG.OPEN);
	  do 
	    {
		profx = (int)(Math.random() * PG.COLS);  
                profy = (int)(Math.random() * PG.ROWS);
	    }
	  while (PG.M[profy][profx]!=PG.OPEN ||
	       astar.hexdist(gemy,gemx,profy,profx)<PG.ROWS*5/5);
	}// if need to generate new starting positions

	// draw map
	drawmap();
	// draw professor and diamond, initial position

	// draw static map
	display = (Graphics)this.getGraphics(); // change display
	display.drawImage(dbuf,0,0,this);
	
	int px = getx(profy,profx), py = gety(profy,profx); // center hx coords
	display.drawImage(imagechar[PG.M[profy][profx]],
			  (px-gap/2),(py-gap/2),gap,gap,null);
	px = getx(gemy,gemx); py = gety(gemy,gemx);
	display.drawImage(diamondgif,px-gap/2,py-gap/2,gap,gap,null);
	/*	
	display.drawImage(imagechar[PG.M[profy][profx]],
			  (profx*gap),(profy*gap)+yoff,gap,gap,null);
	display.drawImage(diamondgif,gemx*gap,gemy*gap+yoff,gap,gap,null);
	*/
	animate();
	if (astar.genmap || downloadconfig)
	    { saverun(); // save the run
	      System.out.println("run saved in "+(fpath+savefile));
	    }
}//try
catch (Exception eee) {}
    } // constructor

    // save map and positions to file:
    void saverun()
    {
	try {
	    mapconfig mc = new mapconfig(profy,profx,gemy,gemx,PG.M,PG.costof);
	    ObjectOutputStream dout = new ObjectOutputStream(new FileOutputStream(fpath+savefile));
	    dout.writeObject(mc);
	    dout.close();
	}
	catch(IOException ie) {System.out.println(ie);}
    }//saverun

    void uploadrun(ObjectOutputStream bout) throws IOException
    { 
	mapconfig mc = new mapconfig(profy,profx,gemy,gemx,PG.M,PG.costof);
	bout.writeObject(mc);
    }//uploadrun

    void loadrun(String filen, boolean loadcost)
    {
	try {
	    ObjectInputStream din = new ObjectInputStream(new FileInputStream(fpath+filen));
	    mapconfig mc = (mapconfig)din.readObject();
	    din.close();
	    profy = mc.chary;  profx = mc.charx;
	    gemy = mc.goaly;  gemx = mc.goalx;
	    PG.genmap=false;
	    PG.ROWS = mc.ROWS;  PG.COLS = mc.COLS;
	    PG.M = mc.Map;
	    if (loadcost) PG.costof = mc.costof;
	}
	catch(Exception ie) {System.out.println(ie); astar.genmap=true;}
    }//loadrun
    static mapconfig relayconfig=null;
    void loadrun(boolean loadcost)
    {   if (relayconfig==null) {astar.genmap=true; return;}
	mapconfig mc=relayconfig;
	profy = mc.chary;  profx = mc.charx;
	gemy = mc.goaly;  gemx = mc.goalx;
	PG.genmap=false;
	PG.ROWS = mc.ROWS;  PG.COLS = mc.COLS;
	PG.M = mc.Map;
	if (loadcost) PG.costof = mc.costof;
    }//loadrun	

    void downloadrun(String srvaddr, int srvport, String cname)
    {	   
	try {
	    csk = new Socket(srvaddr,srvport);
	    csk.setSoTimeout(5000);
	    cskout = new ObjectOutputStream(csk.getOutputStream());
	    cskout.writeObject(cname);
	    cskin = new ObjectInputStream(csk.getInputStream());
	    mapconfig mc = (mapconfig)cskin.readObject();
	    profy = mc.chary;  profx = mc.charx;
	    gemy = mc.goaly;  gemx = mc.goalx;
	    PG.genmap=false;
	    PG.ROWS = mc.ROWS;  PG.COLS = mc.COLS;
	    PG.M = mc.Map;
	    PG.costof = mc.costof;	    
	}catch (Exception e) {
	    System.out.println(e);
	    astar.genmap=true; downloadconfig=false;
	}
    }//downloadrun

    public void trace(int y, int x, char t)
    {
	if (x<0 || x>=cols || y<0 || y>=rows) return;
	display.setColor(Color.black);	
	display.drawString(""+t,getx(y,x)-gap,gety(y,x)-gap-4);
    }
    
    public void animate()
    {
	// invert path.
	coord path = null;
	try {  // student code
	   path = PG.search(gemy,gemx, profy,profx); // call SEARCH here
	} catch (Exception e) {System.out.println("error in call to search: "+e);}

	if (sendpath && csk!=null) try {
		//		ObjectOutputStream bout=new ObjectOutputStream(csk.getOutputStream());
		cskout.writeObject(path);
	    } catch (Exception e) {System.out.println(e);}
	    
	if (path==null) 
	    {   display.setColor(Color.red);
		display.drawString("NO PATH TO TARGET!",50,100); 
		System.out.println("no path"); 
		if (sendpath && csk!=null) try {
	  	  String response = (String)cskin.readObject();
          	  System.out.println("Response from server: "+response);
          	  csk.close();
		}catch(Exception ie){}
		return;
	    }
	int px=0, py=0; // for calculating graphical coords
	while (path!=null)
	    {
		px = getx(path.y,path.x); py = gety(path.y,path.x);
		display.drawImage(imagechar[PG.M[path.y][path.x]],
				(px-gap/2),(py-gap/2),gap,gap,null);
		//	      display.drawImage(imagechar[PG.M[path.y][path.x]],
		//				(path.x*gap),(path.y*gap)+yoff,gap,gap,null);

		//System.out.printf("%d,%d: %d\n",path.y,path.x,PG.M[path.y][path.x]);

	      try{Thread.sleep(delaytime);} catch(Exception se) {}
	      //	      display.drawImage(imageof[PG.M[path.y][path.x]],
	      //				(path.x*gap),(path.y*gap)+yoff,gap,gap,null);
	      //	      display.setColor(Color.red);
	      //      	      display.fillOval((path.x*gap)+8,(path.y*gap)+yoff+8,4,4);
	      // for animation:
	      //	      display.drawImage(diamondgif,gemx*gap,gemy*gap+yoff,gap,gap,null);	      

	      if (!showtrace) // erase trail - redraw static map
		  display.drawImage(dbuf,0,0,this);
	      path = path.prev;
	    }//with path!=null
	px = getx(gemy,gemx); py = gety(gemy,gemx);
	display.drawImage(diamondgif,px-gap/2,py-gap/2,gap,gap,null);
	display.drawImage(imagechar[PG.M[gemy][gemx]],px-gap/2,py-gap/2,gap,gap,null);

	if (sendpath && csk!=null) try {
	  String response = (String)cskin.readObject();
          System.out.println("Response from server: "+response);
          csk.close();
	}catch(Exception ie){}

    }//animate

    public void drawmap()
    {   
	int i, j;
	for(i=0;i<PG.ROWS;i++)
	    for(j=0;j<PG.COLS;j++)
		{

		    display.setColor(colorof[ PG.M[i][j] ]);
		    //		    display.fillPolygon(HX[i][j]);
		    //		    Gg = imageof[PG.M[i][j]].getGraphics();
		    //		    Gg.setClip(HX[i][j]);
		    //		    Gg.drawImage(imageof[PG.M[i][j]],0,0,null);
		    //		    display.drawImage(imageof[PG.M[i][j]],j*gap,(i*gap)+yoff,gap,gap,null);
		    display.setClip(HX[i][j]);
		    display.drawImage(imageof[PG.M[i][j]],yoff/2 + (j*2+(i%2)-1)*hpdist,(3*gap/2*i)+yoff,null);		    
		}
	//try{Thread.sleep(1000);} catch(Exception e) {} 
    } // drawmap

    public static boolean isint(String s) // check if string is an integer
    {   boolean answer=true;
	try {Integer.parseInt(s);}
	catch (NumberFormatException nfe) {answer=false;}
	return answer;
    }

    static int defaultrows = 24;
    static int defaultcols = 36;
    static String defaultserver = "96.57.41.74";
    static int defaultsrvport = 17002;
    protected Socket csk;
    protected ObjectInputStream cskin;
    protected ObjectOutputStream cskout;
    public static void main(String[] args)
    {
	int r = defaultrows, c = defaultcols;
	if (args.length>=2)
	 try
	    {
		r = Integer.parseInt(args[args.length-2]);
		c = Integer.parseInt(args[args.length-1]);
		pathfinder.gap = (pathfinder.gap*defaultrows)/r;
	    }
	 catch (NumberFormatException nfe) {r=defaultrows; c=defaultcols;}
	pathfinder pf = new pathfinder(r,c,args);
    }//main

} // class pathfinder