add wrapper for power flow data

NREL-Sienna · Jan 24, 2024 · d95ef9e · d95ef9e
1 parent 7a2555a
commit d95ef9e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/src/PowerSimulations.jl b/src/PowerSimulations.jl
@@ -450,6 +450,7 @@ include("core/results_by_time.jl")
 
 # Order Required
 include("operation/problem_template.jl")
+include("core/powerflow_data_wrapper.jl")
 include("core/optimization_container.jl")
 include("core/store_common.jl")
 include("initial_conditions/initial_condition_chronologies.jl")

diff --git a/src/core/optimization_container.jl b/src/core/optimization_container.jl
@@ -110,7 +110,7 @@ mutable struct OptimizationContainer <: AbstractModelContainer
     built_for_recurrent_solves::Bool
     metadata::OptimizationContainerMetadata
     default_time_series_type::Type{<:PSY.TimeSeriesData}
-    power_flow_data::Union{PFS.PowerFlowData, Nothing}
+    power_flow_data::Union{<:PowerFlowEvaluationData, Nothing}
 end
 
 function OptimizationContainer(

diff --git a/src/core/powerflow_data_wrapper.jl b/src/core/powerflow_data_wrapper.jl
@@ -0,0 +1,4 @@
+mutable struct PowerFlowEvaluationData{T <: PFS.PowerFlowData}
+    power_flow_data::T
+    injection_key_map::Dict{<:OptimizationContainerKey, Dict{String, Int}}
+end