package mygraphs;

import java.awt.*;
import java.awt.event.*;
import java.awt.font.*;
import java.awt.geom.*;
import java.util.*;
import javax.swing.*;

import genegui.*;
import main.*;
import parameters.*;
import stat.*;


/** CamGraph2.java
 A re-write of the original CamGraph to provide for cleaner visuals and to allow
 integration with the IteratorViewer.
 @author WJS
 */
public class CamGraph2 extends JInternalFrame {
	
	private static final int               INNER_CIRCLE = 100;
	private static final int               OUTER_CIRCLE = 800;
	private static final int               RAY_LENGTH = 420;
	
	
	/** */
	private Color[]                        GraphColors;
	/** */
	private String[]                       GraphNames;
	/** */
	private Polygon[]                      GraphPolygons;
	/** */
	private float[]                        GraphScores;
	/** */
	private boolean[]                      GraphVisible;
	/** */
	private	JToggleButton                  HoldButton;
	/** */
	private Font                           LabelFont = new Font("Arial",Font.BOLD,14);
	/** */
	private JWindow                        LabelWindow;
	/** */
	private JPanel                         MainPanel;
	/** */
  private JComboBox                      ParametersBox;
	/** */
	private String[]                       ParameterNames;
	/** */
	private Rectangle[]                    ParameterNameAreas;
	/** */
	private JToggleButton                  PlotAllButton;
	/** */
	private JTextField                     ScoreJTextField;
	/** */
	private PaintPanel                     ThePaintPanel;
	/** */
	private SuccessList                    TheSuccessList;
	/** */
	private int                            ViewIndex = 0;
	/** */
	private int[]                          XRays;
	/** */
	private int[]                          YRays;
	/** */
	private int[]                          XNames;
	/** */
	private int[]                          YNames;
	
	public CamGraph2(ParameterSet protoTypeParameterSet) {
		
		super("CamGraph",true,true,true,false);
		
		MainPanel = new JPanel();
		setContentPane(MainPanel);
		
		MainPanel.setLayout(new BoxLayout(MainPanel,BoxLayout.Y_AXIS));
		MainPanel.setBackground(Color.lightGray);
		
		// Set up the paint panel used for the display
		ThePaintPanel = new CamGraph2.PaintPanel();
		ThePaintPanel.setBorder(BorderFactory.createMatteBorder(0,0,1,0,Color.black));
		MainPanel.add(ThePaintPanel);
		ThePaintPanel.paintOffScreen(true,false,0);
		
		// Setup the control area
		JPanel controls = new JPanel();
		controls.setBackground(Color.lightGray);
		controls.setOpaque(true);
		controls.setLayout(new BoxLayout(controls,BoxLayout.X_AXIS));
		// Fix the size of the controlspanel so it forces the ThePaintPanel
		// to do all the resizinf when resizing.
		Dimension controlDim = new Dimension(400,60);
		controls.setSize(controlDim);
		controls.setMaximumSize(controlDim);
		controls.setMinimumSize(controlDim);
		controls.setPreferredSize(controlDim);
		MainPanel.add(controls);
		
		// Left Side
		JPanel leftSide = new JPanel();
		leftSide.setBackground(Color.lightGray);
		leftSide.setOpaque(true);
		leftSide.setLayout(new BoxLayout(leftSide,BoxLayout.Y_AXIS));
		controls.add(leftSide);
		
		// Navigation buttons and results list
		JPanel p = new JPanel();
		p.setBackground(Color.lightGray);
		p.setOpaque(true);
		p.setLayout(new BoxLayout(p,BoxLayout.X_AXIS));
		leftSide.add(p);
		JButton LeftLeftArrow = new JButton(new ImageIcon("icons/ArrowLeftLeft13.gif"));
		LeftLeftArrow.setContentAreaFilled(false);
		LeftLeftArrow.setBorderPainted(false);
		LeftLeftArrow.setFocusPainted(false);
		LeftLeftArrow.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent ae) {
						ViewIndex=0;
						setViewIndex(ViewIndex);
					}
				});
		p.add(LeftLeftArrow);
		JButton LeftArrow = new JButton(new ImageIcon("icons/ArrowLeft13.gif"));
		LeftArrow.setContentAreaFilled(false);
		LeftArrow.setBorderPainted(false);
		LeftArrow.setFocusPainted(false);
		LeftArrow.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent ae) {
						ViewIndex--;
						if (ViewIndex<0) ViewIndex=0;
						setViewIndex(ViewIndex);
					}
				});
		p.add(LeftArrow);
		
		// The combobox for displaying results to be viewed
		ParametersBox = new JComboBox();
		ParametersBox.setOpaque(false);
		ParametersBox.setFont(new Font("Arial",Font.BOLD,10));
		ParametersBox.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent ae) {
						ViewIndex = ParametersBox.getSelectedIndex();
						if (ViewIndex<0) ViewIndex=0;
						setViewIndex(ViewIndex);
					}
				});
		p.add(ParametersBox);
		
		JButton RightArrow = new JButton(new ImageIcon("icons/ArrowRight13.gif"));
		RightArrow.setContentAreaFilled(false);
		RightArrow.setBorderPainted(false);
		RightArrow.setFocusPainted(false);
		RightArrow.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent ae) {
						ViewIndex++;
						if (ViewIndex==GraphPolygons.length) ViewIndex--;
						setViewIndex(ViewIndex);
					}
				});
		p.add(RightArrow);
		JButton RightRightArrow = new JButton(new ImageIcon("icons/ArrowRightRight13.gif"));
		RightRightArrow.setContentAreaFilled(false);
		RightRightArrow.setBorderPainted(false);
		RightRightArrow.setFocusPainted(false);
		RightRightArrow.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent ae) {
						ViewIndex=GraphPolygons.length-1;
						setViewIndex(ViewIndex);
					}
				});
		p.add(RightRightArrow);
		
		// Control buttons
		p = new JPanel();
		p.setBackground(Color.lightGray);
		p.setOpaque(true);
		p.setLayout(new BoxLayout(p,BoxLayout.X_AXIS));
		leftSide.add(p);
		// Hold button tells camgraph to keep this parameterset displayed
		HoldButton = new JToggleButton("Hold");
		HoldButton.setOpaque(false);
		HoldButton.setFont(new Font("Arial",Font.BOLD,10));
		HoldButton.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent ae) {
						GraphVisible[ViewIndex] = HoldButton.isSelected();
					}
				});
		p.add(HoldButton);
		// Plot All button causes all parameter sets to be displayed
		PlotAllButton = new JToggleButton("Plot All");
		PlotAllButton.setOpaque(false);
		PlotAllButton.setFont(new Font("Arial",Font.BOLD,10));
		PlotAllButton.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent ae) {
						if (PlotAllButton.isSelected()) plotAll();
						else {
							clearGraphs();
							setViewIndex(ViewIndex);
						}
					}
				});
		p.add(PlotAllButton);
		JPanel filler = new JPanel();
		filler.setOpaque(false);
		// Clear button erases all parameter sets being displayed
		JButton ClearButton = new JButton("Clear");
		ClearButton.setOpaque(false);
		ClearButton.setFont(new Font("Arial",Font.BOLD,10));
		ClearButton.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent ae) {
						clearGraphs();
					}
				});
		p.add(ClearButton);
		
		// Right Side
		JPanel rightSide = new JPanel();
		rightSide.setBackground(Color.lightGray);
		rightSide.setOpaque(true);
		rightSide.setLayout(new BoxLayout(rightSide,BoxLayout.Y_AXIS));
		controls.add(rightSide);
		
		JLabel l;
		Dimension labelDim = new Dimension(50,14);
		
		// Make panel for displaying score
		p = new JPanel();
		p.setBackground(Color.lightGray);
		p.setOpaque(true);
		p.setLayout(new BoxLayout(p,BoxLayout.X_AXIS));
		l = new JLabel("Score: ");
		l.setHorizontalAlignment(SwingConstants.RIGHT);
		l.setSize(labelDim);
		l.setMinimumSize(labelDim);
		l.setPreferredSize(labelDim);
		l.setFont(new Font("Arial",Font.BOLD,10));
		p.add(l);
		// JTextField for displaying the parameter set score
		ScoreJTextField = new JTextField();
		ScoreJTextField.setFont(new Font("Arial",Font.PLAIN,10));
		ScoreJTextField.setEditable(false);
		p.add(ScoreJTextField);
		
		rightSide.add(p);
		
		// Make panels for displaying threshold fields
		p = new JPanel();
		p.setBackground(Color.lightGray);
		p.setOpaque(true);
		p.setLayout(new BoxLayout(p,BoxLayout.X_AXIS));
		l = new JLabel("High: ");
		l.setHorizontalAlignment(SwingConstants.RIGHT);
		l.setSize(labelDim);
		l.setMinimumSize(labelDim);
		l.setPreferredSize(labelDim);
		l.setFont(new Font("Arial",Font.BOLD,10));
		p.add(l);
		// JTextField for high Threshold
		JTextField HighJTextField = new JTextField();
		HighJTextField.setFont(new Font("Arial",Font.PLAIN,10));
		p.add(HighJTextField);
		rightSide.add(p);
		p = new JPanel();
		p.setBackground(Color.lightGray);
		p.setOpaque(true);
		p.setLayout(new BoxLayout(p,BoxLayout.X_AXIS));
		l = new JLabel("Low: ");
		l.setSize(labelDim);
		l.setHorizontalAlignment(SwingConstants.RIGHT);
		l.setMinimumSize(labelDim);
		l.setPreferredSize(labelDim);
		l.setFont(new Font("Arial",Font.BOLD,10));
		p.add(l);
		// JTextField for high Threshold
		JTextField LowJTextField = new JTextField();
		LowJTextField.setFont(new Font("Arial",Font.PLAIN,10));
		p.add(LowJTextField);
		rightSide.add(p);
		
		filler = new JPanel();
		filler.setOpaque(false);
		
		pack();
		
		setVisible(true);
		
		// If a prototype ParameterSet was passed, then build the empty graph
		// and labels.
		if (protoTypeParameterSet!=null)
			setLinesAndLabels(protoTypeParameterSet);
		
		LabelWindow = new JWindow();
		
	}
	
	/**
	 @param ParameterSet ps
	 @author WJS
	 */
	public void addParameterSet(ParameterSet ps, String name, Color color, float score) {
		
		// Check to see if this is the first parameter set
		if (XRays==null) {
			setLinesAndLabels(ps);
			ThePaintPanel.paintOffScreen(true,false,0);
		}
		// Go through the parameter set and build the polygon representation
		Polygon p = new Polygon();
		float l;
		float u;
		float v;
		double z=0;
		int m;
		int n = ps.getNumParams();
		double wedge = 2*Math.PI/(double)n;
		double theta = 0;
		double r = (OUTER_CIRCLE-INNER_CIRCLE)/2d;
		for (int i=0;i<n;i++) {
			v = ps.getValue(i);
			l = ps.getLowerBound(i);
			u = ps.getUpperBound(i);
			m = ps.getVariationMode(i);
			switch (m) {
				case Parameter.LINEAR:
					z = r*(v-l)/(u-l)+INNER_CIRCLE/2d;
					break;
				case Parameter.LOGARITHMIC:
					z = r*(Math.log(v)-Math.log(l))/(Math.log(u)-Math.log(l))+INNER_CIRCLE/2d;
					break;
					
			}
			p.addPoint((int)(z*Math.cos(theta)),(int)(z*Math.sin(theta)));
			theta += wedge;
		}
		
		// Add the polygon to the list and repaint it
		int index;
		if (GraphPolygons!=null) {
			Polygon[] temp = new Polygon[GraphPolygons.length+1];
			System.arraycopy(GraphPolygons,0,temp,0,temp.length-1);
			temp[temp.length-1] = p;
			GraphPolygons = temp;
			boolean[] temp2 = new boolean[GraphVisible.length+1];
			System.arraycopy(GraphVisible,0,temp2,0,temp2.length-1);
			temp2[temp2.length-1] = PlotAllButton.isSelected();
			GraphVisible = temp2;
			Color[] temp3 = new Color[GraphColors.length+1];
			System.arraycopy(GraphColors,0,temp3,0,temp3.length-1);
			temp3[temp3.length-1] = color;
			GraphColors = temp3;
			index = temp3.length-1;
			String[] temp4 = new String[GraphNames.length+1];
			System.arraycopy(GraphNames,0,temp4,0,temp4.length-1);
			temp4[temp4.length-1] = name;
			GraphNames = temp4;
			float[] temp5 = new float[GraphScores.length+1];
			System.arraycopy(GraphScores,0,temp5,0,temp5.length-1);
			temp5[temp5.length-1] = score;
			GraphScores = temp5;
			
			ViewIndex = temp.length-1;
			
		} else {
			GraphPolygons = new Polygon[1];
			GraphPolygons[0] = p;
			GraphVisible = new boolean[1];
			GraphVisible[0] = PlotAllButton.isSelected();
			GraphColors = new Color[1];
			GraphColors[0] = color;
			GraphNames = new String[1];
			GraphNames[0] = name;
			GraphScores = new float[1];
			GraphScores[0] = score;
			
			ViewIndex = 0;
			
		}
		
		ParametersBox.addItem(GraphNames[ViewIndex]);
		setViewIndex(ViewIndex);
		
		repaint();
		
	}
	
	/**
	 @author WJS
	 */
	public void clearGraphs() {
		
		if (GraphVisible==null) return;
		
		for (int i=0;i<GraphVisible.length;i++)
			GraphVisible[i] = false;
		PlotAllButton.setSelected(false);
		setViewIndex(ViewIndex);
		
	}
	
	/**
	 @author WJS
	 */
	public void plotAll() {
		
		if (GraphVisible==null) return;
		
		for (int i=0;i<GraphVisible.length;i++)
			GraphVisible[i] = true;
		setViewIndex(ViewIndex);
		
	}
	
	/**
	 @author WJS
	 */
	public void setLinesAndLabels(ParameterSet ps) {
		
		int n = ps.getNumParams();
		
		// Get the parameter names
		ParameterNames = new String[n];
		ParameterNameAreas = new Rectangle[n];
		for (int i=0;i<n;i++) {
			if (ps.getVariationMode(i)==Parameter.LOGARITHMIC)
			  ParameterNames[i] = "ln "+ps.getName(i);
			else
			  ParameterNames[i] = ps.getName(i);
		}
		
		XRays = new int[n];
		YRays = new int[n];
		double wedge = 2*Math.PI/(double)n;
		double theta = 0;
		for (int i=0;i<n;i++) {
			XRays[i] = (int)(RAY_LENGTH*Math.cos(theta));
			YRays[i] = (int)(RAY_LENGTH*Math.sin(theta));
			theta += wedge;
		}
		
	}
	
	/**
	 @author WJS
	 */
	public void setSuccessList(SuccessList slist) {
		
		TheSuccessList = slist;
		
		// If the slist already has elements, then go through it and add them
		for (int i=0;i<TheSuccessList.size();i++) {
			ParameterSet ps = TheSuccessList.getParameterSet(i);
		  String name = TheSuccessList.getName(i);
			float score = TheSuccessList.getScore(i);
			addParameterSet(ps,name,Color.red,score);
		}
		
	}
	
	/**
	 @author WJS
	 */
	public void setViewIndex(int index) {
		
		ViewIndex = index;
		ThePaintPanel.paintOffScreen(true,false,index);
	  ThePaintPanel.paintOffScreen(false,true,index);
		ThePaintPanel.repaint();
		HoldButton.setSelected(GraphVisible[index]);
		ParametersBox.setSelectedIndex(index);
		ScoreJTextField.setText(Float.toString(GraphScores[index]));
		
  }
	
	/**
	 @author WJS
	 */
	public void successListAppended() {
		
		ParameterSet ps = TheSuccessList.getParameterSet(TheSuccessList.size()-1);
		String name = TheSuccessList.getName(TheSuccessList.size()-1);
		float score = TheSuccessList.getScore(TheSuccessList.size()-1);
		
		addParameterSet(ps,name,Color.red,score);
		
	}
	
	/** CamGraph2.PaintPanel.java
	 @author WJS
	 */
	public class PaintPanel extends JPanel {
		
		/** */
		private Dimension                     DEFAULT_SIZE = new Dimension(400,400);
		/** */
		private Image                         OffScreenImage;
		/** */
		private boolean                       Redrawing = false;
		
		/**
		 @author WJS
		 */
		public PaintPanel() {
			
			super();
			setBackground(Color.white);
			setSize(DEFAULT_SIZE);
			setMinimumSize(DEFAULT_SIZE);
			setPreferredSize(DEFAULT_SIZE);
			
			addMouseListener(new MouseAdapter() {
						public void mousePressed(MouseEvent me) {
							if (ParameterNameAreas!=null) {
								Point p = me.getPoint();
			          double scale = (double)Math.min(getWidth(),getHeight())/1000d;
								p.x = (int)(p.x/scale)-500;
								p.y = (int)(p.y/scale)-500;
								for (int i=0;i<ParameterNameAreas.length;i++) {
									if (ParameterNameAreas[i].contains(p)) {
										JLabel l = new JLabel(ParameterNames[i]);
										l.setBackground(Color.yellow);
										l.setOpaque(true);
										l.setBorder(BorderFactory.createLineBorder(Color.black,1));
										LabelWindow.setContentPane(l);
										p = getLocationOnScreen();
										p.x += me.getX();
										p.y += me.getY()-14;
										LabelWindow.setLocation(p);
										LabelWindow.show(true);
										LabelWindow.pack();
										break;
									}
								}
							}
						}
						public void mouseReleased(MouseEvent me) {
							LabelWindow.show(false);
						}
					});
			
		}
		
		/**
		@author WJS
		 */
		public void paint(Graphics g0) {
			
			if (OffScreenImage==null||(OffScreenImage.getWidth(this)!=getWidth())||(OffScreenImage.getHeight(this)!=getHeight())) {
				OffScreenImage = createImage(getWidth(),getHeight());
				paintOffScreen(true,true,0);
			}
			
			// This line will cause the paint thread to yield if the paintOffScreen
			// method is running. It is crucial to making sure painting doesn't
			// mess up especially when the viewer is being dragged around.
			if (Redrawing) Thread.yield();
			
			g0.drawImage(OffScreenImage,0,0,this);
			
		}
		
		/**
		 @author WJS
		 */
		public void paintOffScreen(boolean paintGrid, boolean paintAll, int index) {
			
			// This will happen if the paint panel isn't visible
			if (OffScreenImage==null) return;
			
			Redrawing = true;
			
			Graphics2D g = (Graphics2D)OffScreenImage.getGraphics();
			g.setRenderingHint(RenderingHints.KEY_ANTIALIASING,RenderingHints.VALUE_ANTIALIAS_ON);
			
			if (paintGrid) {
				g.setColor(Color.white);
				g.fillRect(0,0,getWidth(),getHeight());
				
			}
			
			// If the text labels haven't been positioned yet. Do it.
			// Do this here so we can get the proper graphics context for
			// determining text sizes.
			if (XNames==null) {
				int n = XRays.length;
				XNames = new int[n];
				YNames = new int[n];
				double wedge = 2*Math.PI/(double)n;
				double theta = 0;
				FontRenderContext frc = ((Graphics2D)OffScreenImage.getGraphics()).getFontRenderContext();
				TextLayout tl;
				double a;
				double b;
				Rectangle r;
				for (int i=0;i<n;i++) {
					tl = new TextLayout(ParameterNames[i],LabelFont,frc);
					a = tl.getAdvance();
					a = (a/2) - Math.cos(theta)*(a/2);
					b = tl.getAscent()+tl.getDescent();
					b = (b/2) + Math.sin(theta)*(b/2);
					XNames[i] = (int)((RAY_LENGTH+10+a)*Math.cos(theta));
					YNames[i] = (int)((RAY_LENGTH+10+b)*Math.sin(theta));
					r = new Rectangle();
					r.width = (int)tl.getAdvance();
					r.height=(int)(tl.getAscent()+tl.getDescent());
					r.x = (XNames[i]);
					r.y = (YNames[i]-r.height);
					ParameterNameAreas[i] = r;
					theta += wedge;
				}
			}
			
			double scale = (double)Math.min(getWidth(),getHeight())/1000d;
			g.scale(scale,scale);
			g.translate(500,500);
			
			// Paint the rays, borders and labels if wanted
			if (paintGrid) {
				g.setColor(Color.lightGray);
				g.drawOval(-INNER_CIRCLE/2,-INNER_CIRCLE/2,INNER_CIRCLE,INNER_CIRCLE);
				g.drawOval(-OUTER_CIRCLE/2,-OUTER_CIRCLE/2,OUTER_CIRCLE,OUTER_CIRCLE);
				
				// Indicates that model has not been set yet.
				if (XRays==null) return;
				
				
				for (int i=0;i<XRays.length;i++) {
					g.drawLine(0,0,XRays[i],YRays[i]);
				}
				
				// Paint the parameter names
				if (ParameterNames!=null) {
					g.setFont(LabelFont);
					g.setColor(Color.black);
					for (int i=0;i<ParameterNames.length;i++)
						g.drawString(ParameterNames[i],XNames[i],YNames[i]);
				}
			}
			
			// Paint all the polygons or just one
			if (GraphPolygons!=null) {
				if (paintAll) {
					for (int i=0;i<GraphPolygons.length;i++) {
						if ((GraphVisible[i])||(i==ViewIndex)) {
							g.setColor(GraphColors[i]);
							g.draw(GraphPolygons[i]);
						}
					}
				} else {
					g.setColor(GraphColors[index]);
					g.draw(GraphPolygons[index]);
				}
			}
			
			Redrawing = false;
			
			g.dispose();
			
		}
		
	}
	
}