import java.awt.*;
import java.awt.event.*;
import java.applet.*;
import java.util.*;
import java.lang.Math.*;

public class Virus extends Applet {
    ControlPanel controls;
    
    public void init() {
		setLayout(new BorderLayout());
		DisplayCanvas c = new DisplayCanvas();
		add("Center", c);
		add("North", controls = new ControlPanel(c));
    }

    public void start() {
		controls.setEnabled(true);
    }

    public void stop() {
		controls.setEnabled(false);
    }

    public boolean handleEvent(Event e) {
		if (e.id == Event.WINDOW_DESTROY) {
		    System.exit(0);
		}
		return false;
    }

    public static void main(String args[]) {
		Frame f = new Frame("Virus");
		Virus virus = new Virus();

		virus.init();
		virus.start();

		f.add("Center", virus);
		f.setSize(900, 400);
		f.show();
    }
}
    

class DisplayCanvas extends Canvas {

	static final int SIZE = 200;
	static int FOCAL_RADIUS = 20;
	static int MAX_AGE = 5;
	static double INIT_RATE = 0.10;
	static double INFECT_RATE = 0.10;
	
	static final int normal_cell = 0;
	static final int red_v = 1;
	static final int new_red_v = 2;
	static final int old_red_v = 3;
	static final int green_v = 4;
	static final int new_green_v = 5;
	static final int old_green_v = 6;
	
	static Color darkgreen = Color.green.darker();
		
	static int timestep, randchoice;
	Random randval = new Random();
	
	int[][] cell_state = new int[2*SIZE-1][];
	int[][] cell_age = new int[2*SIZE-1][];
	
	public DisplayCanvas() {
		for (int row=0; row<SIZE; row++) {
			cell_state[row] = new int[SIZE+row];
			cell_age[row] = new int[SIZE+row];
		}
		for (int row=SIZE; row<cell_state.length; row++) {
			cell_state[row] = new int[3*SIZE-(row+2)];
			cell_age[row] = new int[3*SIZE-(row+2)];
		}
	    for (int row=0; row<2*SIZE-1; row++) {
			for (int col=0; col<cell_state[row].length; col++) {
				cell_state[row][col] = normal_cell;
				cell_age[row][col] = 0;
			}
		}
	}

    public void init( int radius, double rate) {
    	FOCAL_RADIUS = radius;
    	INIT_RATE = rate;
    	timestep = 0;
    	for (int row=0; row<2*SIZE-1; row++)
			for (int col=0; col<cell_state[row].length; col++) {
				cell_state[row][col] = normal_cell;
				cell_age[row][col] = 0;
		}
		for (int row=(SIZE-(FOCAL_RADIUS+1)); row<(SIZE+FOCAL_RADIUS); row++)
			for (int col=(SIZE-(FOCAL_RADIUS+1));col<(cell_state[row].length+FOCAL_RADIUS+1-SIZE);col++){
				randchoice = Math.abs( randval.nextInt())%10000;
				if (randchoice < 5000*INIT_RATE) cell_state[row][col] = red_v;
				else if (randchoice < 10000*INIT_RATE) cell_state[row][col] = green_v;
			}
		repaint();
    }

    public void paint(Graphics g) {
		for (int row=0; row<2*SIZE-1; row++)
			for (int col=0; col<cell_state[row].length; col++) {
				switch (cell_state[row][col]) {
					case old_red_v: g.setColor(Color.pink); break;
					case red_v:
					case new_red_v: g.setColor(Color.red); break;
					case old_green_v: g.setColor(Color.green); break;
					case green_v:
					case new_green_v: g.setColor(darkgreen); break;
					case normal_cell: g.setColor(Color.white); break;
				}
				g.drawLine(2*col+Math.abs(1+row-SIZE),row,2*col+Math.abs(2+row-SIZE),row);
			}
		g.setColor(Color.black);
		g.drawString("Time: "+timestep, 10, 20);
    }

