package genegui;

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

import affectors.Affector;
import main.*;
import parameters.*;

/** ParameterViewer.java
 A corgial re-write of the ParInsepctor with a new name and some enhancements.
 @author WJS
 */
public class ParameterViewer extends JInternalFrame implements GuiInterface {
	
	/** */
	private ParameterPanel           FirstParameterPanel;
	/** */
	private Font                     FieldFont = new Font("",Font.PLAIN,10);
	/** */
	private float[]                  InitialLowerBounds;
	/** */
	private float[]                  InitialStepSizes;
	/** */
	private int[]                    InitialTypes;
	/** */
	private float[]                  InitialUpperBounds;
	/** */
	private float[]                  InitialValues;
	/** The MenuBar */
	private JMenuBar                 MainMenu;
	/** The Main panel that stores all */
	private JPanel                   MainPanel;
	/** The ParameterSet being viewed/edited */
	private ParameterSet			       Parameters = new ParameterSet("fullstate");
	/** THe panel for storing all ParameterPanels */
  private JPanel                   ParametersPanel;
	/** Self Reference */
	private ParameterViewer          SaveThis = this;
	/** If set to true, simple display mode is used which just shows values */
	private boolean                  SimpleDisplay = false;
	/** The MainGui object that this belongs to */
	private MainGui                  TheMainGui;
	
	/** Simple constructor.
	 @author WJS
	 */
	public ParameterViewer(MainGui theMainGui, String title, boolean simpleDisplay) {
		
		super(title,true,true,true,true);
		
		TheMainGui = theMainGui;
		SimpleDisplay = simpleDisplay;
		
		// Make main panel
		MainPanel = new JPanel();
		MainPanel.setLayout(new BoxLayout(MainPanel,BoxLayout.Y_AXIS));
		MainPanel.setBackground(Color.lightGray);
		setContentPane(MainPanel);
		
		// Make a panel for storing ParamaterPanels
		ParametersPanel = new JPanel();
		
		// Initialize all the entries
		init();
		
		pack();
		setVisible(true);

		if (SimpleDisplay)
			setSize(220,400);
		else
		  setSize(600,400);
		
		// Make a header panel
		ParameterHeaderPanel header = new ParameterHeaderPanel();
		MainPanel.add(header,0);
		
		// Make a scroll panel for this list of all ParameterPanels
		JScrollPane scrollPane = new JScrollPane(ParametersPanel);
		MainPanel.add(scrollPane);
		
		// Make a JMenuBar
		MainMenu = new JMenuBar();
		MainMenu.setBackground(Color.lightGray);
		
		JButton setAllButton = new JButton("Set All");
		setAllButton.setOpaque(false);
		setAllButton.setFont(FieldFont);
		MainMenu.add(setAllButton);
		
		JButton resetAllButton = new JButton("Reset All");
		resetAllButton.setOpaque(false);
		resetAllButton.setFont(FieldFont);
		MainMenu.add(resetAllButton);
		
		setJMenuBar(MainMenu);
		
		// Add an internal frame listener for capturing frame closing events
		addInternalFrameListener(new InternalFrameAdapter() {
					public void internalFrameClosing(InternalFrameEvent ife) {
						close();
					}
				});
		
		
	}
	
	/** Close up the JFrame properly, and disconnect from listener lists.
	 @author WJS
	 */
	private void close() {
		
		dispose();
		GeneNet.TheMainGui.removeInternalFrame(this);
		GeneNet.TheMainGui.removeGUIListener(this);
		
	}
	
	/** Initialize the display
	 */
	public void init() {
		
		// Make arrays for storing initial values so they may be reset.
		InitialLowerBounds = new float[Parameters.getNumParams()];
		InitialUpperBounds = new float[Parameters.getNumParams()];
		InitialValues = new float[Parameters.getNumParams()];
		InitialStepSizes = new float[Parameters.getNumParams()];
		InitialTypes = new int[Parameters.getNumParams()];
		
		ParameterPanel pPanel;
		ParametersPanel.setLayout(new BoxLayout(ParametersPanel,BoxLayout.Y_AXIS));
		for (int i=0;i<Parameters.getNumParams();i++) {
			// Store values in the Initial arrays
			InitialValues[i] = Parameters.getValue(i);
			InitialLowerBounds[i] = Parameters.getLowerBound(i);
			InitialUpperBounds[i] = Parameters.getUpperBound(i);
			InitialStepSizes[i] = 10;
			InitialTypes[i] = Parameters.getVariationMode(i);
			pPanel = new ParameterPanel(i,Parameters.getName(i),Parameters.getValue(i),Parameters.getLowerBound(i),Parameters.getUpperBound(i),10,Parameters.getVariationMode(i));
			ParametersPanel.add(pPanel);
			if (i==0) FirstParameterPanel = pPanel;
		}
		
		
	}
	
