#include <DBremWatcher.h>

Inheritance diagram for DBremWatcher:

Public Member Functions
	DBremWatcher (edm::ParameterSet const &p)

void	produce (edm::Event &, const edm::EventSetup &) override

void	update (const BeginOfTrack *trk) override
	This routine will be called when the appropriate signal arrives. More...

void	update (const BeginOfEvent *event) override
	This routine will be called when the appropriate signal arrives. More...

void	update (const EndOfEvent *event) override
	This routine will be called when the appropriate signal arrives. More...

void	update (const BeginOfRun *run) override
	This routine will be called when the appropriate signal arrives. More...

void	update (const EndOfTrack *trk) override
	This routine will be called when the appropriate signal arrives. More...

	~DBremWatcher () override

Public Member Functions inherited from SimProducer
void	registerProducts (edm::ProducerBase &iProd)

	SimProducer ()

Public Member Functions inherited from SimWatcher
	SimWatcher ()

virtual	~SimWatcher ()

Public Member Functions inherited from Observer< const BeginOfTrack * >
	Observer ()

void	slotForUpdate (const BeginOfTrack * iT)

virtual	~Observer ()

Public Member Functions inherited from Observer< const BeginOfEvent * >
	Observer ()

void	slotForUpdate (const BeginOfEvent * iT)

virtual	~Observer ()

Public Member Functions inherited from Observer< const BeginOfRun * >
	Observer ()

void	slotForUpdate (const BeginOfRun * iT)

virtual	~Observer ()

Public Member Functions inherited from Observer< const EndOfEvent * >
	Observer ()

void	slotForUpdate (const EndOfEvent * iT)

virtual	~Observer ()

Public Member Functions inherited from Observer< const EndOfTrack * >
	Observer ()

void	slotForUpdate (const EndOfTrack * iT)

virtual	~Observer ()

Private Attributes
G4ThreeVector	aPrimeTraj

double	biasFactor

float	f_energy

G4ThreeVector	finaltraj

bool	foundbrem

float	m_weight

int	MotherId

std::vector< int >	pdgs_

G4ThreeVector	VertexPos

Additional Inherited Members
Protected Member Functions inherited from SimProducer
template<class T >
void	produces ()

template<class T >
void	produces (const std::string &instanceName)

Detailed Description

Definition at line 17 of file DBremWatcher.h.

Constructor & Destructor Documentation

DBremWatcher::DBremWatcher ( edm::ParameterSet const & p )

Definition at line 28 of file DBremWatcher.cc.

References aPrimeTraj, biasFactor, f_energy, finaltraj, foundbrem, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), gen::k, m_weight, pdgs_, and VertexPos.

                                                    {
   edm::ParameterSet ps = p.getParameter<edm::ParameterSet>("DBremWatcher");
   biasFactor = ps.getUntrackedParameter<double>("DBremBiasFactor", 1);
   m_weight = 0;
   foundbrem = false;
   finaltraj = G4ThreeVector(0, 0, 0);
   aPrimeTraj = G4ThreeVector(0, 0, 0);
   VertexPos = G4ThreeVector(0, 0, 0);
   f_energy = 0;
   produces<float>("DBremEventWeight");
   produces<float>("DBremLocationX");
   produces<float>("DBremLocationY");
   produces<float>("DBremLocationZ");
   //produces<std::string>("DBremMaterial");
   produces<float>("DBremAngle");
   produces<float>("DBremInitialEnergy");
   produces<float>("DBremFinalEnergy");
   produces<float>("BiasFactor");
   pdgs_ = ps.getUntrackedParameter<std::vector<int>>("PDGCodes");
   edm::LogInfo("DBremWatcher") << "DBremWatcher:: Save Sim Track if PDG code "
                                << "is one from the list of " << pdgs_.size() << " items";
   for (unsigned int k = 0; k < pdgs_.size(); ++k)
     edm::LogInfo("DBremWatcher") << "[" << k << "] " << pdgs_[k];
 }

DBremWatcher::~DBremWatcher ( )

override

Definition at line 53 of file DBremWatcher.cc.

53 {}

Member Function Documentation

void DBremWatcher::produce	(	edm::Event &	fEvent,
		const edm::EventSetup &
	)

overridevirtual

Implements SimProducer.

Definition at line 116 of file DBremWatcher.cc.

