[WLS] Fix defines to be used in other source files. (#3154)

paparazzi · Nov 2, 2023 · 2ba164c · 2ba164c
1 parent decbf5c
commit 2ba164c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/sw/airborne/math/wls/wls_alloc.c b/sw/airborne/math/wls/wls_alloc.c
@@ -55,13 +55,6 @@ static void print_final_values(int n_u, int n_v, float* u, float** B, float* v,
 static void print_in_and_outputs(int n_c, int n_free, float** A_free_ptr, float* d, float* p_free);
 #endif
 
-#ifndef WLS_N_U
-#define WLS_N_U 6
-#endif
-
-#ifndef WLS_N_V
-#define WLS_N_V 4
-#endif
 
 #define WLS_N_C ((WLS_N_U)+(WLS_N_V))
 

diff --git a/sw/airborne/math/wls/wls_alloc.h b/sw/airborne/math/wls/wls_alloc.h
@@ -52,6 +52,16 @@
 #ifndef WLS_ALLOC_HEADER
 #define WLS_ALLOC_HEADER
 
+#include "generated/airframe.h"
+
+#ifndef WLS_N_U
+#define WLS_N_U 6
+#endif
+
+#ifndef WLS_N_V
+#define WLS_N_V 4
+#endif
+
 
 extern int wls_alloc(float* u, float* v,
               float* umin, float* umax, float** B,