Merge pull request #24 from jonathanrocher/feature/add_stage_4.1

Add end state for stage 4.

Author: jonathanrocher

ets_tutorial/__init__.py

@@ -0,0 +1 @@
+# First real version of the pycasa ETS pyface application!

stage4.1_first_application/README.md

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

stage4.1_first_application/pycasa/__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
+            )
+        ]

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

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

stage4.1_first_application/pycasa/app/app.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

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

@@ -0,0 +1,36 @@
+# General imports
+import PIL.Image
+from PIL.ExifTags import TAGS
+import numpy as np
+
+# ETS imports
+from traits.api import cached_property, Dict, File, HasStrictTraits, Property
+
+
+class ImageFile(HasStrictTraits):
+    """ Model to hold an image file.
+    """
+    filepath = File
+
+    metadata = Property(Dict, depends_on="filepath")
+
+    def to_array(self):
+        if not self.filepath:
+            return np.array([])
+
+        with PIL.Image.open(self.filepath) as img:
+            return np.asarray(img)
+
+    @cached_property
+    def _get_metadata(self):
+        if not self.filepath:
+            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 {}

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

@@ -0,0 +1,33 @@
+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_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)