make balance_statespace work for more system types

closes #835
JuliaControl · May 12, 2023 · d3c75c6 · d3c75c6
1 parent ba2713d
commit d3c75c6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/ControlSystemsBase/src/types/conversion.jl b/lib/ControlSystemsBase/src/types/conversion.jl
@@ -187,9 +187,10 @@ end
 #     balance_statespace(A2, B2, C2, perm)
 # end
 
-function balance_statespace(sys::StateSpace, perm::Bool=false)
+function balance_statespace(sys::AbstractStateSpace, perm::Bool=false)
     A, B, C, T = balance_statespace(sys.A,sys.B,sys.C, perm)
-    return ss(A,B,C,sys.D,sys.timeevol), T
+
+    return basetype(sys)(A,B,C,sys.D, ntuple(i->getfield(sys, i+4), fieldcount(T)-4)...), T
 end
 
 # Method that might fail for some exotic types, such as TrackedArrays