Include parametric uncertainty

Snakefile

@@ -1,3 +1,6 @@
+from itertools import product
+from numpy import unique
+
 configfile: "config.yaml"
 
 wildcard_constraints:
@@ -49,8 +52,47 @@ rule solve_all_bases:
 # based on the variables of the original problem the search directions
 # of the MGA iterations are inferred.
 
+def get_wildcard_sets(config):
+    wildcard_sets = [
+        {**config['scenario-totals'], **config['alternative-totals']}
+    ]
+    if config['include_groups']:
+        wildcard_sets.append(
+            {**config['scenario-groups'], **config['alternative-groups']}
+        )
+    if config['include_hypercube']:
+        wildcard_sets.append(
+            {**config['scenario-hypercube'], **config['alternative-hypercube']}
+        )
+    return wildcard_sets
+
+
+def add_cost_opts(wildcards_opts):
+    if "cost-uncertainty" not in config.keys():
+        return wildcards_opts
+    factors = config["cost-uncertainty"]
+    carrier, values = zip(*factors.items())
+    cost_sets = [dict(zip(carrier, v)) for v in product(*values)]
+
+    new_opts = []
+    for opts in wildcards_opts:
+        for cost_set in cost_sets:
+            cost_opts = "-".join([f"{c}+{v}" for c, v in cost_set.items()])
+            new_opts.append(f"{opts}-{cost_opts}")
+
+    return new_opts
+
+
+def get_checkpoint_opts(config):
+    wcs_set = get_wildcard_sets(config)
+    opts = []
+    for wcs in wcs_set:
+        opts.extend(add_cost_opts(wcs["opts"]))
+    return unique(opts)
+
+
 checkpoint generate_list_of_alternatives:
-    input: "results/networks/elec_s_{clusters}_ec_lcopt_{opts}.nc"
+    input: expand("results/networks/elec_s_{clusters}_ec_lcopt_{opts}.nc", opts=get_checkpoint_opts(config), allow_missing=True)
     output: "results/alternatives/elec_s_{clusters}_ec_lcopt_{opts}_cat-{category}.txt"
     script: "scripts/generate_list_of_alternatives.py"
 
@@ -67,28 +109,14 @@ rule generate_alternative:
     script: "scripts/generate_alternative.py"
 
 
-def get_wildcard_sets(config):
-    wildcard_sets = [
-        {**config['scenario-totals'], **config['alternative-totals']}
-    ]
-    if config['include_groups']:
-        wildcard_sets.append(
-            {**config['scenario-groups'], **config['alternative-groups']}
-        )
-    if config['include_hypercube']:
-        wildcard_sets.append(
-            {**config['scenario-hypercube'], **config['alternative-hypercube']}
-        )
-    return wildcard_sets
-
-
 def input_generate_clusters_alternatives(w):
     wildcard_sets = get_wildcard_sets(config)
     input = []
     for wildcards in wildcard_sets:
         for clusters in wildcards["clusters"]:
             if int(clusters) == int(w.clusters):
-                for opts in wildcards['opts']:
+                wildcards_opts = add_cost_opts(wildcards["opts"])
+                for opts in wildcards_opts:
                     for epsilon in wildcards['epsilon']:
                         for category in wildcards['category']:
                             alternatives = checkpoints.generate_list_of_alternatives.get(

config.yaml

@@ -1,20 +1,27 @@
 logging_level: INFO
 
 scenario-totals:
-  clusters: [100]
-  opts: [2H]
+  clusters: [37]
+  opts: [4H]
 
 alternative-totals:
   epsilon: [0.005, 0.01, 0.02, 0.03, 0.04, 0.05, 0.06, 0.07, 0.08, 0.09, 0.1]
   category: ['totals']
 
 scenario-groups:
-  clusters: [100]
-  opts: [2H]
+  clusters: []
+  opts: []
 
 alternative-groups:
-  epsilon: [0.05]
-  category: ['groups']
+  epsilon: []
+  category: []
+
+cost-uncertainty:
+  onwind: [0.5, 1.0, 1.5]
+  offwind: [0.5, 1.0, 1.5]
+  solar: [0.5 ,1.0, 1.5]
+  H2: [0.5, 1.0, 1.5]
+  battery: [0.5, 1.0, 1.5]
 
 solving:
   options:

cost-uncertainty.yaml

@@ -0,0 +1,5 @@
+onwind: [0.5,1.0,1.5]
+offwind: [0.5,1.0,1.5]
+solar: [0.5,1.0,1.5]
+H2: [0.5,1.0,1.5]
+battery: [0.5,1.0,1.5]