changed: rename to upper case (for consistency)

franckgaga · franckgaga · commit 95208ad2d0c3 · 2026-03-09T14:03:31.000-04:00
diff --git a/src/controller/construct.jl b/src/controller/construct.jl
@@ -176,7 +176,7 @@ struct ControllerConstraint{NT<:Real, GCfunc<:Union{Nothing, Function}}
     # indices of finite numbers in the b vector (linear inequality constraints):
     i_b     ::BitVector
     # A matrices for the linear equality constraints:
-    A_ŝ     ::Matrix{NT}
+    A_Ŝ     ::Matrix{NT}
     Aeq     ::Matrix{NT}
     # b vector for the linear equality constraints:
     beq     ::Vector{NT}
@@ -518,7 +518,7 @@ function setconstraint!(
             con.A_Umin, con.A_Umax, con.A_ΔŨmin, con.A_ΔŨmax, 
             con.A_Ymin, con.A_Ymax, con.A_Wmin, con.A_Wmax,
             con.A_x̂min, con.A_x̂max,
-            con.A_ŝ
+            con.A_Ŝ
         )
         A = con.A[con.i_b, :]
         b = con.b[con.i_b]
@@ -910,7 +910,7 @@ function init_defaultcon_mpc(
     A_Ymin,  A_Ymax, Ẽ  = relaxŶ(E, C_ymin, C_ymax, nϵ)
     A_Wmin,  A_Wmax, Ẽw = relaxW(E, Pu, Hp, W̄y, W̄u, C_wmin, C_wmax, nϵ)
     A_x̂min,  A_x̂max, ẽx̂ = relaxterminal(ex̂, c_x̂min, c_x̂max, nϵ)
-    A_ŝ, Ẽŝ = augmentdefect(Eŝ, nϵ)
+    A_Ŝ, Ẽŝ = augmentdefect(Eŝ, nϵ)
     i_Umin,  i_Umax  = .!isinf.(U0min), .!isinf.(U0max)
     i_ΔŨmin, i_ΔŨmax = .!isinf.(ΔŨmin), .!isinf.(ΔŨmax)
     i_Ymin,  i_Ymax  = .!isinf.(Y0min), .!isinf.(Y0max)
@@ -920,7 +920,7 @@ function init_defaultcon_mpc(
         model, transcription, nc,
         i_Umin, i_Umax, i_ΔŨmin, i_ΔŨmax, i_Ymin, i_Ymax, i_Wmin, i_Wmax, i_x̂min, i_x̂max,
         A_Umin, A_Umax, A_ΔŨmin, A_ΔŨmax, A_Ymin, A_Ymax, A_Wmin, A_Wmax, A_x̂max, A_x̂min,
-        A_ŝ
+        A_Ŝ
     )
     # dummy fx̂, Fw and Fŝ vectors (updated just before optimization)
     fx̂, Fw, Fŝ = zeros(NT, nx̂), zeros(NT, nW), zeros(NT, nx̂*Hp)
@@ -935,7 +935,7 @@ function init_defaultcon_mpc(
         A_Umin  , A_Umax , A_ΔŨmin , A_ΔŨmax  , 
         A_Ymin  , A_Ymax , A_Wmin  , A_Wmax   , A_x̂min , A_x̂max , 
         A       , b      , i_b     , 
-        A_ŝ     ,
+        A_Ŝ     ,
         Aeq     , beq    ,
         neq     ,
         C_ymin  , C_ymax , C_wmin  , C_wmax   , c_x̂min , c_x̂max , 
@@ -1194,14 +1194,14 @@ function relaxterminal(ex̂::AbstractMatrix{NT}, c_x̂min, c_x̂max, nϵ) where
 end
 
 @doc raw"""
-    augmentdefect(Eŝ, nϵ) -> A_ŝ, Ẽŝ
+    augmentdefect(Eŝ, nϵ) -> A_Ŝ, Ẽŝ
 
 Augment defect equality constraints with slack variable ϵ if `nϵ == 1`.
 
 It returns the ``\mathbf{Ẽ_ŝ}`` matrix that appears in the defect equation 
 ``\mathbf{Ŝ = Ẽ_ŝ Z̃ + F_ŝ}`` and the ``\mathbf{A}`` matrix for the equality constraints:
 ```math
-\mathbf{A_ŝ Z̃} = - \mathbf{F_ŝ}
+\mathbf{A_Ŝ Z̃} = - \mathbf{F_ŝ}
 ```
 """
 function augmentdefect(Eŝ::AbstractMatrix{NT}, nϵ) where NT<:Real
@@ -1210,8 +1210,8 @@ function augmentdefect(Eŝ::AbstractMatrix{NT}, nϵ) where NT<:Real
     else # Z̃ = Z (only hard constraints)
         Ẽŝ = Eŝ
     end
-    A_ŝ = Ẽŝ
-    return A_ŝ, Ẽŝ
+    A_Ŝ = Ẽŝ
+    return A_Ŝ, Ẽŝ
 end
 
 @doc raw"""
diff --git a/src/controller/execute.jl b/src/controller/execute.jl
@@ -691,7 +691,7 @@ function setmodel_controller!(mpc::PredictiveController, uop_old, x̂op_old)
     con.bx̂ .= bx̂
     # --- defect matrices ---
     Eŝ, Gŝ, Jŝ, Kŝ, Vŝ, Bŝ = init_defectmat(model, estim, transcription, Hp, Hc, nb)
-    A_ŝ, Ẽŝ = augmentdefect(Eŝ, mpc.nϵ)
+    A_Ŝ, Ẽŝ = augmentdefect(Eŝ, mpc.nϵ)
     con.Ẽŝ .= Ẽŝ
     con.Gŝ .= Gŝ
     con.Jŝ .= Jŝ
@@ -718,8 +718,8 @@ function setmodel_controller!(mpc::PredictiveController, uop_old, x̂op_old)
         con.A_x̂max
     ]
     # --- linear equality constraints ---
-    con.A_ŝ .= A_ŝ
-    con.Aeq .= A_ŝ
+    con.A_Ŝ .= A_Ŝ
+    con.Aeq .= A_Ŝ
     # --- operating points ---
     con.U0min .+= mpc.Uop # convert U0 to U with the old operating point
     con.U0max .+= mpc.Uop # convert U0 to U with the old operating point
diff --git a/src/controller/transcription.jl b/src/controller/transcription.jl
@@ -709,19 +709,27 @@ end
 
 function init_defectmat(
     model::NonLinModel, estim::StateEstimator{NT}, transcription::CollocationMethod, Hp, Hc, _
-)
-    
+) where {NT<:Real}
+    nx̂, nu, nd = estim.nx̂, model.nu, model.nd
+    nZ = get_nZ(estim, transcription, Hp, Hc)
+    Eŝ = zeros(NT, 0, nZ)
+    Gŝ = zeros(NT, 0, nd)
+    Jŝ = zeros(NT, 0, nd*Hp)
+    Kŝ = zeros(NT, 0, nx̂)
+    Vŝ = zeros(NT, 0, nu)
+    Bŝ = zeros(NT, 0)
+    return Eŝ, Gŝ, Jŝ, Kŝ, Vŝ, Bŝ
 end
 
 """
     init_defectmat(
-        model::NonLinModel, estim::IntenalModel{NT}, transcription::CollocationMethod, Hp, Hc, _
+        model::NonLinModel, estim::InternalModel{NT}, transcription::CollocationMethod, Hp, Hc, _
     ) -> Eŝ, Gŝ, Jŝ, Kŝ, Vŝ, Bŝ
 
 Return empty matrices for [`InternalModel`](@ref) (state vector is not augmented).
 """
 function init_defectmat(
-    model::NonLinModel, estim::IntenalModel{NT}, transcription::CollocationMethod, Hp, Hc, _
+    model::NonLinModel, estim::InternalModel{NT}, transcription::CollocationMethod, Hp, Hc, _
 ) where {NT<:Real}
     nx̂, nu, nd = estim.nx̂, model.nu, model.nd
     nZ = get_nZ(estim, transcription, Hp, Hc)
@@ -778,7 +786,7 @@ The argument `nc` is the number of custom nonlinear inequality constraints in
 finite numbers. `i_g` is a similar vector but for the indices of ``\mathbf{g}``. The method
 also returns the ``\mathbf{A, A_{eq}}`` matrices and `neq` if `args` is provided. In such a 
 case, `args`  needs to contain all the inequality and equality constraint matrices: 
-`A_Umin, A_Umax, A_ΔŨmin, A_ΔŨmax, A_Ymin, A_Ymax, A_Wmin, A_Wmax, A_x̂min, A_x̂max, A_ŝ`. 
+`A_Umin, A_Umax, A_ΔŨmin, A_ΔŨmax, A_Ymin, A_Ymax, A_Wmin, A_Wmax, A_x̂min, A_x̂max, A_Ŝ`. 
 The integer `neq` is the number of nonlinear equality constraints in ``\mathbf{g_{eq}}``.
 """
 function init_matconstraint_mpc(
@@ -795,7 +803,7 @@ function init_matconstraint_mpc(
             A_Ymin,  A_Ymax, 
             A_Wmin,  A_Wmax,
             A_x̂min,  A_x̂max,  
-            A_ŝ
+            A_Ŝ
         ) = args
         A = [
             A_Umin;  A_Umax; 
@@ -804,7 +812,7 @@ function init_matconstraint_mpc(
             A_Wmin;  A_Wmax
             A_x̂min;  A_x̂max;
         ]
-        Aeq = A_ŝ
+        Aeq = A_Ŝ
         neq = 0
     end
     i_b = [i_Umin; i_Umax; i_ΔŨmin; i_ΔŨmax; i_Ymin; i_Ymax; i_Wmin; i_Wmax; i_x̂min; i_x̂max]
@@ -821,9 +829,9 @@ function init_matconstraint_mpc(
     if isempty(args)
         A, Aeq, neq = nothing, nothing, nothing
     else
-        A_Umin, A_Umax, A_ΔŨmin, A_ΔŨmax, _ , _ , A_Wmin, A_Wmax, _ , _ , A_ŝ = args
+        A_Umin, A_Umax, A_ΔŨmin, A_ΔŨmax, _ , _ , A_Wmin, A_Wmax, _ , _ , A_Ŝ = args
         A   = [A_Umin; A_Umax; A_ΔŨmin; A_ΔŨmax; A_Wmin; A_Wmax]
-        Aeq = A_ŝ
+        Aeq = A_Ŝ
         neq = 0
     end
     i_b = [i_Umin; i_Umax; i_ΔŨmin; i_ΔŨmax; i_Wmin; i_Wmax]
@@ -840,9 +848,9 @@ function init_matconstraint_mpc(
     if isempty(args)
         A, Aeq, neq = nothing, nothing, nothing
     else    
-        A_Umin, A_Umax, A_ΔŨmin, A_ΔŨmax, _ , _ , A_Wmin, A_Wmax, A_x̂min, A_x̂max, A_ŝ = args
+        A_Umin, A_Umax, A_ΔŨmin, A_ΔŨmax, _ , _ , A_Wmin, A_Wmax, A_x̂min, A_x̂max, A_Ŝ = args
         A   = [A_Umin; A_Umax; A_ΔŨmin; A_ΔŨmax; A_Wmin; A_Wmax; A_x̂min; A_x̂max]
-        Aeq = A_ŝ
+        Aeq = A_Ŝ
         nΔŨ, nZ̃ = size(A_ΔŨmin)
         neq = nZ̃ - nΔŨ
     end
