CMS 3D CMS Logo

List of all members | Public Member Functions | Private Member Functions | Private Attributes
CTPPSHepMCDistributionPlotter Class Reference
Inheritance diagram for CTPPSHepMCDistributionPlotter:
edm::one::EDAnalyzer<> edm::one::EDAnalyzerBase edm::EDConsumerBase

Public Member Functions

 CTPPSHepMCDistributionPlotter (const edm::ParameterSet &)
 
- Public Member Functions inherited from edm::one::EDAnalyzer<>
 EDAnalyzer ()=default
 
SerialTaskQueueglobalLuminosityBlocksQueue () final
 
SerialTaskQueueglobalRunsQueue () final
 
bool wantsGlobalLuminosityBlocks () const final
 
bool wantsGlobalRuns () const final
 
- Public Member Functions inherited from edm::one::EDAnalyzerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
 EDAnalyzerBase ()
 
ModuleDescription const & moduleDescription () const
 
bool wantsStreamLuminosityBlocks () const
 
bool wantsStreamRuns () const
 
 ~EDAnalyzerBase () override
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
void convertCurrentProcessAlias (std::string const &processName)
 Convert "@currentProcess" in InputTag process names to the actual current process name. More...
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ESProxyIndex const * esGetTokenIndices (edm::Transition iTrans) const
 
ProductResolverIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
std::vector< ProductResolverIndexAndSkipBit > const & itemsToGetFrom (BranchType iType) const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
void modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
 
EDConsumerBase const & operator= (EDConsumerBase const &)=delete
 
EDConsumerBaseoperator= (EDConsumerBase &&)=default
 
bool registeredToConsume (ProductResolverIndex, bool, BranchType) const
 
bool registeredToConsumeMany (TypeID const &, BranchType) const
 
ProductResolverIndexAndSkipBit uncheckedIndexFrom (EDGetToken) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
void updateLookup (eventsetup::ESRecordsToProxyIndices const &)
 
virtual ~EDConsumerBase () noexcept(false)
 

Private Member Functions

void analyze (const edm::Event &, const edm::EventSetup &) override
 
void endJob () override
 

Private Attributes

std::unique_ptr< TH1D > h_th_x_
 
std::unique_ptr< TH1D > h_th_y_
 
std::unique_ptr< TH1D > h_vtx_x_
 
std::unique_ptr< TH1D > h_vtx_y_
 
std::unique_ptr< TH1D > h_vtx_z_
 
std::unique_ptr< TH1D > h_xi_
 
std::string lhcInfoLabel_
 
std::string outputFile_
 
edm::EDGetTokenT< edm::HepMCProducttokenHepMC_
 

Additional Inherited Members

- Public Types inherited from edm::one::EDAnalyzerBase
typedef EDAnalyzerBase ModuleType
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- Static Public Member Functions inherited from edm::one::EDAnalyzerBase
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 
- Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > consumes (edm::InputTag const &tag)
 
