Merge branch 'new-design-stewart' of https://github.com/nurkanovic/no…

…snoc into new-design-stewart
nurkanovic · May 23, 2024 · babe151 · babe151
2 parents d64cbe4 + f26b1cd
commit babe151
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/+nosnoc/+discrete_time_problem/Stewart.m b/+nosnoc/+discrete_time_problem/Stewart.m
@@ -103,13 +103,13 @@ function create_variables(obj)
 
                 % Handle x_box settings
                 if ~opts.x_box_at_stg && opts.rk_representation ~= RKRepresentation.differential
-                    obj.w.x(1:opts.N_stages,1:opts.N_finite_elements(ii),1:(opts.n_s+rbp-1)).lb = -inf*ones(dims.n_x, 1);
-                    obj.w.x(1:opts.N_stages,1:opts.N_finite_elements(ii),1:(opts.n_s+rbp-1)).ub = inf*ones(dims.n_x, 1);
+                    obj.w.x(ii,1:opts.N_finite_elements(ii),1:(opts.n_s+rbp-1)).lb = -inf*ones(dims.n_x, 1);
+                    obj.w.x(ii,1:opts.N_finite_elements(ii),1:(opts.n_s+rbp-1)).ub = inf*ones(dims.n_x, 1);
                 end
 
                 if ~opts.x_box_at_fe
-                    obj.w.x(1:opts.N_stages,1:(opts.N_finite_elements(ii)-1),opts.n_s+rbp).lb = -inf*ones(dims.n_x, 1);
-                    obj.w.x(1:opts.N_stages,1:(opts.N_finite_elements(ii)-1),opts.n_s+rbp).ub = inf*ones(dims.n_x, 1);
+                    obj.w.x(ii,1:(opts.N_finite_elements(ii)-1),opts.n_s+rbp).lb = -inf*ones(dims.n_x, 1);
+                    obj.w.x(ii,1:(opts.N_finite_elements(ii)-1),opts.n_s+rbp).ub = inf*ones(dims.n_x, 1);
                 end
             end
         end