Capacity expansion model

Add the model file with capacity expansion constraints and subsequent modifications

relog-web/src/casebuilder/InputPage.js

@@ -207,6 +207,8 @@ const InputPage = () => {
         }
         if (outputFound) {
           delete plant["outputs (tonne/tonne)"][productName];
+          delete plant["disposal cost ($/tonne)"][productName];
+          delete plant["disposal limit (tonne)"][productName];
         }
       }
       save(newData);

src/instance/compress.jl

@@ -24,6 +24,9 @@ function _compress(instance::Instance)::Instance
 
     # Compress products
     for p in compressed.products
+        p.acquisition_cost = [mean(p.acquisition_cost)]
+        p.disposal_cost = [mean(p.disposal_cost)]
+        p.disposal_limit = [sum(p.disposal_limit)]
         p.transportation_cost = [mean(p.transportation_cost)]
         p.transportation_energy = [mean(p.transportation_energy)]
         for (emission_name, emission_value) in p.transportation_emissions

src/instance/parse.jl

@@ -171,7 +171,7 @@ function parse(json)::Instance
                     ),
                 )
             end
-            length(sizes) > 1 || push!(sizes, sizes[1])
+            length(sizes) > 1 || push!(sizes, deepcopy(sizes[1]))
             sort!(sizes, by = x -> x.capacity)
 
             # Initial capacity

src/model/resolve.jl

@@ -17,6 +17,24 @@ function resolve(model_old, instance::Instance; optimizer = nothing)::OrderedDic
         lp_optimizer = _get_default_lp_optimizer()
     end
 
+    @info "Filtering candidate locations..."
+    selected_pairs = Set()
+    for ((node_old, t), var_old) in model_old[:is_open]
+        if JuMP.value(var_old) > 0.1
+            push!(
+                selected_pairs,
+                (node_old.location.plant_name, node_old.location.location_name),
+            )
+        end
+    end
+    filtered_plants = []
+    for p in instance.plants
+        if (p.plant_name, p.location_name) in selected_pairs
+            push!(filtered_plants, p)
+        end
+    end
+    instance.plants = filtered_plants
+
     @info "Building new graph..."
     graph = build_graph(instance)
     _print_graph_stats(instance, graph)
@@ -48,10 +66,9 @@ function _fix_plants!(model_old, model_new)::Nothing
     # Fix open_plant variables
     for ((node_old, t), var_old) in model_old[:open_plant]
         value_old = JuMP.value(var_old)
-        node_new = model_new[:graph].name_to_process_node_map[(
-            node_old.location.plant_name,
-            node_old.location.location_name,
-        )]
+        key = (node_old.location.plant_name, node_old.location.location_name)
+        key ∈ keys(model_new[:graph].name_to_process_node_map) || continue
+        node_new = model_new[:graph].name_to_process_node_map[key]
         var_new = model_new[:open_plant][node_new, t]
         JuMP.unset_binary(var_new)
         JuMP.fix(var_new, value_old)
@@ -61,10 +78,9 @@ function _fix_plants!(model_old, model_new)::Nothing
     for ((node_old, t), var_old) in model_old[:is_open]
         t > 0 || continue
         value_old = JuMP.value(var_old)
-        node_new = model_new[:graph].name_to_process_node_map[(
-            node_old.location.plant_name,
-            node_old.location.location_name,
-        )]
+        key = (node_old.location.plant_name, node_old.location.location_name)
+        key ∈ keys(model_new[:graph].name_to_process_node_map) || continue
+        node_new = model_new[:graph].name_to_process_node_map[key]
         var_new = model_new[:is_open][node_new, t]
         JuMP.unset_binary(var_new)
         JuMP.fix(var_new, value_old)
@@ -73,10 +89,9 @@ function _fix_plants!(model_old, model_new)::Nothing
     # Fix plant capacities
     for ((node_old, t), var_old) in model_old[:capacity]
         value_old = JuMP.value(var_old)
-        node_new = model_new[:graph].name_to_process_node_map[(
-            node_old.location.plant_name,
-            node_old.location.location_name,
-        )]
+        key = (node_old.location.plant_name, node_old.location.location_name)
+        key ∈ keys(model_new[:graph].name_to_process_node_map) || continue
+        node_new = model_new[:graph].name_to_process_node_map[key]
         var_new = model_new[:capacity][node_new, t]
         JuMP.delete_lower_bound(var_new)
         JuMP.delete_upper_bound(var_new)
@@ -87,10 +102,9 @@ function _fix_plants!(model_old, model_new)::Nothing
     for ((node_old, t), var_old) in model_old[:expansion]
         t > 0 || continue
         value_old = JuMP.value(var_old)
