CMS 3D CMS Logo

List of all members | Public Member Functions | Private Attributes
EcalGenEvtSelectorFrag Class Reference

#include <EcalGenEvtSelectorFrag.h>

Inheritance diagram for EcalGenEvtSelectorFrag:
BaseHiGenEvtSelector

Public Member Functions

 EcalGenEvtSelectorFrag (const edm::ParameterSet &pset)
 
bool filter (HepMC::GenEvent *) override
 
 ~EcalGenEvtSelectorFrag () override
 
- Public Member Functions inherited from BaseHiGenEvtSelector
 BaseHiGenEvtSelector (const edm::ParameterSet &)
 
bool selectParticle (HepMC::GenParticle *par, int status, int pdg, double ptMin, double etaMax)
 
virtual ~BaseHiGenEvtSelector ()
 

Private Attributes

double etaMax_
 
std::vector< int > particleId_
 
std::vector< double > particlePt_
 
std::vector< int > particleStatus_
 
std::vector< int > partonId_
 
std::vector< double > partonPt_
 
std::vector< int > partonStatus_
 

Detailed Description

Definition at line 7 of file EcalGenEvtSelectorFrag.h.

Constructor & Destructor Documentation

EcalGenEvtSelectorFrag::EcalGenEvtSelectorFrag ( const edm::ParameterSet pset)

Definition at line 7 of file EcalGenEvtSelectorFrag.cc.

References etaMax_, Exception, edm::ParameterSet::getParameter(), edm::errors::LogicError, particleId_, particlePt_, particleStatus_, partonId_, partonPt_, partonStatus_, and DiDispStaMuonMonitor_cfi::pt.

7  : BaseHiGenEvtSelector(pset) {
8  partonId_ = pset.getParameter<vector<int> >("partons");
9  partonStatus_ = pset.getParameter<vector<int> >("partonStatus");
10  partonPt_ = pset.getParameter<vector<double> >("partonPt");
11 
12  particleId_ = pset.getParameter<vector<int> >("particles");
13  particleStatus_ = pset.getParameter<vector<int> >("particleStatus");
14  particlePt_ = pset.getParameter<vector<double> >("particlePt");
15 
16  etaMax_ = pset.getParameter<double>("etaMax");
17 
18  int id = partonId_.size();
19  int st = partonStatus_.size();
20  int pt = partonPt_.size();
21 
22  if (partonId_.size() != partonStatus_.size() || partonId_.size() != partonPt_.size()) {
23  throw edm::Exception(edm::errors::LogicError) << id << st << pt << endl;
24  }
25 
26  id = particleId_.size();
27  st = particleStatus_.size();
28  pt = particlePt_.size();
29 
30  if (particleId_.size() != particleStatus_.size() || particleId_.size() != particlePt_.size()) {
31  throw edm::Exception(edm::errors::LogicError) << id << st << pt << endl;
32  }
33 }
T getParameter(std::string const &) const
std::vector< int > partonStatus_
std::vector< double > particlePt_
std::vector< int > partonId_
std::vector< int > particleId_
BaseHiGenEvtSelector(const edm::ParameterSet &)
std::vector< double > partonPt_
std::vector< int > particleStatus_
EcalGenEvtSelectorFrag::~EcalGenEvtSelectorFrag ( )
inlineoverride

Definition at line 10 of file EcalGenEvtSelectorFrag.h.

References filter().

10 { ; }

Member Function Documentation

bool EcalGenEvtSelectorFrag::filter ( HepMC::GenEvent evt)
overridevirtual

Reimplemented from BaseHiGenEvtSelector.

Definition at line 35 of file EcalGenEvtSelectorFrag.cc.

References begin, end, etaMax_, mps_fire::i, particleId_, particlePt_, particleStatus_, and BaseHiGenEvtSelector::selectParticle().

Referenced by ~EcalGenEvtSelectorFrag().

35  {
36  HepMC::GenEvent::particle_const_iterator begin = evt->particles_begin();
37  HepMC::GenEvent::particle_const_iterator end = evt->particles_end();
38 
39  bool foundParticle = false;
40  bool foundParton = false;
41 
42  HepMC::GenEvent::particle_const_iterator it = begin;
43  while ((!foundParton || !foundParticle) && it != end) {
44  bool isFrag = false;
45  bool isThisPhoton = false;
46 
47  foundParton = true;
48  /*for(unsigned i = 0; i < partonId_.size(); ++i){
49  if(selectParticle(*it, partonStatus_[i], partonId_[i], partonPt_[i], etaMax_)) foundParton = true;
50  }*/
51 
52  for (unsigned i = 0; i < particleId_.size(); ++i) {
54  isThisPhoton = true;
55  }
56 
57  // Now you have to requre the partcile is "prompt", meaning its mom is parton
58 
59  if (!((*it)->production_vertex())) {
60  isFrag = false;
61  } else {
62  const HepMC::GenVertex* productionVertex = (*it)->production_vertex();
63 
64  size_t numberOfMothers = productionVertex->particles_in_size();
65  if (numberOfMothers <= 0) {
66  isFrag = false;
67  // cout << "number of mothers = " << numberOfMothers << endl;
68  } else {
69  // cout << "number of mothers = " << numberOfMothers << endl;
70  HepMC::GenVertex::particles_in_const_iterator motherIt = productionVertex->particles_in_const_begin();
71  for (; motherIt != productionVertex->particles_in_const_end(); motherIt++) {
72  if (fabs((*motherIt)->pdg_id()) <= 22) {
73  isFrag = true;
74  }
75  }
76  }
77  }
78 
79  if ((isFrag == true) && (isThisPhoton == true)) {
80  //cout << "Found fragmentation photon!!" << endl ;
81  foundParticle = true;
82  }
83 
84  ++it;
85  }
86 
87  foundParton = true; // We don't care of the parton
88  return (foundParton && foundParticle);
89 }
bool selectParticle(HepMC::GenParticle *par, int status, int pdg, double ptMin, double etaMax)
std::vector< double > particlePt_
#define end
Definition: vmac.h:39
std::vector< int > particleId_
#define begin
Definition: vmac.h:32
std::vector< int > particleStatus_

Member Data Documentation

double EcalGenEvtSelectorFrag::etaMax_
private

Definition at line 23 of file EcalGenEvtSelectorFrag.h.

Referenced by EcalGenEvtSelectorFrag(), and filter().

std::vector<int> EcalGenEvtSelectorFrag::particleId_
private

Definition at line 19 of file EcalGenEvtSelectorFrag.h.

Referenced by EcalGenEvtSelectorFrag(), and filter().

std::vector<double> EcalGenEvtSelectorFrag::particlePt_
private

Definition at line 21 of file EcalGenEvtSelectorFrag.h.

Referenced by EcalGenEvtSelectorFrag(), and filter().

std::vector<int> EcalGenEvtSelectorFrag::particleStatus_
private

Definition at line 20 of file EcalGenEvtSelectorFrag.h.

Referenced by EcalGenEvtSelectorFrag(), and filter().

std::vector<int> EcalGenEvtSelectorFrag::partonId_
private

Definition at line 15 of file EcalGenEvtSelectorFrag.h.

Referenced by EcalGenEvtSelectorFrag().

std::vector<double> EcalGenEvtSelectorFrag::partonPt_
private

Definition at line 17 of file EcalGenEvtSelectorFrag.h.

Referenced by EcalGenEvtSelectorFrag().

std::vector<int> EcalGenEvtSelectorFrag::partonStatus_
private

Definition at line 16 of file EcalGenEvtSelectorFrag.h.

Referenced by EcalGenEvtSelectorFrag().