forward balreal and balance_statespace on ess

Author: baggepinnen

src/ExtendedStateSpace.jl

@@ -370,6 +370,18 @@ macro sizecompat(a,b)
     end
 end
 
+function ControlSystems.balance_statespace(G::ExtendedStateSpace, perm::Bool=false)
+    Gs = ss(G)
+    Gs, T = ControlSystems.balance_statespace(Gs, perm)
+    partition(Gs, G.nw, G.nz), T
+end
+
+function ControlSystems.balreal(G::ExtendedStateSpace, args...; kwargs...)
+    Gs = ss(G)
+    Gs, S, T = ControlSystems.balreal(Gs, args...; kwargs...)
+    partition(Gs, G.nw, G.nz), S, T
+end
+
 # This version is not correct for the intended usage
 # function ControlSystems.feedback(s1::ExtendedStateSpace, s2::ExtendedStateSpace;
 #     Wperm=:, Zperm=:)

test/test_extendedstatespace.jl

@@ -54,3 +54,10 @@ connections = [K.y .=> G.u; G.y .=> G.y; K.u .=> K.u]
 Gcl2 = connect([G, K, S], connections; z1, w1)
 
 @test linfnorm(minreal(Gcl1 - Gcl2.sys))[1] < 1e-10
+
+
+##
+G = ssrand(3,4,5)
+Ge = partition(G, 1, 1)
+@test ss(balance_statespace(Ge)[1]) == balance_statespace(G)[1]
+@test ss(balreal(Ge)[1]) == balreal(G)[1]