TreeFiller/src/FillerTracks.cc

// $Id: FillerTracks.cc,v 1.33 2009/07/14 13:44:35 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, const edm::EventSetup &es)
{
  // 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());
        
    // 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 gsf flag, some type gymastics needed...
    if (typeid(*it)==typeid(reco::GsfTrack))
      outTrack->SetIsGsf(kTRUE);
    else
      outTrack->SetIsGsf(kFALSE);
    
    //fill hit layer map
    const reco::HitPattern &hits = it->hitPattern();
    for (Int_t i=0; i<hits.numberOfHits(); i++) {
      uint32_t hit = hits.getHitPattern(i);
      if (hits.validHitFilter(hit))
        if (hits.trackerHitFilter(hit))
          outTrack->SetHit(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));
      }
    }

    //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);
      }
        
      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 dim 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.34
Committed:	Fri Sep 25 08:42:51 2009 UTC (15 years, 7 months ago) by loizides
Content type:	text/plain
Branch:	MAIN
CVS Tags:	Mit_012h, Mit_012g, Mit_012f, Mit_012e, Mit_012d, Mit_012c, Mit_012b, Mit_012a, Mit_012, Mit_011a
Changes since 1.33:	+2 -2 lines
Log Message:	Extended interface of BookDataBlock to contain event setup.
#	User	Rev	Content
1	loizides	1.34	// $Id: FillerTracks.cc,v 1.33 2009/07/14 13:44:35 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	loizides	1.34	void FillerTracks::BookDataBlock(TreeWriter &tws, const edm::EventSetup &es)
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			if (ecalAssocActive_)
122			setup.get<IdealMagneticFieldRecord>().get(bField);
123
124	paus	1.10	// loop through all tracks and fill the information
125	bendavid	1.24	for (View<reco::Track>::const_iterator it = inTracks.begin();
126			it != inTracks.end(); ++it) {
127
128	loizides	1.5	mithep::Track *outTrack = tracks_->Allocate();
129	paus	1.10	// create track and set the core parameters
130	loizides	1.19	new (outTrack) mithep::Track(it->qoverp(),it->lambda(),
131			it->phi(),it->dxy(),it->dsz());
132			outTrack->SetErrors(it->qoverpError(),it->lambdaError(),
133			it->phiError(),it->dxyError(),it->dszError());
134	loizides	1.1
135	loizides	1.19	// fill track quality information
136	bendavid	1.14	outTrack->SetChi2(it->chi2());
137	loizides	1.16	outTrack->SetNdof(static_cast<Int_t>(it->ndof()));
138	bendavid	1.33
139			//fill algo enum
140			outTrack->SetAlgo(static_cast<mithep::Track::ETrackAlgorithm>(it->algo()));
141
142			//fill gsf flag, some type gymastics needed...
143			if (typeid(*it)==typeid(reco::GsfTrack))
144			outTrack->SetIsGsf(kTRUE);
145			else
146			outTrack->SetIsGsf(kFALSE);
147	bendavid	1.14
148	loizides	1.19	//fill hit layer map
149	bendavid	1.14	const reco::HitPattern &hits = it->hitPattern();
150			for (Int_t i=0; i<hits.numberOfHits(); i++) {
151			uint32_t hit = hits.getHitPattern(i);
152	loizides	1.19	if (hits.validHitFilter(hit))
153			if (hits.trackerHitFilter(hit))
154	bendavid	1.22	outTrack->SetHit(hitReader_.Layer(hit));
155	bendavid	1.14
156	loizides	1.29	if (verbose_>2) {
157	loizides	1.19	if (hits.muonDTHitFilter(hit))
158			printf("Muon DT Layer = %i\n", hits.getLayer(hit));
159			if (hits.muonCSCHitFilter(hit))
160			printf("Muon CSC Layer = %i\n", hits.getLayer(hit));
161			if (hits.muonRPCHitFilter(hit))
162			printf("Muon RPC Layer = %i\n", hits.getLayer(hit));
163			}
164	bendavid	1.14	}
165	bendavid	1.23
166			//make ecal associations
167			if (ecalAssocActive_) {
168	bendavid	1.28	TrackDetMatchInfo matchInfo;
169			//Extra check to allow propagation to work in AOD where no TrackExtra is available
170			if (it->extra().isAvailable())
171			matchInfo = trackAssociator.associate(event,setup,*it,assocParams_);
172			else {
173			TrajectoryStateTransform tsTransform;
174			FreeTrajectoryState initialState = tsTransform.initialFreeState(it,&bField);
175			matchInfo = trackAssociator.associate(event, setup, assocParams_, &initialState);
176			}
177
178	bendavid	1.23	outTrack->SetEtaEcal(matchInfo.trkGlobPosAtEcal.eta());
179			outTrack->SetPhiEcal(matchInfo.trkGlobPosAtEcal.phi());
180
181			//fill supercluster link
182			if (barrelSuperClusterIdMap_ \|\| endcapSuperClusterIdMap_) {
183			mithep::SuperCluster *cluster = 0;
184	loizides	1.29	for (std::vector<const ::CaloTower*>::const_iterator iTower =
185			matchInfo.crossedTowers.begin();
186			iTower<matchInfo.crossedTowers.end() && !cluster; iTower++) {
187	bendavid	1.23
188			for (uint ihit=0; ihit<(*iTower)->constituentsSize() && !cluster; ihit++) {
189			DetId hit = (*iTower)->constituent(ihit);
190			if (barrelSuperClusterIdMap_ && barrelSuperClusterIdMap_->HasMit(hit))
191			cluster = barrelSuperClusterIdMap_->GetMit(hit);
192			else if (endcapSuperClusterIdMap_ && endcapSuperClusterIdMap_->HasMit(hit))
193			cluster = endcapSuperClusterIdMap_->GetMit(hit);
194			}
195			}
196			if (cluster)
197			outTrack->SetSCluster(cluster);
198			}
199			}
200	bendavid	1.14
201	loizides	1.1	// add reference between mithep and edm object
202	loizides	1.30	if (trackMap_) {
203			mitedm::TrackPtr thePtr = inTracks.ptrAt(it - inTracks.begin());
204			trackMap_->Add(thePtr, outTrack);
205			}
206
207	bendavid	1.23	//do dim associations
208	loizides	1.19	if (trackingMap_ && !edmSimAssocName_.empty()) {
209	loizides	1.29	if (verbose_>1)
210			printf("Trying Track-Sim association\n");
211	bendavid	1.24	reco::TrackBaseRef theBaseRef = inTracks.refAt(it - inTracks.begin());
212	loizides	1.1	vector<pair<TrackingParticleRef, double> > simRefs;
213	paus	1.10	Bool_t noSimParticle = 0;
214	loizides	1.1	try {
215			simRefs = simAssociation[theBaseRef]; //try to get the sim references if existing
216			}
217	loizides	1.5	catch (edm::Exception &ex) {
218	paus	1.10	noSimParticle = 1;
219	loizides	1.1	}
220	paus	1.10
221	loizides	1.1	if (!noSimParticle) { //loop through sim match candidates
222	loizides	1.29	if (verbose_>1)
223			printf("Applying track-sim association\n");
224	loizides	1.1	for (vector<pair<TrackingParticleRef, double> >::const_iterator simRefPair=simRefs.begin();
225			simRefPair != simRefs.end(); ++simRefPair)
226	paus	1.10	if (simRefPair->second > 0.5) // require more than 50% shared hits between reco and sim
227	bendavid	1.17	outTrack->SetMCPart(trackingMap_->GetMit(simRefPair->first)); //add reco->sim reference
228	loizides	1.1	}
229			}
230			}
231			tracks_->Trim();
232			}