#include <HIPixelMedianVtxProducer.h>

Inheritance diagram for HIPixelMedianVtxProducer:

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
std::vector< ConsumesInfo >	consumesInfo () const

	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

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) 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::EDGetTokenT < reco::TrackCollection >	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 12 of file HIPixelMedianVtxProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 21 of file HIPixelMedianVtxProducer.cc.

                                                                             : 
   theTrackCollection(consumes<reco::TrackCollection>(ps.getParameter<edm::InputTag>("TrackCollection"))),
   thePtMin(ps.getParameter<double>("PtMin")),
   thePeakFindThresh(ps.getParameter<unsigned int>("PeakFindThreshold")),
   thePeakFindMaxZ(ps.getParameter<double>("PeakFindMaxZ")),
   thePeakFindBinning(ps.getParameter<int>("PeakFindBinsPerCm")),
   theFitThreshold(ps.getParameter<int>("FitThreshold")),
   theFitMaxZ(ps.getParameter<double>("FitMaxZ")),
   theFitBinning(ps.getParameter<int>("FitBinsPerCm"))
 {
   produces<reco::VertexCollection>();
 }

HIPixelMedianVtxProducer::~HIPixelMedianVtxProducer ( )

inline

Definition at line 16 of file HIPixelMedianVtxProducer.h.

16 {};

Member Function Documentation

void HIPixelMedianVtxProducer::beginJob ( void )

inlineprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 20 of file HIPixelMedianVtxProducer.h.

