JuliaSetProgram.java

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

public class JuliaSetProgram extends JPanel implements AdjustmentListener, ActionListener
{
	JFrame frame;
	int w, h, pixels;
	double a, b, zoom, dx, dy;
	JScrollBar zoomBar, aBar, bBar, iBar, hueBar, adjustX, adjustY;
	JPanel scrollPanel, boxPanel, southPanel, southEastPanel, southWestPanel;
	JLabel[] labels;
	float  maxIter, hueAdjust;

	JToggleButton pixelation;

	BufferedImage image;

	public JuliaSetProgram()
	{
		frame = new JFrame("Julia Set Program");
		frame.add(this);
		frame.setSize(1000, 800);
		frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);


		labels = new JLabel[8];
		for(int i = 0; i < 8; i++)
		{
			labels[i] = new JLabel();
		}
		w = 1000; h = 700;
		a = -0.7;
		b = 0.27015;
		a = 0;
		b = 0;
		zoom = 1;
		maxIter = 100;
		hueAdjust = 1;
		pixels = 1;
		dx = 0;
		dy = 0;

		scrollPanel = new JPanel();
		boxPanel = new JPanel();
		scrollPanel.setLayout(new GridLayout(4, 2));

		zoomBar = new JScrollBar(JScrollBar.HORIZONTAL, 50, 0, 1, 200);
		aBar = new JScrollBar(JScrollBar.HORIZONTAL, 3000, 0, 0, 6000);
		bBar = new JScrollBar(JScrollBar.HORIZONTAL, 3000, 0, 0, 6000);
		iBar = new JScrollBar(JScrollBar.HORIZONTAL, 100, 0, 1, 1000);
		hueBar = new JScrollBar(JScrollBar.HORIZONTAL, 1, 0, 0, 100);
		adjustX = new JScrollBar(JScrollBar.HORIZONTAL, 50, 0, 0, 100);
		adjustY = new JScrollBar(JScrollBar.HORIZONTAL, 50, 0, 0, 100);
		pixelation = new JRadioButton();

		zoomBar.addAdjustmentListener(this);
		aBar.addAdjustmentListener(this);
		bBar.addAdjustmentListener(this);
		iBar.addAdjustmentListener(this);
		hueBar.addAdjustmentListener(this);
		adjustX.addAdjustmentListener(this);
		adjustY.addAdjustmentListener(this);
		pixelation.addActionListener(this);

		scrollPanel.add(aBar);
		scrollPanel.add(zoomBar);
		scrollPanel.add(bBar);
		scrollPanel.add(iBar);
		scrollPanel.add(adjustX);
		scrollPanel.add(hueBar);
		scrollPanel.add(adjustY);

		boxPanel.add(pixelation);
		scrollPanel.add(boxPanel);


		southPanel = new JPanel();
		southWestPanel = new JPanel();
		southWestPanel.setLayout(new GridLayout(4, 1));

		labels[0].setText("A: "+a);
		southWestPanel.add(labels[0]);

		labels[1].setText("B: "+b);
		southWestPanel.add(labels[1]);

		labels[2].setText("DX: "+dx);
		southWestPanel.add(labels[2]);

		labels[3].setText("DY: "+dy);
		southWestPanel.add(labels[3]);

		southEastPanel = new JPanel();
		southEastPanel.setLayout(new GridLayout(4, 1));

		labels[4].setText("ZOOM: "+zoom);
		southEastPanel.add(labels[4]);

		labels[5].setText("ITERATIONS: "+maxIter);
		southEastPanel.add(labels[5]);

		labels[6].setText("HUE: "+hueAdjust);
		southEastPanel.add(labels[6]);

		labels[7].setText("PIXEL: "+pixelation.isSelected());
		southEastPanel.add(labels[7]);

		southPanel.setLayout(new BorderLayout());
		southPanel.add(southWestPanel, BorderLayout.WEST);
		southPanel.add(southEastPanel, BorderLayout.EAST);
		southPanel.add(scrollPanel, BorderLayout.CENTER);
		//southPanel.add(boxPanel, BorderLayout.EAST);

		frame.add(southPanel, BorderLayout.SOUTH);
		setBackground(Color.white);
		frame.setVisible(true); // should be last thing
	}

	public BufferedImage generateImage()
	{
		double zx, zy;
		image = new BufferedImage(w, h, BufferedImage.TYPE_INT_RGB);

		for (int x = 0; x < w; x+=pixels)
		{
			for (int y = 0; y < h; y+=pixels)
			{
				zx = 1.5 * (x - w / 2) / (0.5 * zoom * w)+dx;
				zy = (y - h / 2) / (0.5 * zoom * h)+dy;

				float i = maxIter;
				while(zx*zx+zy*zy < 6 && i > 0)
				{
					double diff = (zx*zx)-(zy*zy)+a; //plus complex number value
					zy = 2*zx*zy+b;
					zx = diff;
					i--;
				}

				int c;
				if(i>0)
					c = Color.HSBtoRGB(hueAdjust*(maxIter / i) % 1, 1, 1);
				else c = Color.HSBtoRGB((maxIter / i), 1, 0);
				image.setRGB(x, y, c);
			}
		}
		return image;
	}

	public void actionPerformed(ActionEvent e)
	{
		if(pixelation.isSelected())
			pixels = 5;
		else
			pixels = 1;


		repaint();
	}

	public void adjustmentValueChanged(AdjustmentEvent e)
	{
		if(e.getSource() == zoomBar)
		{
			zoom = (0.99/49)*zoomBar.getValue();
			dy = ((adjustY.getValue()*-1 / 100.0)+.5)*(1/zoom);
			dx = ((adjustX.getValue()*-1 / 100.0)+.5)*(1/zoom);
			//dx/=zoomBar.getValue();
		}
		if(e.getSource() == aBar)
		{
			a = (aBar.getValue()/1000.0) - 3.0;
		}
		if(e.getSource() == bBar)
		{
			b = (bBar.getValue()/1000.0) - 3.0;
		}
		if(e.getSource() == iBar)
		{
			maxIter = iBar.getValue();
		}
		if(e.getSource() == hueBar)
		{
			hueAdjust = hueBar.getValue()/100.0f;
		}
		if(e.getSource() == adjustX)
		{
			dx = ((adjustX.getValue()*-1 / 100.0)+.5)*(1/zoom);
		}
		if(e.getSource() == adjustY)
		{
			dy = ((adjustY.getValue()*-1 / 100.0)+.5)*(1/zoom);
		}

		labels[0].setText("A: "+(double)Math.round(a * 1000d) / 1000d);
		labels[1].setText("B: "+(double)Math.round(b * 1000d) / 1000d);
		labels[2].setText("DX: "+(double)Math.round(dx * 1000d) / 1000d);
		labels[3].setText("DY: "+(double)Math.round(dy * 1000d) / 1000d);
		labels[4].setText("ZOOM: "+(double)Math.round(zoom * 1000d) / 1000d);
		labels[5].setText("ITERATIONS: "+maxIter);
		labels[6].setText("HUE: "+hueAdjust);
		labels[7].setText("PIXEL: "+pixelation.isSelected());
		repaint();
	}

	public void paintComponent(Graphics g)
	{
		super.paintComponent(g); //GIANT ERASER, Wipes out everything and redraws everything

		//g.setColor(new Color(redValue, greenValue, blueValue));

		//g.fillRect(0, 0, frame.getWidth(), frame.getHeight());

		g.drawImage(generateImage(),0,0,null);
	}

	public static void main(String[]args)
	{
		JuliaSetProgram app = new JuliaSetProgram();
	}

}