    public void iterate(int iterations, double rate, int max_age) {
    
    int redcount, greencount;
    	this.MAX_AGE = max_age;
		this.INFECT_RATE = rate;
		while (iterations > 0) {    //Loop for number of replications
		    timestep++; iterations--;
		    for (int row=1; row<SIZE-1; row++)
			  for (int col=1; col<cell_state[row].length-1; col++)       // avoid edges
			    if (cell_state[row][col] == normal_cell)          // if uninfected
			      if (Math.abs( randval.nextInt())%10000 < 10000*INFECT_RATE) { // 1 in INVINFRATE chance of infection
				  	redcount = greencount = 0;
				  	if (cell_state[row-1][col-1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row-1][col-1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[row-1][col] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row-1][col] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[row][col-1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row][col-1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[row][col+1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row][col+1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[row+1][col] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row+1][col] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[row+1][col+1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row+1][col+1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (redcount == 0) {
				  	  if (greencount > 0) cell_state[row][col] = new_green_v;
				  	}
				  	else {
				  	  if (Math.abs( randval.nextInt())%(redcount+greencount) < redcount) cell_state[row][col] = new_red_v;
				  	  else cell_state[row][col] = new_green_v;
				  	}
				  }
			  for (int col=1; col<cell_state[SIZE-1].length-1; col++)  // avoid edges
			    if (cell_state[SIZE-1][col] == normal_cell)         // if uninfected
			      if (Math.abs( randval.nextInt())%10000 < 10000*INFECT_RATE) {   // 1 in INVINFRATE chance of infection
				  	redcount = greencount = 0;
				  	if (cell_state[SIZE-2][col-1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[SIZE-2][col-1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[SIZE-2][col] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[SIZE-2][col] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[SIZE-1][col-1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[SIZE-1][col-1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[SIZE-1][col+1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[SIZE-1][col+1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[SIZE][col-1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[SIZE][col-1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[SIZE][col] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[SIZE][col] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (redcount == 0) {
				  	  if (greencount > 0) cell_state[SIZE-1][col] = new_green_v;
				  	}
				  	else {
				  	  if (Math.abs( randval.nextInt())%(redcount+greencount) < redcount) cell_state[SIZE-1][col] = new_red_v;
				  	  else cell_state[SIZE-1][col] = new_green_v;
				  	}
				  }
		    for (int row=SIZE; row<2*SIZE-2; row++)
			  for (int col=1; col<cell_state[row].length-1; col++)		 // avoid edges
			    if (cell_state[row][col] == normal_cell)          // if uninfected
			      if (Math.abs( randval.nextInt())%10000 < 10000*INFECT_RATE) { // 1 in INVINFRATE chance of infection
				  	redcount = greencount = 0;
				  	if (cell_state[row-1][col] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row-1][col] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[row-1][col+1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row-1][col+1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[row][col-1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row][col-1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[row][col+1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row][col+1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[row+1][col-1] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row+1][col-1] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (cell_state[row+1][col] == red_v) redcount += Math.abs( randval.nextInt())%34;
				  	else if (cell_state[row+1][col] == green_v) greencount += Math.abs( randval.nextInt())%34;
				  	if (redcount == 0) {
				  	  if (greencount > 0) cell_state[row][col] = new_green_v;
				  	}
				  	else {
				  	  if (Math.abs( randval.nextInt())%(redcount+greencount) < redcount) cell_state[row][col] = new_red_v;
				  	  else cell_state[row][col] = new_green_v;
				  	}
				  }

		    for (int row=1; row<2*SIZE-2; row++)
			  for (int col=1; col<cell_state[row].length-1; col++)
				switch (cell_state[row][col]) {
					case red_v: if (++cell_age[row][col] == MAX_AGE)
						cell_state[row][col] = old_red_v;
						break;
					case green_v: if (++cell_age[row][col] == MAX_AGE)
						cell_state[row][col] = old_green_v;
						break;
					case new_red_v: cell_state[row][col] = red_v;
						break;
					case new_green_v: cell_state[row][col] = green_v;
						break;
				}
		  }
		repaint();
    }
}

class ControlPanel extends Panel
                  implements ActionListener {
    TextField AgeField;
    TextField InitField;
    TextField FocusField;
    TextField InfectField;
    TextField IterField;
    DisplayCanvas canvas;

    public ControlPanel(DisplayCanvas canvas) {
	Button b = null;

	this.canvas = canvas;
	add(new Label("Max Age"));
	add(AgeField = new TextField("5", 4));
	add(new Label("Init Rate"));
	add(InitField = new TextField("0.10", 5));
	add(new Label("Focal radius"));	
	add(FocusField = new TextField("20", 4));
	add(new Label("Rate"));
	add(InfectField = new TextField("0.10", 5));
	add(new Label("Iterations"));
	add(IterField = new TextField("50", 4));
	b = new Button("Run");
	b.addActionListener(this);
	add(b);
	b = new Button("Restart");
	b.addActionListener(this);
	add(b);
	canvas.init(20, 0.10);
    }
    
    public void actionPerformed(ActionEvent ev) {
    	double infect_rate = new Double(InfectField.getText().trim()).doubleValue();
    	double init_rate = new Double(InitField.getText().trim()).doubleValue();

		String label = ev.getActionCommand();
		if (label.equals("Run"))
			canvas.iterate(Integer.parseInt(IterField.getText().trim()),
				infect_rate, Integer.parseInt(AgeField.getText().trim()));
		else canvas.init(Integer.parseInt(FocusField.getText().trim()),
				init_rate);
    }
}