SimG4FluxProducer.h

Go to the documentation of this file.

#ifndef SimG4CMSForwardSimG4FluxProducer_h
#define SimG4CMSForwardSimG4FluxProducer_h
 
#include "SimG4Core/Notification/interface/BeginOfRun.h"
#include "SimG4Core/Notification/interface/BeginOfEvent.h"
#include "SimG4Core/Notification/interface/EndOfEvent.h"
#include "SimG4Core/Watcher/interface/SimProducer.h"
#include "SimG4Core/Notification/interface/Observer.h"
 
// to retreive hits
#include "SimDataFormats/CaloTest/interface/ParticleFlux.h"
 
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "G4Step.hh"
#include "G4LogicalVolumeStore.hh"
#include "G4PhysicalVolumeStore.hh"
 
#include <vector>
#include <string>
#include <map>
 
class SimG4FluxProducer : public SimProducer,
                          public Observer<const BeginOfRun *>,
                          public Observer<const BeginOfEvent *>,
                          public Observer<const G4Step *> {
public:
  SimG4FluxProducer(const edm::ParameterSet &p);
  ~SimG4FluxProducer() override;
 
  void produce(edm::Event &, const edm::EventSetup &) override;
 
private:
  SimG4FluxProducer(const SimG4FluxProducer &) = delete;  // stop default
  const SimG4FluxProducer &operator=(const SimG4FluxProducer &) = delete;
 
  // observer classes
  void update(const BeginOfRun *run) override;
  void update(const BeginOfEvent *evt) override;
  void update(const G4Step *step) override;
 
  void endOfEvent(ParticleFlux &pflx, unsigned int k);
  G4VPhysicalVolume *getTopPV();
  std::map<G4LogicalVolume *, std::pair<unsigned int, std::string>>::iterator findLV(G4LogicalVolume *plv);
 
private:
  std::vector<std::string> LVNames_;
  std::vector<int> LVTypes_;
  G4VPhysicalVolume *topPV_;
  std::map<G4LogicalVolume *, std::pair<unsigned int, std::string>> mapLV_;
 
  // some private members for ananlysis
  unsigned int count_;
  bool init_;
  std::map<std::pair<G4LogicalVolume *, unsigned int>, ParticleFlux> store_;
};
 
#endif