Add traitsUI widgets to filter a folder view.

Author: jonathanrocher

stage5.3_fuller_application/pycasa/ui/image_folder_view.py

@@ -1,16 +1,22 @@
 # General imports
 
 # ETS imports
-from traits.api import Button, Instance, observe
-from traitsui.api import HGroup, Item, Label, ModelView, Spring, View
+import numpy as np
+import pandas as pd
+from traits.api import Bool, Button, Enum, Instance, List, observe
+from traitsui.api import HGroup, Item, Label, ListStrEditor, ModelView, \
+    Spring, VGroup, View
 from traitsui.ui_editors.data_frame_editor import DataFrameEditor
+
 # Local imports
-from ..model.image_folder import ImageFolder
+from ..model.image_folder import FILENAME_COL, ImageFolder, NUM_FACE_COL
 
 
-DISPLAYED_COLUMNS = [
+DISPLAYED_COLUMNS = [FILENAME_COL, NUM_FACE_COL] + [
     'ApertureValue', 'ExifVersion', 'Model', 'Make', 'LensModel', 'DateTime',
-    'ShutterSpeedValue', 'XResolution', 'YResolution'
+    'ShutterSpeedValue', 'ExposureTime', 'XResolution', 'YResolution',
+    'Orientation', 'GPSInfo', 'DigitalZoomRatio', 'FocalLengthIn35mmFilm',
+    'ISOSpeedRatings', 'SceneType'
 ]
 
 
@@ -21,28 +27,96 @@ class ImageFolderView(ModelView):
 
     scan = Button("Scan for faces...")
 
-    view = View(
-        Item("model.path", style="readonly", show_label=False),
-        Item("model.data", editor=DataFrameEditor(columns=DISPLAYED_COLUMNS,
-                                                  update="data_updated"),
-             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",
-        ),
-        HGroup(
-            Spring(),
-            Item("scan", show_label=False, enabled_when="len(model.data) > 0"),
-            Spring(),
+    # Filters widgets
+    view_filter_controls = Bool
+
+    filtered_data = Instance(pd.DataFrame)
+
+    year_mask = Instance(pd.Series)
+
+    selected_years = List
+
+    all_years = List
+
+    selected_make = Enum(["All", "Canon", "Nikon", "Sony", "Apple", "samsung"])
+
+    make_mask = Instance(pd.Series)
+
+    def traits_view(self):
+        view = View(
+            Item("model.path", style="readonly", show_label=False),
+            HGroup(
+                Spring(),
+                Item("view_filter_controls"),
+            ),
+            HGroup(
+                Item("all_years", label="Years",
+                     editor=ListStrEditor(selected="selected_years",
+                                          multi_select=True)
+                     ),
+                Item("selected_make", label="Camera make"),
+                visible_when="view_filter_controls",
+            ),
+            Item("filtered_data",
+                 editor=DataFrameEditor(columns=DISPLAYED_COLUMNS,
+                                        update="data_updated"),
+                 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",
+            ),
+            HGroup(
+                Spring(),
+                Item("scan", show_label=False,
+                     enabled_when="len(model.data) > 0"),
+                Spring(),
+            ),
+            resizable=True, height=800
         )
-    )
+        return view
+
+    # Listener methods --------------------------------------------------------
 
     @observe("scan")
     def scan_for_faces(self, event):
         self.model.compute_num_faces()
 
+    @observe("selected_years")
+    def update_years(self, event):
+        self.year_mask = self.model.data['Year'].isin(self.selected_years)
+
+    @observe("selected_make")
+    def update_make(self, event):
+        if self.selected_make == "All":
+            self.make_mask = pd.Series(np.ones(len(self.model.data),
+                                               dtype=bool))
+        else:
+            self.make_mask = self.model.data['Make'] == self.selected_make
+
+    @observe("year_mask, make_mask")
+    def update_filtered_data(self, event):
+        self.filtered_data = self.model.data[self.year_mask & self.make_mask]
+
+    # Initialization methods --------------------------------------------------
+
+    def _make_mask_default(self):
+        return pd.Series(np.ones(len(self.model.data), dtype=bool))
+
+    def _year_mask_default(self):
+        return pd.Series(np.ones(len(self.model.data), dtype=bool))
+
+    def _filtered_data_default(self):
+        return self.model.data
+
+    def _all_years_default(self):
+        def parse_year(x):
+            return x.split(":")[0] if isinstance(x, str) else "unknown"
+
+        self.model.data['Year'] = self.model.data['DateTime'].apply(parse_year)
+        return sorted(self.model.data['Year'].unique().tolist())
+
 
 if __name__ == '__main__':
     from os.path import dirname, join