Add a PVManager implementation

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

Author: shsms

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

@@ -6,9 +6,11 @@
 from ._battery_manager import BatteryManager
 from ._component_manager import ComponentManager
 from ._ev_charger_manager import EVChargerManager
+from ._pv_inverter_manager import PVManager
 
 __all__ = [
     "BatteryManager",
     "ComponentManager",
     "EVChargerManager",
+    "PVManager",
 ]

src/frequenz/sdk/actor/power_distributing/_component_managers/_pv_inverter_manager/__init__.py

@@ -0,0 +1,8 @@
+# License: MIT
+# Copyright © 2024 Frequenz Energy-as-a-Service GmbH
+
+"""Manage PV inverters for the power distributor."""
+
+from ._pv_inverter_manager import PVManager
+
+__all__ = ["PVManager"]

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

@@ -0,0 +1,211 @@
+# License: MIT
+# Copyright © 2024 Frequenz Energy-as-a-Service GmbH
+
+"""Manage PV inverters for the power distributor."""
+
+import asyncio
+import collections.abc
+import logging
+from datetime import timedelta
+
+import grpc
+from frequenz.channels import Broadcast, Sender
+from frequenz.client.microgrid import ComponentCategory, InverterData, InverterType
+from typing_extensions import override
+
+from ....._internal._channels import LatestValueCache
+from ....._internal._math import is_close_to_zero
+from .....microgrid import connection_manager
+from .....timeseries import Power
+from ..._component_pool_status_tracker import ComponentPoolStatusTracker
+from ..._component_status import ComponentPoolStatus, PVInverterStatusTracker
+from ...request import Request
+from ...result import PartialFailure, Result, Success
+from .._component_manager import ComponentManager
+
+_logger = logging.getLogger(__name__)
+
+
+class PVManager(ComponentManager):
+    """Manage PV inverters for the power distributor."""
+
+    @override
+    def __init__(
+        self,
+        component_pool_status_sender: Sender[ComponentPoolStatus],
+        results_sender: Sender[Result],
+    ) -> None:
+        """Initialize this instance.
+
+        Args:
+            component_pool_status_sender: Channel for sending information about which
+                components are expected to be working.
+            results_sender: Channel for sending results of power distribution.
+        """
+        self._results_sender = results_sender
+        self._pv_inverter_ids = self._get_pv_inverter_ids()
+
+        self._component_pool_status_tracker = (
+            ComponentPoolStatusTracker(
+                component_ids=self._pv_inverter_ids,
+                component_status_sender=component_pool_status_sender,
+                max_data_age=timedelta(seconds=10.0),
+                max_blocking_duration=timedelta(seconds=30.0),
+                component_status_tracker_type=PVInverterStatusTracker,
+            )
+            if self._pv_inverter_ids
+            else None
+        )
+        self._component_data_caches: dict[int, LatestValueCache[InverterData]] = {}
+        self._target_power = Power.zero()
+        self._target_power_channel = Broadcast[Request](name="target_power")
+        self._target_power_tx = self._target_power_channel.new_sender()
+        self._task: asyncio.Task[None] | None = None
+        self._latest_request: Request = Request(Power.zero(), set())
+
+    @override
+    def component_ids(self) -> collections.abc.Set[int]:
+        """Return the set of PV inverter ids."""
+        return self._pv_inverter_ids
+
+    @override
+    async def start(self) -> None:
+        """Start the PV inverter manager."""
+        self._component_data_caches = {
+            inv_id: LatestValueCache(
+                await connection_manager.get().api_client.inverter_data(inv_id)
+            )
+            for inv_id in self._pv_inverter_ids
+        }
+
+    @override
+    async def stop(self) -> None:
+        """Stop the PV inverter manager."""
+        await asyncio.gather(
+            *[tracker.stop() for tracker in self._component_data_caches.values()]
+        )
+        if self._component_pool_status_tracker:
+            await self._component_pool_status_tracker.stop()
+
+    @override
+    async def distribute_power(self, request: Request) -> None:
+        """Distribute the requested power to the PV inverters.
+
+        Args:
+            request: Request to get the distribution for.
+
+        Raises:
+            ValueError: If no PV inverters are present in the component graph, but
+                component_ids are provided in the request.
+        """
+        remaining_power = request.power
+        allocations: dict[int, Power] = {}
+        if not self._component_pool_status_tracker:
+            if not request.component_ids:
+                await self._results_sender.send(
+                    Success(
+                        succeeded_components=set(),
+                        succeeded_power=Power.zero(),
+                        excess_power=remaining_power,
+                        request=request,
+                    )
+                )
+                return
+            raise ValueError(
+                "Cannot distribute power to PV inverters without any inverters"
+            )
+        for inv_id in self._component_pool_status_tracker.get_working_components(
+            request.component_ids
+        ):
+            if remaining_power < Power.zero() or is_close_to_zero(
+                remaining_power.as_watts()
+            ):
+                break
+            inv_data = self._component_data_caches[inv_id]
+            if not inv_data.has_value():
+                allocations[inv_id] = Power.zero()
+                # Can't get device bounds, so can't use inverter.
+                continue
+            discharge_bounds = Power.from_watts(
+                inv_data.get().active_power_inclusion_lower_bound
+            )
+            allocated_power = min(remaining_power, discharge_bounds)
+            allocations[inv_id] = allocated_power
+            remaining_power -= allocated_power
+
+        _logger.debug(
+            "Distributing %s to PV inverters %s",
+            request.power,
+            allocations,
+        )
+        await self._set_api_power(request, allocations, remaining_power)
+
+    async def _set_api_power(
+        self, request: Request, allocations: dict[int, Power], remaining_power: Power
+    ) -> None:
+        api_client = connection_manager.get().api_client
+        tasks: dict[int, asyncio.Task[None]] = {}
+        for component_id, power in allocations.items():
+            tasks[component_id] = asyncio.create_task(
+                api_client.set_power(component_id, power.as_watts())
+            )
+            _, pending = await asyncio.wait(
+                tasks.values(),
+                timeout=request.request_timeout.total_seconds(),
+                return_when=asyncio.ALL_COMPLETED,
+            )
+        for task in pending:
+            task.cancel()
+            await asyncio.gather(*pending, return_exceptions=True)
+
+        failed_components: set[int] = set()
+        succeeded_components: set[int] = set()
+        failed_power = Power.zero()
+        for component_id, task in tasks.items():
+            exc = task.exception()
+            if exc is not None:
+                failed_components.add(component_id)
+                failed_power += allocations[component_id]
+            else:
+                succeeded_components.add(component_id)
+
+            match task.exception():
+                case asyncio.CancelledError:
+                    _logger.warning(
+                        "Timeout while setting power to EV charger %s", component_id
+                    )
+                case grpc.aio.AioRpcError as err:
+                    _logger.warning(
+                        "Error while setting power to EV charger %s: %s",
+                        component_id,
+                        err,
+                    )
+        if failed_components:
+            await self._results_sender.send(
+                PartialFailure(
+                    failed_components=failed_components,
+                    succeeded_components=succeeded_components,
+                    failed_power=failed_power,
+                    succeeded_power=self._target_power - failed_power,
+                    excess_power=remaining_power,
+                    request=self._latest_request,
+                )
+            )
+        await self._results_sender.send(
+            Success(
+                succeeded_components=succeeded_components,
+                succeeded_power=self._target_power,
+                excess_power=remaining_power,
+                request=self._latest_request,
+            )
+        )
+
+    def _get_pv_inverter_ids(self) -> collections.abc.Set[int]:
+        """Return the IDs of all PV inverters present in the component graph."""
+        return {
+            inv.component_id
+            for inv in connection_manager.get().component_graph.components(
+                component_categories={ComponentCategory.INVERTER}
+            )
+            if inv.type == InverterType.SOLAR
+        }