Add support for folder visualization

Author: jonathanrocher

stage5.1_first_application/pycasa/__init__.py

@@ -0,0 +1,2 @@
+
+__version__ = "0.0.1"

stage5.1_first_application/pycasa/app/__init__.py

@@ -0,0 +1,28 @@
+# coding=utf-8
+""" TaskApplication object for the Pycasa app.
+"""
+import logging
+
+from pyface.tasks.api import TasksApplication, TaskFactory
+from ..ui.tasks.pycasa_task import PycasaTask
+
+logger = logging.getLogger(__name__)
+
+
+class PycasaApplication(TasksApplication):
+    """ An application to say hello.
+    """
+    id = "pycasa_application"
+
+    name = "Pycasa"
+
+    description = "An example Tasks application that explores image files."
+
+    def _task_factories_default(self):
+        return [
+            TaskFactory(
+                id='pycasa.pycasa_task_factory',
+                name="Main Pycasa Task Factory",
+                factory=PycasaTask
+            )
+        ]

stage5.1_first_application/pycasa/app/app.py

@@ -0,0 +1,11 @@
+
+from pycasa.app.app import PycasaApplication
+
+
+def main():
+    app = PycasaApplication()
+    app.run()
+
+
+if __name__ == '__main__':
+    main()

stage5.1_first_application/pycasa/app/main.py

@@ -0,0 +1,9 @@
+from os.path import expanduser
+from traits.api import Directory, Event, HasStrictTraits
+
+
+class FileBrowser(HasStrictTraits):
+    root = Directory(expanduser("~"))
+
+    #: Item last double-clicked on in the tree view
+    requested_item = Event

stage5.1_first_application/pycasa/model/__init__.py

@@ -0,0 +1,41 @@
+# General imports
+import os
+import PIL.Image
+from PIL.ExifTags import TAGS
+import numpy as np
+
+# ETS imports
+from traits.api import cached_property, Dict, File, HasStrictTraits, Property
+
+SUPPORTED_FORMATS = [".png", ".jpg", ".jpeg"]
+
+
+class ImageFile(HasStrictTraits):
+    """ Model to hold an image file.
+    """
+    filepath = File
+
+    metadata = Property(Dict, depends_on="filepath")
+
+    def to_array(self):
+        file_ext = os.path.splitext(self.filepath)[1].lower()
+        if not self.filepath or file_ext not in SUPPORTED_FORMATS:
+            return np.array([])
+
+        with PIL.Image.open(self.filepath) as img:
+            return np.asarray(img)
+
+    @cached_property
+    def _get_metadata(self):
+        file_ext = os.path.splitext(self.filepath)[1].lower()
+        if not self.filepath or file_ext not in SUPPORTED_FORMATS:
+            return {}
+
+        with PIL.Image.open(self.filepath) as img:
+            exif = img._getexif()
+
+        if exif:
+            return {TAGS[k]: v for k, v in exif.items()
+                    if k in TAGS}
+        else:
+            return {}

stage5.1_first_application/pycasa/model/file_browser.py

@@ -0,0 +1,48 @@
+# General imports
+import os
+
+import pandas as pd
+import numpy as np
+
+# ETS imports
+from traits.api import Directory, HasStrictTraits, Instance
+
+# Local imports
+from .image_file import ImageFile, SUPPORTED_FORMATS
+
+
+class ImageFolder(HasStrictTraits):
+    """ Model to hold an image folder.
+    """
+    path = Directory
+
+    data = Instance(pd.DataFrame)
+
+    def __init__(self, **traits):
+        # Don't forget this!
+        super(ImageFolder, self).__init__(**traits)
+        if not os.path.isdir(self.path):
+            msg = f"Unable to create an ImageFolder from {self.path} since" \
+                  f" it is not a valid directory."
+            raise ValueError(msg)
+
+        self.data = self.to_dataframe()
+
+    def to_dataframe(self):
+        if not self.path:
+            return pd.DataFrame({"filename": [], "Num. faces": []})
+
+        data = []
+        for filename in os.listdir(self.path):
+            file_ext = os.path.splitext(filename)[1].lower()
+            if file_ext in SUPPORTED_FORMATS:
+                filepath = os.path.join(self.path, filename)
+                img_file = ImageFile(filepath=filepath)
+                file_data = {"filename": filename, "Num. faces": np.nan}
+                try:
+                    file_data.update(img_file.metadata)
+                except Exception as e:
+                    pass
+                data.append(file_data)
+
+        return pd.DataFrame(data)

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

@@ -0,0 +1,40 @@
+from os.path import dirname, join
+from unittest import TestCase
+
+import numpy as np
+
+from pycasa.model.image_file import ImageFile
+
+import ets_tutorial
+
+TUTORIAL_DIR = dirname(ets_tutorial.__file__)
+
+SAMPLE_IMG_DIR = join(TUTORIAL_DIR, "..", "sample_images")
+
+SAMPLE_IMG1 = join(SAMPLE_IMG_DIR, "IMG-0311_xmas_2020.JPG")
+
+
+class TestImageFile(TestCase):
+    def test_no_image_file(self):
+        img = ImageFile()
+        self.assertEqual(img.metadata, {})
+        data = img.to_array()
+        self.assertIsInstance(data, np.ndarray)
+        self.assertEqual(data.shape, (0,))
+
+    def test_bad_type_image_file(self):
+        img = ImageFile(filepath=__file__)
+        self.assertEqual(img.metadata, {})
+        data = img.to_array()
+        self.assertIsInstance(data, np.ndarray)
+        self.assertEqual(data.shape, (0,))
+
+    def test_image_metadata(self):
+        img = ImageFile(filepath=SAMPLE_IMG1)
+        self.assertNotEqual(img.metadata, {})
+        for key in ['ExifVersion', 'ExifImageWidth', 'ExifImageHeight']:
+            self.assertIn(key, img.metadata.keys())
+        data = img.to_array()
+        expected_shape = (img.metadata['ExifImageHeight'],
+                          img.metadata['ExifImageWidth'], 3)
+        self.assertEqual(data.shape, expected_shape)