Added download method to ImageWrapper

Author: ppouchin

src/main/java/fr/igred/omero/repository/ImageWrapper.java

@@ -39,6 +39,7 @@
 import omero.gateway.exception.DSAccessException;
 import omero.gateway.exception.DSOutOfServiceException;
 import omero.gateway.facility.ROIFacility;
+import omero.gateway.facility.TransferFacility;
 import omero.gateway.model.ChannelData;
 import omero.gateway.model.FolderData;
 import omero.gateway.model.ImageData;
@@ -52,6 +53,7 @@
 import java.awt.Color;
 import java.awt.image.BufferedImage;
 import java.io.ByteArrayInputStream;
+import java.io.File;
 import java.io.IOException;
 import java.sql.Timestamp;
 import java.util.ArrayList;
@@ -62,8 +64,7 @@
 import java.util.logging.Level;
 import java.util.logging.Logger;
 
-import static fr.igred.omero.exception.ExceptionHandler.handleServiceOrAccess;
-import static fr.igred.omero.exception.ExceptionHandler.handleServiceOrServer;
+import static fr.igred.omero.exception.ExceptionHandler.*;
 import static omero.rtypes.rint;
 
 
@@ -572,4 +573,28 @@ public BufferedImage getThumbnail(Client client, int size) throws ServiceExcepti
         return thumbnail;
     }
 
+
+    /**
+     * Downloads the original files from the server.
+     *
+     * @param client The client handling the connection.
+     * @param path   Path to the file.
+     *
+     * @return See above.
+     *
+     * @throws OMEROServerError Server error.
+     * @throws ServiceException Cannot connect to OMERO.
+     * @throws AccessException  Cannot access data.
+     */
+    public List<File> download(Client client, String path)
+    throws OMEROServerError, ServiceException, AccessException {
+        try {
+            TransferFacility transfer = client.getGateway().getFacility(TransferFacility.class);
+            return transfer.downloadImage(client.getCtx(), path, getId());
+        } catch (DSAccessException | DSOutOfServiceException | ExecutionException e) {
+            handleException(e, "Could not download image " + getId() + ": " + e.getMessage());
+        }
+        return new ArrayList<>();
+    }
+
 }

src/test/java/fr/igred/omero/repository/ImageTest.java

@@ -38,6 +38,7 @@
 import java.io.FileOutputStream;
 import java.io.PrintStream;
 import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
 import java.text.SimpleDateFormat;
 import java.util.ArrayList;
 import java.util.Date;
@@ -678,4 +679,15 @@ public void testGetThumbnail() throws Exception {
     }
 
 
+    @Test
+    public void testDownload() throws Exception {
+        ImageWrapper  image     = client.getImage(1L);
+        List<File> files = image.download(client, ".");
+        assertEquals(2, files.size());
+        assertTrue(files.get(0).exists());
+        Files.deleteIfExists(files.get(0).toPath());
+        Files.deleteIfExists(files.get(1).toPath());
+    }
+
+
 }