testZHAireSFile.cc

Go to the documentation of this file.

#include <io/ZHAireSFile.h>
#include <io/ZHAireSIOException.h>
#include <io/IoCodes.h>

#include <evt/Event.h>
#include <evt/ShowerSimData.h>
#include <evt/VGaisserHillasParameter.h>
#include <evt/RadioSimulation.h>
#include <evt/SimRadioPulse.h>

#include <utl/AugerUnits.h>
#include <utl/Particle.h>

#include <fwk/CentralConfig.h>

#include <det/Detector.h>

#include <tst/Verify.h>
#include <cppunit/extensions/HelperMacros.h>

#include <iostream>

using io::ZHAireSFile;
using namespace fwk;
using namespace utl;
using namespace tst;
using namespace std;


class testZHAireSFile : public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(testZHAireSFile);
  CPPUNIT_TEST(testConstruct);
  CPPUNIT_TEST(testGetNEvents);
  CPPUNIT_TEST(testRead);
  CPPUNIT_TEST_SUITE_END();

public:
  // shared data for tests

  ZHAireSFile* fFile = nullptr;
  evt::Event fEvent;

  void
  setUp()
  {
    fFile = new ZHAireSFile();
    fEvent = evt::Event();
  }

  void
  tearDown() {
    delete fFile;
    fFile = nullptr;
  }

  void
  testConstruct()
  {
    ZHAireSFile* file;
    CPPUNIT_ASSERT(file = new ZHAireSFile(TESTFILE));
    delete file;
  }

  void
  testGetNEvents()
  {
    fFile->Open(TESTFILE);
    CPPUNIT_ASSERT(fFile->GetNEvents() == 1);
  }

  void
  testRead()
  {
    fwk::CentralConfig::GetInstance(BOOTSTRAPFILE);

    fFile->Open(TESTFILE);
    int nEventsInFile = fFile->GetNEvents();
    int nEventsSeen = 0;
    fEvent = evt::Event();  // clear event until readers do it.
    while (fFile->Read(fEvent) != io::eEOF) {
      ++nEventsSeen;
      evt::ShowerSimData& shower = fEvent.GetSimShower();

      const CoordinateSystemPtr refCS = det::Detector::GetInstance().GetReferenceCoordinateSystem();
      shower.MakeGeometry(utl::Point(0,0,0, refCS));

      // test if AIRES variables are set up correctly
      CPPUNIT_ASSERT(Verify<Equal>(shower.GetPrimaryParticle(), int(Particle::eProton)));
      CPPUNIT_ASSERT(Verify<CloseTo>(shower.GetEnergy(), 7.4411e17*eV));

      const utl::CoordinateSystemPtr localCS = shower.GetLocalCoordinateSystem();
      const double zenith = (-shower.GetDirection()).GetTheta(localCS);
      CPPUNIT_ASSERT(Verify<CloseTo>(zenith, 51.73*deg));
      const double azimuth = (-shower.GetDirection()).GetPhi(localCS);
      const double airesAzimuthDeg = -105.23;  // from the .sry file
      // 90deg for North -> East, -4.23deg for magnetic-North:North misalignement
      CPPUNIT_ASSERT(Verify<CloseTo>(azimuth, (airesAzimuthDeg + 90 - 4.23)*deg));

      CPPUNIT_ASSERT(Verify<Equal>(shower.GetShowerNumber(), nEventsSeen));
      CPPUNIT_ASSERT(Verify<Equal>(shower.GetShowerRunId(), std::string("1")));
      CPPUNIT_ASSERT(Verify<Equal>(shower.HasGroundParticles(), false)); // no particle file
      CPPUNIT_ASSERT(Verify<Equal>(shower.HasGHParameters(), false));

      // test if ZHAireS contents are set up correctly
      CPPUNIT_ASSERT(Verify<Equal>(shower.HasRadioSimulation(), true));
      evt::RadioSimulation& radsim = shower.GetRadioSimulation();
      CPPUNIT_ASSERT(Verify<Equal>(radsim.GetNumPulses(), 7L));
      CPPUNIT_ASSERT(Verify<Equal>(radsim.HasCorePosition(), true));
      bool ok = false;
      const evt::SimRadioPulse& simpulse = radsim.GetNextSimRadioPulse(ok);
      const unsigned int frontpadding = 0;  // samples used for padding of the timeseries front
      const double binning = 0.5*nanosecond;
      CPPUNIT_ASSERT(Verify<Equal>(ok, true));
      CPPUNIT_ASSERT(Verify<Equal>(simpulse.GetEfieldTimeSeries().GetSize(), TraceV3D::SizeType(22254+frontpadding)));
      CPPUNIT_ASSERT(Verify<CloseTo>(simpulse.GetEfieldTimeSeries().GetBinning(), binning));
      CPPUNIT_ASSERT(Verify<CloseTo>(simpulse.GetBinning(), binning));
      CPPUNIT_ASSERT(Verify<CloseTo>(simpulse.GetStartTime(), -1126.75*nanosecond - frontpadding*binning));

      const unsigned int samplenumber = 3575;  // test this bin (starts at 0)
      const Vector3D& sample = simpulse.GetEfieldTimeSeries()[samplenumber + frontpadding];
      CPPUNIT_ASSERT(Verify<CloseTo>(sample[0]/volt*meter, -2.85945943595946640e-03));  // east component
      CPPUNIT_ASSERT(Verify<CloseTo>(sample[1]/volt*meter, -2.03602779956765009e-03));  // north component
      CPPUNIT_ASSERT(Verify<CloseTo>(sample[2]/volt*meter, 2.61886417865753174e-03));  // vertical up component )
      fEvent = evt::Event();  // clear event until readers do it.
    }

    CPPUNIT_ASSERT(Verify<Equal>(nEventsInFile, nEventsSeen));
  }

};


CPPUNIT_TEST_SUITE_REGISTRATION (testZHAireSFile);