References aPrimeTraj, biasFactor, DEFINE_SIMWATCHER, f_energy, finaltraj, foundbrem, GeV, m_weight, eostools::move(), funct::pow(), edm::Event::put(), mathSSE::sqrt(), and VertexPos.

                                                                  {
   if (foundbrem) {
     std::unique_ptr<float> weight = std::make_unique<float>(m_weight);
     fEvent.put(std::move(weight), "DBremEventWeight");
     std::unique_ptr<float> vtxposx = std::make_unique<float>(VertexPos.x());
     std::unique_ptr<float> vtxposy = std::make_unique<float>(VertexPos.y());
     std::unique_ptr<float> vtxposz = std::make_unique<float>(VertexPos.z());
     fEvent.put(std::move(vtxposx), "DBremLocationX");
     fEvent.put(std::move(vtxposy), "DBremLocationY");
     fEvent.put(std::move(vtxposz), "DBremLocationZ");
     std::unique_ptr<float> finalE = std::make_unique<float>(f_energy / GeV);
     fEvent.put(std::move(finalE), "DBremFinalEnergy");
     float deflectionAngle = -1;
     float initialEnergy = sqrt(pow(finaltraj.x() + aPrimeTraj.x(), 2) + pow(finaltraj.y() + aPrimeTraj.y(), 2) +
                                pow(finaltraj.z() + aPrimeTraj.z(), 2) + pow(0.1056 * GeV, 2));
     G4ThreeVector mother(
         finaltraj.x() + aPrimeTraj.x(), finaltraj.y() + aPrimeTraj.y(), finaltraj.z() + aPrimeTraj.z());
     deflectionAngle = mother.angle(finaltraj);
     std::unique_ptr<float> dAngle = std::make_unique<float>(deflectionAngle);
     std::unique_ptr<float> initialE = std::make_unique<float>(initialEnergy / GeV);
     fEvent.put(std::move(dAngle), "DBremAngle");
     fEvent.put(std::move(initialE), "DBremInitialEnergy");
     std::unique_ptr<float> bias = std::make_unique<float>(biasFactor);
     fEvent.put(std::move(bias), "BiasFactor");
   } else {
     std::unique_ptr<float> weight = std::make_unique<float>(0.);
     fEvent.put(std::move(weight), "DBremEventWeight");
   }
 }

void DBremWatcher::update ( const BeginOfTrack * )

overridevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfTrack * >.

Definition at line 55 of file DBremWatcher.cc.

References aPrimeTraj, f_energy, finaltraj, spr::find(), foundbrem, LogDebug, m_weight, MeV, pdgs_, TrackInformation::storeTrack(), and VertexPos.

                                                  {
   G4Track* theTrack = (G4Track*)((*trk)());
   TrackInformation* trkInfo = (TrackInformation*)(theTrack->GetUserInformation());
   if (trkInfo) {
     int pdg = theTrack->GetDefinition()->GetPDGEncoding();
     G4ThreeVector Vpos = theTrack->GetVertexPosition();
     const G4VProcess* TrPro = theTrack->GetCreatorProcess();
     if (TrPro != nullptr) {
       if ((theTrack->GetCreatorProcess()->GetProcessName()) == "muDBrem") {
         if (std::find(pdgs_.begin(), pdgs_.end(), pdg) == pdgs_.end()) {
           //Found the deflected muon track
           trkInfo->storeTrack(true);
           if (!theTrack->IsGoodForTracking()) {
             theTrack->SetGoodForTrackingFlag(true);
           }
           f_energy = theTrack->GetTotalEnergy();
           foundbrem = true;
           finaltraj = theTrack->GetMomentum();
         } else {
           m_weight = theTrack->GetWeight();
         }
       }
     }
     if (std::find(pdgs_.begin(), pdgs_.end(), pdg) != pdgs_.end()) {
       //Found an A'
       trkInfo->storeTrack(true);
       VertexPos = Vpos;
       aPrimeTraj = theTrack->GetMomentum();
       LogDebug("DBremWatcher") << "Save SimTrack the Track " << theTrack->GetTrackID() << " Type "
                                << theTrack->GetDefinition()->GetParticleName() << " Momentum "
                                << theTrack->GetMomentum() / MeV << " MeV/c";
     }
   }
 }

void DBremWatcher::update ( const BeginOfEvent * )

overridevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfEvent * >.

Definition at line 92 of file DBremWatcher.cc.

References foundbrem, and BPhysicsValidation_cfi::pname.

                                                    {
   G4String pname = "muDBrem";
   G4ProcessTable* ptable = G4ProcessTable::GetProcessTable();
   G4bool state = true;
   ptable->SetProcessActivation(pname, state);
   foundbrem = false;
 }

void DBremWatcher::update ( const EndOfEvent * )

overridevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const EndOfEvent * >.

Definition at line 100 of file DBremWatcher.cc.

100 {}

void DBremWatcher::update ( const BeginOfRun * )

overridevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfRun * >.

Definition at line 90 of file DBremWatcher.cc.

90 {}

void DBremWatcher::update ( const EndOfTrack * )

overridevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const EndOfTrack * >.

Definition at line 102 of file DBremWatcher.cc.

References spr::find(), pdgs_, and TrackInformation::storeTrack().

                                                {
   G4Track* theTrack = (G4Track*)((*trk)());
   TrackInformation* trkInfo = (TrackInformation*)(theTrack->GetUserInformation());
   const G4VProcess* TrPro = theTrack->GetCreatorProcess();
   if (trkInfo && TrPro != nullptr) {
     int pdg = theTrack->GetDefinition()->GetPDGEncoding();
 
     if (std::find(pdgs_.begin(), pdgs_.end(), pdg) == pdgs_.end() &&
         (theTrack->GetCreatorProcess()->GetProcessName()) == "muDBrem") {
       trkInfo->storeTrack(true);
     }
   }
 }