TreeFiller/src/FillerTracks.cc

// $Id: FillerTracks.cc,v 1.38 2010/03/23 16:52:58 bendavid Exp $

#include "MitProd/TreeFiller/interface/FillerTracks.h"
#include "DataFormats/RecoCandidate/interface/TrackAssociation.h"
#include "DataFormats/TrackReco/interface/HitPattern.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "TrackingTools/TrackAssociator/interface/TrackDetectorAssociator.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateTransform.h"
#include "MitAna/DataTree/interface/Names.h"
#include "MitAna/DataTree/interface/TrackCol.h"
#include "MitProd/ObjectService/interface/ObjectService.h"
#include "MitEdm/DataFormats/interface/Types.h"

using namespace std;
using namespace edm;
using namespace mithep;

//--------------------------------------------------------------------------------------------------
FillerTracks::FillerTracks(const ParameterSet &cfg, const char *name, bool active) :
  BaseFiller(cfg,name,active),
  ecalAssocActive_(Conf().getUntrackedParameter<bool>("ecalAssocActive",0)),
  edmName_(Conf().getUntrackedParameter<string>("edmName","generalTracks")),
  mitName_(Conf().getUntrackedParameter<string>("mitName",Names::gkTrackBrn)),
  edmSimAssocName_(Conf().getUntrackedParameter<string>("edmSimAssociationName","")),
  trackingMapName_(Conf().getUntrackedParameter<string>("trackingMapName","TrackingMap")),
  barrelSuperClusterIdMapName_(Conf().getUntrackedParameter<string>
                               ("superClusterIdMapName","barrelSuperClusterIdMap")),
  endcapSuperClusterIdMapName_(Conf().getUntrackedParameter<string>
                               ("endcapClusterIdMapName","endcapSuperClusterIdMap")),
  trackMapName_(Conf().getUntrackedParameter<string>("trackMapName",
                                                     Form("%sMapName",mitName_.c_str()))),
  trackingMap_(0),
  tracks_(new mithep::TrackArr(250)), 
  trackMap_(new mithep::TrackMap)
{
  // Constructor.
  
  if (ecalAssocActive_) // initialize track associator configuration if needed
    assocParams_.loadParameters(
      cfg.getUntrackedParameter<ParameterSet>("TrackAssociatorParameters"));
}

//--------------------------------------------------------------------------------------------------
FillerTracks::~FillerTracks()
{
  // Destructor.

  delete tracks_;
  delete trackMap_;
}

//--------------------------------------------------------------------------------------------------
void FillerTracks::BookDataBlock(TreeWriter &tws)
{
  // Add tracks branch to tree, publish and get our objects.

  tws.AddBranch(mitName_,&tracks_);
  OS()->add<TrackArr>(tracks_,mitName_);

  if (!trackingMapName_.empty()) {
    trackingMap_ = OS()->get<TrackingParticleMap>(trackingMapName_);
    if (trackingMap_)
      AddBranchDep(mitName_,trackingMap_->GetBrName());
  }
  if (ecalAssocActive_ && !barrelSuperClusterIdMapName_.empty()) {
    barrelSuperClusterIdMap_ = OS()->get<SuperClusterIdMap>(barrelSuperClusterIdMapName_);
    if (barrelSuperClusterIdMap_)
      AddBranchDep(mitName_,barrelSuperClusterIdMap_->GetBrName());
  }
  if (ecalAssocActive_ && !endcapSuperClusterIdMapName_.empty()) {
    endcapSuperClusterIdMap_ = OS()->get<SuperClusterIdMap>(endcapSuperClusterIdMapName_);
    if (endcapSuperClusterIdMap_)
      AddBranchDep(mitName_,endcapSuperClusterIdMap_->GetBrName());
  }
  if (!trackMapName_.empty()) {
    trackMap_->SetBrName(mitName_);
    OS()->add<TrackMap>(trackMap_,trackMapName_);
  }
}

