Add to_array for an ImageFile model.

Author: jonathanrocher

stage4.1_first_application/pycasa/model/image_file.py

@@ -1,6 +1,9 @@
+# General imports
 import PIL.Image
 from PIL.ExifTags import TAGS
+import numpy as np
 
+# ETS imports
 from traits.api import Dict, File, HasStrictTraits, observe
 
 
@@ -11,10 +14,15 @@ class ImageFile(HasStrictTraits):
 
     metadata = Dict
 
+    def to_array(self):
+        with PIL.Image.open(self.filepath) as img:
+            return np.asarray(img)
+
     @observe("filepath")
     def load_metadata(self, event):
-        img = PIL.Image.open(self.filepath)
-        exif = img._getexif()
+        with PIL.Image.open(self.filepath) as img:
+            exif = img._getexif()
+
         if exif:
             self.metadata = {TAGS[k]: v for k, v in exif.items()
                              if k in TAGS}

stage4.1_first_application/pycasa/ui/image_file_view.py

@@ -27,6 +27,5 @@ class ImageFileView(ModelView):
     def build_mpl_figure(self, event):
         figure = Figure()
         axes = figure.add_subplot(111)
-        data = imageio.imread(self.model.filepath)
-        axes.imshow(data)
+        axes.imshow(self.model.to_array())
         self.figure = figure