testEVAFile.cc

Go to the documentation of this file.

#include <io/EVAFile.h>
#include <io/EVAIOException.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::EVAFile;
using namespace fwk;
using namespace utl;
using namespace tst;

class testEVAFile : public CppUnit::TestFixture {
private:
  CPPUNIT_TEST_SUITE (testEVAFile);
  CPPUNIT_TEST (testConstruct);
  CPPUNIT_TEST (testGetNEvents);
  CPPUNIT_TEST (testRead);
  CPPUNIT_TEST_SUITE_END();
  
public:
  // shared data for tests
  EVAFile* f;
  evt::Event event;

  void setUp() {
    f = new EVAFile();
      event = evt::Event();
  }

  void tearDown() {
    delete f;
  }

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

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

    fwk::CentralConfig::GetInstance(BOOTSTRAPFILE);

    // Some numbers for checking the data
    const double start_times[6] = {1.0824, 0.8214, 1.0596, 0.1590, 0.9667, 0.8041};
    const double field_sample1[6][3] = {
        { 9.7513824668E-05,  1.6809870176E-05, -3.1268147325E-05}, // station1
        {-1.0457331050E-01, -2.1307854418E-02,  3.5102479773E-02}, // station2
        {-1.3048007407E+00, -2.9347374586E-01,  4.3270475183E-01}, // station3
        {-2.7832267828E+03, -5.0633607442E+02,  8.5908390450E+02}, // station4
        {-1.5242416513E-02, -2.8939296207E-03,  3.6363917902E-03}, // station5
        {-9.3780203407E+02, -2.1832715112E+02,  2.7429037668E+02}  // station6
    };
    const double field_sample2[6][3] = {
        { 1.8118094038E+02,  2.7474426401E+01, -7.2966177649E+01}, // station1
        { 4.2307544703E+02,  8.6342326262E+01, -1.0922502801E+02}, // station2
        { 4.5767736476E+02,  1.0886486729E+02, -9.2860815629E+01}, // station3
        { 4.0576623131E+02,  7.0267573388E+01, -1.2122706669E+02}, // station4
        { 4.8185477571E+02,  1.1248897768E+02, -8.8667336289E+01}, // station5
        { 4.5953447261E+02,  1.0833208790E+02, -8.7527963838E+01}  // station6   
    };
    const double field_sample3[6][3] = {
        { 3.4397880870E-03, -1.0811865284E-02,  1.1460809613E-02}, // station1
        { 5.1979536749E-03, -8.6519291944E-03,  6.9117284726E-03}, // station2
        { 7.0621774784E-03, -9.2785115705E-03,  7.5364749611E-03}, // station3
        { 0.0000000000E+00, -0.0000000000E+00,  0.0000000000E+00}, // station4
        { 1.1726113640E-02, -6.6024960407E-03,  8.2778160319E-03}, // station5
        { 6.3571980942E-06,  1.0960432053E-06, -2.0383258957E-06}  // station6
    };

    f->Open (TESTFILE);
    int num_events_in_file = f->GetNEvents();
    int num_pulses = 0;
    int num_events = 0;
    
    CPPUNIT_ASSERT (Verify<Equal>(num_events_in_file, 1));

    while (f->Read (event) != io::eEOF) {
      evt::ShowerSimData& shower = event.GetSimShower();
      
      const CoordinateSystemPtr refCS = det::Detector::GetInstance().GetReferenceCoordinateSystem();
      shower.MakeGeometry(utl::Point(0,0,0,refCS));
      
      CPPUNIT_ASSERT (Verify <Less> (num_events, 1));

      /* Primary particle not implemented yet. This is however something
      That needs to become part of the EVA Model.
      CPPUNIT_ASSERT (Verify <Equal> (shower.GetPrimaryParticle(),
                                   int(Particle::eProton)));
      */

      CPPUNIT_ASSERT (Verify <CloseTo> (shower.GetEnergy(), 1.0919*EeV));

      const utl::CoordinateSystemPtr localCS = shower.GetLocalCoordinateSystem();
      const double zenith = (-shower.GetDirection()).GetTheta(localCS);
      const double azimuth = (-shower.GetDirection()).GetPhi(localCS); 

      CPPUNIT_ASSERT (Verify <CloseTo> (zenith, 58.357683919*deg));
      CPPUNIT_ASSERT (Verify <CloseTo> (azimuth, -68.990120994*deg));
      CPPUNIT_ASSERT (Verify<Equal>(shower.GetShowerNumber(), num_events));

      // test if EVA contents are set up correctly
      CPPUNIT_ASSERT (Verify<Equal>(shower.HasRadioSimulation(), true));
      evt::RadioSimulation& radsim = shower.GetRadioSimulation();
      CPPUNIT_ASSERT (Verify<Equal>(radsim.GetNumPulses(), (long)6));
      CPPUNIT_ASSERT (Verify<Equal>(radsim.HasCorePosition(), true));
      
      while (true) {
        bool ok;
        const evt::SimRadioPulse& simpulse = radsim.GetNextSimRadioPulse(ok);
        if (not ok) break;
      
        CPPUNIT_ASSERT (Verify<Less> (num_pulses, 6));
        
        CPPUNIT_ASSERT (Verify<Equal>(simpulse.GetEfieldTimeSeries().GetSize(),
          (TraceV3D::SizeType)1047));
        CPPUNIT_ASSERT (Verify<Equal>(simpulse.GetEfieldTimeSeries().GetBinning(), 
          2.5*nanosecond));
        CPPUNIT_ASSERT (Verify<Equal> (simpulse.GetBinning(), 2.5 * nano * second));
        CPPUNIT_ASSERT (Verify<CloseTo>(simpulse.GetStartTime(),
            start_times[num_pulses] * micro * second));
        
        Vector3D sample = simpulse.GetEfieldTimeSeries()[1];
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[0],
            field_sample1[num_pulses][0] * micro * volt / meter));
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[1],
            field_sample1[num_pulses][1] * micro * volt / meter));
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[2],
            field_sample1[num_pulses][2] * micro * volt / meter));

        sample = simpulse.GetEfieldTimeSeries()[9];
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[0],
            field_sample2[num_pulses][0] * micro * volt / meter));
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[1],
            field_sample2[num_pulses][1] * micro * volt / meter));
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[2],
            field_sample2[num_pulses][2] * micro * volt / meter));

        sample = simpulse.GetEfieldTimeSeries()[199];
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[0],
            field_sample3[num_pulses][0] * micro * volt / meter));
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[1],
            field_sample3[num_pulses][1] * micro * volt / meter));
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[2],
            field_sample3[num_pulses][2] * micro * volt / meter));
        
        sample = simpulse.GetEfieldTimeSeries()[0];
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[0], 0 * micro * volt / meter));
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[1], 0 * micro * volt / meter));
        CPPUNIT_ASSERT (Verify<CloseTo>(sample[2], 0 * micro * volt / meter));

        ++num_pulses;
      }
      CPPUNIT_ASSERT (Verify<Equal>(num_pulses, 6));
      ++num_events;
    }
    CPPUNIT_ASSERT (Verify<Equal>(num_events, 1));
  }
};

CPPUNIT_TEST_SUITE_REGISTRATION (testEVAFile);

// Configure (x)emacs for this file ...
// Local Variables:
// mode:c++
// compile-command: "make -C .. -k testEVA && (cd ..; testEVA)"
// End: