package jalview.gui;

import jalview.jbgui.GUserDefinedColours;
import jalview.datamodel.*;
import jalview.io.*;
import javax.swing.*;
import java.awt.*;
import java.awt.event.*;
import javax.swing.event.*;
import jalview.schemes.*;
import java.io.*;
import java.util.*;


public class UserDefinedColours extends GUserDefinedColours implements ChangeListener
{

  AlignmentPanel ap;
  SequenceGroup seqGroup;
  JButton selectedButton;
  Vector oldColours = new Vector();
  ColourSchemeI oldColourScheme;
  JInternalFrame frame;


  public UserDefinedColours(AlignmentPanel ap, SequenceGroup sg)
  {
    super();
    frame = new JInternalFrame();
    frame.setContentPane(this);
    Desktop.addInternalFrame(frame, "User Defined Colours", 450,530, false );
    if(System.getProperty("os.name").startsWith("Mac"))
      frame.setSize(450,560);

    if(sg!=null)
        frame.setTitle( frame.getTitle()+ " ("+sg.getName()+")");

    colorChooser.getSelectionModel().addChangeListener(this);

    this.ap = ap;
    seqGroup = sg;


      if (seqGroup != null)
        oldColourScheme = seqGroup.cs;
      else
        oldColourScheme = ap.av.getGlobalColourScheme();

      for (int i = 0; i < 20; i++)
        makeButton(ResidueProperties.aa2Triplet.get(ResidueProperties.aa[i]) +
                   "", ResidueProperties.aa[i]);

      makeButton("B", "B");
      makeButton("Z", "Z");
      makeButton("X", "X");
      makeButton("Gap", "'.','-',' '");

      if(jalview.bin.Cache.getProperty("USER_DEFINED_COLOUR")!=null)
      {
        loadColours(jalview.bin.Cache.getProperty("USER_DEFINED_COLOUR"));
      }
  }

 public void stateChanged(ChangeEvent evt)
 {
   if(selectedButton!=null)
     selectedButton.setBackground( colorChooser.getColor() );
 }

  public void colourButtonPressed(MouseEvent e)
  {
     selectedButton = (JButton)e.getSource();
     colorChooser.setColor(selectedButton.getBackground());
  }

  void makeButton(String label, String aa)
  {
    final JButton button = new JButton();
    Color col = Color.white;


    try{
      col = oldColourScheme.findColour(aa, -1);
    }catch(Exception ex){}

    button.setBackground(  col );
    oldColours.addElement( col );
    button.setText( label );
    button.setForeground( col.darker().darker().darker() );
    button.setFont(new java.awt.Font("Verdana", 1, 10));
    button.addMouseListener(new java.awt.event.MouseAdapter()
    {
      public void mouseClicked(MouseEvent e)
      {
        colourButtonPressed(e);
      }
    });

    buttonPanel.add(button, null);
  }


  protected void okButton_actionPerformed(ActionEvent e)
  {
    applyButton_actionPerformed(null);
    try{
      frame.setClosed(true);
    }catch(Exception ex){}
  }

  protected void applyButton_actionPerformed(ActionEvent e)
  {
    Color [] newColours = new Color[24];
    for(int i=0; i<24; i++)
    {
          JButton button = (JButton)buttonPanel.getComponent(i);
          newColours[i] = button.getBackground();
    }

    UserColourScheme ucs = new UserColourScheme(newColours);
    ucs.setThreshold(0);

    if(seqGroup!=null)
    {
      seqGroup.cs = ucs;
      ap.repaint();
    }
    else
      ap.alignFrame.changeColour(ucs);

  }

