BeamHaloParticle.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef BEAMHALOPARTICLE_H
#define BEAMHALOPARTICLE_H
 
#include "AtlasHepMC/SimpleVector.h"
 
class MarsParticle;
class FlukaParticle;
namespace HepPDT {
  class ParticleDataTable;
}

class BeamHaloParticle {
 public:
  
  BeamHaloParticle();
  
  BeamHaloParticle(long pdgId, 
           HepMC::FourVector fourVector, 
           HepMC::FourVector positionAtScoringPlane, 
           double weight);
 
  BeamHaloParticle(long pdgId,
                   HepMC::FourVector fourVector,
                   HepMC::FourVector positionAtScoringPlane,
                   double weight,
                   HepMC::FourVector positionAtPrimary);
 
 
  BeamHaloParticle(const BeamHaloParticle& beamHaloParticle) = default;

  BeamHaloParticle & operator=(const BeamHaloParticle &rhs) = default;

  int fill(const HepPDT::ParticleDataTable *particleDataTable,
       MarsParticle *marsParticle);
  
  int fill(const HepPDT::ParticleDataTable *particleDataTable,
       FlukaParticle *flukaParticle);
  
  void print();
  
  long pdgId() const { return m_pdgId; }

  HepMC::FourVector fourVector() const { return m_fourVector; }

  HepMC::FourVector positionAtScoringPlane() const { return m_positionAtScoringPlane; } 

  double weight() const { return m_weight; }

  HepMC::FourVector positionAtPrimary() const { return m_positionAtPrimary; }
 
 private:

  long m_pdgId;

  HepMC::FourVector m_fourVector;
  
  HepMC::FourVector m_positionAtScoringPlane;
  
  double m_weight;

  HepMC::FourVector m_positionAtPrimary;
};
 
#endif