|
|
|
#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 &secTrack, const reco::TrackRef &primTrack, const double &distOfClosestApp, const GlobalPoint &crossPoint) const |
| bool | isGoodSecondaryTrack (const reco::TrackRef &primTrack, const reco::TrackRef &secTrack) const |
Private Attributes | |
| double | chi2Cut_ |
| double | DPtovPtCut_ |
| double | minDistFromPrim_ |
| double | minDistFromVtx_ |
| double | shareFrac_ |
| reco::Vertex | the_vertex |
| const MagneticField * | theMagField |
| const TransientTrackBuilder * | theTransientTrackBuilder |
Definition at line 16 of file NuclearVertexBuilder.h.
| NuclearVertexBuilder::NuclearVertexBuilder | ( | const MagneticField * | mag, |
| const TransientTrackBuilder * | transientTkBuilder, | ||
| const edm::ParameterSet & | iConfig | ||
| ) | [inline] |
Definition at line 19 of file NuclearVertexBuilder.h.
:
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")){}
| void NuclearVertexBuilder::addSecondaryTrack | ( | const reco::TrackRef & | secTrack | ) |
Definition at line 192 of file NuclearVertexBuilder.cc.
References build(), the_vertex, reco::Vertex::tracks_begin(), and reco::Vertex::tracks_end().
{
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(), FillVertexWithLastPrimHit(), and python::multivaluedict::sort().
Referenced by addSecondaryTrack().
{
cleanTrackCollection(primTrack, secTracks);
std::sort(secTracks.begin(),secTracks.end(),cmpTracks());
checkEnergy(primTrack, secTracks);
if( secTracks.size() != 0) {
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 273 of file NuclearVertexBuilder.cc.
References i.
Referenced by build().
{
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 201 of file NuclearVertexBuilder.cc.
References i, isGoodSecondaryTrack(), TrackingRecHit::isValid(), j, LogDebug, shareFrac_, TrackingRecHit::sharesInput(), TrackingRecHit::some, and ExpressReco_HICollisions_FallBack::track.
Referenced by build().
{
// inspired from FinalTrackSelector (S. Wagner) modified by P. Janot
LogDebug("NuclearInteractionMaker") << "cleanTrackCollection number of input tracks : " << tC.size();
std::map<std::vector<reco::TrackRef>::const_iterator, std::vector<const TrackingRecHit*> > rh;
// first remove bad quality tracks and create map
std::vector<bool> selected(tC.size(), false);
int i=0;
for (std::vector<reco::TrackRef>::const_iterator track=tC.begin(); track!=tC.end(); track++){
if( isGoodSecondaryTrack(primTrack, *track)) {
selected[i]=true;
trackingRecHit_iterator itB = (*track)->recHitsBegin();
trackingRecHit_iterator itE = (*track)->recHitsEnd();
for (trackingRecHit_iterator it = itB; it != itE; ++it) {
const TrackingRecHit* hit = &(**it);
rh[track].push_back(hit);
}
}
i++;
}
// then remove duplicated tracks
i=-1;
for (std::vector<reco::TrackRef>::const_iterator track=tC.begin(); track!=tC.end(); track++){
i++;
int j=-1;
for (std::vector<reco::TrackRef>::const_iterator track2=tC.begin(); track2!=tC.end(); track2++){
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 (std::vector<reco::TrackRef>::const_iterator track=tC.begin(); track!=tC.end(); track++){
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 121 of file NuclearVertexBuilder.cc.
References ClosestApproachInRPhi::calculate(), getTrajectory(), NULL, and ntuplemaker::status.
Referenced by FillVertexWithCrossingPoint(), 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 NULL;
}
}
| 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(), exception, i, reco::Vertex::isValid(), LogDebug, the_vertex, theTransientTrackBuilder, reco::Vertex::tracksSize(), AdaptiveVertexFitter::vertex(), cms::Exception::what(), and VertexException::what().
Referenced by build().
{
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(), getTrajectory(), i, the_vertex, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().
Referenced by build().
{
FreeTrajectoryState primTraj = getTrajectory(primTrack);
// 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(), i, LogDebug, and the_vertex.
Referenced by build().
{
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 FillVertexWithCrossingPoint().
{
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] |
| bool NuclearVertexBuilder::isCompatible | ( | const reco::TrackRef & | secTrack | ) | const |
Definition at line 166 of file NuclearVertexBuilder.cc.
References closestApproach(), isGoodSecondaryTrack(), minDistFromVtx_, reco::Vertex::position(), query::result, mathSSE::sqrt(), the_vertex, reco::Vertex::tracks_begin(), 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().
{
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 & | secTrack, |
| const reco::TrackRef & | primTrack, | ||
| const double & | distOfClosestApp, | ||
| const GlobalPoint & | crossPoint | ||
| ) | const [private] |
Definition at line 141 of file NuclearVertexBuilder.cc.
References chi2Cut_, gather_cfg::cout, DPtovPtCut_, minDistFromPrim_, p1, p2, and PV3DBase< T, PVType, FrameType >::perp().
{
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;
}
| bool NuclearVertexBuilder::isGoodSecondaryTrack | ( | const reco::TrackRef & | primTrack, |
| const reco::TrackRef & | secTrack | ||
| ) | const [private] |
Definition at line 132 of file NuclearVertexBuilder.cc.
References closestApproach(), ClosestApproachInRPhi::crossingPoint(), ClosestApproachInRPhi::distance(), and query::result.
Referenced by cleanTrackCollection(), and isCompatible().
{
ClosestApproachInRPhi* theApproach = closestApproach(primTrack, secTrack);
bool result = false;
if(theApproach)
result = isGoodSecondaryTrack(secTrack, primTrack, theApproach->distance() , theApproach->crossingPoint());
delete theApproach;
return result;
}
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().
1.7.3