ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillerElectrons.cc
Revision: 1.43
Committed: Thu May 6 17:31:24 2010 UTC (15 years ago) by bendavid
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_014pre3, Mit_014pre2, Mit_014pre1
Changes since 1.42: +76 -2 lines
Log Message:
Fill conversion partner and impact parameter signficance variables

File Contents

# User Rev Content
1 bendavid 1.43 // $Id: FillerElectrons.cc,v 1.42 2010/03/18 20:21:00 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     #include "TrackingTools/IPTools/interface/IPTools.h"
23     #include "RecoEgamma/EgammaTools/interface/ConversionFinder.h"
24     #include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
25     #include "MagneticField/Engine/interface/MagneticField.h"
26 loizides 1.35 #include "MitAna/DataTree/interface/ElectronCol.h"
27     #include "MitAna/DataTree/interface/Names.h"
28     #include "MitAna/DataTree/interface/Track.h"
29 loizides 1.33 #include "MitEdm/DataFormats/interface/RefToBaseToPtr.h"
30 loizides 1.35 #include "MitProd/ObjectService/interface/ObjectService.h"
31 sixie 1.13
32 loizides 1.1 using namespace std;
33     using namespace edm;
34     using namespace mithep;
35    
36 loizides 1.5 //--------------------------------------------------------------------------------------------------
37 loizides 1.28 FillerElectrons::FillerElectrons(const edm::ParameterSet &cfg, const char *name, bool active) :
38     BaseFiller(cfg,name,active),
39 loizides 1.1 edmName_(Conf().getUntrackedParameter<string>("edmName","pixelMatchGsfElectrons")),
40     mitName_(Conf().getUntrackedParameter<string>("mitName",Names::gkElectronBrn)),
41 loizides 1.10 gsfTrackMapName_(Conf().getUntrackedParameter<string>("gsfTrackMapName","")),
42     trackerTrackMapName_(Conf().getUntrackedParameter<string>("trackerTrackMapName","")),
43 sixie 1.12 barrelSuperClusterMapName_(Conf().getUntrackedParameter<string>("barrelSuperClusterMapName","")),
44     endcapSuperClusterMapName_(Conf().getUntrackedParameter<string>("endcapSuperClusterMapName","")),
45 bendavid 1.36 pfSuperClusterMapName_(Conf().getUntrackedParameter<string>("pfSuperClusterMapName","")),
46 loizides 1.16 eIDCutBasedTightName_(Conf().getUntrackedParameter<string>("eIDCutBasedTightName","eidTight")),
47 bendavid 1.43 eIDCutBasedLooseName_(Conf().getUntrackedParameter<string>("eIDCutBasedLooseName","eidLoose")),
48     pvEdmName_(Conf().getUntrackedParameter<string>("pvEdmName","offlinePrimaryVertices")),
49     pvBSEdmName_(Conf().getUntrackedParameter<string>("pvEdmName","offlinePrimaryVerticesWithBS")),
50 loizides 1.8 electrons_(new mithep::ElectronArr(16)),
51 loizides 1.10 gsfTrackMap_(0),
52 sixie 1.11 trackerTrackMap_(0),
53 sixie 1.12 barrelSuperClusterMap_(0),
54     endcapSuperClusterMap_(0)
55 loizides 1.1 {
56     // Constructor.
57     }
58    
59 loizides 1.5 //--------------------------------------------------------------------------------------------------
60 loizides 1.1 FillerElectrons::~FillerElectrons()
61     {
62     // Destructor.
63 loizides 1.6
64     delete electrons_;
65 loizides 1.1 }
66    
67 loizides 1.5 //--------------------------------------------------------------------------------------------------
68 bendavid 1.42 void FillerElectrons::BookDataBlock(TreeWriter &tws)
69 loizides 1.1 {
70 loizides 1.10 // Add electron branch to our tree and get our maps.
71 loizides 1.1
72 loizides 1.29 tws.AddBranch(mitName_,&electrons_);
73     OS()->add<mithep::ElectronArr>(electrons_,mitName_);
74 loizides 1.10
75 loizides 1.29 if (!gsfTrackMapName_.empty()) {
76     gsfTrackMap_ = OS()->get<TrackMap>(gsfTrackMapName_);
77     if (gsfTrackMap_)
78     AddBranchDep(mitName_,gsfTrackMap_->GetBrName());
79     }
80     if (!trackerTrackMapName_.empty()) {
81     trackerTrackMap_ = OS()->get<TrackMap>(trackerTrackMapName_);
82     if (trackerTrackMap_)
83     AddBranchDep(mitName_,trackerTrackMap_->GetBrName());
84     }
85     if (!barrelSuperClusterMapName_.empty()) {
86     barrelSuperClusterMap_ = OS()->get<SuperClusterMap>(barrelSuperClusterMapName_);
87     if (barrelSuperClusterMap_)
88     AddBranchDep(mitName_,barrelSuperClusterMap_->GetBrName());
89     }
90     if (!endcapSuperClusterMapName_.empty()) {
91     endcapSuperClusterMap_ = OS()->get<SuperClusterMap>(endcapSuperClusterMapName_);
92     if (endcapSuperClusterMap_)
93     AddBranchDep(mitName_,endcapSuperClusterMap_->GetBrName());
94     }
95 bendavid 1.36 if (!pfSuperClusterMapName_.empty()) {
96     pfSuperClusterMap_ = OS()->get<SuperClusterMap>(pfSuperClusterMapName_);
97     if (pfSuperClusterMap_)
98     AddBranchDep(mitName_,pfSuperClusterMap_->GetBrName());
99     }
100 loizides 1.1 }
101    
102 loizides 1.5 //--------------------------------------------------------------------------------------------------
103 loizides 1.29 void FillerElectrons::FillDataBlock(const edm::Event &event, const edm::EventSetup &setup)
104 loizides 1.1 {
105     // Fill electrons from edm collection into our collection.
106    
107 bendavid 1.26 electrons_->Delete();
108 loizides 1.1
109 sixie 1.11 Handle<reco::GsfElectronCollection> hElectronProduct;
110 loizides 1.7 GetProduct(edmName_, hElectronProduct, event);
111 loizides 1.1
112 loizides 1.29 // handles to get the electron ID information
113 sixie 1.13 Handle<edm::ValueMap<float> > eidLooseMap;
114     GetProduct(eIDCutBasedLooseName_, eidLooseMap, event);
115     Handle<edm::ValueMap<float> > eidTightMap;
116     GetProduct(eIDCutBasedTightName_, eidTightMap, event);
117 bendavid 1.43
118     edm::Handle<reco::VertexCollection> hVertex;
119     event.getByLabel(pvEdmName_, hVertex);
120     const reco::VertexCollection *pvCol = hVertex.product();
121    
122     edm::Handle<reco::VertexCollection> hVertexBS;
123     event.getByLabel(pvBSEdmName_, hVertexBS);
124     const reco::VertexCollection *pvBSCol = hVertexBS.product();
125    
126     edm::Handle<reco::TrackCollection> hGeneralTracks;
127     event.getByLabel("generalTracks", hGeneralTracks);
128     //const reco::VertexCollection *trackCol = hGeneralTracks.product();
129    
130     edm::ESHandle<TransientTrackBuilder> hTransientTrackBuilder;
131     setup.get<TransientTrackRecord>().get("TransientTrackBuilder",hTransientTrackBuilder);
132     const TransientTrackBuilder *transientTrackBuilder = hTransientTrackBuilder.product();
133 bendavid 1.19
134 bendavid 1.43 //Get Magnetic Field from event setup, taking value at (0,0,0)
135     edm::ESHandle<MagneticField> magneticField;
136     setup.get<IdealMagneticFieldRecord>().get(magneticField);
137     const double bfield = magneticField->inTesla(GlobalPoint(0.,0.,0.)).z();
138    
139 sixie 1.11 const reco::GsfElectronCollection inElectrons = *(hElectronProduct.product());
140 loizides 1.29 // loop over electrons
141 sixie 1.11 for (reco::GsfElectronCollection::const_iterator iM = inElectrons.begin();
142 loizides 1.1 iM != inElectrons.end(); ++iM) {
143 sixie 1.13
144 loizides 1.29 // the index and Ref are needed for the eID association Map
145 sixie 1.13 unsigned int iElectron = iM - inElectrons.begin();
146     reco::GsfElectronRef eRef(hElectronProduct, iElectron);
147    
148 loizides 1.8 mithep::Electron *outElectron = electrons_->AddNew();
149 bendavid 1.31
150     outElectron->SetPtEtaPhi(iM->pt(),iM->eta(),iM->phi());
151 sixie 1.12
152 bendavid 1.40 outElectron->SetCharge(iM->charge());
153     outElectron->SetScPixCharge(iM->scPixCharge());
154    
155 loizides 1.30 outElectron->SetESuperClusterOverP(iM->eSuperClusterOverP());
156     outElectron->SetESeedClusterOverPout(iM->eSeedClusterOverPout());
157     outElectron->SetPIn(iM->trackMomentumAtVtx().R());
158 loizides 1.29 outElectron->SetPOut(iM->trackMomentumOut().R());
159 loizides 1.30 outElectron->SetDeltaEtaSuperClusterTrackAtVtx(iM->deltaEtaSuperClusterTrackAtVtx());
160     outElectron->SetDeltaEtaSeedClusterTrackAtCalo(iM->deltaEtaSeedClusterTrackAtCalo());
161     outElectron->SetDeltaPhiSuperClusterTrackAtVtx(iM->deltaPhiSuperClusterTrackAtVtx());
162     outElectron->SetDeltaPhiSeedClusterTrackAtCalo(iM->deltaPhiSeedClusterTrackAtCalo());
163     outElectron->SetIsEnergyScaleCorrected(iM->isEnergyScaleCorrected());
164     outElectron->SetIsMomentumCorrected(iM->isMomentumCorrected());
165 bendavid 1.36 outElectron->SetNumberOfClusters(iM->basicClustersSize());
166 loizides 1.30 outElectron->SetClassification(iM->classification());
167 bendavid 1.37 outElectron->SetFBrem(iM->fbrem());
168    
169 loizides 1.38 // pflow electron stuff
170 bendavid 1.41 outElectron->SetIsEcalDriven(iM->ecalDrivenSeed());
171     outElectron->SetIsTrackerDriven(iM->trackerDrivenSeed());
172 bendavid 1.37 outElectron->SetMva(iM->mva());
173    
174 loizides 1.29 // shower shape variables
175 bendavid 1.37 outElectron->SetE15(iM->e1x5());
176     outElectron->SetE25Max(iM->e2x5Max());
177     outElectron->SetE55(iM->e5x5());
178     outElectron->SetCovEtaEta(iM->sigmaEtaEta());
179     outElectron->SetCoviEtaiEta(iM->sigmaIetaIeta());
180     outElectron->SetHadronicOverEm(iM->hcalOverEcal());
181     outElectron->SetHcalDepth1OverEcal(iM->hcalDepth1OverEcal());
182     outElectron->SetHcalDepth2OverEcal(iM->hcalDepth2OverEcal());
183 sixie 1.12
184 loizides 1.38 // fill isolation variables for both cone sizes
185 bendavid 1.37 outElectron->SetEcalRecHitIsoDr04(iM->dr04EcalRecHitSumEt());
186     outElectron->SetHcalDepth1TowerSumEtDr04(iM->dr04HcalDepth1TowerSumEt());
187     outElectron->SetHcalDepth2TowerSumEtDr04(iM->dr04HcalDepth2TowerSumEt());
188     outElectron->SetTrackIsolationDr04(iM->dr04TkSumPt());
189     outElectron->SetEcalRecHitIsoDr03(iM->dr03EcalRecHitSumEt());
190     outElectron->SetHcalTowerSumEtDr03(iM->dr03HcalTowerSumEt());
191     outElectron->SetHcalDepth1TowerSumEtDr03(iM->dr03HcalDepth1TowerSumEt());
192     outElectron->SetHcalDepth2TowerSumEtDr03(iM->dr03HcalDepth2TowerSumEt());
193     outElectron->SetTrackIsolationDr03(iM->dr03TkSumPt());
194    
195 loizides 1.38 // fiducial flags
196 bendavid 1.37 outElectron->SetIsEB(iM->isEB());
197     outElectron->SetIsEE(iM->isEE());
198     outElectron->SetIsEBEEGap(iM->isEBEEGap());
199     outElectron->SetIsEBEtaGap(iM->isEBEtaGap());
200     outElectron->SetIsEBPhiGap(iM->isEBPhiGap());
201     outElectron->SetIsEEDeeGap(iM->isEEDeeGap());
202     outElectron->SetIsEERingGap(iM->isEERingGap());
203 sixie 1.12
204 loizides 1.38 // gsf-tracker match quality
205 bendavid 1.37 outElectron->SetFracSharedHits(iM->shFracInnerHits());
206 sixie 1.15
207 loizides 1.29 // make proper links to Tracks and Super Clusters
208 loizides 1.1 if (gsfTrackMap_ && iM->gsfTrack().isNonnull())
209 bendavid 1.18 outElectron->SetGsfTrk(gsfTrackMap_->GetMit(refToPtr(iM->gsfTrack())));
210 loizides 1.29 // make tracker track links, relinking from gsf track associations if configured and
211     // link is otherwise absent
212 bendavid 1.37 if (trackerTrackMap_ && iM->closestCtfTrackRef().isNonnull()) {
213 bendavid 1.36 outElectron->SetTrackerTrk(trackerTrackMap_->GetMit(refToPtr(iM->closestCtfTrackRef())));
214 bendavid 1.19 }
215 loizides 1.38 if (barrelSuperClusterMap_ && endcapSuperClusterMap_ &&
216     pfSuperClusterMap_ && iM->superCluster().isNonnull()) {
217 bendavid 1.37 if(barrelSuperClusterMap_->HasMit(iM->superCluster())) {
218 sixie 1.12 outElectron->SetSuperCluster(barrelSuperClusterMap_->GetMit(iM->superCluster()));
219 bendavid 1.36 }
220     else if (endcapSuperClusterMap_->HasMit(iM->superCluster())) {
221 sixie 1.12 outElectron->SetSuperCluster(endcapSuperClusterMap_->GetMit(iM->superCluster()));
222     }
223 bendavid 1.36 else if (pfSuperClusterMap_->HasMit(iM->superCluster())) {
224     outElectron->SetSuperCluster(pfSuperClusterMap_->GetMit(iM->superCluster()));
225     }
226     else throw edm::Exception(edm::errors::Configuration, "FillerElectrons:FillDataBlock()\n")
227     << "Error! SuperCluster reference in unmapped collection " << edmName_ << endl;
228     }
229 sixie 1.13
230 bendavid 1.43 //compute impact parameter with respect to PV
231     if (iM->gsfTrack().isNonnull()) {
232     const reco::TransientTrack &tt = transientTrackBuilder->build(iM->gsfTrack());
233    
234     const std::pair<bool,Measurement1D> &d0pv = IPTools::absoluteTransverseImpactParameter(tt,pvCol->at(0));
235     if (d0pv.first) {
236     outElectron->SetD0PV(d0pv.second.value());
237     outElectron->SetD0PVErr(d0pv.second.error());
238     }
239    
240     const std::pair<bool,Measurement1D> &ip3dpv = IPTools::absoluteImpactParameter3D(tt,pvCol->at(0));
241     if (ip3dpv.first) {
242     outElectron->SetIp3dPV(ip3dpv.second.value());
243     outElectron->SetIp3dPVErr(ip3dpv.second.error());
244     }
245    
246     const std::pair<bool,Measurement1D> &d0pvbs = IPTools::absoluteTransverseImpactParameter(tt,pvBSCol->at(0));
247     if (d0pvbs.first) {
248     outElectron->SetD0PVBS(d0pvbs.second.value());
249     outElectron->SetD0PVBSErr(d0pvbs.second.error());
250     }
251    
252     const std::pair<bool,Measurement1D> &ip3dpvbs = IPTools::absoluteImpactParameter3D(tt,pvBSCol->at(0));
253     if (ip3dpvbs.first) {
254     outElectron->SetIp3dPVBS(ip3dpvbs.second.value());
255     outElectron->SetIp3dPVBSErr(ip3dpvbs.second.error());
256     }
257    
258     if (verbose_>1) {
259     printf("gsf track pt = %5f\n",iM->gsfTrack()->pt());
260     printf("gsf track mode pt = %5f\n",iM->gsfTrack()->ptMode());
261     printf("ttrack pt = %5f\n",tt.initialFreeState().momentum().perp());
262     //printf("ttrackgsf pt = %5f\n",ttgsf.innermostMeasurementState().globalMomentum().perp());
263     }
264    
265     }
266    
267     //fill conversion partner track info
268     ConversionFinder convFinder;
269     ConversionInfo convInfo = convFinder.getConversionInfo(*iM, hGeneralTracks, bfield);
270     outElectron->SetConvPartnerDCotTheta(convInfo.dcot());
271     outElectron->SetConvPartnerDist(convInfo.dist());
272     outElectron->SetConvPartnerRadius(convInfo.radiusOfConversion());
273    
274 loizides 1.29 // fill Electron ID information
275 sixie 1.13 outElectron->SetPassLooseID((*eidLooseMap)[eRef]);
276     outElectron->SetPassTightID((*eidTightMap)[eRef]);
277 bendavid 1.31
278     if (verbose_>1) {
279     double recomass = sqrt(iM->energy()*iM->energy() - iM->p()*iM->p());
280 loizides 1.32 printf(" mithep::Electron, pt=%5f, eta=%5f, phi=%5f, energy=%5f, p=%5f, mass=%5f\n",
281     outElectron->Pt(), outElectron->Eta(), outElectron->Phi(),
282     outElectron->E(), outElectron->P(), outElectron->Mass());
283     printf("reco::GsfElectron , pt=%5f, eta=%5f, phi=%5f, energy=%5f, p=%5f, mass=%5f\n",
284     iM->pt(), iM->eta(), iM->phi(), iM->energy(), iM->p(), recomass);
285 bendavid 1.31 }
286 sixie 1.12 }
287 loizides 1.1 electrons_->Trim();
288     }