NuclearVertexBuilder Class Reference

#include <NuclearVertexBuilder.h>

Classes
class	cmpTracks

Public Member Functions
void	addSecondaryTrack (const reco::TrackRef &secTrack)
void	build (const reco::TrackRef &primaryTrack, std::vector< reco::TrackRef > &secondaryTrack)
ClosestApproachInRPhi *	closestApproach (const reco::TrackRef &primTrack, const reco::TrackRef &secTrack) const
reco::Vertex	getVertex () const
bool	isCompatible (const reco::TrackRef &secTrack) const
	NuclearVertexBuilder (const MagneticField *mag, const TransientTrackBuilder *transientTkBuilder, const edm::ParameterSet &iConfig)

Private Member Functions
void	checkEnergy (const reco::TrackRef &primTrack, std::vector< reco::TrackRef > &tC) const
void	cleanTrackCollection (const reco::TrackRef &primTrack, std::vector< reco::TrackRef > &tC) const
bool	FillVertexWithAdaptVtxFitter (const reco::TrackRef &primTrack, const std::vector< reco::TrackRef > &secTracks)
bool	FillVertexWithCrossingPoint (const reco::TrackRef &primTrack, const std::vector< reco::TrackRef > &secTracks)
void	FillVertexWithLastPrimHit (const reco::TrackRef &primTrack, const std::vector< reco::TrackRef > &secTracks)
FreeTrajectoryState	getTrajectory (const reco::TrackRef &track) const
bool	isGoodSecondaryTrack (const reco::TrackRef &primTrack, const reco::TrackRef &secTrack) const
bool	isGoodSecondaryTrack (const reco::TrackRef &secTrack, const reco::TrackRef &primTrack, const double &distOfClosestApp, const GlobalPoint &crossPoint) const

Private Attributes
double	chi2Cut_
double	DPtovPtCut_
double	minDistFromPrim_
double	minDistFromVtx_
double	shareFrac_
reco::Vertex	the_vertex
const MagneticField *	theMagField
const TransientTrackBuilder *	theTransientTrackBuilder

Detailed Description

Definition at line 16 of file NuclearVertexBuilder.h.

Constructor & Destructor Documentation

NuclearVertexBuilder::NuclearVertexBuilder ( const MagneticField *  mag, const TransientTrackBuilder *  transientTkBuilder, const edm::ParameterSet &  iConfig  )

inline

Definition at line 19 of file NuclearVertexBuilder.h.

References build().

                                                                                                                                          : 
               theMagField(mag), 
               theTransientTrackBuilder(transientTkBuilder), 
               minDistFromPrim_( iConfig.getParameter<double>("minDistFromPrimary") ),
               chi2Cut_(iConfig.getParameter<double>("chi2Cut")), 
               DPtovPtCut_(iConfig.getParameter<double>("DPtovPtCut")),
               minDistFromVtx_(iConfig.getParameter<double>("minDistFromVtx")),
               shareFrac_(iConfig.getParameter<double>("shareFrac")){}

Member Function Documentation

void NuclearVertexBuilder::addSecondaryTrack

(

const reco::TrackRef &

secTrack

)

Definition at line 190 of file NuclearVertexBuilder.cc.

References build(), the_vertex, reco::Vertex::tracks_begin(), and reco::Vertex::tracks_end().

Referenced by getVertex().

                                                                           {
        std::vector<reco::TrackRef> allSecondary;
        for( reco::Vertex::trackRef_iterator it=the_vertex.tracks_begin()+1; it != the_vertex.tracks_end(); it++) {
            allSecondary.push_back( (*it).castTo<reco::TrackRef>() );
        } 
        allSecondary.push_back( secTrack );
        build( (*the_vertex.tracks_begin()).castTo<reco::TrackRef>(), allSecondary );
}

void NuclearVertexBuilder::build	(	const reco::TrackRef &	primaryTrack,
		std::vector< reco::TrackRef > &	secondaryTrack
	)

Definition at line 10 of file NuclearVertexBuilder.cc.

References checkEnergy(), cleanTrackCollection(), FillVertexWithAdaptVtxFitter(), FillVertexWithCrossingPoint(), and FillVertexWithLastPrimHit().

