X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FAppJmol.java;h=1c0dfe6c2e534bddc3d733588bfdc9542c17e5a0;hb=37de9310bec3501cbc6381e0c3dcb282fcaad812;hp=0cc04f7dd029d31d668468c5bf648208b35e1975;hpb=6486a5333807ca300db4a28a6bf9899f5dddf5a6;p=jalview.git diff --git a/src/jalview/gui/AppJmol.java b/src/jalview/gui/AppJmol.java index 0cc04f7..1c0dfe6 100644 --- a/src/jalview/gui/AppJmol.java +++ b/src/jalview/gui/AppJmol.java @@ -1,93 +1,100 @@ /* - * Jalview - A Sequence Alignment Editor and Viewer (Version 2.5) - * Copyright (C) 2010 J Procter, AM Waterhouse, G Barton, M Clamp, S Searle + * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$) + * Copyright (C) $$Year-Rel$$ The Jalview Authors * * This file is part of Jalview. * * Jalview is free software: you can redistribute it and/or * modify it under the terms of the GNU General Public License - * as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. - * + * as published by the Free Software Foundation, either version 3 + * of the License, or (at your option) any later version. + * * Jalview is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty * of MERCHANTABILITY or FITNESS FOR A PARTICULAR * PURPOSE. See the GNU General Public License for more details. * - * You should have received a copy of the GNU General Public License along with Jalview. If not, see . + * You should have received a copy of the GNU General Public License + * along with Jalview. If not, see . + * The Jalview Authors are detailed in the 'AUTHORS' file. */ package jalview.gui; -import java.util.regex.*; -import java.util.*; -import java.awt.*; -import javax.swing.*; -import javax.swing.event.*; -import java.awt.event.*; -import java.io.*; - -import jalview.jbgui.GStructureViewer; -import jalview.api.SequenceStructureBinding; import jalview.bin.Cache; -import jalview.datamodel.*; -import jalview.gui.*; -import jalview.structure.*; +import jalview.datamodel.Alignment; +import jalview.datamodel.AlignmentI; +import jalview.datamodel.ColumnSelection; import jalview.datamodel.PDBEntry; -import jalview.io.*; -import jalview.schemes.*; -import jalview.ws.ebi.EBIFetchClient; +import jalview.datamodel.SequenceI; +import jalview.gui.StructureViewer.ViewerType; +import jalview.io.JalviewFileChooser; +import jalview.io.JalviewFileView; +import jalview.schemes.BuriedColourScheme; +import jalview.schemes.ColourSchemeI; +import jalview.schemes.HelixColourScheme; +import jalview.schemes.HydrophobicColourScheme; +import jalview.schemes.PurinePyrimidineColourScheme; +import jalview.schemes.StrandColourScheme; +import jalview.schemes.TaylorColourScheme; +import jalview.schemes.TurnColourScheme; +import jalview.schemes.ZappoColourScheme; +import jalview.structures.models.AAStructureBindingModel; +import jalview.util.MessageManager; +import jalview.util.Platform; +import jalview.ws.dbsources.Pdb; + +import java.awt.BorderLayout; +import java.awt.Color; +import java.awt.Dimension; +import java.awt.Font; +import java.awt.Graphics; +import java.awt.Rectangle; +import java.awt.event.ActionEvent; +import java.awt.event.ActionListener; +import java.awt.event.ItemEvent; +import java.awt.event.ItemListener; +import java.io.BufferedReader; +import java.io.File; +import java.io.FileOutputStream; +import java.io.FileReader; +import java.io.IOException; +import java.io.PrintWriter; +import java.util.ArrayList; +import java.util.List; +import java.util.Vector; + +import javax.swing.JCheckBoxMenuItem; +import javax.swing.JColorChooser; +import javax.swing.JInternalFrame; +import javax.swing.JMenu; +import javax.swing.JMenuItem; +import javax.swing.JOptionPane; +import javax.swing.JPanel; +import javax.swing.JSplitPane; +import javax.swing.SwingUtilities; +import javax.swing.event.InternalFrameAdapter; +import javax.swing.event.InternalFrameEvent; +import javax.swing.event.MenuEvent; +import javax.swing.event.MenuListener; + +public class AppJmol extends StructureViewerBase +{ + // ms to wait for Jmol to load files + private static final int JMOL_LOAD_TIMEOUT = 20000; -import org.jmol.api.*; -import org.jmol.adapter.smarter.SmarterJmolAdapter; -import org.jmol.popup.*; -import org.jmol.viewer.JmolConstants; -import org.openscience.jmol.app.jmolpanel.AppConsole; + private static final String SPACE = " "; -public class AppJmol extends GStructureViewer implements Runnable, - SequenceStructureBinding + private static final String BACKSLASH = "\""; -{ AppJmolBinding jmb; JPanel scriptWindow; - + JSplitPane splitPane; RenderPanel renderPanel; - AlignmentPanel ap; - - Vector atomsPicked = new Vector(); - - private boolean addingStructures = false; - - /** - * - * @param file - * @param id - * @param seq - * @param ap - * @param loadStatus - * @param bounds - * @deprecated defaults to AppJmol(String[] files, ... , viewid); - */ - public AppJmol(String file, String id, SequenceI[] seq, - AlignmentPanel ap, String loadStatus, Rectangle bounds) - { - this(file, id, seq, ap, loadStatus, bounds, null); - } - - /** - * @deprecated - */ - public AppJmol(String file, String id, SequenceI[] seq, - AlignmentPanel ap, String loadStatus, Rectangle bounds, - String viewid) - { - this(new String[] - { file }, new String[] - { id }, new SequenceI[][] - { seq }, ap, loadStatus, bounds, viewid); - } + ViewSelectionMenu seqColourBy; /** * @@ -95,233 +102,266 @@ public class AppJmol extends GStructureViewer implements Runnable, * @param ids * @param seqs * @param ap + * @param usetoColour + * - add the alignment panel to the list used for colouring these + * structures + * @param useToAlign + * - add the alignment panel to the list used for aligning these + * structures + * @param leaveColouringToJmol + * - do not update the colours from any other source. Jmol is + * handling them * @param loadStatus * @param bounds * @param viewid */ public AppJmol(String[] files, String[] ids, SequenceI[][] seqs, - AlignmentPanel ap, String loadStatus, Rectangle bounds, - String viewid) + AlignmentPanel ap, boolean usetoColour, boolean useToAlign, + boolean leaveColouringToJmol, String loadStatus, + Rectangle bounds, String viewid) { PDBEntry[] pdbentrys = new PDBEntry[files.length]; for (int i = 0; i < pdbentrys.length; i++) { - PDBEntry pdbentry = new PDBEntry(); - pdbentry.setFile(files[i]); - pdbentry.setId(ids[i]); + // PDBEntry pdbentry = new PDBEntry(files[i], ids[i]); + PDBEntry pdbentry = new PDBEntry(ids[i], null, PDBEntry.Type.PDB, + files[i]); pdbentrys[i] = pdbentry; } // / TODO: check if protocol is needed to be set, and if chains are // autodiscovered. - jmb = new AppJmolBinding(this, pdbentrys, seqs, null, null); + jmb = new AppJmolBinding(this, ap.getStructureSelectionManager(), + pdbentrys, seqs, null, null); jmb.setLoadingFromArchive(true); - this.ap = ap; + addAlignmentPanel(ap); + if (useToAlign) + { + useAlignmentPanelForSuperposition(ap); + } + if (leaveColouringToJmol || !usetoColour) + { + jmb.setColourBySequence(false); + seqColour.setSelected(false); + viewerColour.setSelected(true); + } + else if (usetoColour) + { + useAlignmentPanelForColourbyseq(ap); + jmb.setColourBySequence(true); + seqColour.setSelected(true); + viewerColour.setSelected(false); + } this.setBounds(bounds); - jmb.setColourBySequence(false); - seqColour.setSelected(false); - viewId = viewid; + initMenus(); + setViewId(viewid); // jalview.gui.Desktop.addInternalFrame(this, "Loading File", // bounds.width,bounds.height); this.addInternalFrameListener(new InternalFrameAdapter() { + @Override public void internalFrameClosing(InternalFrameEvent internalFrameEvent) { - closeViewer(); + closeViewer(false); } }); initJmol(loadStatus); // pdbentry, seq, JBPCHECK! } - IProgressIndicator progressBar = null; - - public AppJmol(PDBEntry pdbentry, SequenceI[] seq, String[] chains, - AlignmentPanel ap) + private void initMenus() { - progressBar = ap.alignFrame; - // //////////////////////////////// - // Is the pdb file already loaded? - String alreadyMapped = StructureSelectionManager - .getStructureSelectionManager().alreadyMappedToFile( - pdbentry.getId()); - - if (alreadyMapped != null) - { - int option = JOptionPane.showInternalConfirmDialog(Desktop.desktop, - pdbentry.getId() + " is already displayed." - + "\nDo you want to re-use this viewer ?", - "Map Sequences to Visible Window: " + pdbentry.getId(), - JOptionPane.YES_NO_OPTION); - - if (option == JOptionPane.YES_OPTION) - { - StructureSelectionManager.getStructureSelectionManager() - .setMapping(seq, chains, alreadyMapped, - AppletFormatAdapter.FILE); - if (ap.seqPanel.seqCanvas.fr != null) - { - ap.seqPanel.seqCanvas.fr.featuresAdded(); - ap.paintAlignment(true); - } + seqColour.setSelected(jmb.isColourBySequence()); + viewerColour.setSelected(!jmb.isColourBySequence()); + if (_colourwith == null) + { + _colourwith = new Vector(); + } + if (_alignwith == null) + { + _alignwith = new Vector(); + } - // Now this AppJmol is mapped to new sequences. We must add them to - // the exisiting array - JInternalFrame[] frames = Desktop.instance.getAllFrames(); + seqColourBy = new ViewSelectionMenu( + MessageManager.getString("label.colour_by"), this, _colourwith, + new ItemListener() + { - for (int i = 0; i < frames.length; i++) - { - if (frames[i] instanceof AppJmol) - { - AppJmol topJmol = ((AppJmol) frames[i]); - // JBPNOTE: this looks like a binding routine, rather than a gui - // routine - for (int pe = 0; pe < topJmol.jmb.pdbentry.length; pe++) + @Override + public void itemStateChanged(ItemEvent e) + { + if (!seqColour.isSelected()) + { + seqColour.doClick(); + } + else + { + // update the jmol display now. + seqColour_actionPerformed(null); + } + } + }); + viewMenu.add(seqColourBy); + final ItemListener handler; + JMenu alpanels = new ViewSelectionMenu( + MessageManager.getString("label.superpose_with"), this, + _alignwith, handler = new ItemListener() { - if (topJmol.jmb.pdbentry[pe].getFile().equals(alreadyMapped)) + + @Override + public void itemStateChanged(ItemEvent e) { - topJmol.jmb.addSequence(pe, seq); - break; + alignStructs.setEnabled(_alignwith.size() > 0); + alignStructs.setToolTipText(MessageManager + .formatMessage( + "label.align_structures_using_linked_alignment_views", + new String[] { new Integer(_alignwith + .size()).toString() })); } - } - } - } + }); + handler.itemStateChanged(null); + viewerActionMenu.add(alpanels); + viewerActionMenu.addMenuListener(new MenuListener() + { - return; + @Override + public void menuSelected(MenuEvent e) + { + handler.itemStateChanged(null); } - } - // ///////////////////////////////// - // Check if there are other Jmol views involving this alignment - // and prompt user about adding this molecule to one of them - Vector existingViews = getJmolsFor(ap); - if (existingViews.size() > 0) - { - Enumeration jm = existingViews.elements(); - while (jm.hasMoreElements()) - { - AppJmol topJmol = (AppJmol) jm.nextElement(); - // TODO: highlight topJmol in view somehow - int option = JOptionPane.showInternalConfirmDialog(Desktop.desktop, - "Do you want to add " + pdbentry.getId() - + " to the view called\n'" + topJmol.getTitle() - + "'\n", "Align to existing structure view", - JOptionPane.YES_NO_OPTION); - if (option == JOptionPane.YES_OPTION) - { - topJmol.addStructure(pdbentry, seq, chains, true, ap.alignFrame); - return; - } + + @Override + public void menuDeselected(MenuEvent e) + { + // TODO Auto-generated method stub + } - } - // ///////////////////////////////// - jmb = new AppJmolBinding(this, new PDBEntry[] - { pdbentry }, new SequenceI[][] - { seq }, null, null); - this.ap = ap; - setSize(400, 400); // probably should be a configurable/dynamic default here + @Override + public void menuCanceled(MenuEvent e) + { + // TODO Auto-generated method stub + + } + }); + } + + IProgressIndicator progressBar = null; - if (pdbentry.getFile() != null) + /** + * add a single PDB structure to a new or existing Jmol view + * + * @param pdbentry + * @param seq + * @param chains + * @param ap + */ + public AppJmol(PDBEntry pdbentry, SequenceI[] seq, String[] chains, + final AlignmentPanel ap) + { + progressBar = ap.alignFrame; + String pdbId = pdbentry.getId(); + + /* + * If the PDB file is already loaded, the user may just choose to add to an + * existing viewer (or cancel) + */ + if (addAlreadyLoadedFile(seq, chains, ap, pdbId)) { - initJmol("load \"" + pdbentry.getFile() + "\""); + return; } - else + + /* + * Check if there are other Jmol views involving this alignment and prompt + * user about adding this molecule to one of them + */ + if (addToExistingViewer(pdbentry, seq, chains, ap, pdbId)) { - addingStructures = false; - worker = new Thread(this); - worker.start(); + return; } + /* + * If the options above are declined or do not apply, open a new viewer + */ + openNewJmol(ap, new PDBEntry[] { pdbentry }, new SequenceI[][] { seq }); + } + + /** + * Answers true if this viewer already involves the given PDB ID + */ + @Override + protected boolean hasPdbId(String pdbId) + { + return jmb.hasPdbId(pdbId); + } + + private void openNewJmol(AlignmentPanel ap, PDBEntry[] pdbentrys, + SequenceI[][] seqs) + { + progressBar = ap.alignFrame; + jmb = new AppJmolBinding(this, ap.getStructureSelectionManager(), + pdbentrys, seqs, null, null); + addAlignmentPanel(ap); + useAlignmentPanelForColourbyseq(ap); + if (pdbentrys.length > 1) + { + alignAddedStructures = true; + useAlignmentPanelForSuperposition(ap); + } + jmb.setColourBySequence(true); + setSize(400, 400); // probably should be a configurable/dynamic default here + initMenus(); + addingStructures = false; + worker = new Thread(this); + worker.start(); + this.addInternalFrameListener(new InternalFrameAdapter() { + @Override public void internalFrameClosing(InternalFrameEvent internalFrameEvent) { - closeViewer(); + closeViewer(false); } }); } /** - * pdb retrieval thread. - */ - private Thread worker = null; - - /** - * add a new structure (with associated sequences and chains) to this viewer, - * retrieving it if necessary first. + * create a new Jmol containing several structures superimposed using the + * given alignPanel. * - * @param pdbentry - * @param seq - * @param chains - * @param alignFrame - * @param align - * if true, new structure(s) will be align using associated alignment + * @param ap + * @param pe + * @param seqs */ - private void addStructure(final PDBEntry pdbentry, final SequenceI[] seq, - final String[] chains, final boolean b, final IProgressIndicator alignFrame) + public AppJmol(AlignmentPanel ap, PDBEntry[] pe, SequenceI[][] seqs) { - if (pdbentry.getFile() == null) - { - if (worker != null && worker.isAlive()) - { - // a retrieval is in progress, wait around and add ourselves to the - // queue. - new Thread(new Runnable() - { - public void run() - { - while (worker != null && worker.isAlive() && _started) - { - try - { - Thread.sleep(100 + ((int) Math.random() * 100)); - - } catch (Exception e) - { - } - - } - // and call ourselves again. - addStructure(pdbentry, seq, chains, b, alignFrame); - } - }).start(); - return; - } - } - // otherwise, start adding the structure. - jmb.addSequenceAndChain(new PDBEntry[] - { pdbentry }, new SequenceI[][] - { seq }, new String[][] - { chains }); - addingStructures = true; - _started = false; - alignAddedStructures = b; - progressBar = alignFrame; // visual indication happens on caller frame. - (worker = new Thread(this)).start(); - return; + openNewJmol(ap, pe, seqs); } - private Vector getJmolsFor(AlignmentPanel ap2) + /** + * Returns a list of any Jmol viewers. The list is restricted to those linked + * to the given alignment panel if it is not null. + */ + @Override + protected List getViewersFor(AlignmentPanel apanel) { - Vector otherJmols = new Vector(); - // Now this AppJmol is mapped to new sequences. We must add them to - // the exisiting array + List result = new ArrayList(); JInternalFrame[] frames = Desktop.instance.getAllFrames(); - for (int i = 0; i < frames.length; i++) + for (JInternalFrame frame : frames) { - if (frames[i] instanceof AppJmol) + if (frame instanceof AppJmol) { - AppJmol topJmol = ((AppJmol) frames[i]); - if (topJmol.ap == ap2) + if (apanel == null + || ((StructureViewerBase) frame).isLinkedWith(apanel)) { - otherJmols.addElement(topJmol); + result.add((StructureViewerBase) frame); } } } - return otherJmols; + return result; } void initJmol(String command) @@ -339,31 +379,42 @@ public class AppJmol extends GStructureViewer implements Runnable, bl.setHgap(0); bl.setVgap(0); scriptWindow = new JPanel(bl); - }; - - jmb.allocateViewer(renderPanel, true, "", null, null, "", scriptWindow, null); - jmb.newJmolPopup(true, "Jmol", true); + scriptWindow.setVisible(false); + } + + jmb.allocateViewer(renderPanel, true, "", null, null, "", + scriptWindow, null); + // jmb.newJmolPopup("Jmol"); + if (command == null) + { + command = ""; + } jmb.evalStateCommand(command); + jmb.evalStateCommand("set hoverDelay=0.1"); jmb.setFinishedInit(true); } - void setChainMenuItems(Vector chains) + void setChainMenuItems(Vector chains) { chainMenu.removeAll(); if (chains == null) { return; } - JMenuItem menuItem = new JMenuItem("All"); + JMenuItem menuItem = new JMenuItem( + MessageManager.getString("label.all")); menuItem.addActionListener(new ActionListener() { + @Override public void actionPerformed(ActionEvent evt) { allChainsSelected = true; for (int i = 0; i < chainMenu.getItemCount(); i++) { if (chainMenu.getItem(i) instanceof JCheckBoxMenuItem) + { ((JCheckBoxMenuItem) chainMenu.getItem(i)).setSelected(true); + } } centerViewer(); allChainsSelected = false; @@ -372,15 +423,18 @@ public class AppJmol extends GStructureViewer implements Runnable, chainMenu.add(menuItem); - for (int c = 0; c < chains.size(); c++) + for (String chain : chains) { - menuItem = new JCheckBoxMenuItem(chains.elementAt(c).toString(), true); + menuItem = new JCheckBoxMenuItem(chain, true); menuItem.addItemListener(new ItemListener() { + @Override public void itemStateChanged(ItemEvent evt) { if (!allChainsSelected) + { centerViewer(); + } } }); @@ -390,13 +444,9 @@ public class AppJmol extends GStructureViewer implements Runnable, boolean allChainsSelected = false; - private boolean alignAddedStructures = false; - void centerViewer() { - Vector toshow = new Vector(); - String lbl; - int mlength, p, mnum; + Vector toshow = new Vector(); for (int i = 0; i < chainMenu.getItemCount(); i++) { if (chainMenu.getItem(i) instanceof JCheckBoxMenuItem) @@ -411,82 +461,244 @@ public class AppJmol extends GStructureViewer implements Runnable, jmb.centerViewer(toshow); } - void closeViewer() + @Override + public void closeViewer(boolean closeExternalViewer) { - jmb.closeViewer(); + // Jmol does not use an external viewer + if (jmb != null) + { + jmb.closeViewer(); + } + setAlignmentPanel(null); + _aps.clear(); + _alignwith.clear(); + _colourwith.clear(); // TODO: check for memory leaks where instance isn't finalised because jmb // holds a reference to the window jmb = null; } + @Override + public void run() + { + _started = true; + try + { + List files = fetchPdbFiles(); + if (files.size() > 0) + { + showFilesInViewer(files); + } + } finally + { + _started = false; + worker = null; + } + } + /** - * state flag for PDB retrieval thread + * Either adds the given files to a structure viewer or opens a new viewer to + * show them + * + * @param files + * list of absolute paths to structure files */ - private boolean _started = false; + void showFilesInViewer(List files) + { + long lastnotify = jmb.getLoadNotifiesHandled(); + StringBuilder fileList = new StringBuilder(); + for (String s : files) + { + fileList.append(SPACE).append(BACKSLASH) + .append(Platform.escapeString(s)).append(BACKSLASH); + } + String filesString = fileList.toString(); - public void run() + if (!addingStructures) + { + try + { + initJmol("load FILES " + filesString); + } catch (OutOfMemoryError oomerror) + { + new OOMWarning("When trying to open the Jmol viewer!", oomerror); + Cache.log.debug("File locations are " + filesString); + } catch (Exception ex) + { + Cache.log.error("Couldn't open Jmol viewer!", ex); + } + } + else + { + StringBuilder cmd = new StringBuilder(); + cmd.append("loadingJalviewdata=true\nload APPEND "); + cmd.append(filesString); + cmd.append("\nloadingJalviewdata=null"); + final String command = cmd.toString(); + lastnotify = jmb.getLoadNotifiesHandled(); + + try + { + jmb.evalStateCommand(command); + } catch (OutOfMemoryError oomerror) + { + new OOMWarning("When trying to add structures to the Jmol viewer!", + oomerror); + Cache.log.debug("File locations are " + filesString); + } catch (Exception ex) + { + Cache.log.error("Couldn't add files to Jmol viewer!", ex); + } + } + + // need to wait around until script has finished + int waitMax = JMOL_LOAD_TIMEOUT; + int waitFor = 35; + int waitTotal = 0; + while (addingStructures ? lastnotify >= jmb.getLoadNotifiesHandled() + : !(jmb.isFinishedInit() && jmb.getPdbFile() != null && jmb + .getPdbFile().length == files.size())) + { + try + { + Cache.log.debug("Waiting around for jmb notify."); + Thread.sleep(waitFor); + waitTotal += waitFor; + } catch (Exception e) + { + } + if (waitTotal > waitMax) + { + System.err + .println("Timed out waiting for Jmol to load files after " + + waitTotal + "ms"); +// System.err.println("finished: " + jmb.isFinishedInit() +// + "; loaded: " + Arrays.toString(jmb.getPdbFile()) +// + "; files: " + files.toString()); + jmb.getPdbFile(); + break; + } + } + + // refresh the sequence colours for the new structure(s) + for (AlignmentPanel ap : _colourwith) + { + jmb.updateColours(ap); + } + // do superposition if asked to + if (Cache.getDefault("AUTOSUPERIMPOSE", true) && alignAddedStructures) + { + alignAddedStructures(); + } + addingStructures = false; + } + + /** + * Queues a thread to align structures with Jalview alignments + */ + void alignAddedStructures() { - _started = true; + javax.swing.SwingUtilities.invokeLater(new Runnable() + { + @Override + public void run() + { + if (jmb.viewer.isScriptExecuting()) + { + SwingUtilities.invokeLater(this); + try + { + Thread.sleep(5); + } catch (InterruptedException q) + { + } + return; + } + else + { + alignStructs_withAllAlignPanels(); + } + } + }); + alignAddedStructures = false; + } + + /** + * Retrieves and saves as file any modelled PDB entries for which we do not + * already have a file saved. Returns a list of absolute paths to structure + * files which were either retrieved, or already stored but not modelled in + * the structure viewer (i.e. files to add to the viewer display). + * + * @return + */ + List fetchPdbFiles() + { + // todo - record which pdbids were successfully imported. + StringBuilder errormsgs = new StringBuilder(); + + List files = new ArrayList(); String pdbid = ""; - // todo - record which pdbids were successfuly imported. - StringBuffer errormsgs = new StringBuffer(), files = new StringBuffer(); try { - String[] curfiles = jmb.getPdbFile(); // files currently in viewer + String[] filesInViewer = jmb.getPdbFile(); // TODO: replace with reference fetching/transfer code (validate PDBentry // as a DBRef?) - jalview.ws.dbsources.Pdb pdbclient = new jalview.ws.dbsources.Pdb(); - for (int pi = 0; pi < jmb.pdbentry.length; pi++) + Pdb pdbclient = new Pdb(); + for (int pi = 0; pi < jmb.getPdbCount(); pi++) { - String file = jmb.pdbentry[pi].getFile(); + String file = jmb.getPdbEntry(pi).getFile(); if (file == null) { // retrieve the pdb and store it locally AlignmentI pdbseq = null; - pdbid = jmb.pdbentry[pi].getId(); - long hdl=pdbid.hashCode()-System.currentTimeMillis(); + pdbid = jmb.getPdbEntry(pi).getId(); + long hdl = pdbid.hashCode() - System.currentTimeMillis(); if (progressBar != null) { - progressBar.setProgressBar("Fetching PDB " + pdbid, - hdl); + progressBar.setProgressBar(MessageManager.formatMessage( + "status.fetching_pdb", new String[] { pdbid }), hdl); } try { - pdbseq = pdbclient.getSequenceRecords(pdbid = jmb.pdbentry[pi] - .getId()); + pdbseq = pdbclient.getSequenceRecords(pdbid); } catch (OutOfMemoryError oomerror) { new OOMWarning("Retrieving PDB id " + pdbid, oomerror); } catch (Exception ex) { ex.printStackTrace(); - errormsgs.append("'" + pdbid + "'"); - } - if (progressBar != null) + errormsgs.append("'").append(pdbid).append("'"); + } finally { - progressBar.setProgressBar("Finished.", hdl); + if (progressBar != null) + { + progressBar.setProgressBar( + MessageManager.getString("label.state_completed"), + hdl); + } } if (pdbseq != null) { // just transfer the file name from the first sequence's first // PDBEntry - jmb.pdbentry[pi].setFile(file = ((PDBEntry) pdbseq - .getSequenceAt(0).getPDBId().elementAt(0)).getFile()); - files.append(" \"" + file + "\""); + file = new File(pdbseq.getSequenceAt(0).getAllPDBEntries() + .elementAt(0).getFile()).getAbsolutePath(); + jmb.getPdbEntry(pi).setFile(file); + files.add(file); } else { - errormsgs.append("'" + pdbid + "' "); + errormsgs.append("'").append(pdbid).append("' "); } } else { - if (curfiles != null && curfiles.length > 0) + if (filesInViewer != null && filesInViewer.length > 0) { addingStructures = true; // already files loaded. - for (int c = 0; c < curfiles.length; c++) + for (int c = 0; c < filesInViewer.length; c++) { - if (curfiles[c].equals(file)) + if (filesInViewer[c].equals(file)) { file = null; break; @@ -495,7 +707,7 @@ public class AppJmol extends GStructureViewer implements Runnable, } if (file != null) { - files.append(" \"" + file + "\""); + files.add(file); } } } @@ -505,107 +717,39 @@ public class AppJmol extends GStructureViewer implements Runnable, } catch (Exception ex) { ex.printStackTrace(); - errormsgs.append("When retrieving pdbfiles : current was: '" + pdbid - + "'"); + errormsgs.append("When retrieving pdbfiles : current was: '") + .append(pdbid).append("'"); } if (errormsgs.length() > 0) { - - JOptionPane.showInternalMessageDialog(Desktop.desktop, - "The following pdb entries could not be retrieved from the PDB:\n" - + errormsgs.toString() - + "\nPlease try downloading them manually.", - "Couldn't load file", JOptionPane.ERROR_MESSAGE); - + JOptionPane.showInternalMessageDialog(Desktop.desktop, MessageManager + .formatMessage("label.pdb_entries_couldnt_be_retrieved", + new String[] { errormsgs.toString() }), + MessageManager.getString("label.couldnt_load_file"), + JOptionPane.ERROR_MESSAGE); } - if (files.length() > 0) - { - if (!addingStructures) - { - - try - { - initJmol("load FILES " + files.toString()); - } catch (OutOfMemoryError oomerror) - { - new OOMWarning("When trying to open the Jmol viewer!", oomerror); - Cache.log.debug("File locations are " + files); - } catch (Exception ex) - { - Cache.log.error("Couldn't open Jmol viewer!", ex); - } - } - else - { - StringBuffer cmd = new StringBuffer(); - cmd.append("loadingJalviewdata=true\nload APPEND "); - cmd.append(files.toString()); - cmd.append("\nloadingJalviewdata=null"); - final String command = cmd.toString(); - cmd = null; - try - { - jmb.evalStateCommand(command); - } catch (OutOfMemoryError oomerror) - { - new OOMWarning( - "When trying to add structures to the Jmol viewer!", - oomerror); - Cache.log.debug("File locations are " + files); - } catch (Exception ex) - { - Cache.log.error("Couldn't add files to Jmol viewer!", ex); - } - long lastnotify=jmb.getLoadNotifiesHandled(); - // need to wait around until script has finished - while (lastnotify>=jmb.getLoadNotifiesHandled()); - { - try - { - Thread.sleep(35); - } catch (Exception e) - { - } - } - // refresh the sequence colours for the new structure(s) - jmb.updateColours(ap); - // do superposition if asked to - if (alignAddedStructures) - { - javax.swing.SwingUtilities.invokeLater(new Runnable() - { - public void run() - { - jmb.superposeStructures(ap.av.getAlignment(), -1, null); - } - }); - alignAddedStructures = false; - } - addingStructures = false; - } - } - _started = false; - worker = null; + return files; } + @Override public void pdbFile_actionPerformed(ActionEvent actionEvent) { JalviewFileChooser chooser = new JalviewFileChooser( jalview.bin.Cache.getProperty("LAST_DIRECTORY")); chooser.setFileView(new JalviewFileView()); - chooser.setDialogTitle("Save PDB File"); - chooser.setToolTipText("Save"); + chooser.setDialogTitle(MessageManager.getString("label.save_pdb_file")); + chooser.setToolTipText(MessageManager.getString("action.save")); int value = chooser.showSaveDialog(this); if (value == JalviewFileChooser.APPROVE_OPTION) { + BufferedReader in = null; try { // TODO: cope with multiple PDB files in view - BufferedReader in = new BufferedReader(new FileReader( - jmb.getPdbFile()[0])); + in = new BufferedReader(new FileReader(jmb.getPdbFile()[0])); File outFile = chooser.getSelectedFile(); PrintWriter out = new PrintWriter(new FileOutputStream(outFile)); @@ -621,22 +765,29 @@ public class AppJmol extends GStructureViewer implements Runnable, } catch (Exception ex) { ex.printStackTrace(); + } finally + { + if (in != null) + { + try + { + in.close(); + } catch (IOException e) + { + // ignore + } + } } } } + @Override public void viewMapping_actionPerformed(ActionEvent actionEvent) { jalview.gui.CutAndPasteTransfer cap = new jalview.gui.CutAndPasteTransfer(); try { - for (int pdbe = 0; pdbe < jmb.pdbentry.length; pdbe++) - { - cap.appendText(StructureSelectionManager - .getStructureSelectionManager().printMapping( - jmb.pdbentry[pdbe].getFile())); - cap.appendText("\n"); - } + cap.appendText(jmb.printMappings()); } catch (OutOfMemoryError e) { new OOMWarning( @@ -645,138 +796,183 @@ public class AppJmol extends GStructureViewer implements Runnable, cap.dispose(); return; } - jalview.gui.Desktop.addInternalFrame(cap, "PDB - Sequence Mapping", - 550, 600); + jalview.gui.Desktop.addInternalFrame(cap, + MessageManager.getString("label.pdb_sequence_mapping"), 550, + 600); } - /** - * DOCUMENT ME! - * - * @param e - * DOCUMENT ME! - */ + @Override public void eps_actionPerformed(ActionEvent e) { - makePDBImage(jalview.util.ImageMaker.EPS); + makePDBImage(jalview.util.ImageMaker.TYPE.EPS); } - /** - * DOCUMENT ME! - * - * @param e - * DOCUMENT ME! - */ + @Override public void png_actionPerformed(ActionEvent e) { - makePDBImage(jalview.util.ImageMaker.PNG); + makePDBImage(jalview.util.ImageMaker.TYPE.PNG); } - void makePDBImage(int type) + void makePDBImage(jalview.util.ImageMaker.TYPE type) { int width = getWidth(); int height = getHeight(); jalview.util.ImageMaker im; - if (type == jalview.util.ImageMaker.PNG) + if (type == jalview.util.ImageMaker.TYPE.PNG) { - im = new jalview.util.ImageMaker(this, jalview.util.ImageMaker.PNG, - "Make PNG image from view", width, height, null, null); + im = new jalview.util.ImageMaker(this, + jalview.util.ImageMaker.TYPE.PNG, "Make PNG image from view", + width, height, null, null, null, 0, false); + } + else if (type == jalview.util.ImageMaker.TYPE.EPS) + { + im = new jalview.util.ImageMaker(this, + jalview.util.ImageMaker.TYPE.EPS, "Make EPS file from view", + width, height, null, this.getTitle(), null, 0, false); } else { - im = new jalview.util.ImageMaker(this, jalview.util.ImageMaker.EPS, - "Make EPS file from view", width, height, null, - this.getTitle()); + + im = new jalview.util.ImageMaker(this, + jalview.util.ImageMaker.TYPE.SVG, "Make SVG file from PCA", + width, height, null, this.getTitle(), null, 0, false); } if (im.getGraphics() != null) { - Rectangle rect = new Rectangle(width, height); - jmb.viewer.renderScreenImage(im.getGraphics(), rect.getSize(), rect); + jmb.viewer.renderScreenImage(im.getGraphics(), width, height); im.writeImage(); } } + @Override + public void viewerColour_actionPerformed(ActionEvent actionEvent) + { + if (viewerColour.isSelected()) + { + // disable automatic sequence colouring. + jmb.setColourBySequence(false); + } + } + + @Override public void seqColour_actionPerformed(ActionEvent actionEvent) { jmb.setColourBySequence(seqColour.isSelected()); - // Set the colour using the current view for the associated alignframe - jmb.colourBySequence(ap.alignFrame.viewport.showSequenceFeatures, - ap.alignFrame.viewport.alignment); + if (_colourwith == null) + { + _colourwith = new Vector(); + } + if (jmb.isColourBySequence()) + { + if (!jmb.isLoadingFromArchive()) + { + if (_colourwith.size() == 0 && getAlignmentPanel() != null) + { + // Make the currently displayed alignment panel the associated view + _colourwith.add(getAlignmentPanel().alignFrame.alignPanel); + } + } + // Set the colour using the current view for the associated alignframe + for (AlignmentPanel ap : _colourwith) + { + jmb.colourBySequence(ap); + } + } } + @Override public void chainColour_actionPerformed(ActionEvent actionEvent) { chainColour.setSelected(true); jmb.colourByChain(); } + @Override public void chargeColour_actionPerformed(ActionEvent actionEvent) { chargeColour.setSelected(true); jmb.colourByCharge(); } + @Override public void zappoColour_actionPerformed(ActionEvent actionEvent) { zappoColour.setSelected(true); jmb.setJalviewColourScheme(new ZappoColourScheme()); } + @Override public void taylorColour_actionPerformed(ActionEvent actionEvent) { taylorColour.setSelected(true); jmb.setJalviewColourScheme(new TaylorColourScheme()); } + @Override public void hydroColour_actionPerformed(ActionEvent actionEvent) { hydroColour.setSelected(true); jmb.setJalviewColourScheme(new HydrophobicColourScheme()); } + @Override public void helixColour_actionPerformed(ActionEvent actionEvent) { helixColour.setSelected(true); jmb.setJalviewColourScheme(new HelixColourScheme()); } + @Override public void strandColour_actionPerformed(ActionEvent actionEvent) { strandColour.setSelected(true); jmb.setJalviewColourScheme(new StrandColourScheme()); } + @Override public void turnColour_actionPerformed(ActionEvent actionEvent) { turnColour.setSelected(true); jmb.setJalviewColourScheme(new TurnColourScheme()); } + @Override public void buriedColour_actionPerformed(ActionEvent actionEvent) { buriedColour.setSelected(true); jmb.setJalviewColourScheme(new BuriedColourScheme()); } + @Override + public void purinePyrimidineColour_actionPerformed(ActionEvent actionEvent) + { + setJalviewColourScheme(new PurinePyrimidineColourScheme()); + } + + @Override public void userColour_actionPerformed(ActionEvent actionEvent) { userColour.setSelected(true); new UserDefinedColours(this, null); } + @Override public void backGround_actionPerformed(ActionEvent actionEvent) { - java.awt.Color col = JColorChooser.showDialog(this, - "Select Background Colour", null); + java.awt.Color col = JColorChooser + .showDialog(this, MessageManager + .getString("label.select_backgroud_colour"), null); if (col != null) { jmb.setBackgroundColour(col); } } - public void jmolHelp_actionPerformed(ActionEvent actionEvent) + @Override + public void showHelp_actionPerformed(ActionEvent actionEvent) { try { @@ -786,6 +982,7 @@ public class AppJmol extends GStructureViewer implements Runnable, { } } + public void showConsole(boolean showConsole) { @@ -807,9 +1004,9 @@ public class AppJmol extends GStructureViewer implements Runnable, else { if (splitPane != null) - { - splitPane.setVisible(false); - } + { + splitPane.setVisible(false); + } splitPane = null; @@ -823,31 +1020,30 @@ public class AppJmol extends GStructureViewer implements Runnable, { final Dimension currentSize = new Dimension(); - final Rectangle rectClip = new Rectangle(); - + @Override public void paintComponent(Graphics g) { getSize(currentSize); - g.getClipBounds(rectClip); - if (jmb.fileLoadingError != null) + if (jmb != null && jmb.fileLoadingError != null) { g.setColor(Color.black); g.fillRect(0, 0, currentSize.width, currentSize.height); g.setColor(Color.white); g.setFont(new Font("Verdana", Font.BOLD, 14)); - g.drawString("Error loading file...", 20, currentSize.height / 2); + g.drawString(MessageManager.getString("label.error_loading_file") + + "...", 20, currentSize.height / 2); StringBuffer sb = new StringBuffer(); int lines = 0; - for (int e = 0; e < jmb.pdbentry.length; e++) + for (int e = 0; e < jmb.getPdbCount(); e++) { - sb.append(jmb.pdbentry[e].getId()); - if (e < jmb.pdbentry.length - 1) + sb.append(jmb.getPdbEntry(e).getId()); + if (e < jmb.getPdbCount() - 1) { sb.append(","); } - if (e == jmb.pdbentry.length - 1 || sb.length() > 20) + if (e == jmb.getPdbCount() - 1 || sb.length() > 20) { lines++; g.drawString(sb.toString(), 20, currentSize.height / 2 - lines @@ -861,26 +1057,17 @@ public class AppJmol extends GStructureViewer implements Runnable, g.fillRect(0, 0, currentSize.width, currentSize.height); g.setColor(Color.white); g.setFont(new Font("Verdana", Font.BOLD, 14)); - g.drawString("Retrieving PDB data....", 20, currentSize.height / 2); + g.drawString(MessageManager.getString("label.retrieving_pdb_data"), + 20, currentSize.height / 2); } else { - jmb.viewer.renderScreenImage(g, currentSize, rectClip); + jmb.viewer.renderScreenImage(g, currentSize.width, + currentSize.height); } } } - String viewId = null; - - public String getViewId() - { - if (viewId == null) - { - viewId = System.currentTimeMillis() + "." + this.hashCode(); - } - return viewId; - } - public void updateTitleAndMenus() { if (jmb.fileLoadingError != null && jmb.fileLoadingError.length() > 0) @@ -889,12 +1076,15 @@ public class AppJmol extends GStructureViewer implements Runnable, return; } setChainMenuItems(jmb.chainNames); - jmb.colourBySequence(ap.av.getShowSequenceFeatures(), ap.av.alignment); this.setTitle(jmb.getViewerTitle()); - if (jmb.getPdbFile().length > 1 && jmb.sequence.length > 1) + if (jmb.getPdbFile().length > 1 && jmb.getSequence().length > 1) + { + viewerActionMenu.setVisible(true); + } + if (!jmb.isLoadingFromArchive()) { - jmolActionMenu.setVisible(true); + seqColour_actionPerformed(null); } } @@ -908,23 +1098,96 @@ public class AppJmol extends GStructureViewer implements Runnable, @Override protected void alignStructs_actionPerformed(ActionEvent actionEvent) { + alignStructs_withAllAlignPanels(); + } + private void alignStructs_withAllAlignPanels() + { + if (getAlignmentPanel() == null) + { + return; + } + ; + if (_alignwith.size() == 0) + { + _alignwith.add(getAlignmentPanel()); + } + ; try { - jmb.superposeStructures(ap.av.getAlignment(), -1, - ap.av.getColumnSelection()); + AlignmentI[] als = new Alignment[_alignwith.size()]; + ColumnSelection[] alc = new ColumnSelection[_alignwith.size()]; + int[] alm = new int[_alignwith.size()]; + int a = 0; + + for (AlignmentPanel ap : _alignwith) + { + als[a] = ap.av.getAlignment(); + alm[a] = -1; + alc[a++] = ap.av.getColumnSelection(); + } + jmb.superposeStructures(als, alm, alc); } catch (Exception e) { - Cache.log.info("Couldn't align structures in alignframe " - + ap.alignFrame.getTitle(), e); + StringBuffer sp = new StringBuffer(); + for (AlignmentPanel ap : _alignwith) + { + sp.append("'" + ap.alignFrame.getTitle() + "' "); + } + Cache.log.info("Couldn't align structures with the " + sp.toString() + + "associated alignment panels.", e); } + } + @Override public void setJalviewColourScheme(ColourSchemeI ucs) { jmb.setJalviewColourScheme(ucs); } + /** + * + * @param alignment + * @return first alignment panel displaying given alignment, or the default + * alignment panel + */ + public AlignmentPanel getAlignmentPanelFor(AlignmentI alignment) + { + for (AlignmentPanel ap : getAllAlignmentPanels()) + { + if (ap.av.getAlignment() == alignment) + { + return ap; + } + } + return getAlignmentPanel(); + } + + @Override + public AAStructureBindingModel getBinding() + { + return this.jmb; + } + + @Override + public String getStateInfo() + { + return jmb == null ? null : jmb.viewer.getStateInfo(); + } + + @Override + public ViewerType getViewerType() + { + return ViewerType.JMOL; + } + + @Override + protected AAStructureBindingModel getBindingModel() + { + return jmb; + } + }