MonopoleSteppingAction.cc

Go to the documentation of this file.

#include "SimG4Core/HelpfulWatchers/interface/MonopoleSteppingAction.h"

#include "SimG4Core/Notification/interface/BeginOfRun.h"
#include "SimG4Core/Notification/interface/BeginOfTrack.h"
#include "SimG4Core/Physics/interface/Monopole.h"

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "G4Event.hh"
#include "G4ParticleTable.hh"
#include "G4Run.hh"
#include "G4Track.hh"

#include "CLHEP/Units/GlobalPhysicalConstants.h"
#include "CLHEP/Units/GlobalSystemOfUnits.h"

MonopoleSteppingAction::MonopoleSteppingAction(edm::ParameterSet const &p) : actOnTrack(false), bZ(0) {
  mode = p.getUntrackedParameter<bool>("ChangeFromFirstStep", true);
  edm::LogVerbatim("SimG4CoreWatcher") << "MonopoleSeppingAction set mode for"
                                       << " start at first step to " << mode;
}

MonopoleSteppingAction::~MonopoleSteppingAction() {}

void MonopoleSteppingAction::registerConsumes(edm::ConsumesCollector cc) {
  tok_bFieldH_ = cc.esConsumes<MagneticField, IdealMagneticFieldRecord, edm::Transition::BeginRun>();
  edm::LogVerbatim("SimG4CoreWatcher") << "MonopoleSteppingAction::Initialize ESGetToken for MagneticField";
}

void MonopoleSteppingAction::beginRun(edm::EventSetup const &es) {
  const MagneticField *bField = &es.getData(tok_bFieldH_);
  const GlobalPoint p(0, 0, 0);
  bZ = (bField->inTesla(p)).z();
  edm::LogVerbatim("SimG4CoreWatcher") << "Magnetic Field (X): " << (bField->inTesla(p)).x()
                                       << " Y: " << (bField->inTesla(p)).y() << " Z: " << bZ;
}

void MonopoleSteppingAction::update(const BeginOfRun *) {
  G4ParticleTable *partTable = G4ParticleTable::GetParticleTable();
  magCharge = CLHEP::e_SI / CLHEP::fine_structure_const * 0.5;
  for (int ii = 0; ii < partTable->size(); ii++) {
    G4ParticleDefinition *particle = partTable->GetParticle(ii);
    std::string particleName = (particle->GetParticleName()).substr(0, 8);
    if (strcmp(particleName.c_str(), "Monopole") == 0) {
      magCharge = CLHEP::e_SI * ((Monopole *)(particle))->MagneticCharge();
      pdgCode.push_back(particle->GetPDGEncoding());
    }
  }
  edm::LogVerbatim("SimG4CoreWatcher") << "MonopoleSeppingAction Finds " << pdgCode.size() << " candidates";
  for (unsigned int ii = 0; ii < pdgCode.size(); ++ii) {
    edm::LogVerbatim("SimG4CoreWatcher") << "PDG Code[" << ii << "] = " << pdgCode[ii];
  }
  cMevToJ = CLHEP::e_SI / CLHEP::eV;
  cMeVToKgMByS = CLHEP::e_SI * CLHEP::meter / (CLHEP::eV * CLHEP::c_light * CLHEP::second);
  cInMByS = CLHEP::c_light * CLHEP::second / CLHEP::meter;
  edm::LogVerbatim("SimG4CoreWatcher") << "MonopoleSeppingAction Constants:"
                                       << " MevToJoules  = " << cMevToJ << ", MevToKgm/s  = " << cMeVToKgMByS
                                       << ", c in m/s    = " << cInMByS << ", mag. charge = " << magCharge;
}

