#include <SimG4FluxProducer.h>

Inheritance diagram for SimG4FluxProducer:

Public Member Functions
void	produce (edm::Event &, const edm::EventSetup &)

	SimG4FluxProducer (const edm::ParameterSet &p)

virtual	~SimG4FluxProducer ()

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 BeginOfRun * >
	Observer ()

void	slotForUpdate (const BeginOfRun * 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 G4Step * >
	Observer ()

void	slotForUpdate (const G4Step * iT)

virtual	~Observer ()

Private Member Functions
void	endOfEvent (ParticleFlux &pflx, unsigned int k)

std::map< G4LogicalVolume *, std::pair< unsigned int, std::string > >::iterator	findLV (G4LogicalVolume *plv)

G4VPhysicalVolume *	getTopPV ()

const SimG4FluxProducer &	operator= (const SimG4FluxProducer &)

	SimG4FluxProducer (const SimG4FluxProducer &)

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

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

void	update (const G4Step *step)
	This routine will be called when the appropriate signal arrives. More...

Private Attributes
unsigned int	count_

bool	init_

std::vector< std::string >	LVNames_

std::vector< int >	LVTypes_

std::map< G4LogicalVolume *, std::pair< unsigned int, std::string > >	mapLV_

std::map< std::pair< G4LogicalVolume *, unsigned int >, ParticleFlux >	store_

G4VPhysicalVolume *	topPV_

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 25 of file SimG4FluxProducer.h.

Constructor & Destructor Documentation

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

Definition at line 18 of file SimG4FluxProducer.cc.

References gather_cfg::cout, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), gen::k, LVNames_, and LVTypes_.

                                                              : count_(0), init_(false) {
 
   edm::ParameterSet m_FP = p.getParameter<edm::ParameterSet>("SimG4FluxProducer");
   LVNames_  = m_FP.getUntrackedParameter<std::vector<std::string> >("LVNames");
   LVTypes_  = m_FP.getUntrackedParameter<std::vector<int> >("LVTypes");
 
   for (unsigned int k=0; k<LVNames_.size(); ++k) {
     produces<ParticleFlux>(Form("%sParticleFlux",LVNames_[k].c_str()));
 #ifdef EDM_ML_DEBUG
     std::cout << "Collection name[" << k << "] ParticleFlux" << LVNames_[k] 
           << " and type " << LVTypes_[k] << std::endl;
 #endif
   }
 } 

SimG4FluxProducer::~SimG4FluxProducer ( )

virtual

Definition at line 33 of file SimG4FluxProducer.cc.

33 { }

SimG4FluxProducer::SimG4FluxProducer ( const SimG4FluxProducer & )

private

Member Function Documentation

void SimG4FluxProducer::endOfEvent	(	ParticleFlux &	pflx,
		unsigned int	k
	)

private

Definition at line 126 of file SimG4FluxProducer.cc.

References gather_cfg::cout, plotBeamSpotDB::first, ParticleFlux::getComponents(), ParticleFlux::getFlux(), ParticleFlux::getId(), ParticleFlux::getName(), gen::k, mapLV_, and store_.

Referenced by produce().

                                                                      {
 
   bool done(false);
   for (const auto& element : store_) {
     G4LogicalVolume* lv = (element.first).first;
     auto it = mapLV_.find(lv);
     if (it != mapLV_.end()) {
       if ((it->second).first == k) {
     flux = element.second;
     done = true;
 #ifdef EDM_ML_DEBUG
     std::cout << "SimG4FluxProducer[" << k << "] Flux " << flux.getName()
           << ":" << flux.getId() << " with " << flux.getComponents()
           << " elements" << std::endl;
     std::vector<ParticleFlux::flux> fluxes = flux.getFlux() ;
     unsigned int k(0);
     for (auto element : fluxes) {
       std::cout << "Flux[" << k << "] PDGId " << element.pdgId << " VT "
             << element.vxType << " ToF " << element.tof << " Vertex "
             << element.vertex << " Hit " << element.hitPoint << " p "
             << element.momentum << std::endl;
       ++k;
     }
 #endif
       }
     }
     if (done) break;
   }
 }

std::map< G4LogicalVolume *, std::pair< unsigned int, std::string > >::iterator SimG4FluxProducer::findLV ( G4LogicalVolume * plv )

private

Definition at line 160 of file SimG4FluxProducer.cc.

References gen::k, LVNames_, mapLV_, dataset::name, and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by update().

                                                                                                                 {
   auto itr = mapLV_.find(plv);
   if (itr == mapLV_.end()) {
     std::string name = plv->GetName();
     for (unsigned int k=0; k<LVNames_.size(); ++k) {
       if (name.find(LVNames_[k]) != std::string::npos) {
     mapLV_[plv] = std::pair<unsigned int,std::string>(k,name);
     itr = mapLV_.find(plv);
     break;
       }
     }
   }
   return itr;
 }

G4VPhysicalVolume * SimG4FluxProducer::getTopPV ( )

private

Definition at line 156 of file SimG4FluxProducer.cc.

Referenced by update().

                                                 {
   return G4TransportationManager::GetTransportationManager()->GetNavigatorForTracking()->GetWorldVolume();
 }

const SimG4FluxProducer& SimG4FluxProducer::operator= ( const SimG4FluxProducer & )

private

void SimG4FluxProducer::produce	(	edm::Event &	e,
		const edm::EventSetup &
	)

virtual

Implements SimProducer.

Definition at line 35 of file SimG4FluxProducer.cc.

References endOfEvent(), gen::k, LVNames_, eostools::move(), dataset::name, edm::Event::put(), and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                  {
   
   for (unsigned int k=0; k<LVNames_.size(); ++k) {
     std::unique_ptr<ParticleFlux> pflux(new ParticleFlux);
     endOfEvent(*pflux, k);
     std::string name = LVNames_[k] + "ParticleFlux";
     e.put(std::move(pflux),name);
   }
 }

void SimG4FluxProducer::update ( const BeginOfRun * )

privatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfRun * >.

Definition at line 45 of file SimG4FluxProducer.cc.

References gather_cfg::cout, findLV(), plotBeamSpotDB::first, getTopPV(), init_, gen::k, mapLV_, and topPV_.

                                                      {
 
   topPV_ = getTopPV();
   if (topPV_ == 0) {
     edm::LogWarning("SimG4FluxProducer") << "Cannot find top level volume\n";
   } else {
     init_ = true;
     const G4LogicalVolumeStore * lvs = G4LogicalVolumeStore::GetInstance();
     for (auto lvcite : *lvs) {
       findLV(lvcite);
     } 
 
 #ifdef EDM_ML_DEBUG
     std::cout << "SimG4FluxProducer::Finds " << mapLV_.size() << " logical volumes\n";
     unsigned int k(0);
     for (const auto& lvs : mapLV_) {
       std::cout << "Entry[" << k << "] " << lvs.first << ": (" 
         << (lvs.second).first << ", " << (lvs.second).second << ")\n";
       ++k;
     }
 #endif
   }
 }

void SimG4FluxProducer::update ( const BeginOfEvent * )

privatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfEvent * >.

Definition at line 69 of file SimG4FluxProducer.cc.

References count_, and store_.

                                                        {
  
   int iev = (*evt)()->GetEventID();
   edm::LogInfo("ValidHGCal") << "SimG4FluxProducer: =====> Begin event = "
                  << iev << std::endl;
   ++count_;
   store_.clear();
 }

void SimG4FluxProducer::update ( const G4Step * )

privatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const G4Step * >.

Definition at line 78 of file SimG4FluxProducer.cc.

