---

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/TrackInformationExtractor.h"
#include "SimG4Core/Notification/interface/SimG4Exception.h"
#include "SimG4Core/Notification/interface/GenParticleInfoExtractor.h"

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

#include "G4VProcess.hh"

#include <iostream>

using std::cout;
using std::endl;

//#define DEBUG

G4TrackToParticleID * TrackWithHistory::theG4TrackToParticleID(0);

TrackWithHistory::TrackWithHistory(const G4Track * g4trk) 
  : trackID_(0),particleID_(0),parentID_(0),momentum_(math::XYZVectorD(0.,0.,0.)),totalEnergy_(0),
    vertexPosition_(math::XYZVectorD(0.,0.,0.)),globalTime_(0),localTime_(0),properTime_(0),
    creatorProcess_(0),weight_(0),storeTrack_(false),saved_(false)
{
    if (theG4TrackToParticleID == 0) theG4TrackToParticleID = new G4TrackToParticleID;  
    if (g4trk!=0) 
    {
        TrackInformationExtractor extractor;
        trackID_ = g4trk->GetTrackID();
        particleID_ = theG4TrackToParticleID->particleID(g4trk);
        parentID_ = g4trk->GetParentID();
        momentum_ = math::XYZVectorD(g4trk->GetMomentum().x(),g4trk->GetMomentum().y(),g4trk->GetMomentum().z());
        totalEnergy_ = g4trk->GetTotalEnergy();
        vertexPosition_ = math::XYZVectorD(g4trk->GetPosition().x(),g4trk->GetPosition().y(),g4trk->GetPosition().z());
        globalTime_  = g4trk->GetGlobalTime();
        localTime_  = g4trk->GetLocalTime();
        properTime_  = g4trk->GetProperTime();
        creatorProcess_ = g4trk->GetCreatorProcess();
        weight_ = g4trk->GetWeight();
        storeTrack_ = extractor(g4trk).storeTrack();
        saved_ = false;
        genParticleID_ = extractGenID( g4trk);
#ifdef DEBUG    
        LogDebug("TrackWithHistory") << " TrackWithHistory : created history for " << trackID_
             << " with mother " << parentID_;
#endif
    }
}

void TrackWithHistory::checkAtEnd(const G4Track * gt)
{

  math::XYZVectorD vposdir(gt->GetVertexPosition().x(),gt->GetVertexPosition().y(),gt->GetVertexPosition().z());
  math::XYZVectorD vmomdir(gt->GetVertexMomentumDirection().x(),gt->GetVertexMomentumDirection().y(),gt->GetVertexMomentumDirection().z());
  bool ok = true;
  double epsilon = 1.e-6;
  double eps2 = epsilon*epsilon;
  if ((vertexPosition_-vposdir).Mag2() > eps2) 
    {
      cout << "TrackWithHistory vertex position check failed" << endl;
      cout << "At construction: " << vertexPosition_ << endl;
      cout << "At end:          " << vposdir << endl;
      ok = false;
    }
  math::XYZVectorD dirDiff = momentum_.Unit() - vmomdir;
  if (dirDiff.Mag2() > eps2) 
    {
      cout << "TrackWithHistory momentum direction check failed" << endl;
      cout << "At construction: " << momentum_.Unit() << endl;
      cout << "At end:          " << vmomdir << endl;
      ok = false;
    }
  if (!ok) throw SimG4Exception("TrackWithHistory::checkAtEnd failed");
}

int TrackWithHistory::extractGenID(const G4Track* gt) const
{
  void * vgprimary = gt->GetDynamicParticle()->GetPrimaryParticle();
  if (vgprimary == 0) return -1;
  // replace old-style cast with appropriate new-style cast...
  G4PrimaryParticle* gprimary = (G4PrimaryParticle*) vgprimary;
  GenParticleInfoExtractor ext;
  return ext(gprimary).id();
}

---