Add feature handling to state

1 files changed, 2 insertions, 1 deletions

diff --git a/src/main.h b/src/main.h
index 174ab0e..0698e68 100644
--- a/src/main.h
+++ b/src/main.h
@@ -7,6 +7,7 @@
 #include <Eigen/Dense>
 #include <istream>
 #include <iostream>
+#include <vector>
 
 #include "body.h"
 #include "camera.h"
@@ -24,7 +25,7 @@ int parseLine(char *line, message *msg);
 timestamp update_dt(const timestamp t, timestamp *t_old);
 
 void covCallback(const message &msg, Matrix<double,9,9> &P, const Quaterniond &q);
-void imgCallback(message &msg);
+void imgCallback(message &msg, State &mu, Camera &cam, const Quaterniond &q);
 void imuCallback(const message &msg, State &mu, const Quaterniond &q, const timestamp dt);
 void pvaCallback(const message &msg, Matrix<double,9,1> &X, Quaterniond &q);
 void utmCallback(const message &msg, State &mu);