CMS 3D CMS Logo
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
List of all members | Public Member Functions | Private Member Functions | Private Attributes
HIPixelMedianVtxProducer Class Reference

#include <HIPixelMedianVtxProducer.h>

Inheritance diagram for HIPixelMedianVtxProducer:
edm::EDProducer edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 HIPixelMedianVtxProducer (const edm::ParameterSet &ps)
 
virtual void produce (edm::Event &ev, const edm::EventSetup &es)
 
 ~HIPixelMedianVtxProducer ()
 
- Public Member Functions inherited from edm::EDProducer
 EDProducer ()
 
ModuleDescription const & moduleDescription () const
 
virtual ~EDProducer ()
 
- Public Member Functions inherited from edm::ProducerBase
 ProducerBase ()
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription
const &)> 		registrationCallback () const
 used by the fwk to register list of products More...
 
virtual ~ProducerBase ()
 
- Public Member Functions inherited from edm::EDConsumerBase
 EDConsumerBase ()
 
ProductHolderIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
 
std::vector
< ProductHolderIndexAndSkipBit >
const & 		itemsToGetFromEvent () const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
bool registeredToConsume (ProductHolderIndex, bool, BranchType) const
 
bool registeredToConsumeMany (TypeID const &, BranchType) const
 
void updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
 
virtual ~EDConsumerBase ()
 

Private Member Functions

void beginJob ()
 

Private Attributes

int theFitBinning
 
double theFitMaxZ
 
int theFitThreshold
 
int thePeakFindBinning
 
double thePeakFindMaxZ
 
unsigned int thePeakFindThresh
 
double thePtMin
 
edm::InputTag theTrackCollection
 

Additional Inherited Members

- Public Types inherited from edm::EDProducer
typedef EDProducer ModuleType
 
- Public Types inherited from edm::ProducerBase
typedef
ProductRegistryHelper::TypeLabelList 		TypeLabelList
 
- Static Public Member Functions inherited from edm::EDProducer
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 
- Protected Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
- Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > consumes (edm::InputTag const &tag)
 
