#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 ()

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

boost::function< void(const BranchDescription &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

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

typedef WorkerT< EDProducer >	WorkerType

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::EDProducer
CurrentProcessingContext const *	currentContext () const

Protected Member Functions inherited from edm::ProducerBase
template<class TProducer , class TMethod >
void	callWhenNewProductsRegistered (TProducer *iProd, TMethod iMethod)

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.

                                                                             : 
   theTrackCollection(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 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(), interpolateCardsSimple::histo, scaleCards::hmax, i, LogTrace, edm::Handle< T >::product(), edm::Event::put(), python.multivaluedict::sort(), testEve_cfg::tracks, and tracks_.

 {
   // Get pixel tracks
   edm::Handle<reco::TrackCollection> trackCollection;
   ev.getByLabel(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 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.

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation