In This Package:

RRawStream.cc

Go to the documentation of this file.

/*
 *
 *   RRawStream, input stream for rraw type raw data.
 *   It provides some similar operations like ifstream to handle root TTree.
 *
 *   Zhe Wang
 *   Feb. 10, 2010 at BNL
 *
 */

#include "RawData/RRawStream.h"
#include <cstring>
#include <stdlib.h>

static void sanity()
{
  if( sizeof(Int32) != 4 ) {
    cerr<<"RawStream:: error: Need a 32-bits type to hold each line of raw data"<<endl;
    cerr<<"RawStream:: Redefine type Int32"<<endl;
    abort();
  }
}

RRawStream::RRawStream():
  m_file(0),m_tree(0),m_size(0),
  m_currentEntry(-1),m_currentPosi(0),m_entryEnd(true)
{
  sanity();
}

RRawStream::RRawStream( const char* filename ):
  m_file(0),m_tree(0),m_size(0),
  m_currentEntry(-1),m_currentPosi(0),m_entryEnd(true)
{
  sanity();
  this->open(filename);
}

RRawStream::~RRawStream()
{
  this->close();
}

void RRawStream::close ( )
{
  if(!m_file) return;

  m_file->Close();
  delete m_file;
  m_file=0;

}

void RRawStream::open ( const char * filename )
{
  // open a file
  m_file = TFile::Open( filename, "READ" );
  if(!m_file)  {
    cout<<"RRawStream::Failed to open "<<filename<<endl;
    abort();
  }
  if( !(m_file->IsOpen()) )  {
    cout<<"RRawStream::Failed to open "<<filename<<endl;
    abort();
  }

  // get the tree
  m_tree = (TTree*)m_file->Get("RRawTree");
  if(!m_tree)  {
    cout<<"RRawStream::Failed to get TTree RRawTree"<<endl;
    abort();
  }

  // set branch address
  m_tree->SetBranchAddress("size",&m_size);
  m_tree->SetBranchAddress("buffer",m_buffer);
    
  // reset current entry number to -1
  m_currentEntry=-1;
        
  // reset reading status of current entry
  m_currentPosi=0;
  m_entryEnd=true;
}

bool RRawStream::is_open ( ) const
{
  if(!m_file) 
    return false;
  else
    return m_file->IsOpen();
}

bool RRawStream::good ( ) const
{
  if (!is_open()) return false;
  return !m_file->IsZombie();
}

// Equivalant to fail() of ifstream
bool RRawStream::operator ! ( ) const
{
  if (!this->is_open()) return true;
  return !good();
}

RRawStream& RRawStream::read ( char* s, int n )
{
  // Need to handle the case of reaching to the end of one entry
  if(m_entryEnd) {
    m_currentEntry++;
    m_tree->GetEntry(m_currentEntry);
    m_entryEnd=false;
    m_currentPosi = 0;

    /*
    cout<<"RRawStream::New Entry "<<m_size<<endl;
    for(int i=0;i<m_size;i++) {
      cout<<(unsigned int)m_buffer[i]<<" ";
    }
    cout<<endl;
    */
  }

  if( m_currentPosi+n <= m_size * 4 )  {
    memcpy(s, (char*)(m_buffer) + m_currentPosi, n);
    m_currentPosi+=n;
  }  else  {
    cout<<"RRawStream::Asking data over one root entry boundary"<<endl;
    abort();
  }

  // Mark the end
  if(m_currentPosi == m_size * 4 ) {
    m_entryEnd = true;
  }

  return *this;
}

RRawStream& RRawStream::ignore ( int n /* 1 */ )
{
  // The same as read(...), just without memcpy
  
  // Need to handle the case of reaching to the end of one entry
  if(m_entryEnd) {
    m_currentEntry++;
    m_tree->GetEntry(m_currentEntry);
    m_entryEnd=false;
    m_currentPosi = 0;

    /*
    cout<<"RRawStream::New Entry "<<m_size<<endl;
    for(int i=0;i<m_size;i++) {
      cout<<(unsigned int)m_buffer[i]<<" ";
    }
    cout<<endl;
    */
  }

  m_currentPosi+=n;

  // Mark the end
  if(m_currentPosi == m_size * 4 ) {
    m_entryEnd = true;
  }

  return *this;
}


// Local Variables: **
// c-basic-offset:2 **
// indent-tabs-mode:nil **
// End: **

---