RdEventAndStationSelector.cc

Go to the documentation of this file.

// version $Id: RdEventAndStationSelector.cc 32946 2019-09-19 15:35:43Z gottowik $

#include "RdEventAndStationSelector.h"
#include <fwk/CentralConfig.h>
#include <det/VManager.h>
#include <det/Detector.h>
#include <rdet/RDetector.h>

#include <utl/Branch.h>
#include <utl/ErrorLogger.h>
#include <utl/String.h>
#include <utl/RadioGeometryUtilities.h>
#include <utl/Vector.h>
#include <utl/Point.h>

#include <evt/ShowerRRecData.h>
#include <evt/ShowerSRecData.h>
#include <evt/ShowerFRecData.h>
#include <evt/ShowerRecData.h>
#include <evt/ShowerSimData.h>

#include <evt/Event.h>
#include <revt/REvent.h>
#include <revt/Header.h>
#include <revt/StationConstants.h>
#include <revt/StationRecData.h>

#include <sstream>
#include <algorithm>
#include <vector>

using namespace std;
using namespace fwk;


namespace RdEventAndStationSelector {

  fwk::VModule::ResultFlag
  RdEventAndStationSelector::Init()
  {
    utl::Branch topB = CentralConfig::GetInstance()->GetTopBranch("RdEventAndStationSelector");
    topB.GetChild("InfoLevel").GetData(fInfoLevel);
    topB.GetChild("filename").GetData(fFilename);

    fstream fin;
    const int MAX_CHARS_PER_LINE = 512;
    char line[MAX_CHARS_PER_LINE];
    fin.open(fFilename.c_str(), ios::in);  // opening data
    //fin.getline(line, MAX_CHARS_PER_LINE); // read/skip first line
    stringstream sstr;
    sstr.str("");
    sstr << "loop through file... filename = " << fFilename;
    INFODebug(sstr.str());

    if (!fin.good()) {
      WARNING("File could not be opened");
      return eFailure;
    }
    while (fin.good()) {
      stringstream output;
      fin.getline(line, MAX_CHARS_PER_LINE);
      stringstream tmp_sstr;
      tmp_sstr << line;
      std::string segment;
      if (!std::getline(tmp_sstr, segment, '\t')) {
        break;
      }
      int runNumber = atoi(segment.c_str());
      std::getline(tmp_sstr, segment, '\t');
      int eventId = atoi(segment.c_str());
      std::vector<int> tmp_station_ids;
      output << "Event " << runNumber << "_" << eventId << ": ";
      while (std::getline(tmp_sstr, segment, '\t')) {
        tmp_station_ids.push_back(atoi(segment.c_str()));
        output << atoi(segment.c_str()) << " ";
      }
      std::pair<int, int> event_identifier(runNumber, eventId);
      fEventSelection[event_identifier] = tmp_station_ids;

      INFOIntermediate(output.str());
    }
    fin.close();

    return eSuccess;
  }


  fwk::VModule::ResultFlag
  RdEventAndStationSelector::Run(evt::Event& event)
  {
    stringstream sstr;
    INFODebug(sstr.str());
    if (!event.HasREvent()) {
      WARNING("No REvent");
      return eBreakLoop;
    }
    revt::REvent& rEvent = event.GetREvent();
    int runNumber = rEvent.GetHeader().GetRunNumber();
    int eventId = rEvent.GetHeader().GetId();
    std::pair<int, int> event_identifier(runNumber, eventId);

    if (fEventSelection.find(event_identifier) == fEventSelection.end()) {
      sstr.str("");
      sstr << "Event " << event_identifier.first << "_" << event_identifier.second
           << " not present in selection, event will be skipped";
      INFOIntermediate(sstr.str());
      return eContinueLoop;
    }

    for (revt::REvent::StationIterator sIt = rEvent.StationsBegin(); sIt != rEvent.StationsEnd(); ++sIt) {
      const int stationId = sIt->GetId();
      std::vector<int> stationList = fEventSelection[event_identifier];
      std::vector<int>::iterator it = std::find(stationList.begin(), stationList.end(), stationId);
      if (it == stationList.end()) {
        sIt->SetRejected(revt::eManuallyRejected);
        sstr.str("");
        sstr << "station " << sIt->GetId() << " will be rejected";
        INFOIntermediate(sstr.str());
      }
    }

    return eSuccess;
  }


  VModule::ResultFlag
  RdEventAndStationSelector::Finish()
  {
    return eSuccess;
  }

}