#include <RecoVertex/PrimaryVertexProducerAlgorithm/src/PrimaryVertexProducerAlgorithm.cc>

Inheritance diagram for PrimaryVertexProducerAlgorithm:

Public Member Functions
virtual PrimaryVertexProducerAlgorithm *	clone () const

	PrimaryVertexProducerAlgorithm (const edm::ParameterSet &)

virtual std::vector < TransientVertex >	vertices (const std::vector< reco::TransientTrack > &tracks) const

virtual std::vector < TransientVertex >	vertices (const std::vector< reco::TransientTrack > &tracks, const reco::BeamSpot &beamSpot) const

	~PrimaryVertexProducerAlgorithm ()

Public Member Functions inherited from VertexReconstructor
	VertexReconstructor ()

virtual std::vector < TransientVertex >	vertices (const std::vector< reco::TransientTrack > &primaries, const std::vector< reco::TransientTrack > &tracks, const reco::BeamSpot &spot) const

virtual	~VertexReconstructor ()

Private Attributes
bool	fapply_finder

bool	fFailsafe

double	fMinNdof

bool	fUseBeamConstraint

bool	fVerbose

edm::ParameterSet	theConfig

KalmanTrimmedVertexFinder	theFinder

VertexFitter< 5 > *	theFitter

TrackClusterizerInZ *	theTrackClusterizer

TrackFilterForPVFindingBase *	theTrackFilter

VertexCompatibleWithBeam	theVertexSelector

Detailed Description

Description: finds primary vertices, compatible with the beam line

Implementation: <Notes on="" implementation>="">

Definition at line 36 of file PrimaryVertexProducerAlgorithm.h.

Constructor & Destructor Documentation

PrimaryVertexProducerAlgorithm::PrimaryVertexProducerAlgorithm ( const edm::ParameterSet & conf )

explicit

Definition at line 22 of file PrimaryVertexProducerAlgorithm.cc.

References algorithm(), fapply_finder, fFailsafe, fMinNdof, fUseBeamConstraint, fVerbose, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), KalmanTrimmedVertexFinder::setParameters(), theFinder, theFitter, theTrackClusterizer, and theTrackFilter.

Referenced by clone().

   : theConfig(conf), 
     theVertexSelector(VertexDistanceXY(), 
               conf.getParameter<edm::ParameterSet>("PVSelParameters").getParameter<double>("maxDistanceToBeam"))
 {
   edm::LogInfo("PVDebugInfo") 
     << "PVSelParameters::maxDistanceToBeam = " 
     << conf.getParameter<edm::ParameterSet>("PVSelParameters").getParameter<double>("maxDistanceToBeam") << "\n";
 
 
   fUseBeamConstraint = conf.getParameter<bool>("useBeamConstraint");
   fVerbose           = conf.getUntrackedParameter<bool>("verbose", false);
   fMinNdof           = conf.getParameter<double>("minNdof");
   fFailsafe          = true; //conf.getUntrackedParameter<bool>("failsafe",true);
 
 
   // select and configure the track selection
   std::string trackSelectionAlgorithm=conf.getParameter<edm::ParameterSet>("TkFilterParameters").getParameter<std::string>("algorithm");
   if(trackSelectionAlgorithm=="filter"){
     theTrackFilter= new TrackFilterForPVFinding( conf.getParameter<edm::ParameterSet>("TkFilterParameters") );
   }else if (trackSelectionAlgorithm=="filterWithThreshold"){
     theTrackFilter= new HITrackFilterForPVFinding(conf.getParameter<edm::ParameterSet>("TkFilterParameters"));
   }else{
     throw VertexException("PrimaryVertexProducerAlgorithm: unknown track selection algorithm: " + trackSelectionAlgorithm);  
   }
 
 
   // select and configure the track clusterizer
   std::string clusteringAlgorithm=conf.getParameter<edm::ParameterSet>("TkClusParameters").getParameter<std::string>("algorithm");
   if (clusteringAlgorithm=="gap"){
     theTrackClusterizer = new GapClusterizerInZ(conf.getParameter<edm::ParameterSet>("TkClusParameters").getParameter<edm::ParameterSet>("TkGapClusParameters"));
   }else if(clusteringAlgorithm=="DA"){
     theTrackClusterizer = new DAClusterizerInZ(conf.getParameter<edm::ParameterSet>("TkClusParameters").getParameter<edm::ParameterSet>("TkDAClusParameters"));
   }else{
     throw VertexException("PrimaryVertexProducerAlgorithm: unknown clustering algorithm: " + clusteringAlgorithm);  
   }
 
   // select and configure the vertex fitter
   std::string algorithm = conf.getParameter<std::string>("algorithm");
   fapply_finder = false;
   if (algorithm == "TrimmedKalmanFinder") {
     fapply_finder = true;
     theFinder.setParameters(conf.getParameter<edm::ParameterSet>("VtxFinderParameters"));
   } else if (algorithm=="KalmanVertexFitter") {
     theFitter=new KalmanVertexFitter();
   } else if( algorithm=="AdaptiveVertexFitter") {
     theFitter=new AdaptiveVertexFitter();
   } else {
     throw VertexException("PrimaryVertexProducerAlgorithm: unknown algorithm: " + algorithm);  
   }
 
   edm::LogInfo("PVDebugInfo") 
     << "Using " << algorithm << "\n";
   edm::LogInfo("PVDebugInfo") 
     << "beam-constraint  " << fUseBeamConstraint << "\n"; 
 
   edm::LogInfo("PVDebugInfo") 
     << "PV producer algorithm initialization: done" << "\n";
 
 }

PrimaryVertexProducerAlgorithm::~PrimaryVertexProducerAlgorithm ( )

Definition at line 85 of file PrimaryVertexProducerAlgorithm.cc.

References theFitter, theTrackClusterizer, and theTrackFilter.

 {
   if (theFitter) delete theFitter;
   if (theTrackFilter) delete theTrackFilter;
   if (theTrackClusterizer) delete theTrackClusterizer;
 }

Member Function Documentation

virtual PrimaryVertexProducerAlgorithm* PrimaryVertexProducerAlgorithm::clone ( void ) const

inlinevirtual

Clone method

Implements VertexReconstructor.

Definition at line 55 of file PrimaryVertexProducerAlgorithm.h.

References PrimaryVertexProducerAlgorithm().

                                                          {
     return new PrimaryVertexProducerAlgorithm(*this);
   }

std::vector< TransientVertex > PrimaryVertexProducerAlgorithm::vertices ( const std::vector< reco::TransientTrack > & tracks ) const

virtual

Find primary vertices

Implements VertexReconstructor.

Definition at line 99 of file PrimaryVertexProducerAlgorithm.cc.

 {
 
    throw VertexException("PrimaryVertexProducerAlgorithm: cannot make a Primary Vertex without a beam spot constraint " );
 
   /*  std::cout<< "PrimaryVertexProducer::vertices> Obsolete function, using dummy beamspot " << std::endl;
     reco::BeamSpot dummyBeamSpot;
     dummyBeamSpot.dummy();
     return vertices(tracks,dummyBeamSpot); */
    return std::vector<TransientVertex>();
 }

std::vector< TransientVertex > PrimaryVertexProducerAlgorithm::vertices	(	const std::vector< reco::TransientTrack > &	t,
		const reco::BeamSpot &
	)		const

virtual

Reconstruct vertices, exploiting the beamspot constraint for the primary vertex

Reimplemented from VertexReconstructor.

Definition at line 113 of file PrimaryVertexProducerAlgorithm.cc.