Referenced by addSecondaryTrack(), and NuclearVertexBuilder().

                                                                                                   {

     cleanTrackCollection(primTrack, secTracks); 
     std::sort(secTracks.begin(),secTracks.end(),cmpTracks());
     checkEnergy(primTrack, secTracks);

     if( !secTracks.empty()) {
         if( FillVertexWithAdaptVtxFitter(primTrack, secTracks) ) return;
         else if( FillVertexWithCrossingPoint(primTrack, secTracks) ) return;
         else FillVertexWithLastPrimHit( primTrack, secTracks);
     }
     else {
       // if no secondary tracks : vertex position = position of last rechit of the primary track 
       FillVertexWithLastPrimHit( primTrack, secTracks);
     }
}

void NuclearVertexBuilder::checkEnergy ( const reco::TrackRef &  primTrack, std::vector< reco::TrackRef > &  tC  ) const

private

Definition at line 266 of file NuclearVertexBuilder.cc.

References mps_fire::i.

Referenced by build(), and getVertex().

                                                                           {
   float totalEnergy=0;
   for(size_t i=0; i< tC.size(); ++i) {
     totalEnergy += tC[i]->p();
   }
   if( totalEnergy > primTrack->p()+0.1*primTrack->p() ) {
           tC.pop_back();
           checkEnergy(primTrack,tC);
   }
}

void NuclearVertexBuilder::cleanTrackCollection ( const reco::TrackRef &  primTrack, std::vector< reco::TrackRef > &  tC  ) const

private

Definition at line 199 of file NuclearVertexBuilder.cc.

References objects.autophobj::float, mps_fire::i, isGoodSecondaryTrack(), TrackingRecHit::isValid(), LogDebug, shareFrac_, TrackingRecHit::sharesInput(), TrackingRecHit::some, and HiIsolationCommonParameters_cff::track.

Referenced by build(), and getVertex().

                                                                                    {

    // inspired from FinalTrackSelector (S. Wagner) modified by P. Janot
    LogDebug("NuclearInteractionMaker") << "cleanTrackCollection number of input tracks : " << tC.size();
   std::map<reco::TrackRef const*, std::vector<const TrackingRecHit*>> rh;

    // first remove bad quality tracks and create map
    std::vector<bool> selected(tC.size(), false);
    int i=0;
    for(auto const& track : tC) {
       if( isGoodSecondaryTrack(primTrack, track)) { 
            selected[i]=true;
            for(auto const& hit : track->recHits()) rh[&track].push_back(hit);
        }
       i++;
    }

    // then remove duplicated tracks
    i=-1;
    for(auto const& track : tC) {
      i++;
      int j=-1;
      for(auto const& track2 : tC) {
        j++;
        if ((!selected[j])||(!selected[i]))continue;
        if ((j<=i))continue;
        int noverlap=0;
        std::vector<const TrackingRecHit*>& iHits = rh[&track];
        for ( unsigned ih=0; ih<iHits.size(); ++ih ) {
          const TrackingRecHit* it = iHits[ih];
          if (it->isValid()){
            std::vector<const TrackingRecHit*>& jHits = rh[&track2];
            for ( unsigned ih2=0; ih2<jHits.size(); ++ih2 ) {
            const TrackingRecHit* jt = jHits[ih2];
              if (jt->isValid()){
                const TrackingRecHit* kt = jt;
                if ( it->sharesInput(kt,TrackingRecHit::some) )noverlap++;
               }
             }
          }
        }
        float fi=float(noverlap)/float(track->recHitsSize()); 
        float fj=float(noverlap)/float(track2->recHitsSize());
        if ((fi>shareFrac_)||(fj>shareFrac_)){
          if (fi<fj){
            selected[j]=false;
          }else{
            if (fi>fj){
              selected[i]=false;
            }else{
              if (track->normalizedChi2() > track2->normalizedChi2()){selected[i]=false;}else{selected[j]=false;}
            }//end fi > or = fj
          }//end fi < fj
        }//end got a duplicate
      }//end track2 loop
    }//end track loop

   std::vector< reco::TrackRef > newTrackColl;
   i=0;
   for(auto const& track : tC) {
         if( selected[i] ) newTrackColl.push_back(track);
         ++i;
   }
   tC = newTrackColl;
}

ClosestApproachInRPhi * NuclearVertexBuilder::closestApproach	(	const reco::TrackRef &	primTrack,
		const reco::TrackRef &	secTrack
	)		const

Definition at line 119 of file NuclearVertexBuilder.cc.

References ClosestApproachInRPhi::calculate(), getTrajectory(), and mps_update::status.

Referenced by FillVertexWithCrossingPoint(), NuclearInteractionEDProducer::findAdditionalSecondaryTracks(), getVertex(), isCompatible(), and isGoodSecondaryTrack().

                                                                                                                                {
            FreeTrajectoryState primTraj = getTrajectory(primTrack);
            ClosestApproachInRPhi *theApproach = new ClosestApproachInRPhi();
            FreeTrajectoryState secTraj = getTrajectory(secTrack);
            bool status = theApproach->calculate(primTraj,secTraj);
            if( status ) { return theApproach; }
            else { 
                   return nullptr;
            }
}

bool NuclearVertexBuilder::FillVertexWithAdaptVtxFitter ( const reco::TrackRef &  primTrack, const std::vector< reco::TrackRef > &  secTracks  )

private

Definition at line 57 of file NuclearVertexBuilder.cc.

References TransientTrackBuilder::build(), cppFunctionSkipper::exception, mps_fire::i, reco::Vertex::isValid(), LogDebug, the_vertex, theTransientTrackBuilder, reco::Vertex::tracksSize(), AdaptiveVertexFitter::vertex(), VertexException::what(), and cms::Exception::what().

Referenced by build(), and getVertex().

                                                                                                                               {
         std::vector<reco::TransientTrack> transientTracks;
         transientTracks.push_back( theTransientTrackBuilder->build(primTrack));
         // get the secondary track with the max number of hits
         for( unsigned short i=0; i != secTracks.size(); i++ ) {
              transientTracks.push_back( theTransientTrackBuilder->build( secTracks[i]) );
         }
         if( transientTracks.size() == 1 ) return  false;
         AdaptiveVertexFitter AVF;
         try {
            TransientVertex tv = AVF.vertex(transientTracks);
            the_vertex = reco::Vertex(tv);
         }
         catch(VertexException& exception){
            // AdaptivevertexFitter does not work
            LogDebug("NuclearInteractionMaker") << exception.what() << "\n";
            return false;
         }
         catch( cms::Exception& exception){
            // AdaptivevertexFitter does not work
            LogDebug("NuclearInteractionMaker") << exception.what() << "\n";
            return false;
         }
         if( the_vertex.isValid() ) {
              LogDebug("NuclearInteractionMaker") << "Try to build from AdaptiveVertexFitter with " << the_vertex.tracksSize() << " tracks";
              return true;
           }
         else return false;
}

bool NuclearVertexBuilder::FillVertexWithCrossingPoint ( const reco::TrackRef &  primTrack, const std::vector< reco::TrackRef > &  secTracks  )

private

Definition at line 88 of file NuclearVertexBuilder.cc.

References reco::Vertex::add(), closestApproach(), ClosestApproachInRPhi::crossingPoint(), mps_fire::i, NuclearSeed_cfi::maxHits, nhits, the_vertex, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by build(), and getVertex().

                                                                                                                              {
            // get the secondary track with the max number of hits
            unsigned short maxHits = 0; int indice=-1;
            for( unsigned short i=0; i < secTracks.size(); i++) {
                   // Add references to daughters
                   unsigned short nhits = secTracks[i]->numberOfValidHits();
                   if( nhits > maxHits ) { maxHits = nhits; indice=i; }
            }

            // Closest points
            if(indice == -1) return false;

            ClosestApproachInRPhi* theApproach = closestApproach( primTrack, secTracks[indice]);
            GlobalPoint crossing = theApproach->crossingPoint();
            delete theApproach;

            // Create vertex (creation point)
            // TODO: add error
            the_vertex = reco::Vertex(reco::Vertex::Point(crossing.x(),
                                                          crossing.y(),
                                                          crossing.z()),
                                      reco::Vertex::Error(), 0.,0.,0);

            the_vertex.add(reco::TrackBaseRef( primTrack ), 1.0);
            for( unsigned short i=0; i < secTracks.size(); i++) {
                 if( i==indice ) the_vertex.add(reco::TrackBaseRef( secTracks[i] ), 1.0);
                 else the_vertex.add(reco::TrackBaseRef( secTracks[i] ), 0.0);
            }
            return true;
}

