tp_test.jl

######
#
# These are toy problem formulations used to test advanced features
# such as storage devices
#
######

"opf with storage"
function run_tp_strg_opf(data::Dict{String,Any}, model_constructor, solver; kwargs...)
    return PMs.run_generic_model(data, model_constructor, solver, post_tp_strg_opf; multiconductor=true, kwargs...)
end

""
function post_tp_strg_opf(pm::PMs.GenericPowerModel)
    variable_tp_voltage(pm)
    variable_tp_branch_flow(pm)
    variable_tp_storage(pm)

    for c in PMs.conductor_ids(pm)
        PMs.variable_generation(pm, cnd=c)
        PMs.variable_dcline_flow(pm, cnd=c)
    end

    constraint_tp_voltage(pm)

    for i in PMs.ids(pm, :ref_buses)
        constraint_tp_theta_ref(pm, i)
    end

    for i in PMs.ids(pm, :bus), c in PMs.conductor_ids(pm)
        PMs.constraint_kcl_shunt_storage(pm, i, cnd=c)
    end

    for i in PMs.ids(pm, :storage)
        PMs.constraint_storage_state(pm, i)
        constraint_tp_storage_exchange(pm, i)
        for c in PMs.conductor_ids(pm)
            PMs.constraint_storage_thermal_limit(pm, i, cnd=c)
        end
    end

    for i in PMs.ids(pm, :branch)
        for c in PMs.conductor_ids(pm)
            constraint_ohms_tp_yt_from(pm, i, cnd=c)
            constraint_ohms_tp_yt_to(pm, i, cnd=c)

            PMs.constraint_voltage_angle_difference(pm, i, cnd=c)

            PMs.constraint_thermal_limit_from(pm, i, cnd=c)
            PMs.constraint_thermal_limit_to(pm, i, cnd=c)
        end
    end

    for i in PMs.ids(pm, :dcline), c in PMs.conductor_ids(pm)
        PMs.constraint_dcline(pm, i, cnd=c)
    end

    PMs.objective_min_fuel_cost(pm)
end




"multi-network opf with storage"
function run_mn_tp_strg_opf(data::Dict{String,Any}, model_constructor, solver; kwargs...)
    return PMs.run_generic_model(data, model_constructor, solver, post_mn_tp_strg_opf; multiconductor=true, multinetwork=true, kwargs...)
end

""
function post_mn_tp_strg_opf(pm::PMs.GenericPowerModel)
    for (n, network) in PMs.nws(pm)
        variable_tp_voltage(pm, nw=n)
        variable_tp_branch_flow(pm, nw=n)
        variable_tp_storage(pm, nw=n)

        for c in PMs.conductor_ids(pm, nw=n)
            PMs.variable_generation(pm, cnd=c, nw=n)
            PMs.variable_dcline_flow(pm, cnd=c, nw=n)
        end

        constraint_tp_voltage(pm, nw=n)

        for i in PMs.ids(pm, :ref_buses, nw=n)
            constraint_tp_theta_ref(pm, i, nw=n)
        end

        for i in PMs.ids(pm, :bus, nw=n), c in PMs.conductor_ids(pm, nw=n)
            PMs.constraint_kcl_shunt_storage(pm, i, cnd=c, nw=n)
        end

        for i in PMs.ids(pm, :storage, nw=n)
            constraint_tp_storage_exchange(pm, i, nw=n)
            for c in PMs.conductor_ids(pm, nw=n)
                PMs.constraint_storage_thermal_limit(pm, i, cnd=c, nw=n)
            end
        end

        for i in PMs.ids(pm, :branch, nw=n)
            for c in PMs.conductor_ids(pm, nw=n)
                constraint_ohms_tp_yt_from(pm, i, cnd=c, nw=n)
                constraint_ohms_tp_yt_to(pm, i, cnd=c, nw=n)

                PMs.constraint_voltage_angle_difference(pm, i, cnd=c, nw=n)

                PMs.constraint_thermal_limit_from(pm, i, cnd=c, nw=n)
                PMs.constraint_thermal_limit_to(pm, i, cnd=c, nw=n)
            end
        end

        for i in PMs.ids(pm, :dcline, nw=n), c in PMs.conductor_ids(pm, nw=n)
            PMs.constraint_dcline(pm, i, cnd=c, nw=n)
        end
    end

    network_ids = sort(collect(PMs.nw_ids(pm)))

    n_1 = network_ids[1]
    for i in PMs.ids(pm, :storage, nw=n_1)
        PMs.constraint_storage_state(pm, i, nw=n_1)
    end

    for n_2 in network_ids[2:end]
        for i in PMs.ids(pm, :storage, nw=n_2)
            PMs.constraint_storage_state(pm, i, n_1, n_2)
        end
        n_1 = n_2
    end

    PMs.objective_min_fuel_cost(pm)
end