	/** Re-initialize the display.
	 @author WJS
	 */
	private void reinitialize() {
		
		int height = getHeight();
		
		ParametersPanel.removeAll();
		MainPanel.remove(0);
		
		ParameterPanel pPanel;
		for (int i=0;i<Parameters.getNumParams();i++) {
			pPanel = new ParameterPanel(i,Parameters.getName(i),Parameters.getValue(i),Parameters.getLowerBound(i),Parameters.getUpperBound(i),10,Parameters.getVariationMode(i));
			ParametersPanel.add(pPanel);
			if (i==0) FirstParameterPanel = pPanel;
		}
		
		// Make a header panel
		ParameterHeaderPanel header = new ParameterHeaderPanel();
		MainPanel.add(header,0);
		
		pack();
		
		if (SimpleDisplay)
			setSize(220,height);
		else
		  setSize(600,height);
		
	}
	
	/** Over-ride native method to allow the small version of the window
	@param boolean simple - If true sets simple display mode
	@author WJS
	 */
	public void setIcon(boolean simple) {
		
		SimpleDisplay = simple;
		setJMenuBar(null);
		reinitialize();
		
	}
	
	/** Over-ride native method to allow the normal version of the window
	@param boolean normal - If true sets normal display mode
	@author WJS
	 */
	public void setMaximum(boolean normal) {
		
		SimpleDisplay = !normal;
		setJMenuBar(MainMenu);
		reinitialize();
		
	}
	
	/** Resets the ParameterSet
	 @param ParameterSet parameters
	 @author WJS
	 */
	public void setParameterSet(ParameterSet parameters) {
		
		Dimension d = getSize();
		
		Parameters = parameters;
		
		ParametersPanel.removeAll();
		
		init();
		
	}
	
	private void setValues(int index, String name, float value, float lower, float upper, float step, int variationType) {
		
		Parameters.setValue(index,value);
		
	}
	
	/** Updates a single Parameter
	 @parm int parameterNumber
	 @author WJS
	 */
	public void updateParameter(int parameterNumber) {
		
		Component[] c = ParametersPanel.getComponents();
		for (int i=0;i<c.length;i++) {
			if (c[i] instanceof ParameterPanel) {
				if (((ParameterPanel)c[i]).getIndex()==parameterNumber)
						((ParameterPanel)c[i]).updateParameters();
			}
		}
		
	}
	
	/** Updates Displayed values for all parameters
	 @author WJS
	 */
	public void updateDisplays() {
		
		Component[] c = ParametersPanel.getComponents();
		for (int i=0;i<c.length;i++) {
			if (c[i] instanceof ParameterPanel) {
				((ParameterPanel)c[i]).updateDisplays();
			}
		}
		
	}
	
	/** Updates all the Parameters
	 @author WJS
	 */
	public void updateParameters() {
		
		Component[] c = ParametersPanel.getComponents();
		for (int i=0;i<c.length;i++) {
			if (c[i] instanceof ParameterPanel) {
				((ParameterPanel)c[i]).updateParameters();
			}
		}
		
	}
	
	/** Changes out the parameter set and updates the display.
	 NOTE: This should only be used if the passed parameter set has the same
	 parameters in the identical order as the current set.
	 @param ParameterSet - The new ParameterSet
	 @author WJS
	 */
	public void updateParameterSet(ParameterSet ps) {
		
		Parameters = ps;
		
		updateDisplays();
		
		
	}
	
	/******************** GUItoNet Interface implementation *********************/
	
	
	/** Called when a change is made to the passed AffectorTemplate
	 @param int operation
	 @param AffectorTemplate affectorTemplate
	 @param GuiInterface source - Originator of this interface change.
	 @author WJS
	 */
	public void affectorTemplateChanged(int operation, AffectorTemplate affectorTemplate, GuiInterface source) {
		
	}
	
	
	
	/** Method for invoking general network IO operations.
	 @param int operation - The network IO operation code.
	 @param File file - The file if needed, else don't care.
	 @author WJS
	 */
	public boolean networkIO(int operation, File file) {
		
		boolean result = true;
		
		return(result);
		
	}
	
	/** Called when a change is made to the passed NodeTemplate
	 @param int operation
	 @param NodeTemplate nodeTemplate
	 @param GuiInterface source - Originator of this interface change.
	 @author WJS
	 */
	public void nodeTemplateChanged(int operation, NodeTemplate nodeTemplate, GuiInterface source) {
		
		// Notify other listeners.
		
	}
	
	/** Called when a change is made to a parameter or the whole parameter set
	 @param int operation
	 @param int parameterNumber
	 @param GuiInterface source - Originator of this interface change.
	 @author WJS
	 */
	public void parameterSetChanged(int operation, int parameterNumber, GuiInterface source) {
		
		switch (operation) {
			case GuiInterface.PARAMETER_ADDED:
				// TODO
				break;
			case GuiInterface.PARAMETER_CHANGED:
				updateParameter(parameterNumber);
				break;
			case GuiInterface.PARAMETER_SET_CHANGED:
				updateParameters();
				break;
			case GuiInterface.PARAMETER_REMOVED:
				// TODO
				break;
		}
		
	}
	
	/** Method for controling model/iterator runs.
	 @param int operation - Run control code.
	 @author WJS
	 */
	public void runControl(int operation) {
		
		
	}
	
	/** Method for reporting status of the model/iterator runs
	 @param int operation - Run status code.
	 @author WJS
	 */
	public void runStatus(int operation) {
		
	}
	
	/** ParamaterPanel.java
	 Internal class for holding all the fields for a parameter.
	 @author WJS
	 */
	private class ParameterPanel extends JPanel {
		
		/** Index of parameter */
		private int                    Index;
		/** Lower bound of value range */
		private JTextField             LowerBoundField;
		/** Name of node */
		private JLabel                 NameLabel;
		/** Button for restoring initial values */
		private JButton                ResetButton;
		/** Button for setting values */
		private JButton                SetButton;
		/** Size of step when progessing though value range */
		private JTextField             StepSizeField;
		/** Upper bound of value range */
		private JTextField             UpperBoundField;
		/** The value of the node */
		private JTextField             ValueField;
		/** The type of progression (linear or log)*/
		private JComboBox              VariationTypeField;
		
		/** Constructor
		 @author WJS
		 */
		public ParameterPanel(int index, String name, float value, float lowerBound, float upperBound, float stepSize, int variationType) {
			
			super();
			setLayout(new BoxLayout(this,BoxLayout.X_AXIS));
			setBackground(Color.lightGray);
			
			Index = index;
			
			// Set up the label that stores the name of the Node. It's size must be fixed
			// Horizontally to make things line up better.
			Dimension labelSize = new Dimension(100,20);
			NameLabel = new JLabel(name);
			NameLabel.setFont(FieldFont);
			NameLabel.setBorder(BorderFactory.createMatteBorder(1,0,1,0,Color.black));
			NameLabel.setHorizontalAlignment(SwingConstants.LEFT);
			NameLabel.setMaximumSize(new Dimension(labelSize.width,60));
			NameLabel.setMinimumSize(labelSize);
			NameLabel.setPreferredSize(labelSize);
			NameLabel.setSize(labelSize);
			add(NameLabel);
			
			// ValueField holds current node value
			ValueField = new JTextField(5);
			ValueField.setFont(FieldFont);
			ValueField.setText(Float.toString(value));
			add(ValueField);
			
			if (!SimpleDisplay) {
				
				// LowerBound field holds smallest allowed value for parameter
				LowerBoundField = new JTextField(5);
				LowerBoundField.setFont(FieldFont);
				LowerBoundField.setText(Float.toString(lowerBound));
				add(LowerBoundField);
				
				// UpperBound field holds largest allowed value for parameter
				UpperBoundField = new JTextField(5);
				UpperBoundField.setFont(FieldFont);
				UpperBoundField.setText(Float.toString(upperBound));
				add(UpperBoundField);
				
				// StepSize used by iterator (?) for steping through values
				// Defaults to 10 for all because ParInsepctor was written that
				// way.
				// TODO - Have ParameterSets store a stepsize?
				StepSizeField = new JTextField(5);
				StepSizeField.setFont(FieldFont);
				StepSizeField.setText(Float.toString(stepSize));
				add(StepSizeField);
				
				// Shows type of progression for steps (linear or logger)
				VariationTypeField = new JComboBox();
				VariationTypeField.setFont(FieldFont);
				VariationTypeField.setBackground(Color.lightGray);
				VariationTypeField.addItem("Linear");
				VariationTypeField.addItem("Log");
				switch (variationType) {
					case Parameter.LINEAR:
						VariationTypeField.setSelectedIndex(0);
						break;
					case Parameter.LOGARITHMIC:
						VariationTypeField.setSelectedIndex(1);
						break;
				}
				add(VariationTypeField);
				
				
				// A button for setting values. Values changes don't take effect until
				// set button is pressed.
				SetButton = new JButton("Set");
				SetButton.setFont(FieldFont);
				SetButton.setOpaque(false);
				SetButton.addActionListener(new ActionListener() {
							public void actionPerformed(ActionEvent ae) {
								try {
									float v = Float.parseFloat(ValueField.getText());
									Parameters.setValue(Index,v);
								} catch (NumberFormatException nfe) {
									ValueField.setText(Float.toString(Parameters.getValue(Index)));
								}
								try {
									float v = Float.parseFloat(LowerBoundField.getText());
									Parameters.setLowerBound(Index,v);
								} catch (NumberFormatException nfe) {
									LowerBoundField.setText(Float.toString(Parameters.getLowerBound(Index)));
								}
								try {
									float v = Float.parseFloat(UpperBoundField.getText());
									Parameters.setUpperBound(Index,v);
								} catch (NumberFormatException nfe) {
									UpperBoundField.setText(Float.toString(Parameters.getUpperBound(Index)));
								}
								try {
									float v = Float.parseFloat(StepSizeField.getText());
									// TODO - Why can't we set step size?
//								Parameters.setStepSize(Index,v);
								} catch (NumberFormatException nfe) {
								}
								Parameters.setVariationMode(Index,VariationTypeField.getSelectedIndex());
								Affector.setParameter(Parameters.getParameter(Index));
								// Notify the MainGui of change so it can be passed on to others
								TheMainGui.parameterSetChanged(GuiInterface.PARAMETER_CHANGED,Index,SaveThis);
							}
						});
				add(SetButton);
				
				// A button for restoring the original values of the fields.
				ResetButton = new JButton("Reset");
				ResetButton.setFont(FieldFont);
				ResetButton.setOpaque(false);
				ResetButton.addActionListener(new ActionListener() {
							public void actionPerformed(ActionEvent ae) {
								ValueField.setText(Float.toString(InitialValues[Index]));
								Parameters.setValue(Index,InitialValues[Index]);
								
								LowerBoundField.setText(Float.toString(InitialLowerBounds[Index]));
								Parameters.setLowerBound(Index,InitialLowerBounds[Index]);
								
								
								UpperBoundField.setText(Float.toString(InitialUpperBounds[Index]));
								Parameters.setUpperBound(Index,InitialUpperBounds[Index]);
								
								StepSizeField.setText(Float.toString(InitialStepSizes[Index]));
								// TODO - Why can't we set step size?
								///							Parameters.setStepSize(Index,InitialStepSizes[Index]);
								
								VariationTypeField.setSelectedIndex(InitialTypes[Index]);
								Parameters.setVariationMode(Index,InitialTypes[Index]);
								
								Affector.setParameter(Parameters.getParameter(Index));
								// Notify the MainGui of change so it can be passed on to others
								TheMainGui.parameterSetChanged(GuiInterface.PARAMETER_CHANGED,Index,SaveThis);
								
							}
						});
				add(ResetButton);
				
			}
			
		}
		
		public int getIndex() {
			
			return(Index);
			
		}
		
		public int getNameLabelWidth() {
			
			return(NameLabel.getWidth());
			
		}
		
		public int getLowerBoundFieldWidth() {
			
			return(LowerBoundField.getWidth());
			
		}
		
		public int getUpperBoundFieldWidth() {
			
			return(UpperBoundField.getWidth());
			
		}
		
		public int getValueFieldWidth() {
			
			return(ValueField.getWidth());
			
		}
		
		public int getVariationTypeFieldWidth() {
			
			return(VariationTypeField.getWidth());
			
		}
		
		public int getStepSizeFieldWidth() {
			
			return(StepSizeField.getWidth());
			
		}
		
		/** Updates the displayed values for this panel based on the current parameter values.
		 @author WJS
		 */
		public void updateDisplays() {
			
			Parameter p = Parameters.getParameter(Index);
			
			ValueField.setText(Float.toString(p.getValue()));
			LowerBoundField.setText(Float.toString(p.lowerBound));
			UpperBoundField.setText(Float.toString(p.upperBound));
			VariationTypeField.setSelectedIndex(p.getVariationMode());
			
		}
		
