import javax.swing.*;
import java.awt.*;
import java.awt.event.*;
public class ArrayTeacher implements ActionListener
{
	//The main frame / panel
	private JPanel primary;
	private JFrame frame;
	private JFrame frame2;

	//Memebers for the main window
	private JPanel main;
	private JPanel intro;
	private JPanel p1;
	private JPanel p2;
	private JPanel p3;
	private JPanel p4;
	private JPanel p5;

	private JLabel p1L1;
	private JLabel p2L1;
	private JLabel p2L2;
	private JLabel p2L3;

	private JLabel p4L1;
	private JTextField p4T1;
	private JTextField p4T2;
	private JButton p4B1;
	private JButton p4B2;
	private JLabel p4L2;
	private JLabel p4L3;

	private JButton p5B1;
	private JButton p5B2;
	private JLabel p5L1;
	private JLabel p5L2;

	//Members for the intro panel
	private JPanel ip1;
	private JLabel iL1;
	private JLabel iL2;
	private JTextField iT1;
	private JButton iB1;
	private JLabel iL3;
	private JLabel iL4;
	private JLabel iL5;
	private JLabel iL6;
	private JLabel iL7;
	private JLabel iL8;
	private JButton iB2;
	private JTextField iT2;
	private JTextField iT3;

	//The layout magic component...
	GridBagConstraints i1 = new GridBagConstraints();

	//State variables
	int type;
	int rows;
	int cols;
	int pos;
	int walki;
	int walkj;
	int walk;

	//The actual information stored by the GUI
	int[] oneD;
	int[][] twoD;

	//The JButton arrays
	JButton[] joneD;
	JButton[][] jtwoD;

	public void actionPerformed(ActionEvent e)
	{
		try{
		//Code for the intro window
		boolean createNewWindow=false;
		if (e.getSource() == iB1 && type == 0)
		{
			if (iT1.getText().length() < 1)
				return;
			type = 1;
			cols= Integer.valueOf(iT1.getText());
			System.out.println("Create 1D array of size " + cols);
			createNewWindow = true;
		}
		if (e.getSource() == iB2 && type == 0)
		{

			if (iT2.getText().length() < 1 || iT3.getText().length() < 1)
				return;
			type = 2;
			rows = Integer.valueOf(iT2.getText());
			cols = Integer.valueOf(iT3.getText());
			createNewWindow = true;
			System.out.println("Create 2D array with " + rows + " rows and " + cols + " columns");
		}


		//Start walk button
		if (e.getSource() == p5B1)
		{
			if (walk == 1)
			{
				//Reset the button array
				if (type == 1)
					joneD[cols-1].setText("?");
				if (type == 2)
					jtwoD[rows-1][cols-1].setText("?");
				walk=0;
				p2L2.setText("");
				p2L3.setText("");
				p5B1.setText("Start Walk");
				p5B2.setVisible(false);
				p5L2.setText("--");
				return;
			}
			p5B1.setText("Stop Walk");
			if (type == 1)
				p2L2.setText("for (int i=0; i<" + String.valueOf(cols) + "; i++)");
			if (type == 2)
			{
				p2L2.setText("for (int i=0; i<" + String.valueOf(rows) + "; i++)");
			        p2L3.setText("     for(int j=0;j<" + String.valueOf(cols) + ";j++)");;
			}
			p5B2.setVisible(true);
			if (type ==1)
			{
				pos=0;
				joneD[0].setText("*****");
				p5L2.setText(String.valueOf(pos));
				p4T1.setText(String.valueOf(pos));
				p4L3.setText(String.valueOf(oneD[pos]));
			}
			if (type == 2)
			{
				walki=0;
				walkj=0;
				jtwoD[0][0].setText("****");
				p5L2.setText(String.valueOf(walki)+","+String.valueOf(walkj));
				p4T1.setText(String.valueOf(walki));
				p4T1.setText(String.valueOf(walkj));
				p4L3.setText(String.valueOf(twoD[walki][walkj]));
			}
			walk=1;
		}
		//Next button
		if (e.getSource() == p5B2)
		{
			if (type == 1)
			{
				if (pos >= cols)
					return;
				//Reset the array of buttons
				if (pos+1 < cols)
					joneD[pos].setText("?"); 
				pos = pos +1;
				if (pos < cols)
				{
					joneD[pos].setText("****"); 
					p4T1.setText(String.valueOf(pos));
					p5L2.setText(String.valueOf(pos));
				}
				else
					return;
			}
			if (type == 2)
			{
				if (walki >= rows)
					return;
				if (walki+1<rows)
					jtwoD[walki][walkj].setText("?");
				else if (walkj < cols-1)
					jtwoD[walki][walkj].setText("?");
					
				walkj = walkj + 1;
				if (walkj == cols && walki<rows)
				{
					walkj=0;
					walki++;
				}
				if  (walki < rows && walkj < cols)
				{
					jtwoD[walki][walkj].setText("****"); 
					p5L2.setText(String.valueOf(walki)+","+String.valueOf(walkj));
					p4T1.setText(String.valueOf(walki));
					p4T2.setText(String.valueOf(walkj));
				}
			}

			int value=0;
			if (type == 1 && pos < oneD.length)
			{
				value=oneD[pos];
				p4L3.setText(String.valueOf(value));
				p4T1.setText(String.valueOf(pos));
			}
			if (type == 2 && walki < rows && walkj < cols)
			{
				value = twoD[walki][walkj];
				p4L3.setText(String.valueOf(value));
				p4T1.setText(String.valueOf(walki));
				p4T2.setText(String.valueOf(walkj));
			}
		}
		//Show button
		if (e.getSource() == p4B1)
		{
			if (type == 1) 
			{
				if (p4T1.getText().length() < 1)
					return;
				pos = Integer.valueOf(p4T1.getText());
				if (pos > -1 && pos < cols)
				{
					int value=oneD[pos];
					p4L3.setText(String.valueOf(value));
				}
				else
					p4L3.setText("Out of range");
			}
			else if (type == 2 && walki < rows && walkj < cols)
			{
				if (p4T1.getText().length() < 1)
					return;
				if (p4T2.getText().length() < 1)
					return;
				walki = Integer.valueOf(p4T1.getText());
				walkj = Integer.valueOf(p4T2.getText());
				if (walki > -1 && walkj >-1 && walki < rows && walkj < cols)
				{
					int value = twoD[walki][walkj];
					p4L3.setText(String.valueOf(value));
				}
				else
					p4L3.setText("Out of range");
			}
			else
				p4L3.setText("--");
		}
		//Set button
		if (e.getSource() == p4B2)
		{
			//Make sure there is data in the text field to set
			if (p4T1.getText().length() < 1)
				return;

			String value=JOptionPane.showInputDialog("Enter a value");
			if (type == 1)
			{
				pos = Integer.valueOf(p4T1.getText()); 
				if (pos < oneD.length)
					oneD[pos] = Integer.valueOf(value);
				else
				{
					p4L3.setText("Invalid array position");
					return;
				}
			}
			if (type == 2) 
			{
				walki = Integer.valueOf(p4T1.getText());
				walkj = Integer.valueOf(p4T2.getText());
				if (walki < rows && walkj < cols)
					twoD[walki][walkj] = Integer.valueOf(value);
				else
				{
					p4L3.setText("Invalid array position");
					return;
				}
			}
			p4L3.setText(String.valueOf(value));
		}
		//Only happens if we are creating the new window for array visualization
		if (!createNewWindow)
			return;
		if (type == 1)
			oneD = new int[cols];
		if (type == 2)
			twoD = new int[rows][cols];
	        System.out.println("Test");
		createMain();
		createMiddle();
	        layoutTop();
	        layoutBottom();
	        //CardLayout cl = (CardLayout)(main.getLayout());
	        //cl.show(main, "main");
			frame2 = new JFrame();
			frame2.setTitle("Array Visualization Main Window");
			frame2.getContentPane().add(primary);
			frame2.pack();
			frame2.setSize(frame2.getWidth()+50,frame2.getHeight()+50);
			frame2.setVisible(true);
		}
		catch (Exception ex)
		{
			System.out.println(ex);
		}
	}
	public void add(int x, int y, JPanel dest, Component toAdd)
	{
		i1.fill = GridBagConstraints.HORIZONTAL;
		i1.gridx = x;
		i1.gridy = y;
		i1.weightx=.5;
		dest.add(toAdd,i1);
	}
	public ArrayTeacher()
	{
		type=0;
		walk=0;
		//Create all the members
		JFrame frame = new JFrame();
		primary = new JPanel();

		createIntro();
		layoutIntro();

		//Show to the world
		frame.getContentPane().add(intro);
		frame.pack();
		frame.setVisible(true);
	}