//--------------------------------------------------------------------------------------------------
void FillerTracks::FillDataBlock(const edm::Event      &event, 
                                 const edm::EventSetup &setup)
{
  // Fill tracks from edm collection into our collection.

  tracks_  ->Delete();
  trackMap_->Reset();

  // initialize handle and get product,
  // this usage allows also to get collections of classes which inherit from reco::Track
  Handle<View<reco::Track> > hTrackProduct;
  GetProduct(edmName_, hTrackProduct, event);  
        
  trackMap_->SetEdmProductId(hTrackProduct.id().id());
  const View<reco::Track> inTracks = *(hTrackProduct.product());  
  
  if (verbose_>1)
    printf("Track Collection: %s, id=%i\n",edmName_.c_str(),hTrackProduct.id().id());
  
  // for MC SimParticle association (reco->sim mappings)
  reco::RecoToSimCollection simAssociation;
  if (trackingMap_ && !edmSimAssocName_.empty()) {
    Handle<reco::RecoToSimCollection> simAssociationProduct;
    GetProduct(edmSimAssocName_, simAssociationProduct, event);  
    simAssociation = *(simAssociationProduct.product());
    if (verbose_>1)
      printf("SimAssociation Map Size = %i\n",simAssociation.size());
  }
  
  // set up associator for Track-Ecal associations
  TrackDetectorAssociator trackAssociator;
  trackAssociator.useDefaultPropagator();
  edm::ESHandle<MagneticField> bField;
  if (ecalAssocActive_) {
    setup.get<IdealMagneticFieldRecord>().get(bField);
  }  


  // loop through all tracks and fill the information
  for (View<reco::Track>::const_iterator it = inTracks.begin();
         it != inTracks.end(); ++it) {
         
    mithep::Track *outTrack = tracks_->Allocate();
    // create track and set the core parameters
    new (outTrack) mithep::Track(it->qoverp(),it->lambda(),
                                 it->phi(),it->dxy(),it->dsz());
    outTrack->SetErrors(it->qoverpError(),it->lambdaError(),
                        it->phiError(),it->dxyError(),it->dszError());
                        
    outTrack->SetPtErr(it->ptError());
        
    // fill track quality information
    outTrack->SetChi2(it->chi2());
    outTrack->SetNdof(static_cast<Int_t>(it->ndof()));

    //fill algo enum
    outTrack->SetAlgo(static_cast<mithep::Track::ETrackAlgorithm>(it->algo()));

    //fill quality bitmask
    outTrack->Quality().SetQualityMask(BitMask8(UChar_t(it->qualityMask())));
    
    if (verbose_>2) {
      printf("reco::Track   high purity = %i,  ", it->quality(reco::TrackBase::highPurity));
      printf("mithep::Track high purity = %i\n",outTrack->Quality().Quality(mithep::TrackQuality::highPurity));
    }
    
    //fill gsf flag, some type gymastics needed...
    if (typeid(*it)==typeid(reco::GsfTrack))
      outTrack->SetIsGsf(kTRUE);
    else
      outTrack->SetIsGsf(kFALSE);
    
    //fill hit layer map and missing hits
    const reco::HitPattern &hits = it->hitPattern();
    BitMask48 hitLayers;
    BitMask48 missingHitLayers;
    for (Int_t i=0; i<hits.numberOfHits(); i++) {
      uint32_t hit = hits.getHitPattern(i);
      if (hits.validHitFilter(hit)) {
        if (hits.trackerHitFilter(hit)) {
          hitLayers.SetBit(hitReader_.Layer(hit));
        }
      }

      if (hits.getHitType(hit)==1) {
        if (hits.trackerHitFilter(hit)) {
          missingHitLayers.SetBit(hitReader_.Layer(hit));
        }
      }
       
      if (verbose_>2) {
        if (hits.muonDTHitFilter(hit))
          printf("Muon DT Layer = %i\n", hits.getLayer(hit));
        if (hits.muonCSCHitFilter(hit))
          printf("Muon CSC Layer = %i\n", hits.getLayer(hit));        
        if (hits.muonRPCHitFilter(hit))
          printf("Muon RPC Layer = %i\n", hits.getLayer(hit));
      }
    }

    outTrack->SetHits(hitLayers);
    outTrack->SetMissingHits(missingHitLayers);

    //set expected inner hits
    const reco::HitPattern &expectedInnerHitPattern = it->trackerExpectedHitsInner();
    BitMask48 expectedHitsInner;
    // search for all good crossed layers (with or without hit)
    for (Int_t hi=0; hi<expectedInnerHitPattern.numberOfHits(); hi++) {
      uint32_t hit = expectedInnerHitPattern.getHitPattern(hi);
      if (expectedInnerHitPattern.getHitType(hit)<=1) {
        if (expectedInnerHitPattern.trackerHitFilter(hit)) {
          expectedHitsInner.SetBit(hitReader_.Layer(hit));
        }
      }
    }

    outTrack->SetExpectedHitsInner(expectedHitsInner);

    //set expected outer hits
    const reco::HitPattern &expectedOuterHitPattern = it->trackerExpectedHitsOuter();
    BitMask48 expectedHitsOuter;
    // search for all good crossed layers (with or without hit)
    for (Int_t hi=0; hi<expectedOuterHitPattern.numberOfHits(); hi++) {
      uint32_t hit = expectedOuterHitPattern.getHitPattern(hi);
      if (expectedOuterHitPattern.getHitType(hit)<=1) {
        if (expectedOuterHitPattern.trackerHitFilter(hit)) {
          expectedHitsOuter.SetBit(hitReader_.Layer(hit));
        }
      }
    }

    outTrack->SetExpectedHitsOuter(expectedHitsOuter);

    //make ecal associations
    if (ecalAssocActive_) {
      TrackDetMatchInfo matchInfo;
      //Extra check to allow propagation to work in AOD where no TrackExtra is available
      if (it->extra().isAvailable())
        matchInfo = trackAssociator.associate(event,setup,*it,assocParams_);
      else {
        TrajectoryStateTransform tsTransform;
        FreeTrajectoryState initialState = tsTransform.initialFreeState(*it,&*bField);
        matchInfo = trackAssociator.associate(event, setup, assocParams_, &initialState);
      }
       
      if (matchInfo.isGoodEcal) {
        outTrack->SetEtaEcal(matchInfo.trkGlobPosAtEcal.eta());
        outTrack->SetPhiEcal(matchInfo.trkGlobPosAtEcal.phi());
      }

      //fill supercluster link
      if (barrelSuperClusterIdMap_ || endcapSuperClusterIdMap_) {
        mithep::SuperCluster *cluster = 0;
        for (std::vector<const ::CaloTower*>::const_iterator iTower =
               matchInfo.crossedTowers.begin(); 
             iTower<matchInfo.crossedTowers.end() && !cluster; iTower++) {

          for (uint ihit=0; ihit<(*iTower)->constituentsSize() && !cluster; ihit++) {
            DetId hit = (*iTower)->constituent(ihit);
            if (barrelSuperClusterIdMap_ && barrelSuperClusterIdMap_->HasMit(hit))
              cluster = barrelSuperClusterIdMap_->GetMit(hit);
            else if (endcapSuperClusterIdMap_ && endcapSuperClusterIdMap_->HasMit(hit))
              cluster = endcapSuperClusterIdMap_->GetMit(hit);
          }
        }
        if (cluster)
          outTrack->SetSCluster(cluster);
      }
    }
    
    // add reference between mithep and edm object
    if (trackMap_) {
      mitedm::TrackPtr thePtr = inTracks.ptrAt(it - inTracks.begin());
      trackMap_->Add(thePtr, outTrack);
    }

    //do sim associations
    if (trackingMap_ && !edmSimAssocName_.empty()) {
      if (verbose_>1)
        printf("Trying Track-Sim association\n");
      reco::TrackBaseRef theBaseRef = inTracks.refAt(it - inTracks.begin());
      vector<pair<TrackingParticleRef, double> > simRefs;
      Bool_t noSimParticle = 0;
      try {
        simRefs = simAssociation[theBaseRef]; //try to get the sim references if existing
      }
      catch (edm::Exception &ex) {
        noSimParticle = 1;
      }

      if (!noSimParticle) { //loop through sim match candidates
        if (verbose_>1)
          printf("Applying track-sim association\n");
        for (vector<pair<TrackingParticleRef, double> >::const_iterator simRefPair=simRefs.begin(); 
             simRefPair != simRefs.end(); ++simRefPair) 
          if (simRefPair->second > 0.5) // require more than 50% shared hits between reco and sim
            outTrack->SetMCPart(trackingMap_->GetMit(simRefPair->first)); //add reco->sim reference
      }
    }
  }
  tracks_->Trim();
}
Revision:	1.39
Committed:	Mon May 10 15:14:19 2010 UTC (14 years, 11 months ago) by bendavid
Content type:	text/plain
Branch:	MAIN
CVS Tags:	Mit_014a, Mit_014, Mit_014pre3, Mit_014pre2, Mit_014pre1
Changes since 1.38:	+3 -1 lines
Log Message:	Add PtError
#	User	Rev	Content
1	bendavid	1.39	// $Id: FillerTracks.cc,v 1.38 2010/03/23 16:52:58 bendavid Exp $
2	loizides	1.1
3	bendavid	1.11	#include "MitProd/TreeFiller/interface/FillerTracks.h"
4	loizides	1.31	#include "DataFormats/RecoCandidate/interface/TrackAssociation.h"
5	bendavid	1.14	#include "DataFormats/TrackReco/interface/HitPattern.h"
6	loizides	1.1	#include "DataFormats/TrackReco/interface/Track.h"
7			#include "DataFormats/TrackReco/interface/TrackFwd.h"
8	bendavid	1.33	#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
9	loizides	1.1	#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
10			#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
11	loizides	1.31	#include "MagneticField/Engine/interface/MagneticField.h"
12			#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
13	bendavid	1.23	#include "TrackingTools/TrackAssociator/interface/TrackDetectorAssociator.h"
14	bendavid	1.28	#include "TrackingTools/TrajectoryState/interface/TrajectoryStateTransform.h"
15	loizides	1.31	#include "MitAna/DataTree/interface/Names.h"
16			#include "MitAna/DataTree/interface/TrackCol.h"
17			#include "MitProd/ObjectService/interface/ObjectService.h"
18	bendavid	1.26	#include "MitEdm/DataFormats/interface/Types.h"
19	loizides	1.1
20			using namespace std;
21			using namespace edm;
22			using namespace mithep;
23
24	loizides	1.4	//--------------------------------------------------------------------------------------------------
25	loizides	1.29	FillerTracks::FillerTracks(const ParameterSet &cfg, const char *name, bool active) :
26	loizides	1.7	BaseFiller(cfg,name,active),
27	loizides	1.29	ecalAssocActive_(Conf().getUntrackedParameter<bool>("ecalAssocActive",0)),
28			edmName_(Conf().getUntrackedParameter<string>("edmName","generalTracks")),
29			mitName_(Conf().getUntrackedParameter<string>("mitName",Names::gkTrackBrn)),
30	loizides	1.19	edmSimAssocName_(Conf().getUntrackedParameter<string>("edmSimAssociationName","")),
31	bendavid	1.20	trackingMapName_(Conf().getUntrackedParameter<string>("trackingMapName","TrackingMap")),
32	loizides	1.29	barrelSuperClusterIdMapName_(Conf().getUntrackedParameter<string>
33			("superClusterIdMapName","barrelSuperClusterIdMap")),
34			endcapSuperClusterIdMapName_(Conf().getUntrackedParameter<string>
35			("endcapClusterIdMapName","endcapSuperClusterIdMap")),
36	loizides	1.13	trackMapName_(Conf().getUntrackedParameter<string>("trackMapName",
37			Form("%sMapName",mitName_.c_str()))),
38	bendavid	1.17	trackingMap_(0),
39	loizides	1.7	tracks_(new mithep::TrackArr(250)),
40	loizides	1.1	trackMap_(new mithep::TrackMap)
41			{
42	loizides	1.13	// Constructor.
43	bendavid	1.23
44	loizides	1.29	if (ecalAssocActive_) // initialize track associator configuration if needed
45			assocParams_.loadParameters(
46			cfg.getUntrackedParameter<ParameterSet>("TrackAssociatorParameters"));
47	loizides	1.1	}
48
49	loizides	1.4	//--------------------------------------------------------------------------------------------------
50	loizides	1.1	FillerTracks::~FillerTracks()
51			{
52	loizides	1.13	// Destructor.
53
54	loizides	1.5	delete tracks_;
55	loizides	1.1	delete trackMap_;
56			}
57
58	loizides	1.4	//--------------------------------------------------------------------------------------------------
59	bendavid	1.37	void FillerTracks::BookDataBlock(TreeWriter &tws)
60	loizides	1.1	{
61	loizides	1.13	// Add tracks branch to tree, publish and get our objects.
62
63	loizides	1.29	tws.AddBranch(mitName_,&tracks_);
64			OS()->add<TrackArr>(tracks_,mitName_);
65	loizides	1.12
66	loizides	1.29	if (!trackingMapName_.empty()) {
67			trackingMap_ = OS()->get<TrackingParticleMap>(trackingMapName_);
68			if (trackingMap_)
69			AddBranchDep(mitName_,trackingMap_->GetBrName());
70			}
71	loizides	1.30	if (ecalAssocActive_ && !barrelSuperClusterIdMapName_.empty()) {
72	loizides	1.29	barrelSuperClusterIdMap_ = OS()->get<SuperClusterIdMap>(barrelSuperClusterIdMapName_);
73			if (barrelSuperClusterIdMap_)
74			AddBranchDep(mitName_,barrelSuperClusterIdMap_->GetBrName());
75			}
76	loizides	1.30	if (ecalAssocActive_ && !endcapSuperClusterIdMapName_.empty()) {
77	loizides	1.29	endcapSuperClusterIdMap_ = OS()->get<SuperClusterIdMap>(endcapSuperClusterIdMapName_);
78			if (endcapSuperClusterIdMap_)
79			AddBranchDep(mitName_,endcapSuperClusterIdMap_->GetBrName());
80			}
81			if (!trackMapName_.empty()) {
82			trackMap_->SetBrName(mitName_);
83			OS()->add<TrackMap>(trackMap_,trackMapName_);
84			}
85	loizides	1.1	}
86
87	loizides	1.4	//--------------------------------------------------------------------------------------------------
88	loizides	1.1	void FillerTracks::FillDataBlock(const edm::Event &event,
89			const edm::EventSetup &setup)
90			{
91	loizides	1.13	// Fill tracks from edm collection into our collection.
92
93	bendavid	1.27	tracks_ ->Delete();
94	loizides	1.1	trackMap_->Reset();
95	loizides	1.6
96	loizides	1.29	// initialize handle and get product,
97			// this usage allows also to get collections of classes which inherit from reco::Track
98	bendavid	1.24	Handle<View<reco::Track> > hTrackProduct;
99	loizides	1.6	GetProduct(edmName_, hTrackProduct, event);
100	loizides	1.1
101	loizides	1.6	trackMap_->SetEdmProductId(hTrackProduct.id().id());
102	bendavid	1.24	const View<reco::Track> inTracks = *(hTrackProduct.product());
103	loizides	1.1
104	bendavid	1.32	if (verbose_>1)
105			printf("Track Collection: %s, id=%i\n",edmName_.c_str(),hTrackProduct.id().id());
106
107	paus	1.10	// for MC SimParticle association (reco->sim mappings)
108	loizides	1.1	reco::RecoToSimCollection simAssociation;
109	loizides	1.19	if (trackingMap_ && !edmSimAssocName_.empty()) {
110	loizides	1.1	Handle<reco::RecoToSimCollection> simAssociationProduct;
111	loizides	1.19	GetProduct(edmSimAssocName_, simAssociationProduct, event);
112	loizides	1.1	simAssociation = *(simAssociationProduct.product());
113	loizides	1.29	if (verbose_>1)
114			printf("SimAssociation Map Size = %i\n",simAssociation.size());
115	loizides	1.1	}
116
117	loizides	1.29	// set up associator for Track-Ecal associations
118	bendavid	1.23	TrackDetectorAssociator trackAssociator;
119			trackAssociator.useDefaultPropagator();
120	bendavid	1.28	edm::ESHandle<MagneticField> bField;
121	bendavid	1.36	if (ecalAssocActive_) {
122	bendavid	1.28	setup.get<IdealMagneticFieldRecord>().get(bField);
123	bendavid	1.36	}
124
125
126	paus	1.10	// loop through all tracks and fill the information
127	bendavid	1.24	for (View<reco::Track>::const_iterator it = inTracks.begin();
128			it != inTracks.end(); ++it) {
129
130	loizides	1.5	mithep::Track *outTrack = tracks_->Allocate();
131	paus	1.10	// create track and set the core parameters
132	loizides	1.19	new (outTrack) mithep::Track(it->qoverp(),it->lambda(),
133			it->phi(),it->dxy(),it->dsz());
134			outTrack->SetErrors(it->qoverpError(),it->lambdaError(),
135			it->phiError(),it->dxyError(),it->dszError());
136	bendavid	1.39
137			outTrack->SetPtErr(it->ptError());
138	loizides	1.1
139	loizides	1.19	// fill track quality information
140	bendavid	1.14	outTrack->SetChi2(it->chi2());
141	loizides	1.16	outTrack->SetNdof(static_cast<Int_t>(it->ndof()));
142	bendavid	1.33
143			//fill algo enum
144			outTrack->SetAlgo(static_cast<mithep::Track::ETrackAlgorithm>(it->algo()));
145
146	bendavid	1.35	//fill quality bitmask
147			outTrack->Quality().SetQualityMask(BitMask8(UChar_t(it->qualityMask())));
148
149			if (verbose_>2) {
150			printf("reco::Track high purity = %i, ", it->quality(reco::TrackBase::highPurity));
151			printf("mithep::Track high purity = %i\n",outTrack->Quality().Quality(mithep::TrackQuality::highPurity));
152			}
153
154	bendavid	1.33	//fill gsf flag, some type gymastics needed...
155			if (typeid(*it)==typeid(reco::GsfTrack))
156			outTrack->SetIsGsf(kTRUE);
157			else
158			outTrack->SetIsGsf(kFALSE);
159	bendavid	1.14
160	bendavid	1.38	//fill hit layer map and missing hits
161	bendavid	1.14	const reco::HitPattern &hits = it->hitPattern();
162	bendavid	1.38	BitMask48 hitLayers;
163			BitMask48 missingHitLayers;
164	bendavid	1.14	for (Int_t i=0; i<hits.numberOfHits(); i++) {
165			uint32_t hit = hits.getHitPattern(i);
166	bendavid	1.38	if (hits.validHitFilter(hit)) {
167			if (hits.trackerHitFilter(hit)) {
168			hitLayers.SetBit(hitReader_.Layer(hit));
169			}
170			}
171
172			if (hits.getHitType(hit)==1) {
173			if (hits.trackerHitFilter(hit)) {
174			missingHitLayers.SetBit(hitReader_.Layer(hit));
175			}
176			}
177
178	loizides	1.29	if (verbose_>2) {
179	loizides	1.19	if (hits.muonDTHitFilter(hit))
180			printf("Muon DT Layer = %i\n", hits.getLayer(hit));
181			if (hits.muonCSCHitFilter(hit))
182			printf("Muon CSC Layer = %i\n", hits.getLayer(hit));
183			if (hits.muonRPCHitFilter(hit))
184			printf("Muon RPC Layer = %i\n", hits.getLayer(hit));
185			}
186	bendavid	1.14	}
187	bendavid	1.23
188	bendavid	1.38	outTrack->SetHits(hitLayers);
189			outTrack->SetMissingHits(missingHitLayers);
190
191			//set expected inner hits
192			const reco::HitPattern &expectedInnerHitPattern = it->trackerExpectedHitsInner();
193			BitMask48 expectedHitsInner;
194			// search for all good crossed layers (with or without hit)
195			for (Int_t hi=0; hi<expectedInnerHitPattern.numberOfHits(); hi++) {
196			uint32_t hit = expectedInnerHitPattern.getHitPattern(hi);
197			if (expectedInnerHitPattern.getHitType(hit)<=1) {
198			if (expectedInnerHitPattern.trackerHitFilter(hit)) {
199			expectedHitsInner.SetBit(hitReader_.Layer(hit));
200			}
201			}
202			}
203
204			outTrack->SetExpectedHitsInner(expectedHitsInner);
205
206			//set expected outer hits
207			const reco::HitPattern &expectedOuterHitPattern = it->trackerExpectedHitsOuter();
208			BitMask48 expectedHitsOuter;
209			// search for all good crossed layers (with or without hit)
210			for (Int_t hi=0; hi<expectedOuterHitPattern.numberOfHits(); hi++) {
211			uint32_t hit = expectedOuterHitPattern.getHitPattern(hi);
212			if (expectedOuterHitPattern.getHitType(hit)<=1) {
213			if (expectedOuterHitPattern.trackerHitFilter(hit)) {
214			expectedHitsOuter.SetBit(hitReader_.Layer(hit));
215			}
216			}
217			}
218
219			outTrack->SetExpectedHitsOuter(expectedHitsOuter);
220
221	bendavid	1.23	//make ecal associations
222			if (ecalAssocActive_) {
223	bendavid	1.28	TrackDetMatchInfo matchInfo;
224			//Extra check to allow propagation to work in AOD where no TrackExtra is available
225			if (it->extra().isAvailable())
226			matchInfo = trackAssociator.associate(event,setup,*it,assocParams_);
227			else {
228			TrajectoryStateTransform tsTransform;
229			FreeTrajectoryState initialState = tsTransform.initialFreeState(it,&bField);
230			matchInfo = trackAssociator.associate(event, setup, assocParams_, &initialState);
231			}
232	bendavid	1.36
233			if (matchInfo.isGoodEcal) {
234			outTrack->SetEtaEcal(matchInfo.trkGlobPosAtEcal.eta());
235			outTrack->SetPhiEcal(matchInfo.trkGlobPosAtEcal.phi());
236			}
237	bendavid	1.23
238			//fill supercluster link
239			if (barrelSuperClusterIdMap_ \|\| endcapSuperClusterIdMap_) {
240			mithep::SuperCluster *cluster = 0;
241	loizides	1.29	for (std::vector<const ::CaloTower*>::const_iterator iTower =
242			matchInfo.crossedTowers.begin();
243			iTower<matchInfo.crossedTowers.end() && !cluster; iTower++) {
244	bendavid	1.23
245			for (uint ihit=0; ihit<(*iTower)->constituentsSize() && !cluster; ihit++) {
246			DetId hit = (*iTower)->constituent(ihit);
247			if (barrelSuperClusterIdMap_ && barrelSuperClusterIdMap_->HasMit(hit))
248			cluster = barrelSuperClusterIdMap_->GetMit(hit);
249			else if (endcapSuperClusterIdMap_ && endcapSuperClusterIdMap_->HasMit(hit))
250			cluster = endcapSuperClusterIdMap_->GetMit(hit);
251			}
252			}
253			if (cluster)
254			outTrack->SetSCluster(cluster);
255			}
256			}
257	bendavid	1.14
258	loizides	1.1	// add reference between mithep and edm object
259	loizides	1.30	if (trackMap_) {
260			mitedm::TrackPtr thePtr = inTracks.ptrAt(it - inTracks.begin());
261			trackMap_->Add(thePtr, outTrack);
262			}
263
264	bendavid	1.35	//do sim associations
265	loizides	1.19	if (trackingMap_ && !edmSimAssocName_.empty()) {
266	loizides	1.29	if (verbose_>1)
267			printf("Trying Track-Sim association\n");
268	bendavid	1.24	reco::TrackBaseRef theBaseRef = inTracks.refAt(it - inTracks.begin());
269	loizides	1.1	vector<pair<TrackingParticleRef, double> > simRefs;
270	paus	1.10	Bool_t noSimParticle = 0;
271	loizides	1.1	try {
272			simRefs = simAssociation[theBaseRef]; //try to get the sim references if existing
273			}
274	loizides	1.5	catch (edm::Exception &ex) {
275	paus	1.10	noSimParticle = 1;
276	loizides	1.1	}
277	paus	1.10
278	loizides	1.1	if (!noSimParticle) { //loop through sim match candidates
279	loizides	1.29	if (verbose_>1)
280			printf("Applying track-sim association\n");
281	loizides	1.1	for (vector<pair<TrackingParticleRef, double> >::const_iterator simRefPair=simRefs.begin();
282			simRefPair != simRefs.end(); ++simRefPair)
283	paus	1.10	if (simRefPair->second > 0.5) // require more than 50% shared hits between reco and sim
284	bendavid	1.17	outTrack->SetMCPart(trackingMap_->GetMit(simRefPair->first)); //add reco->sim reference
285	loizides	1.1	}
286			}
287			}
288			tracks_->Trim();
289			}