Add the min_per_tech module

Author: staadecker

switch_model/policies/min_per_tech.py

@@ -0,0 +1,63 @@
+"""
+This module allows defining a constraint that specifies a minimum buildout for a certain type of gen_tech.
+
+The advantage of this module is that it stills allows switch to decide where to place
+the specified technology.
+"""
+import os
+
+from pyomo.environ import *
+
+from switch_model.reporting import write_table
+
+
+def define_components(mod):
+    mod.GEN_TECH_PER_PERIOD = Set(
+        initialize=lambda m: m.GENERATION_TECHNOLOGIES * m.PERIODS,
+        dimen=2
+    )
+
+    mod.minimum_capacity_mw = Param(
+        mod.GEN_TECH_PER_PERIOD,
+        within=NonNegativeReals,
+        default=0
+    )
+
+    mod.GenCapacityPerTech = Expression(
+        mod.GEN_TECH_PER_PERIOD,
+        rule=lambda m, tech, p: sum(m.GenCapacity[g, p] for g in m.GENS_BY_TECHNOLOGY[tech])
+    )
+
+    mod.Enforce_Minimum_Capacity_Per_Tech = Constraint(
+        mod.GEN_TECH_PER_PERIOD,
+        rule=lambda m, tech, p:
+        Constraint.Skip if m.minimum_capacity_mw[tech, p] == 0 else m.GenCapacityPerTech[tech, p] >=
+                                                                    m.minimum_capacity_mw[tech, p]
+    )
+
+
+def load_inputs(mod, switch_data, inputs_dir):
+    """
+    Expected input file:
+
+     min_per_tech.csv with the following format:
+         gen_tech,period,minimum_capacity_mw
+         Nuclear,2040,10
+    """
+    switch_data.load_aug(
+        filename=os.path.join(inputs_dir, "min_per_tech.csv"),
+        param=mod.minimum_capacity_mw,
+        auto_select=True,
+        # We want this module to run even if we don't specify a constraint so we still get the useful outputs
+        optional=True
+    )
+
+
+def post_solve(mod, outdir):
+    write_table(
+        mod,
+        mod.GEN_TECH_PER_PERIOD,
+        output_file=os.path.join(outdir, "gen_cap_per_tech.csv"),
+        headings=("gen_tech", "period", "gen_capacity", "minimum_capacity_mw"),
+        values=lambda m, tech, p: (tech, p, m.GenCapacityPerTech[tech, p], m.minimum_capacity_mw[tech, p])
+    )

switch_model/wecc/get_inputs.py

@@ -68,6 +68,7 @@ def write_csv(data: Iterable[List], fname, headers: List[str], log=True):
     "switch_model.transmission.transport.dispatch",
     "switch_model.policies.carbon_policies",
     "switch_model.policies.rps_unbundled",
+    "switch_model.policies.min_per_tech",  # Always include since it provides useful outputs even when unused
     # "switch_model.reporting.basic_exports_wecc",
 ]