testCONEXFile.cc

Go to the documentation of this file.

#include <io/CONEXFile.h>
#include <io/CONEXIOException.h>
#include <io/CONEXStructures.h>
#include <io/IoCodes.h>

#include <evt/Event.h>
#include <evt/ShowerSimData.h>

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

#include <fwk/CentralConfig.h>

#include <det/Detector.h>

#include <cppunit/extensions/HelperMacros.h>

#include <iostream>

#include <tst/Verify.h>

using io::CONEXFile;
using namespace fwk;
using namespace utl;
using namespace tst;
using tst::Expected;


class testCONEXFile : public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(testCONEXFile);
  CPPUNIT_TEST(testConstruct);
  CPPUNIT_TEST_EXCEPTION(testConstructException, io::CONEXIOException);
  CPPUNIT_TEST(testOpen);
  CPPUNIT_TEST_EXCEPTION(testOpenNonExisting, io::CONEXIOException);
  CPPUNIT_TEST_EXCEPTION(testOpenForWrite, io::CONEXIOException);
  CPPUNIT_TEST(testClose);
  CPPUNIT_TEST_EXCEPTION(testWrite, io::CONEXIOException);
  CPPUNIT_TEST(testFindEvent);
  CPPUNIT_TEST(testGotoPosition);
  CPPUNIT_TEST(testGetNEvents);
  CPPUNIT_TEST(testRead);
  CPPUNIT_TEST(testProfileRead);
  CPPUNIT_TEST_EXCEPTION(testGetNEventsClosedException, io::CONEXIOException);
  CPPUNIT_TEST_SUITE_END();

public:
  // shared data for tests

  CONEXFile* f;
  evt::Event event;

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

  void tearDown()
  { delete f; }

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

  void
  testConstructException()
  {
    Expected();
    CONEXFile f(TESTFILE, io::eWrite);
  }

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

  void
  testOpenNonExisting()
  {
    Expected();
    f->Open("NoSuchCONEXShowerFile.root");
  }

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

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

  void
  testFindEvent()
  {
    f->Open(TESTFILE);
    CPPUNIT_ASSERT(f->FindEvent(2) == io::eFail);
    CPPUNIT_ASSERT(f->FindEvent(0) != io::eFail);
  }

  void
  testGotoPosition()
  {
    f->Open(TESTFILE);
    CPPUNIT_ASSERT(f->GotoPosition(0) != io::eFail);
    CPPUNIT_ASSERT(f->GotoPosition(2) == io::eFail);
  }

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

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

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

  void
  testRead()
  {
    CentralConfig::GetInstance(BOOTSTRAPFILE);
    f->Open(TESTFILE);
    const int nEventsInFile = f->GetNEvents();
    int nEventsSeen = 0;

    event = evt::Event();      // clear event until readers do it.
    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(), 1.e19*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, 60.*deg));
      CPPUNIT_ASSERT(Verify<CloseTo>(azimuth, 0*deg));
      event = evt::Event();      // clear event until readers do it.
    }

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

  void
  testProfileRead()
  {
    //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();

      CPPUNIT_ASSERT(shower.HasLongitudinalProfile());
      CPPUNIT_ASSERT(shower.HasdEdX());
      //CPPUNIT_ASSERT(shower.HasDirection());
      //CPPUNIT_ASSERT(shower.HasPosition());

      //utl::TabulatedFunction& profile = shower.GetLongitudinalProfile();

      //CPPUNIT_ASSERT(nParticles > 1e11);
      //CPPUNIT_ASSERT(nParticles < 1e12);
    }
  }

};


CPPUNIT_TEST_SUITE_REGISTRATION(testCONEXFile);


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