[DNMY] Support in the loop PowerFlow evaluation

Project.toml

@@ -19,6 +19,7 @@ JuMP = "4076af6c-e467-56ae-b986-b466b2749572"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
 MathOptInterface = "b8f27783-ece8-5eb3-8dc8-9495eed66fee"
+PowerFlows = "94fada2c-fd9a-4e89-8d82-81405f5cb4f6"
 PowerModels = "c36e90e8-916a-50a6-bd94-075b64ef4655"
 PowerNetworkMatrices = "bed98974-b02a-5e2f-9fe0-a103f5c450dd"
 PowerSystems = "bcd98974-b02a-5e2f-9ee0-a103f5c450dd"
@@ -45,8 +46,9 @@ JuMP = "1"
 LinearAlgebra = "1"
 Logging = "1"
 MathOptInterface = "1"
-PowerModels = "^0.20"
+PowerModels = "^0.21"
 PowerNetworkMatrices = "^0.10"
+PowerFlows = "0.7"
 PowerSystems = "^3"
 PrettyTables = "2"
 ProgressMeter = "^1.5"

src/PowerSimulations.jl

@@ -238,6 +238,10 @@ export LowerBoundFeedForwardSlack
 export TimeDurationOn
 export TimeDurationOff
 export PowerOutput
+export PowerFlowVoltageAngle
+export PowerFlowVoltageMagnitude
+export PowerFlowLineReactivePower
+export PowerFlowLineActivePower
 
 # Constraints
 export AbsoluteValueConstraint
@@ -339,6 +343,7 @@ import LinearAlgebra
 import JSON3
 import PowerSystems
 import InfrastructureSystems
+import PowerFlows
 import PowerNetworkMatrices
 import PowerNetworkMatrices: PTDF, VirtualPTDF
 export PTDF
@@ -399,6 +404,7 @@ const MOI = MathOptInterface
 const MOIU = MathOptInterface.Utilities
 const MOPFM = MOI.FileFormats.Model
 const PNM = PowerNetworkMatrices
+const PFS = PowerFlows
 const TS = TimeSeries
 
 ################################################################################
@@ -444,6 +450,7 @@ include("core/results_by_time.jl")
 
 # Order Required
 include("operation/problem_template.jl")
+include("core/power_flow_data_wrapper.jl")
 include("core/optimization_container.jl")
 include("core/store_common.jl")
 include("initial_conditions/initial_condition_chronologies.jl")
@@ -530,6 +537,7 @@ include("network_models/pm_translator.jl")
 include("network_models/network_slack_variables.jl")
 include("network_models/area_balance_model.jl")
 include("network_models/hvdc_networks.jl")
+include("network_models/power_flow_evaluation.jl")
 
 include("initial_conditions/initialization.jl")
 

src/core/auxiliary_variables.jl

@@ -31,7 +31,29 @@ Auxiliary Variable for Thermal Generation Models that solve for power above min
 """
 struct PowerOutput <: AuxVariableType end
 
+"""
+Auxiliary Variable for the bus angle results from power flow evaluation
+"""
+struct PowerFlowVoltageAngle <: AuxVariableType end
+
+"""
+Auxiliary Variable for the bus voltage magnitued results from power flow evaluation
+"""
+struct PowerFlowVoltageMagnitude <: AuxVariableType end
+
+"""
+Auxiliary Variable for the line reactive flow from power flow evaluation
+"""
+struct PowerFlowLineReactivePower <: AuxVariableType end
+
+"""
+Auxiliary Variable for the line active flow from power flow evaluation
+"""
+struct PowerFlowLineActivePower <: AuxVariableType end
+
 should_write_resulting_value(::Type{<:AuxVariableType}) = true
 
 convert_result_to_natural_units(::Type{<:AuxVariableType}) = false
 convert_result_to_natural_units(::Type{PowerOutput}) = true
+convert_result_to_natural_units(::Type{PowerFlowLineReactivePower}) = true
+convert_result_to_natural_units(::Type{PowerFlowLineActivePower}) = true

src/core/network_model.jl

@@ -34,6 +34,7 @@ mutable struct NetworkModel{T <: PM.AbstractPowerModel}
     duals::Vector{DataType}
     radial_network_reduction::PNM.RadialNetworkReduction
     reduce_radial_branches::Bool
+    power_flow_evaluation::Union{Nothing, PFS.PowerFlowEvaluationModel}
 
     function NetworkModel(
         ::Type{T};
@@ -42,6 +43,7 @@ mutable struct NetworkModel{T <: PM.AbstractPowerModel}
         reduce_radial_branches = false,
         subnetworks = Dict{Int, Set{Int}}(),
         duals = Vector{DataType}(),
+        power_flow_evaluation = nothing,
     ) where {T <: PM.AbstractPowerModel}
         _check_pm_formulation(T)
         new{T}(
@@ -52,6 +54,7 @@ mutable struct NetworkModel{T <: PM.AbstractPowerModel}
             duals,
             PNM.RadialNetworkReduction(),
             reduce_radial_branches,
+            power_flow_evaluation,
         )
     end
 end
@@ -66,6 +69,7 @@ get_reference_buses(m::NetworkModel{T}) where {T <: PM.AbstractPowerModel} =
     collect(keys(m.subnetworks))
 get_subnetworks(m::NetworkModel) = m.subnetworks
 get_bus_area_map(m::NetworkModel) = m.bus_area_map
+get_power_flow_evaluation(m::NetworkModel) = m.power_flow_evaluation
 has_subnetworks(m::NetworkModel) = !isempty(m.bus_area_map)
 
 function add_dual!(model::NetworkModel, dual)

src/core/optimization_container.jl

@@ -110,6 +110,7 @@ mutable struct OptimizationContainer <: AbstractModelContainer
     built_for_recurrent_solves::Bool
     metadata::OptimizationContainerMetadata
     default_time_series_type::Type{<:PSY.TimeSeriesData}
+    power_flow_evaluation_data::Union{<:PowerFlowEvaluationData, Nothing}
 end
 
 function OptimizationContainer(
@@ -154,6 +155,7 @@ function OptimizationContainer(
         false,
         OptimizationContainerMetadata(),
         T,
+        nothing,
     )
 end
 
@@ -191,6 +193,8 @@ get_jump_model(container::OptimizationContainer) = container.JuMPmodel
 get_metadata(container::OptimizationContainer) = container.metadata
 get_optimizer_stats(container::OptimizationContainer) = container.optimizer_stats
 get_parameters(container::OptimizationContainer) = container.parameters
+get_power_flow_evaluation_data(container::OptimizationContainer) =
+    container.power_flow_evaluation_data
 get_resolution(container::OptimizationContainer) = container.resolution
 get_settings(container::OptimizationContainer) = container.settings
 get_time_steps(container::OptimizationContainer) = container.time_steps
@@ -650,8 +654,8 @@ function build_impl!(
     @debug "Total operation count $(PSI.get_jump_model(container).operator_counter)" _group =
         LOG_GROUP_OPTIMIZATION_CONTAINER
 
+    add_power_flow_data!(container, get_power_flow_evaluation(transmission_model), sys)
     check_optimization_container(container)
-
     return
 end
 
@@ -684,8 +688,8 @@ function solve_impl!(container::OptimizationContainer, system::PSY.System)
         return RunStatus.FAILED
     end
 
-    status = RunStatus.SUCCESSFUL
-
+    # Order is important because if a dual is needed then it could move the results to the
+    # temporary primal container
     _, optimizer_stats.timed_calculate_aux_variables =
         @timed calculate_aux_variables!(container, system)
 
@@ -694,7 +698,8 @@ function solve_impl!(container::OptimizationContainer, system::PSY.System)
 
     _, optimizer_stats.timed_calculate_dual_variables =
         @timed calculate_dual_variables!(container, system, is_milp(container))
-    return status
+
+    return RunStatus.SUCCESSFUL
 end
 
 function compute_conflict!(container::OptimizationContainer)
@@ -1503,6 +1508,10 @@ function deserialize_key(container::OptimizationContainer, name::AbstractString)
 end
 
 function calculate_aux_variables!(container::OptimizationContainer, system::PSY.System)
+    if !isnothing(get_power_flow_data(container))
+        solve_power_flow!(container, system)
+    end
+
     aux_vars = get_aux_variables(container)
     for key in keys(aux_vars)
         calculate_aux_variable_value!(container, key, system)

src/core/parameters.jl

@@ -64,6 +64,7 @@ function add_component_name!(attr::TimeSeriesAttributes, name::String, uuid::Str
     return
 end
 
+get_component_names(attr::TimeSeriesAttributes) = keys(attr.component_name_to_ts_uuid)
 _get_ts_uuid(attr::TimeSeriesAttributes, name) = attr.component_name_to_ts_uuid[name]
 
 struct VariableValueAttributes{T <: OptimizationContainerKey} <: ParameterAttributes

src/core/power_flow_data_wrapper.jl

@@ -0,0 +1,14 @@
+mutable struct PowerFlowEvaluationData{T <: PFS.PowerFlowData}
+    power_flow_data::T
+    injection_key_map::Dict{<:OptimizationContainerKey, Dict{String, Int}}
+end
+
+function PowerFlowEvaluationData(power_flow_data::T) where {T <: PFS.PowerFlowData}
+    return PowerFlowEvaluationData{T}(
+        power_flow_data,
+        Dict{OptimizationContainerKey, Dict{String, Int}}(),
+    )
+end
+
+get_power_flow_data(ped::PowerFlowEvaluationData) = ped.power_flow_data
+get_injection_key_map(ped::PowerFlowEvaluationData) = ped.injection_key_map

src/initial_conditions/initialization.jl

@@ -12,6 +12,8 @@ function get_initial_conditions_template(model::OperationModel)
     network_model.radial_network_reduction =
         get_radial_network_reduction(get_network_model(model.template))
     network_model.subnetworks = get_subnetworks(get_network_model(model.template))
+    # Initialization does not support PowerFlow evaluation
+    network_model.power_flow_evaluation = nothing
     bus_area_map = get_bus_area_map(get_network_model(model.template))
     if !isempty(bus_area_map)
         network_model.bus_area_map = get_bus_area_map(get_network_model(model.template))