Enable sensitivity analysis

src/PowerSimulationsDynamics.jl

@@ -184,5 +184,6 @@ include("utils/immutable_dicts.jl")
 include("utils/print.jl")
 include("utils/kwargs_check.jl")
 include("utils/logging.jl")
+include("utils/parameters.jl")
 
 end # module

src/base/branch_wrapper.jl

@@ -11,13 +11,15 @@ struct BranchWrapper
     bus_ix_to::Int
     ix_range::Vector{Int}
     ode_range::Vector{Int}
+    p_range::Vector{Int}
     global_index::Base.ImmutableDict{Symbol, Int}
     function BranchWrapper(
         branch::PSY.DynamicBranch,
         bus_ix_from::Int,
         bus_ix_to::Int,
         ix_range,
         ode_range,
+        p_range,
         sys_base_power,
         sys_base_freq,
     )
@@ -31,6 +33,7 @@ struct BranchWrapper
             bus_ix_to,
             ix_range,
             ode_range,
+            p_range,
             Base.ImmutableDict(Dict(branch_states .=> ix_range)...),
         )
     end
@@ -41,6 +44,7 @@ get_bus_ix_from(wrapper::BranchWrapper) = wrapper.bus_ix_from
 get_bus_ix_to(wrapper::BranchWrapper) = wrapper.bus_ix_to
 get_ix_range(wrapper::BranchWrapper) = wrapper.ix_range
 get_ode_ouput_range(wrapper::BranchWrapper) = wrapper.ode_range
+get_p_range(wrapper::BranchWrapper) = wrapper.p_range
 get_global_index(wrapper::BranchWrapper) = wrapper.global_index
 get_branch(wrapper::BranchWrapper) = wrapper.branch
 

src/base/definitions.jl

@@ -101,6 +101,14 @@ end
 Base.to_index(ix::dq_ref) = Int(ix)
 Base.to_index(ix::RI_ref) = Int(ix)
 
+@enum ref_ix begin
+    Q_ref_ix = 1
+    V_ref_ix = 2
+    ω_ref_ix = 3
+    P_ref_ix = 4
+end
+Base.to_index(ix::ref_ix) = Int(ix)
+
 const MAPPING_DICT = Dict{String, OrderedDict{Symbol, Int}}
 const DEVICE_INTERNAL_MAPPING = Base.ImmutableDict{Int, Vector{Int}}