00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018 #include <plObject.h>
00019 #include <plMath.h>
00020 #include <plFloatVector.h>
00021 #include <plFloatMatrix.h>
00022 #include <plKernel.h>
00023 #include <plValues.h>
00024
00025
00026
00027
00028
00029
00030
00031
00032 #ifndef __LINEAR_KALMAN_FILTER_H
00033 #define __LINEAR_KALMAN_FILTER_H
00034
00035
00036 #ifndef INVERSE_SEUIL
00037 #define INVERSE_SEUIL 1.0e-38
00038 #endif
00039
00041 class plKalmanFilter : public plObject
00042 {
00043 protected:
00044
00045 plFloatMatrix I;
00046
00047 unsigned int state_size;
00048 unsigned int obs_size;
00049 unsigned int command_size;
00050
00051
00052
00053 void resize( unsigned int s, unsigned int o, unsigned int c);
00054
00055
00056 void internal_prediction();
00057 void internal_update();
00058
00059 public:
00060 plFloatMatrix A, At, B, H, Ht, Q, R, K, P_update, P_predict;
00061 plFloatVector x_update, x_predict, z, u;
00062
00063
00064 plKalmanFilter() {};
00065 plKalmanFilter(unsigned int n_in, unsigned int m_im, unsigned int c_in=1);
00066
00067
00068 virtual ~plKalmanFilter() {};
00069
00073 void update_matrix(const plFloatMatrix& Ain)
00074 {
00075 A = Ain;
00076 At = Ain.transpose();
00077 }
00078 void control_matrix(const plFloatMatrix& Bin) { B = Bin; }
00079 void sys_noise(const plFloatMatrix& QS) { Q = QS; }
00080 void obs_noise(const plFloatMatrix& Ro) { R = Ro; }
00081 void obs_sys(const plFloatMatrix& Hin)
00082 {
00083 H = Hin;
00084 Ht = Hin.transpose();
00085 }
00086 void initial_prediction(const plFloatVector& Xo,
00087 const plFloatMatrix& Po)
00088 {
00089 x_predict = Xo;
00090 P_predict = Po;
00091 }
00092
00093 void initial_update(const plFloatVector& Xo,
00094 const plFloatMatrix& Po)
00095 {
00096 x_update = Xo;
00097 P_update = Po;
00098 }
00099
00100
00101 void commande(const plFloatVector& Uin)
00102 {
00103 u = Uin;
00104 }
00105
00106
00107 void observation(const plFloatVector& Zin)
00108 {
00109 z = Zin;
00110 }
00111
00112 void prediction();
00113
00114 void update();
00115
00116
00117 plFloatVector& mean() { return x_update;}
00118 plFloatMatrix& covariance() { return P_update;}
00119
00120 };
00121
00122 #endif