TreeFiller/src/FillerTracks.cc

// $Id: FillerTracks.cc,v 1.31 2009/06/15 15:00:26 loizides 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 "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());
        
    // 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(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.32
Committed:	Thu Jun 18 23:02:22 2009 UTC (15 years, 10 months ago) by bendavid
Content type:	text/plain
Branch:	MAIN
Changes since 1.31:	+4 -1 lines
Log Message:	Added debug statement to find linked collections
#	User	Rev	Content
1	bendavid	1.32	// $Id: FillerTracks.cc,v 1.31 2009/06/15 15:00:26 loizides 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			#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
9			#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
10	loizides	1.31	#include "MagneticField/Engine/interface/MagneticField.h"
11			#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
12	bendavid	1.23	#include "TrackingTools/TrackAssociator/interface/TrackDetectorAssociator.h"
13	bendavid	1.28	#include "TrackingTools/TrajectoryState/interface/TrajectoryStateTransform.h"
14	loizides	1.31	#include "MitAna/DataTree/interface/Names.h"
15			#include "MitAna/DataTree/interface/TrackCol.h"
16			#include "MitProd/ObjectService/interface/ObjectService.h"
17	bendavid	1.26	#include "MitEdm/DataFormats/interface/Types.h"
18	loizides	1.1
19			using namespace std;
20			using namespace edm;
21			using namespace mithep;
22
23	loizides	1.4	//--------------------------------------------------------------------------------------------------
24	loizides	1.29	FillerTracks::FillerTracks(const ParameterSet &cfg, const char *name, bool active) :
25	loizides	1.7	BaseFiller(cfg,name,active),
26	loizides	1.29	ecalAssocActive_(Conf().getUntrackedParameter<bool>("ecalAssocActive",0)),
27			edmName_(Conf().getUntrackedParameter<string>("edmName","generalTracks")),
28			mitName_(Conf().getUntrackedParameter<string>("mitName",Names::gkTrackBrn)),
29	loizides	1.19	edmSimAssocName_(Conf().getUntrackedParameter<string>("edmSimAssociationName","")),
30	bendavid	1.20	trackingMapName_(Conf().getUntrackedParameter<string>("trackingMapName","TrackingMap")),
31	loizides	1.29	barrelSuperClusterIdMapName_(Conf().getUntrackedParameter<string>
32			("superClusterIdMapName","barrelSuperClusterIdMap")),
33			endcapSuperClusterIdMapName_(Conf().getUntrackedParameter<string>
34			("endcapClusterIdMapName","endcapSuperClusterIdMap")),
35	loizides	1.13	trackMapName_(Conf().getUntrackedParameter<string>("trackMapName",
36			Form("%sMapName",mitName_.c_str()))),
37	bendavid	1.17	trackingMap_(0),
38	loizides	1.7	tracks_(new mithep::TrackArr(250)),
39	loizides	1.1	trackMap_(new mithep::TrackMap)
40			{
41	loizides	1.13	// Constructor.
42	bendavid	1.23
43	loizides	1.29	if (ecalAssocActive_) // initialize track associator configuration if needed
44			assocParams_.loadParameters(
45			cfg.getUntrackedParameter<ParameterSet>("TrackAssociatorParameters"));
46	loizides	1.1	}
47
48	loizides	1.4	//--------------------------------------------------------------------------------------------------
49	loizides	1.1	FillerTracks::~FillerTracks()
50			{
51	loizides	1.13	// Destructor.
52
53	loizides	1.5	delete tracks_;
54	loizides	1.1	delete trackMap_;
55			}
56
57	loizides	1.4	//--------------------------------------------------------------------------------------------------
58	loizides	1.1	void FillerTracks::BookDataBlock(TreeWriter &tws)
59			{
60	loizides	1.13	// Add tracks branch to tree, publish and get our objects.
61
62	loizides	1.29	tws.AddBranch(mitName_,&tracks_);
63			OS()->add<TrackArr>(tracks_,mitName_);
64	loizides	1.12
65	loizides	1.29	if (!trackingMapName_.empty()) {
66			trackingMap_ = OS()->get<TrackingParticleMap>(trackingMapName_);
67			if (trackingMap_)
68			AddBranchDep(mitName_,trackingMap_->GetBrName());
69			}
70	loizides	1.30	if (ecalAssocActive_ && !barrelSuperClusterIdMapName_.empty()) {
71	loizides	1.29	barrelSuperClusterIdMap_ = OS()->get<SuperClusterIdMap>(barrelSuperClusterIdMapName_);
72			if (barrelSuperClusterIdMap_)
73			AddBranchDep(mitName_,barrelSuperClusterIdMap_->GetBrName());
74			}
75	loizides	1.30	if (ecalAssocActive_ && !endcapSuperClusterIdMapName_.empty()) {
76	loizides	1.29	endcapSuperClusterIdMap_ = OS()->get<SuperClusterIdMap>(endcapSuperClusterIdMapName_);
77			if (endcapSuperClusterIdMap_)
78			AddBranchDep(mitName_,endcapSuperClusterIdMap_->GetBrName());
79			}
80			if (!trackMapName_.empty()) {
81			trackMap_->SetBrName(mitName_);
82			OS()->add<TrackMap>(trackMap_,trackMapName_);
83			}
84	loizides	1.1	}
85
86	loizides	1.4	//--------------------------------------------------------------------------------------------------
87	loizides	1.1	void FillerTracks::FillDataBlock(const edm::Event &event,
88			const edm::EventSetup &setup)
89			{
90	loizides	1.13	// Fill tracks from edm collection into our collection.
91
92	bendavid	1.27	tracks_ ->Delete();
93	loizides	1.1	trackMap_->Reset();
94	loizides	1.6
95	loizides	1.29	// initialize handle and get product,
96			// this usage allows also to get collections of classes which inherit from reco::Track
97	bendavid	1.24	Handle<View<reco::Track> > hTrackProduct;
98	loizides	1.6	GetProduct(edmName_, hTrackProduct, event);
99	loizides	1.1
100	loizides	1.6	trackMap_->SetEdmProductId(hTrackProduct.id().id());
101	bendavid	1.24	const View<reco::Track> inTracks = *(hTrackProduct.product());
102	loizides	1.1
103	bendavid	1.32	if (verbose_>1)
104			printf("Track Collection: %s, id=%i\n",edmName_.c_str(),hTrackProduct.id().id());
105
106	paus	1.10	// for MC SimParticle association (reco->sim mappings)
107	loizides	1.1	reco::RecoToSimCollection simAssociation;
108	loizides	1.19	if (trackingMap_ && !edmSimAssocName_.empty()) {
109	loizides	1.1	Handle<reco::RecoToSimCollection> simAssociationProduct;
110	loizides	1.19	GetProduct(edmSimAssocName_, simAssociationProduct, event);
111	loizides	1.1	simAssociation = *(simAssociationProduct.product());
112	loizides	1.29	if (verbose_>1)
113			printf("SimAssociation Map Size = %i\n",simAssociation.size());
114	loizides	1.1	}
115
116	loizides	1.29	// set up associator for Track-Ecal associations
117	bendavid	1.23	TrackDetectorAssociator trackAssociator;
118			trackAssociator.useDefaultPropagator();
119	bendavid	1.28	edm::ESHandle<MagneticField> bField;
120			if (ecalAssocActive_)
121			setup.get<IdealMagneticFieldRecord>().get(bField);
122
123	paus	1.10	// loop through all tracks and fill the information
124	bendavid	1.24	for (View<reco::Track>::const_iterator it = inTracks.begin();
125			it != inTracks.end(); ++it) {
126
127	loizides	1.5	mithep::Track *outTrack = tracks_->Allocate();
128	paus	1.10	// create track and set the core parameters
129	loizides	1.19	new (outTrack) mithep::Track(it->qoverp(),it->lambda(),
130			it->phi(),it->dxy(),it->dsz());
131			outTrack->SetErrors(it->qoverpError(),it->lambdaError(),
132			it->phiError(),it->dxyError(),it->dszError());
133	loizides	1.1
134	loizides	1.19	// fill track quality information
135	bendavid	1.14	outTrack->SetChi2(it->chi2());
136	loizides	1.16	outTrack->SetNdof(static_cast<Int_t>(it->ndof()));
137	bendavid	1.14
138	loizides	1.19	//fill hit layer map
139	bendavid	1.14	const reco::HitPattern &hits = it->hitPattern();
140			for (Int_t i=0; i<hits.numberOfHits(); i++) {
141			uint32_t hit = hits.getHitPattern(i);
142	loizides	1.19	if (hits.validHitFilter(hit))
143			if (hits.trackerHitFilter(hit))
144	bendavid	1.22	outTrack->SetHit(hitReader_.Layer(hit));
145	bendavid	1.14
146	loizides	1.29	if (verbose_>2) {
147	loizides	1.19	if (hits.muonDTHitFilter(hit))
148			printf("Muon DT Layer = %i\n", hits.getLayer(hit));
149			if (hits.muonCSCHitFilter(hit))
150			printf("Muon CSC Layer = %i\n", hits.getLayer(hit));
151			if (hits.muonRPCHitFilter(hit))
152			printf("Muon RPC Layer = %i\n", hits.getLayer(hit));
153			}
154	bendavid	1.14	}
155	bendavid	1.23
156			//make ecal associations
157			if (ecalAssocActive_) {
158	bendavid	1.28	TrackDetMatchInfo matchInfo;
159			//Extra check to allow propagation to work in AOD where no TrackExtra is available
160			if (it->extra().isAvailable())
161			matchInfo = trackAssociator.associate(event,setup,*it,assocParams_);
162			else {
163			TrajectoryStateTransform tsTransform;
164			FreeTrajectoryState initialState = tsTransform.initialFreeState(it,&bField);
165			matchInfo = trackAssociator.associate(event, setup, assocParams_, &initialState);
166			}
167
168	bendavid	1.23	outTrack->SetEtaEcal(matchInfo.trkGlobPosAtEcal.eta());
169			outTrack->SetPhiEcal(matchInfo.trkGlobPosAtEcal.phi());
170
171			//fill supercluster link
172			if (barrelSuperClusterIdMap_ \|\| endcapSuperClusterIdMap_) {
173			mithep::SuperCluster *cluster = 0;
174	loizides	1.29	for (std::vector<const ::CaloTower*>::const_iterator iTower =
175			matchInfo.crossedTowers.begin();
176			iTower<matchInfo.crossedTowers.end() && !cluster; iTower++) {
177	bendavid	1.23
178			for (uint ihit=0; ihit<(*iTower)->constituentsSize() && !cluster; ihit++) {
179			DetId hit = (*iTower)->constituent(ihit);
180			if (barrelSuperClusterIdMap_ && barrelSuperClusterIdMap_->HasMit(hit))
181			cluster = barrelSuperClusterIdMap_->GetMit(hit);
182			else if (endcapSuperClusterIdMap_ && endcapSuperClusterIdMap_->HasMit(hit))
183			cluster = endcapSuperClusterIdMap_->GetMit(hit);
184			}
185			}
186			if (cluster)
187			outTrack->SetSCluster(cluster);
188			}
189			}
190	bendavid	1.14
191	loizides	1.1	// add reference between mithep and edm object
192	loizides	1.30	if (trackMap_) {
193			mitedm::TrackPtr thePtr = inTracks.ptrAt(it - inTracks.begin());
194			trackMap_->Add(thePtr, outTrack);
195			}
196
197	bendavid	1.23	//do dim associations
198	loizides	1.19	if (trackingMap_ && !edmSimAssocName_.empty()) {
199	loizides	1.29	if (verbose_>1)
200			printf("Trying Track-Sim association\n");
201	bendavid	1.24	reco::TrackBaseRef theBaseRef = inTracks.refAt(it - inTracks.begin());
202	loizides	1.1	vector<pair<TrackingParticleRef, double> > simRefs;
203	paus	1.10	Bool_t noSimParticle = 0;
204	loizides	1.1	try {
205			simRefs = simAssociation[theBaseRef]; //try to get the sim references if existing
206			}
207	loizides	1.5	catch (edm::Exception &ex) {
208	paus	1.10	noSimParticle = 1;
209	loizides	1.1	}
210	paus	1.10
211	loizides	1.1	if (!noSimParticle) { //loop through sim match candidates
212	loizides	1.29	if (verbose_>1)
213			printf("Applying track-sim association\n");
214	loizides	1.1	for (vector<pair<TrackingParticleRef, double> >::const_iterator simRefPair=simRefs.begin();
215			simRefPair != simRefs.end(); ++simRefPair)
216	paus	1.10	if (simRefPair->second > 0.5) // require more than 50% shared hits between reco and sim
217	bendavid	1.17	outTrack->SetMCPart(trackingMap_->GetMit(simRefPair->first)); //add reco->sim reference
218	loizides	1.1	}
219			}
220			}
221			tracks_->Trim();
222			}