G4StationSim/TankValidation.cc

Go to the documentation of this file.

#include <evt/Event.h>
#include <sevt/SEvent.h>
#include <sevt/SortCriteria.h>
#include <sevt/PMTSimData.h>

#include <utl/ErrorLogger.h>
#include <utl/Branch.h>
#include <utl/TimeDistribution.h>
#include <utl/Particle.h>
#include <tst/Validatrix.h>

#include <fwk/CentralConfig.h>

#include "TankValidation.h"

using namespace std;
using namespace evt;
using namespace sevt;
using namespace fwk;
using namespace utl;
using namespace tst::Validatrix;
using namespace TankValidationNS;


VModule::ResultFlag
TankValidation::Init()
{
  auto topB = CentralConfig::GetInstance()->GetTopBranch("TankValidation");
  topB.GetChild("RefFilename").GetData(fRefFile);

  return eSuccess;
}


VModule::ResultFlag
TankValidation::Run(evt::Event& event)
{
  if (!event.HasSEvent()) {
    ERROR("Event should have SEvent.");
    return eFailure;
  }

  const auto newRefFile = fRefFile + string(".new");
  {
    ofstream outFile(newRefFile);

    event.GetSEvent().SortStations(ByIncreasingId());
    const auto& sEvent = event.GetSEvent();

    for (const auto& s : sEvent.StationsRange()) {

      for (const auto& pmt : s.PMTsRange()) {

        if (!pmt.HasSimData())
          continue;

        const auto& pmtSim = pmt.GetSimData();

        for (const auto& pe : pmtSim.PETimeDistributionsRange()) {

          INFO(fRefFile);
          ostringstream label;
          label << "Station_" << s.GetId() << "_PMT_" << pmt.GetId() << "_label_" << pe.GetLabel();
          outFile << BeLabel(label.str()) << BeEqual(pe.GetTimeDistribution());

        }
      }
    }
  }

  ifstream newFile(newRefFile);
  ifstream oldFile(fRefFile);

  return Compare(oldFile, newFile, /*failOnFirst=*/true) ? eSuccess : eFailure;
}


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