Update P during imuCallback

1 files changed, 8 insertions, 3 deletions

diff --git a/src/main.h b/src/main.h
index aa97995..d1932c8 100644
--- a/src/main.h
+++ b/src/main.h
@@ -13,6 +13,7 @@
 #include <iostream>
 
 #include "body.h"
+#include "state.h"
 
 #define MAXLINE 8192
 #define MAXFILENAME 1024
@@ -83,10 +84,14 @@ timestamp update_dt(const timestamp t, timestamp *t_old);
 #ifdef  USE_ROS
 void imuCallback();
 #else      /* -----  not USE_ROS  ----- */
-void covCallback(const message &msg, Eigen::Matrix<double,9,9> &P, const Eigen::Quaternion<double> &q);
+void covCallback(const message &msg, Eigen::Matrix<double,9,9> &P,
+        const Eigen::Quaternion<double> &q);
 void imgCallback(const message *msg);
-void imuCallback(const message &msg, Eigen::Matrix<double,9,1> &X, const Eigen::Quaternion<double> &q, const timestamp dt);
-void pvaCallback(const message &msg, Eigen::Matrix<double,9,1> &X, Eigen::Quaternion<double> &q);
+void imuCallback(const message &msg, Eigen::Matrix<double,9,1> &X,
+        Eigen::Matrix<double,9,9> &P, const Eigen::Quaternion<double> &q,
+        const timestamp dt);
+void pvaCallback(const message &msg, Eigen::Matrix<double,9,1> &X,
+        Eigen::Quaternion<double> &q);
 void utmCallback(const message &msg, Eigen::Matrix<double,9,1> &X);
 #endif     /* -----  not USE_ROS  ----- */