SdHasDataRec/Validation.cc

Go to the documentation of this file.

#include <evt/Event.h>
#include <evt/ShowerRecData.h>
#include <evt/ShowerSRecData.h>

#include <sevt/SEvent.h>
#include <sevt/SortCriteria.h>

#include <utl/ErrorLogger.h>
#include <utl/Branch.h>
#include <utl/AugerUnits.h>

#include <tst/Validatrix.h>

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

#include "Validation.h"

using namespace std;
using namespace evt;
using namespace sevt;
using namespace fevt;
using namespace fwk;
using namespace utl;
using namespace tst::Validatrix;
using namespace ValidationNS;


VModule::ResultFlag
Validation::Init()
{
  Branch topB = CentralConfig::GetInstance()->GetTopBranch("Validation");
  topB.GetChild("RefFilename").GetData(fRefFileName);

  fResults = new ofstream((fRefFileName + string(".new")).c_str());

  return eSuccess;
}


VModule::ResultFlag
Validation::Run(evt::Event& event)
{
  // SD
  if (!event.HasSEvent()) {
    ERROR("Event should contain an SEvent.");
    return eFailure;
  }

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

  const double tolerance = 0.02;

  *fResults << BeLabel("SD_rec");

  for (const auto& station : sEvent.StationsRange()) {
    if (station.HasRecData()) {
      ostringstream label;
      label << "Station_" << station.GetId();
      *fResults << BeLabel(label.str());
      *fResults << BeCloseRel(station.GetRecData().GetTotalSignal(), tolerance);
    }
  }

  *fResults << BeLabel("Shower_rec");

  // Shower-level rec data
  if (!event.HasRecShower())
    ERROR("Event does not contain a RecShower");
  const auto& showerRec = event.GetRecShower();

  // SD rec (shower level)

  if (!showerRec.HasSRecShower())
    ERROR("RecShower does not have SRecShower");

  *fResults << BeLabel("Shower_S_rec");

  const auto& showerSRec = showerRec.GetSRecShower();

  const auto& corePos = showerSRec.GetCorePosition();
  const auto& baryPos = showerSRec.GetBarycenter();
  const auto& coreCS = LocalCoordinateSystem::Create(corePos);
  const auto& baryCS = LocalCoordinateSystem::Create(baryPos);

#define SCOMPARE(item, eps) \
  *fResults << BeLabel(#item); \
  *fResults << BeCloseRel(showerSRec.Get##item(), eps)

  SCOMPARE(ShowerSize, tolerance);
  SCOMPARE(Curvature, tolerance);
  SCOMPARE(ShowerSizeError, tolerance);
  SCOMPARE(CurvatureError, tolerance);

  SCOMPARE(LDFChi2, tolerance);
  *fResults << BeLabel("LDFNdof");
  *fResults << BeEqual(showerSRec.GetLDFNdof());

  *fResults << BeLabel("Theta");
  *fResults << BeCloseRel(showerSRec.GetAxis().GetTheta(coreCS), tolerance);
  *fResults << BeLabel("Phi");
  *fResults << BeCloseRel(showerSRec.GetAxis().GetPhi(coreCS), tolerance);
  SCOMPARE(ThetaError, tolerance);
  SCOMPARE(PhiError, tolerance);

  *fResults << BeLabel("CoreX");
  *fResults << BeCloseRel(corePos.GetX(baryCS), tolerance);
  *fResults << BeLabel("CoreY");
  *fResults << BeCloseRel(corePos.GetY(baryCS), tolerance);
  *fResults << BeLabel("CoreXError");
  *fResults << BeCloseRel(showerSRec.GetCoreError().GetX(baryCS), tolerance);
  *fResults << BeLabel("CoreYError");
  *fResults << BeCloseRel(showerSRec.GetCoreError().GetY(baryCS), tolerance);
#undef SCOMPARE

  return eSuccess;
}


VModule::ResultFlag
Validation::Finish()
{
  delete fResults;
  fResults = nullptr;

  ifstream newFile((fRefFileName + string(".new")).c_str());
  ifstream oldFile(fRefFileName.c_str());

  const bool comp = Compare(oldFile, newFile, /*failOnFirst=*/true);

  return comp ? eSuccess : eFailure;
}