Changeset 263 for bdm/stat/libDS.cpp

Timestamp:

02/09/09 23:11:35 (15 years ago)

Author:

smidl

Message:

UIArxDS test

Files:

• bdm/stat/libDS.cpp (modified) (1 diff)

bdm/stat/libDS.cpp

@@ -1 @@
-#include <itpp/itbase.h>
+
 #include "libDS.h"
 
 using namespace bdm;
 
-void MemDS::getdata(vec &dt){
+void MemDS::getdata ( vec &dt ) {
         int i;
-        
-        it_assert_debug(dt.length()==rowid.length(), "MemDS:getdata incompatible dt");
-        for(i=0;i<rowid.length();i++){
-                dt(i) = Data(rowid(i),time-delays(i));
+
+        it_assert_debug ( dt.length() ==rowid.length(), "MemDS:getdata incompatible dt" );
+        for ( i=0;i<rowid.length();i++ ) {
+                dt ( i ) = Data ( rowid ( i ),time-delays ( i ) );
         }
 }
 
-void MemDS::getdata(vec &dt,ivec &indeces){
+void MemDS::getdata ( vec &dt,const ivec &indeces ) {
         int j,i;
-        it_assert_debug(dt.length()==indeces.length(), "MemDS:getdata incompatible dt");
-        for(i=0;i<indeces.length();i++){
-                j = indeces (i);
-                dt(i) = Data(rowid(j),time-delays(j));
+        it_assert_debug ( dt.length() ==indeces.length(), "MemDS:getdata incompatible dt" );
+        for ( i=0;i<indeces.length();i++ ) {
+                j = indeces ( i );
+                dt ( i ) = Data ( rowid ( j ),time-delays ( j ) );
         }
 }
 
 void MemDS::step() {
-        if (time<Data.cols()) {time++;}
+        if ( time<Data.cols() ) {time++;}
 }
 
-void MemDS::linkrvs(RV &drv, RV &urv) {
-        it_assert_debug(drv.count()==rowid.length(),"MemDS::linkrvs incompatible drv");
-        it_assert_debug(urv.count()==0,"MemDS does not support urv.");
-        
+void MemDS::linkrvs ( RV &drv, RV &urv ) {
+        it_assert_debug ( drv.count() ==rowid.length(),"MemDS::linkrvs incompatible drv" );
+        it_assert_debug ( urv.count() ==0,"MemDS does not support urv." );
+
         Drv = drv;
         Urv = urv;
 }
 
-MemDS::MemDS(mat &Dat, ivec &rowid, ivec &delays){
-        it_assert_debug(max(rowid)<=Dat.rows(),"MemDS rowid is too high for given Dat.");
-        it_assert_debug(max(delays)<Dat.cols(),"MemDS delays are too high.");
-        
-        time = max(delays); 
+MemDS::MemDS ( mat &Dat, ivec &rowid, ivec &delays ) {
+        it_assert_debug ( max ( rowid ) <=Dat.rows(),"MemDS rowid is too high for given Dat." );
+        it_assert_debug ( max ( delays ) <Dat.cols(),"MemDS delays are too high." );
+
+        time = max ( delays );
         Data = Dat;
 }
+
+void ArxDS::step() {
+        double tmp;
+        //get regressor
+        rgr = rgrlnk.get_val ( H );
+        // Eval Y
+        Y = model.samplecond ( rgr,tmp );
+
+        //shift history
+        H.shift_right ( 0, Drv.count() +Urv.count() );
+        //fill new data
+        H.set_subvector ( Drv.count(),Y );
+
+        //Leaving U.length() empty slots - these will be filled by write()
+}
+
+//! Auxiliary function building full history of rv0
+RV fullrgr ( const RV &drv0, const RV &urv0, const RV &rrv0 ) {
+        RV T ( urv0 );
+        RV pom = concat(drv0, urv0);
+        int mint = rrv0.mint();
+        for ( int i=0; i>mint; i-- ) {
+                pom.t ( -1 );
+                T.add ( pom );
+        }
+        return T;
+}
+
+ArxDS::ArxDS ( RV &drv, RV &urv, RV &rrv ) : 
+                DS(drv,urv), Rrv(rrv), Hrv( fullrgr ( drv,urv,rrv )), //RVs
+                Y(drv.count()), H(Hrv.count()) ,rgr ( rrv.count() ),  //tmp variables
+                rgrlnk (Hrv ,rrv ) ,model ( drv,rrv ) {
+}