import java.awt.*;
import java.awt.image.*;
import javax.swing.*;

public class ZPanel extends JPanel {
    BufferedImage src;
    BufferedImage dst;

    ConvolveOp convolve;

    int width, height;               // image width and height
    int kwidth=21, kheight=21;         // convolution kernel width and height
    float astrength=3, istrength=2; // activator and inhibitor strengths
    

    public ZPanel(int width, int height) { 
	this.width=width; this.height=height;

	src = new BufferedImage(width,height,BufferedImage.TYPE_INT_RGB);
	dst = new BufferedImage(width,height,BufferedImage.TYPE_INT_RGB);

	WritableRaster raster = src.getRaster();

	int arrayLength = width*height;
	int[] randomImage=new int[arrayLength];
	for (int i=0; i<arrayLength; i++) {
	    int value = 0xFF*((int)(2*Math.random()));
	    randomImage[i] = value << 16 | value << 8 | value;
	}

	raster.setDataElements(0,0,width,height,randomImage);

	computeKernel();
    }

    public ZPanel(Image image) {
	width = image.getWidth(this);
	height= image.getHeight(this);

	src = new BufferedImage(width,height,BufferedImage.TYPE_INT_RGB);
	dst = new BufferedImage(width,height,BufferedImage.TYPE_INT_RGB);

	Graphics2D gc = src.createGraphics();
	gc.drawImage(image,0,0,width,height,this);
	
	computeKernel();
    }

    public void setKernelSize(int width, int height){
	//Make kwidth and kheight odd numbers so there is a definite middle
	if (kwidth%2 == 0) width++;
	if (kheight%2 == 0) height++;

	kwidth=width; kheight=height;
	computeKernel();
    }

    public void setHormoneStrengths(int activatorStrength, int inhibitorStrength) {
	astrength = activatorStrength; 
	istrength = inhibitorStrength;
	computeKernel();
    }

    public void computeKernel() {
	float[] elements = new float[kwidth*kheight];
	
	float a,i;
	int radius;
	int current=0;
	final int ymax = (int)Math.floor(kheight/2.0);
	final int ymin = -ymax;
	final int xmax = (int)Math.floor(kwidth/2.0);
	final int xmin = -xmax;
	for (int y=ymin; y<=ymax; y++) {
	    for (int x=xmin; x<=xmax; x++) {
		a = (float)(astrength/Math.exp(((x*x)  +(y*y)/3)/6));
		i = (float)(istrength/Math.exp(((x*x)/4+(y*y))/9  ));
		elements[current++]=a-i;
		System.out.print( (a-i) + " ");
	    }
	    System.out.println();
	}

	elements[1+current/2] = 0; //set center of kernel to 0

  	Kernel kernel = new Kernel(kwidth, kheight, elements);
  	convolve = new ConvolveOp(kernel, ConvolveOp.EDGE_NO_OP, null);
    }

    public void update() {
	//apply convolution
	convolve.filter(src,dst);

	//swap source and destination images
	BufferedImage tmp = src;
	src=dst;
	dst=tmp;
    }

    public void paintComponent(Graphics g) {
	int width = getWidth();
	int height = getHeight();

	g.drawImage(src,0,0,width,height,this);
    }
}