SimG4Core/Notification/src/TrackWithHistory.cc - master - githubmirror/cmssw

SimG4Core/Notification/src/TrackWithHistory.cc
66 строк3 KB
Vladimir
code format
23 апр 2026, 16:04
23 апр 2026, 16:044a762ed
#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();
  const 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());
  if (nullptr != trkinfo) {
    storeTrack_ = trkinfo->storeTrack();
  }
  auto vgprimary = g4trk->GetDynamicParticle()->GetPrimaryParticle();
  // GetPrimaryParticle() returns the pointer to the corresponding G4PrimaryParticle object
  // if this particle is a primary particle OR is defined as a
  // pre-assigned decay product. Otherwise return nullptr.
  // Therefore here the genParticleID_ is -1 for not primary particles
  if (vgprimary != nullptr) {
    auto priminfo = static_cast<GenParticleInfo*>(vgprimary->GetUserInformation());
    if (nullptr != priminfo) {
      genParticleID_ = priminfo->id();
      isPrimary_ = true;
    }
  }
  // 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();
    isPrimary_ = true;
  }
  weight_ = 10000. * genParticleID_;
}