TrackWithHistory.cc

Go to the documentation of this file.

#include "SimG4Core/Notification/interface/TrackWithHistory.h"
#include "SimG4Core/Notification/interface/G4TrackToParticleID.h"
#include "SimG4Core/Notification/interface/TrackInformation.h"
#include "SimG4Core/Notification/interface/GenParticleInfo.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "G4VProcess.hh"
#include "G4DynamicParticle.hh"
#include "G4PrimaryParticle.hh"
#include "G4ThreeVector.hh"

#include <iostream>

G4ThreadLocal G4Allocator<TrackWithHistory>* fpTrackWithHistoryAllocator = nullptr;

TrackWithHistory::TrackWithHistory(const G4Track* g4trk, int pID) {
  trackID_ = g4trk->GetTrackID();
  pdgID_ = G4TrackToParticleID::particleID(g4trk);
  parentID_ = pID;
  auto mom = g4trk->GetMomentum();
  momentum_ = math::XYZVectorD(mom.x(), mom.y(), mom.z());
  totalEnergy_ = g4trk->GetTotalEnergy();
  auto pos = g4trk->GetPosition();
  vertexPosition_ = math::XYZVectorD(pos.x(), pos.y(), pos.z());
  time_ = g4trk->GetGlobalTime();
  auto p = g4trk->GetCreatorProcess();
  procType_ = (nullptr != p) ? p->GetProcessSubType() : 0;
  TrackInformation* trkinfo = static_cast<TrackInformation*>(g4trk->GetUserInformation());
  storeTrack_ = trkinfo->storeTrack();
  auto vgprimary = g4trk->GetDynamicParticle()->GetPrimaryParticle();
  if (vgprimary != nullptr) {
    auto priminfo = static_cast<GenParticleInfo*>(vgprimary->GetUserInformation());
    if (nullptr != priminfo) {
      genParticleID_ = priminfo->id();
    }
  }
  // V.I. weight is computed in the same way as before
  // without usage of G4Track::GetWeight()
  weight_ = 10000 * genParticleID_;
}

TrackWithHistory::TrackWithHistory(const G4PrimaryParticle* ptr, int trackID, const math::XYZVectorD& pos, double time) {
  trackID_ = trackID;
  pdgID_ = G4TrackToParticleID::particleID(ptr, trackID_);
  parentID_ = trackID;
  auto mom = ptr->GetMomentum();
  momentum_ = math::XYZVectorD(mom.x(), mom.y(), mom.z());
  totalEnergy_ = ptr->GetTotalEnergy();
  vertexPosition_ = math::XYZVectorD(pos.x(), pos.y(), pos.z());
  time_ = time;
  storeTrack_ = true;
  auto priminfo = static_cast<GenParticleInfo*>(ptr->GetUserInformation());
  if (nullptr != priminfo) {
    genParticleID_ = priminfo->id();
  }
  weight_ = 10000. * genParticleID_;
}