src/jalview/ext/jmol/JalviewJmolBinding.java

   1 /*
   2  * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
   3  * Copyright (C) $$Year-Rel$$ 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.ext.jmol;
  22
  23 import jalview.api.AlignmentViewPanel;
  24 import jalview.api.FeatureRenderer;
  25 import jalview.api.SequenceRenderer;
  26 import jalview.datamodel.AlignmentI;
  27 import jalview.datamodel.HiddenColumns;
  28 import jalview.datamodel.PDBEntry;
  29 import jalview.datamodel.SequenceI;
  30 import jalview.gui.IProgressIndicator;
  31 import jalview.io.DataSourceType;
  32 import jalview.io.StructureFile;
  33 import jalview.schemes.ColourSchemeI;
  34 import jalview.schemes.ResidueProperties;
  35 import jalview.structure.AtomSpec;
  36 import jalview.structure.StructureMappingcommandSet;
  37 import jalview.structure.StructureSelectionManager;
  38 import jalview.structures.models.AAStructureBindingModel;
  39 import jalview.util.MessageManager;
  40
  41 import java.awt.Color;
  42 import java.awt.Container;
  43 import java.awt.event.ComponentEvent;
  44 import java.awt.event.ComponentListener;
  45 import java.io.File;
  46 import java.net.URL;
  47 import java.security.AccessControlException;
  48 import java.util.ArrayList;
  49 import java.util.BitSet;
  50 import java.util.Hashtable;
  51 import java.util.List;
  52 import java.util.Map;
  53 import java.util.Vector;
  54
  55 import org.jmol.adapter.smarter.SmarterJmolAdapter;
  56 import org.jmol.api.JmolAppConsoleInterface;
  57 import org.jmol.api.JmolSelectionListener;
  58 import org.jmol.api.JmolStatusListener;
  59 import org.jmol.api.JmolViewer;
  60 import org.jmol.c.CBK;
  61 import org.jmol.script.T;
  62 import org.jmol.viewer.Viewer;
  63
  64 public abstract class JalviewJmolBinding extends AAStructureBindingModel
  65         implements JmolStatusListener, JmolSelectionListener,
  66         ComponentListener
  67 {
  68   boolean allChainsSelected = false;
  69
  70   /*
  71    * when true, try to search the associated datamodel for sequences that are
  72    * associated with any unknown structures in the Jmol view.
  73    */
  74   private boolean associateNewStructs = false;
  75
  76   Vector<String> atomsPicked = new Vector<>();
  77
  78   private List<String> chainNames;
  79
  80   Hashtable<String, String> chainFile;
  81
  82   /*
  83    * the default or current model displayed if the model cannot be identified
  84    * from the selection message
  85    */
  86   int frameNo = 0;
  87
  88   // protected JmolGenericPopup jmolpopup; // not used - remove?
  89
  90   String lastCommand;
  91
  92   String lastMessage;
  93
  94   boolean loadedInline;
  95
  96   StringBuffer resetLastRes = new StringBuffer();
  97
  98   public Viewer viewer;
  99
 100   public JalviewJmolBinding(StructureSelectionManager ssm,
 101           PDBEntry[] pdbentry, SequenceI[][] sequenceIs,
 102           DataSourceType protocol)
 103   {
 104     super(ssm, pdbentry, sequenceIs, protocol);
 105     /*
 106      * viewer = JmolViewer.allocateViewer(renderPanel, new SmarterJmolAdapter(),
 107      * "jalviewJmol", ap.av.applet .getDocumentBase(),
 108      * ap.av.applet.getCodeBase(), "", this);
 109      *
 110      * jmolpopup = JmolPopup.newJmolPopup(viewer, true, "Jmol", true);
 111      */
 112   }
 113
 114   public JalviewJmolBinding(StructureSelectionManager ssm,
 115           SequenceI[][] seqs, Viewer theViewer)
 116   {
 117     super(ssm, seqs);
 118
 119     viewer = theViewer;
 120     viewer.setJmolStatusListener(this);
 121     viewer.addSelectionListener(this);
 122   }
 123
 124   /**
 125    * construct a title string for the viewer window based on the data jalview
 126    * knows about
 127    *
 128    * @return
 129    */
 130   public String getViewerTitle()
 131   {
 132     return getViewerTitle("Jmol", true);
 133   }
 134
 135   /**
 136    * prepare the view for a given set of models/chains. chainList contains
 137    * strings of the form 'pdbfilename:Chaincode'
 138    *
 139    * @param chainList
 140    *          list of chains to make visible
 141    */
 142   public void centerViewer(Vector<String> chainList)
 143   {
 144     StringBuilder cmd = new StringBuilder(128);
 145     int mlength, p;
 146     for (String lbl : chainList)
 147     {
 148       mlength = 0;
 149       do
 150       {
 151         p = mlength;
 152         mlength = lbl.indexOf(":", p);
 153       } while (p < mlength && mlength < (lbl.length() - 2));
 154       // TODO: lookup each pdb id and recover proper model number for it.
 155       cmd.append(":" + lbl.substring(mlength + 1) + " /"
 156               + (1 + getModelNum(chainFile.get(lbl))) + " or ");
 157     }
 158     if (cmd.length() > 0)
 159     {
 160       cmd.setLength(cmd.length() - 4);
 161     }
 162     evalStateCommand("select *;restrict " + cmd + ";cartoon;center " + cmd);
 163   }
 164
 165   public void closeViewer()
 166   {
 167     // remove listeners for all structures in viewer
 168     getSsm().removeStructureViewerListener(this, this.getStructureFiles());
 169     viewer.dispose();
 170     lastCommand = null;
 171     viewer = null;
 172     releaseUIResources();
 173   }
 174
 175   @Override
 176   public void colourByChain()
 177   {
 178     colourBySequence = false;
 179     // TODO: colour by chain should colour each chain distinctly across all
 180     // visible models
 181     // TODO: http://issues.jalview.org/browse/JAL-628
 182     evalStateCommand("select *;color chain");
 183   }
 184
 185   @Override
 186   public void colourByCharge()
 187   {
 188     colourBySequence = false;
 189     evalStateCommand("select *;color white;select ASP,GLU;color red;"
 190             + "select LYS,ARG;color blue;select CYS;color yellow");
 191   }
 192
 193   /**
 194    * superpose the structures associated with sequences in the alignment
 195    * according to their corresponding positions.
 196    */
 197   public void superposeStructures(AlignmentI alignment)
 198   {
 199     superposeStructures(alignment, -1, null);
 200   }
 201
 202   /**
 203    * superpose the structures associated with sequences in the alignment
 204    * according to their corresponding positions. ded)
 205    *
 206    * @param refStructure
 207    *          - select which pdb file to use as reference (default is -1 - the
 208    *          first structure in the alignment)
 209    */
 210   public void superposeStructures(AlignmentI alignment, int refStructure)
 211   {
 212     superposeStructures(alignment, refStructure, null);
 213   }
 214
 215   /**
 216    * superpose the structures associated with sequences in the alignment
 217    * according to their corresponding positions. ded)
 218    *
 219    * @param refStructure
 220    *          - select which pdb file to use as reference (default is -1 - the
 221    *          first structure in the alignment)
 222    * @param hiddenCols
 223    *          TODO
 224    */
 225   public void superposeStructures(AlignmentI alignment, int refStructure,
 226           HiddenColumns hiddenCols)
 227   {
 228     superposeStructures(new AlignmentI[] { alignment },
 229             new int[]
 230             { refStructure }, new HiddenColumns[] { hiddenCols });
 231   }
 232
 233   /**
 234    * {@inheritDoc}
 235    */
 236   @Override
 237   public String superposeStructures(AlignmentI[] _alignment,
 238           int[] _refStructure, HiddenColumns[] _hiddenCols)
 239   {
 240     while (viewer.isScriptExecuting())
 241     {
 242       try
 243       {
 244         Thread.sleep(10);
 245       } catch (InterruptedException i)
 246       {
 247       }
 248     }
 249
 250     /*
 251      * get the distinct structure files modelled
 252      * (a file with multiple chains may map to multiple sequences)
 253      */
 254     String[] files = getStructureFiles();
 255     if (!waitForFileLoad(files))
 256     {
 257       return null;
 258     }
 259
 260     StringBuilder selectioncom = new StringBuilder(256);
 261     // In principle - nSeconds specifies the speed of animation for each
 262     // superposition - but is seems to behave weirdly, so we don't specify it.
 263     String nSeconds = " ";
 264     if (files.length > 10)
 265     {
 266       nSeconds = " 0.005 ";
 267     }
 268     else
 269     {
 270       nSeconds = " " + (2.0 / files.length) + " ";
 271       // if (nSeconds).substring(0,5)+" ";
 272     }
 273
 274     // see JAL-1345 - should really automatically turn off the animation for
 275     // large numbers of structures, but Jmol doesn't seem to allow that.
 276     // nSeconds = " ";
 277     // union of all aligned positions are collected together.
 278     for (int a = 0; a < _alignment.length; a++)
 279     {
 280       int refStructure = _refStructure[a];
 281       AlignmentI alignment = _alignment[a];
 282       HiddenColumns hiddenCols = _hiddenCols[a];
 283       if (a > 0 && selectioncom.length() > 0 && !selectioncom
 284               .substring(selectioncom.length() - 1).equals("|"))
 285       {
 286         selectioncom.append("|");
 287       }
 288       // process this alignment
 289       if (refStructure >= files.length)
 290       {
 291         System.err.println(
 292                 "Invalid reference structure value " + refStructure);
 293         refStructure = -1;
 294       }
 295
 296       /*
 297        * 'matched' bit j will be set for visible alignment columns j where
 298        * all sequences have a residue with a mapping to the PDB structure
 299        */
 300       BitSet matched = new BitSet();
 301       for (int m = 0; m < alignment.getWidth(); m++)
 302       {
 303         if (hiddenCols == null || hiddenCols.isVisible(m))
 304         {
 305           matched.set(m);
 306         }
 307       }
 308
 309       SuperposeData[] structures = new SuperposeData[files.length];
 310       for (int f = 0; f < files.length; f++)
 311       {
 312         structures[f] = new SuperposeData(alignment.getWidth());
 313       }
 314
 315       /*
 316        * Calculate the superposable alignment columns ('matched'), and the
 317        * corresponding structure residue positions (structures.pdbResNo)
 318        */
 319       int candidateRefStructure = findSuperposableResidues(alignment,
 320               matched, structures);
 321       if (refStructure < 0)
 322       {
 323         /*
 324          * If no reference structure was specified, pick the first one that has
 325          * a mapping in the alignment
 326          */
 327         refStructure = candidateRefStructure;
 328       }
 329
 330       String[] selcom = new String[files.length];
 331       int nmatched = matched.cardinality();
 332       if (nmatched < 4)
 333       {
 334         return (MessageManager.formatMessage("label.insufficient_residues",
 335                 nmatched));
 336       }
 337
 338       /*
 339        * generate select statements to select regions to superimpose structures
 340        */
 341       {
 342         // TODO extract method to construct selection statements
 343         for (int pdbfnum = 0; pdbfnum < files.length; pdbfnum++)
 344         {
 345           String chainCd = ":" + structures[pdbfnum].chain;
 346           int lpos = -1;
 347           boolean run = false;
 348           StringBuilder molsel = new StringBuilder();
 349           molsel.append("{");
 350
 351           int nextColumnMatch = matched.nextSetBit(0);
 352           while (nextColumnMatch != -1)
 353           {
 354             int pdbResNo = structures[pdbfnum].pdbResNo[nextColumnMatch];
 355             if (lpos != pdbResNo - 1)
 356             {
 357               // discontinuity
 358               if (lpos != -1)
 359               {
 360                 molsel.append(lpos);
 361                 molsel.append(chainCd);
 362                 molsel.append("|");
 363               }
 364               run = false;
 365             }
 366             else
 367             {
 368               // continuous run - and lpos >-1
 369               if (!run)
 370               {
 371                 // at the beginning, so add dash
 372                 molsel.append(lpos);
 373                 molsel.append("-");
 374               }
 375               run = true;
 376             }
 377             lpos = pdbResNo;
 378             nextColumnMatch = matched.nextSetBit(nextColumnMatch + 1);
 379           }
 380           /*
 381            * add final selection phrase
 382            */
 383           if (lpos != -1)
 384           {
 385             molsel.append(lpos);
 386             molsel.append(chainCd);
 387             molsel.append("}");
 388           }
 389           if (molsel.length() > 1)
 390           {
 391             selcom[pdbfnum] = molsel.toString();
 392             selectioncom.append("((");
 393             selectioncom.append(selcom[pdbfnum].substring(1,
 394                     selcom[pdbfnum].length() - 1));
 395             selectioncom.append(" )& ");
 396             selectioncom.append(pdbfnum + 1);
 397             selectioncom.append(".1)");
 398             if (pdbfnum < files.length - 1)
 399             {
 400               selectioncom.append("|");
 401             }
 402           }
 403           else
 404           {
 405             selcom[pdbfnum] = null;
 406           }
 407         }
 408       }
 409       StringBuilder command = new StringBuilder(256);
 410       // command.append("set spinFps 10;\n");
 411
 412       for (int pdbfnum = 0; pdbfnum < files.length; pdbfnum++)
 413       {
 414         if (pdbfnum == refStructure || selcom[pdbfnum] == null
 415                 || selcom[refStructure] == null)
 416         {
 417           continue;
 418         }
 419         command.append("echo ");
 420         command.append("\"Superposing (");
 421         command.append(structures[pdbfnum].pdbId);
 422         command.append(") against reference (");
 423         command.append(structures[refStructure].pdbId);
 424         command.append(")\";\ncompare " + nSeconds);
 425         command.append("{");
 426         command.append(Integer.toString(1 + pdbfnum));
 427         command.append(".1} {");
 428         command.append(Integer.toString(1 + refStructure));
 429         // conformation=1 excludes alternate locations for CA (JAL-1757)
 430         command.append(
 431                 ".1} SUBSET {(*.CA | *.P) and conformation=1} ATOMS ");
 432
 433         // for (int s = 0; s < 2; s++)
 434         // {
 435         // command.append(selcom[(s == 0 ? pdbfnum : refStructure)]);
 436         // }
 437         command.append(selcom[pdbfnum]);
 438         command.append(selcom[refStructure]);
 439         command.append(" ROTATE TRANSLATE;\n");
 440       }
 441       if (selectioncom.length() > 0)
 442       {
 443         // TODO is performing selectioncom redundant here? is done later on
 444         // System.out.println("Select regions:\n" + selectioncom.toString());
 445         evalStateCommand("select *; cartoons off; backbone; select ("
 446                 + selectioncom.toString() + "); cartoons; ");
 447         // selcom.append("; ribbons; ");
 448         String cmdString = command.toString();
 449         // System.out.println("Superimpose command(s):\n" + cmdString);
 450
 451         evalStateCommand(cmdString);
 452       }
 453     }
 454     if (selectioncom.length() > 0)
 455     {// finally, mark all regions that were superposed.
 456       if (selectioncom.substring(selectioncom.length() - 1).equals("|"))
 457       {
 458         selectioncom.setLength(selectioncom.length() - 1);
 459       }
 460       // System.out.println("Select regions:\n" + selectioncom.toString());
 461       evalStateCommand("select *; cartoons off; backbone; select ("
 462               + selectioncom.toString() + "); cartoons; ");
 463       // evalStateCommand("select *; backbone; select "+selcom.toString()+";
 464       // cartoons; center "+selcom.toString());
 465     }
 466
 467     return null;
 468   }
 469
 470   public void evalStateCommand(String command)
 471   {
 472     jmolHistory(false);
 473     if (lastCommand == null || !lastCommand.equals(command))
 474     {
 475       viewer.evalStringQuiet(command + "\n");
 476     }
 477     jmolHistory(true);
 478     lastCommand = command;
 479   }
 480
 481   Thread colourby = null;
 482   /**
 483    * Sends a set of colour commands to the structure viewer
 484    *
 485    * @param colourBySequenceCommands
 486    */
 487   @Override
 488   protected void colourBySequence(
 489           final StructureMappingcommandSet[] colourBySequenceCommands)
 490   {
 491     if (colourby != null)
 492     {
 493       colourby.interrupt();
 494       colourby = null;
 495     }
 496     colourby = new Thread(new Runnable()
 497     {
 498       @Override
 499       public void run()
 500       {
 501         for (StructureMappingcommandSet cpdbbyseq : colourBySequenceCommands)
 502         {
 503           for (String cbyseq : cpdbbyseq.commands)
 504           {
 505             executeWhenReady(cbyseq);
 506           }
 507         }
 508       }
 509     });
 510     colourby.start();
 511   }
 512
 513   /**
 514    * @param files
 515    * @param sr
 516    * @param viewPanel
 517    * @return
 518    */
 519   @Override
 520   protected StructureMappingcommandSet[] getColourBySequenceCommands(
 521           String[] files, SequenceRenderer sr, AlignmentViewPanel viewPanel)
 522   {
 523     return JmolCommands.getColourBySequenceCommand(getSsm(), files,
 524             getSequence(), sr, viewPanel);
 525   }
 526
 527   /**
 528    * @param command
 529    */
 530   protected void executeWhenReady(String command)
 531   {
 532     evalStateCommand(command);
 533   }
 534
 535   public void createImage(String file, String type, int quality)
 536   {
 537     System.out.println("JMOL CREATE IMAGE");
 538   }
 539
 540   @Override
 541   public String createImage(String fileName, String type,
 542           Object textOrBytes, int quality)
 543   {
 544     System.out.println("JMOL CREATE IMAGE");
 545     return null;
 546   }
 547
 548   @Override
 549   public String eval(String strEval)
 550   {
 551     // System.out.println(strEval);
 552     // "# 'eval' is implemented only for the applet.";
 553     return null;
 554   }
 555
 556   // End StructureListener
 557   // //////////////////////////
 558
 559   @Override
 560   public float[][] functionXY(String functionName, int x, int y)
 561   {
 562     return null;
 563   }
 564
 565   @Override
 566   public float[][][] functionXYZ(String functionName, int nx, int ny,
 567           int nz)
 568   {
 569     // TODO Auto-generated method stub
 570     return null;
 571   }
 572
 573   public Color getColour(int atomIndex, int pdbResNum, String chain,
 574           String pdbfile)
 575   {
 576     if (getModelNum(pdbfile) < 0)
 577     {
 578       return null;
 579     }
 580     // TODO: verify atomIndex is selecting correct model.
 581     // return new Color(viewer.getAtomArgb(atomIndex)); Jmol 12.2.4
 582     int colour = viewer.ms.at[atomIndex].atomPropertyInt(T.color);
 583     return new Color(colour);
 584   }
 585
 586   /**
 587    * instruct the Jalview binding to update the pdbentries vector if necessary
 588    * prior to matching the jmol view's contents to the list of structure files
 589    * Jalview knows about.
 590    */
 591   public abstract void refreshPdbEntries();
 592
 593   private int getModelNum(String modelFileName)
 594   {
 595     String[] mfn = getStructureFiles();
 596     if (mfn == null)
 597     {
 598       return -1;
 599     }
 600     for (int i = 0; i < mfn.length; i++)
 601     {
 602       if (mfn[i].equalsIgnoreCase(modelFileName))
 603       {
 604         return i;
 605       }
 606     }
 607     return -1;
 608   }
 609
 610   /**
 611    * map between index of model filename returned from getPdbFile and the first
 612    * index of models from this file in the viewer. Note - this is not trimmed -
 613    * use getPdbFile to get number of unique models.
 614    */
 615   private int _modelFileNameMap[];
 616
 617   // ////////////////////////////////
 618   // /StructureListener
 619   // @Override
 620   public synchronized String[] getPdbFilex()
 621   {
 622     if (viewer == null)
 623     {
 624       return new String[0];
 625     }
 626     if (modelFileNames == null)
 627     {
 628       List<String> mset = new ArrayList<>();
 629       _modelFileNameMap = new int[viewer.ms.mc];
 630       String m = viewer.ms.getModelFileName(0);
 631       if (m != null)
 632       {
 633         String filePath = m;
 634         try
 635         {
 636           filePath = new File(m).getAbsolutePath();
 637         } catch (AccessControlException x)
 638         {
 639           // usually not allowed to do this in applet
 640           System.err.println(
 641                   "jmolBinding: Using local file string from Jmol: " + m);
 642         }
 643         if (filePath.indexOf("/file:") != -1)
 644         {
 645           // applet path with docroot - discard as format won't match pdbfile
 646           filePath = m;
 647         }
 648         mset.add(filePath);
 649         _modelFileNameMap[0] = 0; // filename index for first model is always 0.
 650       }
 651       int j = 1;
 652       for (int i = 1; i < viewer.ms.mc; i++)
 653       {
 654         m = viewer.ms.getModelFileName(i);
 655         String filePath = m;
 656         if (m != null)
 657         {
 658           try
 659           {
 660             filePath = new File(m).getAbsolutePath();
 661           } catch (AccessControlException x)
 662           {
 663             // usually not allowed to do this in applet, so keep raw handle
 664             // System.err.println("jmolBinding: Using local file string from
 665             // Jmol: "+m);
 666           }
 667         }
 668
 669         /*
 670          * add this model unless it is read from a structure file we have
 671          * already seen (example: 2MJW is an NMR structure with 10 models)
 672          */
 673         if (!mset.contains(filePath))
 674         {
 675           mset.add(filePath);
 676           _modelFileNameMap[j] = i; // record the model index for the filename
 677           j++;
 678         }
 679       }
 680       modelFileNames = mset.toArray(new String[mset.size()]);
 681     }
 682     return modelFileNames;
 683   }
 684
 685   @Override
 686   public synchronized String[] getStructureFiles()
 687   {
 688     List<String> mset = new ArrayList<>();
 689     if (viewer == null)
 690     {
 691       return new String[0];
 692     }
 693
 694     if (modelFileNames == null)
 695     {
 696       int modelCount = viewer.ms.mc;
 697       String filePath = null;
 698       for (int i = 0; i < modelCount; ++i)
 699       {
 700         filePath = viewer.ms.getModelFileName(i);
 701         if (!mset.contains(filePath))
 702         {
 703           mset.add(filePath);
 704         }
 705       }
 706       modelFileNames = mset.toArray(new String[mset.size()]);
 707     }
 708
 709     return modelFileNames;
 710   }
 711
 712   /**
 713    * map from string to applet
 714    */
 715   @Override
 716   public Map<String, Object> getRegistryInfo()
 717   {
 718     // TODO Auto-generated method stub
 719     return null;
 720   }
 721
 722   // ///////////////////////////////
 723   // JmolStatusListener
 724
 725   public void handlePopupMenu(int x, int y)
 726   {
 727     // jmolpopup.show(x, y);
 728     // jmolpopup.jpiShow(x, y);
 729   }
 730
 731   /**
 732    * Highlight zero, one or more atoms on the structure
 733    */
 734   @Override
 735   public void highlightAtoms(List<AtomSpec> atoms)
 736   {
 737     if (atoms != null)
 738     {
 739       if (resetLastRes.length() > 0)
 740       {
 741         viewer.evalStringQuiet(resetLastRes.toString());
 742         resetLastRes.setLength(0);
 743       }
 744       for (AtomSpec atom : atoms)
 745       {
 746         highlightAtom(atom.getAtomIndex(), atom.getPdbResNum(),
 747                 atom.getChain(), atom.getPdbFile());
 748       }
 749     }
 750   }
 751
 752   // jmol/ssm only
 753   public void highlightAtom(int atomIndex, int pdbResNum, String chain,
 754           String pdbfile)
 755   {
 756     if (modelFileNames == null)
 757     {
 758       return;
 759     }
 760
 761     // look up file model number for this pdbfile
 762     int mdlNum = 0;
 763     // may need to adjust for URLencoding here - we don't worry about that yet.
 764     while (mdlNum < modelFileNames.length
 765             && !pdbfile.equals(modelFileNames[mdlNum]))
 766     {
 767       mdlNum++;
 768     }
 769     if (mdlNum == modelFileNames.length)
 770     {
 771       return;
 772     }
 773
 774     jmolHistory(false);
 775
 776     StringBuilder cmd = new StringBuilder(64);
 777     cmd.append("select " + pdbResNum); // +modelNum
 778
 779     resetLastRes.append("select " + pdbResNum); // +modelNum
 780
 781     cmd.append(":");
 782     resetLastRes.append(":");
 783     if (!chain.equals(" "))
 784     {
 785       cmd.append(chain);
 786       resetLastRes.append(chain);
 787     }
 788     {
 789       cmd.append(" /" + (mdlNum + 1));
 790       resetLastRes.append("/" + (mdlNum + 1));
 791     }
 792     cmd.append(";wireframe 100;" + cmd.toString() + " and not hetero;");
 793
 794     resetLastRes.append(";wireframe 0;" + resetLastRes.toString()
 795             + " and not hetero; spacefill 0;");
 796
 797     cmd.append("spacefill 200;select none");
 798
 799     viewer.evalStringQuiet(cmd.toString());
 800     jmolHistory(true);
 801
 802   }
 803
 804   boolean debug = true;
 805
 806   private void jmolHistory(boolean enable)
 807   {
 808     viewer.evalStringQuiet("History " + ((debug || enable) ? "on" : "off"));
 809   }
 810
 811   public void loadInline(String string)
 812   {
 813     loadedInline = true;
 814     // TODO: re JAL-623
 815     // viewer.loadInline(strModel, isAppend);
 816     // could do this:
 817     // construct fake fullPathName and fileName so we can identify the file
 818     // later.
 819     // Then, construct pass a reader for the string to Jmol.
 820     // ((org.jmol.Viewer.Viewer) viewer).loadModelFromFile(fullPathName,
 821     // fileName, null, reader, false, null, null, 0);
 822     viewer.openStringInline(string);
 823   }
 824
 825   public void mouseOverStructure(int atomIndex, String strInfo)
 826   {
 827     int pdbResNum;
 828     int alocsep = strInfo.indexOf("^");
 829     int mdlSep = strInfo.indexOf("/");
 830     int chainSeparator = strInfo.indexOf(":"), chainSeparator1 = -1;
 831
 832     if (chainSeparator == -1)
 833     {
 834       chainSeparator = strInfo.indexOf(".");
 835       if (mdlSep > -1 && mdlSep < chainSeparator)
 836       {
 837         chainSeparator1 = chainSeparator;
 838         chainSeparator = mdlSep;
 839       }
 840     }
 841     // handle insertion codes
 842     if (alocsep != -1)
 843     {
 844       pdbResNum = Integer.parseInt(
 845               strInfo.substring(strInfo.indexOf("]") + 1, alocsep));
 846
 847     }
 848     else
 849     {
 850       pdbResNum = Integer.parseInt(
 851               strInfo.substring(strInfo.indexOf("]") + 1, chainSeparator));
 852     }
 853     String chainId;
 854
 855     if (strInfo.indexOf(":") > -1)
 856     {
 857       chainId = strInfo.substring(strInfo.indexOf(":") + 1,
 858               strInfo.indexOf("."));
 859     }
 860     else
 861     {
 862       chainId = " ";
 863     }
 864
 865     String pdbfilename = modelFileNames[frameNo]; // default is first or current
 866     // model
 867     if (mdlSep > -1)
 868     {
 869       if (chainSeparator1 == -1)
 870       {
 871         chainSeparator1 = strInfo.indexOf(".", mdlSep);
 872       }
 873       String mdlId = (chainSeparator1 > -1)
 874               ? strInfo.substring(mdlSep + 1, chainSeparator1)
 875               : strInfo.substring(mdlSep + 1);
 876       try
 877       {
 878         // recover PDB filename for the model hovered over.
 879         int mnumber = new Integer(mdlId).intValue() - 1;
 880         if (_modelFileNameMap != null)
 881         {
 882           int _mp = _modelFileNameMap.length - 1;
 883
 884           while (mnumber < _modelFileNameMap[_mp])
 885           {
 886             _mp--;
 887           }
 888           pdbfilename = modelFileNames[_mp];
 889         }
 890         else
 891         {
 892           if (mnumber >= 0 && mnumber < modelFileNames.length)
 893           {
 894             pdbfilename = modelFileNames[mnumber];
 895           }
 896
 897           if (pdbfilename == null)
 898           {
 899             pdbfilename = new File(viewer.ms.getModelFileName(mnumber))
 900                     .getAbsolutePath();
 901           }
 902         }
 903       } catch (Exception e)
 904       {
 905       }
 906       ;
 907     }
 908     if (lastMessage == null || !lastMessage.equals(strInfo))
 909     {
 910       getSsm().mouseOverStructure(pdbResNum, chainId, pdbfilename);
 911     }
 912
 913     lastMessage = strInfo;
 914   }
 915
 916   public void notifyAtomHovered(int atomIndex, String strInfo, String data)
 917   {
 918     if (data != null)
 919     {
 920       System.err.println("Ignoring additional hover info: " + data
 921               + " (other info: '" + strInfo + "' pos " + atomIndex + ")");
 922     }
 923     mouseOverStructure(atomIndex, strInfo);
 924   }
 925
 926   /*
 927    * { if (history != null && strStatus != null &&
 928    * !strStatus.equals("Script completed")) { history.append("\n" + strStatus);
 929    * } }
 930    */
 931
 932   public void notifyAtomPicked(int atomIndex, String strInfo,
 933           String strData)
 934   {
 935     /**
 936      * this implements the toggle label behaviour copied from the original
 937      * structure viewer, MCView
 938      */
 939     if (strData != null)
 940     {
 941       System.err.println("Ignoring additional pick data string " + strData);
 942     }
 943     int chainSeparator = strInfo.indexOf(":");
 944     int p = 0;
 945     if (chainSeparator == -1)
 946     {
 947       chainSeparator = strInfo.indexOf(".");
 948     }
 949
 950     String picked = strInfo.substring(strInfo.indexOf("]") + 1,
 951             chainSeparator);
 952     String mdlString = "";
 953     if ((p = strInfo.indexOf(":")) > -1)
 954     {
 955       picked += strInfo.substring(p, strInfo.indexOf("."));
 956     }
 957
 958     if ((p = strInfo.indexOf("/")) > -1)
 959     {
 960       mdlString += strInfo.substring(p, strInfo.indexOf(" #"));
 961     }
 962     picked = "((" + picked + ".CA" + mdlString + ")|(" + picked + ".P"
 963             + mdlString + "))";
 964     jmolHistory(false);
 965
 966     if (!atomsPicked.contains(picked))
 967     {
 968       viewer.evalStringQuiet("select " + picked + ";label %n %r:%c");
 969       atomsPicked.addElement(picked);
 970     }
 971     else
 972     {
 973       viewer.evalString("select " + picked + ";label off");
 974       atomsPicked.removeElement(picked);
 975     }
 976     jmolHistory(true);
 977     // TODO: in application this happens
 978     //
 979     // if (scriptWindow != null)
 980     // {
 981     // scriptWindow.sendConsoleMessage(strInfo);
 982     // scriptWindow.sendConsoleMessage("\n");
 983     // }
 984
 985   }
 986
 987   @Override
 988   public void notifyCallback(CBK type, Object[] data)
 989   {
 990     try
 991     {
 992       switch (type)
 993       {
 994       case LOADSTRUCT:
 995         notifyFileLoaded((String) data[1], (String) data[2],
 996                 (String) data[3], (String) data[4],
 997                 ((Integer) data[5]).intValue());
 998
 999         break;
1000       case PICK:
1001         notifyAtomPicked(((Integer) data[2]).intValue(), (String) data[1],
1002                 (String) data[0]);
1003         // also highlight in alignment
1004         // deliberate fall through
1005       case HOVER:
1006         notifyAtomHovered(((Integer) data[2]).intValue(), (String) data[1],
1007                 (String) data[0]);
1008         break;
1009       case SCRIPT:
1010         notifyScriptTermination((String) data[2],
1011                 ((Integer) data[3]).intValue());
1012         break;
1013       case ECHO:
1014         sendConsoleEcho((String) data[1]);
1015         break;
1016       case MESSAGE:
1017         sendConsoleMessage(
1018                 (data == null) ? ((String) null) : (String) data[1]);
1019         break;
1020       case ERROR:
1021         // System.err.println("Ignoring error callback.");
1022         break;
1023       case SYNC:
1024       case RESIZE:
1025         refreshGUI();
1026         break;
1027       case MEASURE:
1028
1029       case CLICK:
1030       default:
1031         System.err.println(
1032                 "Unhandled callback " + type + " " + data[1].toString());
1033         break;
1034       }
1035     } catch (Exception e)
1036     {
1037       System.err.println("Squashed Jmol callback handler error:");
1038       e.printStackTrace();
1039     }
1040   }
1041
1042   @Override
1043   public boolean notifyEnabled(CBK callbackPick)
1044   {
1045     switch (callbackPick)
1046     {
1047     case ECHO:
1048     case LOADSTRUCT:
1049     case MEASURE:
1050     case MESSAGE:
1051     case PICK:
1052     case SCRIPT:
1053     case HOVER:
1054     case ERROR:
1055       return true;
1056     default:
1057       return false;
1058     }
1059   }
1060
1061   // incremented every time a load notification is successfully handled -
1062   // lightweight mechanism for other threads to detect when they can start
1063   // referrring to new structures.
1064   private long loadNotifiesHandled = 0;
1065
1066   public long getLoadNotifiesHandled()
1067   {
1068     return loadNotifiesHandled;
1069   }
1070
1071   public void notifyFileLoaded(String fullPathName, String fileName2,
1072           String modelName, String errorMsg, int modelParts)
1073   {
1074     if (errorMsg != null)
1075     {
1076       fileLoadingError = errorMsg;
1077       refreshGUI();
1078       return;
1079     }
1080     // TODO: deal sensibly with models loaded inLine:
1081     // modelName will be null, as will fullPathName.
1082
1083     // the rest of this routine ignores the arguments, and simply interrogates
1084     // the Jmol view to find out what structures it contains, and adds them to
1085     // the structure selection manager.
1086     fileLoadingError = null;
1087     String[] oldmodels = modelFileNames;
1088     modelFileNames = null;
1089     chainNames = new ArrayList<>();
1090     chainFile = new Hashtable<>();
1091     boolean notifyLoaded = false;
1092     String[] modelfilenames = getStructureFiles();
1093     // first check if we've lost any structures
1094     if (oldmodels != null && oldmodels.length > 0)
1095     {
1096       int oldm = 0;
1097       for (int i = 0; i < oldmodels.length; i++)
1098       {
1099         for (int n = 0; n < modelfilenames.length; n++)
1100         {
1101           if (modelfilenames[n] == oldmodels[i])
1102           {
1103             oldmodels[i] = null;
1104             break;
1105           }
1106         }
1107         if (oldmodels[i] != null)
1108         {
1109           oldm++;
1110         }
1111       }
1112       if (oldm > 0)
1113       {
1114         String[] oldmfn = new String[oldm];
1115         oldm = 0;
1116         for (int i = 0; i < oldmodels.length; i++)
1117         {
1118           if (oldmodels[i] != null)
1119           {
1120             oldmfn[oldm++] = oldmodels[i];
1121           }
1122         }
1123         // deregister the Jmol instance for these structures - we'll add
1124         // ourselves again at the end for the current structure set.
1125         getSsm().removeStructureViewerListener(this, oldmfn);
1126       }
1127     }
1128     refreshPdbEntries();
1129     for (int modelnum = 0; modelnum < modelfilenames.length; modelnum++)
1130     {
1131       String fileName = modelfilenames[modelnum];
1132       boolean foundEntry = false;
1133       StructureFile pdb = null;
1134       String pdbfile = null;
1135       // model was probably loaded inline - so check the pdb file hashcode
1136       if (loadedInline)
1137       {
1138         // calculate essential attributes for the pdb data imported inline.
1139         // prolly need to resolve modelnumber properly - for now just use our
1140         // 'best guess'
1141         pdbfile = viewer.getData(
1142                 "" + (1 + _modelFileNameMap[modelnum]) + ".0", "PDB");
1143       }
1144       // search pdbentries and sequences to find correct pdbentry for this
1145       // model
1146       for (int pe = 0; pe < getPdbCount(); pe++)
1147       {
1148         boolean matches = false;
1149         addSequence(pe, getSequence()[pe]);
1150         if (fileName == null)
1151         {
1152           if (false)
1153           // see JAL-623 - need method of matching pasted data up
1154           {
1155             pdb = getSsm().setMapping(getSequence()[pe], getChains()[pe],
1156                     pdbfile, DataSourceType.PASTE,
1157                     getIProgressIndicator());
1158             getPdbEntry(modelnum).setFile("INLINE" + pdb.getId());
1159             matches = true;
1160             foundEntry = true;
1161           }
1162         }
1163         else
1164         {
1165           File fl = new File(getPdbEntry(pe).getFile());
1166           matches = fl.equals(new File(fileName));
1167           if (matches)
1168           {
1169             foundEntry = true;
1170             // TODO: Jmol can in principle retrieve from CLASSLOADER but
1171             // this
1172             // needs
1173             // to be tested. See mantis bug
1174             // https://mantis.lifesci.dundee.ac.uk/view.php?id=36605
1175             DataSourceType protocol = DataSourceType.URL;
1176             try
1177             {
1178               if (fl.exists())
1179               {
1180                 protocol = DataSourceType.FILE;
1181               }
1182             } catch (Exception e)
1183             {
1184             } catch (Error e)
1185             {
1186             }
1187             // Explicitly map to the filename used by Jmol ;
1188             pdb = getSsm().setMapping(getSequence()[pe], getChains()[pe],
1189                     fileName, protocol, getIProgressIndicator());
1190             // pdbentry[pe].getFile(), protocol);
1191
1192           }
1193         }
1194         if (matches)
1195         {
1196           // add an entry for every chain in the model
1197           for (int i = 0; i < pdb.getChains().size(); i++)
1198           {
1199             String chid = new String(
1200                     pdb.getId() + ":" + pdb.getChains().elementAt(i).id);
1201             chainFile.put(chid, fileName);
1202             chainNames.add(chid);
1203           }
1204           notifyLoaded = true;
1205         }
1206       }
1207
1208       if (!foundEntry && associateNewStructs)
1209       {
1210         // this is a foreign pdb file that jalview doesn't know about - add
1211         // it to the dataset and try to find a home - either on a matching
1212         // sequence or as a new sequence.
1213         String pdbcontent = viewer.getData("/" + (modelnum + 1) + ".1",
1214                 "PDB");
1215         // parse pdb file into a chain, etc.
1216         // locate best match for pdb in associated views and add mapping to
1217         // ssm
1218         // if properly registered then
1219         notifyLoaded = true;
1220
1221       }
1222     }
1223     // FILE LOADED OK
1224     // so finally, update the jmol bits and pieces
1225     // if (jmolpopup != null)
1226     // {
1227     // // potential for deadlock here:
1228     // // jmolpopup.updateComputedMenus();
1229     // }
1230     if (!isLoadingFromArchive())
1231     {
1232       viewer.evalStringQuiet(
1233               "model *; select backbone;restrict;cartoon;wireframe off;spacefill off");
1234     }
1235     // register ourselves as a listener and notify the gui that it needs to
1236     // update itself.
1237     getSsm().addStructureViewerListener(this);
1238     if (notifyLoaded)
1239     {
1240       FeatureRenderer fr = getFeatureRenderer(null);
1241       if (fr != null)
1242       {
1243         fr.featuresAdded();
1244       }
1245       refreshGUI();
1246       loadNotifiesHandled++;
1247     }
1248     setLoadingFromArchive(false);
1249   }
1250
1251   @Override
1252   public List<String> getChainNames()
1253   {
1254     return chainNames;
1255   }
1256
1257   protected IProgressIndicator getIProgressIndicator()
1258   {
1259     return null;
1260   }
1261
1262   public void notifyNewPickingModeMeasurement(int iatom, String strMeasure)
1263   {
1264     notifyAtomPicked(iatom, strMeasure, null);
1265   }
1266
1267   public abstract void notifyScriptTermination(String strStatus,
1268           int msWalltime);
1269
1270   /**
1271    * display a message echoed from the jmol viewer
1272    *
1273    * @param strEcho
1274    */
1275   public abstract void sendConsoleEcho(String strEcho); /*
1276                                                          * { showConsole(true);
1277                                                          *
1278                                                          * history.append("\n" +
1279                                                          * strEcho); }
1280                                                          */
1281
1282   // /End JmolStatusListener
1283   // /////////////////////////////
1284
1285   /**
1286    * @param strStatus
1287    *          status message - usually the response received after a script
1288    *          executed
1289    */
1290   public abstract void sendConsoleMessage(String strStatus);
1291
1292   @Override
1293   public void setCallbackFunction(String callbackType,
1294           String callbackFunction)
1295   {
1296     System.err.println("Ignoring set-callback request to associate "
1297             + callbackType + " with function " + callbackFunction);
1298
1299   }
1300
1301   @Override
1302   public void setJalviewColourScheme(ColourSchemeI cs)
1303   {
1304     colourBySequence = false;
1305
1306     if (cs == null)
1307     {
1308       return;
1309     }
1310
1311     jmolHistory(false);
1312     StringBuilder command = new StringBuilder(128);
1313     command.append("select *;color white;");
1314     List<String> residueSet = ResidueProperties.getResidues(isNucleotide(),
1315             false);
1316     for (String resName : residueSet)
1317     {
1318       char res = resName.length() == 3
1319               ? ResidueProperties.getSingleCharacterCode(resName)
1320               : resName.charAt(0);
1321       Color col = cs.findColour(res, 0, null, null, 0f);
1322       command.append("select " + resName + ";color[" + col.getRed() + ","
1323               + col.getGreen() + "," + col.getBlue() + "];");
1324     }
1325
1326     evalStateCommand(command.toString());
1327     jmolHistory(true);
1328   }
1329
1330   public void showHelp()
1331   {
1332     showUrl("http://jmol.sourceforge.net/docs/JmolUserGuide/", "jmolHelp");
1333   }
1334
1335   /**
1336    * open the URL somehow
1337    *
1338    * @param target
1339    */
1340   public abstract void showUrl(String url, String target);
1341
1342   /**
1343    * called when the binding thinks the UI needs to be refreshed after a Jmol
1344    * state change. this could be because structures were loaded, or because an
1345    * error has occured.
1346    */
1347   public abstract void refreshGUI();
1348
1349   /**
1350    * called to show or hide the associated console window container.
1351    *
1352    * @param show
1353    */
1354   public abstract void showConsole(boolean show);
1355
1356   /**
1357    * @param renderPanel
1358    * @param jmolfileio
1359    *          - when true will initialise jmol's file IO system (should be false
1360    *          in applet context)
1361    * @param htmlName
1362    * @param documentBase
1363    * @param codeBase
1364    * @param commandOptions
1365    */
1366   public void allocateViewer(Container renderPanel, boolean jmolfileio,
1367           String htmlName, URL documentBase, URL codeBase,
1368           String commandOptions)
1369   {
1370     allocateViewer(renderPanel, jmolfileio, htmlName, documentBase,
1371             codeBase, commandOptions, null, null);
1372   }
1373
1374   /**
1375    *
1376    * @param renderPanel
1377    * @param jmolfileio
1378    *          - when true will initialise jmol's file IO system (should be false
1379    *          in applet context)
1380    * @param htmlName
1381    * @param documentBase
1382    * @param codeBase
1383    * @param commandOptions
1384    * @param consolePanel
1385    *          - panel to contain Jmol console
1386    * @param buttonsToShow
1387    *          - buttons to show on the console, in ordr
1388    */
1389   public void allocateViewer(Container renderPanel, boolean jmolfileio,
1390           String htmlName, URL documentBase, URL codeBase,
1391           String commandOptions, final Container consolePanel,
1392           String buttonsToShow)
1393   {
1394     if (commandOptions == null)
1395     {
1396       commandOptions = "";
1397     }
1398     viewer = (Viewer) JmolViewer.allocateViewer(renderPanel,
1399             (jmolfileio ? new SmarterJmolAdapter() : null),
1400             htmlName + ((Object) this).toString(), documentBase, codeBase,
1401             commandOptions, this);
1402
1403     viewer.setJmolStatusListener(this); // extends JmolCallbackListener
1404
1405     console = createJmolConsole(consolePanel, buttonsToShow);
1406     if (consolePanel != null)
1407     {
1408       consolePanel.addComponentListener(this);
1409
1410     }
1411
1412   }
1413
1414   protected abstract JmolAppConsoleInterface createJmolConsole(
1415           Container consolePanel, String buttonsToShow);
1416
1417   protected org.jmol.api.JmolAppConsoleInterface console = null;
1418
1419   @Override
1420   public void setBackgroundColour(java.awt.Color col)
1421   {
1422     jmolHistory(false);
1423     viewer.evalStringQuiet("background [" + col.getRed() + ","
1424             + col.getGreen() + "," + col.getBlue() + "];");
1425     jmolHistory(true);
1426   }
1427
1428   @Override
1429   public int[] resizeInnerPanel(String data)
1430   {
1431     // Jalview doesn't honour resize panel requests
1432     return null;
1433   }
1434
1435   /**
1436    *
1437    */
1438   protected void closeConsole()
1439   {
1440     if (console != null)
1441     {
1442       try
1443       {
1444         console.setVisible(false);
1445       } catch (Error e)
1446       {
1447       } catch (Exception x)
1448       {
1449       }
1450       ;
1451       console = null;
1452     }
1453   }
1454
1455   /**
1456    * ComponentListener method
1457    */
1458   @Override
1459   public void componentMoved(ComponentEvent e)
1460   {
1461   }
1462
1463   /**
1464    * ComponentListener method
1465    */
1466   @Override
1467   public void componentResized(ComponentEvent e)
1468   {
1469   }
1470
1471   /**
1472    * ComponentListener method
1473    */
1474   @Override
1475   public void componentShown(ComponentEvent e)
1476   {
1477     showConsole(true);
1478   }
1479
1480   /**
1481    * ComponentListener method
1482    */
1483   @Override
1484   public void componentHidden(ComponentEvent e)
1485   {
1486     showConsole(false);
1487   }
1488 }