/*
 *                  BioJava development code
 *
 * This code may be freely distributed and modified under the
 * terms of the GNU Lesser General Public Licence.  This should
 * be distributed with the code.  If you do not have a copy,
 * see:
 *
 *      http://www.gnu.org/copyleft/lesser.html
 *
 * Copyright for this code is held jointly by the individual
 * authors.  These should be listed in @author doc comments.
 *
 * For more information on the BioJava project and its aims,
 * or to join the biojava-l mailing list, visit the home page
 * at:
 *
 *      http://www.biojava.org/
 * 
 * Created on Aug 3, 2007
 * 
 */

package org.biojava.nbio.structure.gui;

import org.biojava.nbio.structure.Structure;
import org.biojava.nbio.structure.align.StrucAligParameters;
import org.biojava.nbio.structure.align.StructurePairAligner;
import org.biojava.nbio.structure.align.pairwise.AlternativeAlignment;
import org.biojava.nbio.structure.align.pairwise.FragmentPair;
import org.biojava.nbio.structure.gui.util.color.*;
import org.biojava.nbio.structure.gui.util.color.LinearColorInterpolator.InterpolationDirection;
import org.biojava.nbio.structure.io.PDBFileReader;
import org.biojava.nbio.structure.jama.Matrix;

import javax.swing.*;
import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
import java.awt.*;
import java.awt.color.ColorSpace;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.util.Map;
import java.util.SortedMap;
import java.util.TreeMap;


/** A JPanel that can display the underlying distance matrix 
 * data of the protein structure alignment algorithm. It adds a 
 * JSlider to a JMatrixPanel.
 * 
 * see also JMatrixPanel.
 * 
 */
