Add investment method to the graph structure (#788)

TulipaEnergy · Sep 11, 2024 · b2a73e6 · b2a73e6
1 parent 410fc5d
commit b2a73e6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/io.jl b/src/io.jl
@@ -92,6 +92,7 @@ function create_internal_structures(connection)
         row.name => GraphAssetData(
             row.type,
             row.group,
+            row.investment_method,
             _get_stuff("assets_data", "active"; name = row.name),
             _get_stuff("assets_data", "investable"; name = row.name),
             _get_stuff("assets_data", "investment_integer"; name = row.name),

diff --git a/src/structures.jl b/src/structures.jl
@@ -50,6 +50,7 @@ Structure to hold the asset data in the graph.
 mutable struct GraphAssetData
     type::String
     group::Union{Missing,String}
+    investment_method::Union{Missing,String}
     active::Dict{Int,Bool}
     investable::Dict{Int,Bool}
     investment_integer::Dict{Int,Bool}
@@ -98,6 +99,7 @@ mutable struct GraphAssetData
     function GraphAssetData(
         type,
         group,
+        investment_method,
         active,
         investable,
         investment_integer,
@@ -138,6 +140,7 @@ mutable struct GraphAssetData
         return new(
             type,
             group,
+            investment_method,
             active,
             investable,
             investment_integer,