Check if colos are null
[jalview.git] / src / jalview / gui / UserDefinedColours.java
index 1fcc923..5fdcdf8 100755 (executable)
@@ -2,6 +2,7 @@ package jalview.gui;
 \r
 import jalview.jbgui.GUserDefinedColours;\r
 import jalview.datamodel.*;\r
+import jalview.io.*;\r
 import javax.swing.*;\r
 import java.awt.*;\r
 import java.awt.event.*;\r
@@ -18,24 +19,46 @@ public class UserDefinedColours extends GUserDefinedColours implements ChangeLis
   SequenceGroup seqGroup;\r
   JButton selectedButton;\r
   Vector oldColours = new Vector();\r
+  ColourSchemeI oldColourScheme;\r
   JInternalFrame frame;\r
 \r
-  public UserDefinedColours(JInternalFrame f, AlignmentPanel ap, SequenceGroup sg)\r
+\r
+  public UserDefinedColours(AlignmentPanel ap, SequenceGroup sg)\r
   {\r
     super();\r
+    frame = new JInternalFrame();\r
+    frame.setContentPane(this);\r
+    Desktop.addInternalFrame(frame, "User Defined Colours", 450,530, false );\r
+    if(System.getProperty("os.name").startsWith("Mac"))\r
+      frame.setSize(450,560);\r
+\r
+    if(sg!=null)\r
+        frame.setTitle( frame.getTitle()+ " ("+sg.getName()+")");\r
+\r
     colorChooser.getSelectionModel().addChangeListener(this);\r
+\r
     this.ap = ap;\r
-    frame = f;\r
-   seqGroup = sg;\r
+    seqGroup = sg;\r
+\r
+\r
+      if (seqGroup != null)\r
+        oldColourScheme = seqGroup.cs;\r
+      else\r
+        oldColourScheme = ap.av.getGlobalColourScheme();\r
 \r
-    for(int i=0; i<20; i++)\r
-      makeButton(ResidueProperties.aa2Triplet.get( ResidueProperties.aa[i] )+"", ResidueProperties.aa[i]);\r
+      for (int i = 0; i < 20; i++)\r
+        makeButton(ResidueProperties.aa2Triplet.get(ResidueProperties.aa[i]) +\r
+                   "", ResidueProperties.aa[i]);\r
 \r
-    makeButton("B","B");\r
-    makeButton("Z","Z");\r
-    makeButton("X","X");\r
-    makeButton("'.','-',' '", "-");\r
+      makeButton("B", "B");\r
+      makeButton("Z", "Z");\r
+      makeButton("X", "X");\r
+      makeButton("Gap", "'.','-',' '");\r
 \r
+      if(jalview.bin.Cache.getProperty("USER_DEFINED_COLOUR")!=null)\r
+      {\r
+        loadColours(jalview.bin.Cache.getProperty("USER_DEFINED_COLOUR"));\r
+      }\r
   }\r
 \r
  public void stateChanged(ChangeEvent evt)\r
@@ -54,18 +77,16 @@ public class UserDefinedColours extends GUserDefinedColours implements ChangeLis
   {\r
     final JButton button = new JButton();\r
     Color col = Color.white;\r
-    ColourSchemeI cs = null;\r
-    if(seqGroup!=null)\r
-      cs = seqGroup.cs;\r
-    else\r
-      cs = ap.av.getGlobalColourScheme();\r
 \r
-    if(cs!=null)\r
-      col = cs.findColour(null, aa, -1, null);\r
+\r
+    try{\r
+      col = oldColourScheme.findColour(aa, -1);\r
+    }catch(Exception ex){}\r
 \r
     button.setBackground(  col );\r
     oldColours.addElement( col );\r
     button.setText( label );\r
+    button.setForeground( col.darker().darker().darker() );\r
     button.setFont(new java.awt.Font("Verdana", 1, 10));\r
     button.addMouseListener(new java.awt.event.MouseAdapter()\r
     {\r
@@ -75,7 +96,7 @@ public class UserDefinedColours extends GUserDefinedColours implements ChangeLis
       }\r
     });\r
 \r