-        node_new = model_new[:graph].name_to_process_node_map[(
-            node_old.location.plant_name,
-            node_old.location.location_name,
-        )]
+        key = (node_old.location.plant_name, node_old.location.location_name)
+        key ∈ keys(model_new[:graph].name_to_process_node_map) || continue
+        node_new = model_new[:graph].name_to_process_node_map[key]
         var_new = model_new[:expansion][node_new, t]
         JuMP.delete_lower_bound(var_new)
         JuMP.delete_upper_bound(var_new)

src/model/solve.jl

@@ -32,6 +32,7 @@ function solve(
     output = nothing,
     marginal_costs = true,
     return_model = false,
+    graph = nothing,
 )
 
     if lp_optimizer == nothing
@@ -51,7 +52,9 @@ function solve(
 
 
     @info "Building graph..."
-    graph = RELOG.build_graph(instance)
+    if graph === nothing
+        graph = RELOG.build_graph(instance)
+    end
     _print_graph_stats(instance, graph)
 
     @info "Building optimization model..."

src/reports/products.jl

@@ -5,7 +5,7 @@
 using DataFrames
 using CSV
 
-function products_report(solution; marginal_costs = true)::DataFrame
+function products_report(solution; marginal_costs)::DataFrame
     df = DataFrame()
     df."product name" = String[]
     df."location name" = String[]
@@ -21,7 +21,11 @@ function products_report(solution; marginal_costs = true)::DataFrame
     for (prod_name, prod_dict) in solution["Products"]
         for (location_name, location_dict) in prod_dict
             for year = 1:T
-                marginal_cost = location_dict["Marginal cost (\$/tonne)"][year]
+                if marginal_costs
+                    marginal_cost = location_dict["Marginal cost (\$/tonne)"][year]
+                else
+                    marginal_cost = 0.0
+                end
                 latitude = round(location_dict["Latitude (deg)"], digits = 6)
                 longitude = round(location_dict["Longitude (deg)"], digits = 6)
                 amount = location_dict["Amount (tonne)"][year]
@@ -49,4 +53,5 @@ function products_report(solution; marginal_costs = true)::DataFrame
     return df
 end
 
-write_products_report(solution, filename) = CSV.write(filename, products_report(solution))
+write_products_report(solution, filename; marginal_costs = true) =
+    CSV.write(filename, products_report(solution; marginal_costs))

src/reports/write.jl

@@ -13,8 +13,12 @@ function write(solution::AbstractDict, filename::AbstractString)
     end
 end
 
-function write_reports(solution::AbstractDict, basename::AbstractString)
-    RELOG.write_products_report(solution, "$(basename)_products.csv")
+function write_reports(
+    solution::AbstractDict,
+    basename::AbstractString;
+    marginal_costs = true,
+)
+    RELOG.write_products_report(solution, "$(basename)_products.csv"; marginal_costs)
     RELOG.write_plants_report(solution, "$(basename)_plants.csv")
     RELOG.write_plant_outputs_report(solution, "$(basename)_plant_outputs.csv")
     RELOG.write_plant_emissions_report(solution, "$(basename)_plant_emissions.csv")

test/src/instance/parse_test.jl

@@ -72,7 +72,10 @@ function instance_parse_test()
         @test plant.sizes[1].opening_cost == [3000, 3000]
         @test plant.sizes[1].fixed_operating_cost == [50, 50]
         @test plant.sizes[1].variable_operating_cost == [50, 50]
-        @test plant.sizes[1] == plant.sizes[2]
+        @test plant.sizes[2].capacity == 1000.0
+        @test plant.sizes[2].opening_cost == [3000, 3000]
+        @test plant.sizes[2].fixed_operating_cost == [50, 50]
+        @test plant.sizes[2].variable_operating_cost == [50, 50]
 
         p4 = product_name_to_product["P4"]
         @test plant.output[p3] == 0.05

test/src/model/resolve_test.jl

@@ -4,10 +4,18 @@
 using RELOG
 
 function model_resolve_test()
-    @testset "Resolve" begin
+    @testset "Resolve (exact)" begin
         # Shoud not crash
         filename = fixture("s1.json")
         solution_old, model_old = RELOG.solve(filename, return_model = true)
         solution_new = RELOG.resolve(model_old, filename)
     end
+
+    @testset "Resolve (heuristic)" begin
+        # Shoud not crash
+        filename = fixture("s1.json")
+        solution_old, model_old =
+            RELOG.solve(filename, return_model = true, heuristic = true)
+        solution_new = RELOG.resolve(model_old, filename)
+    end
 end