ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillerElectrons.cc
Revision: 1.50
Committed: Mon Nov 22 16:55:51 2010 UTC (14 years, 5 months ago) by bendavid
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_017, Mit_017pre3, Mit_017pre2, Mit_017pre1
Changes since 1.49: +21 -87 lines
Log Message:
Adapt to new expected hits inner

File Contents

# User Rev Content
1 bendavid 1.50 // $Id: FillerElectrons.cc,v 1.49 2010/10/18 01:34:48 bendavid Exp $
2 loizides 1.1
3     #include "MitProd/TreeFiller/interface/FillerElectrons.h"
4     #include "DataFormats/TrackReco/interface/Track.h"
5     #include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
6     #include "DataFormats/TrackReco/interface/TrackFwd.h"
7 sixie 1.11 #include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
8     #include "DataFormats/EgammaCandidates/interface/GsfElectronFwd.h"
9     #include "DataFormats/EgammaReco/interface/ClusterShape.h"
10     #include "DataFormats/EgammaReco/interface/BasicClusterShapeAssociation.h"
11 bendavid 1.18 #include "DataFormats/Common/interface/RefToPtr.h"
12 loizides 1.35 #include "DataFormats/Common/interface/ValueMap.h"
13     #include "AnalysisDataFormats/Egamma/interface/ElectronID.h"
14     #include "AnalysisDataFormats/Egamma/interface/ElectronIDAssociation.h"
15 sixie 1.11 #include "RecoEgamma/EgammaIsolationAlgos/interface/ElectronTkIsolation.h"
16     #include "RecoEgamma/EgammaIsolationAlgos/interface/EgammaEcalIsolation.h"
17 sixie 1.14 #include "RecoEgamma/EgammaIsolationAlgos/interface/EgammaTowerIsolation.h"
18 loizides 1.35 #include "RecoEcal/EgammaCoreTools/interface/EcalClusterLazyTools.h"
19 bendavid 1.43 #include "DataFormats/VertexReco/interface/VertexFwd.h"
20     #include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
21     #include "TrackingTools/TransientTrack/plugins/TransientTrackBuilderESProducer.h"
22 bendavid 1.44 #include "RecoVertex/GaussianSumVertexFit/interface/GsfVertexTrackCompatibilityEstimator.h"
23 bendavid 1.43 #include "TrackingTools/IPTools/interface/IPTools.h"
24     #include "RecoEgamma/EgammaTools/interface/ConversionFinder.h"
25     #include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
26     #include "MagneticField/Engine/interface/MagneticField.h"
27 loizides 1.35 #include "MitAna/DataTree/interface/ElectronCol.h"
28     #include "MitAna/DataTree/interface/Names.h"
29     #include "MitAna/DataTree/interface/Track.h"
30 loizides 1.33 #include "MitEdm/DataFormats/interface/RefToBaseToPtr.h"
31 loizides 1.35 #include "MitProd/ObjectService/interface/ObjectService.h"
32 bendavid 1.45 #include "MitEdm/DataFormats/interface/DecayPart.h"
33     #include "MitEdm/ConversionRejection/interface/ConversionMatcher.h"
34 sixie 1.13
35 loizides 1.1 using namespace std;
36     using namespace edm;
37     using namespace mithep;
38    
39 loizides 1.5 //--------------------------------------------------------------------------------------------------
40 loizides 1.28 FillerElectrons::FillerElectrons(const edm::ParameterSet &cfg, const char *name, bool active) :
41     BaseFiller(cfg,name,active),
42 loizides 1.1 edmName_(Conf().getUntrackedParameter<string>("edmName","pixelMatchGsfElectrons")),
43 bendavid 1.49 expectedHitsName_(Conf().getUntrackedParameter<string>("expectedHitsName","")),
44 loizides 1.1 mitName_(Conf().getUntrackedParameter<string>("mitName",Names::gkElectronBrn)),
45 loizides 1.10 gsfTrackMapName_(Conf().getUntrackedParameter<string>("gsfTrackMapName","")),
46     trackerTrackMapName_(Conf().getUntrackedParameter<string>("trackerTrackMapName","")),
47 sixie 1.12 barrelSuperClusterMapName_(Conf().getUntrackedParameter<string>("barrelSuperClusterMapName","")),
48     endcapSuperClusterMapName_(Conf().getUntrackedParameter<string>("endcapSuperClusterMapName","")),
49 bendavid 1.36 pfSuperClusterMapName_(Conf().getUntrackedParameter<string>("pfSuperClusterMapName","")),
50 loizides 1.16 eIDCutBasedTightName_(Conf().getUntrackedParameter<string>("eIDCutBasedTightName","eidTight")),
51 bendavid 1.43 eIDCutBasedLooseName_(Conf().getUntrackedParameter<string>("eIDCutBasedLooseName","eidLoose")),
52 bendavid 1.49 eIDLikelihoodName_(Conf().getUntrackedParameter<string>("eIDLikelihoodName","")),
53 bendavid 1.43 pvEdmName_(Conf().getUntrackedParameter<string>("pvEdmName","offlinePrimaryVertices")),
54     pvBSEdmName_(Conf().getUntrackedParameter<string>("pvEdmName","offlinePrimaryVerticesWithBS")),
55 loizides 1.8 electrons_(new mithep::ElectronArr(16)),
56 loizides 1.10 gsfTrackMap_(0),
57 sixie 1.11 trackerTrackMap_(0),
58 sixie 1.12 barrelSuperClusterMap_(0),
59     endcapSuperClusterMap_(0)
60 loizides 1.1 {
61     // Constructor.
62     }
63    
64 loizides 1.5 //--------------------------------------------------------------------------------------------------
65 loizides 1.1 FillerElectrons::~FillerElectrons()
66     {
67     // Destructor.
68 loizides 1.6
69     delete electrons_;
70 loizides 1.1 }
71    
72 loizides 1.5 //--------------------------------------------------------------------------------------------------
73 bendavid 1.42 void FillerElectrons::BookDataBlock(TreeWriter &tws)
74 loizides 1.1 {
75 loizides 1.10 // Add electron branch to our tree and get our maps.
76 loizides 1.1
77 loizides 1.29 tws.AddBranch(mitName_,&electrons_);
78     OS()->add<mithep::ElectronArr>(electrons_,mitName_);
79 loizides 1.10
80 loizides 1.29 if (!gsfTrackMapName_.empty()) {
81     gsfTrackMap_ = OS()->get<TrackMap>(gsfTrackMapName_);
82     if (gsfTrackMap_)
83     AddBranchDep(mitName_,gsfTrackMap_->GetBrName());
84     }
85     if (!trackerTrackMapName_.empty()) {
86     trackerTrackMap_ = OS()->get<TrackMap>(trackerTrackMapName_);
87     if (trackerTrackMap_)
88     AddBranchDep(mitName_,trackerTrackMap_->GetBrName());
89     }
90     if (!barrelSuperClusterMapName_.empty()) {
91     barrelSuperClusterMap_ = OS()->get<SuperClusterMap>(barrelSuperClusterMapName_);
92     if (barrelSuperClusterMap_)
93     AddBranchDep(mitName_,barrelSuperClusterMap_->GetBrName());
94     }
95     if (!endcapSuperClusterMapName_.empty()) {
96     endcapSuperClusterMap_ = OS()->get<SuperClusterMap>(endcapSuperClusterMapName_);
97     if (endcapSuperClusterMap_)
98     AddBranchDep(mitName_,endcapSuperClusterMap_->GetBrName());
99     }
100 bendavid 1.36 if (!pfSuperClusterMapName_.empty()) {
101     pfSuperClusterMap_ = OS()->get<SuperClusterMap>(pfSuperClusterMapName_);
102     if (pfSuperClusterMap_)
103     AddBranchDep(mitName_,pfSuperClusterMap_->GetBrName());
104     }
105 loizides 1.1 }
106    
107 loizides 1.5 //--------------------------------------------------------------------------------------------------
108 loizides 1.29 void FillerElectrons::FillDataBlock(const edm::Event &event, const edm::EventSetup &setup)
109 loizides 1.1 {
110     // Fill electrons from edm collection into our collection.
111    
112 bendavid 1.26 electrons_->Delete();
113 loizides 1.1
114 sixie 1.11 Handle<reco::GsfElectronCollection> hElectronProduct;
115 loizides 1.7 GetProduct(edmName_, hElectronProduct, event);
116 loizides 1.1
117 loizides 1.29 // handles to get the electron ID information
118 peveraer 1.47 Handle<edm::ValueMap<float> > eidLooseMap;
119     GetProduct(eIDCutBasedLooseName_, eidLooseMap, event);
120     Handle<edm::ValueMap<float> > eidTightMap;
121 peveraer 1.48 GetProduct(eIDCutBasedTightName_, eidTightMap, event);
122 bendavid 1.49 edm::Handle<edm::ValueMap<float> > eidLikelihoodMap;
123     if (!eIDLikelihoodName_.empty()) {
124     GetProduct(eIDLikelihoodName_, eidLikelihoodMap, event);
125     }
126    
127 bendavid 1.43 edm::Handle<reco::VertexCollection> hVertex;
128     event.getByLabel(pvEdmName_, hVertex);
129     const reco::VertexCollection *pvCol = hVertex.product();
130    
131     edm::Handle<reco::VertexCollection> hVertexBS;
132     event.getByLabel(pvBSEdmName_, hVertexBS);
133     const reco::VertexCollection *pvBSCol = hVertexBS.product();
134    
135     edm::Handle<reco::TrackCollection> hGeneralTracks;
136     event.getByLabel("generalTracks", hGeneralTracks);
137     //const reco::VertexCollection *trackCol = hGeneralTracks.product();
138 bendavid 1.45
139     edm::Handle<std::vector<mitedm::DecayPart> > hConversions;
140     event.getByLabel("mvfConversionRemoval", hConversions);
141    
142     mitedm::ConversionMatcher convMatcher;
143 bendavid 1.50
144 bendavid 1.43 edm::ESHandle<TransientTrackBuilder> hTransientTrackBuilder;
145     setup.get<TransientTrackRecord>().get("TransientTrackBuilder",hTransientTrackBuilder);
146     const TransientTrackBuilder *transientTrackBuilder = hTransientTrackBuilder.product();
147 bendavid 1.19
148 bendavid 1.44 GsfVertexTrackCompatibilityEstimator gsfEstimator;
149    
150 bendavid 1.43 //Get Magnetic Field from event setup, taking value at (0,0,0)
151 bendavid 1.50 // edm::ESHandle<MagneticField> magneticField;
152     // setup.get<IdealMagneticFieldRecord>().get(magneticField);
153     // const double bfield = magneticField->inTesla(GlobalPoint(0.,0.,0.)).z();
154 bendavid 1.43
155 sixie 1.11 const reco::GsfElectronCollection inElectrons = *(hElectronProduct.product());
156 loizides 1.29 // loop over electrons
157 sixie 1.11 for (reco::GsfElectronCollection::const_iterator iM = inElectrons.begin();
158 loizides 1.1 iM != inElectrons.end(); ++iM) {
159 sixie 1.13
160 loizides 1.29 // the index and Ref are needed for the eID association Map
161 sixie 1.13 unsigned int iElectron = iM - inElectrons.begin();
162     reco::GsfElectronRef eRef(hElectronProduct, iElectron);
163    
164 loizides 1.8 mithep::Electron *outElectron = electrons_->AddNew();
165 bendavid 1.31
166     outElectron->SetPtEtaPhi(iM->pt(),iM->eta(),iM->phi());
167 sixie 1.12
168 bendavid 1.40 outElectron->SetCharge(iM->charge());
169     outElectron->SetScPixCharge(iM->scPixCharge());
170    
171 loizides 1.30 outElectron->SetESuperClusterOverP(iM->eSuperClusterOverP());
172     outElectron->SetESeedClusterOverPout(iM->eSeedClusterOverPout());
173     outElectron->SetPIn(iM->trackMomentumAtVtx().R());
174 loizides 1.29 outElectron->SetPOut(iM->trackMomentumOut().R());
175 loizides 1.30 outElectron->SetDeltaEtaSuperClusterTrackAtVtx(iM->deltaEtaSuperClusterTrackAtVtx());
176     outElectron->SetDeltaEtaSeedClusterTrackAtCalo(iM->deltaEtaSeedClusterTrackAtCalo());
177     outElectron->SetDeltaPhiSuperClusterTrackAtVtx(iM->deltaPhiSuperClusterTrackAtVtx());
178     outElectron->SetDeltaPhiSeedClusterTrackAtCalo(iM->deltaPhiSeedClusterTrackAtCalo());
179     outElectron->SetIsEnergyScaleCorrected(iM->isEnergyScaleCorrected());
180     outElectron->SetIsMomentumCorrected(iM->isMomentumCorrected());
181 bendavid 1.36 outElectron->SetNumberOfClusters(iM->basicClustersSize());
182 loizides 1.30 outElectron->SetClassification(iM->classification());
183 bendavid 1.37 outElectron->SetFBrem(iM->fbrem());
184    
185 loizides 1.38 // pflow electron stuff
186 bendavid 1.41 outElectron->SetIsEcalDriven(iM->ecalDrivenSeed());
187     outElectron->SetIsTrackerDriven(iM->trackerDrivenSeed());
188 bendavid 1.37 outElectron->SetMva(iM->mva());
189    
190 loizides 1.29 // shower shape variables
191 bendavid 1.37 outElectron->SetE15(iM->e1x5());
192     outElectron->SetE25Max(iM->e2x5Max());
193     outElectron->SetE55(iM->e5x5());
194     outElectron->SetCovEtaEta(iM->sigmaEtaEta());
195     outElectron->SetCoviEtaiEta(iM->sigmaIetaIeta());
196     outElectron->SetHadronicOverEm(iM->hcalOverEcal());
197     outElectron->SetHcalDepth1OverEcal(iM->hcalDepth1OverEcal());
198     outElectron->SetHcalDepth2OverEcal(iM->hcalDepth2OverEcal());
199 sixie 1.12
200 loizides 1.38 // fill isolation variables for both cone sizes
201 bendavid 1.37 outElectron->SetEcalRecHitIsoDr04(iM->dr04EcalRecHitSumEt());
202     outElectron->SetHcalDepth1TowerSumEtDr04(iM->dr04HcalDepth1TowerSumEt());
203     outElectron->SetHcalDepth2TowerSumEtDr04(iM->dr04HcalDepth2TowerSumEt());
204     outElectron->SetTrackIsolationDr04(iM->dr04TkSumPt());
205     outElectron->SetEcalRecHitIsoDr03(iM->dr03EcalRecHitSumEt());
206     outElectron->SetHcalTowerSumEtDr03(iM->dr03HcalTowerSumEt());
207     outElectron->SetHcalDepth1TowerSumEtDr03(iM->dr03HcalDepth1TowerSumEt());
208     outElectron->SetHcalDepth2TowerSumEtDr03(iM->dr03HcalDepth2TowerSumEt());
209     outElectron->SetTrackIsolationDr03(iM->dr03TkSumPt());
210    
211 loizides 1.38 // fiducial flags
212 bendavid 1.37 outElectron->SetIsEB(iM->isEB());
213     outElectron->SetIsEE(iM->isEE());
214     outElectron->SetIsEBEEGap(iM->isEBEEGap());
215     outElectron->SetIsEBEtaGap(iM->isEBEtaGap());
216     outElectron->SetIsEBPhiGap(iM->isEBPhiGap());
217     outElectron->SetIsEEDeeGap(iM->isEEDeeGap());
218     outElectron->SetIsEERingGap(iM->isEERingGap());
219 sixie 1.12
220 loizides 1.38 // gsf-tracker match quality
221 bendavid 1.37 outElectron->SetFracSharedHits(iM->shFracInnerHits());
222 sixie 1.15
223 loizides 1.29 // make proper links to Tracks and Super Clusters
224 bendavid 1.49 if (gsfTrackMap_ && iM->gsfTrack().isNonnull()) {
225 bendavid 1.18 outElectron->SetGsfTrk(gsfTrackMap_->GetMit(refToPtr(iM->gsfTrack())));
226 bendavid 1.49 }
227     // make links to ambigous gsf tracks
228     if (gsfTrackMap_) {
229     for (reco::GsfTrackRefVector::const_iterator agsfi = iM->ambiguousGsfTracksBegin(); agsfi != iM->ambiguousGsfTracksEnd(); ++agsfi) {
230     outElectron->AddAmbiguousGsfTrack(gsfTrackMap_->GetMit(refToPtr(*agsfi)));
231     }
232     }
233    
234     // make tracker track links,
235 bendavid 1.37 if (trackerTrackMap_ && iM->closestCtfTrackRef().isNonnull()) {
236 bendavid 1.36 outElectron->SetTrackerTrk(trackerTrackMap_->GetMit(refToPtr(iM->closestCtfTrackRef())));
237 bendavid 1.19 }
238 loizides 1.38 if (barrelSuperClusterMap_ && endcapSuperClusterMap_ &&
239     pfSuperClusterMap_ && iM->superCluster().isNonnull()) {
240 bendavid 1.37 if(barrelSuperClusterMap_->HasMit(iM->superCluster())) {
241 sixie 1.12 outElectron->SetSuperCluster(barrelSuperClusterMap_->GetMit(iM->superCluster()));
242 bendavid 1.36 }
243     else if (endcapSuperClusterMap_->HasMit(iM->superCluster())) {
244 sixie 1.12 outElectron->SetSuperCluster(endcapSuperClusterMap_->GetMit(iM->superCluster()));
245     }
246 bendavid 1.36 else if (pfSuperClusterMap_->HasMit(iM->superCluster())) {
247     outElectron->SetSuperCluster(pfSuperClusterMap_->GetMit(iM->superCluster()));
248     }
249     else throw edm::Exception(edm::errors::Configuration, "FillerElectrons:FillDataBlock()\n")
250     << "Error! SuperCluster reference in unmapped collection " << edmName_ << endl;
251     }
252 sixie 1.13
253 bendavid 1.43 //compute impact parameter with respect to PV
254     if (iM->gsfTrack().isNonnull()) {
255     const reco::TransientTrack &tt = transientTrackBuilder->build(iM->gsfTrack());
256    
257     const std::pair<bool,Measurement1D> &d0pv = IPTools::absoluteTransverseImpactParameter(tt,pvCol->at(0));
258     if (d0pv.first) {
259     outElectron->SetD0PV(d0pv.second.value());
260     outElectron->SetD0PVErr(d0pv.second.error());
261     }
262 bendavid 1.44 else {
263     outElectron->SetD0PV(-99.0);
264     }
265    
266 bendavid 1.43
267     const std::pair<bool,Measurement1D> &ip3dpv = IPTools::absoluteImpactParameter3D(tt,pvCol->at(0));
268     if (ip3dpv.first) {
269     outElectron->SetIp3dPV(ip3dpv.second.value());
270     outElectron->SetIp3dPVErr(ip3dpv.second.error());
271     }
272 bendavid 1.44 else {
273     outElectron->SetIp3dPV(-99.0);
274     }
275 bendavid 1.43
276     const std::pair<bool,Measurement1D> &d0pvbs = IPTools::absoluteTransverseImpactParameter(tt,pvBSCol->at(0));
277     if (d0pvbs.first) {
278     outElectron->SetD0PVBS(d0pvbs.second.value());
279     outElectron->SetD0PVBSErr(d0pvbs.second.error());
280     }
281 bendavid 1.44 else {
282     outElectron->SetD0PVBS(-99.0);
283     }
284 bendavid 1.43
285     const std::pair<bool,Measurement1D> &ip3dpvbs = IPTools::absoluteImpactParameter3D(tt,pvBSCol->at(0));
286     if (ip3dpvbs.first) {
287     outElectron->SetIp3dPVBS(ip3dpvbs.second.value());
288     outElectron->SetIp3dPVBSErr(ip3dpvbs.second.error());
289     }
290 bendavid 1.44 else {
291     outElectron->SetIp3dPVBS(-99.0);
292     }
293    
294 bendavid 1.50
295 bendavid 1.43 if (verbose_>1) {
296     printf("gsf track pt = %5f\n",iM->gsfTrack()->pt());
297     printf("gsf track mode pt = %5f\n",iM->gsfTrack()->ptMode());
298     printf("ttrack pt = %5f\n",tt.initialFreeState().momentum().perp());
299     //printf("ttrackgsf pt = %5f\n",ttgsf.innermostMeasurementState().globalMomentum().perp());
300 bendavid 1.44 printf("ip3dpv reduced chisquared = %5f, probability = %5f\n", ip3dpv.second.value()/ip3dpv.second.error(), TMath::Prob(ip3dpv.second.value()/ip3dpv.second.error(),1));
301     //printf("gsf reduced chisquared = %5f, probability = %5f\n", pvGsfCompat.second/2, TMath::Prob(pvGsfCompat.second,2));
302 bendavid 1.43 }
303    
304     }
305    
306     //fill conversion partner track info
307 bendavid 1.50 outElectron->SetConvPartnerDCotTheta(iM->convDcot());
308     outElectron->SetConvPartnerDist(iM->convDist());
309     outElectron->SetConvPartnerRadius(iM->convRadius());
310     reco::TrackBaseRef convTrackRef = iM->convPartner();
311     if (convTrackRef.isNonnull()) {
312     if ( dynamic_cast<const reco::GsfTrack*>(convTrackRef.get()) && gsfTrackMap_ ) {
313     outElectron->SetConvPartnerTrk(gsfTrackMap_->GetMit(mitedm::refToBaseToPtr(convTrackRef)));
314     }
315     else if (trackerTrackMap_) {
316     outElectron->SetConvPartnerTrk(trackerTrackMap_->GetMit(mitedm::refToBaseToPtr(convTrackRef)));
317     }
318 bendavid 1.45 }
319 bendavid 1.43
320 bendavid 1.50
321    
322 loizides 1.29 // fill Electron ID information
323 peveraer 1.47 outElectron->SetPassLooseID((*eidLooseMap)[eRef]);
324     outElectron->SetPassTightID((*eidTightMap)[eRef]);
325 bendavid 1.49 if (!eIDLikelihoodName_.empty()) {
326     outElectron->SetIDLikelihood((*eidLikelihoodMap)[eRef]);
327     }
328    
329     // fill corrected expected inner hits
330 bendavid 1.50 if(iM->gsfTrack().isNonnull()) {
331     outElectron->SetCorrectedNExpectedHitsInner(iM->gsfTrack()->trackerExpectedHitsInner().numberOfHits());
332 bendavid 1.49 }
333 bendavid 1.31
334 bendavid 1.45 //fill additional conversion flag
335     outElectron->SetMatchesVertexConversion(convMatcher.matchesGoodConversion(*iM,hConversions));
336    
337 bendavid 1.31 if (verbose_>1) {
338     double recomass = sqrt(iM->energy()*iM->energy() - iM->p()*iM->p());
339 loizides 1.32 printf(" mithep::Electron, pt=%5f, eta=%5f, phi=%5f, energy=%5f, p=%5f, mass=%5f\n",
340     outElectron->Pt(), outElectron->Eta(), outElectron->Phi(),
341     outElectron->E(), outElectron->P(), outElectron->Mass());
342     printf("reco::GsfElectron , pt=%5f, eta=%5f, phi=%5f, energy=%5f, p=%5f, mass=%5f\n",
343     iM->pt(), iM->eta(), iM->phi(), iM->energy(), iM->p(), recomass);
344 bendavid 1.31 }
345 sixie 1.12 }
346 loizides 1.1 electrons_->Trim();
347     }