work/git/mixpp/bdm/stat/loggers.h

Go to the documentation of this file.

#ifndef LGR_H
#define LGR_H

#include <itpp/itbase.h>
#include "libBM.h"
#include "../itpp_ext.h"

using namespace itpp;
using std::string;

class logger {
protected:
        Array<RV> entries;
        Array<string> names;
public:
        logger ( ) : entries(0),names ( 0 ) {}

        virtual int add (const RV &rv, string name="" ) {
                int id=entries.length();
                names=concat ( names, name ); // diff
                entries.set_length(id+1,true);
                entries(id)= rv;
                return id; // identifier of the last entry
        }

        virtual void logit ( int id, const vec &v ) =0;

        virtual void step() =0;

        virtual void finalize() {};

        virtual ~logger() {};
};


class memlog : public logger {

protected:
        int maxlen;
        int ind;
        Array<mat> vectors;
public:
        memlog ( int maxlen0 ) : logger(),maxlen ( maxlen0 ),ind ( 0 ),vectors ( 0 ) {}
        void init() {
                int i; int n =entries.length();
                vectors.set_size ( n ); 
                for ( i=0;i<n;i++ ) {vectors(i).set_size (maxlen,entries(i).count() );}
                ;
        }
        void step() {if ( ind<maxlen ) ind++; else it_error ( "memlog::ind is too high;" );}
        void logit ( int id, const vec &v ) {
                it_assert_debug(id<vectors.length(),"Logger was not initialized, run init().");
                vectors ( id ).set_row ( ind,v );}
        void itsave(const char* fname);
};

class dirfilelog : public memlog {

protected:
        string dirname;
        Array<string> scalarnames;
public:
        dirfilelog ( std::string dirname0, int maxlen0 ) : memlog ( maxlen0 ), dirname ( dirname0 ), scalarnames ( 0 ) {}
        void init();
        void step();
        void finalize();
        void write_buffers ( int Len );
};


#endif // LGR_H

---