public class ScaleableMatrixPanel 
extends JPanel 
implements ChangeListener, ActionListener {

	/**
	 * 
	 */
	private static final long serialVersionUID = -8082261434322968652L;

	protected JMatrixPanel mPanel;
	protected JSlider slider;
	protected JScrollPane scroll;
	protected JComboBox coloring;
	
	protected Map<String,ContinuousColorMapper> gradients;
	
	protected static final int SLIDER_STEPS = 8; // Number of minor ticks per unit scaled
	
	
	public static void main(String[] args){

		PDBFileReader pdbr = new PDBFileReader();  
		pdbr.setPath("/tmp/");


		//String pdb1 = "1crl";
		//String pdb2 = "1ede";

		String pdb1 = "1buz";
		String pdb2 = "1ali";            

		//String pdb1 = "5pti";
		//String pdb2 = "5pti";

		// NO NEED TO DO CHANGE ANYTHING BELOW HERE...

		StructurePairAligner sc = new StructurePairAligner();
		StrucAligParameters params = new StrucAligParameters();
		params.setMaxIter(1);
		sc.setParams(params);

		// step1 : read molecules
		try {
			Structure s1 = pdbr.getStructureById(pdb1);
			Structure s2 = pdbr.getStructureById(pdb2);      

			System.out.println("aligning " + pdb1 + " vs. " + pdb2);
			System.out.println(s1);
			System.out.println();
			System.out.println(s2);
			// step 2 : do the calculations
			sc.align(s1,s2);


			ScaleableMatrixPanel smp = new ScaleableMatrixPanel();
			JFrame frame = new JFrame();
			frame.addWindowListener(new WindowAdapter(){
				@Override
				public void windowClosing(WindowEvent e){
					JFrame f = (JFrame) e.getSource();
					f.setVisible(false);
					f.dispose();
				}

						
				
			});
						
			smp.setMatrix(sc.getDistMat());
			smp.setFragmentPairs(sc.getFragmentPairs());
			smp.setAlternativeAligs(sc.getAlignments());
			
			for (int i = 0; i < sc.getAlignments().length; i++) {
				AlternativeAlignment aa =sc.getAlignments()[i];
				System.out.println(aa);
				
			}
			
			frame.getContentPane().add(smp);

			frame.pack();
			frame.setVisible(true);
			
		} catch (Exception e) {
			e.printStackTrace();
		}

	}

	public ScaleableMatrixPanel(){

		mPanel   = new JMatrixPanel();
		Box vBox = Box.createVerticalBox();
		
		Box gradientBox = Box.createHorizontalBox();
		vBox.add(gradientBox);
		gradientBox.add(new JLabel("Coloring:"));
		gradients = createGradients(); //sets gradients
		//Set first gradient
		this.setCellColor(gradients.values().iterator().next());
		
		coloring = new JComboBox(gradients.keySet().toArray(new String[] {}));
		coloring.setRenderer(new GradientRenderer());
		coloring.addActionListener(this);
		coloring.setMaximumSize(new Dimension(1000,30));
		gradientBox.add(coloring);
		
		int RES_MIN  = 0*SLIDER_STEPS;
		int RES_MAX  = 8*SLIDER_STEPS;
		int RES_INIT = 1*SLIDER_STEPS;

		slider = new JSlider(JSlider.HORIZONTAL, RES_MIN,RES_MAX,RES_INIT);
		slider.setInverted(false);
		slider.setPaintTicks(true);
		//slider.setMinorTickSpacing(1);
		slider.setMajorTickSpacing(SLIDER_STEPS);
		slider.setSnapToTicks(false);
		slider.setPaintLabels(false);
		slider.setPaintTrack(true);
		//slider.setLabelTable(slider.createStandardLabels(SLIDER_STEPS));
		slider.addChangeListener(this);
		slider.setPreferredSize(new Dimension(100,slider.getPreferredSize().height));

		vBox.add(slider);

		scroll = new JScrollPane(mPanel);
		scroll.getHorizontalScrollBar().setUnitIncrement(60);
		scroll.getVerticalScrollBar().setUnitIncrement(60);
		scroll.getHorizontalScrollBar().setBlockIncrement(60);
		scroll.getVerticalScrollBar().setBlockIncrement(60);
		vBox.add(scroll);
		this.setPreferredSize(new Dimension(400,400));
		this.add(vBox);


		mPanel.setLayout(new BoxLayout(mPanel,BoxLayout.Y_AXIS));
		this.setLayout(new BoxLayout(this,BoxLayout.Y_AXIS));

	}




	protected static Map<String,ContinuousColorMapper> createGradients() {
		SortedMap<String,ContinuousColorMapper> gradients = new TreeMap<String,ContinuousColorMapper>();
		
		int i = 0; //prepend number, since sorted alphabetically
		ColorSpace hsv = HSVColorSpace.getHSVColorSpace();
		LinearColorInterpolator interp;
		GradientMapper gradient;
		
		/*
		DefaultMatrixMapper defaultMapper = new DefaultMatrixMapper(10., .9f);
		gradients.put((++i)+". Default", defaultMapper);
		defaultMapper = new DefaultMatrixMapper(5., .9f);
		gradients.put((++i)+". Sensitive", defaultMapper);
		
		
		gradients.put((++i)+". Rainbow", GradientMapper.getGradientMapper(GradientMapper.RAINBOW_INTENSITY_GRADIENT, 0, 10));
		gradients.put((++i)+". Rainbow", GradientMapper.getGradientMapper(GradientMapper.RAINBOW_INTENSITY_GRADIENT, 10, 0));
		*/
		

		interp = new LinearColorInterpolator(hsv);
		interp.setInterpolationDirection(0, InterpolationDirection.INNER);
		gradient = new GradientMapper(Color.green, Color.black, hsv);
		gradient.put( -50., new Color(hsv,new float[] {0f, .9f, 0f},1f));
		gradient.put( 10., new Color(hsv,new float[] {1f, .9f, 1f},1f));
		gradient.setInterpolator(interp);
		
		gradients.put((++i)+". -50 to 10", gradient);	
		
		
		
		// Mimic DefaultMapper
		interp = new LinearColorInterpolator(hsv);
		interp.setInterpolationDirection(0, InterpolationDirection.INNER);
		gradient = new GradientMapper(Color.green, Color.black, hsv);
		gradient.put( 0., new Color(hsv,new float[] {1f, .9f, 1f},1f));
		gradient.put(10., new Color(hsv,new float[] {0f, .9f, 0f},1f));
		gradient.setInterpolator(interp);
		
		gradients.put((++i)+". Default", gradient);
		
		// Sensitive DefaultMapper
		interp = new LinearColorInterpolator(hsv);
		interp.setInterpolationDirection(0, InterpolationDirection.INNER);
		gradient = new GradientMapper(Color.green, Color.black, hsv);
		gradient.put( 0., new Color(hsv,new float[] {1f, .9f, 1f},1f));
		gradient.put( 5., new Color(hsv,new float[] {0f, .9f, 0f},1f));
		gradient.setInterpolator(interp);
		
		gradients.put((++i)+". Sensitive", gradient);	
		
		// color [0,1]import java.awt.Color;

		interp = new LinearColorInterpolator(hsv);
		interp.setInterpolationDirection(0, InterpolationDirection.INNER);
		gradient = new GradientMapper(Color.green, Color.black, hsv);
		gradient.put( 0., new Color(hsv,new float[] {1f, .9f, 1f},1f));
		gradient.put( 1., new Color(hsv,new float[] {.2f, .9f, 1f},1f));
		gradient.put( 1+1e-6, Color.white);
		gradient.put(5., Color.black);
		gradient.setInterpolator(interp);
		
		gradients.put((++i)+". Emphasize low", gradient);
		
		
		interp = new LinearColorInterpolator(hsv);
		interp.setInterpolationDirection(0, InterpolationDirection.INNER);
		gradient = new GradientMapper(Color.green, Color.black, hsv);
		gradient.put( 0., new Color(hsv,new float[] {0f, .9f, 0f},1f));
		gradient.put( 100., new Color(hsv,new float[] {1f, .9f, 1f},1f));
		gradient.setInterpolator(interp);
		
		gradients.put((++i)+". 0 to 100", gradient);	
		
		// log color
		interp = new LinearColorInterpolator(hsv);
		interp.setInterpolationDirection(0, InterpolationDirection.INNER);
		gradient = new GradientMapper(Color.red, Color.black, hsv);
		gradient.put( 0., new Color(hsv,new float[] {1f, .9f, 1f},1f));
		gradient.put( 10., new Color(hsv,new float[] {0f, .9f, 0f},1f));
		gradient.setInterpolator(interp);
		
		ContinuousColorMapper logGradient = new LogColorMapper(gradient,2);
		gradients.put((++i)+". Logorithmic", logGradient);	
		
		// sqrt color
		interp = new LinearColorInterpolator(hsv);
		interp.setInterpolationDirection(0, InterpolationDirection.INNER);
		gradient = new GradientMapper(Color.red, Color.black, hsv);
		gradient.put( 0., new Color(hsv,new float[] {1f, .9f, 1f},1f));
		gradient.put( 4., new Color(hsv,new float[] {0f, .9f, 0f},1f));
		gradient.setInterpolator(interp);
		
		ContinuousColorMapper sqrtGradient = new SqrtColorMapper(gradient);
		gradients.put((++i)+". Square Root", sqrtGradient);
		
		GradientMapper black = new GradientMapper(Color.BLACK, Color.BLACK);
		gradients.put((++i)+". Black", black);

		return gradients;
	}

	@Override
	public void stateChanged(ChangeEvent e) {
		
		JSlider source = (JSlider)e.getSource();
		
		if ( source.getValueIsAdjusting()) {
			//return;
		}
		
		//System.out.println("Changed scale to "+source.getValue());
		mPanel.setScale((float)source.getValue()/SLIDER_STEPS);
		
		scroll.repaint();
		scroll.updateUI();
	}

	public Matrix getMatrix() {
		return mPanel.getMatrix();
	}

	public void setMatrix(Matrix matrix) {
		mPanel.setMatrix(matrix);
	
		
	}

	public JMatrixPanel getMatrixPanel(){
		return mPanel;
	}
	
	public FragmentPair[] getFragmentPairs(){
		return mPanel.getFragmentPairs();
	}
	public void setFragmentPairs(FragmentPair[] pairs){
		mPanel.setFragmentPairs(pairs);
	}

	public AlternativeAlignment[] getAlternativeAligs() {
		return mPanel.getAlternativeAligs();
	}



	public void setAlternativeAligs(AlternativeAlignment[] aligs) {
		mPanel.setAlternativeAligs(aligs);
	}
	
	public int getSelectedAlignmentPos() {
		return mPanel.getSelectedAlignmentPos();
	}

	public void setSelectedAlignmentPos(int selectedAlignmentPos) {
		mPanel.setSelectedAlignmentPos(selectedAlignmentPos);
	}

	/**
	 * @return the color mapping of the JMatrixPanel
	 */
	public ContinuousColorMapper getCellColor() {
		return mPanel.getCellColor();
	}

	/**
	 * @param cellColor the color mapping of the JMatrixPanel
	 */
	public void setCellColor(ContinuousColorMapper cellColor) {
		mPanel.setCellColor(cellColor);
	}

	/**
	 * A renderer for the the gradient dropdown menu at the top of scaleableMatrixPanes.
	 * Knows how to draw a gradient and nicely label it.
	 * 
	 * @author Spencer Bliven
	 *
	 */
	protected class GradientRenderer extends JPanel
	implements ListCellRenderer {

		private static final long serialVersionUID = -2000575579184232365L;
		private int min,max;
		JLabel title;
		JPanel gradientContainer;
		
		public GradientRenderer() {
			this.setPreferredSize(new Dimension(100,25));
			this.setLayout(new BorderLayout());
			this.min = -1;
			this.max = 10;
			
			JPanel gradientBounds = new JPanel();
			gradientBounds.setLayout(new BorderLayout());
			//gradientBounds.setBorder(BorderFactory.createLineBorder(Color.GRAY));
			gradientBounds.add(new JLabel(Integer.toString(min)),BorderLayout.WEST);
			
			gradientContainer = new JPanel();
			gradientContainer.setLayout(new BorderLayout());
			gradientContainer.setOpaque(false);
			gradientContainer.add(new JLabel("<No gradient>"),BorderLayout.CENTER);
			//gradientContainer.setMinimumSize(new Dimension(50,20));
			//gradientContainer.setBorder(BorderFactory.createLineBorder(Color.DARK_GRAY));
			//gradientBounds.setPreferredSize(new Dimension(200,25));
			gradientBounds.add(gradientContainer,BorderLayout.CENTER);
			gradientBounds.setOpaque(false);
			
			gradientBounds.add(new JLabel(Integer.toString(max)),BorderLayout.EAST);
			
			this.add(gradientBounds, BorderLayout.EAST);
			
			this.title = new JLabel("No gradient");
			//this.title.setOpaque(true);
			this.title.setHorizontalAlignment(JLabel.CENTER);
			this.title.setVerticalAlignment(JLabel.CENTER);
			this.add(this.title,BorderLayout.CENTER);
		}

		/*
		 * This method finds the image and text corresponding
		 * to the selected value and returns the label, set up
		 * to display the text and image.
		 */
		@Override
		public Component getListCellRendererComponent(
				JList list,
				Object value,
				int index,
				boolean isSelected,
				boolean cellHasFocus) {
			//Get the selected index. (The index param isn't
			//always valid, so just use the value.)
			String gradientLabel = (String)value;
			
			if (isSelected) {
				setBackground(list.getSelectionBackground());
				setForeground(list.getSelectionForeground());
			} else {
				setBackground(list.getBackground());
				setForeground(list.getForeground());
			}
			
			//Set the icon and text.  If icon was null, say so.
			GradientPanel gradPanel = new GradientPanel(gradients.get(gradientLabel),min,max);
			gradPanel.setPreferredSize(new Dimension(100,20));
			//gradPanel.setBorder(BorderFactory.createLineBorder(Color.cyan));
			gradientContainer.removeAll();
			gradientContainer.add(gradPanel,BorderLayout.CENTER);

			title.setText(gradientLabel);

			this.validate();
			
			return this;
		}
	}

	/**
	 * @param e
	 * @see java.awt.event.ActionListener#actionPerformed(java.awt.event.ActionEvent)
	 */
	@Override
	public void actionPerformed(ActionEvent e) {
        JComboBox cb = (JComboBox)e.getSource(); // == coloring
        String gradientName = (String)cb.getSelectedItem();
        ContinuousColorMapper gradient = gradients.get(gradientName);
        assert(gradient != null);
        this.setCellColor(gradient);
        this.repaint();
	}

	
}
