TrackInformation.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MCTRUTH_TRACKINFORMATION_H
#define MCTRUTH_TRACKINFORMATION_H
 
#include "VTrackInformation.h"
#include "TruthUtils/MagicNumbers.h" //HepMC::INVALID_PARTICLE_BARCODE
#include "CxxUtils/checker_macros.h"
 
namespace ISF {
  class ISFParticle;
}

class TrackInformation: public VTrackInformation {
public:
  TrackInformation();
  TrackInformation(HepMC::GenParticlePtr p, ISF::ISFParticle* baseIsp=nullptr);
  virtual HepMC::ConstGenParticlePtr GetCurrentGenParticle() const override {return m_currentGenParticle;}
  virtual HepMC::GenParticlePtr GetCurrentGenParticle() override {return m_currentGenParticle;}
  virtual void SetCurrentGenParticle(HepMC::GenParticlePtr) override;

  virtual HepMC::ConstGenParticlePtr GetGenerationZeroGenParticle() const override {return m_generationZeroGenParticle;}
  virtual HepMC::GenParticlePtr GetGenerationZeroGenParticle() override {return m_generationZeroGenParticle;}
  virtual void SetGenerationZeroGenParticle(HepMC::GenParticlePtr) override;

  virtual const ISF::ISFParticle *GetBaseISFParticle() const override {return m_theBaseISFParticle;}
  virtual ISF::ISFParticle *GetBaseISFParticle() override {return m_theBaseISFParticle;}
  virtual void SetBaseISFParticle(ISF::ISFParticle*) override;

  virtual bool GetReturnedToISF() const override {return m_returnedToISF;}
  virtual void SetReturnedToISF(bool returned) override {m_returnedToISF=returned;}

  int GetRegenerationNr() const {return m_regenerationNr;}
  void SetRegenerationNr(int i) {m_regenerationNr=i;}
 
  virtual int GetParticleBarcode() const override; // TODO Drop this once UniqueID and Status are used instead
  virtual int GetParticleUniqueID() const override;
  virtual int GetParticleStatus() const override;
private:
  int m_regenerationNr{0};
  HepMC::GenParticlePtr m_currentGenParticle{};
  HepMC::GenParticlePtr m_generationZeroGenParticle{};
  mutable int m_barcode ATLAS_THREAD_SAFE = HepMC::INVALID_PARTICLE_BARCODE; // TODO Drop this once UniqueID and Status are used instead
  mutable int m_uniqueID ATLAS_THREAD_SAFE = HepMC::INVALID_PARTICLE_BARCODE;
  ISF::ISFParticle *m_theBaseISFParticle{};
  bool m_returnedToISF{false};
};
 
#endif // MCTRUTH_TRACKINFORMATION_H