X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2Fjalview%2Fgui%2FAssociatePdbFileWithSeq.java;h=fd89c503eac593533f0f48b2b0e970c90cd65ce8;hb=7e3a6674abdd31bf48e7e249a74eff50fd2ce589;hp=cbfbbc46d093ef649f92adc13795d9117e358c37;hpb=28c13a67af0d57c53de532d0d50add7ab83f7d42;p=jalview.git

diff --git a/src/jalview/gui/AssociatePdbFileWithSeq.java b/src/jalview/gui/AssociatePdbFileWithSeq.java
index cbfbbc4..fd89c50 100644
--- a/src/jalview/gui/AssociatePdbFileWithSeq.java
+++ b/src/jalview/gui/AssociatePdbFileWithSeq.java
@@ -1,61 +1,105 @@
+/*
+ * 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.
+ *  
+ * 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 <http://www.gnu.org/licenses/>.
+ * The Jalview Authors are detailed in the 'AUTHORS' file.
+ */
 package jalview.gui;
 
-import javax.swing.JOptionPane;
-
+import jalview.api.StructureSelectionManagerProvider;
 import jalview.datamodel.PDBEntry;
-import jalview.datamodel.Sequence;
 import jalview.datamodel.SequenceI;
+import jalview.io.DataSourceType;
+import jalview.io.StructureFile;
+import jalview.structure.StructureImportSettings.TFType;
+import jalview.structure.StructureSelectionManager;
+import jalview.util.MessageManager;
 
 /**
  * GUI related routines for associating PDB files with sequences
+ * 
  * @author JimP
- *
+ * 
  */
 public class AssociatePdbFileWithSeq
 {
 
-/**
- * assocate the given PDB file with 
- * @param choice
- * @param sequence
- */
-  public PDBEntry associatePdbWithSeq(String choice, String protocol, SequenceI sequence, boolean prompt)
+  /**
+   * assocate the given PDB file with
+   * 
+   * @param choice
+   * @param sequence
+   */
+  public PDBEntry associatePdbWithSeq(String choice, DataSourceType file,
+          SequenceI sequence, boolean prompt,
+          StructureSelectionManagerProvider ssmp)
+  {
+    return associatePdbWithSeq(choice, file, sequence, prompt, ssmp,
+            TFType.DEFAULT, null);
+  }
+
+  public PDBEntry associatePdbWithSeq(String choice, DataSourceType file,
+          SequenceI sequence, boolean prompt,
+          StructureSelectionManagerProvider ssmp, TFType tft,
+          String paeFilename)
   {
     PDBEntry entry = new PDBEntry();
-    try
+    StructureFile pdbfile = StructureSelectionManager
+            .getStructureSelectionManager(ssmp)
+            .setMapping(false, new SequenceI[]
+            { sequence }, null, choice, file, tft, paeFilename);
+    if (pdbfile == null)
     {
-      MCview.PDBfile pdbfile = new MCview.PDBfile(choice,
-              protocol);
+      // stacktrace already thrown so just return
+      return null;
+    }
+    if (pdbfile.getId() == null)
+    {
+      String reply = null;
 
-      if (pdbfile.id == null)
+      if (prompt)
       {
-        String reply = null;
-      
-        if (prompt) { reply = JOptionPane
-                .showInternalInputDialog(
-                        Desktop.desktop,
-                        "Couldn't find a PDB id in the file supplied."
-                                + "Please enter an Id to identify this structure.",
-                        "No PDB Id in File", JOptionPane.QUESTION_MESSAGE);}
-        if (reply == null)
-        {
-          return null;
-        }
-
-        entry.setId(reply);
+        reply = JvOptionPane.showInternalInputDialog(Desktop.desktop,
+                MessageManager
+                        .getString("label.couldnt_find_pdb_id_in_file"),
+                MessageManager.getString("label.no_pdb_id_in_file"),
+                JvOptionPane.QUESTION_MESSAGE);
       }
-      else
+      if (reply == null)
       {
-        entry.setId(pdbfile.id);
+        return null;
       }
-    } catch (java.io.IOException ex)
+
+      entry.setId(reply);
+    }
+    else
     {
-      ex.printStackTrace();
+      entry.setId(pdbfile.getId());
     }
+    entry.setType(PDBEntry.Type.FILE);
 
-    entry.setFile(choice);
-    sequence.getDatasetSequence().addPDBId(entry);
+    if (pdbfile != null)
+    {
+      entry.setFile(choice);
+      sequence.getDatasetSequence().addPDBId(entry);
+      StructureSelectionManager.getStructureSelectionManager(ssmp)
+              .registerPDBEntry(entry);
+      entry.setStructureFile(pdbfile);
+    }
     return entry;
   }
-
 }