References popcon2dropbox::copy(), gather_cfg::cout, findLV(), plotBeamSpotDB::first, ParticleFlux::flux::hitPoint, init_, createfilelist::int, crabWrapper::key, LVTypes_, mapLV_, ParticleFlux::flux::momentum, NULL, BPhysicsValidation_cfi::pdgid, store_, ntuplemaker::time, HiIsolationCommonParameters_cff::track, and ParticleFlux::flux::vertex.

                                                    {
 
   if (aStep != NULL) {
 
     G4TouchableHistory* touchable = (G4TouchableHistory*)aStep->GetPreStepPoint()->GetTouchable();
     G4LogicalVolume* plv = (G4LogicalVolume*)touchable->GetVolume()->GetLogicalVolume();
     auto it = (init_) ? mapLV_.find(plv) : findLV(plv);
 //  std::cout << plv->GetName() << " Flag " << (it != mapLV_.end()) << " step " << aStep->IsFirstStepInVolume() << ":"  << aStep->IsLastStepInVolume() << std::endl;
     if (it != mapLV_.end()) {
       int type = LVTypes_[(it->second).first];
       if ((type == 0 && aStep->IsFirstStepInVolume()) ||
       (type == 1 && aStep->IsLastStepInVolume())) {
     unsigned int copy = (unsigned int)(touchable->GetReplicaNumber(0));
     std::pair<G4LogicalVolume*,unsigned int> key(plv,copy);
     auto itr = store_.find(key);
     if (itr == store_.end()) {
       store_[key] = ParticleFlux((it->second).second,copy);
       itr         = store_.find(key);
     }
     G4Track* track = aStep->GetTrack();
     int      pdgid = track->GetDefinition()->GetPDGEncoding();
     int      vxtyp = (track->GetCreatorProcess() == nullptr) ? 0 : 1;
     double   time  = (aStep->GetPostStepPoint()->GetGlobalTime());
     const double mmTocm(0.1), MeVToGeV(0.001);
     ParticleFlux::flux flx(pdgid,vxtyp,time);
     flx.vertex   = math::GlobalPoint(mmTocm*track->GetVertexPosition().x(),
                      mmTocm*track->GetVertexPosition().y(),
                      mmTocm*track->GetVertexPosition().z());
     flx.hitPoint = math::GlobalPoint(mmTocm*track->GetPosition().x(),
                      mmTocm*track->GetPosition().y(),
                      mmTocm*track->GetPosition().z());
     flx.momentum = math::GlobalVector(MeVToGeV*track->GetMomentum().x(),
                       MeVToGeV*track->GetMomentum().y(),
                       MeVToGeV*track->GetMomentum().z());
     (itr->second).addFlux(flx);
 #ifdef EDM_ML_DEBUG
     std::cout << "SimG4FluxProducer: Element " << (it->second).first << ":"
           << (it->second).second << ":" << copy << " ID " << pdgid
           << " VxType " << vxtyp << " TOF " << time << " Hit Point "
           << flx.hitPoint << " p " << flx.momentum << " Vertex "
           << flx.vertex << std::endl;
 #endif
       }//if(Step ok)
     }//if( it != map.end() )
   }//if (aStep != NULL)
 
 }

Member Data Documentation

unsigned int SimG4FluxProducer::count_

private

Definition at line 58 of file SimG4FluxProducer.h.

Referenced by update().

bool SimG4FluxProducer::init_

private

Definition at line 59 of file SimG4FluxProducer.h.

Referenced by update().

std::vector<std::string> SimG4FluxProducer::LVNames_

private

Definition at line 52 of file SimG4FluxProducer.h.

Referenced by findLV(), produce(), and SimG4FluxProducer().

std::vector<int> SimG4FluxProducer::LVTypes_

private

Definition at line 53 of file SimG4FluxProducer.h.

Referenced by SimG4FluxProducer(), and update().

std::map<G4LogicalVolume*,std::pair<unsigned int,std::string> > SimG4FluxProducer::mapLV_

private

Definition at line 55 of file SimG4FluxProducer.h.

Referenced by endOfEvent(), findLV(), and update().

std::map<std::pair<G4LogicalVolume*,unsigned int>,ParticleFlux> SimG4FluxProducer::store_

private

Definition at line 60 of file SimG4FluxProducer.h.

Referenced by endOfEvent(), and update().

G4VPhysicalVolume* SimG4FluxProducer::topPV_

private

Definition at line 54 of file SimG4FluxProducer.h.

Referenced by update().

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation