TreeFiller/src/FillerTracks.cc

// $Id: FillerTracks.cc,v 1.18 2008/09/06 19:25:43 bendavid Exp $

#include "MitProd/TreeFiller/interface/FillerTracks.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/TrackReco/interface/HitPattern.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
#include "DataFormats/RecoCandidate/interface/TrackAssociation.h"

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

//--------------------------------------------------------------------------------------------------
FillerTracks::FillerTracks(const ParameterSet &cfg, const char *name, bool active,
                           const char *edmName, const char *mitName) :
  BaseFiller(cfg,name,active),
  edmName_(Conf().getUntrackedParameter<string>("edmName",edmName)),
  mitName_(Conf().getUntrackedParameter<string>("mitName",mitName)),
  edmSimAssocName_(Conf().getUntrackedParameter<string>("edmSimAssociationName","")),
  trackingMapName_(Conf().getUntrackedParameter<string>("simMapName","TrackingMap")),
  trackMapName_(Conf().getUntrackedParameter<string>("trackMapName",
                                                     Form("%sMapName",mitName_.c_str()))),
  trackingMap_(0),
  tracks_(new mithep::TrackArr(250)), 
  trackMap_(new mithep::TrackMap)
{
  // Constructor.

  InitLayerMap();
}

//--------------------------------------------------------------------------------------------------
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_.c_str(),&tracks_);

  trackingMap_ = OS()->get<TrackingParticleMap>(trackingMapName_.c_str());
  OS()->add<TrackMap>(trackMap_,trackMapName_.c_str());
  OS()->add<TrackArr>(tracks_,mitName_.c_str());
}

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

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

  Handle<reco::TrackCollection> hTrackProduct;
  GetProduct(edmName_, hTrackProduct, event);  
        
  trackMap_->SetEdmProductId(hTrackProduct.id().id());
  const reco::TrackCollection inTracks = *(hTrackProduct.product());  
  
  // 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());
    //printf("SimAssociation Map Size = %i\n",simAssociation.size());
  }
  
  // loop through all tracks and fill the information
  for (reco::TrackCollection::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 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(layerMap_[hit]);
                
      if (0) {
        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));
      }
    }
    
    // add reference between mithep and edm object
    reco::TrackRef theRef(hTrackProduct, it - inTracks.begin());
    trackMap_->Add(theRef, outTrack);
        
    if (trackingMap_ && !edmSimAssocName_.empty()) {
      //printf("Trying Track-Sim association\n");
      reco::TrackBaseRef theBaseRef(theRef);
      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
        //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();
}

