allow templates to be read from a URL

git-svn-id: svn+ssh://svn.internal.sanger.ac.uk/repos/svn/pathsoft/artemis/trunk@9048 ee4ac58c-ac51-4696-9907-e4b3aa274f04

uk/ac/sanger/artemis/circular/DNADraw.java

@@ -62,7 +62,6 @@ import java.awt.dnd.DropTargetDropEvent;
 import java.awt.dnd.DropTargetEvent;
 import java.awt.dnd.DropTargetListener;
 import java.io.BufferedReader;
-import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.InputStreamReader;
@@ -1884,10 +1883,7 @@ public class DNADraw extends ScrollPanel
     final Wizard wiz;
     
     if(arg.length > 0 && arg[0].equals("-t"))
-    {
-      final File fileTemplate = new File(arg[1]);
-      wiz = new Wizard(fileTemplate);
-    }
+      wiz = new Wizard(arg[1]);
     else
       wiz = new Wizard((DNADraw)null);
     final DNADraw dna = wiz.getDNADraw();

uk/ac/sanger/artemis/circular/Wizard.java

@@ -29,6 +29,9 @@ import java.io.File;
 import java.io.FileNotFoundException;
 import java.io.FileReader;
 import java.io.IOException;
+import java.io.InputStreamReader;
+import java.net.MalformedURLException;
+import java.net.URL;
 import java.util.Vector;
 import java.util.Hashtable;
 
@@ -94,7 +97,7 @@ public class Wizard
         JOptionPane.showMessageDialog(null, 
             fileTemplate.getName()+" cannot be found!", 
             "Missing File", JOptionPane.WARNING_MESSAGE);
-      loadTemplate(fileTemplate);
+      loadTemplate(chooser.getSelectedFile());
     }
     else if(n == 1 || n == 2)
     {
@@ -195,29 +198,60 @@ public class Wizard
    * Open a DNA plot based on a template file
    * @param template
    */
-  public Wizard(final File template)
+  public Wizard(final String templateName)
   {
-    loadTemplate(template);
+    loadTemplate(templateName);
   }
   
   /**
    * Load from a template file
    * @param template
    */
-  private void loadTemplate(final File template)
+  private void loadTemplate(final String templateName)
   {
     final ProgressFrame progress = new ProgressFrame();
-    progress.setString("Reading from "+template.getName()+"   ");
+    progress.setString("Reading from "+templateName+"   ");
     progress.setValue(2);
     
     if(dna == null)
       dna = new DNADraw();
     Options.getOptions();
+    final BufferedReader inputStream = getReader(templateName);
+    loadTemplate(inputStream, templateName, progress);
+  }
+  
+  /**
+   * Load from a template file
+   * @param template
+   */
+  private void loadTemplate(final File templateFile)
+  {
+    final ProgressFrame progress = new ProgressFrame();
+    progress.setString("Reading from "+templateFile.getName()+"   ");
+    progress.setValue(2);
     
+    if(dna == null)
+      dna = new DNADraw();
+    Options.getOptions();
+    FileReader reader;
     try
     {
-      final FileReader reader = new FileReader(template);
+      reader = new FileReader(templateFile);
       final BufferedReader inputStream = new BufferedReader(reader);
+      loadTemplate(inputStream, templateFile.getName(), progress);
+    }
+    catch(FileNotFoundException e)
+    {
+      e.printStackTrace();
+    }
+  }
+  
+  private void loadTemplate(final BufferedReader inputStream,
+                            final String templateName,
+                            final ProgressFrame progress)
+  {
+    try
+    {
       final EntryGroup entryGroup = new SimpleEntryGroup();
       final Hashtable fileEntrys = new Hashtable();
       Vector v_tracks = new Vector();
@@ -273,9 +307,8 @@ public class Wizard
         v_tracks.add(track);
       }
       inputStream.close();
-      reader.close();
       
-      progress.setString("Read template "+template.getName());
+      progress.setString("Read template "+templateName);
       progress.setValue(7);
       Track[] newTracks = new Track[v_tracks.size()];
       for(int i=0; i<v_tracks.size(); i++)
@@ -758,5 +791,50 @@ public class Wizard
 
     Wizard.tracks = newTracks;
   }
+  
+  /**
+   * Return reader for a file or a URL
+   * @param templateName
+   * @return
+   */
+  private BufferedReader getReader(final String templateName)
+  {
+    final File fileTemplate = new File(templateName);
+    BufferedReader inputStream = null;
+    if(!fileTemplate.exists())
+    {
+      if (templateName.indexOf ("://") != -1) 
+      {
+        URL template;
+        try
+        {
+          template = new URL(templateName);
+          inputStream = new BufferedReader(
+              new InputStreamReader(template.openStream()));
+        }
+        catch(MalformedURLException e)
+        {
+          e.printStackTrace();
+        }
+        catch(IOException e)
+        {
+          e.printStackTrace();
+        }
+      } 
+    }
+    else
+    {
+      try
+      {
+        FileReader reader = new FileReader(fileTemplate);
+        inputStream = new BufferedReader(reader);
+      }
+      catch(FileNotFoundException e)
+      {
+        e.printStackTrace();
+      }
+    }
+    return inputStream;
+  }
 }