EDGetToken consumes (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken consumes (TypeToGet const &id, edm::InputTag const &tag)
 
ConsumesCollector consumesCollector ()
 Use a ConsumesCollector to gather consumes information from helper functions. More...
 
template<typename ProductType , BranchType B = InEvent>
void consumesMany ()
 
void consumesMany (const TypeToGet &id)
 
template<BranchType B>
void consumesMany (const TypeToGet &id)
 
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto esConsumes ()
 
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto esConsumes (ESInputTag const &tag)
 
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > mayConsume (edm::InputTag const &tag)
 
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 

Detailed Description

Definition at line 28 of file CTPPSHepMCDistributionPlotter.cc.

Constructor & Destructor Documentation

CTPPSHepMCDistributionPlotter::CTPPSHepMCDistributionPlotter ( const edm::ParameterSet iConfig)
explicit

Definition at line 53 of file CTPPSHepMCDistributionPlotter.cc.

53  :
54  tokenHepMC_( consumes<edm::HepMCProduct>(iConfig.getParameter<edm::InputTag>("tagHepMC")) ),
55  lhcInfoLabel_(iConfig.getParameter<std::string>("lhcInfoLabel")),
56  outputFile_(iConfig.getParameter<string>("outputFile")),
57 
58  h_vtx_x_(new TH1D("h_vtx_x", ";vtx_x (mm)", 100, 0., 0.)),
59  h_vtx_y_(new TH1D("h_vtx_y", ";vtx_y (mm)", 100, 0., 0.)),
60  h_vtx_z_(new TH1D("h_vtx_z", ";vtx_y (mm)", 100, 0., 0.)),
61 
62  h_xi_(new TH1D("h_xi", ";#xi", 100, 0., 0.30)),
63  h_th_x_(new TH1D("h_th_x", ";#theta^{*}_{x}", 100, -300E-6, +300E-6)),
64  h_th_y_(new TH1D("h_th_y", ";#theta^{*}_{y}", 100, -300E-6, +300E-6))
65 {}
T getParameter(std::string const &) const
edm::EDGetTokenT< edm::HepMCProduct > tokenHepMC_

Member Function Documentation

void CTPPSHepMCDistributionPlotter::analyze ( const edm::Event iEvent,
const edm::EventSetup iSetup 
)
overrideprivate

Definition at line 69 of file CTPPSHepMCDistributionPlotter.cc.

References LHCInfo::energy(), edm::EventSetup::get(), edm::Event::getByToken(), edm::HepMCProduct::GetEvent(), h_th_x_, h_th_y_, h_vtx_x_, h_vtx_y_, h_vtx_z_, h_xi_, lhcInfoLabel_, and tokenHepMC_.

70 {
71  // get conditions
72  edm::ESHandle<LHCInfo> hLHCInfo;
73  iSetup.get<LHCInfoRcd>().get(lhcInfoLabel_, hLHCInfo);
74 
75  // get input
77  iEvent.getByToken(tokenHepMC_, hHepMC);
78  HepMC::GenEvent *hepMCEvent = (HepMC::GenEvent *) hHepMC->GetEvent();
79 
80  // plot vertices
81  for (HepMC::GenEvent::vertex_iterator vit = hepMCEvent->vertices_begin(); vit != hepMCEvent->vertices_end(); ++vit)
82  {
83  const auto pos = (*vit)->position();
84  h_vtx_x_->Fill(pos.x());
85  h_vtx_y_->Fill(pos.y());
86  h_vtx_z_->Fill(pos.z());
87  }
88 
89  // extract protons
90  for (auto it = hepMCEvent->particles_begin(); it != hepMCEvent->particles_end(); ++it)
91  {
92  const auto &part = *it;
93 
94  // accept only stable non-beam protons
95  if (part->pdg_id() != 2212)
96  continue;
97 
98  if (part->status() != 1)
99  continue;
100 
101  if (part->is_beam())
102  continue;
103 
104  const auto &mom = part->momentum();
105  const double p_nom = hLHCInfo->energy();
106 
107  if (mom.rho() / p_nom < 0.7)
108  continue;
109 
110  const double xi_simu = (p_nom - mom.e()) / p_nom;
111  const double th_x_simu = mom.x() / mom.rho();
112  const double th_y_simu = mom.y() / mom.rho();
113 
114  h_xi_->Fill(xi_simu);
115  h_th_x_->Fill(th_x_simu);
116  h_th_y_->Fill(th_y_simu);
117  }
118 }
edm::EDGetTokenT< edm::HepMCProduct > tokenHepMC_
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:517
const HepMC::GenEvent * GetEvent() const
Definition: HepMCProduct.h:38
part
Definition: HCALResponse.h:20
T get() const
Definition: EventSetup.h:71
float const energy() const
Definition: LHCInfo.cc:192
VertexRefVector::iterator vertex_iterator
iterator over a vector of references to Vertex objects in the same collection
Definition: VertexFwd.h:19
void CTPPSHepMCDistributionPlotter::endJob ( void  )
overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 122 of file CTPPSHepMCDistributionPlotter.cc.

References DEFINE_FWK_MODULE, h_th_x_, h_th_y_, h_vtx_x_, h_vtx_y_, h_vtx_z_, h_xi_, and outputFile_.

123 {
124  auto f_out = std::make_unique<TFile>(outputFile_.c_str(), "recreate");
125 
126  h_vtx_x_->Write();
127  h_vtx_y_->Write();
128  h_vtx_z_->Write();
129 
130  h_xi_->Write();
131  h_th_x_->Write();
132  h_th_y_->Write();
133 }

Member Data Documentation

std::unique_ptr<TH1D> CTPPSHepMCDistributionPlotter::h_th_x_
private

Definition at line 42 of file CTPPSHepMCDistributionPlotter.cc.

Referenced by analyze(), and endJob().

std::unique_ptr<TH1D> CTPPSHepMCDistributionPlotter::h_th_y_
private

Definition at line 42 of file CTPPSHepMCDistributionPlotter.cc.

Referenced by analyze(), and endJob().

std::unique_ptr<TH1D> CTPPSHepMCDistributionPlotter::h_vtx_x_
private

Definition at line 41 of file CTPPSHepMCDistributionPlotter.cc.

Referenced by analyze(), and endJob().

std::unique_ptr<TH1D> CTPPSHepMCDistributionPlotter::h_vtx_y_
private

Definition at line 41 of file CTPPSHepMCDistributionPlotter.cc.

Referenced by analyze(), and endJob().

std::unique_ptr<TH1D> CTPPSHepMCDistributionPlotter::h_vtx_z_
private

Definition at line 41 of file CTPPSHepMCDistributionPlotter.cc.

Referenced by analyze(), and endJob().

std::unique_ptr<TH1D> CTPPSHepMCDistributionPlotter::h_xi_
private

Definition at line 42 of file CTPPSHepMCDistributionPlotter.cc.

Referenced by analyze(), and endJob().

std::string CTPPSHepMCDistributionPlotter::lhcInfoLabel_
private

Definition at line 38 of file CTPPSHepMCDistributionPlotter.cc.

Referenced by analyze().

std::string CTPPSHepMCDistributionPlotter::outputFile_
private

Definition at line 39 of file CTPPSHepMCDistributionPlotter.cc.

Referenced by endJob().

edm::EDGetTokenT<edm::HepMCProduct> CTPPSHepMCDistributionPlotter::tokenHepMC_
private

Definition at line 37 of file CTPPSHepMCDistributionPlotter.cc.

Referenced by analyze().