TrackRecord.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRACKRECORD_TRACKRECORD_H
#define TRACKRECORD_TRACKRECORD_H
 
#include "CLHEP/Vector/ThreeVector.h"
#include "TruthUtils/MagicNumbers.h"
#include "GeneratorObjects/HepMcParticleLink.h"
 
class TrackRecord {
public:
  TrackRecord() = default;
 
  virtual ~TrackRecord() = default;
 
  TrackRecord(
              int pdg,
              int status,
              double energy,
              const CLHEP::Hep3Vector& momentum,
              const CLHEP::Hep3Vector& postition,
              double time,
              int barcode,
              int id,
              const std::string& volumeName)
    : m_pdgCode(pdg)
    , m_status(status)
    , m_energy(energy)
    , m_momentum(momentum)
    , m_position(postition)
    , m_time(time)
    , m_barcode(barcode)
    , m_id(id)
    , m_volName(volumeName)
  {
    if (m_id == HepMC::INVALID_PARTICLE_ID && m_barcode != HepMC::INVALID_PARTICLE_ID) {
      m_partLink = std::make_unique<HepMcParticleLink>(m_barcode, 0, HepMcParticleLink::IS_POSITION, HepMcParticleLink::IS_BARCODE); // FIXME is barcode-based
    }
  }
 
  TrackRecord(const TrackRecord& trc)
    : m_pdgCode(trc.m_pdgCode)
    , m_status(trc.m_status)
    , m_energy(trc.m_energy)
    , m_momentum(trc.m_momentum)
    , m_position(trc.m_position)
    , m_time(trc.m_time)
    , m_barcode(trc.m_barcode)
    , m_id(trc.m_id)
    , m_volName(trc.m_volName){
      if (m_id == HepMC::INVALID_PARTICLE_ID && m_barcode != HepMC::INVALID_PARTICLE_ID) {
        m_id = trc.id(); // Try a look-up via HepMcParticleLink
    }
}
 
  TrackRecord &operator=(const TrackRecord& trc) {
    if (this != &trc) {
      m_pdgCode = trc.m_pdgCode;
      m_status = trc.m_status;
      m_energy = trc.m_energy;
      m_momentum = trc.m_momentum;
      m_position = trc.m_position;
      m_time = trc.m_time;
      m_barcode = trc.m_barcode;
      m_id = trc.m_id;
      if (m_id == HepMC::INVALID_PARTICLE_ID && m_barcode != HepMC::INVALID_PARTICLE_ID) {
        m_id = trc.id(); // Try a look-up via HepMcParticleLink
      }
      m_volName = trc.m_volName;
    }
    return *this;
  }
 
  double GetEnergy() const {return m_energy;}
 
  void SetEnergy(double e) {m_energy = e;}
 
  CLHEP::Hep3Vector GetPosition() const {return m_position;}
 
  void SetPosition(CLHEP::Hep3Vector p) {m_position = p;}
 
  CLHEP::Hep3Vector GetMomentum() const {return m_momentum;}
 
  void SetMomentum(CLHEP::Hep3Vector e) {m_momentum = e;}
 
  int GetPDGCode() const {return m_pdgCode;}
 
  void SetPDGCode(int pcode) {m_pdgCode = pcode;}
 
  double GetTime() const {return m_time;}
 
  void SetTime(double time) {m_time = time;}
 
  const std::string& GetVolName() const {return m_volName;}
 
  void SetVolName(const std::string& theName){m_volName = theName;}
 
  int status() const {return m_status;}
 
  void SetStatus(int status) {m_status = status;}
 
  int id() const {if (m_id == HepMC::INVALID_PARTICLE_ID && m_partLink) { return m_partLink->id(); } else { return m_id; } }
 
  void SetID(int uniqueID){
    m_id = uniqueID;
    if (m_id == HepMC::INVALID_PARTICLE_ID && m_barcode != HepMC::INVALID_PARTICLE_ID) {
      m_partLink = std::make_unique<HepMcParticleLink>(m_barcode, 0, HepMcParticleLink::IS_POSITION, HepMcParticleLink::IS_BARCODE); // FIXME is barcode-based
    }
  }
 
  int barcode() const {return m_barcode;}
 
  void SetBarcode(int barcode){
    m_barcode = barcode;
    if (m_id == HepMC::INVALID_PARTICLE_ID && m_barcode != HepMC::INVALID_PARTICLE_ID) {
      m_partLink = std::make_unique<HepMcParticleLink>(m_barcode, 0, HepMcParticleLink::IS_POSITION, HepMcParticleLink::IS_BARCODE); // FIXME is barcode-based
    }
  }
 
private:
  int m_pdgCode{0};
  int m_status{0};
  double m_energy{0};
  CLHEP::Hep3Vector m_momentum{0,0,0};
  CLHEP::Hep3Vector m_position{0,0,0};
  double m_time{0.};
  int m_barcode{HepMC::UNDEFINED_ID};
  int m_id{HepMC::UNDEFINED_ID};
  std::string m_volName{""};
  std::unique_ptr<HepMcParticleLink> m_partLink{}; // nullptr unless object was produced by reading TrackRecord_p1.
};
 
#endif // TRACKRECORD_TRACKRECORD_H