add inhomogenous model

Author: hexaeder

benchmark/benchmark_cases.jl

@@ -77,5 +77,29 @@ BENCHMARK_CASES = [
         # [10,20],
         [100, 1_000, 10_000, 100_000],
         "Power grid network with PQ nodes and generators"
+    ),
+
+    BenchmarkCase(
+        "powergrid_inhomogeneous_pq",
+        N -> begin
+            g = watts_strogatz(N, 3, 0.8; rng=StableRNG(1))
+            rng = StableRNG(1)
+
+            # Create fixed P/Q values for half the nodes
+            pq_count = N ÷ 2
+            P_vals = -1*rand(rng, pq_count)
+            Q_vals = -1*rand(rng, pq_count)
+
+            # Create vertex models with baked-in P/Q values
+            pq_vertices = [pqnode_inhomogeneous(P_vals[i], Q_vals[i]) for i in 1:pq_count]
+            gen_vertices = [generator() for _ in 1:(N-pq_count)]
+
+            # Shuffle the vertex types
+            vertices = shuffle(rng, vcat(pq_vertices, gen_vertices))
+
+            Network(g, vertices, piline())
+        end,
+        [100, 1_000, 10_000, 100_000],
+        "Power grid with heterogeneous PQ nodes having fixed P/Q values"
     )
 ]

benchmark/benchmark_models.jl

@@ -109,3 +109,15 @@ Base.@propagate_inbounds function gen!(dv, v, isum, p, T)
 end
 generator() = VertexModel(; f=gen!, sym=[:u_r, :u_i, :θ, :ω], g=StateMask(1:2),
             psym=[:H, :P, :D, :Ω, :E_f, :T_d_dash, :T_q_dash, :X_d_dash, :X_q_dash, :X_d, :X_q])
+
+function pqnode_inhomogeneous(P, Q)
+    Base.@propagate_inbounds function pq!(du, u, isum, _, t)
+        ic = Complex(isum[1], isum[2])
+        S = Complex(P, Q)
+        uc = S/conj(ic)
+        du[1] = real(uc)
+        du[2] = imag(uc)
+        nothing
+    end
+    VertexModel(; f=pq!, sym=[:u_r, :u_i], g=StateMask(1:2), psym=[], mass_matrix=0)
+end

benchmark/benchmark_utils.jl

@@ -244,10 +244,11 @@ function plot_over_N(target, baseline=nothing)
         comp = target
     end
 
-    fig = Makie.Figure(size=(2000,2000))
+    fig = Makie.Figure(size=(2000,2500))
 
     bmkeys = [
         "powergrid",
+        "powergrid_inhomogeneous_pq",
         "diffusion_static_edge",
         "diffusion_ode_edge",
         "kuramoto_homogeneous",