JAL-1588 part work save/restore of Chimera session
[jalview.git] / src / jalview / gui / AppJmol.java
1 /*
2  * Jalview - A Sequence Alignment Editor and Viewer (Version 2.8.2)
3  * Copyright (C) 2014 The Jalview Authors
4  * 
5  * This file is part of Jalview.
6  * 
7  * Jalview is free software: you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License 
9  * as published by the Free Software Foundation, either version 3
10  * of the License, or (at your option) any later version.
11  *  
12  * Jalview is distributed in the hope that it will be useful, but 
13  * WITHOUT ANY WARRANTY; without even the implied warranty 
14  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
15  * PURPOSE.  See the GNU General Public License for more details.
16  * 
17  * You should have received a copy of the GNU General Public License
18  * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
19  * The Jalview Authors are detailed in the 'AUTHORS' file.
20  */
21 package jalview.gui;
22
23 import jalview.bin.Cache;
24 import jalview.datamodel.Alignment;
25 import jalview.datamodel.AlignmentI;
26 import jalview.datamodel.ColumnSelection;
27 import jalview.datamodel.PDBEntry;
28 import jalview.datamodel.SequenceI;
29 import jalview.io.AppletFormatAdapter;
30 import jalview.io.JalviewFileChooser;
31 import jalview.io.JalviewFileView;
32 import jalview.schemes.BuriedColourScheme;
33 import jalview.schemes.ColourSchemeI;
34 import jalview.schemes.HelixColourScheme;
35 import jalview.schemes.HydrophobicColourScheme;
36 import jalview.schemes.PurinePyrimidineColourScheme;
37 import jalview.schemes.StrandColourScheme;
38 import jalview.schemes.TaylorColourScheme;
39 import jalview.schemes.TurnColourScheme;
40 import jalview.schemes.ZappoColourScheme;
41 import jalview.structures.models.AAStructureBindingModel;
42 import jalview.util.MessageManager;
43 import jalview.util.Platform;
44
45 import java.awt.BorderLayout;
46 import java.awt.Color;
47 import java.awt.Dimension;
48 import java.awt.Font;
49 import java.awt.Graphics;
50 import java.awt.Rectangle;
51 import java.awt.event.ActionEvent;
52 import java.awt.event.ActionListener;
53 import java.awt.event.ItemEvent;
54 import java.awt.event.ItemListener;
55 import java.io.BufferedReader;
56 import java.io.File;
57 import java.io.FileOutputStream;
58 import java.io.FileReader;
59 import java.io.PrintWriter;
60 import java.util.Enumeration;
61 import java.util.Vector;
62
63 import javax.swing.JCheckBoxMenuItem;
64 import javax.swing.JColorChooser;
65 import javax.swing.JInternalFrame;
66 import javax.swing.JMenu;
67 import javax.swing.JMenuItem;
68 import javax.swing.JOptionPane;
69 import javax.swing.JPanel;
70 import javax.swing.JSplitPane;
71 import javax.swing.event.InternalFrameAdapter;
72 import javax.swing.event.InternalFrameEvent;
73 import javax.swing.event.MenuEvent;
74 import javax.swing.event.MenuListener;
75
76 public class AppJmol extends StructureViewerBase
77 {
78   AppJmolBinding jmb;
79
80   JPanel scriptWindow;
81
82   JSplitPane splitPane;
83
84   RenderPanel renderPanel;
85
86   Vector atomsPicked = new Vector();
87
88   private boolean addingStructures = false;
89
90   /**
91    * 
92    * @param file
93    * @param id
94    * @param seq
95    * @param ap
96    * @param loadStatus
97    * @param bounds
98    * @deprecated defaults to AppJmol(String[] files, ... , viewid);
99    */
100   @Deprecated
101   public AppJmol(String file, String id, SequenceI[] seq,
102           AlignmentPanel ap, String loadStatus, Rectangle bounds)
103   {
104     this(file, id, seq, ap, loadStatus, bounds, null);
105   }
106
107   /**
108    * @deprecated
109    */
110   @Deprecated
111   public AppJmol(String file, String id, SequenceI[] seq,
112           AlignmentPanel ap, String loadStatus, Rectangle bounds,
113           String viewid)
114   {
115     this(new String[]
116     { file }, new String[]
117     { id }, new SequenceI[][]
118     { seq }, ap, true, true, false, loadStatus, bounds, viewid);
119   }
120
121   ViewSelectionMenu seqColourBy;
122
123   /**
124    * 
125    * @param files
126    * @param ids
127    * @param seqs
128    * @param ap
129    * @param usetoColour
130    *          - add the alignment panel to the list used for colouring these
131    *          structures
132    * @param useToAlign
133    *          - add the alignment panel to the list used for aligning these
134    *          structures
135    * @param leaveColouringToJmol
136    *          - do not update the colours from any other source. Jmol is
137    *          handling them
138    * @param loadStatus
139    * @param bounds
140    * @param viewid
141    */
142   public AppJmol(String[] files, String[] ids, SequenceI[][] seqs,
143           AlignmentPanel ap, boolean usetoColour, boolean useToAlign,
144           boolean leaveColouringToJmol, String loadStatus,
145           Rectangle bounds, String viewid)
146   {
147     PDBEntry[] pdbentrys = new PDBEntry[files.length];
148     for (int i = 0; i < pdbentrys.length; i++)
149     {
150       PDBEntry pdbentry = new PDBEntry();
151       pdbentry.setFile(files[i]);
152       pdbentry.setId(ids[i]);
153       pdbentrys[i] = pdbentry;
154     }
155     // / TODO: check if protocol is needed to be set, and if chains are
156     // autodiscovered.
157     jmb = new AppJmolBinding(this, ap.getStructureSelectionManager(),
158             pdbentrys, seqs, null, null);
159
160     jmb.setLoadingFromArchive(true);
161     addAlignmentPanel(ap);
162     if (useToAlign)
163     {
164       useAlignmentPanelForSuperposition(ap);
165     }
166     if (leaveColouringToJmol || !usetoColour)
167     {
168       jmb.setColourBySequence(false);
169       seqColour.setSelected(false);
170       viewerColour.setSelected(true);
171     }
172     if (usetoColour)
173     {
174       useAlignmentPanelForColourbyseq(ap);
175       jmb.setColourBySequence(true);
176       seqColour.setSelected(true);
177       viewerColour.setSelected(false);
178     }
179     this.setBounds(bounds);
180     initMenus();
181     setViewId(viewid);
182     // jalview.gui.Desktop.addInternalFrame(this, "Loading File",
183     // bounds.width,bounds.height);
184
185     this.addInternalFrameListener(new InternalFrameAdapter()
186     {
187       public void internalFrameClosing(InternalFrameEvent internalFrameEvent)
188       {
189         closeViewer();
190       }
191     });
192     initJmol(loadStatus); // pdbentry, seq, JBPCHECK!
193
194   }
195
196   private void initMenus()
197   {
198     seqColour.setSelected(jmb.isColourBySequence());
199     viewerColour.setSelected(!jmb.isColourBySequence());
200     if (_colourwith == null)
201     {
202       _colourwith = new Vector<AlignmentPanel>();
203     }
204     if (_alignwith == null)
205     {
206       _alignwith = new Vector<AlignmentPanel>();
207     }
208
209     seqColourBy = new ViewSelectionMenu(MessageManager.getString("label.colour_by"), this, _colourwith,
210             new ItemListener()
211             {
212
213               @Override
214               public void itemStateChanged(ItemEvent e)
215               {
216                 if (!seqColour.isSelected())
217                 {
218                   seqColour.doClick();
219                 }
220                 else
221                 {
222                   // update the jmol display now.
223                   seqColour_actionPerformed(null);
224                 }
225               }
226             });
227     viewMenu.add(seqColourBy);
228     final ItemListener handler;
229     JMenu alpanels = new ViewSelectionMenu(MessageManager.getString("label.superpose_with"), this,
230             _alignwith, handler = new ItemListener()
231             {
232
233               @Override
234               public void itemStateChanged(ItemEvent e)
235               {
236                 alignStructs.setEnabled(_alignwith.size() > 0);
237                 alignStructs.setToolTipText(MessageManager
238                         .formatMessage(
239                                 "label.align_structures_using_linked_alignment_views",
240                                 new String[]
241                                 { new Integer(_alignwith.size()).toString() }));
242               }
243             });
244     handler.itemStateChanged(null);
245     viewerActionMenu.add(alpanels);
246     viewerActionMenu.addMenuListener(new MenuListener()
247     {
248
249       @Override
250       public void menuSelected(MenuEvent e)
251       {
252         handler.itemStateChanged(null);
253       }
254
255       @Override
256       public void menuDeselected(MenuEvent e)
257       {
258         // TODO Auto-generated method stub
259
260       }
261
262       @Override
263       public void menuCanceled(MenuEvent e)
264       {
265         // TODO Auto-generated method stub
266
267       }
268     });
269   }
270
271   IProgressIndicator progressBar = null;
272
273   /**
274    * add a single PDB structure to a new or existing Jmol view
275    * 
276    * @param pdbentry
277    * @param seq
278    * @param chains
279    * @param ap
280    */
281   public AppJmol(PDBEntry pdbentry, SequenceI[] seq, String[] chains,
282           final AlignmentPanel ap)
283   {
284     progressBar = ap.alignFrame;
285     // ////////////////////////////////
286     // Is the pdb file already loaded?
287     String alreadyMapped = ap.getStructureSelectionManager()
288             .alreadyMappedToFile(pdbentry.getId());
289
290     if (alreadyMapped != null)
291     {
292       int option = JOptionPane.showInternalConfirmDialog(Desktop.desktop,
293               MessageManager.formatMessage(
294                       "label.pdb_entry_is_already_displayed", new String[]
295                       { pdbentry.getId() }), MessageManager.formatMessage(
296                       "label.map_sequences_to_visible_window", new String[]
297                       { pdbentry.getId() }),
298               JOptionPane.YES_NO_CANCEL_OPTION);
299
300       if (option == JOptionPane.CANCEL_OPTION)
301       {
302         return;
303       }
304       if (option == JOptionPane.YES_OPTION)
305       {
306         // TODO : Fix multiple seq to one chain issue here.
307         ap.getStructureSelectionManager().setMapping(seq, chains,
308                 alreadyMapped, AppletFormatAdapter.FILE);
309         if (ap.getSeqPanel().seqCanvas.fr != null)
310         {
311           ap.getSeqPanel().seqCanvas.fr.featuresAdded();
312           ap.paintAlignment(true);
313         }
314
315         // Now this AppJmol is mapped to new sequences. We must add them to
316         // the exisiting array
317         JInternalFrame[] frames = Desktop.instance.getAllFrames();
318
319         for (int i = 0; i < frames.length; i++)
320         {
321           if (frames[i] instanceof AppJmol)
322           {
323             final AppJmol topJmol = ((AppJmol) frames[i]);
324             // JBPNOTE: this looks like a binding routine, rather than a gui
325             // routine
326             for (int pe = 0; pe < topJmol.jmb.getPdbCount(); pe++)
327             {
328               if (topJmol.jmb.getPdbEntry(pe).getFile()
329                       .equals(alreadyMapped))
330               {
331                 topJmol.jmb.addSequence(pe, seq);
332                 topJmol.addAlignmentPanel(ap);
333                 // add it to the set used for colouring
334                 topJmol.useAlignmentPanelForColourbyseq(ap);
335                 topJmol.buildActionMenu();
336                 ap.getStructureSelectionManager()
337                         .sequenceColoursChanged(ap);
338                 break;
339               }
340             }
341           }
342         }
343
344         return;
345       }
346     }
347     // /////////////////////////////////
348     // Check if there are other Jmol views involving this alignment
349     // and prompt user about adding this molecule to one of them
350     Vector existingViews = getJmolsFor(ap);
351     if (existingViews.size() > 0)
352     {
353       Enumeration jm = existingViews.elements();
354       while (jm.hasMoreElements())
355       {
356         AppJmol topJmol = (AppJmol) jm.nextElement();
357         // TODO: highlight topJmol in view somehow
358         int option = JOptionPane
359                 .showInternalConfirmDialog(
360                         Desktop.desktop,
361                         MessageManager.formatMessage(
362                                 "label.add_pdbentry_to_view", new String[]
363                                 { pdbentry.getId(), topJmol.getTitle() }),
364                         MessageManager
365                                 .getString("label.align_to_existing_structure_view"),
366                         JOptionPane.YES_NO_CANCEL_OPTION);
367         if (option == JOptionPane.CANCEL_OPTION)
368         {
369           return;
370         }
371         if (option == JOptionPane.YES_OPTION)
372         {
373           topJmol.useAlignmentPanelForSuperposition(ap);
374           topJmol.addStructure(pdbentry, seq, chains, true, ap.alignFrame);
375           return;
376         }
377       }
378     }
379     // /////////////////////////////////
380     openNewJmol(ap, new PDBEntry[]
381     { pdbentry }, new SequenceI[][]
382     { seq });
383   }
384
385   private void openNewJmol(AlignmentPanel ap, PDBEntry[] pdbentrys,
386           SequenceI[][] seqs)
387   {
388     progressBar = ap.alignFrame;
389     jmb = new AppJmolBinding(this, ap.getStructureSelectionManager(),
390             pdbentrys, seqs, null, null);
391     addAlignmentPanel(ap);
392     useAlignmentPanelForColourbyseq(ap);
393     if (pdbentrys.length > 1)
394     {
395       alignAddedStructures = true;
396       useAlignmentPanelForSuperposition(ap);
397     }
398     jmb.setColourBySequence(true);
399     setSize(400, 400); // probably should be a configurable/dynamic default here
400     initMenus();
401     worker = null;
402     {
403       addingStructures = false;
404       worker = new Thread(this);
405       worker.start();
406     }
407     this.addInternalFrameListener(new InternalFrameAdapter()
408     {
409       public void internalFrameClosing(InternalFrameEvent internalFrameEvent)
410       {
411         closeViewer();
412       }
413     });
414
415   }
416
417   /**
418    * create a new Jmol containing several structures superimposed using the
419    * given alignPanel.
420    * 
421    * @param ap
422    * @param pe
423    * @param seqs
424    */
425   public AppJmol(AlignmentPanel ap, PDBEntry[] pe, SequenceI[][] seqs)
426   {
427     openNewJmol(ap, pe, seqs);
428   }
429
430   /**
431    * pdb retrieval thread.
432    */
433   private Thread worker = null;
434
435   /**
436    * add a new structure (with associated sequences and chains) to this viewer,
437    * retrieving it if necessary first.
438    * 
439    * @param pdbentry
440    * @param seq
441    * @param chains
442    * @param alignFrame
443    * @param align
444    *          if true, new structure(s) will be align using associated alignment
445    */
446   private void addStructure(final PDBEntry pdbentry, final SequenceI[] seq,
447           final String[] chains, final boolean b,
448           final IProgressIndicator alignFrame)
449   {
450     if (pdbentry.getFile() == null)
451     {
452       if (worker != null && worker.isAlive())
453       {
454         // a retrieval is in progress, wait around and add ourselves to the
455         // queue.
456         new Thread(new Runnable()
457         {
458           public void run()
459           {
460             while (worker != null && worker.isAlive() && _started)
461             {
462               try
463               {
464                 Thread.sleep(100 + ((int) Math.random() * 100));
465
466               } catch (Exception e)
467               {
468               }
469
470             }
471             // and call ourselves again.
472             addStructure(pdbentry, seq, chains, b, alignFrame);
473           }
474         }).start();
475         return;
476       }
477     }
478     // otherwise, start adding the structure.
479     jmb.addSequenceAndChain(new PDBEntry[]
480     { pdbentry }, new SequenceI[][]
481     { seq }, new String[][]
482     { chains });
483     addingStructures = true;
484     _started = false;
485     alignAddedStructures = b;
486     progressBar = alignFrame; // visual indication happens on caller frame.
487     (worker = new Thread(this)).start();
488     return;
489   }
490
491   private Vector getJmolsFor(AlignmentPanel apanel)
492   {
493     Vector result = new Vector();
494     JInternalFrame[] frames = Desktop.instance.getAllFrames();
495
496     for (JInternalFrame frame : frames)
497     {
498       if (frame instanceof AppJmol)
499       {
500         if (((StructureViewerBase) frame).isLinkedWith(apanel))
501         {
502           result.addElement(frame);
503         }
504       }
505     }
506     return result;
507   }
508
509   void initJmol(String command)
510   {
511     jmb.setFinishedInit(false);
512     renderPanel = new RenderPanel();
513     // TODO: consider waiting until the structure/view is fully loaded before
514     // displaying
515     this.getContentPane().add(renderPanel, java.awt.BorderLayout.CENTER);
516     jalview.gui.Desktop.addInternalFrame(this, jmb.getViewerTitle(),
517             getBounds().width, getBounds().height);
518     if (scriptWindow == null)
519     {
520       BorderLayout bl = new BorderLayout();
521       bl.setHgap(0);
522       bl.setVgap(0);
523       scriptWindow = new JPanel(bl);
524       scriptWindow.setVisible(false);
525     }
526     ;
527     jmb.allocateViewer(renderPanel, true, "", null, null, "", scriptWindow,
528             null);
529     jmb.newJmolPopup(true, "Jmol", true);
530     if (command == null)
531     {
532       command = "";
533     }
534     jmb.evalStateCommand(command);
535     jmb.setFinishedInit(true);
536   }
537
538   void setChainMenuItems(Vector chains)
539   {
540     chainMenu.removeAll();
541     if (chains == null)
542     {
543       return;
544     }
545     JMenuItem menuItem = new JMenuItem(
546             MessageManager.getString("label.all"));
547     menuItem.addActionListener(new ActionListener()
548     {
549       public void actionPerformed(ActionEvent evt)
550       {
551         allChainsSelected = true;
552         for (int i = 0; i < chainMenu.getItemCount(); i++)
553         {
554           if (chainMenu.getItem(i) instanceof JCheckBoxMenuItem)
555           {
556             ((JCheckBoxMenuItem) chainMenu.getItem(i)).setSelected(true);
557           }
558         }
559         centerViewer();
560         allChainsSelected = false;
561       }
562     });
563
564     chainMenu.add(menuItem);
565
566     for (int c = 0; c < chains.size(); c++)
567     {
568       menuItem = new JCheckBoxMenuItem(chains.elementAt(c).toString(), true);
569       menuItem.addItemListener(new ItemListener()
570       {
571         public void itemStateChanged(ItemEvent evt)
572         {
573           if (!allChainsSelected)
574           {
575             centerViewer();
576           }
577         }
578       });
579
580       chainMenu.add(menuItem);
581     }
582   }
583
584   boolean allChainsSelected = false;
585
586   private boolean alignAddedStructures = false;
587
588   void centerViewer()
589   {
590     Vector toshow = new Vector();
591     String lbl;
592     int mlength, p, mnum;
593     for (int i = 0; i < chainMenu.getItemCount(); i++)
594     {
595       if (chainMenu.getItem(i) instanceof JCheckBoxMenuItem)
596       {
597         JCheckBoxMenuItem item = (JCheckBoxMenuItem) chainMenu.getItem(i);
598         if (item.isSelected())
599         {
600           toshow.addElement(item.getText());
601         }
602       }
603     }
604     jmb.centerViewer(toshow);
605   }
606
607   public void closeViewer()
608   {
609     jmb.closeViewer();
610     setAlignmentPanel(null);
611     _aps.clear();
612     _alignwith.clear();
613     _colourwith.clear();
614     // TODO: check for memory leaks where instance isn't finalised because jmb
615     // holds a reference to the window
616     jmb = null;
617   }
618
619   /**
620    * state flag for PDB retrieval thread
621    */
622   private boolean _started = false;
623
624   public void run()
625   {
626     _started = true;
627     String pdbid = "";
628     // todo - record which pdbids were successfuly imported.
629     StringBuffer errormsgs = new StringBuffer(), files = new StringBuffer();
630     try
631     {
632       String[] curfiles = jmb.getPdbFile(); // files currently in viewer
633       // TODO: replace with reference fetching/transfer code (validate PDBentry
634       // as a DBRef?)
635       jalview.ws.dbsources.Pdb pdbclient = new jalview.ws.dbsources.Pdb();
636       for (int pi = 0; pi < jmb.getPdbCount(); pi++)
637       {
638         String file = jmb.getPdbEntry(pi).getFile();
639         if (file == null)
640         {
641           // retrieve the pdb and store it locally
642           AlignmentI pdbseq = null;
643           pdbid = jmb.getPdbEntry(pi).getId();
644           long hdl = pdbid.hashCode() - System.currentTimeMillis();
645           if (progressBar != null)
646           {
647             progressBar.setProgressBar(MessageManager.formatMessage("status.fetching_pdb", new String[]{pdbid}), hdl);
648           }
649           try
650           {
651             pdbseq = pdbclient.getSequenceRecords(pdbid = jmb.getPdbEntry(
652                     pi)
653                     .getId());
654           } catch (OutOfMemoryError oomerror)
655           {
656             new OOMWarning("Retrieving PDB id " + pdbid, oomerror);
657           } catch (Exception ex)
658           {
659             ex.printStackTrace();
660             errormsgs.append("'" + pdbid + "'");
661           }
662           if (progressBar != null)
663           {
664             progressBar.setProgressBar(MessageManager.getString("label.state_completed"), hdl);
665           }
666           if (pdbseq != null)
667           {
668             // just transfer the file name from the first sequence's first
669             // PDBEntry
670             file = new File(((PDBEntry) pdbseq.getSequenceAt(0).getPDBId()
671                     .elementAt(0)).getFile()).getAbsolutePath();
672             jmb.getPdbEntry(pi).setFile(file);
673
674             files.append(" \"" + Platform.escapeString(file) + "\"");
675           }
676           else
677           {
678             errormsgs.append("'" + pdbid + "' ");
679           }
680         }
681         else
682         {
683           if (curfiles != null && curfiles.length > 0)
684           {
685             addingStructures = true; // already files loaded.
686             for (int c = 0; c < curfiles.length; c++)
687             {
688               if (curfiles[c].equals(file))
689               {
690                 file = null;
691                 break;
692               }
693             }
694           }
695           if (file != null)
696           {
697             files.append(" \"" + Platform.escapeString(file) + "\"");
698           }
699         }
700       }
701     } catch (OutOfMemoryError oomerror)
702     {
703       new OOMWarning("Retrieving PDB files: " + pdbid, oomerror);
704     } catch (Exception ex)
705     {
706       ex.printStackTrace();
707       errormsgs.append("When retrieving pdbfiles : current was: '" + pdbid
708               + "'");
709     }
710     if (errormsgs.length() > 0)
711     {
712
713       JOptionPane.showInternalMessageDialog(Desktop.desktop, MessageManager
714               .formatMessage("label.pdb_entries_couldnt_be_retrieved",
715                       new String[]
716                       { errormsgs.toString() }), MessageManager
717               .getString("label.couldnt_load_file"),
718               JOptionPane.ERROR_MESSAGE);
719
720     }
721     long lastnotify = jmb.getLoadNotifiesHandled();
722     if (files.length() > 0)
723     {
724       if (!addingStructures)
725       {
726
727         try
728         {
729           initJmol("load FILES " + files.toString());
730         } catch (OutOfMemoryError oomerror)
731         {
732           new OOMWarning("When trying to open the Jmol viewer!", oomerror);
733           Cache.log.debug("File locations are " + files);
734         } catch (Exception ex)
735         {
736           Cache.log.error("Couldn't open Jmol viewer!", ex);
737         }
738       }
739       else
740       {
741         StringBuffer cmd = new StringBuffer();
742         cmd.append("loadingJalviewdata=true\nload APPEND ");
743         cmd.append(files.toString());
744         cmd.append("\nloadingJalviewdata=null");
745         final String command = cmd.toString();
746         cmd = null;
747         lastnotify = jmb.getLoadNotifiesHandled();
748
749         try
750         {
751           jmb.evalStateCommand(command);
752         } catch (OutOfMemoryError oomerror)
753         {
754           new OOMWarning(
755                   "When trying to add structures to the Jmol viewer!",
756                   oomerror);
757           Cache.log.debug("File locations are " + files);
758         } catch (Exception ex)
759         {
760           Cache.log.error("Couldn't add files to Jmol viewer!", ex);
761         }
762       }
763
764       // need to wait around until script has finished
765       while (addingStructures ? lastnotify >= jmb.getLoadNotifiesHandled()
766               : (jmb.isFinishedInit() && jmb.getPdbFile() != null && jmb
767                       .getPdbFile().length != jmb.getPdbCount()))
768       {
769         try
770         {
771           Cache.log.debug("Waiting around for jmb notify.");
772           Thread.sleep(35);
773         } catch (Exception e)
774         {
775         }
776       }
777       // refresh the sequence colours for the new structure(s)
778       for (AlignmentPanel ap : _colourwith)
779       {
780         jmb.updateColours(ap);
781       }
782       // do superposition if asked to
783       if (alignAddedStructures)
784       {
785         javax.swing.SwingUtilities.invokeLater(new Runnable()
786         {
787           public void run()
788           {
789             alignStructs_withAllAlignPanels();
790             // jmb.superposeStructures(ap.av.getAlignment(), -1, null);
791           }
792         });
793         alignAddedStructures = false;
794       }
795       addingStructures = false;
796
797     }
798     _started = false;
799     worker = null;
800   }
801
802   @Override
803   public void pdbFile_actionPerformed(ActionEvent actionEvent)
804   {
805     JalviewFileChooser chooser = new JalviewFileChooser(
806             jalview.bin.Cache.getProperty("LAST_DIRECTORY"));
807
808     chooser.setFileView(new JalviewFileView());
809     chooser.setDialogTitle(MessageManager.getString("label.save_pdb_file"));
810     chooser.setToolTipText(MessageManager.getString("action.save"));
811
812     int value = chooser.showSaveDialog(this);
813
814     if (value == JalviewFileChooser.APPROVE_OPTION)
815     {
816       try
817       {
818         // TODO: cope with multiple PDB files in view
819         BufferedReader in = new BufferedReader(new FileReader(
820                 jmb.getPdbFile()[0]));
821         File outFile = chooser.getSelectedFile();
822
823         PrintWriter out = new PrintWriter(new FileOutputStream(outFile));
824         String data;
825         while ((data = in.readLine()) != null)
826         {
827           if (!(data.indexOf("<PRE>") > -1 || data.indexOf("</PRE>") > -1))
828           {
829             out.println(data);
830           }
831         }
832         out.close();
833       } catch (Exception ex)
834       {
835         ex.printStackTrace();
836       }
837     }
838   }
839
840   @Override
841   public void viewMapping_actionPerformed(ActionEvent actionEvent)
842   {
843     jalview.gui.CutAndPasteTransfer cap = new jalview.gui.CutAndPasteTransfer();
844     try
845     {
846       for (int pdbe = 0; pdbe < jmb.getPdbCount(); pdbe++)
847       {
848         cap.appendText(jmb.printMapping(jmb.getPdbEntry(pdbe).getFile()));
849         cap.appendText("\n");
850       }
851     } catch (OutOfMemoryError e)
852     {
853       new OOMWarning(
854               "composing sequence-structure alignments for display in text box.",
855               e);
856       cap.dispose();
857       return;
858     }
859     jalview.gui.Desktop.addInternalFrame(cap,
860             MessageManager.getString("label.pdb_sequence_mapping"), 550,
861             600);
862   }
863
864   @Override
865   public void eps_actionPerformed(ActionEvent e)
866   {
867     makePDBImage(jalview.util.ImageMaker.TYPE.EPS);
868   }
869
870   @Override
871   public void png_actionPerformed(ActionEvent e)
872   {
873     makePDBImage(jalview.util.ImageMaker.TYPE.PNG);
874   }
875
876   void makePDBImage(jalview.util.ImageMaker.TYPE type)
877   {
878     int width = getWidth();
879     int height = getHeight();
880
881     jalview.util.ImageMaker im;
882
883     if (type == jalview.util.ImageMaker.TYPE.PNG)
884     {
885       im = new jalview.util.ImageMaker(this,
886               jalview.util.ImageMaker.TYPE.PNG,
887               "Make PNG image from view", width, height, null, null);
888     }
889     else if (type == jalview.util.ImageMaker.TYPE.EPS)
890     {
891       im = new jalview.util.ImageMaker(this,
892               jalview.util.ImageMaker.TYPE.EPS,
893               "Make EPS file from view", width, height, null,
894               this.getTitle());
895     }
896     else
897     {
898
899       im = new jalview.util.ImageMaker(this,
900               jalview.util.ImageMaker.TYPE.SVG, "Make SVG file from PCA",
901               width, height, null, this.getTitle());
902     }
903
904     if (im.getGraphics() != null)
905     {
906       Rectangle rect = new Rectangle(width, height);
907       jmb.viewer.renderScreenImage(im.getGraphics(), rect.getSize(), rect);
908       im.writeImage();
909     }
910   }
911
912   @Override
913   public void viewerColour_actionPerformed(ActionEvent actionEvent)
914   {
915     if (viewerColour.isSelected())
916     {
917       // disable automatic sequence colouring.
918       jmb.setColourBySequence(false);
919     }
920   }
921
922   @Override
923   public void seqColour_actionPerformed(ActionEvent actionEvent)
924   {
925     jmb.setColourBySequence(seqColour.isSelected());
926     if (_colourwith == null)
927     {
928       _colourwith = new Vector<AlignmentPanel>();
929     }
930     if (jmb.isColourBySequence())
931     {
932       if (!jmb.isLoadingFromArchive())
933       {
934         if (_colourwith.size() == 0 && getAlignmentPanel() != null)
935         {
936           // Make the currently displayed alignment panel the associated view
937           _colourwith.add(getAlignmentPanel().alignFrame.alignPanel);
938         }
939       }
940       // Set the colour using the current view for the associated alignframe
941       for (AlignmentPanel ap : _colourwith)
942       {
943         jmb.colourBySequence(ap.av.showSequenceFeatures, ap);
944       }
945     }
946   }
947
948   @Override
949   public void chainColour_actionPerformed(ActionEvent actionEvent)
950   {
951     chainColour.setSelected(true);
952     jmb.colourByChain();
953   }
954
955   @Override
956   public void chargeColour_actionPerformed(ActionEvent actionEvent)
957   {
958     chargeColour.setSelected(true);
959     jmb.colourByCharge();
960   }
961
962   @Override
963   public void zappoColour_actionPerformed(ActionEvent actionEvent)
964   {
965     zappoColour.setSelected(true);
966     jmb.setJalviewColourScheme(new ZappoColourScheme());
967   }
968
969   @Override
970   public void taylorColour_actionPerformed(ActionEvent actionEvent)
971   {
972     taylorColour.setSelected(true);
973     jmb.setJalviewColourScheme(new TaylorColourScheme());
974   }
975
976   @Override
977   public void hydroColour_actionPerformed(ActionEvent actionEvent)
978   {
979     hydroColour.setSelected(true);
980     jmb.setJalviewColourScheme(new HydrophobicColourScheme());
981   }
982
983   @Override
984   public void helixColour_actionPerformed(ActionEvent actionEvent)
985   {
986     helixColour.setSelected(true);
987     jmb.setJalviewColourScheme(new HelixColourScheme());
988   }
989
990   @Override
991   public void strandColour_actionPerformed(ActionEvent actionEvent)
992   {
993     strandColour.setSelected(true);
994     jmb.setJalviewColourScheme(new StrandColourScheme());
995   }
996
997   @Override
998   public void turnColour_actionPerformed(ActionEvent actionEvent)
999   {
1000     turnColour.setSelected(true);
1001     jmb.setJalviewColourScheme(new TurnColourScheme());
1002   }
1003
1004   @Override
1005   public void buriedColour_actionPerformed(ActionEvent actionEvent)
1006   {
1007     buriedColour.setSelected(true);
1008     jmb.setJalviewColourScheme(new BuriedColourScheme());
1009   }
1010
1011   @Override
1012   public void purinePyrimidineColour_actionPerformed(ActionEvent actionEvent)
1013   {
1014     setJalviewColourScheme(new PurinePyrimidineColourScheme());
1015   }
1016
1017   @Override
1018   public void userColour_actionPerformed(ActionEvent actionEvent)
1019   {
1020     userColour.setSelected(true);
1021     new UserDefinedColours(this, null);
1022   }
1023
1024   @Override
1025   public void backGround_actionPerformed(ActionEvent actionEvent)
1026   {
1027     java.awt.Color col = JColorChooser.showDialog(this,
1028             MessageManager.getString("label.select_backgroud_colour"), null);
1029     if (col != null)
1030     {
1031       jmb.setBackgroundColour(col);
1032     }
1033   }
1034
1035   @Override
1036   public void showHelp_actionPerformed(ActionEvent actionEvent)
1037   {
1038     try
1039     {
1040       jalview.util.BrowserLauncher
1041               .openURL("http://jmol.sourceforge.net/docs/JmolUserGuide/");
1042     } catch (Exception ex)
1043     {
1044     }
1045   }
1046
1047   public void showConsole(boolean showConsole)
1048   {
1049
1050     if (showConsole)
1051     {
1052       if (splitPane == null)
1053       {
1054         splitPane = new JSplitPane(JSplitPane.VERTICAL_SPLIT);
1055         splitPane.setTopComponent(renderPanel);
1056         splitPane.setBottomComponent(scriptWindow);
1057         this.getContentPane().add(splitPane, BorderLayout.CENTER);
1058         splitPane.setDividerLocation(getHeight() - 200);
1059         scriptWindow.setVisible(true);
1060         scriptWindow.validate();
1061         splitPane.validate();
1062       }
1063
1064     }
1065     else
1066     {
1067       if (splitPane != null)
1068       {
1069         splitPane.setVisible(false);
1070       }
1071
1072       splitPane = null;
1073
1074       this.getContentPane().add(renderPanel, BorderLayout.CENTER);
1075     }
1076
1077     validate();
1078   }
1079
1080   class RenderPanel extends JPanel
1081   {
1082     final Dimension currentSize = new Dimension();
1083
1084     final Rectangle rectClip = new Rectangle();
1085
1086     public void paintComponent(Graphics g)
1087     {
1088       getSize(currentSize);
1089       g.getClipBounds(rectClip);
1090
1091       if (jmb.fileLoadingError != null)
1092       {
1093         g.setColor(Color.black);
1094         g.fillRect(0, 0, currentSize.width, currentSize.height);
1095         g.setColor(Color.white);
1096         g.setFont(new Font("Verdana", Font.BOLD, 14));
1097         g.drawString(MessageManager.getString("label.error_loading_file")
1098                 + "...", 20, currentSize.height / 2);
1099         StringBuffer sb = new StringBuffer();
1100         int lines = 0;
1101         for (int e = 0; e < jmb.getPdbCount(); e++)
1102         {
1103           sb.append(jmb.getPdbEntry(e).getId());
1104           if (e < jmb.getPdbCount() - 1)
1105           {
1106             sb.append(",");
1107           }
1108
1109           if (e == jmb.getPdbCount() - 1 || sb.length() > 20)
1110           {
1111             lines++;
1112             g.drawString(sb.toString(), 20, currentSize.height / 2 - lines
1113                     * g.getFontMetrics().getHeight());
1114           }
1115         }
1116       }
1117       else if (jmb == null || jmb.viewer == null || !jmb.isFinishedInit())
1118       {
1119         g.setColor(Color.black);
1120         g.fillRect(0, 0, currentSize.width, currentSize.height);
1121         g.setColor(Color.white);
1122         g.setFont(new Font("Verdana", Font.BOLD, 14));
1123         g.drawString(MessageManager.getString("label.retrieving_pdb_data"),
1124                 20, currentSize.height / 2);
1125       }
1126       else
1127       {
1128         jmb.viewer.renderScreenImage(g, currentSize, rectClip);
1129       }
1130     }
1131   }
1132
1133   public void updateTitleAndMenus()
1134   {
1135     if (jmb.fileLoadingError != null && jmb.fileLoadingError.length() > 0)
1136     {
1137       repaint();
1138       return;
1139     }
1140     setChainMenuItems(jmb.chainNames);
1141
1142     this.setTitle(jmb.getViewerTitle());
1143     if (jmb.getPdbFile().length > 1 && jmb.getSequence().length > 1)
1144     {
1145       viewerActionMenu.setVisible(true);
1146     }
1147     if (!jmb.isLoadingFromArchive())
1148     {
1149       seqColour_actionPerformed(null);
1150     }
1151   }
1152
1153   /*
1154    * (non-Javadoc)
1155    * 
1156    * @see
1157    * jalview.jbgui.GStructureViewer#alignStructs_actionPerformed(java.awt.event
1158    * .ActionEvent)
1159    */
1160   @Override
1161   protected void alignStructs_actionPerformed(ActionEvent actionEvent)
1162   {
1163     alignStructs_withAllAlignPanels();
1164   }
1165
1166   private void alignStructs_withAllAlignPanels()
1167   {
1168     if (getAlignmentPanel() == null)
1169     {
1170       return;
1171     }
1172     ;
1173     if (_alignwith.size() == 0)
1174     {
1175       _alignwith.add(getAlignmentPanel());
1176     }
1177     ;
1178     try
1179     {
1180       AlignmentI[] als = new Alignment[_alignwith.size()];
1181       ColumnSelection[] alc = new ColumnSelection[_alignwith.size()];
1182       int[] alm = new int[_alignwith.size()];
1183       int a = 0;
1184
1185       for (AlignmentPanel ap : _alignwith)
1186       {
1187         als[a] = ap.av.getAlignment();
1188         alm[a] = -1;
1189         alc[a++] = ap.av.getColumnSelection();
1190       }
1191       jmb.superposeStructures(als, alm, alc);
1192     } catch (Exception e)
1193     {
1194       StringBuffer sp = new StringBuffer();
1195       for (AlignmentPanel ap : _alignwith)
1196       {
1197         sp.append("'" + ap.alignFrame.getTitle() + "' ");
1198       }
1199       Cache.log.info("Couldn't align structures with the " + sp.toString()
1200               + "associated alignment panels.", e);
1201
1202     }
1203
1204   }
1205
1206   public void setJalviewColourScheme(ColourSchemeI ucs)
1207   {
1208     jmb.setJalviewColourScheme(ucs);
1209
1210   }
1211
1212   /**
1213    * 
1214    * @param alignment
1215    * @return first alignment panel displaying given alignment, or the default
1216    *         alignment panel
1217    */
1218   public AlignmentPanel getAlignmentPanelFor(AlignmentI alignment)
1219   {
1220     for (AlignmentPanel ap : getAllAlignmentPanels())
1221     {
1222       if (ap.av.getAlignment() == alignment)
1223       {
1224         return ap;
1225       }
1226     }
1227     return getAlignmentPanel();
1228   }
1229
1230   @Override
1231   public AAStructureBindingModel getBinding()
1232   {
1233     return this.jmb;
1234   }
1235
1236   @Override
1237   public String getStateInfo()
1238   {
1239     return jmb == null ? null : jmb.viewer.getStateInfo();
1240   }
1241
1242 }