root/tests/testKF.cpp @ 18

Revision 18, 1.2 kB (checked in by smidl, 17 years ago)

update

Line 
1#include <itpp/itbase.h>
2#include "../bdm/estim/libKF.h"
3
4using namespace itpp;
5
6//These lines are needed for use of cout and endl
7using std::cout;
8using std::endl;
9
10int main() {
11
12        // Klaman filter
13        mat A, B,C,D,R,Q,P0;
14        vec mu0;
15        mat Mu0;;
16        // input from Matlab
17        it_file fin( "testKF.it" );
18
19        mat Dt, Xt,Xt2;
20        int Ndat;
21
22        bool xxx= fin.seek( "d" );
23        fin >>Dt;
24        fin.seek( "A" ); 
25        fin >> A;
26        fin.seek( "B" ); 
27        fin >> B;
28        fin.seek( "C" ); 
29        fin >> C;
30        fin.seek( "D" ); 
31        fin >> D;
32        fin.seek( "R" ); 
33        fin >> R;
34        fin.seek( "Q" ); fin >> Q;
35        fin.seek( "P0" ); fin >> P0;
36        fin.seek( "mu0" ); fin >> Mu0; 
37        mu0=Mu0.get_col(0);
38       
39        Ndat = Dt.cols();
40        Xt=zeros( 2,Ndat );
41        Xt2=zeros( 2,Ndat );
42
43        Kalman<ldmat> KF( A,B,C,D,ldmat(R),ldmat(Q),ldmat(P0),mu0 );
44//      cout << KF;
45        KalmanFull KF2( A,B,C,D,R,Q,P0,mu0 );
46
47        Xt.set_col( 0,KF.mu );
48        Xt2.set_col( 0,KF.mu );
49        for ( int t=1;t<Ndat;t++ ) {
50                KF.bayes( Dt.get_col( t ));
51                KF2.bayes( Dt.get_col( t ));
52                Xt.set_col(t,KF.mu);
53                Xt2.set_col(t,KF2.mu);
54//              Kmu = KF.mu;
55//              cout <<  "t:" <<t<< "  " << dt<<"  "<<Kmu <<endl;
56        }
57
58        it_file fou( "testKF_res.it" );
59        fou << Name("xth") << Xt;
60        fou << Name("xth2") << Xt2;
61        //Exit program:
62        return 0;
63
64}
Note: See TracBrowser for help on using the browser.