  protected void loadbutton_actionPerformed(ActionEvent e)
  {
    JalviewFileChooser chooser = new JalviewFileChooser(
      jalview.bin.Cache.getProperty("LAST_DIRECTORY"),
        new String[]{"jc"},
        new String[]{"Jalview User Colours"},
          "Jalview User Colours");
    chooser.setFileView(new jalview.io.JalviewFileView());
    chooser.setDialogTitle("Load colour scheme");
    chooser.setToolTipText("Load");
    int value = chooser.showOpenDialog(this);
    if (value == JalviewFileChooser.APPROVE_OPTION)
    {
      File choice = chooser.getSelectedFile();
      jalview.bin.Cache.setProperty("LAST_DIRECTORY",  choice.getParent());
      jalview.bin.Cache.setProperty("USER_DEFINED_COLOUR", choice.getPath());
      Color[] colors = loadColours(choice.getAbsolutePath());
      for (int i = 0; i < colors.length; i++)
      {
        JButton button = (JButton) buttonPanel.getComponent(i);
        button.setBackground(colors[i]);
      }

    }
  }

  public static UserColourScheme loadDefaultColours()
  {
    if(jalview.bin.Cache.getProperty("USER_DEFINED_COLOUR")!=null)
      return loadDefaultColours(jalview.bin.Cache.getProperty("USER_DEFINED_COLOUR"));
    else
      return null;
  }

  public static UserColourScheme loadDefaultColours(String file)
  {
      UserColourScheme ucs = null;
      Color cols[] = loadColours(file);
      if(cols!=null)
      {
        ucs = new UserColourScheme(cols);
        ucs.setThreshold(0);
      }
      return ucs;
  }

  static Color [] loadColours(String file)
  {
    Color [] newColours=null;
    try
    {
      InputStreamReader in = new InputStreamReader(
          new FileInputStream(file), "UTF-8");

      jalview.binding.JalviewUserColours ucs = new jalview.binding.JalviewUserColours();
      ucs = (jalview.binding.JalviewUserColours)ucs.unmarshal(in);

      newColours = new Color[ucs.getColourCount()];
      for (int i = 0; i < 24; i++)
        newColours[i] = new Color(Integer.parseInt(ucs.getColour(i).getRGB(), 16));

    }
    catch (Exception ex)
    {System.out.println("Error loading UserColourFile "+file);}

    return newColours;
  }

  protected void savebutton_actionPerformed(ActionEvent e)
  {
    JalviewFileChooser chooser = new JalviewFileChooser(
    jalview.bin.Cache.getProperty("LAST_DIRECTORY"),
    new String[]{"jc"},
    new String[]{"Jalview User Colours"},
      "Jalview User Colours");

    chooser.setFileView(new jalview.io.JalviewFileView());
    chooser.setDialogTitle("Save colour scheme");
    chooser.setToolTipText("Save");
    int value = chooser.showSaveDialog(this);
    if (value == JalviewFileChooser.APPROVE_OPTION)
    {
      String choice = chooser.getSelectedFile().getPath();
      jalview.bin.Cache.setProperty("USER_DEFINED_COLOUR", choice);
      jalview.binding.JalviewUserColours ucs = new jalview.binding.JalviewUserColours();

      try{
        PrintWriter out = new PrintWriter(new OutputStreamWriter(
                                     new FileOutputStream(choice), "UTF-8"));
        for(int i=0; i<24; i++)
        {
         JButton button = (JButton)buttonPanel.getComponent(i);
         jalview.binding.Colour col = new jalview.binding.Colour();
         col.setName(button.getText());
         col.setRGB( jalview.util.Format.getHexString( button.getBackground() ));
         ucs.addColour(col);
        }

        ucs.marshal(out);
        out.close();
      }
      catch(Exception ex)
      {ex.printStackTrace();}
    }

  }

  protected void cancelButton_actionPerformed(ActionEvent e)
  {

    Color [] newColours = new Color[24];
    for(int i=0; i<24; i++)
    {
          newColours[i] = (Color)oldColours.elementAt(i);
          buttonPanel.getComponent(i).setBackground(newColours[i]);
    }

    UserColourScheme ucs = new UserColourScheme( newColours );

    if (seqGroup != null)
      seqGroup.cs = ucs;
    else
      ap.av.setGlobalColourScheme(ucs);

    ap.repaint();

    try{
      frame.setClosed(true);
    }catch(Exception ex){}
  }


}