SdBadStationRejector.cc

Go to the documentation of this file.

#include "SdBadStationRejector.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 <sevt/SEvent.h>
#include <sevt/StationConstants.h>

#include <sstream>

using namespace std;
using namespace fwk;
using namespace det;
using namespace utl;
using namespace evt;
using namespace sevt;
using namespace SdBadStationRejectorKG;


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

  fRejectMap.clear();

  ostringstream info;
  info << "registered stations:";
  for (auto b = topB.GetFirstChild(); b; b = b.GetNextSibling()) {
    const auto stationId = VManager::FindComponent<size_t>("id", b.GetAttributes());
    const auto reason = VManager::FindComponent<string>("reason", b.GetAttributes());

    const auto range = b.Get<vector<utl::TimeStamp>>();

    fRejectMap.insert(make_pair(range[0], make_pair(range[1], stationId)));

    info << ' ' << stationId << " [" << range[0] << ", " << range[1] << ')';
  }
  INFO(info);

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


VModule::ResultFlag
SdBadStationRejector::Run(Event& event)
{
  if (!event.HasSEvent())
    return eSuccess;

  auto& sEvent = event.GetSEvent();

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

  const auto badIds = fRejectMap.GetBadStations(time);

  vector<int> infoList;
  for (const auto& sId : badIds) {
    if (sEvent.HasStation(sId)) {
      infoList.push_back(sId);
      // other rejection status?
      sEvent.GetStation(sId).SetRejected(StationConstants::eOffGrid);
    }
  }

  if (!infoList.empty()) {
    ostringstream info;
    info << Join(" ", infoList) << "rejected.";
    INFO(info);
  }

  return eSuccess;
}


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

  return eSuccess;
}