root/bdm/stat/libBM.cpp @ 14

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

using namespace itpp;

using std::cout;

void RV::init( ivec in_ids, Array<std::string> in_names, ivec in_sizes, ivec in_times, ivec in_obs ) {
        //
        int i;
        len = in_ids.length();
        //PRUDENT_MODE
        // All vetors should be of same length
        if (     ( len != in_ids.length() ) || \
                ( len != in_names.length() ) || \
                ( len != in_sizes.length() ) || \
                ( len != in_times.length() ) || \
                ( len != in_obs.length() ) ) {
                it_error( "RV::RV inconsistent length of input vectors." );
        }

        ids = in_ids;
        names = in_names;
        sizes = in_sizes;
        times = in_times;
        obs = in_obs;
        size = 0;
        for(i=0;i<len;i++){size+=sizes(i);}
};

RV::RV ( ivec in_ids, Array<std::string> in_names, ivec in_sizes, ivec in_times, ivec in_obs ) {
        init ( in_ids, in_names, in_sizes, in_times, in_obs );
}

RV::RV () {};

RV::RV ( ivec in_ids ) {
        
        len = in_ids.length();
        Array<std::string> A( len );
        std::string rvstr = "rv";

        for ( int i = 0; i < len;i++ ) {
                A( i ) = rvstr + to_str(i);
        }

        init ( in_ids, A, ones_i( len ), zeros_i( len ), zeros_i( len ) );
}

RV RV::subselect(ivec ind){
        return RV(ids(ind), names(to_Arr(ind)), sizes(ind), times(ind), obs(ind));
}

void RV::t(int delta){ times +=delta;}

RV RV::operator()(ivec ind){
        return RV(ids(ind), names(to_Arr(ind)), sizes(ind), times(ind), obs(ind));
}

std::ostream &operator<<( std::ostream &os, const RV &rv ) {

        for ( int i = 0; i < rv.len ;i++ ) {
                os << rv.ids( i ) << "(" << rv.sizes( i ) << ")" <<  // id(size)=
                "=" << rv.names( i )  << "_{"  << rv.times( i ) << "}; "; //name_{time}
        }
        return os;
}