-    buttonPanel.add(button);\r
+    buttonPanel.add(button, null);\r
   }\r
 \r
 \r
@@ -89,7 +110,6 @@ public class UserDefinedColours extends GUserDefinedColours implements ChangeLis
 \r
   protected void applyButton_actionPerformed(ActionEvent e)\r
   {\r
-    UserColourScheme ucs = new UserColourScheme();\r
     Color [] newColours = new Color[24];\r
     for(int i=0; i<24; i++)\r
     {\r
@@ -97,63 +117,118 @@ public class UserDefinedColours extends GUserDefinedColours implements ChangeLis
           newColours[i] = button.getBackground();\r
     }\r
 \r
+    UserColourScheme ucs = new UserColourScheme(newColours);\r
     ucs.setThreshold(0);\r
-    ucs.setColourScheme( newColours );\r
 \r
     if(seqGroup!=null)\r
+    {\r
       seqGroup.cs = ucs;\r
+      ap.repaint();\r
+    }\r
     else\r
-      ap.av.setGlobalColourScheme(ucs);\r
+      ap.alignFrame.changeColour(ucs);\r
 \r
-    ap.RefreshPanels();\r
   }\r
 \r
   protected void loadbutton_actionPerformed(ActionEvent e)\r
   {\r
-    JFileChooser chooser = new JFileChooser(jalview.bin.Cache.LAST_DIRECTORY);\r
+    JalviewFileChooser chooser = new JalviewFileChooser(\r
+      jalview.bin.Cache.getProperty("LAST_DIRECTORY"),\r
+        new String[]{"jc"},\r
+        new String[]{"Jalview User Colours"},\r
+          "Jalview User Colours");\r
+    chooser.setFileView(new jalview.io.JalviewFileView());\r
     chooser.setDialogTitle("Load colour scheme");\r
     chooser.setToolTipText("Load");\r
     int value = chooser.showOpenDialog(this);\r
-    if (value == JFileChooser.APPROVE_OPTION)\r
+    if (value == JalviewFileChooser.APPROVE_OPTION)\r
     {\r
-      String choice = chooser.getSelectedFile().getPath();\r
-      jalview.bin.Cache.LAST_DIRECTORY = choice;\r
+      File choice = chooser.getSelectedFile();\r
+      jalview.bin.Cache.setProperty("LAST_DIRECTORY",  choice.getParent());\r
+      jalview.bin.Cache.setProperty("USER_DEFINED_COLOUR", choice.getPath());\r
+      Color[] colors = loadColours(choice.getAbsolutePath());\r
+      for (int i = 0; i < colors.length; i++)\r
+      {\r
+        JButton button = (JButton) buttonPanel.getComponent(i);\r
+        button.setBackground(colors[i]);\r
+      }\r
 \r
-      try{\r
-        BufferedReader in = new BufferedReader(new FileReader(choice));\r
-        for(int i=0; i<24; i++)\r
-        {\r
-          JButton button = (JButton)buttonPanel.getComponent(i);\r
-          Color c = new Color(Integer.parseInt(in.readLine()));\r
-          button.setBackground(c);\r
-        }\r
+    }\r
+  }\r
+\r
+  public static UserColourScheme loadDefaultColours()\r
+  {\r
+    if(jalview.bin.Cache.getProperty("USER_DEFINED_COLOUR")!=null)\r
+      return loadDefaultColours(jalview.bin.Cache.getProperty("USER_DEFINED_COLOUR"));\r
+    else\r
+      return null;\r
+  }\r
 \r
+  public static UserColourScheme loadDefaultColours(String file)\r
+  {\r
+      UserColourScheme ucs = null;\r
+      Color cols[] = loadColours(file);\r
+      if(cols!=null)\r
+      {\r
+        ucs = new UserColourScheme(cols);\r
+        ucs.setThreshold(0);\r
       }\r
-      catch(Exception ex)\r
-      {}\r
+      return ucs;\r
+  }\r
+\r
+  static Color [] loadColours(String file)\r
+  {\r
+    Color [] newColours=null;\r
+    try\r
+    {\r
+      InputStreamReader in = new InputStreamReader(\r
+          new FileInputStream(file), "UTF-8");\r
+\r
+      jalview.binding.JalviewUserColours ucs = new jalview.binding.JalviewUserColours();\r
+      ucs = (jalview.binding.JalviewUserColours)ucs.unmarshal(in);\r
+\r
+      newColours = new Color[ucs.getColourCount()];\r
+      for (int i = 0; i < 24; i++)\r
+        newColours[i] = new Color(Integer.parseInt(ucs.getColour(i).getRGB(), 16));\r
+\r
     }\r
+    catch (Exception ex)\r
+    {System.out.println("Error loading UserColourFile "+file);}\r
 \r
+    return newColours;\r
   }\r
 \r
   protected void savebutton_actionPerformed(ActionEvent e)\r
   {\r
-    JFileChooser chooser = new JFileChooser(jalview.bin.Cache.LAST_DIRECTORY);\r
+    JalviewFileChooser chooser = new JalviewFileChooser(\r
+    jalview.bin.Cache.getProperty("LAST_DIRECTORY"),\r
+    new String[]{"jc"},\r
+    new String[]{"Jalview User Colours"},\r
+      "Jalview User Colours");\r
+\r
+    chooser.setFileView(new jalview.io.JalviewFileView());\r
     chooser.setDialogTitle("Save colour scheme");\r
     chooser.setToolTipText("Save");\r
     int value = chooser.showSaveDialog(this);\r
-    if (value == JFileChooser.APPROVE_OPTION)\r
+    if (value == JalviewFileChooser.APPROVE_OPTION)\r
     {\r
       String choice = chooser.getSelectedFile().getPath();\r
-      jalview.bin.Cache.LAST_DIRECTORY = choice;\r
+      jalview.bin.Cache.setProperty("USER_DEFINED_COLOUR", choice);\r
+      jalview.binding.JalviewUserColours ucs = new jalview.binding.JalviewUserColours();\r
 \r
       try{\r
-        PrintWriter out = new PrintWriter(new FileWriter(choice));\r
+        PrintWriter out = new PrintWriter(new OutputStreamWriter(\r
+                                     new FileOutputStream(choice), "UTF-8"));\r
         for(int i=0; i<24; i++)\r
         {\r
-          JButton button = (JButton)buttonPanel.getComponent(i);\r
-          out.println(button.getBackground().getRGB() + "");\r
+         JButton button = (JButton)buttonPanel.getComponent(i);\r
+         jalview.binding.Colour col = new jalview.binding.Colour();\r
+         col.setName(button.getText());\r
+         col.setRGB( jalview.util.Format.getHexString( button.getBackground() ));\r
+         ucs.addColour(col);\r
         }\r
 \r
+        ucs.marshal(out);\r
         out.close();\r
       }\r
       catch(Exception ex)\r
@@ -164,7 +239,7 @@ public class UserDefinedColours extends GUserDefinedColours implements ChangeLis
 \r
   protected void cancelButton_actionPerformed(ActionEvent e)\r
   {\r
-    UserColourScheme ucs = new UserColourScheme();\r
+\r
     Color [] newColours = new Color[24];\r
     for(int i=0; i<24; i++)\r
     {\r
@@ -172,14 +247,18 @@ public class UserDefinedColours extends GUserDefinedColours implements ChangeLis
           buttonPanel.getComponent(i).setBackground(newColours[i]);\r
     }\r
 \r
-    ucs.setColourScheme( newColours );\r
+    UserColourScheme ucs = new UserColourScheme( newColours );\r
+\r
     if (seqGroup != null)\r
       seqGroup.cs = ucs;\r
     else\r
       ap.av.setGlobalColourScheme(ucs);\r
 \r
-    ap.RefreshPanels();\r
+    ap.repaint();\r
 \r
+    try{\r
+      frame.setClosed(true);\r
+    }catch(Exception ex){}\r
   }\r
 \r
 \r