Fix post-detection particle display

Fixes particles not being displayable post-detection. See #94.
ANP-Granular · Jul 23, 2024 · 0273b46 · 0273b46
1 parent 1eb0404
commit 0273b46
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/RodTracker/src/RodTracker/backend/rod_data.py b/RodTracker/src/RodTracker/backend/rod_data.py
@@ -835,7 +835,13 @@ def add_data(self, data: pd.DataFrame):
             cols_pos_3d = [
                 col for col in columns if re.fullmatch(RE_3D_POS, col)
             ]
-            self.cols_2D = [*cols_pos_2d, *cols_seen, "particle", "frame"]
+            self.cols_2D = [
+                *cols_pos_2d,
+                *cols_seen,
+                "particle",
+                "frame",
+                "color",
+            ]
             self.cols_3D = [*cols_pos_3d, "particle", "frame", "color"]
 
             # Display as a tree

diff --git a/RodTracker/src/RodTracker/ui/rodimagewidget.py b/RodTracker/src/RodTracker/ui/rodimagewidget.py
@@ -1273,7 +1273,8 @@ def extract_rods(self, data: pd.DataFrame, color: str) -> None:
             data = data[col_list]
         except KeyError:
             _logger.info(
-                f"Couldn't extract rods. Didn't find columns: " f"{col_list}"
+                f"Couldn't extract rods. Didn't find all necessary columns: "
+                f"{col_list}"
             )
             del self.rods
             return

diff --git a/RodTracker/src/RodTracker/ui/view3d.py b/RodTracker/src/RodTracker/ui/view3d.py
@@ -152,7 +152,7 @@ def update_rods(self, data: pd.DataFrame) -> None:
                     *mpl_colors.to_rgba(color, alpha=1.0)
                 )
             except ValueError as e:
-                _logger.error(
+                _logger.warning(
                     f"Unknown color for 3D display!\n{e.args}\n"
                     f"Using 'pink' instead."
                 )