void MonopoleSteppingAction::update(const BeginOfTrack *trk) {
  actOnTrack = false;
  if (!pdgCode.empty()) {
    const G4Track *aTrack = (*trk)();
    int code = aTrack->GetDefinition()->GetPDGEncoding();
    if (std::count(pdgCode.begin(), pdgCode.end(), code) > 0) {
      actOnTrack = true;
      eStart = aTrack->GetTotalEnergy();
      pxStart = aTrack->GetMomentum().x();
      pyStart = aTrack->GetMomentum().y();
      pzStart = aTrack->GetMomentum().z();
      dirxStart = aTrack->GetMomentumDirection().x();
      diryStart = aTrack->GetMomentumDirection().y();
      dirzStart = aTrack->GetMomentumDirection().z();
      LogDebug("SimG4CoreWatcher") << "MonopoleSeppingAction Track " << code << " Flag " << actOnTrack
                                   << " (px,py,pz,E) = (" << pxStart / GeV << ", " << pyStart / GeV << ", "
                                   << pzStart / GeV << ", " << eStart / GeV << ")";
    }
  }
}

void MonopoleSteppingAction::update(const G4Step *aStep) {
  if (actOnTrack) {
    double eT, pT, pZ, tStep;
    G4Track *aTrack = aStep->GetTrack();
    G4ThreeVector initialPosition(0, 0, 0);
    if (mode) {
      tStep = aTrack->GetGlobalTime();
      eT = eStart * std::sqrt(dirxStart * dirxStart + diryStart * diryStart);
      pT = std::sqrt(pxStart * pxStart + pyStart * pyStart);
      pZ = pzStart;
    } else {
      const G4ThreeVector &dirStep = aTrack->GetMomentumDirection();
      double lStep = aTrack->GetStepLength();
      double xStep = aTrack->GetPosition().x() - lStep * dirStep.x();
      double yStep = aTrack->GetPosition().y() - lStep * dirStep.y();
      double zStep = aTrack->GetPosition().z() - lStep * dirStep.z();
      double vStep = aTrack->GetVelocity();
      initialPosition = G4ThreeVector(xStep, yStep, zStep);
      tStep = (vStep > 0. ? (lStep / vStep) : 0.);
      double eStep = aTrack->GetTotalEnergy();
      eT = eStep * dirStep.perp();
      pT = aTrack->GetMomentum().perp();
      pZ = aTrack->GetMomentum().z();
    }
    LogDebug("SimG4CoreWatcher") << "MonopoleSeppingAction: tStep " << tStep << " eT " << eT << " pT " << pT << " pZ "
                                 << pZ;
    double eT1 = eT * cMevToJ;
    double pT1 = pT * cMeVToKgMByS;
    double pZ1 = pZ * cMeVToKgMByS;
    double ts1 = tStep / CLHEP::second;
    double eT2 = (eT1 > 0. ? (1. / eT1) : 0.);
    double fac0 = magCharge * bZ * cInMByS;
    double fac2 = pZ1 * cInMByS * eT2;
    double fac1 = fac0 * cInMByS * ts1 * eT2 + fac2;
    double fact1 = (eT1 / fac0) * (std::sqrt(1 + fac1 * fac1) - std::sqrt(1 + fac2 * fac2));
    double fact2 = (pT1 / (magCharge * bZ) * (asinh(fac1) - asinh(fac2)));
    LogDebug("SimG4CoreWatcher") << "MonopoleSeppingAction: Factor eT = " << eT << " " << eT1 << " " << eT2
                                 << ", pT = " << pT << " " << pT1 << ", pZ = " << pZ << " " << pZ1
                                 << ", time = " << tStep << " " << ts1 << ", Factors " << fac0 << " " << fac1 << " "
                                 << fac2 << ", Final ones " << fact1 << " " << fact2;
    G4ThreeVector ez(0, 0, 1), et(std::sqrt(0.5), std::sqrt(0.5), 0);
    G4ThreeVector displacement = (fact1 * ez + fact2 * et) * CLHEP::m;

    LogDebug("SimG4CoreWatcher") << "MonopoleSeppingAction: Initial " << initialPosition << " Displacement "
                                 << displacement << " Final " << (initialPosition + displacement);
    aTrack->SetPosition(initialPosition + displacement);
  }
}