MCFsbostream.h

Go to the documentation of this file.

// MCFsbostream.h
//////////////////////////////////////////////////////////////////////
//          (c) Copyright 2001-2002 Brown Deer Technology, LLC.
//                        All rights reserved.
//////////////////////////////////////////////////////////////////////

#ifndef MCFSBOSTREAM_H
#define MCFSBOSTREAM_H

#ifdef USE_STD  
#include <iostream>
#include <string>
using namespace std;
#else
#include <iostream.h>
#include <string.h>
#endif

#include "utility.h"
#include "MCFmp.h"
#include "MCF_Code.h"

#include "MCFsbostreamBase.h"

template < class Rtmra_t >
class MCFsbostream : public MCFsbostreamBase 
{
   public:
      typedef typename Rtmra_t::DataType Data_t;
      MCFsbostream( const string& descript = "_", ostream& os = cout );
      ~MCFsbostream() {}
      virtual void final( Rtmra_t& );
      MCFsbostream<Rtmra_t>& operator << ( Rtmra_t& );
      void threshold( const Data_t threshold );
   protected:
      Data_t _threshold;
};

template <class Rtmra_t>
MCFsbostream<Rtmra_t>::MCFsbostream( const string& descript, ostream& os) 
  : MCFsbostreamBase( descript, os ), _threshold(0)
{ 
   _arraydim = Rtmra_t::ArrayDim; 
   _datatype_code = DataType_Code<Data_t>::eval();
}

template <class Rtmra_t>
void MCFsbostream<Rtmra_t>:: final( Rtmra_t& x ) 
{
   bool eob = ( x.mcycle() == x.mmax() );
   _lockfinalstep = true;
   while ( !eob ) {
//      x.assign( 0.0 );
      x.assign( typename Rtmra_t::DataType(0) );
      x++;
      (*this) << x;
      eob = ( x.mcycle() == x.mmax() );
   }
}

template <class Rtmra_t>
MCFsbostream<Rtmra_t>& MCFsbostream<Rtmra_t>::operator << ( Rtmra_t& x ) 
{

   MCFmp<Rtmra_t> mp( _bplen_mpsizeb, _bplen_coef, _bplen_n, _bplen_a );
   mp.threshold(_threshold);

   if ( !isopen() ) open();

   if ( isopen() ) {
      if ( _needheader ) outheader();
      int m = x.mcycle();
      bool eob = ( m == x.mmax() );
      if (eob) { 
         (*_os) << mp( x, char(m) ); 
      }
      while ( m != 0 ) { 
        (*_os) << mp( x, char(0x80|m) );
        m--;
      }

      if ( !_lockfinalstep ) {
//         _finalstep = x.mrastep() & ( ( 1 << _mmax ) - 1 );  HACK
         _finalstep = x.mrastep() & ( ( 1 << x.mmax() ) - 1 );
#ifdef STORM_DEBUG
         cout << "locked with _finalstep = " << _finalstep << endl;
#endif
      }
      if (eob) {
         close();
         _sbn++;
         _needheader = true;
      }
   }
   return *this;
}

template <class Rtmra_t>
void MCFsbostream<Rtmra_t>::threshold ( const Data_t threshold )
{
   _threshold = threshold;
}

#endif

---