EDGetToken consumes (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken consumes (TypeToGet const &id, edm::InputTag const &tag)
 
ConsumesCollector consumesCollector ()
 Use a ConsumesCollector to gather consumes information from helper functions. More...
 
template<typename ProductType , BranchType B = InEvent>
void consumesMany ()
 
void consumesMany (const TypeToGet &id)
 
template<BranchType B>
void consumesMany (const TypeToGet &id)
 
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > mayConsume (edm::InputTag const &tag)
 
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 

Detailed Description

Definition at line 14 of file HIPixelMedianVtxProducer.h.

Constructor & Destructor Documentation

HIPixelMedianVtxProducer::HIPixelMedianVtxProducer ( const edm::ParameterSet &  ps)
explicit

Definition at line 21 of file HIPixelMedianVtxProducer.cc.

21  :
22  theTrackCollection(ps.getParameter<edm::InputTag>("TrackCollection")),
23  thePtMin(ps.getParameter<double>("PtMin")),
24  thePeakFindThresh(ps.getParameter<unsigned int>("PeakFindThreshold")),
25  thePeakFindMaxZ(ps.getParameter<double>("PeakFindMaxZ")),
26  thePeakFindBinning(ps.getParameter<int>("PeakFindBinsPerCm")),
27  theFitThreshold(ps.getParameter<int>("FitThreshold")),
28  theFitMaxZ(ps.getParameter<double>("FitMaxZ")),
29  theFitBinning(ps.getParameter<int>("FitBinsPerCm"))
30 {
31  produces<reco::VertexCollection>();
32 }
edm::ParameterSet::getParameter
T getParameter(std::string const &) const
HIPixelMedianVtxProducer::~HIPixelMedianVtxProducer ( )
inline

Definition at line 18 of file HIPixelMedianVtxProducer.h.

18 {};

Member Function Documentation

void HIPixelMedianVtxProducer::beginJob ( void  )
inlineprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 22 of file HIPixelMedianVtxProducer.h.

22 {};
void HIPixelMedianVtxProducer::produce ( edm::Event &  ev,
const edm::EventSetup &  es 
)
virtual

Implements edm::EDProducer.

Definition at line 36 of file HIPixelMedianVtxProducer.cc.

References python.connectstrParser::f1, edm::Event::getByLabel(), timingPdfMaker::histo, i, LogTrace, pileupDistInMC::num, edm::Handle< T >::product(), RecoTauCleanerPlugins::pt, edm::Event::put(), python.multivaluedict::sort(), testEve_cfg::tracks, and tracks_.

37 {
38  // Get pixel tracks
39  edm::Handle<reco::TrackCollection> trackCollection;
40  ev.getByLabel(theTrackCollection, trackCollection);
41  const reco::TrackCollection tracks_ = *(trackCollection.product());
42 
43  // Select tracks above minimum pt
44  std::vector<const reco::Track *> tracks;
45  for (unsigned int i=0; i<tracks_.size(); i++) {
46  if (tracks_[i].pt() < thePtMin && std::fabs(tracks_[i].vz()) < 100000.) continue;
47  reco::TrackRef recTrack(trackCollection, i);
48  tracks.push_back( &(*recTrack));
49  }
50 
51  LogTrace("MinBiasTracking")
52  << " [VertexProducer] selected tracks: "
53  << tracks.size() << " (out of " << tracks_.size()
54  << ") above pt = " << thePtMin;
55 
56  // Output vertex collection
57  std::auto_ptr<reco::VertexCollection> vertices(new reco::VertexCollection);
58 
59  // No tracks -> return empty collection
60  if(tracks.size() == 0) {
61  ev.put(vertices);
62  return;
63  }
64 
65  // Sort tracks according to vertex z position
66  std::sort(tracks.begin(), tracks.end(), ComparePairs());
67 
68  // Calculate median vz
69  float med;
70  if(tracks.size() % 2 == 0)
71  med = (tracks[tracks.size()/2-1]->vz() + tracks[tracks.size()/2]->vz())/2;
72  else
73  med = tracks[tracks.size()/2 ]->vz();
74 
75  LogTrace("MinBiasTracking")
76  << " [vertex position] median = " << med << " cm";
77 
78  // In high multiplicity events, fit around most probable position
79  if(tracks.size() > thePeakFindThresh) {
80 
81  // Find maximum bin
82  TH1F hmax("hmax","hmax",thePeakFindBinning*2.0*thePeakFindMaxZ,-1.0*thePeakFindMaxZ,thePeakFindMaxZ);
83 
84  for(std::vector<const reco::Track *>::const_iterator
85  track = tracks.begin(); track!= tracks.end(); track++)
86  if(fabs((*track)->vz()) < thePeakFindMaxZ)
87  hmax.Fill((*track)->vz());
88 
89  int maxBin = hmax.GetMaximumBin();
90 
91  LogTrace("MinBiasTracking")
92  << " [vertex position] most prob = "
93  << hmax.GetBinCenter(maxBin) << " cm";
94 
95  // Find 3-bin weighted average
96  float num=0.0, denom=0.0;
97  for(int i=-1; i<=1; i++) {
98  num += hmax.GetBinContent(maxBin+i)*hmax.GetBinCenter(maxBin+i);
99  denom += hmax.GetBinContent(maxBin+i);
100  }
101 
102  if(denom==0)
103  {
104  reco::Vertex::Error err;
105  err(2,2) = 0.1 * 0.1;
106  reco::Vertex ver(reco::Vertex::Point(0,0,99999.),
107  err, 0, 0, 1);
108  vertices->push_back(ver);
109  ev.put(vertices);
110  return;
111  }
112 
113  float nBinAvg = num/denom;
114 
115  // Center fit at 3-bin weighted average around max bin
116  med = nBinAvg;
117 
118  LogTrace("MinBiasTracking")
119  << " [vertex position] 3-bin weighted average = "
120  << nBinAvg << " cm";
121  }
122 
123  // Bin vz-values around most probable value (or median) for fitting
124  TH1F histo("histo","histo", theFitBinning*2.0*theFitMaxZ,-1.0*theFitMaxZ,theFitMaxZ);
125  histo.Sumw2();
126 
127  for(std::vector<const reco::Track *>::const_iterator
128  track = tracks.begin(); track!= tracks.end(); track++)
129  if(fabs((*track)->vz() - med) < theFitMaxZ)
130  histo.Fill((*track)->vz() - med);
131 
132  LogTrace("MinBiasTracking")
133  << " [vertex position] most prob for fit = "
134  << med + histo.GetBinCenter(histo.GetMaximumBin()) << " cm";
135 
136  // If there are very few entries, don't do the fit
137  if(histo.GetEntries() <= theFitThreshold) {
138 
139  LogTrace("MinBiasTracking")
140  << " [vertex position] Fewer than" << theFitThreshold
141  << " entries in fit histogram. Returning median.";
142 
143  reco::Vertex::Error err;
144  err(2,2) = 0.1 * 0.1;
145  reco::Vertex ver(reco::Vertex::Point(0,0,med),
146  err, 0, 1, 1);
147  vertices->push_back(ver);
148  ev.put(vertices);
149  return;
150  }
151 
152  // Otherwise, there are enough entries to refine the estimate with a fit
153  TF1 f1("f1","[0]*exp(-0.5 * ((x-[1])/[2])^2) + [3]");
154  f1.SetParameters(10.,0.,0.02,0.002*tracks.size());
155  f1.SetParLimits(1,-0.1,0.1);
156  f1.SetParLimits(2,0.001,0.05);
157  f1.SetParLimits(3,0.0,0.005*tracks.size());
158 
159  histo.Fit("f1","QN");
160 
161  LogTrace("MinBiasTracking")
162  << " [vertex position] fitted = "
163  << med + f1.GetParameter(1) << " +- " << f1.GetParError(1) << " cm";
164 
165  reco::Vertex::Error err;
166  err(2,2) = f1.GetParError(1) * f1.GetParError(1);
167  reco::Vertex ver(reco::Vertex::Point(0,0,med + f1.GetParameter(1)),
168  err, 0, 1, 1);
169  vertices->push_back(ver);
170 
171  ev.put(vertices);
172  return;
173 
174 }
i
int i
Definition: DBlmapReader.cc:9
tracks_
const std::vector< reco::PFCandidatePtr > & tracks_
Definition: RecoTauIsolationMasking.cc:47
reco::TrackCollection
std::vector< Track > TrackCollection
collection of Tracks
Definition: TrackFwd.h:10
reco::Vertex::Error
math::Error< dimension >::type Error
covariance error matrix (3x3)
Definition: Vertex.h:43
reco::VertexCollection
std::vector< Vertex > VertexCollection
collection of Vertex objects
Definition: VertexFwd.h:9
edm::Event::put
OrphanHandle< PROD > put(std::auto_ptr< PROD > product)
Put a new product.
Definition: Event.h:116
reco::Vertex::Point
math::XYZPoint Point
point in the space
Definition: Vertex.h:39
edm::Event::getByLabel
bool getByLabel(InputTag const &tag, Handle< PROD > &result) const
Definition: Event.h:390
LogTrace
#define LogTrace(id)
Definition: MessageLogger.h:502
testEve_cfg.tracks
tuple tracks
Definition: testEve_cfg.py:39
edm::Handle::product
T const * product() const
Definition: Handle.h:81

Member Data Documentation

int HIPixelMedianVtxProducer::theFitBinning
private

Definition at line 31 of file HIPixelMedianVtxProducer.h.

double HIPixelMedianVtxProducer::theFitMaxZ
private

Definition at line 30 of file HIPixelMedianVtxProducer.h.

int HIPixelMedianVtxProducer::theFitThreshold
private

Definition at line 29 of file HIPixelMedianVtxProducer.h.

int HIPixelMedianVtxProducer::thePeakFindBinning
private

Definition at line 28 of file HIPixelMedianVtxProducer.h.

double HIPixelMedianVtxProducer::thePeakFindMaxZ
private

Definition at line 27 of file HIPixelMedianVtxProducer.h.

unsigned int HIPixelMedianVtxProducer::thePeakFindThresh
private

Definition at line 26 of file HIPixelMedianVtxProducer.h.

double HIPixelMedianVtxProducer::thePtMin
private

Definition at line 25 of file HIPixelMedianVtxProducer.h.

edm::InputTag HIPixelMedianVtxProducer::theTrackCollection
private

Definition at line 22 of file HIPixelMedianVtxProducer.h.

Generated for CMSSW Reference Manual by doxygen 1.8.5