Reorganize folders to support slides

- Remove stage3.0 as it's no longer needed
- Add the image folder back to stages 2 and 3

Author: siddhantwahal

stage2.1_traited_script/image_folder.py

@@ -0,0 +1,62 @@
+
+# General imports
+import glob
+from os.path import basename, expanduser, isdir
+
+import numpy as np
+import pandas as pd
+
+# ETS imports
+from traits.api import (
+    Directory, HasStrictTraits, Instance, List, observe,
+)
+
+# Local imports
+from pycasa.model.image_file import ImageFile, SUPPORTED_FORMATS
+
+FILENAME_COL = "filename"
+NUM_FACE_COL = "Num. faces"
+
+
+class ImageFolder(HasStrictTraits):
+    """ Model for a folder of images.
+    """
+    directory = Directory(expanduser("~"))
+
+    images = List(Instance(ImageFile))
+
+    data = Instance(pd.DataFrame)
+
+    def __init__(self, **traits):
+        # Don't forget this!
+        super(ImageFolder, self).__init__(**traits)
+        if not isdir(self.directory):
+            msg = f"The provided directory isn't a real directory: " \
+                  f"{self.directory}"
+            raise ValueError(msg)
+        self.data = self._create_metadata_df()
+
+    @observe("directory")
+    def _update_images(self, event):
+        self.images = [
+            ImageFile(filepath=file)
+            for fmt in SUPPORTED_FORMATS
+            for file in glob.glob(f"{self.directory}/*{fmt}")
+        ]
+
+    @observe("images.items")
+    def _update_metadata(self, event):
+        self.data = self._create_metadata_df()
+
+    def _create_metadata_df(self):
+        if not self.images:
+            return pd.DataFrame({FILENAME_COL: [], NUM_FACE_COL: []})
+        return pd.DataFrame([
+                {
+                    FILENAME_COL: basename(img.filepath),
+                    NUM_FACE_COL: np.nan,
+                    **img.metadata
+
+                }
+                for img in self.images
+        ])

stage3.0_first_views/README.md

@@ -0,0 +1,62 @@
+
+# General imports
+import glob
+from os.path import basename, expanduser, isdir
+
+import numpy as np
+import pandas as pd
+
+# ETS imports
+from traits.api import (
+    Directory, HasStrictTraits, Instance, List, observe,
+)
+
+# Local imports
+from pycasa.model.image_file import ImageFile, SUPPORTED_FORMATS
+
+FILENAME_COL = "filename"
+NUM_FACE_COL = "Num. faces"
+
+
+class ImageFolder(HasStrictTraits):
+    """ Model for a folder of images.
+    """
+    directory = Directory(expanduser("~"))
+
+    images = List(Instance(ImageFile))
+
+    data = Instance(pd.DataFrame)
+
+    def __init__(self, **traits):
+        # Don't forget this!
+        super(ImageFolder, self).__init__(**traits)
+        if not isdir(self.directory):
+            msg = f"The provided directory isn't a real directory: " \
+                  f"{self.directory}"
+            raise ValueError(msg)
+        self.data = self._create_metadata_df()
+
+    @observe("directory")
+    def _update_images(self, event):
+        self.images = [
+            ImageFile(filepath=file)
+            for fmt in SUPPORTED_FORMATS
+            for file in glob.glob(f"{self.directory}/*{fmt}")
+        ]
+
+    @observe("images.items")
+    def _update_metadata(self, event):
+        self.data = self._create_metadata_df()
+
+    def _create_metadata_df(self):
+        if not self.images:
+            return pd.DataFrame({FILENAME_COL: [], NUM_FACE_COL: []})
+        return pd.DataFrame([
+                {
+                    FILENAME_COL: basename(img.filepath),
+                    NUM_FACE_COL: np.nan,
+                    **img.metadata
+
+                }
+                for img in self.images
+        ])

stage3.0_first_views/__init__.py

@@ -0,0 +1,51 @@
+from traits.api import Instance
+from traitsui.api import (
+    HGroup, Item, Label, ModelView, Spring, View
+)
+from traitsui.ui_editors.data_frame_editor import DataFrameEditor
+
+# Local imports
+from pycasa.model.image_folder import FILENAME_COL, ImageFolder, NUM_FACE_COL
+
+
+DISPLAYED_COLUMNS = [FILENAME_COL, NUM_FACE_COL] + [
+    'ApertureValue', 'ExifVersion', 'Model', 'Make', 'LensModel', 'DateTime',
+    'ShutterSpeedValue', 'ExposureTime', 'XResolution', 'YResolution',
+    'Orientation', 'GPSInfo', 'DigitalZoomRatio', 'FocalLengthIn35mmFilm',
+    'ISOSpeedRatings', 'SceneType'
+]
+
+
+class ImageFolderView(ModelView):
+    """ ModelView for a folder of images.
+    """
+    model = Instance(ImageFolder)
+
+    view = View(
+        Item('model.directory', style="readonly", show_label=False),
+        Item(
+            'model.data',
+            editor=DataFrameEditor(columns=DISPLAYED_COLUMNS),
+            show_label=False,
+            visible_when="len(model.data) > 0",
+        ),
+        HGroup(
+            Spring(),
+            Label("No images found. No data to show"),
+            Spring(),
+            visible_when="len(model.data) == 0",
+        ),
+        resizable=True
+    )
+
+
+if __name__ == '__main__':
+    from os.path import dirname, join
+    import ets_tutorial
+
+    TUTORIAL_DIR = dirname(ets_tutorial.__file__)
+    SAMPLE_IMG_DIR = join(TUTORIAL_DIR, "..", "sample_images")
+
+    image_file = ImageFolder(directory=SAMPLE_IMG_DIR)
+    view = ImageFolderView(model=image_file)
+    view.configure_traits()