		/** Updates the values for this panel based on the current parameter values.
		 @author WJS
		 */
		public void updateParameters() {
			
			Parameter p = Affector.getParameter(Index);
			Parameters.setValue(Index,p.getValue());
			ValueField.setText(Float.toString(p.getValue()));
			Parameters.setLowerBound(Index,p.lowerBound);
			LowerBoundField.setText(Float.toString(p.lowerBound));
			Parameters.setUpperBound(Index,p.upperBound);
			UpperBoundField.setText(Float.toString(p.upperBound));
			Parameters.setVariationMode(Index,p.getVariationMode());
			VariationTypeField.setSelectedIndex(p.getVariationMode());
			
		}
		
	}
	
	/** ParameterHeaderPanel.java
	 Private inner class extends ParameterPanel and has its way with it by
	 using it to display column headers. Hope is that the column headers
	 will line up nicely with the ParameterPanel columns. Its works suprisingly
	 well.
	 @author WJS
	 */
	private class ParameterHeaderPanel extends JPanel {
		
		/** Lower bound of value range */
		private JLabel                 LowerBoundFieldTitle;
		/** Name of node */
		private JLabel                 NameTitle;
		/** Size of step when progessing though value range */
		private JLabel                 StepSizeFieldTitle;
		/** Upper bound of value range */
		private JLabel                 UpperBoundFieldTitle;
		/** The value of the node */
		private JLabel                 ValueFieldTitle;
		/** The type of progression (linear or log)*/
		private JLabel                 VariationTypeFieldTitle;
		
		/** Excitingly complex constructor.
		 @author WJS
		 */
		public ParameterHeaderPanel() {
			
			setLayout(new BoxLayout(this,BoxLayout.X_AXIS));
			setBackground(Color.lightGray);
			
			NameTitle = new JLabel("Name");
			NameTitle.setFont(FieldFont);
			add(NameTitle);
			ValueFieldTitle = new JLabel("Value");
			ValueFieldTitle.setFont(FieldFont);
			add(ValueFieldTitle);
			
			if (!SimpleDisplay) {
				LowerBoundFieldTitle = new JLabel("Low");
				LowerBoundFieldTitle.setFont(FieldFont);
				add(LowerBoundFieldTitle);
				UpperBoundFieldTitle = new JLabel("High");
				UpperBoundFieldTitle.setFont(FieldFont);
				add(UpperBoundFieldTitle);
				StepSizeFieldTitle = new JLabel("Step");
				StepSizeFieldTitle.setFont(FieldFont);
				add(StepSizeFieldTitle);
				VariationTypeFieldTitle = new  JLabel("Type");
				VariationTypeFieldTitle.setFont(FieldFont);
				add(VariationTypeFieldTitle);
			}
			
			JPanel filler = new JPanel();
			filler.setOpaque(false);
			add(filler);
			
		}
		
		public void paint(Graphics g) {
			
			// Resize the labels to match the fields
			Dimension d;
			d = NameTitle.getSize();
			d.width = FirstParameterPanel.getNameLabelWidth();
			NameTitle.setSize(d);
			NameTitle.setPreferredSize(d);
			d = ValueFieldTitle.getSize();
			d.width = FirstParameterPanel.getValueFieldWidth();
			ValueFieldTitle.setSize(d);
			ValueFieldTitle.setPreferredSize(d);
			if (!SimpleDisplay) {
				d = LowerBoundFieldTitle.getSize();
				d.width = FirstParameterPanel.getLowerBoundFieldWidth();
				LowerBoundFieldTitle.setSize(d);
				LowerBoundFieldTitle.setPreferredSize(d);
				d = UpperBoundFieldTitle.getSize();
				d.width = FirstParameterPanel.getUpperBoundFieldWidth();
				UpperBoundFieldTitle.setSize(d);
				UpperBoundFieldTitle.setPreferredSize(d);
				d = StepSizeFieldTitle.getSize();
				d.width = FirstParameterPanel.getStepSizeFieldWidth();
				StepSizeFieldTitle.setSize(d);
				StepSizeFieldTitle.setPreferredSize(d);
				d = VariationTypeFieldTitle.getSize();
				d.width = FirstParameterPanel.getVariationTypeFieldWidth();
				VariationTypeFieldTitle.setSize(d);
				VariationTypeFieldTitle.setPreferredSize(d);
			}
			
			super.paint(g);
			
		}
		
	}
	
}