RawRecCounts.cxx

Go to the documentation of this file.

// $Id: RawRecCounts.cxx,v 1.6 2007/01/30 14:26:51 rhatcher Exp $
//
// A simple job module for characterizing a RawData file
// including giving the first and last VldContext (and those of snarls).
// It should give sensible output when presented with Daq, Dcs or Beam files.
//
// rhatcher@fnal.gov
#include "DataUtil/RawRecCounts.h"

#include <cstdio>
#include <iomanip>

#include "TSystem.h"
#include "TFile.h"
#include "TROOT.h"

#include "MessageService/MsgService.h"
#include "MinosObjectMap/MomNavigator.h"
#include "JobControl/JobCModuleRegistry.h" // For JOBMODULE macro

#include "RawData/RawRecord.h"
#include "RawData/RawDaqSnarlHeader.h"
#include "RawData/RawDigitDataBlock.h"
#include "RawData/RawDigit.h"

using namespace std;

JOBMODULE(RawRecCounts, "RawRecCounts",
          "print at EndJob first/last VldContext");
CVSID("$Id: RawRecCounts.cxx,v 1.6 2007/01/30 14:26:51 rhatcher Exp $");

static const VldContext defaultVldContext(Detector::kUnknown,
                                          SimFlag::kUnknown,
                                          VldTimeStamp((time_t)0,0));

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

RawRecCounts::RawRecCounts()
  : fFirstVldc(defaultVldContext),      fLastVldc(defaultVldContext), 
    fFirstSnarlVldc(defaultVldContext), fLastSnarlVldc(defaultVldContext),
    fNumRecords(0), fNumRecSets(0), fFirstSnarl(-1), fLastSnarl(-1),
    // initialize classes for drawing, writing
    fRunningStatus(0), fFileRootVersion(0)
{
//======================================================================
// JobModule ctor
//======================================================================
  ClearStatus();
}

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

RawRecCounts::~RawRecCounts()
{
//======================================================================
// JobModule dtor
//======================================================================
}

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

void RawRecCounts::EndJob()
{
//======================================================================
// JobModule EndJob
//======================================================================

  Report();

}

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

JobCResult RawRecCounts::Ana(const MomNavigator* mom)
{
//======================================================================
// JobCModule analysis hook
//======================================================================

  fNumRecSets++;
  if (fRunningStatus>0 && fNumRecSets%fRunningStatus == 0)
    MSG("DataUtil",Msg::kInfo)
      << "\r processed " << fNumRecSets << " record sets\r" << flush;

  TObject *tobj = 0;
  TIter reciter = mom->FragmentIter();
  while ( ( tobj = reciter() ) ) {
    RawRecord *rawrec  = dynamic_cast<RawRecord*>(tobj);
    if (!rawrec) continue;

    // check for new file
    const Registry& registry = rawrec->GetTempTags();
    const char* tmpfile = 0;
    registry.Get("file",tmpfile);
    // we only want the basename
    const char* basename = gSystem->BaseName(tmpfile);
    std::string thisfile = basename;
    if ( thisfile != fCurrentFile ) {
      if ( fCurrentFile != "" ) Report();
      ClearStatus();
      fCurrentFile = thisfile;

      // find the TFile for this new filename
      TSeqCollection* listOfFiles = gROOT->GetListOfFiles();
      TIter fileItr(listOfFiles);
      TFile* afile = 0;
      std::string currentBaseName = gSystem->BaseName(fCurrentFile.c_str());
      
      while ( ( afile = (TFile*)fileItr.Next() ) ) {
        std::string fname = afile->GetName();
        std::string fbasename = gSystem->BaseName(fname.c_str());

        if ( currentBaseName == fbasename ) {  // test without any path info
          fFileRootVersion = afile->GetVersion();
        }
      }
    }  // new file

    // process the record
    fNumRecords++;

    const RawHeader *rawhead = rawrec->GetRawHeader();
    const RawDaqSnarlHeader *rawdaqsnarlhead =
                         dynamic_cast<const RawDaqSnarlHeader*>(rawhead);

    if (fFirstVldc == defaultVldContext)
        fFirstVldc = rawhead->GetVldContext();
    fLastVldc = rawhead->GetVldContext();

    if (rawdaqsnarlhead) {
      if (fFirstSnarlVldc == defaultVldContext) {
          fFirstSnarlVldc = rawdaqsnarlhead->GetVldContext();
          fFirstSnarl     = rawdaqsnarlhead->GetSnarl();
      }
      fLastSnarlVldc = rawhead->GetVldContext();
      fLastSnarl = rawdaqsnarlhead->GetSnarl();
    }

    ProcessBlocksInRecord(rawrec);

  }  // loop over records

  return JobCResult::kPassed; // kNoDecision, kFailed, etc.
}

