IoRerootStreamItr.cxx

Go to the documentation of this file.

// $Id: IoRerootStreamItr.cxx,v 1.8 2006/06/02 02:36:20 schubert Exp $
//
// Class for walking over a reroot input file
//
// messier@huhepl.harvard.edu
#include "IoModules/IoRerootStreamItr.h"
#include "MessageService/MsgService.h"
#include "Validity/VldContext.h"

#include <cassert>

CVSID("$Id: IoRerootStreamItr.cxx,v 1.8 2006/06/02 02:36:20 schubert Exp $");

//......................................................................

IoRerootStreamItr::IoRerootStreamItr(const char* sourceName) : fIndex(-1),
                           fBOFIndex(-1),fEOFIndex(-1),fIsValid(true)
{
  // Need to setup the MINFAST environment to work with reroot data
  if (gMINFast == 0) {
    // Create a MINFAST object w/o the display
    gMINFast = new MINFast("gMinFast","MINFAST",kFALSE,"");
    static IoRerootStreamItr::Cleaner c; c.UseMe();
    assert(gMINFast);
  }
  this -> SetSourceName(sourceName);

}

//......................................................................

IoRerootStreamItr::~IoRerootStreamItr()
{
  this->CloseFile();
}

//......................................................................

int IoRerootStreamItr::LoadRecords(MomNavigator* /* mom */) 
{
  if (!IsOpenFile() || fIndex<0 || fIndex>=fEOFIndex) return 0;
  MSG("Io",Msg::kDebug) << "IoRerootStreamItr::LoadRecord at index " 
                        << fIndex << endl;
  if (gMINFast->GetEvent(fIndex)) return 1;
  return 0;
}

//......................................................................

int IoRerootStreamItr::Increment(int n, MomNavigator* m)
{

  if (!IsOpenFile() || n<=0) return 0;

  // Case where requested advance steps past start of file
  if ((fIndex + n) >= fEOFIndex) {
    int isave = fIndex;
    fIndex = fEOFIndex;
    if (m) this->LoadRecords(m);
    return (fEOFIndex-isave-1);
  }

  // Normal case
  fIndex += n;
  if (m) this->LoadRecords(m);
  return n;
}

//......................................................................

int IoRerootStreamItr::Decrement(int n, MomNavigator* m)
{
//======================================================================
// Back up n positions in the stream.
//======================================================================
  if (!IsOpenFile() || n<=0) return 0;

  // Case where requested advance steps past start of file
  if ((fIndex - n) <= fBOFIndex) {
    int isave = fIndex;
    fIndex = fBOFIndex;
    if (m) this->LoadRecords(m);
    return (isave-fBOFIndex-1);
  }

  // Normal case
  fIndex -= n;
  if (m) this->LoadRecords(m);
  return n;
}

//......................................................................

JobCResult IoRerootStreamItr::GoTo(const VldContext& vld, MomNavigator* m)
{
//======================================================================
// Go to record set with specified validity context -- for reroot I
// don't know what this means...
//======================================================================
  MSG("Io",Msg::kWarning) << 
    "GoTo(" << vld << ") not supported for REROOT data.\n";
  if (m) this->LoadRecords(m);
  return JobCResult::kWarning;
}

//......................................................................

int IoRerootStreamItr::GoToEOF() { fIndex = fEOFIndex; return 1;}

//......................................................................

JobCResult IoRerootStreamItr::OpenFile() 
{
//======================================================================
// Open the file attached to current file iterator
//======================================================================
  std::string fullfilepathname = this -> GetCurrentFile();
  if ( fullfilepathname == "" ) return JobCResult::kWarning;
  this -> SetSourceName(fullfilepathname.c_str());

  gMINFast->OpenRerootFile(fullfilepathname.c_str());
  
  // Test if this is a valid file
  if ( !IsOpenFile() ) return JobCResult::kWarning;
  
  // Set up the indicies
  fBOFIndex = -1;
  fEOFIndex = gMINFast->GetMINFile()->Getnevts();
  fIndex    = fBOFIndex;

  return JobCResult::kAOK;
}

//......................................................................

void IoRerootStreamItr::CloseFile() {
  // Handled automatically by gMINFast
  this -> SetSourceName("");
  return;
}

bool IoRerootStreamItr::IsOpenFile() const {
  bool isOpen = true;
  if (gMINFast->GetMINFile()==0 || 
      gMINFast->GetMINFile()->Getnevts() ==0) isOpen = false;
  return isOpen;
}

---