SimG4FluxProducer Class Reference

Inheritance diagram for SimG4FluxProducer:

Public Member Functions
const SimG4FluxProducer &	operator= (const SimG4FluxProducer &)=delete
void	produce (edm::Event &, const edm::EventSetup &) override
	SimG4FluxProducer (const edm::ParameterSet &p)
	SimG4FluxProducer (const SimG4FluxProducer &)=delete
	~SimG4FluxProducer () override
Public Member Functions inherited from SimProducer
const SimProducer &	operator= (const SimProducer &)=delete
void	registerProducts (edm::ProducesCollector producesCollector)
	SimProducer ()
	SimProducer (const SimProducer &)=delete
Public Member Functions inherited from SimWatcher
virtual void	beginRun (edm::EventSetup const &)
bool	isMT () const
const SimWatcher &	operator= (const SimWatcher &)=delete
virtual void	registerConsumes (edm::ConsumesCollector)
	SimWatcher ()
	SimWatcher (const SimWatcher &)=delete
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 ()
void	update (const BeginOfRun *run) override
	This routine will be called when the appropriate signal arrives. More...
void	update (const BeginOfEvent *evt) override
	This routine will be called when the appropriate signal arrives. More...
void	update (const G4Step *step) override
	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)
Protected Member Functions inherited from SimWatcher
void	setMT (bool val)

Detailed Description

Definition at line 35 of file SimG4FluxProducer.cc.

Constructor & Destructor Documentation

SimG4FluxProducer() [1/2]

SimG4FluxProducer::SimG4FluxProducer

(

const edm::ParameterSet &

p

)

Definition at line 68 of file SimG4FluxProducer.cc.

References edm::ParameterSet::getUntrackedParameter(), dqmdumpme::k, LVNames_, LVTypes_, and AlCaHLTBitMon_ParallelJobs::p.

                                                             : 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
    edm::LogVerbatim("SimG4FluxProducer")
        << "SimG4FluxProducer::Collection name[" << k << "] ParticleFlux" << LVNames_[k] << " and type " << LVTypes_[k];
#endif
  }
}

SimG4FluxProducer() [2/2]

SimG4FluxProducer::SimG4FluxProducer ( const SimG4FluxProducer &  )

delete

~SimG4FluxProducer()

SimG4FluxProducer::~SimG4FluxProducer ( )

override

Definition at line 82 of file SimG4FluxProducer.cc.

{}

Member Function Documentation

endOfEvent()

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

private

Definition at line 171 of file SimG4FluxProducer.cc.

References fileCollector::done, first, ParticleFlux::getComponents(), ParticleFlux::getFlux(), ParticleFlux::getId(), ParticleFlux::getName(), dqmdumpme::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
        edm::LogVerbatim("SimG4FluxProducer") << "SimG4FluxProducer[" << k << "] Flux " << flux.getName() << ":"
                                              << flux.getId() << " with " << flux.getComponents() << " elements";
        std::vector<ParticleFlux::flux> fluxes = flux.getFlux();
        unsigned int k(0);
        for (auto element : fluxes) {
          edm::LogVerbatim("SimG4FluxProducer")
              << "Flux[" << k << "] PDGId " << element.pdgId << " VT " << element.vxType << " ToF " << element.tof
              << " Vertex " << element.vertex << " Hit " << element.hitPoint << " p " << element.momentum;
          ++k;
        }
#endif
      }
    }
    if (done)
      break;
  }
}

findLV()

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

private

Definition at line 203 of file SimG4FluxProducer.cc.

References dqmdumpme::k, LVNames_, mapLV_, Skims_PA_cff::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;
}

getTopPV()

G4VPhysicalVolume * SimG4FluxProducer::getTopPV ( )

private

Definition at line 199 of file SimG4FluxProducer.cc.

Referenced by update().

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

operator=()

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

delete

produce()

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

overridevirtual

Implements SimProducer.

Definition at line 84 of file SimG4FluxProducer.cc.

References MillePedeFileConverter_cfg::e, endOfEvent(), dqmdumpme::k, LVNames_, eostools::move(), Skims_PA_cff::name, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                  {
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("SimG4FluxProducer") << "SimG4FluxProducer: enters produce with " << LVNames_.size() << " LV's";
#endif
  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);
  }
}

update() [1/3]

void SimG4FluxProducer::update ( const BeginOfRun *  )

overrideprivatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfRun *>.

Definition at line 96 of file SimG4FluxProducer.cc.

References findLV(), first, getTopPV(), init_, dqmdumpme::k, mapLV_, and topPV_.

Referenced by progressbar.ProgressBar::__next__(), MatrixUtil.Matrix::__setitem__(), MatrixUtil.Steps::__setitem__(), progressbar.ProgressBar::finish(), and MatrixUtil.Steps::overwrite().

                                                    {
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("SimG4FluxProducer") << "SimG4FluxProducer: enters BeginOfRun";
#endif
  topPV_ = getTopPV();
  if (topPV_ == nullptr) {
    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
    edm::LogVerbatim("SimG4FluxProducer") << "SimG4FluxProducer::Finds " << mapLV_.size() << " logical volumes";
    unsigned int k(0);
    for (const auto &lvs : mapLV_) {
      edm::LogVerbatim("SimG4FluxProducer")
          << "Entry[" << k << "] " << lvs.first << ": (" << (lvs.second).first << ", " << (lvs.second).second << ")";
      ++k;
    }
#endif
  }
}

update() [2/3]

void SimG4FluxProducer::update ( const BeginOfEvent *  )

overrideprivatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfEvent *>.

Definition at line 122 of file SimG4FluxProducer.cc.

References count_, iev, and store_.

Referenced by progressbar.ProgressBar::__next__(), MatrixUtil.Matrix::__setitem__(), MatrixUtil.Steps::__setitem__(), progressbar.ProgressBar::finish(), and MatrixUtil.Steps::overwrite().

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

update() [3/3]

void SimG4FluxProducer::update ( const G4Step *  )

overrideprivatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const G4Step *>.

Definition at line 129 of file SimG4FluxProducer.cc.

References filterCSVwithJSON::copy, findLV(), first, ParticleFlux::flux::hitPoint, init_, createfilelist::int, crabWrapper::key, LVTypes_, mapLV_, ParticleFlux::flux::momentum, EgammaValidation_cff::pdgid, store_, protons_cff::time, HLT_2022v15_cff::track, and ParticleFlux::flux::vertex.

Referenced by progressbar.ProgressBar::__next__(), MatrixUtil.Matrix::__setitem__(), MatrixUtil.Steps::__setitem__(), progressbar.ProgressBar::finish(), and MatrixUtil.Steps::overwrite().

                                                  {
  if (aStep != nullptr) {
    G4TouchableHistory *touchable = (G4TouchableHistory *)aStep->GetPreStepPoint()->GetTouchable();
    G4LogicalVolume *plv = (G4LogicalVolume *)touchable->GetVolume()->GetLogicalVolume();
    auto it = (init_) ? mapLV_.find(plv) : findLV(plv);
    //  edm::LogVerbatim("SimG4FluxProducer") << plv->GetName() << " Flag " << (it != mapLV_.end()) << " step " << aStep->IsFirstStepInVolume() << ":"  << aStep->IsLastStepInVolume();
    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
        edm::LogVerbatim("SimG4FluxProducer")
            << "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;
#endif
      }  //if(Step ok)
    }    //if( it != map.end() )
  }      //if (aStep != NULL)
}

Member Data Documentation

count_

unsigned int SimG4FluxProducer::count_

private

Definition at line 63 of file SimG4FluxProducer.cc.

Referenced by update().

init_

bool SimG4FluxProducer::init_

private

Definition at line 64 of file SimG4FluxProducer.cc.

Referenced by update().

LVNames_

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

private

Definition at line 57 of file SimG4FluxProducer.cc.

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

LVTypes_

std::vector<int> SimG4FluxProducer::LVTypes_

private

Definition at line 58 of file SimG4FluxProducer.cc.

Referenced by SimG4FluxProducer(), and update().

mapLV_

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

private

Definition at line 60 of file SimG4FluxProducer.cc.

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

store_

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

private

Definition at line 65 of file SimG4FluxProducer.cc.

Referenced by endOfEvent(), and update().

topPV_

G4VPhysicalVolume* SimG4FluxProducer::topPV_

private

Definition at line 59 of file SimG4FluxProducer.cc.

Referenced by update().