#include <TrackVertexArbitratration.h>

Public Member Functions
	TrackVertexArbitration (const edm::ParameterSet &params)

reco::VertexCollection	trackVertexArbitrator (edm::Handle< reco::BeamSpot > &beamSpot, const reco::Vertex &pv, edm::ESHandle< TransientTrackBuilder > &trackBuilder, const edm::RefVector< reco::TrackCollection > &selectedTracks, reco::VertexCollection &secondaryVertices)

Private Member Functions
bool	trackFilterArbitrator (const reco::TrackRef &track) const

Private Attributes
edm::InputTag	beamSpotCollection

double	distCut

double	dLenFraction

double	dRCut

edm::InputTag	primaryVertexCollection

edm::InputTag	secondaryVertexCollection

double	sigCut

edm::InputTag	trackCollection

Detailed Description

Definition at line 39 of file TrackVertexArbitratration.h.

Constructor & Destructor Documentation

TrackVertexArbitration::TrackVertexArbitration ( const edm::ParameterSet & params )

Definition at line 5 of file TrackVertexArbitratration.cc.

                                                                             :
     primaryVertexCollection   (params.getParameter<edm::InputTag>("primaryVertices")),
     secondaryVertexCollection (params.getParameter<edm::InputTag>("secondaryVertices")),
     trackCollection           (params.getParameter<edm::InputTag>("tracks")),
         beamSpotCollection        (params.getParameter<edm::InputTag>("beamSpot")),
     dRCut                     (params.getParameter<double>("dRCut")),
     distCut                   (params.getParameter<double>("distCut")),
     sigCut                    (params.getParameter<double>("sigCut")),
     dLenFraction              (params.getParameter<double>("dLenFraction"))
 {
     
 }

Member Function Documentation

bool TrackVertexArbitration::trackFilterArbitrator ( const reco::TrackRef & track ) const

private

Definition at line 18 of file TrackVertexArbitratration.cc.

Referenced by trackVertexArbitrator().

 {
         if (track->hitPattern().trackerLayersWithMeasurement() < 4)
                 return false;
         if (track->pt() < 0.4 )
                 return false;
         if (track->hitPattern().numberOfValidPixelHits() < 1)
                 return false;
 
         return true;
 }

reco::VertexCollection TrackVertexArbitration::trackVertexArbitrator	(	edm::Handle< reco::BeamSpot > &	beamSpot,
		const reco::Vertex &	pv,
		edm::ESHandle< TransientTrackBuilder > &	trackBuilder,
		const edm::RefVector< reco::TrackCollection > &	selectedTracks,
		reco::VertexCollection &	secondaryVertices
	)

Definition at line 34 of file TrackVertexArbitratration.cc.

References IPTools::absoluteImpactParameter3D(), IPTools::absoluteTransverseImpactParameter(), gather_cfg::cout, Geom::deltaR(), distCut, dLenFraction, PFRecoTauDiscriminationAgainstElectronDeadECAL_cfi::dR, dRCut, TransientVertex::isValid(), edm::Ref< C, T, F >::key(), reco::Vertex::p4(), reco::Vertex::position(), dt_dqm_sourceclient_common_cff::reco, reco::Track::residualX(), reco::Track::residualY(), TrackCollections2monitor_cff::selectedTracks, reco::TransientTrack::setBeamSpot(), sigCut, edm::RefVector< C, T, F >::size(), reco::TransientTrack::track(), trackFilterArbitrator(), groupFilesInBlocks::tt, Measurement1D::value(), AdaptiveVertexFitter::vertex(), and w().

Referenced by TrackVertexArbitrator::produce().

 {
     using namespace reco;
 
          //std::cout << "PV: " << pv.position() << std::endl;
         VertexDistance3D dist;
 
   double sigmacut = 3.0;
   double Tini     = 256.;
   double ratio    = 0.25;
 
   AdaptiveVertexFitter theAdaptiveFitter(
                                             GeometricAnnealing(sigmacut, Tini, ratio),
                                             DefaultLinearizationPointFinder(),
                                             KalmanVertexUpdator<5>(),
                                             KalmanVertexTrackCompatibilityEstimator<5>(),
                                             KalmanVertexSmoother() );
 
 
 
     reco::VertexCollection recoVertices;
 
        VertexDistance3D vdist;
 
 for(std::vector<reco::Vertex>::const_iterator sv = secondaryVertices.begin();
         sv != secondaryVertices.end(); ++sv) {
 /*          recoVertices->push_back(*sv);
         
 
        for(std::vector<reco::Vertex>::iterator sv = recoVertices->begin();
         sv != recoVertices->end(); ++sv) {
 */
         GlobalPoint ppv(pv.position().x(),pv.position().y(),pv.position().z());
         GlobalPoint ssv(sv->position().x(),sv->position().y(),sv->position().z());
             GlobalVector flightDir = ssv-ppv;
 //            std::cout << "Vertex : " << sv-secondaryVertices->begin() << " " << sv->position() << std::endl;
             Measurement1D dlen= vdist.distance(pv,*sv);
             std::vector<reco::TransientTrack>  selTracks;
        for(unsigned int itrack = 0; itrack < selectedTracks.size(); itrack++){
             TrackRef ref = (selectedTracks)[itrack];
 
         //for(TrackCollection::const_iterator track = tracks->begin();
           //  track != tracks->end(); ++track) {
 
             //    TrackRef ref(tracks, track - tracks->begin());
             if (!trackFilterArbitrator(ref))                         continue;
 
                 TransientTrack tt = trackBuilder->build(ref);
                 tt.setBeamSpot(*beamSpot);
             float w = sv->trackWeight(ref);
                 std::pair<bool,Measurement1D> ipv = IPTools::absoluteImpactParameter3D(tt,pv);
                 std::pair<bool,Measurement1D> itpv = IPTools::absoluteTransverseImpactParameter(tt,pv);
                 std::pair<bool,Measurement1D> isv = IPTools::absoluteImpactParameter3D(tt,*sv);
         float dR = Geom::deltaR(flightDir,tt.track()); //.eta(), flightDir.phi(), tt.track().eta(), tt.track().phi());
 
                 if( w > 0 || ( isv.second.significance() < sigCut && isv.second.value() < distCut && isv.second.value() < dlen.value()*dLenFraction ) )
                 {
 
                   if(( isv.second.value() < ipv.second.value()  ) && isv.second.value() < distCut && isv.second.value() < dlen.value()*dLenFraction 
                   && dR < dRCut  ) 
                   {
 #ifdef VTXDEBUG
                      if(w > 0.5) std::cout << " = ";
                     else std::cout << " + ";
 #endif 
                      selTracks.push_back(tt);
                   } else
                   {
 #ifdef VTXDEBUG
                      if(w > 0.5 && isv.second.value() > ipv.second.value() ) std::cout << " - ";
                   else std::cout << "   ";
 #endif
                      //add also the tracks used in previous fitting that are still closer to Sv than Pv 
                      if(w > 0.5 && isv.second.value() <= ipv.second.value() && dR < dRCut) {  
                        selTracks.push_back(tt);
 #ifdef VTXDEBUG
                        std::cout << " = ";
 #endif
                      }
                      if(w > 0.5 && isv.second.value() <= ipv.second.value() && dR >= dRCut) {
 #ifdef VTXDEBUG
                        std::cout << " - ";
 #endif
 
                      }
 
                     
                   }
 #ifdef VTXDEBUG
 
                   std::cout << "t : " << itrack << " ref " << ref.key() << " w: " << w 
                   << " svip: " << isv.second.significance() << " " << isv.second.value()  
                   << " pvip: " << ipv.second.significance() << " " << ipv.second.value()  << " res " << tt.track().residualX(0)   << "," << tt.track().residualY(0) 
                   << " tpvip: " << itpv.second.significance() << " " << itpv.second.value()  << " dr: "   << dR << std::endl;
 #endif
  
                 }
                else
                  {
 #ifdef VTXDEBUG
 
                   std::cout << " . t : " << itrack << " ref " << ref.key()<<  " w: " << w 
                   << " svip: " << isv.second.significance() << " " << isv.second.value()  
                   << " pvip: " << ipv.second.significance() << " " << ipv.second.value()  << " dr: "   << dR << std::endl;
 #endif
 
                  }
            }      
 
            if(selTracks.size() >= 2)
               { 
                  TransientVertex singleFitVertex;
                  singleFitVertex = theAdaptiveFitter.vertex(selTracks,ssv);
                 if(singleFitVertex.isValid())  { recoVertices.push_back(reco::Vertex(singleFitVertex));
 
 #ifdef VTXDEBUG
                 const reco::Vertex & extVertex = recoVertices.back();
                       GlobalVector vtxDir = GlobalVector(extVertex.p4().X(),extVertex.p4().Y(),extVertex.p4().Z());
 
 
         std::cout << " pointing : " << Geom::deltaR(extVertex.position() - pv.position(), vtxDir) << std::endl;
 #endif
         }
               } 
     }
     return recoVertices;
 }