	public void createMain()
	{
		p1 = new JPanel();
		p2 = new JPanel();
		p3 = new JPanel();
		p4 = new JPanel();
		p5 = new JPanel();
		p1L1 = new JLabel();
		if (type == 1)
			p1L1.setText("Array of length:  " + String.valueOf(cols));
		if (type == 2)
			p1L1.setText("Array with " + String.valueOf(rows) + " Rows and " + String.valueOf(cols) + " columns ");

		//Panel 2 components
		p2L1 = new JLabel();
		p2L1.setText("Code:");
		p2L2 = new JLabel();
		p2L3 = new JLabel();


		//Panel 4 components
		p4L1 = new JLabel();
		if (type == 1)
			p4L1.setText("Position");
		if (type == 2)
			p4L1.setText("Row /  Column");
		p4T1 = new JTextField(2);
		p4T2 = new JTextField(2);
		p4B1 = new JButton();
		p4B1.setText("Show");
		p4B1.addActionListener(this);
		p4B2 = new JButton();
		p4B2.addActionListener(this);
		p4B2.setText("Set");
		p4L2 = new JLabel();
		p4L2.setText("Value");
		p4L3 = new JLabel();
		p4L3.setText("---");

		p5B1 = new JButton();
		p5B1.addActionListener(this);
		p5B1.setText("Start walk");
		p5B2 = new JButton();
		p5B2.setText("Next");
		p5B2.addActionListener(this);
		p5B2.setVisible(false);
		p5L1 = new JLabel();
		p5L2 = new JLabel();
		p5L1.setText("Position:  ");
		p5L2.setText("--");
	}
	//Create the array of buttons
	public void createMiddle()
	{
		if (type == 1 )
		{
			p3.setLayout(new GridLayout(2,cols+1));
			joneD = new JButton[cols];
			for(int i=0; i<cols; i++)
			{
				joneD[i]  = new JButton();
				joneD[i].setBackground(Color.black);
				joneD[i].setForeground(Color.red);
				joneD[i].setText("?");
				p3.add(joneD[i]);
			}
			JLabel tutor1 = new JLabel();
			tutor1.setText("Element");
			p3.add(tutor1);
			for (int i=0; i<cols; i++)
			{
				JLabel t = new JLabel();
				t.setText("        " + String.valueOf(i));
				p3.add(t);
			}
			JLabel tutor2 = new JLabel();
			tutor2.setText("Index");
			p3.add(tutor2);
		}
		if (type == 2)
		{
			p3.setLayout(new GridLayout(2*rows,cols+1));
			jtwoD = new JButton[rows][cols];
			for(int i=0; i<rows; i++)
			{
				for(int j=0; j<cols; j++)
				{
					jtwoD[i][j]  = new JButton();
					jtwoD[i][j].setBackground(Color.black);
					jtwoD[i][j].setForeground(Color.red);
					jtwoD[i][j].setText("?");
					p3.add(jtwoD[i][j]);
				}
				JLabel tutor1 = new JLabel();
				tutor1.setText("Element");
				p3.add(tutor1);
				for (int j=0; j<cols; j++)
				{
					JLabel t = new JLabel();
					t.setText("        " + String.valueOf(j));
					p3.add(t);
				}
				JLabel tutor2 = new JLabel();
				tutor2.setText("Index");
				p3.add(tutor2);
			}
		}
	}
	public void createIntro()
	{
		//Create and setup components for the intro window
		intro = new JPanel();
		ip1 = new JPanel();
		iL1 = new JLabel();
		iL1.setText("Array Teacher");
		iL2 = new JLabel();
		iL2.setText("Create 1D Array of size");
		iT1 = new JTextField(2);
		iB1 = new JButton();
		iB1.setText("Create");
		iL3 = new JLabel();
		iL3.setText("Create 2D Array with");
		iL4 = new JLabel();
		iL4.setText("Rows");
		iL5 = new JLabel();
		iL5.setText("Columns");
		iT2 = new JTextField(2);
		iT3 = new JTextField(2);
		iB2 = new JButton();
		iB2.setText("Create");
		iL6 = new JLabel();
		iL6.setText("int[] array = new int[?];");
		iL7 = new JLabel();
		iL7.setText("int[][] array = new int[x][y];");
		iL8 = new JLabel();
		iL8.setText("Java Code:");

		iB1.addActionListener(this);
		iB2.addActionListener(this);

	}
	public void layoutTop()
	{
		//Setup the main scene
		primary.setLayout(new GridBagLayout());
		add(0,0,primary,p1);
		add(0,1,primary,p2);
		add(0,2,primary,p3);
		add(0,3,primary,p4);
		add(0,4,primary,p5);

		//Panel 1 layout code
		p1.setLayout(new GridBagLayout());
		//p1.add(p1L1);
		add(0,0,p1,p1L1);
		//p1.add(p1L2);

		//Panel 2 layout code
		p2.setLayout(new GridBagLayout());
		add(0,0, p2, p2L1);
		add(0,1, p2, p2L2);
		add(0,2, p2, p2L3);
	}
	public void layoutBottom()
	{

		p4.setLayout(new GridBagLayout());
		add(0,0,p4, p4L1);
		add(0,1,p4, p4T1);
		add(2,0,p4, p4L2);
		add(2,1,p4, p4L3);
		if (type == 1)
		{
			add(1,0,p4, p4B1);
			add(1,1,p4, p4B2);
		}
		if (type == 2)
		{
			add(1,1,p4, p4B1);
			add(1,2,p4, p4B2);
			add(0,2,p4,p4T2);
		}

		p5.setLayout(new GridBagLayout());
		p5.add(p5B1);
		p5.add(p5B2);
		p5.add(p5L1);
		p5.add(p5L2);
	}
	public void layoutIntro()
	{
		//Layout the intro window
		ip1.setLayout(new GridBagLayout());
		add(0,0,ip1,iL1);
		add(0,1,ip1,iL2);
		add(1,2,ip1,iT1);
		add(2,2,ip1,iB1);
		add(0,3,ip1,iL3);
		add(0,4,ip1,iL4);
		add(0,5,ip1,iL5);
		add(1,4,ip1,iT2);
		add(1,5,ip1,iT3);
		add(2,5,ip1,iB2);
		add(0,6,ip1,iL8);
		add(0,7,ip1,iL6);
		add(0,8,ip1,iL7);
		intro.add(ip1);
		//Layout the main window
	}
	public static void main(String[] args)
	{
		ArrayTeacher funcs = new ArrayTeacher();
	}
}