MdModuleRejector.cc

Go to the documentation of this file.

#include "MdModuleRejector.h"
#include <fwk/CentralConfig.h>
#include <det/VManager.h>
#include <utl/Branch.h>
#include <utl/ErrorLogger.h>
#include <utl/String.h>
#include <evt/Event.h>
#include <mevt/MEvent.h>
#include <mevt/Module.h>

#include <sstream>

using namespace fwk;
using namespace det;
using namespace utl;
using namespace evt;
using namespace mevt;
using namespace MdModuleRejectorAG;


VModule::ResultFlag
MdModuleRejector::Init()
{
  Branch topB =
    CentralConfig::GetInstance()->GetTopBranch("MdModuleRejector");

  fRejectMap.clear();

  std::ostringstream msg;
  for (Branch b = topB.GetFirstChild(); b;
       b = b.GetNextSibling())
  {
    const size_t counterId = VManager::FindComponent<size_t>("counterId", b.GetAttributes());
    const size_t moduleId = VManager::FindComponent<size_t>("moduleId", b.GetAttributes());
    const std::string reason = VManager::FindComponent<std::string>("reason", b.GetAttributes());

    std::vector<utl::TimeStamp> range;
    b.GetData(range);

    fRejectMap.insert( std::make_pair( range[0], boost::make_tuple(range[1], counterId, moduleId) ) );

    msg << "\n  registered: counter = " << counterId << ", module = " << moduleId;
    msg << " [" << range[0] << ", " << range[1] << ") due to " << reason;
  }
  INFO(msg);

//  return fRejectMap.empty() ? eFailure : eSuccess;
  return eSuccess;
}


VModule::ResultFlag
MdModuleRejector::Run(Event& event)
{
  std::stringstream msg (std::ostringstream::out);


  if (!event.HasMEvent())
    return eSuccess;

  MEvent& mEvent = event.GetMEvent();

  const TimeStamp& time = det::Detector::GetInstance().GetTime();

  TimeRangeMap::IdCollection badIds = fRejectMap.GetBadStations(time);

  std::vector<std::string> rejectedIds;
  for (TimeRangeMap::IdCollection::const_iterator it = badIds.begin();
       it != badIds.end(); ++it)
  {
    size_t counterId = (*it).first;

    if (mEvent.HasCounter(counterId))
    {
      mevt::Counter& counter = mEvent.GetCounter(counterId);

      size_t moduleId = (*it).second;

      if (moduleId==0) {   // reject the counter
        counter.SetRejected("Rejected by MdModuleRejector");
        std::ostringstream counterModule("");
        counterModule << counterId << "-all";
        rejectedIds.push_back(counterModule.str());
      } else if (counter.HasModule(moduleId)) {  // reject one module
        counter.GetModule(moduleId).SetRejected("Rejected by MdModuleRejector");   // TO-DO put the reason defined in the xml file
        std::ostringstream counterModule("");
        counterModule << counterId << "-" << moduleId;
        rejectedIds.push_back(counterModule.str());
      }

    } 

  }  // end IdCollection loop

  // if there are rejected modules output info message
  if  (rejectedIds.size()!=0) {
    msg.str("");
    msg << "rejected counter-module: ";    
    for(std::vector<std::string>::iterator it = rejectedIds.begin(); it!=rejectedIds.end(); ++it) {
      msg << *it << ", ";    
    }
    INFO(msg.str());
  }

  return eSuccess;
}

VModule::ResultFlag
MdModuleRejector::Finish()
{
  TimeRangeMap().swap(fRejectMap);

  return eSuccess;
}

// Configure (x)emacs for this file ...
// Local Variables:
// mode: c++
// End: