testSenecaShowerFile.cc

Go to the documentation of this file.

#include <io/SenecaShowerFile.h>
#include <evt/Event.h>
#include <evt/ShowerSimData.h>
#include <utl/AugerException.h>

#include <string>
#include <fstream>
#include <iostream>

#include <io/IoCodes.h>
#include <evt/Event.h>
#include <evt/ShowerSimData.h>
#include <utl/AugerUnits.h>
#include <utl/Particle.h>
#include <utl/ShowerParticleIterator.h>
//#include <fwk/CoordinateSystemRegistry.h>
#include <fwk/CentralConfig.h>
#include <det/Detector.h>

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

#include <iostream>
#include <string>

using namespace fwk;
using namespace evt;
using namespace io;
using namespace utl;
using namespace std;
using namespace tst;
using tst::Expected;


class testSenecaShowerFile : public CppUnit::TestFixture {

private:
  CPPUNIT_TEST_SUITE(testSenecaShowerFile);
  CPPUNIT_TEST(testConstruct);
  CPPUNIT_TEST_EXCEPTION(testConstructException, utl::IOFailureException);
  CPPUNIT_TEST(testOpen);
  CPPUNIT_TEST_EXCEPTION(testOpenOpen, utl::IOFailureException);
  CPPUNIT_TEST_EXCEPTION(testOpenNonExisting, utl::IOFailureException);
  CPPUNIT_TEST_EXCEPTION(testOpenForWrite, utl::IOFailureException);
  CPPUNIT_TEST(testClose);
  CPPUNIT_TEST(testCloseClose);
  CPPUNIT_TEST_EXCEPTION(testWrite, utl::IOFailureException);
  CPPUNIT_TEST(testFindEvent);
  CPPUNIT_TEST(testGotoPosition);
  CPPUNIT_TEST(testGetNEvents);
  CPPUNIT_TEST(testRead);
  CPPUNIT_TEST(testParticleRead);
  CPPUNIT_TEST_EXCEPTION(testGetNEventsClosedException,
                         utl::IOFailureException);
  CPPUNIT_TEST_SUITE_END();

public:
  // shared data for tests

  SenecaShowerFile* f;
  evt::Event event;

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

  void
  tearDown()
  {
    delete f;
    f = 0;
  }

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

  void
  testConstructException()
  {
    Expected();
    SenecaShowerFile ftest(TESTFILE,io::eWrite);
  }

  void testOpen()
  { f->Open(TESTFILE); }

  void
  testOpenNonExisting()
  {
    Expected();
    f->Open("NoSuchSenecaShowerFile.ptcl");
  }

  void
  testOpenOpen()
  {
    f->Open(TESTFILE);
    Expected();
    f->Open(TESTFILE);
  }

  void
  testOpenForWrite()
  {
    Expected();
    f->Open(TESTFILE, io::eWrite);
  }

  void testClose()
  { f->Close(); }

  void
  testCloseClose()
  {
    f->Close();
    Expected();
    f->Close();
  }

  void
  testFindEvent()
  {
    f->Open(TESTFILE);
    CPPUNIT_ASSERT(Verify<Not<Equal> >(f->FindEvent(2), io::eFail));
    CPPUNIT_ASSERT(Verify<Equal>(f->FindEvent(0), io::eFail));
  }

  void
  testGotoPosition()
  {
    f->Open(TESTFILE);
    CPPUNIT_ASSERT(Verify<Not<Equal> >(f->GotoPosition(0), io::eFail));
  }

  void
  testWrite()
  {
    f->Open(TESTFILE);
    Expected();
    f->Write(event);
  }

  void
  testGetNEvents()
  {
    f->Open(TESTFILE);
    CPPUNIT_ASSERT(Verify<Equal>(f->GetNEvents(), 2));
  }

  void
  testGetNEventsClosedException()
  {
    Expected();
    f->GetNEvents();
  }

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

    f->Open(TESTFILE);
    int nEventsInFile = f->GetNEvents();
    int nEventsSeen = 0;
    while (f->Read(event) != io::eEOF) {
      ++nEventsSeen;

      evt::ShowerSimData& shower = event.GetSimShower();

      const CoordinateSystemPtr refCS = det::Detector::GetInstance().GetReferenceCoordinateSystem();
      shower.MakeGeometry(utl::Point(0,0,0,refCS));
 
      CPPUNIT_ASSERT(Verify<Equal>(shower.GetPrimaryParticle(),
                                 int(Particle::eProton)));
      CPPUNIT_ASSERT(Verify<CloseTo>(shower.GetEnergy(), 9.99995e19*eV));

      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, 32.4199*deg));
      CPPUNIT_ASSERT(Verify<CloseTo>(azimuth, 91.0196*deg - 180*deg)); // todo: for an expert to check if the azimuth angle definition for Seneca is indeed correct like this, and thus in the rest of Offline !
      CPPUNIT_ASSERT(Verify<Equal>(shower.GetShowerNumber(),
                                 nEventsSeen));
      CPPUNIT_ASSERT(Verify<Equal>(shower.GetShowerRunId(), std::string("1")));

      event = evt::Event(); // reset
    }

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

  void
  testParticleRead()
  {
    CentralConfig::GetInstance(BOOTSTRAPFILE);
    f->Open(TESTFILE);
    for (int i = 0; i < f->GetNEvents(); ++i) {
      event = evt::Event();  // clear event until readers do it.
      f->Read(event);
      evt::ShowerSimData& shower = event.GetSimShower();

      shower.MakeGeometry(
        Point(0,0,0, det::Detector::GetInstance().GetSiteCoordinateSystem())
      );

      long double nParticles = 0;
      for (ShowerParticleIterator p = shower.GroundParticlesBegin();
           p != shower.GroundParticlesEnd(); ++p) {
        nParticles += p->GetWeight();
      }
      CPPUNIT_ASSERT(Verify<Greater>(nParticles, 1e11L));
      CPPUNIT_ASSERT(Verify<Less>(nParticles, 1e12L));

      event = evt::Event(); // reset
    }
  }
};


CPPUNIT_TEST_SUITE_REGISTRATION(testSenecaShowerFile);


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