Make battery power manager algorithm configurable

With this, `SHIFTING_MATRYOSHKA` remains the default algorithm for
batteries, but can now be changed to the original `MATRYOSHKA`
algorithm when calling `microgrid.initialize()`.

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

Author: shsms

src/frequenz/sdk/microgrid/__init__.py

@@ -383,13 +383,16 @@
     producer,
     voltage_per_phase,
 )
+from ._power_managing import PowerManagerAlgorithm
 
 
 async def initialize(
     server_url: str,
     resampler_config: ResamplerConfig,
     *,
     api_power_request_timeout: timedelta = timedelta(seconds=5.0),
+    # pylint: disable-next: line-too-long
+    battery_power_manager_algorithm: PowerManagerAlgorithm = PowerManagerAlgorithm.SHIFTING_MATRYOSHKA,  # noqa: E501
 ) -> None:
     """Initialize the microgrid connection manager and the data pipeline.
 
@@ -404,15 +407,19 @@ async def initialize(
             the microgrid API.  When requests to components timeout, they will
             be marked as blocked for a short duration, during which time they
             will be unavailable from the corresponding component pools.
+        battery_power_manager_algorithm: The power manager algorithm to use for
+            batteries.
     """
     await connection_manager.initialize(server_url)
     await _data_pipeline.initialize(
         resampler_config,
         api_power_request_timeout=api_power_request_timeout,
+        battery_power_manager_algorithm=battery_power_manager_algorithm,
     )
 
 
 __all__ = [
+    "PowerManagerAlgorithm",
     "initialize",
     "consumer",
     "grid",

src/frequenz/sdk/microgrid/_data_pipeline.py

@@ -83,13 +83,16 @@ def __init__(
         self,
         resampler_config: ResamplerConfig,
         api_power_request_timeout: timedelta = timedelta(seconds=5.0),
+        battery_power_manager_algorithm: PowerManagerAlgorithm = PowerManagerAlgorithm.SHIFTING_MATRYOSHKA,  # noqa: E501
     ) -> None:
         """Create a `DataPipeline` instance.
 
         Args:
             resampler_config: Config to pass on to the resampler.
             api_power_request_timeout: Timeout to use when making power requests to
                 the microgrid API.
+            battery_power_manager_algorithm: The power manager algorithm to use for
+                batteries.
         """
         self._resampler_config: ResamplerConfig = resampler_config
 
@@ -103,7 +106,7 @@ def __init__(
         self._battery_power_wrapper = PowerWrapper(
             self._channel_registry,
             api_power_request_timeout=api_power_request_timeout,
-            power_manager_algorithm=PowerManagerAlgorithm.SHIFTING_MATRYOSHKA,
+            power_manager_algorithm=battery_power_manager_algorithm,
             default_power=DefaultPower.ZERO,
             component_class=Battery,
         )
@@ -514,6 +517,7 @@ async def _stop(self) -> None:
 async def initialize(
     resampler_config: ResamplerConfig,
     api_power_request_timeout: timedelta = timedelta(seconds=5.0),
+    battery_power_manager_algorithm: PowerManagerAlgorithm = PowerManagerAlgorithm.SHIFTING_MATRYOSHKA,  # noqa: E501
 ) -> None:
     """Initialize a `DataPipeline` instance.
 
@@ -523,6 +527,8 @@ async def initialize(
             the microgrid API.  When requests to components timeout, they will
             be marked as blocked for a short duration, during which time they
             will be unavailable from the corresponding component pools.
+        battery_power_manager_algorithm: The power manager algorithm to use for
+            batteries.
 
     Raises:
         RuntimeError: if the DataPipeline is already initialized.
@@ -531,7 +537,9 @@ async def initialize(
 
     if _DATA_PIPELINE is not None:
         raise RuntimeError("DataPipeline is already initialized.")
-    _DATA_PIPELINE = _DataPipeline(resampler_config, api_power_request_timeout)
+    _DATA_PIPELINE = _DataPipeline(
+        resampler_config, api_power_request_timeout, battery_power_manager_algorithm
+    )
 
 
 def frequency() -> GridFrequency: