feat: add "hide unverified" button

Closes #55
mbari-org · Dec 13, 2023 · 52b477d · 52b477d
1 parent b0ba51f
commit 52b477d
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 1 deletion.
diff --git a/vars_gridview/assets/gridview.ui b/vars_gridview/assets/gridview.ui
@@ -245,6 +245,23 @@
                     </property>
                   </widget>
                 </item>
+                <item>
+                  <widget class="QCheckBox" name="hideUnlabeled">
+                    <property name="font">
+                      <font>
+                        <pointsize>10</pointsize>
+                        <weight>75</weight>
+                        <bold>true</bold>
+                      </font>
+                    </property>
+                    <property name="text">
+                      <string>Hide Unverified</string>
+                    </property>
+                    <property name="checked">
+                      <bool>false</bool>
+                    </property>
+                  </widget>
+                </item>
                 <item>
                   <widget class="QPushButton" name="sortButton">
                     <property name="font">

diff --git a/vars_gridview/lib/image_mosaic.py b/vars_gridview/lib/image_mosaic.py
@@ -48,6 +48,7 @@ def __init__(
         self._rect_widgets: List[RectWidget] = []
         self.roi_map = {}
         self._hide_labeled = True
+        self._hide_unlabeled = True
         self.hide_discarded = True
         self.hide_to_review = True
         self.n_columns = 0
@@ -650,7 +651,7 @@ def render_mosaic(self):
         rect_widgets_to_render = [
             rw
             for rw in self._rect_widgets
-            if not (self._hide_labeled and rw.is_verified)
+            if (not rw.is_verified and not self._hide_unlabeled) or (rw.is_verified and not self._hide_labeled)
         ]
 
         # Hide all rect widgets that we aren't rendering

diff --git a/vars_gridview/scripts/run.py b/vars_gridview/scripts/run.py
@@ -124,6 +124,7 @@ def __init__(self, app):
         self.ui.zoomSpinBox.valueChanged.connect(self.update_zoom)
         # self.ui.sortMethod.currentTextChanged.connect(self.update_layout)
         self.ui.hideLabeled.stateChanged.connect(self.update_layout)
+        self.ui.hideUnlabeled.stateChanged.connect(self.update_layout)
         self.ui.styleComboBox.currentTextChanged.connect(self._style_gui)
         self.ui.openVideo.clicked.connect(self.open_video)
         self.ui.sortButton.clicked.connect(self._sort_widgets)
@@ -420,6 +421,7 @@ def _do_query(self):
         self.image_mosaic.hide_discarded = False
         self.image_mosaic.hide_to_review = False
         self.image_mosaic._hide_labeled = self.ui.hideLabeled.isChecked()
+        self.image_mosaic._hide_unlabeled = self.ui.hideUnlabeled.isChecked()
 
         default_sort_method = RecordedTimestampSort
         self.image_mosaic.sort_rect_widgets(default_sort_method)
@@ -667,6 +669,7 @@ def update_layout(self):
         self.image_mosaic.hide_discarded = False
         self.image_mosaic.hide_to_review = False
         self.image_mosaic._hide_labeled = self.ui.hideLabeled.isChecked()
+        self.image_mosaic._hide_unlabeled = self.ui.hideUnlabeled.isChecked()
         # self.image_mosaic.sort_rect_widgets(method)
         self.image_mosaic.render_mosaic()