20 {};

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::getByToken(), timingPdfMaker::histo, i, LogTrace, pileupDistInMC::num, edm::Handle< T >::product(), EnergyCorrector::pt, edm::Event::put(), python.multivaluedict::sort(), HLT_50ns_5e33_v1_cff::trackCollection, testEve_cfg::tracks, tracks_, and HLT_25ns14e33_v1_cff::vertices.

 {
   // Get pixel tracks
   edm::Handle<reco::TrackCollection> trackCollection;
   ev.getByToken(theTrackCollection, trackCollection);
   const reco::TrackCollection tracks_ = *(trackCollection.product());
   
   // Select tracks above minimum pt
   std::vector<const reco::Track *> tracks;
   for (unsigned int i=0; i<tracks_.size(); i++) {
     if (tracks_[i].pt() <  thePtMin && std::fabs(tracks_[i].vz()) < 100000.) continue;
     reco::TrackRef recTrack(trackCollection, i);
     tracks.push_back( &(*recTrack));
   }
     
   LogTrace("MinBiasTracking")
     << " [VertexProducer] selected tracks: "
     << tracks.size() << " (out of " << tracks_.size()
     << ") above pt = " << thePtMin; 
     
   // Output vertex collection
   std::auto_ptr<reco::VertexCollection> vertices(new reco::VertexCollection);
 
   // No tracks -> return empty collection
   if(tracks.size() == 0) {
       ev.put(vertices);
       return;
   }
 
   // Sort tracks according to vertex z position
   std::sort(tracks.begin(), tracks.end(), ComparePairs());
   
   // Calculate median vz
   float med;
   if(tracks.size() % 2 == 0)
     med = (tracks[tracks.size()/2-1]->vz() + tracks[tracks.size()/2]->vz())/2;
   else
     med =  tracks[tracks.size()/2  ]->vz();
   
   LogTrace("MinBiasTracking")
     << "  [vertex position] median    = " << med << " cm";
   
   // In high multiplicity events, fit around most probable position
   if(tracks.size() > thePeakFindThresh) { 
       
     // Find maximum bin
     TH1F hmax("hmax","hmax",thePeakFindBinning*2.0*thePeakFindMaxZ,-1.0*thePeakFindMaxZ,thePeakFindMaxZ);
     
     for(std::vector<const reco::Track *>::const_iterator
       track = tracks.begin(); track!= tracks.end(); track++)
       if(fabs((*track)->vz()) < thePeakFindMaxZ)
     hmax.Fill((*track)->vz());
     
     int maxBin = hmax.GetMaximumBin();
     
     LogTrace("MinBiasTracking")
       << "  [vertex position] most prob = "
       << hmax.GetBinCenter(maxBin) << " cm";
     
     // Find 3-bin weighted average
     float num=0.0, denom=0.0;
     for(int i=-1; i<=1; i++) {
       num += hmax.GetBinContent(maxBin+i)*hmax.GetBinCenter(maxBin+i);
       denom += hmax.GetBinContent(maxBin+i); 
     }
 
     if(denom==0) 
     {
       reco::Vertex::Error err;
       err(2,2) = 0.1 * 0.1;
       reco::Vertex ver(reco::Vertex::Point(0,0,99999.),
                        err, 0, 0, 1);
       vertices->push_back(ver);
       ev.put(vertices);
       return;
     }
 
     float nBinAvg = num/denom;
 
     // Center fit at 3-bin weighted average around max bin
     med = nBinAvg; 
     
     LogTrace("MinBiasTracking")
       << "  [vertex position] 3-bin weighted average = "
       << nBinAvg << " cm";
   }
   
   // Bin vz-values around most probable value (or median) for fitting
   TH1F histo("histo","histo", theFitBinning*2.0*theFitMaxZ,-1.0*theFitMaxZ,theFitMaxZ);
   histo.Sumw2();
 
   for(std::vector<const reco::Track *>::const_iterator
     track = tracks.begin(); track!= tracks.end(); track++)
     if(fabs((*track)->vz() - med) < theFitMaxZ)
       histo.Fill((*track)->vz() - med);
   
   LogTrace("MinBiasTracking")
     << "  [vertex position] most prob for fit = "
     << med + histo.GetBinCenter(histo.GetMaximumBin()) << " cm";
 
   // If there are very few entries, don't do the fit
   if(histo.GetEntries() <= theFitThreshold) {
 
     LogTrace("MinBiasTracking")
       << "  [vertex position] Fewer than" << theFitThreshold
       <<  " entries in fit histogram. Returning median.";
     
     reco::Vertex::Error err;
     err(2,2) = 0.1 * 0.1;
     reco::Vertex ver(reco::Vertex::Point(0,0,med),
              err, 0, 1, 1);
     vertices->push_back(ver);
     ev.put(vertices);
     return;
   }
 
   // Otherwise, there are enough entries to refine the estimate with a fit
   TF1 f1("f1","[0]*exp(-0.5 * ((x-[1])/[2])^2) + [3]");
   f1.SetParameters(10.,0.,0.02,0.002*tracks.size());
   f1.SetParLimits(1,-0.1,0.1);
   f1.SetParLimits(2,0.001,0.05);
   f1.SetParLimits(3,0.0,0.005*tracks.size());
     
   histo.Fit(&f1,"QN");
     
   LogTrace("MinBiasTracking")
     << "  [vertex position] fitted    = "
     << med + f1.GetParameter(1) << " +- " << f1.GetParError(1) << " cm";
   
   reco::Vertex::Error err;
   err(2,2) = f1.GetParError(1) * f1.GetParError(1); 
   reco::Vertex ver(reco::Vertex::Point(0,0,med + f1.GetParameter(1)),
            err, 0, 1, 1);
   vertices->push_back(ver);
   
   ev.put(vertices);
   return;
 
 }

Member Data Documentation

int HIPixelMedianVtxProducer::theFitBinning

private

Definition at line 29 of file HIPixelMedianVtxProducer.h.

double HIPixelMedianVtxProducer::theFitMaxZ

private

Definition at line 28 of file HIPixelMedianVtxProducer.h.

int HIPixelMedianVtxProducer::theFitThreshold

private

Definition at line 27 of file HIPixelMedianVtxProducer.h.

int HIPixelMedianVtxProducer::thePeakFindBinning

private

Definition at line 26 of file HIPixelMedianVtxProducer.h.

double HIPixelMedianVtxProducer::thePeakFindMaxZ

private

Definition at line 25 of file HIPixelMedianVtxProducer.h.

unsigned int HIPixelMedianVtxProducer::thePeakFindThresh

private

Definition at line 24 of file HIPixelMedianVtxProducer.h.

double HIPixelMedianVtxProducer::thePtMin

private

Definition at line 23 of file HIPixelMedianVtxProducer.h.

edm::EDGetTokenT<reco::TrackCollection> HIPixelMedianVtxProducer::theTrackCollection

private

Definition at line 20 of file HIPixelMedianVtxProducer.h.

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation