JAL-629 Some refactoring to allow more flexible use of PAE files. Not working as...

[jalview.git] / src / jalview / io / StructureFile.java

diff --git a/src/jalview/io/StructureFile.java b/src/jalview/io/StructureFile.java
index 0a0504e..f498c1e 100644 (file)
--- a/src/jalview/io/StructureFile.java
+++ b/src/jalview/io/StructureFile.java
@@ -70,6 +70,25 @@ public abstract class StructureFile extends AlignFile
 
   private StructureImportSettings.TFType temperatureFactorType = TFType.DEFAULT;
 
+  private String paeMatrix = null;
+
+  private boolean alphaFoldModel;
+
+  public void setPAEMatrix(String paeFilename)
+  {
+    paeMatrix = paeFilename;
+  }
+
+  public String getPAEMatrix()
+  {
+    return paeMatrix;
+  }
+
+  public boolean hasPAEMatrix()
+  {
+    return paeMatrix != null;
+  }
+
   public void setTemperatureFactorType(StructureImportSettings.TFType t)
   {
     this.temperatureFactorType = t;
@@ -80,6 +99,16 @@ public abstract class StructureFile extends AlignFile
     return temperatureFactorType;
   }
 
+  public void setAlphafoldModel(boolean afm)
+  {
+    alphaFoldModel = afm;
+  }
+
+  public boolean isAlphafoldModel()
+  {
+    return alphaFoldModel;
+  }
+
   public StructureFile(Object inFile, DataSourceType sourceType)
           throws IOException
   {