//......................................................................
void RawRecCounts::ClearStatus()
{
//======================================================================
// clear all the current counts
//======================================================================
  fFirstVldc      = defaultVldContext;
  fLastVldc       = defaultVldContext; 
  fFirstSnarlVldc = defaultVldContext;
  fLastSnarlVldc  = defaultVldContext;
  
  fNumRecords     =  0;
  fNumRecSets     =  0;
  fFirstSnarl     = -1;
  fLastSnarl      = -1;
}

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

void RawRecCounts::ProcessBlocksInRecord(const RawRecord* rawrec)
{
//======================================================================
// Loop over raw blocks in the record
//======================================================================

  TIter rbitr = rawrec->GetRawBlockIter();
  const RawDataBlock* rdb = 0;
  while ( ( rdb = dynamic_cast<const RawDataBlock*>(rbitr()) ) ) {
    CountRawBlockTypes(rdb);
  } // loop over raw blocks
}

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

void RawRecCounts::CountRawBlockTypes(const RawDataBlock* rdb)
{
//======================================================================
// Accumulate count of RawDataBlock types
//======================================================================

  string name = rdb->GetName();
  if ( fRawBlockCounts.find(name) == fRawBlockCounts.end() )
    fRawBlockCounts[name] = 0;
  fRawBlockCounts[name] += 1;

}

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

const Registry& RawRecCounts::DefaultConfig() const
{
//======================================================================
// Supply the default configuration for the module
//======================================================================
  static Registry r; // Default configuration for module

  // Set name of config
  std::string name = this->GetName();
  name += ".config.default";
  r.SetName(name.c_str());

  // Set values in configuration
  r.UnLockValues();
  r.Set("RunningStatus", 0); 
  r.LockValues();

  return r;
}

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

void RawRecCounts::Config(const Registry& r)
{
//======================================================================
// Configure the module given the Registry r
//======================================================================
  int    tmpi;
  if (r.Get("RunningStatus",tmpi)) { fRunningStatus = tmpi; }
}

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

void RawRecCounts::Report()
{
//======================================================================
// Print the JobCModule's status
//======================================================================

  int rvi = fFileRootVersion;
  int rvmajor = rvi/10000;
  int rvminor = rvi/100 - 100*rvmajor;
  int rvtiny  = rvi%100;
  string rootVersionString = Form("v%02d-%02d-%02d",rvmajor,rvminor,rvtiny);

  MSG("DataUtil",Msg::kInfo) 
    << endl << endl
    << "RawRecCounts Report: " << fCurrentFile << endl
    << "       root version: " << rootVersionString
    << endl << endl;

  MSG("DataUtil",Msg::kInfo) 
    << "  VldContexts: " << endl
    << "  First:       " << fFirstVldc << endl
    << "  First Snarl: " << fFirstSnarlVldc 
    << " # " << setw(10) << fFirstSnarl << endl
    << "  Last:        " << fLastVldc << endl
    << "  Last  Snarl: " << fLastSnarlVldc 
    << " # " << setw(10) << fLastSnarl << endl
    << "  in " << fNumRecords << " records of "
    << fNumRecSets << " record sets " << endl
    << endl;

  std::map<std::string,int>::iterator mitr = fRawBlockCounts.begin();
  for ( ; mitr != fRawBlockCounts.end(); mitr++) {
    MSG("DataUtil",Msg::kInfo)
      << "  " << setw(30) << left << (*mitr).first << " " 
      << setw(10) << right << (*mitr).second
      << resetiosflags(ios::adjustfield) // undo "right"
      << endl;
  }

  MSG("DataUtil",Msg::kInfo) 
    << endl
    << "RawRecCounts done"
    << endl << endl;

}

---