Add a _get_bounds method in PowerDistributingActor

This method will replace the `_get_{upper,lower}_bound` methods in
subsequent commits.

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

src/frequenz/sdk/actor/power_distributing/power_distributing.py

@@ -39,7 +39,7 @@
 from ...power import DistributionAlgorithm, DistributionResult, InvBatPair
 from ._battery_pool_status import BatteryPoolStatus, BatteryStatus
 from .request import Request
-from .result import Error, OutOfBound, PartialFailure, Result, Success
+from .result import Error, OutOfBound, PartialFailure, PowerBounds, Result, Success
 
 _logger = logging.getLogger(__name__)
 
@@ -226,6 +226,51 @@ def __init__(
             bat_id: None for bat_id, _ in self._bat_inv_map.items()
         }
 
+    def _get_bounds(self, batteries: abc.Set[int], include_broken: bool) -> PowerBounds:
+        """Get power bounds for given batteries.
+
+        Args:
+            batteries: List of batteries
+            include_broken: whether all batteries in the batteries set in the
+                request must be used regardless the status.
+
+        Returns:
+            Power bounds for given batteries.
+        """
+        pairs_data: List[InvBatPair] = self._get_components_data(
+            batteries, include_broken
+        )
+        return PowerBounds(
+            inclusion_lower=sum(
+                max(
+                    battery.power_inclusion_lower_bound,
+                    inverter.active_power_inclusion_lower_bound,
+                )
+                for battery, inverter in pairs_data
+            ),
+            inclusion_upper=sum(
+                min(
+                    battery.power_inclusion_upper_bound,
+                    inverter.active_power_inclusion_upper_bound,
+                )
+                for battery, inverter in pairs_data
+            ),
+            exclusion_lower=sum(
+                min(
+                    battery.power_exclusion_lower_bound,
+                    inverter.active_power_exclusion_lower_bound,
+                )
+                for battery, inverter in pairs_data
+            ),
+            exclusion_upper=sum(
+                max(
+                    battery.power_exclusion_upper_bound,
+                    inverter.active_power_exclusion_upper_bound,
+                )
+                for battery, inverter in pairs_data
+            ),
+        )
+
     def _get_upper_bound(self, batteries: abc.Set[int], include_broken: bool) -> float:
         """Get total upper bound of power to be set for given batteries.
 

src/frequenz/sdk/actor/power_distributing/result.py

@@ -77,6 +77,16 @@ class Error(Result):
     """The error message explaining why error happened."""
 
 
+@dataclasses.dataclass
+class PowerBounds:
+    """Inclusion and exclusion power bounds for requested batteries."""
+
+    inclusion_lower: float
+    exclusion_lower: float
+    exclusion_upper: float
+    inclusion_upper: float
+
+
 @dataclasses.dataclass
 class OutOfBound(Result):
     """Result returned when the power was not set because it was out of bounds.