EventFileReader.cc

Go to the documentation of this file.

#include "EventFileReader.h"

#include <fwk/CentralConfig.h>
#include <fwk/RunController.h>

#include <det/Detector.h>
#include <sdet/SDetector.h>

#include <io/EventFileChain.h>
#include <io/OfflineROOTFile.h>

#include <evt/Header.h>
#include <evt/Event.h>

#include <sevt/SEvent.h>
#include <sevt/Header.h>
#include <sevt/EventTrigger.h>

#include <fevt/FEvent.h>
#include <fevt/Eye.h>
#include <fevt/EyeHeader.h>

#include <utl/ErrorLogger.h>
#include <utl/Reader.h>
#include <utl/TimeStamp.h>
#include <utl/config.h>

#include <revt/REvent.h>
#include <revt/Header.h>

#include <io/CDASToOfflineEventConverter.h>
#include <AugerEvent.h>
#include <IoSdData.h>

#include <MdEvent.h>
#include <mevt/MEvent.h>
#include <mevt/Header.h>
#include <mdet/MDetector.h>

#include <TStopwatch.h>

#include <fstream>

using namespace std;
using namespace utl;
using sevt::SEvent;
using namespace io;
using sdet::SDetector;
using namespace fwk;
using namespace EventFileReaderOG;


// DV: we want this in the data segment to get the initialization
// at the library load time
namespace EventFileReaderOG {
  TStopwatch gStopwatch;
}


EventFileReader::~EventFileReader()
{
  delete fEventFiles;
}


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

  fReportFilenames = bool(topBranch.GetChild("ReportFilenames"));
  fIgnoreExpeptions = bool(topBranch.GetChild("IgnoreExceptions"));

  fBranch = topBranch.GetFirstChild();
  if (fReportFilenames) // need to exclude this particular flag
    fBranch = fBranch.GetNextSibling();
  if (fIgnoreExpeptions) // need to exclude this particular flag
    fBranch = fBranch.GetNextSibling();

  fEventFiles = nullptr;
  fEventsRead = 0;
  fCurrentFilename = "n/a"; // to trigger output of first file even if it is ""

  return eSuccess;
}


VModule::ResultFlag
EventFileReader::OpenFiles()
{
  if (!fBranch) {
    ostringstream info;
    info << "No more data to read. ";
    if (fReportFilenames) {
      info << "Last file: '" << fCurrentFilename << "'";
    }
    info << '.';
    INFO(info);
    return eBreakLoop;
  }

  //Branch* nB = new Branch(fBranch);
  fEventFiles = new EventFileChain(fBranch);
  fEventFiles->SetIgnoreExceptions(fIgnoreExpeptions);
  fBranch = fBranch.GetNextSibling(); // move to next sibling in input file
  return eSuccess;
}


VModule::ResultFlag
EventFileReader::Run(evt::Event& event)
{
  if (!fEventFiles) {
    VModule::ResultFlag result = OpenFiles();
    if (result != eSuccess)
      return result;
  }

  const io::Status status = fEventFiles->Read(event);

  if (status == io::eEOF || status == io::eEndOfFiles) {

    delete fEventFiles;
    fEventFiles = nullptr;
    return eBreakLoop;

  } else if (status == io::eFail) {

    INFO("Failed to read event.");
    return eFailure;

  } else if (status == io::eSkipEvent) {

    INFO("Failed to read event, will skip it");
    return eContinueLoop;

  }

  ++fEventsRead;
  ++RunController::GetInstance().GetRunData().GetNamedCounters()["EventFileReader/TotalEvents"];

  const string& file = fEventFiles->GetCurrentFilename();

  if (fCurrentFilename != file) {

    if (fReportFilenames) {
      ostringstream info;
      info << "Switching to file '" << file << "'.";
      INFO(info);
    }
    fCurrentFilename = file;
    RunController::GetInstance().GetRunData().SetCurrentFilename(file);
    ++RunController::GetInstance().GetRunData().GetNamedCounters()["EventFileReader/Files"];

  }

  const evt::Header& header = event.GetHeader();

  io::Type fileType = fEventFiles->GetFileType();
  if (fileType != eCONEXRandom &&
      fileType != eCONEX &&
      fileType != eCorsika &&
      fileType != eAires &&
      fileType != eSeneca &&
      fileType != eRadioAERA && // for eRadioAERA it is already done in the reader
      fileType != eRadioAERAroot &&
      fileType != eREAS &&
      fileType != eZHAireS &&
      fileType != eEVA &&
      fileType != eSELFAS) {
    // This configures the detector.
    // For simulations the time is defined by the EventGenerator,
    // which is why we exclude those cases here.
    det::Detector::GetInstance().Update(header.GetTime());
  }

  // copy in the sd part of the raw event
  if (event.HasRawEvent() && event.GetRawEvent().HasSd()) {
    event << event.GetRawEvent().Sd();
    ++RunController::GetInstance().GetRunData().GetNamedCounters()["EventFileReader/SdEvents"];
  }

  if (event.HasRawEvent() && event.GetRawEvent().HasMd()) {
    event << event.GetRawEvent().Md();
    ++RunController::GetInstance().GetRunData().GetNamedCounters()["EventFileReader/MdEvents"];
  }

  ostringstream info;
  info << "Event id = '" << header.GetId() << '\'';
  INFO(info);

  return eSuccess;
}


VModule::ResultFlag
EventFileReader::Finish()
{
  ostringstream info;
  info << "Read " << fEventsRead << " events";

  gStopwatch.Stop();
  const double realTime = gStopwatch.RealTime();
  const double cpuTime = gStopwatch.CpuTime();

  info << " in " << realTime << " sec, " << (fEventsRead/realTime) << " ev/sec, with "
       << (cpuTime/realTime*100) << "% cpu usage.";
  INFO(info);

  delete fEventFiles;
  fEventFiles = nullptr;

  return eSuccess;
}


const string&
EventFileReader::GetCurrentFilename()
  const
{
  return fEventFiles->GetCurrentFilename();
}