void NuclearVertexBuilder::FillVertexWithLastPrimHit ( const reco::TrackRef &  primTrack, const std::vector< reco::TrackRef > &  secTracks  )

private

Definition at line 45 of file NuclearVertexBuilder.cc.

References reco::Vertex::add(), mps_fire::i, LogDebug, and the_vertex.

Referenced by build(), and getVertex().

                                                                                                                            {
   LogDebug("NuclearInteractionMaker") << "Vertex build from the end point of the primary track";
       the_vertex = reco::Vertex(reco::Vertex::Point(primTrack->outerPosition().x(),
                                         primTrack->outerPosition().y(),
                                         primTrack->outerPosition().z()),
                                         reco::Vertex::Error(), 0.,0.,0);
       the_vertex.add(reco::TrackBaseRef( primTrack ), 1.0);
       for( unsigned short i=0; i != secTracks.size(); i++) {
             the_vertex.add(reco::TrackBaseRef( secTracks[i] ), 0.0);
       }
}

FreeTrajectoryState NuclearVertexBuilder::getTrajectory ( const reco::TrackRef &  track ) const

private

Definition at line 27 of file NuclearVertexBuilder.cc.

References position, and theMagField.

Referenced by closestApproach(), and getVertex().

{ 
  GlobalPoint position(track->vertex().x(),
                       track->vertex().y(),
                       track->vertex().z());

  GlobalVector momentum(track->momentum().x(),
                        track->momentum().y(),
                        track->momentum().z());

  GlobalTrajectoryParameters gtp(position,momentum,
                                 track->charge(),theMagField);
  
  FreeTrajectoryState fts(gtp);

  return fts; 
} 

reco::Vertex NuclearVertexBuilder::getVertex ( ) const

inline

Definition at line 29 of file NuclearVertexBuilder.h.

References addSecondaryTrack(), checkEnergy(), cleanTrackCollection(), closestApproach(), FillVertexWithAdaptVtxFitter(), FillVertexWithCrossingPoint(), FillVertexWithLastPrimHit(), getTrajectory(), isCompatible(), isGoodSecondaryTrack(), the_vertex, and HiIsolationCommonParameters_cff::track.

{ return the_vertex; } 

bool NuclearVertexBuilder::isCompatible

(

const reco::TrackRef &

secTrack

)

const

Definition at line 164 of file NuclearVertexBuilder.cc.

References closestApproach(), isGoodSecondaryTrack(), minDistFromVtx_, reco::Vertex::position(), mps_fire::result, mathSSE::sqrt(), the_vertex, reco::Vertex::tracks_begin(), badGlobalMuonTaggersAOD_cff::vtx, PV3DBase< T, PVType, FrameType >::x(), reco::Vertex::xError(), PV3DBase< T, PVType, FrameType >::y(), reco::Vertex::yError(), PV3DBase< T, PVType, FrameType >::z(), and reco::Vertex::zError().

Referenced by getVertex().

                                                                           {
         
         reco::TrackRef primTrack = (*(the_vertex.tracks_begin())).castTo<reco::TrackRef>();
         ClosestApproachInRPhi *theApproach = closestApproach(primTrack, secTrack);
         bool result = false;
         if( theApproach ) {
           GlobalPoint crp = theApproach->crossingPoint();
           math::XYZPoint vtx = the_vertex.position();
           float dist = sqrt((crp.x()-vtx.x())*(crp.x()-vtx.x()) +
                        (crp.y()-vtx.y())*(crp.y()-vtx.y()) +
                        (crp.z()-vtx.z())*(crp.z()-vtx.z()));

           float distError = sqrt(the_vertex.xError()*the_vertex.xError() +
                                  the_vertex.yError()*the_vertex.yError() +
                                  the_vertex.zError()*the_vertex.zError());
 
           //std::cout << "Distance between Additional track and vertex =" << dist << " +/- " << distError << std::endl;
           // TODO : add condition on distance between last rechit of the primary and the first rec hit of the secondary
 
           result = (isGoodSecondaryTrack(secTrack, primTrack, theApproach->distance(), crp) 
                     && dist-distError<minDistFromVtx_);
         }
         delete theApproach;
         return result;
}

bool NuclearVertexBuilder::isGoodSecondaryTrack ( const reco::TrackRef &  primTrack, const reco::TrackRef &  secTrack  ) const

private

Definition at line 130 of file NuclearVertexBuilder.cc.

References closestApproach(), ClosestApproachInRPhi::crossingPoint(), ClosestApproachInRPhi::distance(), and mps_fire::result.

Referenced by cleanTrackCollection(), getVertex(), and isCompatible().

                                                                                                                   {
           ClosestApproachInRPhi* theApproach = closestApproach(primTrack, secTrack);
           bool result = false;
           if(theApproach)
             result = isGoodSecondaryTrack(secTrack, primTrack, theApproach->distance() , theApproach->crossingPoint());
           delete theApproach;
           return result;
} 

bool NuclearVertexBuilder::isGoodSecondaryTrack ( const reco::TrackRef &  secTrack, const reco::TrackRef &  primTrack, const double &  distOfClosestApp, const GlobalPoint &  crossPoint  ) const

private

Definition at line 139 of file NuclearVertexBuilder.cc.

References chi2Cut_, gather_cfg::cout, DPtovPtCut_, minDistFromPrim_, p1, p2, PV3DBase< T, PVType, FrameType >::perp(), and hiDetachedQuadStep_cff::pt2.

                                                                                       {
          float TRACKER_RADIUS=129;
          double pt2 = secTrack->pt();
          double Dpt2 = secTrack->ptError();
          double p1 = primTrack->p();
          double Dp1 = primTrack->qoverpError()*p1*p1;
          double p2 = secTrack->p();
          double Dp2 = secTrack->qoverpError()*p2*p2;
          std::cout << "1)" << distOfClosestApp << " < " << minDistFromPrim_ << " " << (distOfClosestApp < minDistFromPrim_)  << "\n";
          std::cout << "2)" << secTrack->normalizedChi2() << " < " << chi2Cut_ << " " << (secTrack->normalizedChi2() < chi2Cut_) << "\n";
          std::cout << "3)" << crossPoint.perp() << " < " << TRACKER_RADIUS << " " << (crossPoint.perp() < TRACKER_RADIUS) << "\n";
          std::cout << "4)" << (Dpt2/pt2) << " < " << DPtovPtCut_ << " " << ((Dpt2/pt2) < DPtovPtCut_)  << "\n";
          std::cout << "5)" << (p2-2*Dp2) << " < " << (p1+2*Dp1) << " " << ((p2-2*Dp2) < (p1+2*Dp1))<< "\n";
          if( distOfClosestApp < minDistFromPrim_ &&
              secTrack->normalizedChi2() < chi2Cut_ && 
              crossPoint.perp() < TRACKER_RADIUS && 
              (Dpt2/pt2) < DPtovPtCut_ &&
              (p2-2*Dp2) < (p1+2*Dp1)) return true;
          else return false;
}

Member Data Documentation

double NuclearVertexBuilder::chi2Cut_

private

Definition at line 56 of file NuclearVertexBuilder.h.

Referenced by isGoodSecondaryTrack().

double NuclearVertexBuilder::DPtovPtCut_

private

Definition at line 57 of file NuclearVertexBuilder.h.

Referenced by isGoodSecondaryTrack().

double NuclearVertexBuilder::minDistFromPrim_

private

Definition at line 55 of file NuclearVertexBuilder.h.

Referenced by isGoodSecondaryTrack().

double NuclearVertexBuilder::minDistFromVtx_

private

Definition at line 58 of file NuclearVertexBuilder.h.

Referenced by isCompatible().

double NuclearVertexBuilder::shareFrac_

private

Definition at line 59 of file NuclearVertexBuilder.h.

Referenced by cleanTrackCollection().

reco::Vertex NuclearVertexBuilder::the_vertex

private

Definition at line 50 of file NuclearVertexBuilder.h.

Referenced by addSecondaryTrack(), FillVertexWithAdaptVtxFitter(), FillVertexWithCrossingPoint(), FillVertexWithLastPrimHit(), getVertex(), and isCompatible().

const MagneticField* NuclearVertexBuilder::theMagField

private

Definition at line 53 of file NuclearVertexBuilder.h.

Referenced by getTrajectory().

const TransientTrackBuilder* NuclearVertexBuilder::theTransientTrackBuilder

private

Definition at line 54 of file NuclearVertexBuilder.h.

Referenced by FillVertexWithAdaptVtxFitter().