SenecaShowerFileParticleIterator.cc

Go to the documentation of this file.

#include "SenecaShowerFileParticleIterator.h"

#include <iostream>
#include <utl/PhysicalConstants.h>


using namespace std;
using namespace utl;
using namespace io;


SenecaShowerFileParticleIterator::SenecaShowerFileParticleIterator(SenecaFileParser* const senecaParser) :
  fCurrentParticle(utl::Particle::eUndefined,
                   utl::Particle::eShower,
                   Point(0,0,0, CoordinateSystem::GetRootCoordinateSystem()),
                   Vector(0,0,0, CoordinateSystem::GetRootCoordinateSystem()),
                   TimeInterval(0), 0)
{
  fSenecaFile = senecaParser;

  const Status check = fSenecaFile->FileStatus();

  if (check != eSuccess) {
    fCurrentPosition = fOriginalPosition = 0;
    fIteratorValid = false;
  } else {
    fCurrentPosition = fOriginalPosition = fSenecaFile->GetPosition();
    fIteratorValid = true;
  }
}


SenecaShowerFileParticleIterator::SenecaShowerFileParticleIterator() :
  fCurrentParticle(utl::Particle::eUndefined,
                   utl::Particle::eShower,
                   Point(0,0,0, CoordinateSystem::GetRootCoordinateSystem()),
                   Vector(0,0,0, CoordinateSystem::GetRootCoordinateSystem()),
                   TimeInterval(0), 0)
{
}


SenecaShowerFileParticleIterator::~SenecaShowerFileParticleIterator()
{
}


Particle*
SenecaShowerFileParticleIterator::GetOneParticle(const utl::CoordinateSystemPtr& groundCS)
{
  const Status check = fSenecaFile->ReadParticle();

  if (check != eSuccess)
    return NULL;

  do
    fCurrentSenecaParticle = fSenecaFile->GetParticleRecord();
  while (!fCurrentSenecaParticle);

  fCurrentPosition = fSenecaFile->GetPosition();

  return SenecaToDPAParticle(groundCS);
}


void
SenecaShowerFileParticleIterator::Rewind()
{
  fSenecaFile->GotoPosition(fOriginalPosition);
  fCurrentPosition = fSenecaFile->GetPosition();
}


utl::Particle*
io::SenecaShowerFileParticleIterator::SenecaToDPAParticle(const utl::CoordinateSystemPtr& groundCS)
{
  if (fCurrentSenecaParticle == NULL)
    return NULL;

  const int particleID = SenecaToDPAParticle(fCurrentSenecaParticle->ID);

  if (particleID == utl::Particle::eUndefined)
    return NULL;

  fCurrentParticle =
    Particle(particleID, Particle::eShower,
             Point(fCurrentSenecaParticle->X * m,
                   fCurrentSenecaParticle->Y * m,
                   0.0,
                   groundCS),
             Vector(fCurrentSenecaParticle->Px * GeV,
                    fCurrentSenecaParticle->Py * GeV,
                    -fCurrentSenecaParticle->Pz * GeV,
                    groundCS),
             TimeInterval(fCurrentSenecaParticle->T / .299 * ns), fCurrentSenecaParticle->Weight);

  return &fCurrentParticle;
}


int
SenecaShowerFileParticleIterator::SenecaToDPAParticle(const int senecaParticleID)
{
  switch (senecaParticleID) {
  case 11:
    return utl::Particle::eNuElectron;
  case -11:
    return utl::Particle::eAntiNuElectron;
  case 15:
    return utl::Particle::eNuTau;
  case -15:
    return utl::Particle::eAntiNuTau;
  case 13:
    return utl::Particle::eNuMuon;
  case -13:
    return utl::Particle::eAntiNuMuon;
  case 10:
    return utl::Particle::ePhoton;
  case 12:
    return utl::Particle::eElectron;
  case -12:
    return utl::Particle::ePositron;
  case 14:
    return utl::Particle::eMuon;
  case -14:
    return utl::Particle::eAntiMuon;
  case 110:
    return utl::Particle::ePiZero;
  case 120:
    return utl::Particle::ePiPlus;
  case -120:
    return utl::Particle::ePiMinus;
  case 220:
    return utl::Particle::eEta;
  case 130:
    return utl::Particle::eKaonPlus;
  case -130:
    return utl::Particle::eKaonMinus;
  case 230:
    return utl::Particle::eUndefined;
  case -230:
    return utl::Particle::eUndefined;
  case 330:
    return utl::Particle::eUndefined;
  case 1120:
    return utl::Particle::eProton;
  case -1120:
    return utl::Particle::eAntiProton;
  case 1220:
    return utl::Particle::eNeutron;
  case -1220:
    return utl::Particle::eAntiNeutron;
  case 1130:
    return utl::Particle::eUndefined;
  case 1230:
    return utl::Particle::eUndefined;
  case 2130:
    return utl::Particle::eLambda;
  case -2130:
    return utl::Particle::eAntiLambda;
  case -20:
    return utl::Particle::eKaon0L;
  case 20:
    return utl::Particle::eKaon0S;
  default:
    return utl::Particle::eUndefined;
  }
}


Status
SenecaShowerFileParticleIterator::SetSenecaFileParser(SenecaFileParser* const senecaParser)
{
  fSenecaFile = senecaParser;

  const Status check = fSenecaFile->FileStatus();

  if (check != eSuccess) {
    fCurrentPosition = fOriginalPosition = 0;
    fIteratorValid = false;
  } else {
    fCurrentPosition = fOriginalPosition = fSenecaFile->GetPosition();
    fIteratorValid = true;
  }

  return check;
}