Simple-Track 2.0.4__tar.gz → 2.0.6__tar.gz

{simple_track-2.0.4 → simple_track-2.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Simple-Track
-Version: 2.0.4
+Version: 2.0.6
 Summary: Threshold-based object tracking algorithm for 2D data 
 Author-email: Adam Gainford <adam.gainford@reading.ac.uk>, Thorwald Stein <t.h.m.stein@reading.ac.uk>
 License-Expression: MPL-2.0
@@ -33,13 +33,15 @@ Features are tracked between consecutive frames of data by projecting feature fi
 
 # Installation
 
-Simple-Track can be installed using PyPi:
+Simple-Track can be installed using PyPi or conda-forge:
 
 ```
 python3 -m pip install simple-track
 ```
-
-Coming soon to conda-forge and uv
+```
+conda install conda-forge::simple-track
+```
+Coming soon to uv
 
 # User Guide
 
@@ -177,10 +179,10 @@ Fields (`.field` files):
 
 Features (`.csv` or `.txt` files):
 * ID: Unique feature identifier that persists between frames (i.e., a feature retains the same id across all frames that it is tracked).
-* Centroid: (y, x) tuple containing central location of feature.
-* Size: Number of pixels spanned by the feature.
+* centroid: (y, x) tuple containing central location of feature.
+* size: Number of pixels spanned by the feature.
 * dydx: (dy, dx) tuple containing motion vector that translated feature to its location in the current frame from the previous frame.
-* extreme: Maximum value contained within the feature in the input data.
+* max: Maximum value contained within the feature in the input data.
 * lifetime: Number of timesteps the feature has existed for.
 * accreted: List of IDs of features that were accreted by this feature, if applicable.
 * parent: ID of parent feature that this feature split from, if applicable

{simple_track-2.0.4 → simple_track-2.0.6}/README.md

@@ -11,13 +11,15 @@ Features are tracked between consecutive frames of data by projecting feature fi
 
 # Installation
 
-Simple-Track can be installed using PyPi:
+Simple-Track can be installed using PyPi or conda-forge:
 
 ```
 python3 -m pip install simple-track
 ```
-
-Coming soon to conda-forge and uv
+```
+conda install conda-forge::simple-track
+```
+Coming soon to uv
 
 # User Guide
 
@@ -155,10 +157,10 @@ Fields (`.field` files):
 
 Features (`.csv` or `.txt` files):
 * ID: Unique feature identifier that persists between frames (i.e., a feature retains the same id across all frames that it is tracked).
-* Centroid: (y, x) tuple containing central location of feature.
-* Size: Number of pixels spanned by the feature.
+* centroid: (y, x) tuple containing central location of feature.
+* size: Number of pixels spanned by the feature.
 * dydx: (dy, dx) tuple containing motion vector that translated feature to its location in the current frame from the previous frame.
-* extreme: Maximum value contained within the feature in the input data.
+* max: Maximum value contained within the feature in the input data.
 * lifetime: Number of timesteps the feature has existed for.
 * accreted: List of IDs of features that were accreted by this feature, if applicable.
 * parent: ID of parent feature that this feature split from, if applicable

{simple_track-2.0.4 → simple_track-2.0.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "Simple-Track"
-version = "2.0.4"
+version = "2.0.6"
 authors = [
   { name="Adam Gainford", email="adam.gainford@reading.ac.uk" },
   { name="Thorwald Stein", email="t.h.m.stein@reading.ac.uk"} 

{simple_track-2.0.4 → simple_track-2.0.6}/src/Simple_Track.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Simple-Track
-Version: 2.0.4
+Version: 2.0.6
 Summary: Threshold-based object tracking algorithm for 2D data 
 Author-email: Adam Gainford <adam.gainford@reading.ac.uk>, Thorwald Stein <t.h.m.stein@reading.ac.uk>
 License-Expression: MPL-2.0
@@ -33,13 +33,15 @@ Features are tracked between consecutive frames of data by projecting feature fi
 
 # Installation
 
-Simple-Track can be installed using PyPi:
+Simple-Track can be installed using PyPi or conda-forge:
 
 ```
 python3 -m pip install simple-track
 ```
-
-Coming soon to conda-forge and uv
+```
+conda install conda-forge::simple-track
+```
+Coming soon to uv
 
 # User Guide
 
@@ -177,10 +179,10 @@ Fields (`.field` files):
 
 Features (`.csv` or `.txt` files):
 * ID: Unique feature identifier that persists between frames (i.e., a feature retains the same id across all frames that it is tracked).
-* Centroid: (y, x) tuple containing central location of feature.
-* Size: Number of pixels spanned by the feature.
+* centroid: (y, x) tuple containing central location of feature.
+* size: Number of pixels spanned by the feature.
 * dydx: (dy, dx) tuple containing motion vector that translated feature to its location in the current frame from the previous frame.
-* extreme: Maximum value contained within the feature in the input data.
+* max: Maximum value contained within the feature in the input data.
 * lifetime: Number of timesteps the feature has existed for.
 * accreted: List of IDs of features that were accreted by this feature, if applicable.
 * parent: ID of parent feature that this feature split from, if applicable

{simple_track-2.0.4 → simple_track-2.0.6}/src/run_simple_track.py

@@ -8,7 +8,9 @@ def run_tracking():
     # Need to make sure that changes don't affect pyproject.toml entry points
     # easiest just to pass the parser in to run_tracking.
     parser = argparse.ArgumentParser(description="Run Simple-Track")
-    parser.add_argument("configs", help="Path to one or more yaml config files.")
+    parser.add_argument(
+        "configs", nargs="+", help="Path to one or more yaml config files."
+    )
     args = parser.parse_args()
 
     for config_path in args.configs:

{simple_track-2.0.4 → simple_track-2.0.6}/src/simpletrack/feature.py

@@ -9,7 +9,7 @@ from simpletrack.utils import check_arrays, check_valid_ids, native
 class Feature:
     """
     Object containing details about a specific feature, including its id, time,
-    centroid, extreme value, lifetime, and whether it has undergone any
+    centroid, maximum value, lifetime, and whether it has undergone any
     mergers of splits in the current timestep.
     """
 
@@ -30,7 +30,8 @@ class Feature:
         self._parent = None
         self._children = []
         self._dydx = ()
-        self._extreme = None
+        self._max = None
+        self._mean = None
 
     def __repr__(self) -> str:
         repr_str = f"Feature id: {self._id} (provisionally {self._provisional_id}), "
@@ -140,11 +141,18 @@ class Feature:
         return native(self._dydx)
 
     @property
-    def extreme(self) -> float:
+    def max(self) -> float:
         """
         Maximum value of the Feature in the raw input data
         """
-        return self._extreme
+        return self._max
+
+    @property
+    def mean(self) -> float:
+        """
+        Mean value of the Feature in the raw input data
+        """
+        return self._mean
 
     @coords.setter
     def coords(self, new_coords: NDArray[np.integer]) -> None:
@@ -184,9 +192,13 @@ class Feature:
         id_of_accreting_feature = check_valid_ids(id_of_accreting_feature)
         self._accreted_in_next_frame_by = id_of_accreting_feature
 
-    @extreme.setter
-    def extreme(self, extreme_val: float) -> None:
-        self._extreme = extreme_val
+    @max.setter
+    def max(self, max_val: float) -> None:
+        self._max = max_val
+
+    @mean.setter
+    def mean(self, mean_val: float) -> None:
+        self._mean = mean_val
 
     def calculate_centroid(self) -> tuple:
         """
@@ -279,7 +291,8 @@ class Feature:
             "size": self.get_size(),
             # native() does not convert dydx to python type for some reason
             "dydx": tuple([val.item() for val in self._dydx]),
-            "extreme": self._extreme,
+            "max": self._max,
+            "mean": self._mean,
             "lifetime": self._lifetime,
             "accreted": self._accreted,
             # This will not be output properly in the current workflow, since each

{simple_track-2.0.4 → simple_track-2.0.6}/src/simpletrack/frame.py

@@ -176,6 +176,8 @@ class Frame:
         # Provisionally set the lifetime field to 1 anywhere there is a feature
         self._lifetime_field = np.zeros_like(self._feature_field)
         self._lifetime_field[self._feature_field > 0] = 1
+        if np.max(self._feature_field) == 0:
+            return
         self.max_id = int(np.max(self._feature_field))
         self.populate_features()
 
@@ -211,7 +213,8 @@ class Frame:
             )
             # If raw field is not None, use this to find max value within Feature
             if self.raw_field is not None:
-                feature.extreme = max(self.raw_field[feature_mask])
+                feature.max = np.max(self.raw_field[feature_mask])
+                feature.mean = np.mean(self.raw_field[feature_mask])
             self._features[feature_id] = feature
 
     def assign_displacements(self, y_flow: NDArray, x_flow: NDArray) -> None:
@@ -387,6 +390,9 @@ class Timeline:
     def __init__(self):
         self.timeline = {}
 
+    def __len__(self) -> int:
+        return len(self.timeline)
+
     def add_to_timelime(self, frame: Frame) -> None:
         """
         Add the input frame to the timeline, using the frame.get_time() to

{simple_track-2.0.4 → simple_track-2.0.6}/src/simpletrack/track.py

@@ -58,7 +58,7 @@ class Tracker:
             self.frame_tracker = FrameTracker()
 
         if "OUTPUT" in self.config:
-            self.skip_tracking = self.config["TRACKING"].get("skip_tracking", False)
+            self.skip_tracking = self.config["OUTPUT"].get("skip_tracking", False)
             output_path = self.config["OUTPUT"].get("path", "./output")
             expt_name = self.config["OUTPUT"].get(
                 "experiment_name", "Simple-Track Experiment"

{simple_track-2.0.4 → simple_track-2.0.6}/tests/test_frame.py

@@ -103,7 +103,7 @@ def test_populate_features_valid_feature_field():
     assert test_frame.features == expected_dict
 
 
-def test_populate_features_sets_extreme_property():
+def test_populate_features_sets_max_property():
     test_time = dt.datetime.now()
     test_frame = Frame()
     test_frame.time = test_time
@@ -123,8 +123,33 @@ def test_populate_features_sets_extreme_property():
     test_frame.raw_field = test_raw_field
     test_frame.populate_features()
 
-    assert test_frame.get_feature(1).extreme == 10
-    assert test_frame.get_feature(2).extreme == 20
+    assert test_frame.get_feature(1).max == 10
+    assert test_frame.get_feature(2).max == 20
+
+
+def test_populate_features_sets_mean_property():
+    test_time = dt.datetime.now()
+    test_frame = Frame()
+    test_frame.time = test_time
+
+    test_feature_field = test_field.copy()
+    test_feature_field[2:6, 2:6] = 1
+    test_feature_field[6:9, 6:9] = 2
+
+    test_raw_field = test_field.copy()
+    test_raw_field[2:6, 2:6] = 10
+    test_raw_field[2:6, 4:6] = 50
+    test_raw_field[6:9, 6:9] = 20
+    # Set another higher maximum within the field to check mean
+    # only picks up the values within the feature mask
+    test_raw_field[0:2, 0:2] = 100
+
+    test_frame.feature_field = test_feature_field
+    test_frame.raw_field = test_raw_field
+    test_frame.populate_features()
+
+    assert test_frame.get_feature(1).mean == 30
+    assert test_frame.get_feature(2).mean == 20
 
 
 def test_populate_features_invalid_negative_features():
@@ -175,6 +200,39 @@ def test_populate_features_with_no_feature_field():
         )
 
 
+def test_populate_features_with_feature_field_but_no_identifiable_features():
+    test_time = dt.datetime.now()
+    test_frame = Frame()
+    test_frame.time = test_time
+
+    test_feature_field = np.zeros((10, 10))
+    test_frame.feature_field = test_feature_field
+
+    test_frame.populate_features()
+
+    if len(test_frame.features) != 0:
+        raise TypeError(
+            f"No features expected for this test, got {len(test_frame.get_features())}"
+        )
+
+
+def test_identify_and_populate_features_with_raw_field_but_no_identifiable_features():
+    test_time = dt.datetime.now()
+    test_frame = Frame()
+    test_frame.time = test_time
+
+    test_raw_field = np.zeros((10, 10))
+    test_frame.raw_field = test_raw_field
+
+    test_frame.identify_features(threshold=1, min_size=1)
+    # test_frame.populate_features() automatically happens within identify_features
+
+    if len(test_frame.features) != 0:
+        raise TypeError(
+            f"No features expected for this test, got {len(test_frame.get_features())}"
+        )
+
+
 def test_assign_displacements_valid_inputs():
     test_frame = Frame()
     y_flow = np.ones((10, 10))