DemoSnarlList.cxx

Go to the documentation of this file.

// $Id: DemoSnarlList.cxx,v 1.2 2009/07/21 22:10:17 gmieg Exp $
//
// Given a list of run/snarl numbers, this module passes only those
// records whose run/snarl number is in the list
//
// messier@huhepl.harvard.edu
#include "Demo/DemoSnarlList.h"
#include <algorithm>
#include <fstream>
#include "JobControl/JobCommand.h"
#include "JobControl/JobCModuleRegistry.h" // JOBMODULE registration macro
#include "MessageService/MsgService.h"
#include "MinosObjectMap/MomNavigator.h"
#include "RawData/RawRecord.h"
#include "RawData/RawDaqSnarlHeader.h"
#include "Util/UtilString.h"

CVSID("$Id: DemoSnarlList.cxx,v 1.2 2009/07/21 22:10:17 gmieg Exp $");
JOBMODULE(DemoSnarlList,
          "SnarlList","Filter records based on run/snarl numbers.");

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

DemoSnarlList::DemoSnarlList() : fIsSorted(false), fVetoMode(false) { }

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

JobCResult DemoSnarlList::Ana(const MomNavigator* mom) 
{
//======================================================================
// Are there records in mom whose run/snarl number matchs one in the
// list?
//======================================================================
  TObject* obj;
  for (int i=0; (obj=mom->At(i)); ++i) {
    const RawRecord* rr = dynamic_cast<RawRecord*>(obj);
    if (rr) {
      const RawDaqSnarlHeader* 
        rdsh = dynamic_cast<const RawDaqSnarlHeader*>(rr->GetRawHeader());
      if (rdsh) {
        int run   = rdsh->GetRun();
        int snarl = rdsh->GetSnarl();
        std::string k(this->Key(run,snarl));
        if (fIsSorted==false) { // Make sure we're sorted prior to search...
          fSnarlList.sort();
          fIsSorted = true;
        }
        if (std::binary_search(fSnarlList.begin(),fSnarlList.end(),k)) {
          if (fVetoMode) {

            int nrun, nsnarl;
            MsgService::Instance()->GetCurrentRunSnarl(nrun, nsnarl);
            MSG("Demo",Msg::kWarning)
                                << "DemoSnarlList reject at nsnarl = "
                                <<  nsnarl << endl;

            return JobCResult::kFailed;
          }
          else return JobCResult::kPassed;
        }
      }
    }
  }
  if (fVetoMode) return JobCResult::kPassed;
  else {

    int nrun, nsnarl;
    MsgService::Instance()->GetCurrentRunSnarl(nrun, nsnarl);
    MSG("Demo",Msg::kWarning) << "DemoSnarlList reject at nsnarl = "
                              <<  nsnarl << endl;
    return JobCResult::kFailed;

  }
}

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

void DemoSnarlList::Report() 
{
//======================================================================
// Print the snarl numbers in the list
//======================================================================
  std::list<std::string>::iterator itr(fSnarlList.begin());
  std::list<std::string>::iterator itrEnd(fSnarlList.end());
  MSG("Demo",Msg::kInfo) << "Snarls in list:\n";
  for (; itr!=itrEnd; ++itr) {
    MSG("Demo",Msg::kInfo) << (*itr) << "\n";
  }
}

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

void DemoSnarlList::HandleCommand(JobCommand *cmd) 
{
  std::string c = cmd->PopCmd();
  if (!(c=="Add"||c=="Remove"||c=="VetoMode")) {
    MSG("Demo",Msg::kWarning) << 
      "Only: Add [run#] [snarl#] [filename] and \n" <<
      "      Remove [run#] [snarl#] [filename] and \n" <<
      "      VetoMode <on|off> are supported.\n";
    return;
  }

  int run   = -1;
  int snarl = -1;
  while (cmd->HaveOpt()) {
    std::string opt = cmd->PopOpt();
    if (UtilString::IsInt(opt.c_str())) {
      if (run!=-1) {
        snarl = atoi(opt.c_str());
        if (c=="Add") {
          this->AddRunSnarl(run,snarl);
        }
        else if (c=="Remove") {
          this->RemoveRunSnarl(run,snarl);
        }
        run   = -1;
        snarl = -1;
      }
      else {
        run = atoi(opt.c_str());
      }
    }
    else if (c=="VetoMode"&&opt.c_str()!="off") {
      fVetoMode = true;
    }
    else { // Assume all strings are file names
      if (c=="Add") {
        this->AddUsingFile(opt.c_str());
      }
      else if (c=="Remove") {
        this->RemoveUsingFile(opt.c_str());
      }
    }
  }
}

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

void DemoSnarlList::AddRunSnarl(int r, int s) 
{
//======================================================================
// Add the run/snarl (r,s) to the list of records to pass
//======================================================================
  std::string k(this->Key(r,s));
  fSnarlList.push_back(k);
  fIsSorted = false;
}

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

void DemoSnarlList::RemoveRunSnarl(int r, int s) 
{
//======================================================================
// Remove the run/snarl (r,s) to the list of records to pass
//======================================================================
  std::string k(this->Key(r,s));
  std::list<std::string>::iterator itr;
  itr = std::find(fSnarlList.begin(), fSnarlList.end(), k);
  if (itr!=fSnarlList.end()) {
    fSnarlList.erase(itr);
  }
}

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

void DemoSnarlList::AddUsingFile(const char* f) 
{
//======================================================================
// Add all the run/snarls from the file f to the list. Assumes format
// is run<space>snarl
//======================================================================
  std::ifstream ifs(f);
  int run, snarl;
  while (ifs) { 
    ifs >> run >> snarl;
    this->AddRunSnarl(run,snarl);
  }
}

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

void DemoSnarlList::RemoveUsingFile(const char* f) 
{
//======================================================================
// Remove all the run/snarls from the file f to the list. Assumes format
// is run<space>snarl
//======================================================================
  std::ifstream ifs(f);
  int run, snarl;
  while (ifs) { 
    ifs >> run >> snarl;
    this->RemoveRunSnarl(run,snarl);
  }
}

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

std::string DemoSnarlList::Key(int run, int snarl) 
{
//======================================================================
// Construct a sortable key using run and snarl numbers
//======================================================================
  char k[1024];
  sprintf(k,"%12d.%12.12d",run,snarl);
  return k;
}

---