CMS 3D CMS Logo
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
ShallowTrackClustersProducer.cc
Go to the documentation of this file.
1 #include "CalibTracker/SiStripCommon/interface/ShallowTrackClustersProducer.h"
2 
3 #include "CalibTracker/SiStripCommon/interface/ShallowTools.h"
4 
5 #include "TrackingTools/PatternTools/interface/Trajectory.h"
6 #include "TrackingTools/PatternTools/interface/TrajTrackAssociation.h"
7 #include "TrackingTools/TrackFitters/interface/TrajectoryStateCombiner.h"
8 #include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit1D.h"
9 #include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2D.h"
10 #include "DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2D.h"
11 
12 #include "MagneticField/Engine/interface/MagneticField.h"
13 #include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
14 #include "CondFormats/SiStripObjects/interface/SiStripLorentzAngle.h"
15 #include "CondFormats/DataRecord/interface/SiStripLorentzAngleRcd.h"
16 #include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
17 #include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
18 #include "Geometry/TrackerGeometryBuilder/interface/StripGeomDetUnit.h"
19 #include "Geometry/CommonTopologies/interface/StripTopology.h"
20 #include "FWCore/Framework/interface/ESHandle.h"
21 #include "FWCore/Framework/interface/Event.h"
22 #include "FWCore/ParameterSet/interface/ParameterSet.h"
23 #include "boost/foreach.hpp"
24 
25 
26 ShallowTrackClustersProducer::ShallowTrackClustersProducer(const edm::ParameterSet& iConfig)
27  : theTracksLabel( iConfig.getParameter<edm::InputTag>("Tracks") ),
28  theClustersLabel( iConfig.getParameter<edm::InputTag>("Clusters") ),
29  Suffix ( iConfig.getParameter<std::string>("Suffix") ),
30  Prefix ( iConfig.getParameter<std::string>("Prefix") )
31 {
32  produces <std::vector<unsigned int> > ( Prefix + "trackmulti" + Suffix );
33  produces <std::vector<int> > ( Prefix + "trackindex" + Suffix );
34  produces <std::vector<float> > ( Prefix + "localtheta" + Suffix );
35  produces <std::vector<float> > ( Prefix + "localphi" + Suffix );
36  produces <std::vector<float> > ( Prefix + "localpitch" + Suffix );
37  produces <std::vector<float> > ( Prefix + "localx" + Suffix );
38  produces <std::vector<float> > ( Prefix + "localy" + Suffix );
39  produces <std::vector<float> > ( Prefix + "localz" + Suffix );
40  produces <std::vector<float> > ( Prefix + "strip" + Suffix );
41  produces <std::vector<float> > ( Prefix + "globaltheta" + Suffix );
42  produces <std::vector<float> > ( Prefix + "globalphi" + Suffix );
43  produces <std::vector<float> > ( Prefix + "globalx" + Suffix );
44  produces <std::vector<float> > ( Prefix + "globaly" + Suffix );
45  produces <std::vector<float> > ( Prefix + "globalz" + Suffix );
46  produces <std::vector<float> > ( Prefix + "insidistance"+ Suffix );
47  produces <std::vector<float> > ( Prefix + "covered" + Suffix );
48  produces <std::vector<float> > ( Prefix + "projwidth" + Suffix );
49  produces <std::vector<float> > ( Prefix + "BdotY" + Suffix );
50 
51  produces <std::vector<float> > ( Prefix + "rhlocalx" + Suffix );
52  produces <std::vector<float> > ( Prefix + "rhlocaly" + Suffix );
53  produces <std::vector<float> > ( Prefix + "rhlocalxerr" + Suffix );
54  produces <std::vector<float> > ( Prefix + "rhlocalyerr" + Suffix );
55  produces <std::vector<float> > ( Prefix + "rhglobalx" + Suffix );
56  produces <std::vector<float> > ( Prefix + "rhglobaly" + Suffix );
57  produces <std::vector<float> > ( Prefix + "rhglobalz" + Suffix );
58  produces <std::vector<float> > ( Prefix + "rhstrip" + Suffix );
59  produces <std::vector<float> > ( Prefix + "rhmerr" + Suffix );
60 
61  produces <std::vector<float> > ( Prefix + "ubstrip" + Suffix );
62  produces <std::vector<float> > ( Prefix + "ubmerr" + Suffix );
63 
64  produces <std::vector<float> > ( Prefix + "driftx" + Suffix );
65  produces <std::vector<float> > ( Prefix + "drifty" + Suffix );
66  produces <std::vector<float> > ( Prefix + "driftz" + Suffix );
67  produces <std::vector<float> > ( Prefix + "globalZofunitlocalY" + Suffix );
68 }
69 
70 void ShallowTrackClustersProducer::
71 produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
72  shallow::CLUSTERMAP clustermap = shallow::make_cluster_map(iEvent,theClustersLabel);
73 
74  int size = clustermap.size();
75  std::auto_ptr<std::vector<unsigned int> > trackmulti ( new std::vector<unsigned int>(size, 0) );
76  std::auto_ptr<std::vector<int> > trackindex ( new std::vector<int> (size, -1) );
77  std::auto_ptr<std::vector<float> > localtheta ( new std::vector<float> (size, -100) );
78  std::auto_ptr<std::vector<float> > localphi ( new std::vector<float> (size, -100) );
79  std::auto_ptr<std::vector<float> > localpitch ( new std::vector<float> (size, -100) );
80  std::auto_ptr<std::vector<float> > localx ( new std::vector<float> (size, -100) );
81  std::auto_ptr<std::vector<float> > localy ( new std::vector<float> (size, -100) );
82  std::auto_ptr<std::vector<float> > localz ( new std::vector<float> (size, -100) );
83  std::auto_ptr<std::vector<float> > strip ( new std::vector<float> (size, -100) );
84  std::auto_ptr<std::vector<float> > globaltheta ( new std::vector<float> (size, -100) );
85  std::auto_ptr<std::vector<float> > globalphi ( new std::vector<float> (size, -100) );
86  std::auto_ptr<std::vector<float> > globalx ( new std::vector<float> (size, -10000) );
87  std::auto_ptr<std::vector<float> > globaly ( new std::vector<float> (size, -10000) );
88  std::auto_ptr<std::vector<float> > globalz ( new std::vector<float> (size, -10000) );
89  std::auto_ptr<std::vector<float> > insidistance ( new std::vector<float> (size, -1) );
90  std::auto_ptr<std::vector<float> > projwidth ( new std::vector<float> (size, -1000) );
91  std::auto_ptr<std::vector<float> > BdotY ( new std::vector<float> (size, -1000) );
92  std::auto_ptr<std::vector<float> > covered ( new std::vector<float> (size, -1000) );
93  std::auto_ptr<std::vector<float> > rhlocalx ( new std::vector<float>(size, -10000 ));
94  std::auto_ptr<std::vector<float> > rhlocaly ( new std::vector<float>(size, -10000 ));
95  std::auto_ptr<std::vector<float> > rhlocalxerr ( new std::vector<float>(size, -1 ));
96  std::auto_ptr<std::vector<float> > rhlocalyerr ( new std::vector<float>(size, -1 ));
97  std::auto_ptr<std::vector<float> > rhglobalx ( new std::vector<float>(size, -10000 ));
98  std::auto_ptr<std::vector<float> > rhglobaly ( new std::vector<float>(size, -10000 ));
99  std::auto_ptr<std::vector<float> > rhglobalz ( new std::vector<float>(size, -10000 ));
100  std::auto_ptr<std::vector<float> > rhstrip ( new std::vector<float>(size, -10000 ));
101  std::auto_ptr<std::vector<float> > rhmerr ( new std::vector<float>(size, -10000 ));
102  std::auto_ptr<std::vector<float> > ubstrip ( new std::vector<float>(size, -10000 ));
103  std::auto_ptr<std::vector<float> > ubmerr ( new std::vector<float>(size, -10000 ));
104  std::auto_ptr<std::vector<float> > driftx ( new std::vector<float>(size, -10000 ));
105  std::auto_ptr<std::vector<float> > drifty ( new std::vector<float>(size, -10000 ));
106  std::auto_ptr<std::vector<float> > driftz ( new std::vector<float>(size, -10000 ));
107  std::auto_ptr<std::vector<float> > globalZofunitlocalY ( new std::vector<float>(size, -1000));
108 
109  edm::ESHandle<TrackerGeometry> theTrackerGeometry; iSetup.get<TrackerDigiGeometryRecord>().get( theTrackerGeometry );
110  edm::ESHandle<MagneticField> magfield; iSetup.get<IdealMagneticFieldRecord>().get(magfield);
111  edm::ESHandle<SiStripLorentzAngle> SiStripLorentzAngle; iSetup.get<SiStripLorentzAngleRcd>().get(SiStripLorentzAngle);
112 
113  edm::Handle<edm::View<reco::Track> > tracks; iEvent.getByLabel(theTracksLabel, tracks);
114  edm::Handle<TrajTrackAssociationCollection> associations; iEvent.getByLabel(theTracksLabel, associations);
115 
116  TrajectoryStateCombiner combiner;
117 
118  for( TrajTrackAssociationCollection::const_iterator association = associations->begin();
119  association != associations->end(); association++) {
120  const Trajectory* traj = association->key.get();
121  const reco::Track* track = association->val.get();
122 
123  BOOST_FOREACH( const TrajectoryMeasurement measurement, traj->measurements() ) {
124  const TrajectoryStateOnSurface tsos = measurement.updatedState();
125  const TrajectoryStateOnSurface unbiased = combiner(measurement.forwardPredictedState(), measurement.backwardPredictedState());
126 
127  const TrackingRecHit* hit = measurement.recHit()->hit();
128  const SiStripRecHit1D* hit1D = dynamic_cast<const SiStripRecHit1D*>(hit);
129  const SiStripRecHit2D* hit2D = dynamic_cast<const SiStripRecHit2D*>(hit);
130  const SiStripMatchedRecHit2D* matchedhit = dynamic_cast<const SiStripMatchedRecHit2D*>(hit);
131 
132  for(unsigned h=0; h<2; h++) {
133  const SiStripCluster* cluster_ptr;
134  if(!matchedhit && h==1) continue; else
135  if( matchedhit && h==0) cluster_ptr = (matchedhit->monoHit() ->cluster()).get(); else
136  if( matchedhit && h==1) cluster_ptr = (matchedhit->stereoHit()->cluster()).get(); else
137  if(hit2D) cluster_ptr = (hit2D->cluster()).get(); else
138  if(hit1D) cluster_ptr = (hit1D->cluster()).get();
139  else continue;
140 
141  shallow::CLUSTERMAP::const_iterator cluster = clustermap.find( std::make_pair( hit->geographicalId().rawId(), cluster_ptr->firstStrip() ));
142  if(cluster == clustermap.end() ) throw cms::Exception("Logic Error") << "Cluster not found: this could be a configuration error" << std::endl;
143 
144  unsigned i = cluster->second;
145  if( 0 == (trackmulti->at(i))++ ) {
146  const StripGeomDetUnit* theStripDet = dynamic_cast<const StripGeomDetUnit*>( theTrackerGeometry->idToDet( hit->geographicalId() ) );
147  LocalVector drift = shallow::drift( theStripDet, *magfield, *SiStripLorentzAngle);
148 
149  trackindex->at(i) = shallow::findTrackIndex(tracks, track);
150  localtheta->at(i) = (theStripDet->toLocal(tsos.globalDirection())).theta();
151  localphi->at(i) = (theStripDet->toLocal(tsos.globalDirection())).phi();
152  localpitch->at(i) = (theStripDet->specificTopology()).localPitch(theStripDet->toLocal(tsos.globalPosition()));
153  localx->at(i) = (theStripDet->toLocal(tsos.globalPosition())).x();
154  localy->at(i) = (theStripDet->toLocal(tsos.globalPosition())).y();
155  localz->at(i) = (theStripDet->toLocal(tsos.globalPosition())).z();
156  strip->at(i) = (theStripDet->specificTopology()).strip(theStripDet->toLocal(tsos.globalPosition()));
157  globaltheta->at(i) = tsos.globalDirection().theta();
158  globalphi->at(i) = tsos.globalDirection().phi();
159  globalx->at(i) = tsos.globalPosition().x();
160  globaly->at(i) = tsos.globalPosition().y();
161  globalz->at(i) = tsos.globalPosition().z();
162  insidistance->at(i) = 1./fabs(cos(localtheta->at(i)));
163  projwidth->at(i) = tan(localtheta->at(i))*cos(localphi->at(i));
164  BdotY->at(i) = (theStripDet->surface()).toLocal( magfield->inTesla(theStripDet->surface().position())).y();
165  covered->at(i) = drift.z()/localpitch->at(i) * fabs(projwidth->at(i) - drift.x()/drift.z());
166  rhlocalx->at(i) = hit->localPosition().x();
167  rhlocaly->at(i) = hit->localPosition().y();
168  rhlocalxerr->at(i) = sqrt(hit->localPositionError().xx());
169  rhlocalyerr->at(i) = sqrt(hit->localPositionError().yy());
170  rhglobalx->at(i) = theStripDet->toGlobal(hit->localPosition()).x();
171  rhglobaly->at(i) = theStripDet->toGlobal(hit->localPosition()).y();
172  rhglobalz->at(i) = theStripDet->toGlobal(hit->localPosition()).z();
173  rhstrip->at(i) = theStripDet->specificTopology().strip(hit->localPosition());
174  rhmerr->at(i) = sqrt(theStripDet->specificTopology().measurementError(hit->localPosition(), hit->localPositionError()).uu());
175  ubstrip->at(i) = theStripDet->specificTopology().strip(unbiased.localPosition());
176  ubmerr->at(i) = sqrt(theStripDet->specificTopology().measurementError(unbiased.localPosition(), unbiased.localError().positionError()).uu());
177  driftx->at(i) = drift.x();
178  drifty->at(i) = drift.y();
179  driftz->at(i) = drift.z();
180  globalZofunitlocalY->at(i) = (theStripDet->toGlobal(LocalVector(0,1,0))).z();
181  }
182  }
183  }
184  }
185 
186  iEvent.put(trackmulti, Prefix + "trackmulti" + Suffix );
187  iEvent.put(trackindex, Prefix + "trackindex" + Suffix );
188  iEvent.put(localtheta, Prefix + "localtheta" + Suffix );
189  iEvent.put(localphi, Prefix + "localphi" + Suffix );
190  iEvent.put(localpitch, Prefix + "localpitch" + Suffix );
191  iEvent.put(localx, Prefix + "localx" + Suffix );
192  iEvent.put(localy, Prefix + "localy" + Suffix );
193  iEvent.put(localz, Prefix + "localz" + Suffix );
194  iEvent.put(strip, Prefix + "strip" + Suffix );
195  iEvent.put(globaltheta, Prefix + "globaltheta" + Suffix );
196  iEvent.put(globalphi, Prefix + "globalphi" + Suffix );
197  iEvent.put(globalx, Prefix + "globalx" + Suffix );
198  iEvent.put(globaly, Prefix + "globaly" + Suffix );
199  iEvent.put(globalz, Prefix + "globalz" + Suffix );
200  iEvent.put(insidistance,Prefix + "insidistance"+ Suffix );
201  iEvent.put(covered, Prefix + "covered" + Suffix );
202  iEvent.put(projwidth, Prefix + "projwidth" + Suffix );
203  iEvent.put(BdotY, Prefix + "BdotY" + Suffix );
204  iEvent.put(rhlocalx, Prefix + "rhlocalx" + Suffix );
205  iEvent.put(rhlocaly, Prefix + "rhlocaly" + Suffix );
206  iEvent.put(rhlocalxerr, Prefix + "rhlocalxerr" + Suffix );
207  iEvent.put(rhlocalyerr, Prefix + "rhlocalyerr" + Suffix );
208  iEvent.put(rhglobalx, Prefix + "rhglobalx" + Suffix );
209  iEvent.put(rhglobaly, Prefix + "rhglobaly" + Suffix );
210  iEvent.put(rhglobalz, Prefix + "rhglobalz" + Suffix );
211  iEvent.put(rhstrip, Prefix + "rhstrip" + Suffix );
212  iEvent.put(rhmerr, Prefix + "rhmerr" + Suffix );
213  iEvent.put(ubstrip, Prefix + "ubstrip" + Suffix );
214  iEvent.put(ubmerr, Prefix + "ubmerr" + Suffix );
215  iEvent.put( driftx, Prefix + "driftx" + Suffix );
216  iEvent.put( drifty, Prefix + "drifty" + Suffix );
217  iEvent.put( driftz, Prefix + "driftz" + Suffix );
218  iEvent.put( globalZofunitlocalY, Prefix + "globalZofunitlocalY" + Suffix );
219 }
i
int i
Definition: DBlmapReader.cc:9
LocalVector
Local3DVector LocalVector
Definition: LocalVector.h:12
redigi_cff.SiStripLorentzAngle
tuple SiStripLorentzAngle
Definition: redigi_cff.py:15
strip
void strip(std::string &input, const std::string &blanks=" \n\t")
Definition: stringTools.cc:16
ShallowTrackClustersProducer::ShallowTrackClustersProducer
ShallowTrackClustersProducer(const edm::ParameterSet &)
Definition: ShallowTrackClustersProducer.cc:26
TrajectoryMeasurement::forwardPredictedState
TrajectoryStateOnSurface forwardPredictedState() const
Access to forward predicted state (from fitter or builder)
Definition: TrajectoryMeasurement.h:151
shallow::findTrackIndex
int findTrackIndex(const edm::Handle< edm::View< reco::Track > > &h, const reco::Track *t)
Definition: ShallowTools.cc:29
SiStripMatchedRecHit2D::stereoHit
const SiStripRecHit2D * stereoHit() const
Definition: SiStripMatchedRecHit2D.h:14
edm::AssociationMap::const_iterator
const iterator
Definition: AssociationMap.h:47
GeomDet::toGlobal
GlobalPoint toGlobal(const Local2DPoint &lp) const
Conversion to the global R.F. from the R.F. of the GeomDet.
Definition: GeomDet.h:49
shallow::drift
LocalVector drift(const StripGeomDetUnit *, const MagneticField &, const SiStripLorentzAngle &)
Definition: ShallowTools.cc:39
PV3DBase::phi
Geom::Phi< T > phi() const
Definition: PV3DBase.h:63
PV3DBase::y
T y() const
Definition: PV3DBase.h:57
ShallowTrackClustersProducer::produce
void produce(edm::Event &, const edm::EventSetup &)
Definition: ShallowTrackClustersProducer.cc:71
TrajectoryStateOnSurface::globalPosition
GlobalPoint globalPosition() const
Definition: TrajectoryStateOnSurface.h:128
GeomDet::toLocal
LocalPoint toLocal(const GlobalPoint &gp) const
Conversion to the R.F. of the GeomDet.
Definition: GeomDet.h:64
TrajectoryMeasurement::recHit
ConstRecHitPointer recHit() const
Definition: TrajectoryMeasurement.h:166
SiStripCluster::firstStrip
uint16_t firstStrip() const
Definition: SiStripCluster.h:29
toLocal
LocalVector toLocal(const reco::Track::Vector &v, const Surface &s)
Definition: ConversionProducer.h:201
StripGeomDetUnit::specificTopology
virtual const StripTopology & specificTopology() const
Returns a reference to the strip proxy topology.
Definition: StripGeomDetUnit.cc:19
LocalTrajectoryError::positionError
LocalError positionError() const
Definition: LocalTrajectoryError.h:77
detailsBasic3DVector::z
double double double z
Definition: newBasic3DVector.h:17
StripTopology::strip
virtual float strip(const LocalPoint &) const =0
PV3DBase::theta
Geom::Theta< T > theta() const
Definition: PV3DBase.h:69
Trajectory::measurements
DataContainer const & measurements() const
Definition: Trajectory.h:169
iEvent
int iEvent
Definition: GenABIO.cc:243
Topology::measurementError
virtual MeasurementError measurementError(const LocalPoint &, const LocalError &) const =0
edm::Event::put
OrphanHandle< PROD > put(std::auto_ptr< PROD > product)
Put a new product.
Definition: Event.h:84
EnsembleCalibrationLA_cfg.Prefix
tuple Prefix
Definition: EnsembleCalibrationLA_cfg.py:18
mathSSE::sqrt
T sqrt(T t)
Definition: SSEVec.h:28
PV3DBase::z
T z() const
Definition: PV3DBase.h:58
funct::cos
Cos< T >::type cos(const T &t)
Definition: Cos.h:22
heavyFlavorValidationHarvestingSequence_cff.combiner
tuple combiner
Definition: heavyFlavorValidationHarvestingSequence_cff.py:22
funct::tan
Tan< T >::type tan(const T &t)
Definition: Tan.h:22
TrajectoryMeasurement::updatedState
TrajectoryStateOnSurface updatedState() const
Definition: TrajectoryMeasurement.h:162
TrajectoryStateOnSurface::localError
const LocalTrajectoryError & localError() const
Definition: TrajectoryStateOnSurface.h:164
edm::Event::getByLabel
bool getByLabel(InputTag const &tag, Handle< PROD > &result) const
Definition: Event.h:355
shallow::make_cluster_map
CLUSTERMAP make_cluster_map(const edm::Event &, edm::InputTag &)
Definition: ShallowTools.cc:15
testEve_cfg.tracks
tuple tracks
Definition: testEve_cfg.py:39
edm::EventSetup::get
const T & get() const
Definition: EventSetup.h:55
SiStripRecHit2D::cluster
ClusterRef const & cluster() const
Definition: SiStripRecHit2D.h:32
autoMagneticFieldProducer_cfi.magfield
tuple magfield
Definition: autoMagneticFieldProducer_cfi.py:6
shallow::CLUSTERMAP
std::map< std::pair< uint32_t, uint16_t >, unsigned int > CLUSTERMAP
Definition: ShallowTools.h:16
SiStripRecHit1D::cluster
ClusterRef const & cluster() const
Definition: SiStripRecHit1D.h:38
x
Definition: DDAxes.h:10
h
The Signals That Services Can Subscribe To This is based on ActivityRegistry h
Helper function to determine trigger accepts.
Definition: Activities.doc:4
SiStripMatchedRecHit2D::monoHit
const SiStripRecHit2D * monoHit() const
Definition: SiStripMatchedRecHit2D.h:15
PV3DBase::x
T x() const
Definition: PV3DBase.h:56
GloballyPositioned::position
const PositionType & position() const
Definition: GloballyPositioned.h:41
GeomDet::surface
virtual const BoundPlane & surface() const
The nominal surface of the GeomDet.
Definition: GeomDet.h:37
findQualityFiles.size
tuple size
Write out results.
Definition: findQualityFiles.py:322
TrajectoryMeasurement::backwardPredictedState
TrajectoryStateOnSurface backwardPredictedState() const
Access to backward predicted state (from smoother)
Definition: TrajectoryMeasurement.h:155
TrajectoryStateOnSurface::globalDirection
GlobalVector globalDirection() const
Definition: TrajectoryStateOnSurface.h:134
Generated for CMSSW Reference Manual by doxygen 1.8.5