//--------------------------------------------------------------------------------------------------
void FillerTracks::InitLayerMap()
{
  // Initialize mapping between hit layer format in reco::HitPattern and the one used in
  // mithep::Track. Note in 21x stereo layers are treated separatelely.
  
  layerMap_[1160] = mithep::Track::PXB1;
  layerMap_[1168] = mithep::Track::PXB2;
  layerMap_[1176] = mithep::Track::PXB3;
  layerMap_[1288] = mithep::Track::PXF1;
  layerMap_[1296] = mithep::Track::PXF2;
  layerMap_[1416] = mithep::Track::TIB1;
  layerMap_[1420] = mithep::Track::TIB1S;
  layerMap_[1424] = mithep::Track::TIB2;
  layerMap_[1428] = mithep::Track::TIB2S;
  layerMap_[1432] = mithep::Track::TIB3;
  layerMap_[1440] = mithep::Track::TIB4;
  layerMap_[1544] = mithep::Track::TID1;
  layerMap_[1548] = mithep::Track::TID1S;
  layerMap_[1552] = mithep::Track::TID2;
  layerMap_[1556] = mithep::Track::TID2S;
  layerMap_[1560] = mithep::Track::TID3;
  layerMap_[1672] = mithep::Track::TOB1;
  layerMap_[1676] = mithep::Track::TOB1S;
  layerMap_[1680] = mithep::Track::TOB2;
  layerMap_[1684] = mithep::Track::TOB2S;
  layerMap_[1688] = mithep::Track::TOB3;
  layerMap_[1696] = mithep::Track::TOB4;
  layerMap_[1704] = mithep::Track::TOB5;
  layerMap_[1712] = mithep::Track::TOB6;
  layerMap_[1800] = mithep::Track::TEC1;
  layerMap_[1804] = mithep::Track::TEC1S;
  layerMap_[1808] = mithep::Track::TEC2;
  layerMap_[1812] = mithep::Track::TEC2S;
  layerMap_[1816] = mithep::Track::TEC3;
  layerMap_[1824] = mithep::Track::TEC4;
  layerMap_[1832] = mithep::Track::TEC5;
  layerMap_[1836] = mithep::Track::TEC5S;
  layerMap_[1840] = mithep::Track::TEC6;
  layerMap_[1848] = mithep::Track::TEC7;
  layerMap_[1856] = mithep::Track::TEC8;
  layerMap_[1864] = mithep::Track::TEC9;
}
Revision:	1.19
Committed:	Wed Sep 17 04:26:12 2008 UTC (16 years, 7 months ago) by loizides
Content type:	text/plain
Branch:	MAIN
Changes since 1.18:	+21 -18 lines
Log Message:	Cleanup
#	User	Rev	Content
1	loizides	1.19	// $Id: FillerTracks.cc,v 1.18 2008/09/06 19:25:43 bendavid Exp $
2	loizides	1.1
3	bendavid	1.11	#include "MitProd/TreeFiller/interface/FillerTracks.h"
4	loizides	1.1	#include "FWCore/MessageLogger/interface/MessageLogger.h"
5			#include "DataFormats/Common/interface/Handle.h"
6	bendavid	1.14	#include "DataFormats/TrackReco/interface/HitPattern.h"
7	loizides	1.1	#include "DataFormats/TrackReco/interface/Track.h"
8			#include "DataFormats/TrackReco/interface/TrackFwd.h"
9			#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
10			#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
11			#include "DataFormats/RecoCandidate/interface/TrackAssociation.h"
12
13			using namespace std;
14			using namespace edm;
15			using namespace mithep;
16
17	loizides	1.4	//--------------------------------------------------------------------------------------------------
18	loizides	1.16	FillerTracks::FillerTracks(const ParameterSet &cfg, const char *name, bool active,
19			const char edmName, const char mitName) :
20	loizides	1.7	BaseFiller(cfg,name,active),
21	loizides	1.16	edmName_(Conf().getUntrackedParameter<string>("edmName",edmName)),
22			mitName_(Conf().getUntrackedParameter<string>("mitName",mitName)),
23	loizides	1.19	edmSimAssocName_(Conf().getUntrackedParameter<string>("edmSimAssociationName","")),
24	bendavid	1.18	trackingMapName_(Conf().getUntrackedParameter<string>("simMapName","TrackingMap")),
25	loizides	1.13	trackMapName_(Conf().getUntrackedParameter<string>("trackMapName",
26			Form("%sMapName",mitName_.c_str()))),
27	bendavid	1.17	trackingMap_(0),
28	loizides	1.7	tracks_(new mithep::TrackArr(250)),
29	loizides	1.1	trackMap_(new mithep::TrackMap)
30			{
31	loizides	1.13	// Constructor.
32	loizides	1.16
33			InitLayerMap();
34	loizides	1.1	}
35
36	loizides	1.4	//--------------------------------------------------------------------------------------------------
37	loizides	1.1	FillerTracks::~FillerTracks()
38			{
39	loizides	1.13	// Destructor.
40
41	loizides	1.5	delete tracks_;
42	loizides	1.1	delete trackMap_;
43			}
44
45	loizides	1.4	//--------------------------------------------------------------------------------------------------
46	loizides	1.1	void FillerTracks::BookDataBlock(TreeWriter &tws)
47			{
48	loizides	1.13	// Add tracks branch to tree, publish and get our objects.
49
50	loizides	1.1	tws.AddBranch(mitName_.c_str(),&tracks_);
51	loizides	1.12
52	bendavid	1.17	trackingMap_ = OS()->get<TrackingParticleMap>(trackingMapName_.c_str());
53	loizides	1.13	OS()->add<TrackMap>(trackMap_,trackMapName_.c_str());
54			OS()->add<TrackArr>(tracks_,mitName_.c_str());
55	loizides	1.1	}
56
57	loizides	1.4	//--------------------------------------------------------------------------------------------------
58	loizides	1.1	void FillerTracks::FillDataBlock(const edm::Event &event,
59			const edm::EventSetup &setup)
60			{
61	loizides	1.13	// Fill tracks from edm collection into our collection.
62
63	paus	1.10	tracks_ ->Reset();
64	loizides	1.1	trackMap_->Reset();
65	loizides	1.6
66			Handle<reco::TrackCollection> hTrackProduct;
67			GetProduct(edmName_, hTrackProduct, event);
68	loizides	1.1
69	loizides	1.6	trackMap_->SetEdmProductId(hTrackProduct.id().id());
70			const reco::TrackCollection inTracks = *(hTrackProduct.product());
71	loizides	1.1
72	paus	1.10	// for MC SimParticle association (reco->sim mappings)
73	loizides	1.1	reco::RecoToSimCollection simAssociation;
74	loizides	1.19	if (trackingMap_ && !edmSimAssocName_.empty()) {
75	loizides	1.1	Handle<reco::RecoToSimCollection> simAssociationProduct;
76	loizides	1.19	GetProduct(edmSimAssocName_, simAssociationProduct, event);
77	loizides	1.1	simAssociation = *(simAssociationProduct.product());
78	bendavid	1.17	//printf("SimAssociation Map Size = %i\n",simAssociation.size());
79	loizides	1.1	}
80
81	paus	1.10	// loop through all tracks and fill the information
82			for (reco::TrackCollection::const_iterator it = inTracks.begin();
83			it != inTracks.end(); ++it) {
84	loizides	1.5	mithep::Track *outTrack = tracks_->Allocate();
85	paus	1.10	// create track and set the core parameters
86	loizides	1.19	new (outTrack) mithep::Track(it->qoverp(),it->lambda(),
87			it->phi(),it->dxy(),it->dsz());
88			outTrack->SetErrors(it->qoverpError(),it->lambdaError(),
89			it->phiError(),it->dxyError(),it->dszError());
90	loizides	1.1
91	loizides	1.19	// fill track quality information
92	bendavid	1.14	outTrack->SetChi2(it->chi2());
93	loizides	1.16	outTrack->SetNdof(static_cast<Int_t>(it->ndof()));
94	bendavid	1.14
95	loizides	1.19	//fill hit layer map
96	bendavid	1.14	const reco::HitPattern &hits = it->hitPattern();
97			for (Int_t i=0; i<hits.numberOfHits(); i++) {
98			uint32_t hit = hits.getHitPattern(i);
99	loizides	1.19	if (hits.validHitFilter(hit))
100			if (hits.trackerHitFilter(hit))
101	bendavid	1.14	outTrack->SetHit(layerMap_[hit]);
102
103	loizides	1.19	if (0) {
104			if (hits.muonDTHitFilter(hit))
105			printf("Muon DT Layer = %i\n", hits.getLayer(hit));
106			if (hits.muonCSCHitFilter(hit))
107			printf("Muon CSC Layer = %i\n", hits.getLayer(hit));
108			if (hits.muonRPCHitFilter(hit))
109			printf("Muon RPC Layer = %i\n", hits.getLayer(hit));
110			}
111	bendavid	1.14	}
112
113	loizides	1.1	// add reference between mithep and edm object
114	paus	1.10	reco::TrackRef theRef(hTrackProduct, it - inTracks.begin());
115	loizides	1.1	trackMap_->Add(theRef, outTrack);
116
117	loizides	1.19	if (trackingMap_ && !edmSimAssocName_.empty()) {
118	bendavid	1.17	//printf("Trying Track-Sim association\n");
119	loizides	1.1	reco::TrackBaseRef theBaseRef(theRef);
120			vector<pair<TrackingParticleRef, double> > simRefs;
121	paus	1.10	Bool_t noSimParticle = 0;
122	loizides	1.1	try {
123			simRefs = simAssociation[theBaseRef]; //try to get the sim references if existing
124			}
125	loizides	1.5	catch (edm::Exception &ex) {
126	paus	1.10	noSimParticle = 1;
127	loizides	1.1	}
128	paus	1.10
129	loizides	1.1	if (!noSimParticle) { //loop through sim match candidates
130	bendavid	1.17	//printf("Applying track-sim association\n");
131	loizides	1.1	for (vector<pair<TrackingParticleRef, double> >::const_iterator simRefPair=simRefs.begin();
132			simRefPair != simRefs.end(); ++simRefPair)
133
134	paus	1.10	if (simRefPair->second > 0.5) // require more than 50% shared hits between reco and sim
135	bendavid	1.17	outTrack->SetMCPart(trackingMap_->GetMit(simRefPair->first)); //add reco->sim reference
136	loizides	1.1	}
137			}
138			}
139			tracks_->Trim();
140			}
141	bendavid	1.14
142			//--------------------------------------------------------------------------------------------------
143			void FillerTracks::InitLayerMap()
144			{
145			// Initialize mapping between hit layer format in reco::HitPattern and the one used in
146	loizides	1.15	// mithep::Track. Note in 21x stereo layers are treated separatelely.
147	bendavid	1.14
148	loizides	1.15	layerMap_[1160] = mithep::Track::PXB1;
149			layerMap_[1168] = mithep::Track::PXB2;
150			layerMap_[1176] = mithep::Track::PXB3;
151			layerMap_[1288] = mithep::Track::PXF1;
152			layerMap_[1296] = mithep::Track::PXF2;
153			layerMap_[1416] = mithep::Track::TIB1;
154			layerMap_[1420] = mithep::Track::TIB1S;
155			layerMap_[1424] = mithep::Track::TIB2;
156			layerMap_[1428] = mithep::Track::TIB2S;
157			layerMap_[1432] = mithep::Track::TIB3;
158			layerMap_[1440] = mithep::Track::TIB4;
159			layerMap_[1544] = mithep::Track::TID1;
160			layerMap_[1548] = mithep::Track::TID1S;
161			layerMap_[1552] = mithep::Track::TID2;
162			layerMap_[1556] = mithep::Track::TID2S;
163			layerMap_[1560] = mithep::Track::TID3;
164			layerMap_[1672] = mithep::Track::TOB1;
165			layerMap_[1676] = mithep::Track::TOB1S;
166			layerMap_[1680] = mithep::Track::TOB2;
167			layerMap_[1684] = mithep::Track::TOB2S;
168			layerMap_[1688] = mithep::Track::TOB3;
169			layerMap_[1696] = mithep::Track::TOB4;
170			layerMap_[1704] = mithep::Track::TOB5;
171			layerMap_[1712] = mithep::Track::TOB6;
172			layerMap_[1800] = mithep::Track::TEC1;
173			layerMap_[1804] = mithep::Track::TEC1S;
174			layerMap_[1808] = mithep::Track::TEC2;
175			layerMap_[1812] = mithep::Track::TEC2S;
176			layerMap_[1816] = mithep::Track::TEC3;
177			layerMap_[1824] = mithep::Track::TEC4;
178			layerMap_[1832] = mithep::Track::TEC5;
179			layerMap_[1836] = mithep::Track::TEC5S;
180			layerMap_[1840] = mithep::Track::TEC6;
181			layerMap_[1848] = mithep::Track::TEC7;
182			layerMap_[1856] = mithep::Track::TEC8;
183			layerMap_[1864] = mithep::Track::TEC9;
184	bendavid	1.14	}