References TrackClusterizerInZ::clusterize(), gather_cfg::cout, GlobalErrorBase< T, ErrorWeightType >::cxx(), GlobalErrorBase< T, ErrorWeightType >::cyy(), GlobalErrorBase< T, ErrorWeightType >::czz(), TransientVertex::degreesOfFreedom(), VertexState::error(), fapply_finder, fFailsafe, fMinNdof, fUseBeamConstraint, fVerbose, i, TransientVertex::isValid(), GlobalErrorBase< T, ErrorWeightType >::matrix(), TransientVertex::position(), TrackFilterForPVFindingBase::select(), VertexHigherPtSquared::sumPtSquared(), theFinder, theFitter, theTrackClusterizer, theTrackFilter, theVertexSelector, v, VertexFitter< N >::vertex(), KalmanTrimmedVertexFinder::vertices(), PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

 {
   bool validBS = true;
   VertexState beamVertexState(beamSpot);
   if ( (beamVertexState.error().cxx() <= 0.) || 
        (beamVertexState.error().cyy() <= 0.) ||
        (beamVertexState.error().czz() <= 0.) ) {
     validBS = false;
     edm::LogError("UnusableBeamSpot") << "Beamspot with invalid errors "<<beamVertexState.error().matrix();
   }
 
   if ( fapply_finder) {
         return theFinder.vertices( tracks );
   }
   std::vector<TransientVertex> pvs;
 
 
   // select tracks
   std::vector<TransientTrack> seltks = theTrackFilter->select( tracks );
 
 
   // clusterize tracks in Z
   std::vector< std::vector<reco::TransientTrack> > clusters =  theTrackClusterizer->clusterize(seltks);
   if (fVerbose){std::cout <<  " clustering returned  "<< clusters.size() << " clusters  from " << seltks.size() << " selected tracks" <<std::endl;}
 
 
   // look for primary vertices in each cluster
   std::vector<TransientVertex> pvCand;
   int nclu=0;
   for (std::vector< std::vector<reco::TransientTrack> >::const_iterator iclus
      = clusters.begin(); iclus != clusters.end(); iclus++) {
 
 
     TransientVertex v;
     if( fUseBeamConstraint && validBS &&((*iclus).size()>1) ){
       if (fVerbose){std::cout <<  " constrained fit with "<< (*iclus).size() << " tracks"  <<std::endl;}
       v = theFitter->vertex(*iclus, beamSpot);
       if (v.isValid() && (v.degreesOfFreedom()>=fMinNdof)) pvCand.push_back(v);
 
       if (fVerbose){
     if (v.isValid()) std::cout << "x,y,z=" << v.position().x() <<" " << v.position().y() << " " <<  v.position().z() << std::endl;
     else std::cout <<"Invalid fitted vertex\n";
       }
 
     }else if((*iclus).size()>1){
       if (fVerbose){std::cout <<  " unconstrained fit with "<< (*iclus).size() << " tracks"  << std::endl;}
 
       v = theFitter->vertex(*iclus); 
       if (v.isValid() && (v.degreesOfFreedom()>=fMinNdof)) pvCand.push_back(v);
 
       if (fVerbose){
     if (v.isValid()) std::cout << "x,y,z=" << v.position().x() <<" " << v.position().y() << " " <<  v.position().z() << std::endl;
     else std::cout <<"Invalid fitted vertex\n";
       }
 
     }
 
     nclu++;
 
   }// end of cluster loop
 
   if(fVerbose){
     std::cout << "PrimaryVertexProducerAlgorithm::vertices  candidates =" << pvCand.size() << std::endl;
   }
 
 
 
   // select vertices compatible with beam
   int npv=0;
   for (std::vector<TransientVertex>::const_iterator ipv = pvCand.begin();
        ipv != pvCand.end(); ipv++) {
     if(fVerbose){
       std::cout << "PrimaryVertexProducerAlgorithm::vertices cand " << npv++ << " sel=" <<
     (validBS && theVertexSelector(*ipv,beamVertexState)) << "   z="  << ipv->position().z() << std::endl;
     }
     if (!validBS || theVertexSelector(*ipv,beamVertexState)) pvs.push_back(*ipv);
   }
 
 
   if(pvs.size()>0){
 
     // sort vertices by pt**2  vertex (aka signal vertex tagging)
     // sort(pvs.begin(), pvs.end(), VertexHigherPtSquared());
 
     // avoid re-evaluating sumptsquared for each comparison
     VertexHigherPtSquared V;
     std::vector< std::pair< double , unsigned int > > ptsqpv;
     for(unsigned int i=0; i<pvs.size(); i++){ ptsqpv.push_back( std::make_pair(V.sumPtSquared(pvs.at(i)), i));}
     std::stable_sort(ptsqpv.begin(), ptsqpv.end());
     std::vector<TransientVertex> pvsorted( pvs.size());
     for(unsigned int i=0; i<pvs.size(); i++){ pvsorted.push_back(pvs.at(ptsqpv[i].second));}
     return pvsorted;
 
   }else{
 
     if (    fFailsafe 
         && (seltks.size()>1) 
         && ( (clusters.size()!=1)  ||  ( (clusters.size()==1) && (clusters.begin()->size()<seltks.size())) )
         )
       { 
     // if no vertex was found, try fitting all selected tracks, unless this has already been tried
     // in low/no pile-up situations with low multiplicity vertices, this can recover vertices lost in clustering
     // with very small zSep. Only makes sense when used with a robust fitter, like the AdaptiveVertexFitter
     TransientVertex v;
     if( fUseBeamConstraint && validBS ){
       v = theFitter->vertex(seltks, beamSpot);
     }else{
       v = theFitter->vertex(seltks); 
     }
     if (fVerbose){ std::cout << "PrimaryVertexProducerAlgorithm: failsafe vertex "
                 <<"  tracks="  << seltks.size()
                 <<"  valid()=" << v.isValid() << " ndof()=" << v.degreesOfFreedom() 
                 <<"  selected="<< theVertexSelector(v,beamVertexState) << std::endl; }
     if ( v.isValid() && (v.degreesOfFreedom()>=fMinNdof) && (theVertexSelector(v,beamVertexState)) )  pvs.push_back(v);
       }
     
   }
 
 
   return pvs;
   
 }