CMS 3D CMS Logo

All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
PATHemisphereProducer.cc
Go to the documentation of this file.
1 
2 // -*- C++ -*-
3 //
4 // Package: PatShapeAna
5 // Class: PatShapeAna
6 //
14 //
15 // Original Author: Tanja Rommerskirchen
16 // Created: Sat Mar 22 12:58:04 CET 2008
17 //
18 //
19 
20 
21 //system
22 #include <vector>
23 #include <memory>
24 //PAT
32 //DataFormats
37 //User
39 
40 
41 using namespace pat;
42 
43 
44 //
45 // constants, enums and typedefs
46 //
47 
48 //
49 // static data member definitions
50 //
51 
52 //
53 // constructors and destructor
54 //
56  _patJetsToken ( consumes<reco::CandidateView> ( iConfig.getParameter<edm::InputTag>( "patJets" ) ) ),
57  _patMuonsToken ( consumes<reco::CandidateView> ( iConfig.getParameter<edm::InputTag>( "patMuons" ) ) ),
58  _patElectronsToken ( consumes<reco::CandidateView> ( iConfig.getParameter<edm::InputTag>( "patElectrons" ) ) ),
59  _patPhotonsToken ( consumes<reco::CandidateView> ( iConfig.getParameter<edm::InputTag>( "patPhotons" ) ) ),
60  _patTausToken ( consumes<reco::CandidateView> ( iConfig.getParameter<edm::InputTag>( "patTaus" ) ) ),
61 
62  _minJetEt ( iConfig.getParameter<double>("minJetEt") ),
63  _minMuonEt ( iConfig.getParameter<double>("minMuonEt") ),
64  _minElectronEt ( iConfig.getParameter<double>("minElectronEt") ),
65  _minTauEt ( iConfig.getParameter<double>("minTauEt") ),
66  _minPhotonEt ( iConfig.getParameter<double>("minPhotonEt") ),
67 
68  _maxJetEta ( iConfig.getParameter<double>("maxJetEta") ),
69  _maxMuonEta ( iConfig.getParameter<double>("maxMuonEta") ),
70  _maxElectronEta ( iConfig.getParameter<double>("maxElectronEta") ),
71  _maxTauEta ( iConfig.getParameter<double>("maxTauEta") ),
72  _maxPhotonEta ( iConfig.getParameter<double>("maxPhotonEta") ),
73 
74  _seedMethod ( iConfig.getParameter<int>("seedMethod") ),
75  _combinationMethod ( iConfig.getParameter<int>("combinationMethod") )
76 
77 {
78 
79 
80  produces< std::vector<pat::Hemisphere> >();
81 }
82 
83 
85 {
86 
87  // do anything here that needs to be done at desctruction time
88  // (e.g. close files, deallocate resources etc.)
89 
90 }
91 
92 
93 //
94 // member functions
95 //
96 
97 // ------------ method called to produce the data ------------
98 void
100  using namespace edm;
101  using namespace std;
102 
103  std::vector<float> vPx, vPy, vPz, vE;
104  std::vector<float> vA1, vA2;
105  std::vector<int> vgroups;
106  std::vector<reco::CandidatePtr> componentPtrs;
107 
108  //Jets
110  iEvent.getByToken(_patJetsToken,pJets);
111 
112  //Muons
114  iEvent.getByToken(_patMuonsToken,pMuons);
115 
116  //Electrons
117  Handle<reco::CandidateView> pElectrons;
118  iEvent.getByToken(_patElectronsToken,pElectrons);
119 
120  //Photons
122  iEvent.getByToken(_patPhotonsToken,pPhotons);
123 
124  //Taus
126  iEvent.getByToken(_patTausToken,pTaus);
127 
128 
129  //fill e,p vector with information from all objects (hopefully cleaned before)
130  for(int i = 0; i < (int) (*pJets).size() ; i++){
131  if((*pJets)[i].pt() < _minJetEt || fabs((*pJets)[i].eta()) > _maxJetEta) continue;
132 
133  componentPtrs.push_back(pJets->ptrAt(i));
134  }
135 
136  for(int i = 0; i < (int) (*pMuons).size() ; i++){
137  if((*pMuons)[i].pt() < _minMuonEt || fabs((*pMuons)[i].eta()) > _maxMuonEta) continue;
138 
139  componentPtrs.push_back(pMuons->ptrAt(i));
140  }
141 
142  for(int i = 0; i < (int) (*pElectrons).size() ; i++){
143  if((*pElectrons)[i].pt() < _minElectronEt || fabs((*pElectrons)[i].eta()) > _maxElectronEta) continue;
144 
145  componentPtrs.push_back(pElectrons->ptrAt(i));
146  }
147 
148  for(int i = 0; i < (int) (*pPhotons).size() ; i++){
149  if((*pPhotons)[i].pt() < _minPhotonEt || fabs((*pPhotons)[i].eta()) > _maxPhotonEta) continue;
150 
151  componentPtrs.push_back(pPhotons->ptrAt(i));
152  }
153 
154  //aren't taus included in jets?
155  for(int i = 0; i < (int) (*pTaus).size() ; i++){
156  if((*pTaus)[i].pt() < _minTauEt || fabs((*pTaus)[i].eta()) > _maxTauEta) continue;
157 
158  componentPtrs.push_back(pTaus->ptrAt(i));
159  }
160 
161  // create product
162  auto hemispheres = std::make_unique<std::vector<Hemisphere>>();
163  hemispheres->reserve(2);
164 
165  //calls HemiAlgorithm for seed method 3 (transv. inv. Mass) and association method 3 (Lund algo)
166  HemisphereAlgo myHemi(componentPtrs,_seedMethod,_combinationMethod);
167 
168  //get Hemisphere Axis
169  vA1 = myHemi.getAxis1();
170  vA2 = myHemi.getAxis2();
171 
172  reco::Particle::LorentzVector p1(vA1[0]*vA1[3],vA1[1]*vA1[3],vA1[2]*vA1[3],vA1[4]);
173  hemispheres->push_back(Hemisphere(p1));
174 
175  reco::Particle::LorentzVector p2(vA2[0]*vA2[3],vA2[1]*vA2[3],vA2[2]*vA2[3],vA2[4]);
176  hemispheres->push_back(Hemisphere(p2));
177 
178  //get information to which Hemisphere each object belongs
179  vgroups = myHemi.getGrouping();
180 
181  for ( unsigned int i=0; i<vgroups.size(); ++i ) {
182  if ( vgroups[i]==1 ) {
183  (*hemispheres)[0].addDaughter(componentPtrs[i]);
184  }
185  else {
186  (*hemispheres)[1].addDaughter(componentPtrs[i]);
187  }
188  }
189 
190 
191  iEvent.put(std::move(hemispheres));
192 }
193 
194 //define this as a plug-in
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:125
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:517
Ptr< value_type > ptrAt(size_type i) const
const edm::EDGetTokenT< reco::CandidateView > _patElectronsToken
void produce(edm::StreamID, edm::Event &, const edm::EventSetup &) const override
const edm::EDGetTokenT< reco::CandidateView > _patTausToken
std::vector< float > getAxis1()
Definition: HeavyIon.h:7
int iEvent
Definition: GenABIO.cc:224
#define DEFINE_FWK_MODULE(type)
Definition: MakerMacros.h:16
double p2[4]
Definition: TauolaWrapper.h:90
const edm::EDGetTokenT< reco::CandidateView > _patJetsToken
Input: All PAT objects that are to cross-clean or needed for that.
std::vector< int > getGrouping()
PATHemisphereProducer(const edm::ParameterSet &)
fixed size matrix
HLT enums.
double p1[4]
Definition: TauolaWrapper.h:89
const edm::EDGetTokenT< reco::CandidateView > _patPhotonsToken
math::XYZTLorentzVector LorentzVector
Lorentz vector.
Definition: Particle.h:21
const edm::EDGetTokenT< reco::CandidateView > _patMuonsToken
std::vector< float > getAxis2()
def move(src, dest)
Definition: eostools.py:511
edm::View< Candidate > CandidateView
view of a collection containing candidates
Definition: CandidateFwd.h:23