#include <StackingAction.h>
Public Member Functions | |
virtual G4ClassificationOfNewTrack | ClassifyNewTrack (const G4Track *aTrack) |
virtual void | NewStage () |
virtual void | PrepareNewEvent () |
StackingAction (const edm::ParameterSet &ps) | |
virtual | ~StackingAction () |
Private Member Functions | |
void | initPointer () |
int | isItFromPrimary (const G4Track &, int) const |
bool | isItLongLived (const G4Track *) const |
int | isItPrimaryDecayProductOrConversion (const G4Track *, const G4Track &) const |
bool | isThisVolume (const G4VTouchable *, std::vector< G4LogicalVolume * > &) const |
Private Attributes | |
std::vector< G4LogicalVolume * > | calo |
bool | killDeltaRay |
bool | killHeavy |
bool | killInCalo |
bool | killInCaloEfH |
double | kmaxIon |
double | kmaxNeutron |
double | kmaxProton |
std::vector< std::string > | maxTimeNames |
std::vector< G4Region * > | maxTimeRegions |
double | maxTrackTime |
std::vector< double > | maxTrackTimes |
std::vector< G4LogicalVolume * > | muon |
double | nRusRoEcal |
double | nRusRoEcalLim |
double | nRusRoHcal |
double | nRusRoHcalLim |
double | pRusRoEcal |
double | pRusRoEcalLim |
double | pRusRoHcal |
double | pRusRoHcalLim |
G4Region * | regionEcal |
G4Region * | regionHcal |
bool | saveFirstSecondary |
bool | savePDandCinCalo |
bool | savePDandCinMuon |
bool | savePDandCinTracker |
std::vector< G4LogicalVolume * > | tracker |
bool | trackNeutrino |
Definition at line 14 of file StackingAction.h.
StackingAction::StackingAction | ( | const edm::ParameterSet & | ps | ) |
Definition at line 19 of file StackingAction.cc.
References edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), i, initPointer(), killDeltaRay, killHeavy, killInCalo, killInCaloEfH, kmaxIon, kmaxNeutron, kmaxProton, maxTimeNames, maxTrackTime, maxTrackTimes, nRusRoEcal, nRusRoEcalLim, nRusRoHcal, nRusRoHcalLim, pRusRoEcal, pRusRoEcalLim, pRusRoHcal, pRusRoHcalLim, regionEcal, regionHcal, saveFirstSecondary, savePDandCinCalo, savePDandCinMuon, savePDandCinTracker, and trackNeutrino.
{ trackNeutrino = p.getParameter<bool>("TrackNeutrino"); killHeavy = p.getParameter<bool>("KillHeavy"); kmaxIon = p.getParameter<double>("IonThreshold")*MeV; kmaxProton = p.getParameter<double>("ProtonThreshold")*MeV; kmaxNeutron = p.getParameter<double>("NeutronThreshold")*MeV; killDeltaRay = p.getParameter<bool>("KillDeltaRay"); maxTrackTime = p.getParameter<double>("MaxTrackTime")*ns; maxTrackTimes = p.getParameter<std::vector<double> >("MaxTrackTimes"); for(unsigned int i=0; i<maxTrackTimes.size(); ++i) { maxTrackTimes[i] *= ns; } maxTimeNames = p.getParameter<std::vector<std::string> >("MaxTimeNames"); savePDandCinTracker = p.getUntrackedParameter<bool>("SavePrimaryDecayProductsAndConversionsInTracker",false); savePDandCinCalo = p.getUntrackedParameter<bool>("SavePrimaryDecayProductsAndConversionsInCalo",false); savePDandCinMuon = p.getUntrackedParameter<bool>("SavePrimaryDecayProductsAndConversionsInMuon",false); saveFirstSecondary = p.getUntrackedParameter<bool>("SaveFirstLevelSecondary",false); killInCalo = false; killInCaloEfH = false; // Russian Roulette nRusRoEcal = p.getParameter<double>("RusRoEcalNeutron"); nRusRoHcal = p.getParameter<double>("RusRoHcalNeutron"); nRusRoEcalLim = p.getParameter<double>("RusRoEcalNeutronLimit")*MeV; nRusRoHcalLim = p.getParameter<double>("RusRoHcalNeutronLimit")*MeV; pRusRoEcal = p.getParameter<double>("RusRoEcalProton"); pRusRoHcal = p.getParameter<double>("RusRoHcalProton"); pRusRoEcalLim = p.getParameter<double>("RusRoEcalProtonLimit")*MeV; pRusRoHcalLim = p.getParameter<double>("RusRoHcalProtonLimit")*MeV; regionEcal = 0; regionHcal = 0; if ( p.exists("TestKillingOptions") ) { killInCalo = (p.getParameter<edm::ParameterSet>("TestKillingOptions")).getParameter<bool>("KillInCalo"); killInCaloEfH = (p.getParameter<edm::ParameterSet>("TestKillingOptions")).getParameter<bool>("KillInCaloEfH"); edm::LogWarning("SimG4CoreApplication") << " *** Activating special test killing options in StackingAction \n" << " *** Kill secondaries in Calorimetetrs volume = " << killInCalo << "\n" << " *** Kill electromagnetic secondaries from hadrons in Calorimeters volume = " << killInCaloEfH; } edm::LogInfo("SimG4CoreApplication") << "StackingAction initiated with" << " flag for saving decay products in " << " Tracker: " << savePDandCinTracker << " in Calo: " << savePDandCinCalo << " in Muon: " << savePDandCinMuon << "\n saveFirstSecondary" << ": " << saveFirstSecondary << " Flag for tracking neutrino: " << trackNeutrino << " Killing Flag " << killHeavy << " protons below " << kmaxProton <<" MeV, neutrons below " << kmaxNeutron << " MeV and ions" << " below " << kmaxIon << " MeV\n" << " kill tracks with " << "time larger than " << maxTrackTime << " ns and kill Delta Ray flag set to " << killDeltaRay; for (unsigned int i=0; i<maxTrackTimes.size(); i++) { maxTrackTimes[i] *= ns; edm::LogInfo("SimG4CoreApplication") << "StackingAction::MaxTrackTime for " << maxTimeNames[i] << " is " << maxTrackTimes[i]; } if(nRusRoEcal < 1.0 || nRusRoHcal < 1.0) { edm::LogInfo("SimG4CoreApplication") << "StackingAction: " << "Russian Roulette for neutron in ECAL Prob= " << nRusRoEcal << " Elimit(MeV)= " << nRusRoEcalLim << "\n" << " " << "Russian Roulette for neutron in HCAL Prob= " << nRusRoHcal << " Elimit(MeV)= " << nRusRoHcalLim; } if(pRusRoEcal < 1.0 || pRusRoHcal < 1.0) { edm::LogInfo("SimG4CoreApplication") << "StackingAction: " << "Russian Roulette for proton in ECAL Prob= " << pRusRoEcal << " Elimit(MeV)= " << pRusRoEcalLim << "\n" << " " << "Russian Roulette for proton in HCAL Prob= " << pRusRoHcal << " Elimit(MeV)= " << pRusRoHcalLim; } initPointer(); }
StackingAction::~StackingAction | ( | ) | [virtual] |
Definition at line 106 of file StackingAction.cc.
{}
G4ClassificationOfNewTrack StackingAction::ClassifyNewTrack | ( | const G4Track * | aTrack | ) | [virtual] |
Definition at line 108 of file StackingAction.cc.
References abs, calo, isItFromPrimary(), isItLongLived(), isItPrimaryDecayProductOrConversion(), isThisVolume(), ke, killDeltaRay, killHeavy, killInCalo, killInCaloEfH, kmaxIon, kmaxNeutron, kmaxProton, LogDebug, muon, nRusRoEcal, nRusRoEcalLim, nRusRoHcal, nRusRoHcalLim, NewTrackAction::primary(), pRusRoEcal, pRusRoEcalLim, pRusRoHcal, pRusRoHcalLim, regionEcal, regionHcal, saveFirstSecondary, savePDandCinCalo, savePDandCinMuon, savePDandCinTracker, NewTrackAction::secondary(), CurrentG4Track::track(), tracker, and trackNeutrino.
{ // G4 interface part G4ClassificationOfNewTrack classification = fUrgent; int flag = 0; NewTrackAction newTA; if (aTrack->GetCreatorProcess()==0 || aTrack->GetParentID()==0) { /* std::cout << "StackingAction: primary weight= " << aTrack->GetWeight() << " " << aTrack->GetDefinition()->GetParticleName() << " " << aTrack->GetKineticEnergy() << " Id=" << aTrack->GetTrackID() << " trackInfo " << aTrack->GetUserInformation() << std::endl; */ newTA.primary(aTrack); if (!trackNeutrino) { int pdg = std::abs(aTrack->GetDefinition()->GetPDGEncoding()); if (pdg == 12 || pdg == 14 || pdg == 16 || pdg == 18) { classification = fKill; } } } else if (aTrack->GetTouchable() == 0) { edm::LogError("SimG4CoreApplication") << "StackingAction: no touchable for track " << aTrack->GetTrackID() << " from " << aTrack->GetParentID() << " with PDG code " << aTrack->GetDefinition()->GetParticleName(); classification = fKill; } else { const G4Track * mother = CurrentG4Track::track(); if ((savePDandCinTracker && isThisVolume(aTrack->GetTouchable(),tracker))|| (savePDandCinCalo && isThisVolume(aTrack->GetTouchable(),calo)) || (savePDandCinMuon && isThisVolume(aTrack->GetTouchable(),muon))) flag = isItPrimaryDecayProductOrConversion(aTrack, *mother); if (saveFirstSecondary) flag = isItFromPrimary(*mother, flag); newTA.secondary(aTrack, *mother, flag); if (aTrack->GetTrackStatus() == fStopAndKill) classification = fKill; if (killHeavy && classification != fKill) { int pdg = aTrack->GetDefinition()->GetPDGEncoding(); double ke = aTrack->GetKineticEnergy()/MeV; if (((pdg/1000000000 == 1) && (((pdg/10000)%100) > 0) && (((pdg/10)%100) > 0) && (ke<kmaxIon)) || ((pdg == 2212) && (ke < kmaxProton)) || ((pdg == 2112) && (ke < kmaxNeutron))) classification = fKill; } if (!trackNeutrino && classification != fKill) { int pdg = std::abs(aTrack->GetDefinition()->GetPDGEncoding()); if (pdg == 12 || pdg == 14 || pdg == 16 || pdg == 18) classification = fKill; } if (isItLongLived(aTrack)) classification = fKill; if (killDeltaRay) { if (aTrack->GetCreatorProcess()->GetProcessType() == fElectromagnetic && aTrack->GetCreatorProcess()->GetProcessSubType() == fIonisation) classification = fKill; } if (killInCalo && classification != fKill) { if ( isThisVolume(aTrack->GetTouchable(),calo)) { classification = fKill; } } if (killInCaloEfH && classification != fKill) { int pdg = aTrack->GetDefinition()->GetPDGEncoding(); int pdgMother = mother->GetDefinition()->GetPDGEncoding(); if ( (pdg == 22 || std::abs(pdg) == 11) && (std::abs(pdgMother) < 11 || std::abs(pdgMother) > 17) && pdgMother != 22 ) { if ( isThisVolume(aTrack->GetTouchable(),calo)) { classification = fKill; } } } double currentWeight = aTrack->GetWeight(); if((regionEcal || regionHcal) && classification != fKill && 1.0 >= currentWeight) { int pdg = aTrack->GetDefinition()->GetPDGEncoding(); if(2112 == pdg || 2212 == pdg) { G4Region* reg = aTrack->GetVolume()->GetLogicalVolume()->GetRegion(); double prob = 1.0; double elim = 0.0; if(reg == regionEcal) { if(2112 == pdg) { prob = nRusRoEcal; elim = nRusRoEcalLim; } else { prob = pRusRoEcal; elim = pRusRoEcalLim; } } else if(reg == regionHcal) { if(2112 == pdg) { prob = nRusRoHcal; elim = nRusRoHcalLim; } else { prob = pRusRoHcal; elim = pRusRoHcalLim; } } if(prob < 1.0 && aTrack->GetKineticEnergy() < elim) { if(G4UniformRand() < prob) { const_cast<G4Track*>(aTrack)->SetWeight(currentWeight/prob); } else { classification = fKill; } } } } /* double wt2 = aTrack->GetWeight(); if(wt2 != 1.0) { G4Region* reg = aTrack->GetVolume()->GetLogicalVolume()->GetRegion(); std::cout << "StackingAction: weight= " << wt2 << " " << aTrack->GetDefinition()->GetParticleName() << " " << aTrack->GetKineticEnergy() << " Id=" << aTrack->GetTrackID() << " IdP=" << aTrack->GetParentID(); const G4VProcess* pr = aTrack->GetCreatorProcess(); if(pr) std::cout << " from " << pr->GetProcessName(); if(reg) std::cout << " in " << reg->GetName() << " trackInfo " << aTrack->GetUserInformation(); std::cout << std::endl; } */ #ifdef DebugLog LogDebug("SimG4CoreApplication") << "StackingAction:Classify Track " << aTrack->GetTrackID() << " Parent " << aTrack->GetParentID() << " Type " << aTrack->GetDefinition()->GetParticleName() << " K.E. " << aTrack->GetKineticEnergy()/MeV << " MeV from process/subprocess " << aTrack->GetCreatorProcess()->GetProcessType() << "|" << aTrack->GetCreatorProcess()->GetProcessSubType() << " as " << classification << " Flag " << flag; #endif } return classification; }
void StackingAction::initPointer | ( | ) | [private] |
Definition at line 252 of file StackingAction.cc.
References calo, i, killInCalo, killInCaloEfH, maxTimeNames, maxTimeRegions, maxTrackTimes, muon, mergeVDriftHistosByStation::name, nRusRoEcal, nRusRoHcal, pRusRoEcal, pRusRoHcal, regionEcal, regionHcal, savePDandCinCalo, savePDandCinMuon, savePDandCinTracker, and tracker.
Referenced by StackingAction().
{ const G4LogicalVolumeStore * lvs = G4LogicalVolumeStore::GetInstance(); if (lvs) { std::vector<G4LogicalVolume*>::const_iterator lvcite; for (lvcite = lvs->begin(); lvcite != lvs->end(); lvcite++) { if (savePDandCinTracker) { if (strcmp("Tracker",(*lvcite)->GetName().c_str()) == 0) tracker.push_back(*lvcite); if (strcmp("BEAM",(*lvcite)->GetName().substr(0,4).c_str()) == 0) tracker.push_back(*lvcite); } if (savePDandCinCalo || killInCalo || killInCaloEfH ) { if (strcmp("CALO",(*lvcite)->GetName().c_str()) == 0) calo.push_back(*lvcite); if (strcmp("VCAL",(*lvcite)->GetName().c_str()) == 0) calo.push_back(*lvcite); } if (savePDandCinMuon) { if (strcmp("MUON",(*lvcite)->GetName().c_str()) == 0) muon.push_back(*lvcite); } } edm::LogInfo("SimG4CoreApplication") << "# of LV for Tracker " << tracker.size() << " for Calo " << calo.size() << " for Muon " << muon.size(); for (unsigned int i=0; i<tracker.size(); ++i) edm::LogInfo("SimG4CoreApplication") << "Tracker vol " << i << " name " << tracker[i]->GetName(); for (unsigned int i=0; i<calo.size(); ++i) edm::LogInfo("SimG4CoreApplication") << "Calorimeter vol " <<i <<" name " << calo[i]->GetName(); for (unsigned int i=0; i<muon.size(); ++i) edm::LogInfo("SimG4CoreApplication") << "Muon vol " << i << " name " << muon[i]->GetName(); } const G4RegionStore * rs = G4RegionStore::GetInstance(); unsigned int num = maxTimeNames.size(); if (num > 0) { std::vector<double> tofs; if (rs) { std::vector<G4Region*>::const_iterator rcite; for (rcite = rs->begin(); rcite != rs->end(); rcite++) { if ((nRusRoEcal < 1.0 || pRusRoEcal < 1.0) && (*rcite)->GetName() == "EcalRegion") { regionEcal = (*rcite); } if ((nRusRoHcal < 1.0 || pRusRoHcal < 1.0) && (*rcite)->GetName() == "HcalRegion") { regionHcal = (*rcite); } for (unsigned int i=0; i<num; i++) { if ((*rcite)->GetName() == (G4String)(maxTimeNames[i])) { maxTimeRegions.push_back(*rcite); tofs.push_back(maxTrackTimes[i]); break; } } if (tofs.size() == num) break; } } for (unsigned int i=0; i<tofs.size(); i++) { maxTrackTimes[i] = tofs[i]; G4String name = "Unknown"; if (maxTimeRegions[i]) name = maxTimeRegions[i]->GetName(); edm::LogInfo("SimG4CoreApplication") << name << " with pointer " << maxTimeRegions[i]<<" KE cut off " << maxTrackTimes[i]; } } }
int StackingAction::isItFromPrimary | ( | const G4Track & | mother, |
int | flagIn | ||
) | const [private] |
Definition at line 346 of file StackingAction.cc.
References TrackInformation::isPrimary().
Referenced by ClassifyNewTrack().
{ int flag = flagIn; if (flag != 1) { TrackInformationExtractor extractor; const TrackInformation & motherInfo(extractor(mother)); if (motherInfo.isPrimary()) flag = 3; } return flag; }
bool StackingAction::isItLongLived | ( | const G4Track * | aTrack | ) | const [private] |
Definition at line 357 of file StackingAction.cc.
References i, maxTimeRegions, maxTrackTime, maxTrackTimes, and cond::rpcobgas::time.
Referenced by ClassifyNewTrack().
{ bool flag = false; double time = (aTrack->GetGlobalTime())/nanosecond; double tofM = maxTrackTime; if (maxTimeRegions.size() > 0) { G4Region* reg = aTrack->GetTouchable()->GetVolume(0)->GetLogicalVolume()->GetRegion(); for (unsigned int i=0; i<maxTimeRegions.size(); i++) { if (reg == maxTimeRegions[i]) { tofM = maxTrackTimes[i]; break; } } } if (time > tofM) flag = true; return flag; }
int StackingAction::isItPrimaryDecayProductOrConversion | ( | const G4Track * | aTrack, |
const G4Track & | mother | ||
) | const [private] |
Definition at line 331 of file StackingAction.cc.
References TrackInformation::isPrimary().
Referenced by ClassifyNewTrack().
{ int flag = 0; TrackInformationExtractor extractor; const TrackInformation & motherInfo(extractor(mother)); // Check whether mother is a primary if (motherInfo.isPrimary()) { if (aTrack->GetCreatorProcess()->GetProcessType() == fDecay) flag = 1; else if (aTrack->GetCreatorProcess()->GetProcessType() == fElectromagnetic && aTrack->GetCreatorProcess()->GetProcessSubType() == fGammaConversion) flag = 2; } return flag; }
bool StackingAction::isThisVolume | ( | const G4VTouchable * | touch, |
std::vector< G4LogicalVolume * > & | lvs | ||
) | const [private] |
Definition at line 317 of file StackingAction.cc.
References prof2calltree::count, and testEve_cfg::level.
Referenced by ClassifyNewTrack().
{ bool flag = false; if (lvs.size() > 0 && touch !=0) { int level = ((touch->GetHistoryDepth())+1); if (level >= 3) { unsigned int ii = (unsigned int)(level - 3); flag = (std::count(lvs.begin(),lvs.end(),(touch->GetVolume(ii)->GetLogicalVolume())) != 0); } } return flag; }
void StackingAction::NewStage | ( | ) | [virtual] |
Definition at line 248 of file StackingAction.cc.
{}
void StackingAction::PrepareNewEvent | ( | ) | [virtual] |
Definition at line 250 of file StackingAction.cc.
{}
std::vector<G4LogicalVolume*> StackingAction::calo [private] |
Definition at line 38 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and initPointer().
bool StackingAction::killDeltaRay [private] |
Definition at line 32 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and StackingAction().
bool StackingAction::killHeavy [private] |
Definition at line 32 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and StackingAction().
bool StackingAction::killInCalo [private] |
Definition at line 31 of file StackingAction.h.
Referenced by ClassifyNewTrack(), initPointer(), and StackingAction().
bool StackingAction::killInCaloEfH [private] |
Definition at line 31 of file StackingAction.h.
Referenced by ClassifyNewTrack(), initPointer(), and StackingAction().
double StackingAction::kmaxIon [private] |
Definition at line 33 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and StackingAction().
double StackingAction::kmaxNeutron [private] |
Definition at line 33 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and StackingAction().
double StackingAction::kmaxProton [private] |
Definition at line 33 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and StackingAction().
std::vector<std::string> StackingAction::maxTimeNames [private] |
Definition at line 36 of file StackingAction.h.
Referenced by initPointer(), and StackingAction().
std::vector<G4Region*> StackingAction::maxTimeRegions [private] |
Definition at line 37 of file StackingAction.h.
Referenced by initPointer(), and isItLongLived().
double StackingAction::maxTrackTime [private] |
Definition at line 34 of file StackingAction.h.
Referenced by isItLongLived(), and StackingAction().
std::vector<double> StackingAction::maxTrackTimes [private] |
Definition at line 35 of file StackingAction.h.
Referenced by initPointer(), isItLongLived(), and StackingAction().
std::vector<G4LogicalVolume*> StackingAction::muon [private] |
Definition at line 38 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and initPointer().
double StackingAction::nRusRoEcal [private] |
Definition at line 41 of file StackingAction.h.
Referenced by ClassifyNewTrack(), initPointer(), and StackingAction().
double StackingAction::nRusRoEcalLim [private] |
Definition at line 42 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and StackingAction().
double StackingAction::nRusRoHcal [private] |
Definition at line 45 of file StackingAction.h.
Referenced by ClassifyNewTrack(), initPointer(), and StackingAction().
double StackingAction::nRusRoHcalLim [private] |
Definition at line 46 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and StackingAction().
double StackingAction::pRusRoEcal [private] |
Definition at line 43 of file StackingAction.h.
Referenced by ClassifyNewTrack(), initPointer(), and StackingAction().
double StackingAction::pRusRoEcalLim [private] |
Definition at line 44 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and StackingAction().
double StackingAction::pRusRoHcal [private] |
Definition at line 47 of file StackingAction.h.
Referenced by ClassifyNewTrack(), initPointer(), and StackingAction().
double StackingAction::pRusRoHcalLim [private] |
Definition at line 48 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and StackingAction().
G4Region* StackingAction::regionEcal [private] |
Definition at line 39 of file StackingAction.h.
Referenced by ClassifyNewTrack(), initPointer(), and StackingAction().
G4Region* StackingAction::regionHcal [private] |
Definition at line 40 of file StackingAction.h.
Referenced by ClassifyNewTrack(), initPointer(), and StackingAction().
bool StackingAction::saveFirstSecondary [private] |
Definition at line 30 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and StackingAction().
bool StackingAction::savePDandCinCalo [private] |
Definition at line 29 of file StackingAction.h.
Referenced by ClassifyNewTrack(), initPointer(), and StackingAction().
bool StackingAction::savePDandCinMuon [private] |
Definition at line 30 of file StackingAction.h.
Referenced by ClassifyNewTrack(), initPointer(), and StackingAction().
bool StackingAction::savePDandCinTracker [private] |
Definition at line 29 of file StackingAction.h.
Referenced by ClassifyNewTrack(), initPointer(), and StackingAction().
std::vector<G4LogicalVolume*> StackingAction::tracker [private] |
Definition at line 38 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and initPointer().
bool StackingAction::trackNeutrino [private] |
Definition at line 32 of file StackingAction.h.
Referenced by ClassifyNewTrack(), and StackingAction().