Add unit commitment extension

Signed-off-by: Davey Elder <iandavidelder@gmail.com>

Author: idelder

temoa/_internal/table_data_puller.py

@@ -246,6 +246,11 @@ def poll_flow_results(model: TemoaModel, epsilon: float = 1e-5) -> dict[FI, dict
                         continue
                     res[fi][FlowType.OUT] = flow
 
+    if 'unit_commitment' in model.enabled_extensions:
+        import temoa.extensions.unit_commitment.core.data_puller as uc
+
+        uc.poll_startup_input_results(model, res, epsilon)
+
     return res
 
 
@@ -494,6 +499,10 @@ def poll_cost_results(
         import temoa.extensions.economies_of_scale.core.data_puller as eos
 
         eos.poll_costs(model, exchange_costs, entries, p_0_true, epsilon)
+    if 'unit_commitment' in model.enabled_extensions:
+        import temoa.extensions.unit_commitment.core.data_puller as uc
+
+        uc.poll_startup_cost_results(model, exchange_costs, entries, p_0_true, epsilon)
 
     exchange_entries = exchange_costs.get_entries()
     return entries, exchange_entries
@@ -558,6 +567,11 @@ def poll_emissions(
             * model.emission_activity[r, e, i, t, v, o]
         )
 
+    if 'unit_commitment' in model.enabled_extensions:
+        from temoa.extensions.unit_commitment.core import data_puller as uc
+
+        uc.poll_emission_results(model, flows)
+
     # gather costs
     ud_costs: dict[tuple[Region, Period, Technology, Vintage], float] = defaultdict(float)
     d_costs: dict[tuple[Region, Period, Technology, Vintage], float] = defaultdict(float)

temoa/_internal/table_writer.py

@@ -483,6 +483,16 @@ def write_capacity_tables(self, model: TemoaModel, iteration: int | None = None)
         cap_data = poll_capacity_results(model=model, epsilon=self.output_threshold_capacity)
         self._insert_capacity_results(cap_data=cap_data, iteration=iteration)
 
+        if 'unit_commitment' in model.enabled_extensions:
+            import temoa.extensions.unit_commitment.core.data_puller as uc_puller
+
+            uc_puller.write_uc_results(
+                model=model,
+                writer=self,
+                iteration=iteration,
+                epsilon=self.output_threshold_capacity,
+            )
+
     def _insert_capacity_results(self, cap_data: CapData, iteration: int | None) -> None:
         if self.tech_sectors is None:
             raise RuntimeError('tech sectors not available... code error')

temoa/components/commodities.py

@@ -425,6 +425,11 @@ def commodity_balance_constraint(
             for s_o in model.process_outputs_by_input[r, p, s_t, s_v, c]
         )
 
+    if 'unit_commitment' in model.enabled_extensions:
+        from temoa.extensions.unit_commitment.components import startup
+
+        consumed += startup.uc_startup_input_rpsdc(model, r, p, s, d, c)
+
     if (r, p, c) in model.capacity_consumption_techs:
         # Consumed by building capacity
         # Assume evenly distributed over a year
@@ -573,6 +578,11 @@ def annual_commodity_balance_constraint(
             for s_o in model.process_outputs_by_input[r, p, s_t, s_v, c]
         )
 
+    if 'unit_commitment' in model.enabled_extensions:
+        from temoa.extensions.unit_commitment.components import startup
+
+        consumed += startup.uc_startup_input_rpc(model, r, p, c)
+
     if (r, p, c) in model.capacity_consumption_techs:
         # Consumed by building capacity
         # Assume evenly distributed over a year

temoa/components/limits.py

@@ -795,6 +795,11 @@ def limit_emission_constraint(
         if (reg, p, S_t, S_v) in model.process_inputs
     )
 
+    if 'unit_commitment' in model.enabled_extensions:
+        from temoa.extensions.unit_commitment.components import startup
+
+        process_emissions += startup.uc_startup_emissions_rpe(model, regions, p, e)
+
     embodied_emissions = quicksum(
         model.v_new_capacity[reg, t, v]
         * value(model.emission_embodied[reg, e, t, v])

temoa/components/time.py

@@ -336,9 +336,8 @@ def create_time_season_to_sequential(model: TemoaModel) -> None:
                 f'No data in time_season_sequential but time_sequencing parameter set to '
                 f'{model.time_sequencing.first()} and inter-season features used. '
                 'time_season_sequential must be filled for this type of time sequencing if '
-                'seasonal storage or inter-season constraints like ramp_up/ramp_down are used. '
-                'Check '
-                'the config file.'
+                'seasonal storage or inter-season constraints like ramp_up/ramp_down or'
+                'unit_commitment are used. Check the config file.'
             )
             logger.error(msg)
             raise ValueError(msg)

temoa/core/model.py

@@ -831,10 +831,15 @@ def __init__(
 
         self.progress_marker_4 = BuildAction(['Starting to build Constraints'], rule=progress_check)
 
-        # Declare constraints to calculate derived decision variables
-        self.capacity_constraint = Constraint(
-            self.capacity_constraint_rpsdtv, rule=capacity.capacity_constraint
-        )
+        if 'unit_commitment' in self.enabled_extensions:
+            # Hijack the capacity constraint and load params for other injections
+            import temoa.extensions.unit_commitment.core.model as uc
+
+            uc.register_early_components(self)
+        else:
+            self.capacity_constraint = Constraint(
+                self.capacity_constraint_rpsdtv, rule=capacity.capacity_constraint
+            )
 
         self.capacity_annual_constraint_rptv = Set(
             dimen=4, initialize=capacity.capacity_annual_constraint_indices
@@ -962,12 +967,13 @@ def __init__(
         self.ramp_down_constraint_rpsdtv = Set(
             dimen=6, initialize=operations.ramp_down_constraint_indices
         )
-        self.ramp_down_constraint = Constraint(
-            self.ramp_down_constraint_rpsdtv, rule=operations.ramp_down_constraint
-        )
-        self.ramp_up_constraint = Constraint(
-            self.ramp_up_constraint_rpsdtv, rule=operations.ramp_up_constraint
-        )
+        if 'unit_commitment' not in self.enabled_extensions:
+            self.ramp_down_constraint = Constraint(
+                self.ramp_down_constraint_rpsdtv, rule=operations.ramp_down_constraint
+            )
+            self.ramp_up_constraint = Constraint(
+                self.ramp_up_constraint_rpsdtv, rule=operations.ramp_up_constraint
+            )
 
         self.reserve_margin_rpsd = Set(dimen=4, initialize=reserves.reserve_margin_indices)
         self.validate_reserve_margin = BuildAction(rule=validate_reserve_margin)

temoa/extensions/framework.py

@@ -59,8 +59,14 @@ def get_known_extension_specs() -> dict[str, ExtensionSpec]:
     from temoa.extensions.discrete_capacity.extension import DISCRETE_CAPACITY_EXTENSION
     from temoa.extensions.economies_of_scale.extension import EOS_EXTENSION
     from temoa.extensions.growth_rates.extension import GROWTH_RATES_EXTENSION
-
-    specs = [GROWTH_RATES_EXTENSION, DISCRETE_CAPACITY_EXTENSION, EOS_EXTENSION]
+    from temoa.extensions.unit_commitment.extension import UNIT_COMMITMENT_EXTENSION
+
+    specs = [
+        GROWTH_RATES_EXTENSION,
+        DISCRETE_CAPACITY_EXTENSION,
+        EOS_EXTENSION,
+        UNIT_COMMITMENT_EXTENSION,
+    ]
     return {spec.extension_id: spec for spec in specs}
 
 

temoa/extensions/unit_commitment/__init__.py

@@ -0,0 +1,3 @@
+from temoa.extensions.unit_commitment.extension import UNIT_COMMITMENT_EXTENSION
+
+__all__ = ['UNIT_COMMITMENT_EXTENSION']