Jd/implement state estimator

Project.toml

@@ -11,16 +11,17 @@ InfrastructureSystems = "2cd47ed4-ca9b-11e9-27f2-ab636a7671f1"
 JuMP = "4076af6c-e467-56ae-b986-b466b2749572"
 MPI = "da04e1cc-30fd-572f-bb4f-1f8673147195"
 MathOptInterface = "b8f27783-ece8-5eb3-8dc8-9495eed66fee"
+PowerNetworkMatrices = "bed98974-b02a-5e2f-9fe0-a103f5c450dd"
 PowerSimulations = "e690365d-45e2-57bb-ac84-44ba829e73c4"
 PowerSystems = "bcd98974-b02a-5e2f-9ee0-a103f5c450dd"
 
 [compat]
-Dates = "1"
 DataStructures = "^0.18"
+Dates = "1"
 DocStringExtensions = "~0.8, ~0.9"
-InfrastructureSystems = "^1.21"
+InfrastructureSystems = "2"
 JuMP = "1"
 MPI = "^0.20"
 MathOptInterface = "1"
-PowerSystems = "^3"
+PowerSystems = "4"
 julia = "^1.6"

src/PowerSimulationsDecomposition.jl

@@ -3,7 +3,10 @@ module PowerSimulationsDecomposition
 export MultiRegionProblem
 export MultiProblemTemplate
 
+export SplitAreaPTDFPowerModel
+
 import PowerSimulations
+import PowerNetworkMatrices
 import PowerSystems
 import InfrastructureSystems
 import InfrastructureSystems: @assert_op
@@ -15,7 +18,10 @@ import DataStructures: OrderedDict, SortedDict
 
 const PSI = PowerSimulations
 const PSY = PowerSystems
+const PNM = PowerNetworkMatrices
 const IS = InfrastructureSystems
+const ISOPT = InfrastructureSystems.Optimization
+const ISSIM = InfrastructureSystems.Simulation
 const PM = PSI.PM
 const MOI = MathOptInterface
 
@@ -25,13 +31,19 @@ using DocStringExtensions
                                  $(DOCSTRING)
                                  """
 
-include("definitions.jl")
-include("core.jl")
+include("core/algorithms.jl")
+include("core/definitions.jl")
+include("core/formulations.jl")
+include("core/mpi_info.jl")
+include("core/parameters.jl")
+include("core/auxiliary_variables.jl")
+
 include("multiproblem_template.jl")
 include("multi_optimization_container.jl")
 include("algorithms/sequential_algorithm.jl")
 include("algorithms/mpi_parallel_algorithm.jl")
 include("problems/multi_region_problem.jl")
+include("models/network_models.jl")
 include("print.jl")
 
 end

src/algorithms/sequential_algorithm.jl

@@ -19,7 +19,8 @@ function build_main_problem!(
     container::MultiOptimizationContainer{SequentialAlgorithm},
     template::MultiProblemTemplate,
     sys::PSY.System,
-) end
+)
+end
 
 # The drawback of this approach is that it will loop over the results twice
 # once to write into the main container and a second time when writing into the
@@ -29,23 +30,26 @@ function write_results_to_main_container(container::MultiOptimizationContainer)
     # TODO: This process needs to work in parallel almost right away
     # TODO: This doesn't handle the case where subproblems have an overlap in axis names.
 
-    for subproblem in values(container.subproblems)
+    for (k, subproblem) in container.subproblems
         for field in CONTAINER_FIELDS
             subproblem_data_field = getproperty(subproblem, field)
             main_container_data_field = getproperty(container, field)
             for (key, src) in subproblem_data_field
                 if src isa JuMP.Containers.SparseAxisArray
-                    # @warn "Skip SparseAxisArray" field key
+                    @debug "Skip SparseAxisArray" field key
                     continue
                 end
                 num_dims = ndims(src)
                 num_dims > 2 && error("ndims = $(num_dims) is not supported yet")
                 data = nothing
                 try
+                    if key == ISOPT.ExpressionKey{PSI.ActivePowerBalance, PSY.ACBus}("")
+                        @error "mt exp $k" data = PSI.jump_value.(src)[10313, :]
+                    end
                     data = PSI.jump_value.(src)
                 catch e
                     if e isa UndefRefError
-                        #@warn "Skip UndefRefError for" field key
+                        @error "Skip UndefRefError for" field key
                         continue
                     end
                     rethrow()
@@ -103,10 +107,13 @@ function solve_impl!(
     sys::PSY.System,
 )
     # Solve main problem
-    status = PSI.RunStatus.SUCCESSFUL
+    status = ISSIM.RunStatus.RUNNING
     for (index, subproblem) in container.subproblems
-        @info "Solving problem $index"
+        @debug "Solving problem $index"
         status = PSI.solve_impl!(subproblem, sys)
+        if status != ISSIM.RunStatus.SUCCESSFULLY_FINALIZED
+            return status
+        end
     end
     write_results_to_main_container(container)
     return status

src/core/algorithms.jl

@@ -0,0 +1,4 @@
+abstract type DecompositionAlgorithm end
+
+struct SequentialAlgorithm <: DecompositionAlgorithm end
+struct MPIParallelAlgorithm <: DecompositionAlgorithm end

src/core/auxiliary_variables.jl

@@ -0,0 +1 @@
+

src/core/formulations.jl

@@ -0,0 +1,3 @@
+# Formulations
+
+struct SplitAreaPTDFPowerModel <: PSI.AbstractPTDFModel end

src/core.jl → src/core/mpi_info.jl

@@ -1,8 +1,3 @@
-abstract type DecompositionAlgorithm end
-
-struct SequentialAlgorithm <: DecompositionAlgorithm end
-struct MPIParallelAlgorithm <: DecompositionAlgorithm end
-
 # Taken from https://github.com/ANL-CEEESA/UnitCommitment.jl/blob/dev/src/solution/methods/ProgressiveHedging/structs.jl
 struct MpiInfo
     comm::Any

src/core/parameters.jl

@@ -0,0 +1 @@
+struct StateEstimationInjections <: PSI.VariableValueParameter end