implemented label in cluster estimation

modules/cluster_estimation/cluster_estimation.py

@@ -55,6 +55,32 @@ class ClusterEstimation:
     __WEIGHT_DROP_THRESHOLD = 0.1
     __MAX_COVARIANCE_THRESHOLD = 10
 
+    @staticmethod
+    def check_create_arguments(
+        min_activation_threshold: int,
+        min_new_points_to_run: int,
+        max_num_components: int,
+        random_state: int,
+    ) -> bool:
+        """
+        Checks if a valid cluster estimation object can be constructed.
+
+        See `ClusterEstimation` for parameter descriptions.
+        """
+        if min_activation_threshold < max_num_components:
+            return False
+
+        if min_new_points_to_run < 0:
+            return False
+
+        if max_num_components < 1:
+            return False
+
+        if random_state < 0:
+            return False
+
+        return True
+
     @classmethod
     def create(
         cls,
@@ -85,16 +111,11 @@ def create(
 
         RETURNS: The ClusterEstimation object if all conditions pass, otherwise False, None
         """
-        if min_activation_threshold < max_num_components:
-            return False, None
-
-        if min_new_points_to_run < 0:
-            return False, None
-
-        if max_num_components < 1:
-            return False, None
+        is_valid_arguments = ClusterEstimation.check_create_arguments(
+            min_activation_threshold, min_new_points_to_run, max_num_components, random_state
+        )
 
-        if random_state < 0:
+        if not is_valid_arguments:
             return False, None
 
         return True, ClusterEstimation(
@@ -211,21 +232,20 @@ def run(
         model_output = self.__filter_by_covariances(model_output)
 
         # Create output list of remaining valid clusters
-        detections_in_world = []
+        objects_in_world = []
         for cluster in model_output:
             result, landing_pad = object_in_world.ObjectInWorld.create(
-                cluster[0][0],
-                cluster[0][1],
-                cluster[2],
+                cluster[0][0], cluster[0][1], cluster[2]
             )
 
-            if result:
-                detections_in_world.append(landing_pad)
-            else:
-                self.__logger.warning("Failed to create ObjectInWorld object")
+            if not result:
+                self.__logger.error("Failed to create ObjectInWorld object")
+                return False, None
+
+            objects_in_world.append(landing_pad)
 
-        self.__logger.info(detections_in_world)
-        return True, detections_in_world
+        self.__logger.info(objects_in_world)
+        return True, objects_in_world
 
     def __decide_to_run(self, run_override: bool) -> bool:
         """

modules/cluster_estimation/cluster_estimation_by_label.py

@@ -0,0 +1,162 @@
+"""
+Cluster estimation by label.
+"""
+
+from . import cluster_estimation
+from .. import detection_in_world
+from .. import object_in_world
+from ..common.modules.logger import logger
+
+
+class ClusterEstimationByLabel:
+    """
+    Cluster estimation filtered on label.
+
+    ATTRIBUTES
+    ----------
+    min_activation_threshold: int
+        Minimum total data points before model runs. Must be at least max_num_components.
+
+    min_new_points_to_run: int
+        Minimum number of new data points that must be collected before running model.
+
+    max_num_components: int
+        Max number of real landing pads. Must be at least 1.
+
+    random_state: int
+        Seed for randomizer, to get consistent results.
+
+    local_logger: Logger
+        For logging error and debug messages.
+
+    METHODS
+    -------
+    run()
+        Cluster estimation filtered by label.
+    """
+
+    __create_key = object()
+
+    @classmethod
+    def create(
+        cls,
+        min_activation_threshold: int,
+        min_new_points_to_run: int,
+        max_num_components: int,
+        random_state: int,
+        local_logger: logger.Logger,
+    ) -> "tuple[True, ClusterEstimationByLabel] | tuple[False, None]":
+        """
+        See `ClusterEstimation` for parameter descriptions.
+
+        Return: Success, cluster estimation by label object.
+        """
+
+        is_valid_arguments = cluster_estimation.ClusterEstimation.check_create_arguments(
+            min_activation_threshold, min_new_points_to_run, max_num_components, random_state
+        )
+
+        if not is_valid_arguments:
+            return False, None
+
+        return True, ClusterEstimationByLabel(
+            cls.__create_key,
+            min_activation_threshold,
+            min_new_points_to_run,
+            max_num_components,
+            random_state,
+            local_logger,
+        )
+
+    def __init__(
+        self,
+        class_private_create_key: object,
+        min_activation_threshold: int,
+        min_new_points_to_run: int,
+        max_num_components: int,
+        random_state: int,
+        local_logger: logger.Logger,
+    ) -> None:
+        """
+        Private constructor, use create() method.
+        """
+        assert (
+            class_private_create_key is ClusterEstimationByLabel.__create_key
+        ), "Use create() method"
+
+        # Construction arguments for `ClusterEstimation`
+        self.__min_activation_threshold = min_activation_threshold
+        self.__min_new_points_to_run = min_new_points_to_run
+        self.__max_num_components = max_num_components
+        self.__random_state = random_state
+        self.__local_logger = local_logger
+
+        # Cluster model corresponding to each label
+        # Each cluster estimation object stores the detections given to in its __all_points bucket across runs
+        self.__label_to_cluster_estimation_model: dict[
+            int, cluster_estimation.ClusterEstimation
+        ] = {}
+
+    def run(
+        self,
+        input_detections: list[detection_in_world.DetectionInWorld],
+        run_override: bool,
+    ) -> tuple[True, dict[int, list[object_in_world.ObjectInWorld]]] | tuple[False, None]:
+        """
+        See `ClusterEstimation` for parameter descriptions.
+
+        RETURNS
+        -------
+        model_ran: bool
+            True if ClusterEstimation object successfully ran its estimation model, False otherwise.
+
+        labels_to_objects: dict[int, list[object_in_world.ObjectInWorld] or None.
+            Dictionary where the key is a label and the value is a list of all cluster detections with that label.
+            ObjectInWorld objects don't have a label property, but they are sorted into label categories in the dictionary.
+        """
+        label_to_detections: dict[int, list[detection_in_world.DetectionInWorld]] = {}
+
+        # Filtering detections by label
+        for detection in input_detections:
+            if not detection.label in label_to_detections:
+                label_to_detections[detection.label] = []
+
+            label_to_detections[detection.label].append(detection)
+
+        labels_to_objects: dict[int, list[object_in_world.ObjectInWorld]] = {}
+
+        for label, detections in label_to_detections.items():
+            # Create cluster estimation for label if it doesn't exist
+            if not label in self.__label_to_cluster_estimation_model:
+                result, cluster_model = cluster_estimation.ClusterEstimation.create(
+                    self.__min_activation_threshold,
+                    self.__min_new_points_to_run,
+                    self.__max_num_components,
+                    self.__random_state,
+                    self.__local_logger,
+                )
+                if not result:
+                    self.__local_logger.error(
+                        f"Failed to create cluster estimation for label {label}"
+                    )
+                    return False, None
+
+                self.__label_to_cluster_estimation_model[label] = cluster_model
+
+            # Runs cluster estimation for specific label
+            result, clusters = self.__label_to_cluster_estimation_model[label].run(
+                detections,
+                run_override,
+            )
+
+            if not result:
+                self.__local_logger.error(
+                    f"Failed to run cluster estimation model for label {label}"
+                )
+                return False, None
+
+            if not label in labels_to_objects:
+                labels_to_objects[label] = []
+            labels_to_objects[label] += clusters
+
+        return True, labels_to_objects

modules/cluster_estimation/cluster_estimation_worker.py

@@ -26,26 +26,9 @@ def cluster_estimation_worker(
 
     PARAMETERS
     ----------
-    min_activation_threshold: int
-        Minimum total data points before model runs.
 
-    min_new_points_to_run: int
-        Minimum number of new data points that must be collected before running model.
+    See `ClusterEstimation` for parameter descriptions.
 
-    max_num_components: int
-        Max number of real landing pads.
-
-    random_state: int
-        Seed for randomizer, to get consistent results.
-
-    input_queue: queue_proxy_wrapper.QueuePRoxyWrapper
-        Data queue.
-
-    output_queue: queue_proxy_wrapper.QueuePRoxyWrapper
-        Data queue.
-
-    worker_controller: worker_controller.WorkerController
-        How the main process communicates to this worker process.
     """
     worker_name = pathlib.Path(__file__).stem
     process_id = os.getpid()

modules/object_in_world.py

@@ -21,7 +21,12 @@ def create(
         if spherical_variance < 0.0:
             return False, None
 
-        return True, ObjectInWorld(cls.__create_key, location_x, location_y, spherical_variance)
+        return True, ObjectInWorld(
+            cls.__create_key,
+            location_x,
+            location_y,
+            spherical_variance,
+        )
 
     def __init__(
         self,