TrackWithHistory.h

Go to the documentation of this file.

#ifndef SimG4Core_TrackWithHistory_H
#define SimG4Core_TrackWithHistory_H

#include "G4Track.hh"
#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/Math/interface/LorentzVector.h"

#include "G4Allocator.hh"

class G4VProcess;
class G4TrackToParticleID;
class TrackWithHistory {
public:
  TrackWithHistory(const G4Track *g4track);
  ~TrackWithHistory() {}

  inline void *operator new(size_t);
  inline void operator delete(void *TrackWithHistory);

  void save() { saved_ = true; }
  unsigned int trackID() const { return trackID_; }
  int particleID() const { return particleID_; }
  int parentID() const { return parentID_; }
  int genParticleID() const { return genParticleID_; }
  const math::XYZVectorD &momentum() const { return momentum_; }
  double totalEnergy() const { return totalEnergy_; }
  const math::XYZVectorD &vertexPosition() const { return vertexPosition_; }
  double globalTime() const { return globalTime_; }
  double localTime() const { return localTime_; }
  double properTime() const { return properTime_; }
  const G4VProcess *creatorProcess() const { return creatorProcess_; }
  double weight() const { return weight_; }
  void setTrackID(int i) { trackID_ = i; }
  void setParentID(int i) { parentID_ = i; }
  void setGenParticleID(int i) { genParticleID_ = i; }
  bool storeTrack() const { return storeTrack_; }
  bool saved() const { return saved_; }

  // Boundary crossing variables
  void setCrossedBoundaryPosMom(int id,
                                const math::XYZTLorentzVectorF position,
                                const math::XYZTLorentzVectorF momentum) {
    crossedBoundary_ = true;
    idAtBoundary_ = id;
    positionAtBoundary_ = position;
    momentumAtBoundary_ = momentum;
  }
  bool crossedBoundary() const { return crossedBoundary_; }
  const math::XYZTLorentzVectorF &getPositionAtBoundary() const { return positionAtBoundary_; }
  const math::XYZTLorentzVectorF &getMomentumAtBoundary() const { return momentumAtBoundary_; }
  int getIDAtBoundary() const { return idAtBoundary_; }
  void checkAtEnd(const G4Track *);

private:
  unsigned int trackID_;
  int particleID_;
  int parentID_;
  int genParticleID_;
  math::XYZVectorD momentum_;
  double totalEnergy_;
  math::XYZVectorD vertexPosition_;
  double globalTime_;
  double localTime_;
  double properTime_;
  const G4VProcess *creatorProcess_;
  double weight_;
  bool storeTrack_;
  bool saved_;

  bool isPrimary_;
  bool crossedBoundary_;
  int idAtBoundary_;
  math::XYZTLorentzVectorF positionAtBoundary_;
  math::XYZTLorentzVectorF momentumAtBoundary_;

  int extractGenID(const G4Track *gt) const;
};

extern G4ThreadLocal G4Allocator<TrackWithHistory> *fpTrackWithHistoryAllocator;

inline void *TrackWithHistory::operator new(size_t) {
  if (!fpTrackWithHistoryAllocator)
    fpTrackWithHistoryAllocator = new G4Allocator<TrackWithHistory>;
  return (void *)fpTrackWithHistoryAllocator->MallocSingle();
}

inline void TrackWithHistory::operator delete(void *aTwH) {
  fpTrackWithHistoryAllocator->FreeSingle((TrackWithHistory *)aTwH);
}

#endif