ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillerElectrons.cc
Revision: 1.61.2.3
Committed: Fri May 31 22:58:45 2013 UTC (11 years, 11 months ago) by pharris
Content type: text/plain
Branch: Mit_025c_branch
CVS Tags: Mit_025c_branch1
Changes since 1.61.2.2: +20 -15 lines
Log Message:
updated 42X for Tau

File Contents

# User Rev Content
1 pharris 1.61.2.3 // $Id: FillerElectrons.cc,v 1.61.2.2 2012/05/23 03:24:07 paus Exp $
2 paus 1.61.2.1
3 pharris 1.59 #include "FWCore/MessageLogger/interface/MessageLogger.h"
4 loizides 1.1 #include "MitProd/TreeFiller/interface/FillerElectrons.h"
5     #include "DataFormats/TrackReco/interface/Track.h"
6     #include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
7     #include "DataFormats/TrackReco/interface/TrackFwd.h"
8 sixie 1.11 #include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
9     #include "DataFormats/EgammaCandidates/interface/GsfElectronFwd.h"
10     #include "DataFormats/EgammaReco/interface/ClusterShape.h"
11     #include "DataFormats/EgammaReco/interface/BasicClusterShapeAssociation.h"
12 bendavid 1.18 #include "DataFormats/Common/interface/RefToPtr.h"
13 loizides 1.35 #include "DataFormats/Common/interface/ValueMap.h"
14     #include "AnalysisDataFormats/Egamma/interface/ElectronID.h"
15     #include "AnalysisDataFormats/Egamma/interface/ElectronIDAssociation.h"
16 sixie 1.11 #include "RecoEgamma/EgammaIsolationAlgos/interface/ElectronTkIsolation.h"
17     #include "RecoEgamma/EgammaIsolationAlgos/interface/EgammaEcalIsolation.h"
18 sixie 1.14 #include "RecoEgamma/EgammaIsolationAlgos/interface/EgammaTowerIsolation.h"
19 loizides 1.35 #include "RecoEcal/EgammaCoreTools/interface/EcalClusterLazyTools.h"
20 bendavid 1.43 #include "DataFormats/VertexReco/interface/VertexFwd.h"
21     #include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
22     #include "TrackingTools/TransientTrack/plugins/TransientTrackBuilderESProducer.h"
23 bendavid 1.44 #include "RecoVertex/GaussianSumVertexFit/interface/GsfVertexTrackCompatibilityEstimator.h"
24 bendavid 1.43 #include "TrackingTools/IPTools/interface/IPTools.h"
25     #include "RecoEgamma/EgammaTools/interface/ConversionFinder.h"
26     #include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
27     #include "MagneticField/Engine/interface/MagneticField.h"
28 loizides 1.35 #include "MitAna/DataTree/interface/ElectronCol.h"
29     #include "MitAna/DataTree/interface/Names.h"
30     #include "MitAna/DataTree/interface/Track.h"
31 loizides 1.33 #include "MitEdm/DataFormats/interface/RefToBaseToPtr.h"
32 loizides 1.35 #include "MitProd/ObjectService/interface/ObjectService.h"
33 bendavid 1.45 #include "MitEdm/DataFormats/interface/DecayPart.h"
34     #include "MitEdm/ConversionRejection/interface/ConversionMatcher.h"
35 bendavid 1.51 #include "RecoVertex/VertexTools/interface/LinearizedTrackStateFactory.h"
36     #include "RecoVertex/VertexTools/interface/VertexTrackFactory.h"
37     #include "RecoVertex/VertexPrimitives/interface/VertexTrack.h"
38     #include "RecoVertex/VertexPrimitives/interface/CachingVertex.h"
39     #include "RecoVertex/KalmanVertexFit/interface/KalmanVertexUpdator.h"
40 ksung 1.57 #include "MitEdm/Tools/interface/VertexReProducer.h"
41 pharris 1.61.2.3 //#include "RecoEgamma/EgammaTools/interface/ggPFClusters.h"
42 paus 1.61.2.1 // #include "RecoEgamma/EgammaTools/interface/ggPFClusters.h"
43     #include "Geometry/CaloGeometry/interface/CaloGeometry.h"
44     #include "Geometry/CaloTopology/interface/CaloSubdetectorTopology.h"
45     #include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
46     #include "Geometry/Records/interface/CaloGeometryRecord.h"
47     #include "FWCore/Framework/interface/ESHandle.h"
48     #include "DataFormats/EcalDetId/interface/EcalSubdetector.h"
49     #include "DataFormats/Math/interface/deltaR.h"
50 sixie 1.13
51 loizides 1.1 using namespace std;
52     using namespace edm;
53     using namespace mithep;
54    
55 loizides 1.5 //--------------------------------------------------------------------------------------------------
56 loizides 1.28 FillerElectrons::FillerElectrons(const edm::ParameterSet &cfg, const char *name, bool active) :
57 paus 1.61.2.1 BaseFiller (cfg,name,active),
58     edmName_ (Conf().getUntrackedParameter<string>("edmName","pixelMatchGsfElectrons")),
59     expectedHitsName_ (Conf().getUntrackedParameter<string>("expectedHitsName","")),
60     mitName_ (Conf().getUntrackedParameter<string>("mitName",Names::gkElectronBrn)),
61     gsfTrackMapName_ (Conf().getUntrackedParameter<string>("gsfTrackMapName","")),
62     trackerTrackMapName_ (Conf().getUntrackedParameter<string>("trackerTrackMapName","")),
63 sixie 1.12 barrelSuperClusterMapName_(Conf().getUntrackedParameter<string>("barrelSuperClusterMapName","")),
64     endcapSuperClusterMapName_(Conf().getUntrackedParameter<string>("endcapSuperClusterMapName","")),
65 paus 1.61.2.1 checkClusterActive_ (Conf().getUntrackedParameter<bool>("requireClusterAndGsfMap",true)),
66     pfSuperClusterMapName_ (Conf().getUntrackedParameter<string>("pfSuperClusterMapName","")),
67     pfClusterMapName_ (Conf().getUntrackedParameter<string>("pfClusterMapName","")),
68     electronMapName_ (Conf().getUntrackedParameter<string>("electronMapName","")),
69     eIDCutBasedTightName_ (Conf().getUntrackedParameter<string>("eIDCutBasedTightName","eidTight")),
70     eIDCutBasedLooseName_ (Conf().getUntrackedParameter<string>("eIDCutBasedLooseName","eidLoose")),
71     eIDLikelihoodName_ (Conf().getUntrackedParameter<string>("eIDLikelihoodName","")),
72     pvEdmName_ (Conf().getUntrackedParameter<string>("pvEdmName","offlinePrimaryVertices")),
73     pvBSEdmName_ (Conf().getUntrackedParameter<string>("pvBSEdmName","offlinePrimaryVerticesWithBS")),
74     recomputeConversionInfo_ (Conf().getUntrackedParameter<bool>("recomputeConversionInfo",false)),
75     fitUnbiasedVertex_ (Conf().getUntrackedParameter<bool>("fitUnbiasedVertex",true)),
76 pharris 1.61.2.3 useHcalTowers_ (Conf().getUntrackedParameter<bool>("useHcalTowers",false)),
77 paus 1.61.2.1 electronMap_ (new mithep::ElectronMap),
78     electrons_ (new mithep::ElectronArr(16)),
79     gsfTrackMap_ (0),
80     trackerTrackMap_ (0),
81     barrelSuperClusterMap_ (0),
82     endcapSuperClusterMap_ (0),
83     pfSuperClusterMap_ (0),
84     pfClusterMap_ (0)
85 loizides 1.1 {
86     // Constructor.
87 paus 1.61.2.1
88     // patch 4x
89     hcalCfg_.hOverEConeSize = 0.15;
90     hcalCfg_.useTowers = true;
91     hcalCfg_.hcalTowers = edm::InputTag("towerMaker");
92     hcalCfg_.hOverEPtMin = 0;
93 pharris 1.61.2.3 if(useHcalTowers_) hcalHelper_ = new ElectronHcalHelper(hcalCfg_);
94 loizides 1.1 }
95    
96 loizides 1.5 //--------------------------------------------------------------------------------------------------
97 loizides 1.1 FillerElectrons::~FillerElectrons()
98     {
99     // Destructor.
100 loizides 1.6
101     delete electrons_;
102 paus 1.61.2.1 delete electronMap_;
103 loizides 1.1 }
104    
105 loizides 1.5 //--------------------------------------------------------------------------------------------------
106 bendavid 1.42 void FillerElectrons::BookDataBlock(TreeWriter &tws)
107 loizides 1.1 {
108 loizides 1.10 // Add electron branch to our tree and get our maps.
109 loizides 1.1
110 loizides 1.29 tws.AddBranch(mitName_,&electrons_);
111     OS()->add<mithep::ElectronArr>(electrons_,mitName_);
112     if (!gsfTrackMapName_.empty()) {
113     gsfTrackMap_ = OS()->get<TrackMap>(gsfTrackMapName_);
114     if (gsfTrackMap_)
115     AddBranchDep(mitName_,gsfTrackMap_->GetBrName());
116     }
117     if (!trackerTrackMapName_.empty()) {
118     trackerTrackMap_ = OS()->get<TrackMap>(trackerTrackMapName_);
119     if (trackerTrackMap_)
120     AddBranchDep(mitName_,trackerTrackMap_->GetBrName());
121     }
122     if (!barrelSuperClusterMapName_.empty()) {
123     barrelSuperClusterMap_ = OS()->get<SuperClusterMap>(barrelSuperClusterMapName_);
124     if (barrelSuperClusterMap_)
125     AddBranchDep(mitName_,barrelSuperClusterMap_->GetBrName());
126     }
127     if (!endcapSuperClusterMapName_.empty()) {
128     endcapSuperClusterMap_ = OS()->get<SuperClusterMap>(endcapSuperClusterMapName_);
129     if (endcapSuperClusterMap_)
130     AddBranchDep(mitName_,endcapSuperClusterMap_->GetBrName());
131     }
132 bendavid 1.36 if (!pfSuperClusterMapName_.empty()) {
133     pfSuperClusterMap_ = OS()->get<SuperClusterMap>(pfSuperClusterMapName_);
134     if (pfSuperClusterMap_)
135     AddBranchDep(mitName_,pfSuperClusterMap_->GetBrName());
136     }
137 paus 1.61.2.1 if (!pfClusterMapName_.empty()) {
138     pfClusterMap_ = OS()->get<BasicClusterMap>(pfClusterMapName_);
139     if (pfClusterMap_)
140     AddBranchDep(mitName_,pfClusterMap_->GetBrName());
141     }
142 paus 1.61.2.2
143    
144     printf(" FillerElectrons::BookDataBlock -- Electron filler\n");
145 paus 1.61.2.1 if (!electronMapName_.empty()) {
146 paus 1.61.2.2 printf(" FillerElectrons::BookDataBlock -- Electron map name: %s (MIT) -- %s (mapName)\n\n",
147     mitName_.data(),electronMapName_.data());
148 paus 1.61.2.1 electronMap_->SetBrName(mitName_);
149     OS()->add<ElectronMap>(electronMap_,electronMapName_);
150     }
151 loizides 1.1 }
152    
153 loizides 1.5 //--------------------------------------------------------------------------------------------------
154 loizides 1.29 void FillerElectrons::FillDataBlock(const edm::Event &event, const edm::EventSetup &setup)
155 loizides 1.1 {
156     // Fill electrons from edm collection into our collection.
157    
158 paus 1.61.2.1 electrons_ ->Delete();
159     electronMap_->Reset();
160    
161 sixie 1.11 Handle<reco::GsfElectronCollection> hElectronProduct;
162 loizides 1.7 GetProduct(edmName_, hElectronProduct, event);
163 loizides 1.1
164 loizides 1.29 // handles to get the electron ID information
165 peveraer 1.47 Handle<edm::ValueMap<float> > eidLooseMap;
166     GetProduct(eIDCutBasedLooseName_, eidLooseMap, event);
167     Handle<edm::ValueMap<float> > eidTightMap;
168 peveraer 1.48 GetProduct(eIDCutBasedTightName_, eidTightMap, event);
169 bendavid 1.49 edm::Handle<edm::ValueMap<float> > eidLikelihoodMap;
170     if (!eIDLikelihoodName_.empty()) {
171     GetProduct(eIDLikelihoodName_, eidLikelihoodMap, event);
172     }
173    
174 bendavid 1.43 edm::Handle<reco::VertexCollection> hVertex;
175     event.getByLabel(pvEdmName_, hVertex);
176     const reco::VertexCollection *pvCol = hVertex.product();
177     edm::Handle<reco::VertexCollection> hVertexBS;
178     event.getByLabel(pvBSEdmName_, hVertexBS);
179     const reco::VertexCollection *pvBSCol = hVertexBS.product();
180     edm::Handle<reco::TrackCollection> hGeneralTracks;
181     event.getByLabel("generalTracks", hGeneralTracks);
182     //const reco::VertexCollection *trackCol = hGeneralTracks.product();
183 bendavid 1.45
184 bendavid 1.54 edm::Handle<reco::GsfTrackCollection> hGsfTracks;
185     event.getByLabel("electronGsfTracks", hGsfTracks);
186    
187 bendavid 1.45 edm::Handle<std::vector<mitedm::DecayPart> > hConversions;
188     event.getByLabel("mvfConversionRemoval", hConversions);
189    
190 paus 1.61.2.1 mitedm::ConversionMatcher convMatcher;
191 bendavid 1.50
192 paus 1.61.2.1 edm::ESHandle<TransientTrackBuilder> hTransientTrackBuilder;
193 bendavid 1.43 setup.get<TransientTrackRecord>().get("TransientTrackBuilder",hTransientTrackBuilder);
194 paus 1.61.2.1 const TransientTrackBuilder *transientTrackBuilder = hTransientTrackBuilder.product();
195    
196 pharris 1.61.2.3 //GsfVertexTrackCompatibilityEstimator gsfEstimator;
197     //LinearizedTrackStateFactory lTrackFactory;
198     //VertexTrackFactory<5> vTrackFactory;
199     //KalmanVertexUpdator<5> updator;
200 paus 1.61.2.1
201     // patch 4x
202 pharris 1.61.2.3 if(useHcalTowers_) hcalHelper_->checkSetup(setup);
203     if(useHcalTowers_) hcalHelper_->readEvent (const_cast<edm::Event&>(event));
204 paus 1.61.2.1
205     ////pf photon stuff
206     //edm::Handle< EcalRecHitCollection > pEBRecHits;
207     //event.getByLabel("reducedEcalRecHitsEB", pEBRecHits );
208     //edm::Handle< EcalRecHitCollection > pEERecHits;
209     //event.getByLabel( "reducedEcalRecHitsEE", pEERecHits );
210     //edm::ESHandle<CaloGeometry> pGeometry;
211     //setup.get<CaloGeometryRecord>().get(pGeometry);
212     //const CaloSubdetectorGeometry *geometryEB = pGeometry->getSubdetectorGeometry(DetId::Ecal, EcalBarrel);
213     //const CaloSubdetectorGeometry *geometryEE = pGeometry->getSubdetectorGeometry(DetId::Ecal, EcalEndcap);
214     //ggPFClusters pfclusters(pEBRecHits, pEERecHits, geometryEB, geometryEE);
215 bendavid 1.19
216 pharris 1.61 //Get Magnetic Field from event setup, taking value at (0,0,0)
217 bendavid 1.54 edm::ESHandle<MagneticField> magneticField;
218     setup.get<IdealMagneticFieldRecord>().get(magneticField);
219     const double bfield = magneticField->inTesla(GlobalPoint(0.,0.,0.)).z();
220 bendavid 1.43
221 sixie 1.11 const reco::GsfElectronCollection inElectrons = *(hElectronProduct.product());
222 loizides 1.29 // loop over electrons
223 sixie 1.11 for (reco::GsfElectronCollection::const_iterator iM = inElectrons.begin();
224 loizides 1.1 iM != inElectrons.end(); ++iM) {
225 sixie 1.13
226 loizides 1.29 // the index and Ref are needed for the eID association Map
227 sixie 1.13 unsigned int iElectron = iM - inElectrons.begin();
228     reco::GsfElectronRef eRef(hElectronProduct, iElectron);
229    
230 loizides 1.8 mithep::Electron *outElectron = electrons_->AddNew();
231 bendavid 1.31
232     outElectron->SetPtEtaPhi(iM->pt(),iM->eta(),iM->phi());
233 bendavid 1.40 outElectron->SetCharge(iM->charge());
234     outElectron->SetScPixCharge(iM->scPixCharge());
235 paus 1.61.2.1 outElectron->SetESuperClusterOverP(iM->eSuperClusterOverP());
236 loizides 1.30 outElectron->SetESeedClusterOverPout(iM->eSeedClusterOverPout());
237 paus 1.61.2.1 outElectron->SetEEleClusterOverPout(iM->eEleClusterOverPout());
238 loizides 1.30 outElectron->SetPIn(iM->trackMomentumAtVtx().R());
239 loizides 1.29 outElectron->SetPOut(iM->trackMomentumOut().R());
240 loizides 1.30 outElectron->SetDeltaEtaSuperClusterTrackAtVtx(iM->deltaEtaSuperClusterTrackAtVtx());
241     outElectron->SetDeltaEtaSeedClusterTrackAtCalo(iM->deltaEtaSeedClusterTrackAtCalo());
242     outElectron->SetDeltaPhiSuperClusterTrackAtVtx(iM->deltaPhiSuperClusterTrackAtVtx());
243     outElectron->SetDeltaPhiSeedClusterTrackAtCalo(iM->deltaPhiSeedClusterTrackAtCalo());
244     outElectron->SetIsEnergyScaleCorrected(iM->isEnergyScaleCorrected());
245 bendavid 1.55 //outElectron->SetIsMomentumCorrected(iM->isMomentumCorrected());
246 bendavid 1.36 outElectron->SetNumberOfClusters(iM->basicClustersSize());
247 loizides 1.30 outElectron->SetClassification(iM->classification());
248 bendavid 1.37 outElectron->SetFBrem(iM->fbrem());
249 paus 1.61.2.1 outElectron->SetEcalEnergy(iM->ecalEnergy());
250     outElectron->SetEcalEnergyError(iM->ecalEnergyError());
251 pharris 1.61.2.3 //outElectron->SetTrackMomentumError(iM->trackMomentumError());
252    
253 loizides 1.38 // pflow electron stuff
254 bendavid 1.41 outElectron->SetIsEcalDriven(iM->ecalDrivenSeed());
255     outElectron->SetIsTrackerDriven(iM->trackerDrivenSeed());
256 bendavid 1.37 outElectron->SetMva(iM->mva());
257    
258 loizides 1.29 // shower shape variables
259 bendavid 1.37 outElectron->SetE15(iM->e1x5());
260     outElectron->SetE25Max(iM->e2x5Max());
261     outElectron->SetE55(iM->e5x5());
262     outElectron->SetCovEtaEta(iM->sigmaEtaEta());
263     outElectron->SetCoviEtaiEta(iM->sigmaIetaIeta());
264     outElectron->SetHadronicOverEm(iM->hcalOverEcal());
265     outElectron->SetHcalDepth1OverEcal(iM->hcalDepth1OverEcal());
266     outElectron->SetHcalDepth2OverEcal(iM->hcalDepth2OverEcal());
267 paus 1.61.2.1
268     // patch 4x
269     //outElectron->SetHadOverEmTow(iM->hcalOverEcalBc());
270 pharris 1.61.2.3 if(useHcalTowers_) {
271     std::vector<CaloTowerDetId> hcalTowersBehindClusters = hcalHelper_->hcalTowersBehindClusters(*(iM->superCluster()));
272     float hcalDepth1 = hcalHelper_->hcalESumDepth1BehindClusters(hcalTowersBehindClusters);
273     float hcalDepth2 = hcalHelper_->hcalESumDepth2BehindClusters(hcalTowersBehindClusters);
274     outElectron->SetHadOverEmTow((hcalDepth1 + hcalDepth2)/iM->superCluster()->energy());
275     }
276 paus 1.61.2.1
277 loizides 1.38 // fill isolation variables for both cone sizes
278 bendavid 1.37 outElectron->SetEcalRecHitIsoDr04(iM->dr04EcalRecHitSumEt());
279     outElectron->SetHcalDepth1TowerSumEtDr04(iM->dr04HcalDepth1TowerSumEt());
280     outElectron->SetHcalDepth2TowerSumEtDr04(iM->dr04HcalDepth2TowerSumEt());
281     outElectron->SetTrackIsolationDr04(iM->dr04TkSumPt());
282 paus 1.61.2.1 outElectron->SetHCalIsoTowDr04( iM->dr04HcalTowerSumEtBc() );
283 bendavid 1.37 outElectron->SetEcalRecHitIsoDr03(iM->dr03EcalRecHitSumEt());
284     outElectron->SetHcalTowerSumEtDr03(iM->dr03HcalTowerSumEt());
285     outElectron->SetHcalDepth1TowerSumEtDr03(iM->dr03HcalDepth1TowerSumEt());
286     outElectron->SetHcalDepth2TowerSumEtDr03(iM->dr03HcalDepth2TowerSumEt());
287     outElectron->SetTrackIsolationDr03(iM->dr03TkSumPt());
288 paus 1.61.2.1 outElectron->SetHCalIsoTowDr03( iM->dr03HcalTowerSumEtBc() );
289    
290    
291 bendavid 1.56 //pflow isolation
292     outElectron->SetPFChargedHadronIso(iM->pfIsolationVariables().chargedHadronIso);
293     outElectron->SetPFChargedHadronIso(iM->pfIsolationVariables().neutralHadronIso);
294     outElectron->SetPFChargedHadronIso(iM->pfIsolationVariables().photonIso);
295    
296 loizides 1.38 // fiducial flags
297 bendavid 1.37 outElectron->SetIsEB(iM->isEB());
298     outElectron->SetIsEE(iM->isEE());
299     outElectron->SetIsEBEEGap(iM->isEBEEGap());
300     outElectron->SetIsEBEtaGap(iM->isEBEtaGap());
301     outElectron->SetIsEBPhiGap(iM->isEBPhiGap());
302     outElectron->SetIsEEDeeGap(iM->isEEDeeGap());
303     outElectron->SetIsEERingGap(iM->isEERingGap());
304 sixie 1.12
305 loizides 1.38 // gsf-tracker match quality
306 bendavid 1.37 outElectron->SetFracSharedHits(iM->shFracInnerHits());
307 sixie 1.15
308 loizides 1.29 // make proper links to Tracks and Super Clusters
309 bendavid 1.49 if (gsfTrackMap_ && iM->gsfTrack().isNonnull()) {
310 pharris 1.59 try{outElectron->SetGsfTrk(gsfTrackMap_->GetMit(refToPtr(iM->gsfTrack())));}
311     catch(...) {
312     if(checkClusterActive_) throw edm::Exception(edm::errors::Configuration, "FillerElectrons:FillDataBlock()\n")
313     << "Error! GSF track unmapped collection " << edmName_ << endl;
314     }
315 bendavid 1.49 }
316     // make links to ambigous gsf tracks
317     if (gsfTrackMap_) {
318     for (reco::GsfTrackRefVector::const_iterator agsfi = iM->ambiguousGsfTracksBegin(); agsfi != iM->ambiguousGsfTracksEnd(); ++agsfi) {
319 pharris 1.59 try{outElectron->AddAmbiguousGsfTrack(gsfTrackMap_->GetMit(refToPtr(*agsfi)));}
320     catch(...) {
321     if(checkClusterActive_) throw edm::Exception(edm::errors::Configuration, "FillerElectrons:FillDataBlock()\n")
322     << "Error! GSF track unmapped collection " << edmName_ << endl;
323     }
324 bendavid 1.49 }
325     }
326    
327     // make tracker track links,
328 bendavid 1.37 if (trackerTrackMap_ && iM->closestCtfTrackRef().isNonnull()) {
329 bendavid 1.36 outElectron->SetTrackerTrk(trackerTrackMap_->GetMit(refToPtr(iM->closestCtfTrackRef())));
330 bendavid 1.19 }
331 pharris 1.61
332 loizides 1.38 if (barrelSuperClusterMap_ && endcapSuperClusterMap_ &&
333     pfSuperClusterMap_ && iM->superCluster().isNonnull()) {
334 bendavid 1.37 if(barrelSuperClusterMap_->HasMit(iM->superCluster())) {
335 sixie 1.12 outElectron->SetSuperCluster(barrelSuperClusterMap_->GetMit(iM->superCluster()));
336 bendavid 1.36 }
337     else if (endcapSuperClusterMap_->HasMit(iM->superCluster())) {
338 sixie 1.12 outElectron->SetSuperCluster(endcapSuperClusterMap_->GetMit(iM->superCluster()));
339     }
340 bendavid 1.36 else if (pfSuperClusterMap_->HasMit(iM->superCluster())) {
341     outElectron->SetSuperCluster(pfSuperClusterMap_->GetMit(iM->superCluster()));
342     }
343 pharris 1.59 else if(checkClusterActive_) {
344     throw edm::Exception(edm::errors::Configuration, "FillerElectrons:FillDataBlock()\n")
345 bendavid 1.36 << "Error! SuperCluster reference in unmapped collection " << edmName_ << endl;
346 pharris 1.59 }
347 bendavid 1.36 }
348 paus 1.61.2.1
349     if (pfSuperClusterMap_ && iM->pflowSuperCluster().isNonnull()) {
350     outElectron->SetPFSuperCluster(pfSuperClusterMap_->GetMit(iM->pflowSuperCluster()));
351     }
352    
353     //find matching egamma supercluster first by ref, or by geometric matching if only pfsupercluster is linked
354     const reco::SuperCluster *egammasc = 0;
355     double mindr = 999.;
356     if (iM->superCluster().isNonnull() && iM->superCluster() != iM->pflowSuperCluster()) {
357     egammasc = iM->superCluster().get();
358     }
359     else {
360     for (SuperClusterMap::fwdMapType::const_iterator scit = barrelSuperClusterMap_->FwdMap().begin(); scit != barrelSuperClusterMap_->FwdMap().end(); ++scit) {
361     double dr = reco::deltaR(*iM->pflowSuperCluster(),*scit->first);
362     if (dr<0.1 && dr<mindr) {
363     egammasc = scit->first.get();
364     mindr = dr;
365     }
366     }
367     for (SuperClusterMap::fwdMapType::const_iterator scit = endcapSuperClusterMap_->FwdMap().begin(); scit != endcapSuperClusterMap_->FwdMap().end(); ++scit) {
368     double dr = reco::deltaR(*iM->pflowSuperCluster(),*scit->first);
369     if (dr<0.1 && dr<mindr) {
370     egammasc = scit->first.get();
371     mindr = dr;
372     }
373     }
374     }
375    
376     // Loosing propeor setting of the Matched energy in Bambu [ pfClusterMap_ -> pfCluster(MatchedEnergy) ]
377     ////tag overlapping energy of pflow clusters
378     //if (pfClusterMap_ && iM->pflowSuperCluster().isNonnull() && egammasc) {
379     // for (reco::CaloCluster_iterator pfcit = iM->pflowSuperCluster()->clustersBegin();
380     // pfcit!=iM->pflowSuperCluster()->clustersEnd(); ++pfcit) {
381     // float eoverlap = pfclusters.getPFSuperclusterOverlap( **pfcit, *egammasc );
382     // const_cast<mithep::BasicCluster*>(pfClusterMap_->GetMit(*pfcit))->SetMatchedEnergy(eoverlap);
383     // }
384     //}
385    
386     //compute NLayersWithMeasurement for associated ctf track
387 pharris 1.61.2.3
388 paus 1.61.2.1 if (iM->closestCtfTrackRef().isNonnull()) {
389     outElectron->SetCTFTrkNLayersWithMeasurement(iM->closestCtfTrackRef()->hitPattern().trackerLayersWithMeasurement());
390     } else {
391     outElectron->SetCTFTrkNLayersWithMeasurement(-1);
392     }
393    
394 bendavid 1.43 //compute impact parameter with respect to PV
395     if (iM->gsfTrack().isNonnull()) {
396     const reco::TransientTrack &tt = transientTrackBuilder->build(iM->gsfTrack());
397 bendavid 1.51
398     reco::TransientTrack ttckf;
399    
400 bendavid 1.58 reco::Vertex thevtx = pvCol->at(0);
401     reco::Vertex thevtxbs = pvBSCol->at(0);
402 bendavid 1.52
403 bendavid 1.58 reco::Vertex thevtxub = pvCol->at(0);
404     reco::Vertex thevtxubbs = pvBSCol->at(0);
405 pharris 1.61
406 bendavid 1.51 //check if closest ctf track is included in PV and if so, remove it before computing impact parameters and uncertainties
407     if (iM->closestCtfTrackRef().isNonnull()) {
408 pharris 1.61.2.3 ttckf = transientTrackBuilder->build(iM->closestCtfTrackRef());
409 ksung 1.57
410     //check if closest ctf track is included in PV and if so, remove it from the collection of
411     //tracks associated with the PV and perform a refit before computing impact parameters and uncertainties
412     reco::TrackCollection newTkCollection;
413     bool foundMatch = false;
414     for(reco::Vertex::trackRef_iterator itk = thevtx.tracks_begin(); itk!=thevtx.tracks_end(); itk++) {
415     if(iM->closestCtfTrack().ctfTrack.isNonnull()) {
416     bool refMatching = (itk->get() == &*(iM->closestCtfTrack().ctfTrack));
417     float shFraction = iM->closestCtfTrack().shFracInnerHits;
418     if(refMatching && shFraction > 0.5) {
419 pharris 1.61 foundMatch = true;
420     continue;
421 ksung 1.57 }
422     }
423     newTkCollection.push_back(*itk->get());
424     }
425 bendavid 1.51
426 pharris 1.61 if(foundMatch && fitUnbiasedVertex_) {
427 ksung 1.57 edm::Handle<reco::BeamSpot> bs;
428     event.getByLabel(edm::InputTag("offlineBeamSpot"),bs);
429 pharris 1.61
430     VertexReProducer revertex(hVertex,event); //Needs to be fixed
431    
432     edm::Handle<reco::BeamSpot> pvbeamspot;
433 ksung 1.57 event.getByLabel(revertex.inputBeamSpot(),pvbeamspot);
434 pharris 1.61 vector<TransientVertex> pvs = revertex.makeVertices(newTkCollection,*pvbeamspot,setup);
435     if(pvs.size()>0) {
436 bendavid 1.58 thevtxub = pvs.front(); // take the first in the list
437     }
438 ksung 1.57 VertexReProducer revertexbs(hVertexBS,event);
439     edm::Handle<reco::BeamSpot> pvbsbeamspot;
440     event.getByLabel(revertexbs.inputBeamSpot(),pvbsbeamspot);
441     vector<TransientVertex> pvbss = revertexbs.makeVertices(newTkCollection,*pvbsbeamspot,setup);
442 pharris 1.61 if(pvbss.size()>0) {
443 bendavid 1.58 thevtxubbs = pvbss.front(); // take the first in the list
444     }
445 ksung 1.57 }
446 bendavid 1.51 }
447     //preserve sign of transverse impact parameter (cross-product definition from track, not lifetime-signing)
448     const double gsfsign = ( (-iM->gsfTrack()->dxy(thevtx.position())) >=0 ) ? 1. : -1.;
449     const double gsfsignbs = ( (-iM->gsfTrack()->dxy(thevtxbs.position())) >=0 ) ? 1. : -1.;
450     const std::pair<bool,Measurement1D> &d0pv = IPTools::absoluteTransverseImpactParameter(tt,thevtx);
451 bendavid 1.43 if (d0pv.first) {
452 bendavid 1.51 outElectron->SetD0PV(gsfsign*d0pv.second.value());
453 bendavid 1.43 outElectron->SetD0PVErr(d0pv.second.error());
454     }
455 bendavid 1.44 else {
456 bendavid 1.51 outElectron->SetD0PV(-999.0);
457 bendavid 1.44 }
458    
459 bendavid 1.51 const std::pair<bool,Measurement1D> &ip3dpv = IPTools::absoluteImpactParameter3D(tt,thevtx);
460 bendavid 1.43 if (ip3dpv.first) {
461 bendavid 1.51 outElectron->SetIp3dPV(gsfsign*ip3dpv.second.value());
462 bendavid 1.43 outElectron->SetIp3dPVErr(ip3dpv.second.error());
463     }
464 bendavid 1.44 else {
465 bendavid 1.51 outElectron->SetIp3dPV(-999.0);
466 bendavid 1.44 }
467 bendavid 1.43
468 bendavid 1.51 const std::pair<bool,Measurement1D> &d0pvbs = IPTools::absoluteTransverseImpactParameter(tt,thevtxbs);
469 bendavid 1.43 if (d0pvbs.first) {
470 bendavid 1.51 outElectron->SetD0PVBS(gsfsignbs*d0pvbs.second.value());
471 bendavid 1.43 outElectron->SetD0PVBSErr(d0pvbs.second.error());
472     }
473 bendavid 1.44 else {
474 bendavid 1.51 outElectron->SetD0PVBS(-999.0);
475 bendavid 1.44 }
476 bendavid 1.43
477 bendavid 1.51 const std::pair<bool,Measurement1D> &ip3dpvbs = IPTools::absoluteImpactParameter3D(tt,thevtxbs);
478 bendavid 1.43 if (ip3dpvbs.first) {
479 bendavid 1.51 outElectron->SetIp3dPVBS(gsfsignbs*ip3dpvbs.second.value());
480 bendavid 1.43 outElectron->SetIp3dPVBSErr(ip3dpvbs.second.error());
481     }
482 bendavid 1.44 else {
483 bendavid 1.51 outElectron->SetIp3dPVBS(-999.0);
484 bendavid 1.44 }
485    
486 bendavid 1.58 const std::pair<bool,Measurement1D> &d0pvub = IPTools::absoluteTransverseImpactParameter(tt,thevtxub);
487     if (d0pvub.first) {
488     outElectron->SetD0PVUB(gsfsign*d0pvub.second.value());
489     outElectron->SetD0PVUBErr(d0pvub.second.error());
490     }
491     else {
492     outElectron->SetD0PVUB(-999.0);
493     }
494    
495    
496     const std::pair<bool,Measurement1D> &ip3dpvub = IPTools::absoluteImpactParameter3D(tt,thevtxub);
497     if (ip3dpvub.first) {
498     outElectron->SetIp3dPVUB(gsfsign*ip3dpvub.second.value());
499     outElectron->SetIp3dPVUBErr(ip3dpvub.second.error());
500     }
501     else {
502     outElectron->SetIp3dPVUB(-999.0);
503     }
504    
505     const std::pair<bool,Measurement1D> &d0pvubbs = IPTools::absoluteTransverseImpactParameter(tt,thevtxubbs);
506     if (d0pvubbs.first) {
507     outElectron->SetD0PVUBBS(gsfsignbs*d0pvubbs.second.value());
508     outElectron->SetD0PVUBBSErr(d0pvubbs.second.error());
509     }
510     else {
511     outElectron->SetD0PVUBBS(-999.0);
512     }
513    
514     const std::pair<bool,Measurement1D> &ip3dpvubbs = IPTools::absoluteImpactParameter3D(tt,thevtxubbs);
515     if (ip3dpvubbs.first) {
516     outElectron->SetIp3dPVUBBS(gsfsignbs*ip3dpvubbs.second.value());
517     outElectron->SetIp3dPVUBBSErr(ip3dpvubbs.second.error());
518     }
519     else {
520     outElectron->SetIp3dPVUBBS(-999.0);
521     }
522    
523 bendavid 1.51 if (iM->closestCtfTrackRef().isNonnull()) {
524    
525     const double ckfsign = ( (-iM->closestCtfTrackRef()->dxy(thevtx.position())) >=0 ) ? 1. : -1.;
526     const double ckfsignbs = ( (-iM->closestCtfTrackRef()->dxy(thevtxbs.position())) >=0 ) ? 1. : -1.;
527    
528     const std::pair<bool,Measurement1D> &d0pvckf = IPTools::absoluteTransverseImpactParameter(ttckf,thevtx);
529     if (d0pvckf.first) {
530     outElectron->SetD0PVCkf(ckfsign*d0pvckf.second.value());
531     outElectron->SetD0PVCkfErr(d0pvckf.second.error());
532     }
533     else {
534     outElectron->SetD0PVCkf(-999.0);
535     }
536    
537    
538     const std::pair<bool,Measurement1D> &ip3dpvckf = IPTools::absoluteImpactParameter3D(ttckf,thevtx);
539     if (ip3dpvckf.first) {
540     outElectron->SetIp3dPVCkf(ckfsign*ip3dpvckf.second.value());
541     outElectron->SetIp3dPVCkfErr(ip3dpvckf.second.error());
542     }
543     else {
544     outElectron->SetIp3dPVCkf(-999.0);
545     }
546    
547     const std::pair<bool,Measurement1D> &d0pvbsckf = IPTools::absoluteTransverseImpactParameter(ttckf,thevtxbs);
548     if (d0pvbsckf.first) {
549     outElectron->SetD0PVBSCkf(ckfsignbs*d0pvbsckf.second.value());
550     outElectron->SetD0PVBSCkfErr(d0pvbsckf.second.error());
551     }
552     else {
553     outElectron->SetD0PVBSCkf(-999.0);
554     }
555    
556     const std::pair<bool,Measurement1D> &ip3dpvbsckf = IPTools::absoluteImpactParameter3D(ttckf,thevtxbs);
557     if (ip3dpvbsckf.first) {
558     outElectron->SetIp3dPVBSCkf(ckfsignbs*ip3dpvbsckf.second.value());
559     outElectron->SetIp3dPVBSCkfErr(ip3dpvbsckf.second.error());
560     }
561     else {
562     outElectron->SetIp3dPVBSCkf(-999.0);
563     }
564 bendavid 1.58
565     const std::pair<bool,Measurement1D> &d0pvubckf = IPTools::absoluteTransverseImpactParameter(ttckf,thevtxub);
566     if (d0pvubckf.first) {
567     outElectron->SetD0PVUBCkf(ckfsign*d0pvubckf.second.value());
568     outElectron->SetD0PVUBCkfErr(d0pvubckf.second.error());
569     }
570     else {
571     outElectron->SetD0PVUBCkf(-999.0);
572     }
573    
574    
575     const std::pair<bool,Measurement1D> &ip3dpvubckf = IPTools::absoluteImpactParameter3D(ttckf,thevtxub);
576     if (ip3dpvubckf.first) {
577     outElectron->SetIp3dPVUBCkf(ckfsign*ip3dpvubckf.second.value());
578     outElectron->SetIp3dPVUBCkfErr(ip3dpvubckf.second.error());
579     }
580     else {
581     outElectron->SetIp3dPVUBCkf(-999.0);
582     }
583    
584     const std::pair<bool,Measurement1D> &d0pvubbsckf = IPTools::absoluteTransverseImpactParameter(ttckf,thevtxubbs);
585     if (d0pvubbsckf.first) {
586     outElectron->SetD0PVUBBSCkf(ckfsignbs*d0pvubbsckf.second.value());
587     outElectron->SetD0PVUBBSCkfErr(d0pvubbsckf.second.error());
588     }
589     else {
590     outElectron->SetD0PVUBBSCkf(-999.0);
591     }
592    
593     const std::pair<bool,Measurement1D> &ip3dpvubbsckf = IPTools::absoluteImpactParameter3D(ttckf,thevtxubbs);
594     if (ip3dpvubbsckf.first) {
595     outElectron->SetIp3dPVUBBSCkf(ckfsignbs*ip3dpvubbsckf.second.value());
596     outElectron->SetIp3dPVUBBSCkfErr(ip3dpvubbsckf.second.error());
597     }
598     else {
599     outElectron->SetIp3dPVUBBSCkf(-999.0);
600     }
601    
602 bendavid 1.51 }
603     else {
604     outElectron->SetD0PVCkf(-999.0);
605     outElectron->SetIp3dPVCkf(-999.0);
606     outElectron->SetD0PVBSCkf(-999.0);
607     outElectron->SetIp3dPVBSCkf(-999.0);
608 bendavid 1.58
609     outElectron->SetD0PVUBCkf(-999.0);
610     outElectron->SetIp3dPVUBCkf(-999.0);
611     outElectron->SetD0PVUBBSCkf(-999.0);
612     outElectron->SetIp3dPVUBBSCkf(-999.0);
613 bendavid 1.51 }
614    
615 bendavid 1.43 if (verbose_>1) {
616     printf("gsf track pt = %5f\n",iM->gsfTrack()->pt());
617     printf("gsf track mode pt = %5f\n",iM->gsfTrack()->ptMode());
618     printf("ttrack pt = %5f\n",tt.initialFreeState().momentum().perp());
619     //printf("ttrackgsf pt = %5f\n",ttgsf.innermostMeasurementState().globalMomentum().perp());
620 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));
621     //printf("gsf reduced chisquared = %5f, probability = %5f\n", pvGsfCompat.second/2, TMath::Prob(pvGsfCompat.second,2));
622 bendavid 1.43 }
623     }
624 bendavid 1.53
625 bendavid 1.43 //fill conversion partner track info
626 bendavid 1.54 if (recomputeConversionInfo_) {
627     ConversionFinder convFinder; outElectron->SetConvPartnerDCotTheta(iM->convDcot());
628     ConversionInfo convInfo = convFinder.getConversionInfo(*iM, hGeneralTracks, hGsfTracks, bfield);
629    
630     outElectron->SetConvFlag(convInfo.flag());
631     outElectron->SetConvPartnerDCotTheta(convInfo.dcot());
632     outElectron->SetConvPartnerDist(convInfo.dist());
633     outElectron->SetConvPartnerRadius(convInfo.radiusOfConversion());
634     reco::TrackRef ckfconvTrackRef = convInfo.conversionPartnerCtfTk();
635     reco::GsfTrackRef gsfconvTrackRef = convInfo.conversionPartnerGsfTk();
636    
637    
638     if ( gsfconvTrackRef.isNonnull() && gsfTrackMap_ ) {
639 pharris 1.59 try {outElectron->SetConvPartnerTrk(gsfTrackMap_->GetMit(edm::refToPtr(gsfconvTrackRef)));}
640     catch(...) {
641     if(checkClusterActive_) throw edm::Exception(edm::errors::Configuration, "FillerElectrons:FillDataBlock()\n")
642     << "Error! GSF track unmapped collection " << edmName_ << endl;
643     }
644    
645 bendavid 1.54 }
646     else if (ckfconvTrackRef.isNonnull() && trackerTrackMap_) {
647     outElectron->SetConvPartnerTrk(trackerTrackMap_->GetMit(edm::refToPtr(ckfconvTrackRef)));
648 bendavid 1.50 }
649 bendavid 1.54 }
650     else {
651     outElectron->SetConvFlag(iM->convFlags());
652     outElectron->SetConvPartnerDCotTheta(iM->convDcot());
653     outElectron->SetConvPartnerDist(iM->convDist());
654     outElectron->SetConvPartnerRadius(iM->convRadius());
655     reco::TrackBaseRef convTrackRef = iM->convPartner();
656     if (convTrackRef.isNonnull()) {
657     if ( dynamic_cast<const reco::GsfTrack*>(convTrackRef.get()) && gsfTrackMap_ ) {
658 pharris 1.59 try{outElectron->SetConvPartnerTrk(gsfTrackMap_->GetMit(mitedm::refToBaseToPtr(convTrackRef)));}
659     catch(...) {
660     if(checkClusterActive_) throw edm::Exception(edm::errors::Configuration, "FillerElectrons:FillDataBlock()\n")
661     << "Error! GSF track unmapped collection " << edmName_ << endl;
662     }
663 bendavid 1.54 }
664     else if (trackerTrackMap_) {
665     outElectron->SetConvPartnerTrk(trackerTrackMap_->GetMit(mitedm::refToBaseToPtr(convTrackRef)));
666     }
667 bendavid 1.50 }
668 bendavid 1.45 }
669 bendavid 1.50
670 loizides 1.29 // fill Electron ID information
671 peveraer 1.47 outElectron->SetPassLooseID((*eidLooseMap)[eRef]);
672     outElectron->SetPassTightID((*eidTightMap)[eRef]);
673 bendavid 1.49 if (!eIDLikelihoodName_.empty()) {
674     outElectron->SetIDLikelihood((*eidLikelihoodMap)[eRef]);
675     }
676     // fill corrected expected inner hits
677 bendavid 1.50 if(iM->gsfTrack().isNonnull()) {
678     outElectron->SetCorrectedNExpectedHitsInner(iM->gsfTrack()->trackerExpectedHitsInner().numberOfHits());
679 bendavid 1.49 }
680 bendavid 1.45 //fill additional conversion flag
681     outElectron->SetMatchesVertexConversion(convMatcher.matchesGoodConversion(*iM,hConversions));
682    
683 paus 1.61.2.1 // add electron to map
684     edm::Ptr<reco::GsfElectron> thePtr(hElectronProduct, iM - inElectrons.begin());
685     electronMap_->Add(thePtr, outElectron);
686    
687 bendavid 1.31 if (verbose_>1) {
688     double recomass = sqrt(iM->energy()*iM->energy() - iM->p()*iM->p());
689 loizides 1.32 printf(" mithep::Electron, pt=%5f, eta=%5f, phi=%5f, energy=%5f, p=%5f, mass=%5f\n",
690     outElectron->Pt(), outElectron->Eta(), outElectron->Phi(),
691     outElectron->E(), outElectron->P(), outElectron->Mass());
692     printf("reco::GsfElectron , pt=%5f, eta=%5f, phi=%5f, energy=%5f, p=%5f, mass=%5f\n",
693     iM->pt(), iM->eta(), iM->phi(), iM->energy(), iM->p(), recomass);
694 bendavid 1.31 }
695 pharris 1.59 }
696 loizides 1.1 electrons_->Trim();
697     }