2 * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
3 * Copyright (C) $$Year-Rel$$ The Jalview Authors
5 * This file is part of Jalview.
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.
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.
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.
23 import jalview.bin.Cache;
24 import jalview.datamodel.Alignment;
25 import jalview.datamodel.AlignmentI;
26 import jalview.datamodel.HiddenColumns;
27 import jalview.datamodel.PDBEntry;
28 import jalview.datamodel.SequenceI;
29 import jalview.gui.StructureViewer.ViewerType;
30 import jalview.gui.ViewSelectionMenu.ViewSetProvider;
31 import jalview.io.DataSourceType;
32 import jalview.io.JalviewFileChooser;
33 import jalview.io.JalviewFileView;
34 import jalview.jbgui.GStructureViewer;
35 import jalview.schemes.ColourSchemeI;
36 import jalview.schemes.ColourSchemes;
37 import jalview.structures.models.AAStructureBindingModel;
38 import jalview.util.MessageManager;
40 import java.awt.Color;
41 import java.awt.Component;
42 import java.awt.event.ActionEvent;
43 import java.awt.event.ActionListener;
44 import java.awt.event.ItemEvent;
45 import java.awt.event.ItemListener;
46 import java.io.BufferedReader;
48 import java.io.FileOutputStream;
49 import java.io.FileReader;
50 import java.io.IOException;
51 import java.io.PrintWriter;
52 import java.util.ArrayList;
53 import java.util.List;
54 import java.util.Vector;
56 import javax.swing.ButtonGroup;
57 import javax.swing.JCheckBoxMenuItem;
58 import javax.swing.JColorChooser;
59 import javax.swing.JMenu;
60 import javax.swing.JMenuItem;
61 import javax.swing.JRadioButtonMenuItem;
62 import javax.swing.event.MenuEvent;
63 import javax.swing.event.MenuListener;
66 * Base class with common functionality for JMol, Chimera or other structure
72 public abstract class StructureViewerBase extends GStructureViewer
73 implements Runnable, ViewSetProvider
76 * names for colour options (additional to Jalview colour schemes)
80 BySequence, ByChain, ChargeCysteine, ByViewer
84 * list of sequenceSet ids associated with the view
86 protected List<String> _aps = new ArrayList<>();
89 * list of alignment panels to use for superposition
91 protected Vector<AlignmentPanel> _alignwith = new Vector<>();
94 * list of alignment panels that are used for colouring structures by aligned
97 protected Vector<AlignmentPanel> _colourwith = new Vector<>();
99 private String viewId = null;
101 private AlignmentPanel ap;
103 protected boolean alignAddedStructures = false;
105 protected boolean _started = false;
107 protected boolean addingStructures = false;
109 protected Thread worker = null;
111 protected boolean allChainsSelected = false;
113 protected JMenu viewSelectionMenu;
116 * Default constructor
118 public StructureViewerBase()
126 * @return true if this Jmol instance is linked with the given alignPanel
128 public boolean isLinkedWith(AlignmentPanel ap2)
130 return _aps.contains(ap2.av.getSequenceSetId());
133 public boolean isUsedforaligment(AlignmentPanel ap2)
136 return (_alignwith != null) && _alignwith.contains(ap2);
139 public boolean isUsedforcolourby(AlignmentPanel ap2)
141 return (_colourwith != null) && _colourwith.contains(ap2);
146 * @return TRUE if the view is NOT being coloured by the alignment colours.
148 public boolean isColouredByViewer()
150 return !getBinding().isColourBySequence();
153 public String getViewId()
157 viewId = System.currentTimeMillis() + "." + this.hashCode();
162 protected void setViewId(String viewId)
164 this.viewId = viewId;
167 public abstract String getStateInfo();
169 protected void buildActionMenu()
171 if (_alignwith == null)
173 _alignwith = new Vector<>();
175 if (_alignwith.size() == 0 && ap != null)
180 for (Component c : viewerActionMenu.getMenuComponents())
182 if (c != alignStructs)
184 viewerActionMenu.remove((JMenuItem) c);
189 public AlignmentPanel getAlignmentPanel()
194 protected void setAlignmentPanel(AlignmentPanel alp)
200 public AlignmentPanel[] getAllAlignmentPanels()
202 AlignmentPanel[] t, list = new AlignmentPanel[0];
203 for (String setid : _aps)
205 AlignmentPanel[] panels = PaintRefresher.getAssociatedPanels(setid);
208 t = new AlignmentPanel[list.length + panels.length];
209 System.arraycopy(list, 0, t, 0, list.length);
210 System.arraycopy(panels, 0, t, list.length, panels.length);
219 * set the primary alignmentPanel reference and add another alignPanel to the
220 * list of ones to use for colouring and aligning
224 public void addAlignmentPanel(AlignmentPanel nap)
226 if (getAlignmentPanel() == null)
228 setAlignmentPanel(nap);
230 if (!_aps.contains(nap.av.getSequenceSetId()))
232 _aps.add(nap.av.getSequenceSetId());
237 * remove any references held to the given alignment panel
241 public void removeAlignmentPanel(AlignmentPanel nap)
245 _alignwith.remove(nap);
246 _colourwith.remove(nap);
247 if (getAlignmentPanel() == nap)
249 setAlignmentPanel(null);
250 for (AlignmentPanel aps : getAllAlignmentPanels())
254 setAlignmentPanel(aps);
259 } catch (Exception ex)
262 if (getAlignmentPanel() != null)
268 public void useAlignmentPanelForSuperposition(AlignmentPanel nap)
270 addAlignmentPanel(nap);
271 if (!_alignwith.contains(nap))
277 public void excludeAlignmentPanelForSuperposition(AlignmentPanel nap)
279 if (_alignwith.contains(nap))
281 _alignwith.remove(nap);
285 public void useAlignmentPanelForColourbyseq(AlignmentPanel nap,
286 boolean enableColourBySeq)
288 useAlignmentPanelForColourbyseq(nap);
289 getBinding().setColourBySequence(enableColourBySeq);
290 seqColour.setSelected(enableColourBySeq);
291 viewerColour.setSelected(!enableColourBySeq);
294 public void useAlignmentPanelForColourbyseq(AlignmentPanel nap)
296 addAlignmentPanel(nap);
297 if (!_colourwith.contains(nap))
299 _colourwith.add(nap);
303 public void excludeAlignmentPanelForColourbyseq(AlignmentPanel nap)
305 if (_colourwith.contains(nap))
307 _colourwith.remove(nap);
311 public abstract ViewerType getViewerType();
313 protected abstract IProgressIndicator getIProgressIndicator();
316 * add a new structure (with associated sequences and chains) to this viewer,
317 * retrieving it if necessary first.
323 * if true, new structure(s) will be aligned using associated
327 protected void addStructure(final PDBEntry pdbentry,
328 final SequenceI[] seqs, final String[] chains,
329 final boolean align, final IProgressIndicator alignFrame)
331 if (pdbentry.getFile() == null)
333 if (worker != null && worker.isAlive())
335 // a retrieval is in progress, wait around and add ourselves to the
337 new Thread(new Runnable()
342 while (worker != null && worker.isAlive() && _started)
346 Thread.sleep(100 + ((int) Math.random() * 100));
348 } catch (Exception e)
352 // and call ourselves again.
353 addStructure(pdbentry, seqs, chains, align, alignFrame);
359 // otherwise, start adding the structure.
360 getBinding().addSequenceAndChain(new PDBEntry[] { pdbentry },
362 { seqs }, new String[][] { chains });
363 addingStructures = true;
365 alignAddedStructures = align;
366 worker = new Thread(this);
372 * Presents a dialog with the option to add an align a structure to an
373 * existing structure view
377 * @return YES, NO or CANCEL JvOptionPane code
379 protected int chooseAlignStructureToViewer(String pdbId,
380 StructureViewerBase view)
382 int option = JvOptionPane.showInternalConfirmDialog(Desktop.desktop,
383 MessageManager.formatMessage("label.add_pdbentry_to_view",
385 { pdbId, view.getTitle() }),
387 .getString("label.align_to_existing_structure_view"),
388 JvOptionPane.YES_NO_CANCEL_OPTION);
392 protected boolean hasPdbId(String pdbId)
394 return getBinding().hasPdbId(pdbId);
397 protected abstract List<StructureViewerBase> getViewersFor(
401 * Check for any existing views involving this alignment and give user the
402 * option to add and align this molecule to one of them
409 * @return true if user adds to a view, or cancels entirely, else false
411 protected boolean addToExistingViewer(PDBEntry pdbentry, SequenceI[] seq,
412 String[] chains, final AlignmentPanel apanel, String pdbId)
414 for (StructureViewerBase view : getViewersFor(apanel))
416 // TODO: highlight the view somehow
418 * JAL-1742 exclude view with this structure already mapped (don't offer
419 * to align chain B to chain A of the same structure)
421 if (view.hasPdbId(pdbId))
425 int option = chooseAlignStructureToViewer(pdbId, view);
426 if (option == JvOptionPane.CANCEL_OPTION)
430 else if (option == JvOptionPane.YES_OPTION)
432 view.useAlignmentPanelForSuperposition(apanel);
433 view.addStructure(pdbentry, seq, chains, true, apanel.alignFrame);
438 // NO_OPTION - offer the next viewer if any
443 * nothing offered and selected
449 * Adds mappings for the given sequences to an already opened PDB structure,
450 * and updates any viewers that have the PDB file
457 protected void addSequenceMappingsToStructure(SequenceI[] seq,
458 String[] chains, final AlignmentPanel apanel, String pdbFilename)
460 // TODO : Fix multiple seq to one chain issue here.
462 * create the mappings
464 apanel.getStructureSelectionManager().setMapping(seq, chains,
465 pdbFilename, DataSourceType.FILE, getIProgressIndicator());
468 * alert the FeatureRenderer to show new (PDB RESNUM) features
470 if (apanel.getSeqPanel().seqCanvas.fr != null)
472 apanel.getSeqPanel().seqCanvas.fr.featuresAdded();
473 // note - we don't do a refresh for structure here because we do it
474 // explicitly for all panels later on
475 apanel.paintAlignment(true, false);
479 * add the sequences to any other viewers (of the same type) for this pdb
482 // JBPNOTE: this looks like a binding routine, rather than a gui routine
483 for (StructureViewerBase viewer : getViewersFor(null))
485 AAStructureBindingModel bindingModel = viewer.getBinding();
486 for (int pe = 0; pe < bindingModel.getPdbCount(); pe++)
488 if (bindingModel.getPdbEntry(pe).getFile().equals(pdbFilename))
490 bindingModel.addSequence(pe, seq);
491 viewer.addAlignmentPanel(apanel);
493 * add it to the set of alignments used for colouring structure by
496 viewer.useAlignmentPanelForColourbyseq(apanel);
497 viewer.buildActionMenu();
498 apanel.getStructureSelectionManager()
499 .sequenceColoursChanged(apanel);
507 * Check if the PDB file is already loaded, if so offer to add it to the
514 * @return true if the user chooses to add to a viewer, or to cancel entirely
516 protected boolean addAlreadyLoadedFile(SequenceI[] seq, String[] chains,
517 final AlignmentPanel apanel, String pdbId)
519 boolean finished = false;
520 String alreadyMapped = apanel.getStructureSelectionManager()
521 .alreadyMappedToFile(pdbId);
523 if (alreadyMapped != null)
526 * the PDB file is already loaded
528 int option = JvOptionPane.showInternalConfirmDialog(Desktop.desktop,
529 MessageManager.formatMessage(
530 "label.pdb_entry_is_already_displayed", new Object[]
532 MessageManager.formatMessage(
533 "label.map_sequences_to_visible_window", new Object[]
535 JvOptionPane.YES_NO_CANCEL_OPTION);
536 if (option == JvOptionPane.CANCEL_OPTION)
540 else if (option == JvOptionPane.YES_OPTION)
542 addSequenceMappingsToStructure(seq, chains, apanel, alreadyMapped);
549 void setChainMenuItems(List<String> chainNames)
551 chainMenu.removeAll();
552 if (chainNames == null || chainNames.isEmpty())
556 JMenuItem menuItem = new JMenuItem(
557 MessageManager.getString("label.all"));
558 menuItem.addActionListener(new ActionListener()
561 public void actionPerformed(ActionEvent evt)
563 allChainsSelected = true;
564 for (int i = 0; i < chainMenu.getItemCount(); i++)
566 if (chainMenu.getItem(i) instanceof JCheckBoxMenuItem)
568 ((JCheckBoxMenuItem) chainMenu.getItem(i)).setSelected(true);
571 showSelectedChains();
572 allChainsSelected = false;
576 chainMenu.add(menuItem);
578 for (String chain : chainNames)
580 menuItem = new JCheckBoxMenuItem(chain, true);
581 menuItem.addItemListener(new ItemListener()
584 public void itemStateChanged(ItemEvent evt)
586 if (!allChainsSelected)
588 showSelectedChains();
593 chainMenu.add(menuItem);
597 abstract void showSelectedChains();
600 * Action on selecting one of Jalview's registered colour schemes
603 public void changeColour_actionPerformed(String colourSchemeName)
605 AlignmentI al = getAlignmentPanel().av.getAlignment();
606 ColourSchemeI cs = ColourSchemes.getInstance()
607 .getColourScheme(colourSchemeName, al, null);
608 getBinding().setJalviewColourScheme(cs);
612 * Builds the colour menu
614 protected void buildColourMenu()
616 colourMenu.removeAll();
617 AlignmentI al = getAlignmentPanel().av.getAlignment();
620 * add colour by sequence, by chain, by charge and cysteine
622 colourMenu.add(seqColour);
623 colourMenu.add(chainColour);
624 colourMenu.add(chargeColour);
625 chargeColour.setEnabled(!al.isNucleotide());
628 * add all 'simple' (per-residue) colour schemes registered to Jalview
630 ButtonGroup itemGroup = ColourMenuHelper.addMenuItems(colourMenu, this,
634 * add 'colour by viewer' (menu item text is set in subclasses)
636 viewerColour.setSelected(false);
637 viewerColour.addActionListener(new ActionListener()
640 public void actionPerformed(ActionEvent actionEvent)
642 viewerColour_actionPerformed(actionEvent);
645 colourMenu.add(viewerColour);
648 * add 'set background colour'
650 JMenuItem backGround = new JMenuItem();
652 .setText(MessageManager.getString("action.background_colour"));
653 backGround.addActionListener(new ActionListener()
656 public void actionPerformed(ActionEvent actionEvent)
658 background_actionPerformed(actionEvent);
661 colourMenu.add(backGround);
664 * add colour buttons to a group so their selection is
665 * mutually exclusive (background colour is a separate option)
667 itemGroup.add(seqColour);
668 itemGroup.add(chainColour);
669 itemGroup.add(chargeColour);
670 itemGroup.add(viewerColour);
674 * Construct menu items
676 protected void initMenus()
678 AAStructureBindingModel binding = getBinding();
680 seqColour = new JRadioButtonMenuItem();
681 seqColour.setText(MessageManager.getString("action.by_sequence"));
682 seqColour.setName(ViewerColour.BySequence.name());
683 seqColour.setSelected(binding.isColourBySequence());
684 seqColour.addActionListener(new ActionListener()
687 public void actionPerformed(ActionEvent actionEvent)
689 seqColour_actionPerformed(actionEvent);
693 chainColour = new JRadioButtonMenuItem();
694 chainColour.setText(MessageManager.getString("action.by_chain"));
695 chainColour.setName(ViewerColour.ByChain.name());
696 chainColour.addActionListener(new ActionListener()
699 public void actionPerformed(ActionEvent actionEvent)
701 chainColour_actionPerformed(actionEvent);
705 chargeColour = new JRadioButtonMenuItem();
706 chargeColour.setText(MessageManager.getString("label.charge_cysteine"));
707 chargeColour.setName(ViewerColour.ChargeCysteine.name());
708 chargeColour.addActionListener(new ActionListener()
711 public void actionPerformed(ActionEvent actionEvent)
713 chargeColour_actionPerformed(actionEvent);
717 viewerColour = new JRadioButtonMenuItem();
718 // text is set in overrides of this method
719 viewerColour.setName(ViewerColour.ByViewer.name());
720 viewerColour.setSelected(!binding.isColourBySequence());
722 if (_colourwith == null)
724 _colourwith = new Vector<>();
726 if (_alignwith == null)
728 _alignwith = new Vector<>();
731 ViewSelectionMenu seqColourBy = new ViewSelectionMenu(
732 MessageManager.getString("label.colour_by"), this, _colourwith,
736 public void itemStateChanged(ItemEvent e)
738 if (!seqColour.isSelected())
744 // update the Chimera display now.
745 seqColour_actionPerformed(null);
749 viewMenu.add(seqColourBy);
751 final ItemListener handler = new ItemListener()
754 public void itemStateChanged(ItemEvent e)
756 alignStructs.setEnabled(!_alignwith.isEmpty());
757 alignStructs.setToolTipText(MessageManager.formatMessage(
758 "label.align_structures_using_linked_alignment_views",
762 viewSelectionMenu = new ViewSelectionMenu(
763 MessageManager.getString("label.superpose_with"), this,
764 _alignwith, handler);
765 handler.itemStateChanged(null);
766 viewerActionMenu.add(viewSelectionMenu, 0);
767 viewerActionMenu.addMenuListener(new MenuListener()
770 public void menuSelected(MenuEvent e)
772 handler.itemStateChanged(null);
776 public void menuDeselected(MenuEvent e)
781 public void menuCanceled(MenuEvent e)
790 public void setJalviewColourScheme(ColourSchemeI cs)
792 getBinding().setJalviewColourScheme(cs);
796 * Sends commands to the structure viewer to superimpose structures based on
797 * currently associated alignments. May optionally return an error message for
801 protected String alignStructs_actionPerformed(ActionEvent actionEvent)
803 return alignStructs_withAllAlignPanels();
806 protected String alignStructs_withAllAlignPanels()
808 if (getAlignmentPanel() == null)
813 if (_alignwith.size() == 0)
815 _alignwith.add(getAlignmentPanel());
821 AlignmentI[] als = new Alignment[_alignwith.size()];
822 HiddenColumns[] alc = new HiddenColumns[_alignwith.size()];
823 int[] alm = new int[_alignwith.size()];
826 for (AlignmentPanel ap : _alignwith)
828 als[a] = ap.av.getAlignment();
830 alc[a++] = ap.av.getAlignment().getHiddenColumns();
832 reply = getBinding().superposeStructures(als, alm, alc);
835 String text = MessageManager
836 .formatMessage("error.superposition_failed", reply);
837 statusBar.setText(text);
839 } catch (Exception e)
841 StringBuffer sp = new StringBuffer();
842 for (AlignmentPanel ap : _alignwith)
844 sp.append("'" + ap.alignFrame.getTitle() + "' ");
846 Cache.log.info("Couldn't align structures with the " + sp.toString()
847 + "associated alignment panels.", e);
853 public void background_actionPerformed(ActionEvent actionEvent)
855 Color col = JColorChooser.showDialog(this,
856 MessageManager.getString("label.select_background_colour"),
860 getBinding().setBackgroundColour(col);
865 public void viewerColour_actionPerformed(ActionEvent actionEvent)
867 if (viewerColour.isSelected())
869 // disable automatic sequence colouring.
870 getBinding().setColourBySequence(false);
875 public void chainColour_actionPerformed(ActionEvent actionEvent)
877 chainColour.setSelected(true);
878 getBinding().colourByChain();
882 public void chargeColour_actionPerformed(ActionEvent actionEvent)
884 chargeColour.setSelected(true);
885 getBinding().colourByCharge();
889 public void seqColour_actionPerformed(ActionEvent actionEvent)
891 AAStructureBindingModel binding = getBinding();
892 binding.setColourBySequence(seqColour.isSelected());
893 if (_colourwith == null)
895 _colourwith = new Vector<>();
897 if (binding.isColourBySequence())
899 if (!binding.isLoadingFromArchive())
901 if (_colourwith.size() == 0 && getAlignmentPanel() != null)
903 // Make the currently displayed alignment panel the associated view
904 _colourwith.add(getAlignmentPanel().alignFrame.alignPanel);
907 // Set the colour using the current view for the associated alignframe
908 for (AlignmentPanel ap : _colourwith)
910 binding.colourBySequence(ap);
916 public void pdbFile_actionPerformed(ActionEvent actionEvent)
918 JalviewFileChooser chooser = new JalviewFileChooser(
919 Cache.getProperty("LAST_DIRECTORY"));
921 chooser.setFileView(new JalviewFileView());
922 chooser.setDialogTitle(MessageManager.getString("label.save_pdb_file"));
923 chooser.setToolTipText(MessageManager.getString("action.save"));
925 int value = chooser.showSaveDialog(this);
927 if (value == JalviewFileChooser.APPROVE_OPTION)
929 BufferedReader in = null;
932 // TODO: cope with multiple PDB files in view
933 in = new BufferedReader(
934 new FileReader(getBinding().getStructureFiles()[0]));
935 File outFile = chooser.getSelectedFile();
937 PrintWriter out = new PrintWriter(new FileOutputStream(outFile));
939 while ((data = in.readLine()) != null)
941 if (!(data.indexOf("<PRE>") > -1 || data.indexOf("</PRE>") > -1))
947 } catch (Exception ex)
949 ex.printStackTrace();
957 } catch (IOException e)
967 public void viewMapping_actionPerformed(ActionEvent actionEvent)
969 CutAndPasteTransfer cap = new CutAndPasteTransfer();
972 cap.appendText(getBinding().printMappings());
973 } catch (OutOfMemoryError e)
976 "composing sequence-structure alignments for display in text box.",
981 Desktop.addInternalFrame(cap,
982 MessageManager.getString("label.pdb_sequence_mapping"), 550,
986 protected abstract String getViewerName();
989 * Configures the title and menu items of the viewer panel.
991 public void updateTitleAndMenus()
993 AAStructureBindingModel binding = getBinding();
994 if (binding.hasFileLoadingError())
999 setChainMenuItems(binding.getChainNames());
1001 this.setTitle(binding.getViewerTitle(getViewerName(), true));
1004 * enable 'Superpose with' if more than one mapped structure
1006 viewSelectionMenu.setEnabled(false);
1007 if (getBinding().getStructureFiles().length > 1
1008 && getBinding().getSequence().length > 1)
1010 viewSelectionMenu.setEnabled(true);
1014 * Show action menu if it has any enabled items
1016 viewerActionMenu.setVisible(false);
1017 for (int i = 0; i < viewerActionMenu.getItemCount(); i++)
1019 if (viewerActionMenu.getItem(i).isEnabled())
1021 viewerActionMenu.setVisible(true);
1026 if (!binding.isLoadingFromArchive())
1028 seqColour_actionPerformed(null);