From 8198114122f94e9a31a32baa66e200c42732fc53 Mon Sep 17 00:00:00 2001
From: odri <odri@furano.laas.fr>
Date: Fri, 24 Sep 2021 18:04:05 +0200
Subject: [PATCH] Tweak mpc comparison script

---
 scripts/crocoddyl_eval/test_6/compare_mpcs.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/scripts/crocoddyl_eval/test_6/compare_mpcs.py b/scripts/crocoddyl_eval/test_6/compare_mpcs.py
index 01743216..8094b8fc 100644
--- a/scripts/crocoddyl_eval/test_6/compare_mpcs.py
+++ b/scripts/crocoddyl_eval/test_6/compare_mpcs.py
@@ -58,12 +58,14 @@ def compute_RMSE(array, norm):
 ##############
 
 # [Linear, Non Linear, Planner, OSQP]
-MPCs = [True, True, True, True] # Boolean to choose which MPC to plot
+MPCs = [True, True, True, False] # Boolean to choose which MPC to plot
 MPCs_names = ["No FF", "-afeet", "JT fmpc + M ddq", "Free flyer + Tasks"]
 name_files = ["data_2021_08_27_16_08_0.npz", "data_2021_08_27_15_53_0.npz", "data_2021_08_27_15_42_0.npz", "data_2021_08_27_15_44_0.npz"] # Names of the files
 name_files = ["data_2021_09_02_16_51_0.npz", "data_2021_09_02_16_53_0.npz", "data_2021_09_02_16_55_0.npz", "data_2021_09_02_17_52_0.npz"] # Names of the files
 folder_path = "" # Folder containing the 4 .npz files
 
+name_files = ["data_2021_09_16_18_08_0.npz", "data_2021_09_16_18_12_0.npz", "data_2021_09_16_18_20_0.npz", "data_2021_09_16_18_12_0.npz"]
+
 # Common data shared by 4 MPCs
 params = lqrw.Params()  # Object that holds all controller parameters
 logs = np.load(folder_path + name_files[0])
-- 
GitLab