G4TankSim/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"

#include <boost/range/iterator_range.hpp>

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()
{
  const Branch 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 string newRefFile = fRefFile + string(".new");
  {
    ofstream outFile(newRefFile);

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

    for (const Station& s : boost::make_iterator_range(sEvent.StationsBegin(), sEvent.StationsEnd())) {

      for (const PMT& pmt : boost::make_iterator_range(s.PMTsBegin(), s.PMTsEnd())) {

        if (!pmt.HasSimData())
          continue;

        const PMTSimData& pmtSim = pmt.GetSimData();

        for (const auto& pe :
             boost::make_iterator_range(pmtSim.PETimeDistributionsBegin(), pmtSim.PETimeDistributionsEnd())) {

          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;
}