|
|
|
#include <ConvBremPFTrackFinder.h>
Definition at line 29 of file ConvBremPFTrackFinder.h.
| ConvBremPFTrackFinder::ConvBremPFTrackFinder | ( | const TransientTrackBuilder & | builder, |
| double | mvaBremConvCut, | ||
| std::string | mvaWeightFileConvBrem | ||
| ) |
Definition at line 22 of file ConvBremPFTrackFinder.cc.
References detaBremKF, Epout, nHITS1, pfcalib_, ptRatioGsfKF, secPin, secR, sTIP, and tmvaReader_.
: builder_(builder), mvaBremConvCut_(mvaBremConvCut), mvaWeightFileConvBrem_(mvaWeightFileConvBrem) { tmvaReader_ = new TMVA::Reader("!Color:Silent"); tmvaReader_->AddVariable("secR",&secR); tmvaReader_->AddVariable("sTIP",&sTIP); tmvaReader_->AddVariable("nHITS1",&nHITS1); tmvaReader_->AddVariable("secPin",&secPin); tmvaReader_->AddVariable("Epout",&Epout); tmvaReader_->AddVariable("detaBremKF",&detaBremKF); tmvaReader_->AddVariable("ptRatioGsfKF",&ptRatioGsfKF); tmvaReader_->BookMVA("BDT",mvaWeightFileConvBrem.c_str()); pfcalib_ = new PFEnergyCalibration(); }
| ConvBremPFTrackFinder::~ConvBremPFTrackFinder | ( | ) |
Definition at line 42 of file ConvBremPFTrackFinder.cc.
References pfcalib_, and tmvaReader_.
{delete tmvaReader_; delete pfcalib_; }
| bool ConvBremPFTrackFinder::foundConvBremPFRecTrack | ( | const edm::Handle< reco::PFRecTrackCollection > & | thePfRecTrackCol, |
| const edm::Handle< reco::VertexCollection > & | primaryVertex, | ||
| const edm::Handle< reco::PFDisplacedTrackerVertexCollection > & | pfNuclears, | ||
| const edm::Handle< reco::PFConversionCollection > & | pfConversions, | ||
| const edm::Handle< reco::PFV0Collection > & | pfV0, | ||
| bool | useNuclear, | ||
| bool | useConversions, | ||
| bool | useV0, | ||
| const reco::PFClusterCollection & | theEClus, | ||
| reco::GsfPFRecTrack | gsfpfrectk | ||
| ) | [inline] |
Definition at line 37 of file ConvBremPFTrackFinder.h.
References found_, and runConvBremFinder().
Referenced by PFElecTkProducer::produce().
{
found_ = false;
runConvBremFinder(thePfRecTrackCol,primaryVertex,
pfNuclears,pfConversions,
pfV0,useNuclear,
useConversions,useV0,
theEClus,gsfpfrectk);
return found_;};
| const std::vector<reco::PFRecTrackRef>& ConvBremPFTrackFinder::getConvBremPFRecTracks | ( | ) | [inline] |
Definition at line 57 of file ConvBremPFTrackFinder.h.
References pfRecTrRef_vec_.
Referenced by PFElecTkProducer::produce().
{return pfRecTrRef_vec_;};
| void ConvBremPFTrackFinder::runConvBremFinder | ( | const edm::Handle< reco::PFRecTrackCollection > & | thePfRecTrackCol, |
| const edm::Handle< reco::VertexCollection > & | primaryVertex, | ||
| const edm::Handle< reco::PFDisplacedTrackerVertexCollection > & | pfNuclears, | ||
| const edm::Handle< reco::PFConversionCollection > & | pfConversions, | ||
| const edm::Handle< reco::PFV0Collection > & | pfV0, | ||
| bool | useNuclear, | ||
| bool | useConversions, | ||
| bool | useV0, | ||
| const reco::PFClusterCollection & | theEClus, | ||
| reco::GsfPFRecTrack | gsfpfrectk | ||
| ) | [private] |
Definition at line 45 of file ConvBremPFTrackFinder.cc.
References TrackingRecHit::all, TransientTrackBuilder::build(), builder_, reco::PFTrack::calculatePositionREP(), reco::PFCluster::calculatePositionREP(), gather_cfg::cout, debug, detaBremKF, alignCSCRings::e, reco::PFTrajectoryPoint::ECALEntrance, reco::PFTrajectoryPoint::ECALShowerMax, PFEnergyCalibration::energyEm(), Epout, eta, reco::PFTrack::extrapolatedPoint(), found_, reco::GsfPFRecTrack::gsfTrackRef(), i, edm::Ref< C, T, F >::isNonnull(), reco::PFTrajectoryPoint::isValid(), reco::GsfPFRecTrack::kfPFRecTrackRef(), reco::PFTrajectoryPoint::momentum(), mvaBremConvCut_, nHITS1, notFound, AlCaHLTBitMon_ParallelJobs::p, pfcalib_, reco::GsfPFRecTrack::PFRecBrem(), pfRecTrRef_vec_, phi, Pi, edm::Handle< T >::product(), ptRatioGsfKF, secPin, secPout, secR, IPTools::signedTransverseImpactParameter(), findQualityFiles::size, mathSSE::sqrt(), sTIP, LinkByRecHit::testTrackAndClusterByRecHit(), tmvaReader_, and TwoPi.
Referenced by foundConvBremPFRecTrack().
{
found_ = false;
bool debug = false;
bool debugRef = false;
if(debug)
cout << "runConvBremFinder:: Entering " << endl;
reco::GsfTrackRef refGsf = gsfpfrectk.gsfTrackRef();
reco::PFRecTrackRef pfTrackRef = gsfpfrectk.kfPFRecTrackRef();
vector<PFBrem> primPFBrem = gsfpfrectk.PFRecBrem();
const PFRecTrackCollection& PfRTkColl = *(thePfRecTrackCol.product());
reco::PFRecTrackCollection::const_iterator pft=PfRTkColl.begin();
reco::PFRecTrackCollection::const_iterator pftend=PfRTkColl.end();
//PFEnergyCalibration pfcalib_;
vector<PFRecTrackRef> AllPFRecTracks;
AllPFRecTracks.clear();
unsigned int ipft = 0;
for(;pft!=pftend;++pft,ipft++){
// do not consider the kf track already associated to the seed
if(pfTrackRef.isNonnull())
if(pfTrackRef->trackRef() == pft->trackRef()) continue;
PFRecTrackRef pfRecTrRef(thePfRecTrackCol,ipft);
TrackRef trackRef = pfRecTrRef->trackRef();
reco::TrackBaseRef selTrackBaseRef(trackRef);
if(debug)
cout << "runConvBremFinder:: pushing_back High Purity " << pft->trackRef()->pt()
<< " eta,phi " << pft->trackRef()->eta() << ", " << pft->trackRef()->phi()
<< " Memory Address Ref " << &*trackRef << " Memory Address BaseRef " << &*selTrackBaseRef << endl;
AllPFRecTracks.push_back(pfRecTrRef);
}
if(useConversions) {
const PFConversionCollection& PfConvColl = *(pfConversions.product());
for(unsigned i=0;i<PfConvColl.size(); i++) {
reco::PFConversionRef convRef(pfConversions,i);
unsigned int trackSize=(convRef->pfTracks()).size();
if ( convRef->pfTracks().size() < 2) continue;
for(unsigned iTk=0;iTk<trackSize; iTk++) {
PFRecTrackRef compPFTkRef = convRef->pfTracks()[iTk];
reco::TrackBaseRef newTrackBaseRef(compPFTkRef->trackRef());
// do not consider the kf track already associated to the seed
if(pfTrackRef.isNonnull()) {
reco::TrackBaseRef primaryTrackBaseRef(pfTrackRef->trackRef());
if(primaryTrackBaseRef == newTrackBaseRef) continue;
}
bool notFound = true;
for(unsigned iPF = 0; iPF < AllPFRecTracks.size(); iPF++) {
reco::TrackBaseRef selTrackBaseRef(AllPFRecTracks[iPF]->trackRef());
if(debugRef)
cout << "## Track 1 HP pt " << AllPFRecTracks[iPF]->trackRef()->pt() << " eta, phi " << AllPFRecTracks[iPF]->trackRef()->eta() << ", " << AllPFRecTracks[iPF]->trackRef()->phi()
<< " Memory Address Ref " << &(*AllPFRecTracks[iPF]->trackRef()) << " Memory Address BaseRef " << &*selTrackBaseRef << endl;
if(debugRef)
cout << "** Track 2 CONV pt " << compPFTkRef->trackRef()->pt() << " eta, phi " << compPFTkRef->trackRef()->eta() << ", " << compPFTkRef->trackRef()->phi()
<< " Memory Address Ref " << &*compPFTkRef->trackRef() << " Memory Address BaseRef " << &*newTrackBaseRef << endl;
//if(selTrackBaseRef == newTrackBaseRef || AllPFRecTracks[iPF]->trackRef()== compPFTkRef->trackRef()) {
if(AllPFRecTracks[iPF]->trackRef()== compPFTkRef->trackRef()) {
if(debugRef)
cout << " SAME BREM REF " << endl;
notFound = false;
}
}
if(notFound) {
if(debug)
cout << "runConvBremFinder:: pushing_back Conversions " << compPFTkRef->trackRef()->pt()
<< " eta,phi " << compPFTkRef->trackRef()->eta() << " phi " << compPFTkRef->trackRef()->phi() <<endl;
AllPFRecTracks.push_back(compPFTkRef);
}
}
}
}
if(useNuclear) {
const PFDisplacedTrackerVertexCollection& PfNuclColl = *(pfNuclears.product());
for(unsigned i=0;i<PfNuclColl.size(); i++) {
const reco::PFDisplacedTrackerVertexRef dispacedVertexRef(pfNuclears, i );
unsigned int trackSize= dispacedVertexRef->pfRecTracks().size();
for(unsigned iTk=0;iTk < trackSize; iTk++) {
reco::PFRecTrackRef newPFRecTrackRef = dispacedVertexRef->pfRecTracks()[iTk];
reco::TrackBaseRef newTrackBaseRef(newPFRecTrackRef->trackRef());
// do not consider the kf track already associated to the seed
if(pfTrackRef.isNonnull()) {
reco::TrackBaseRef primaryTrackBaseRef(pfTrackRef->trackRef());
if(primaryTrackBaseRef == newTrackBaseRef) continue;
}
bool notFound = true;
for(unsigned iPF = 0; iPF < AllPFRecTracks.size(); iPF++) {
reco::TrackBaseRef selTrackBaseRef(AllPFRecTracks[iPF]->trackRef());
if(selTrackBaseRef == newTrackBaseRef) notFound = false;
}
if(notFound) {
if(debug)
cout << "runConvBremFinder:: pushing_back displaced Vertex pt " << newPFRecTrackRef->trackRef()->pt()
<< " eta,phi " << newPFRecTrackRef->trackRef()->eta() << ", " << newPFRecTrackRef->trackRef()->phi() << endl;
AllPFRecTracks.push_back(newPFRecTrackRef);
}
}
}
}
if(useV0) {
const PFV0Collection& PfV0Coll = *(pfV0.product());
for(unsigned i=0;i<PfV0Coll.size(); i++) {
reco::PFV0Ref v0Ref( pfV0, i );
unsigned int trackSize=(v0Ref->pfTracks()).size();
for(unsigned iTk=0;iTk<trackSize; iTk++) {
reco::PFRecTrackRef newPFRecTrackRef = (v0Ref->pfTracks())[iTk];
reco::TrackBaseRef newTrackBaseRef(newPFRecTrackRef->trackRef());
// do not consider the kf track already associated to the seed
if(pfTrackRef.isNonnull()) {
reco::TrackBaseRef primaryTrackBaseRef(pfTrackRef->trackRef());
if(primaryTrackBaseRef == newTrackBaseRef) continue;
}
bool notFound = true;
for(unsigned iPF = 0; iPF < AllPFRecTracks.size(); iPF++) {
reco::TrackBaseRef selTrackBaseRef(AllPFRecTracks[iPF]->trackRef());
if(selTrackBaseRef == newTrackBaseRef) notFound = false;
}
if(notFound) {
if(debug)
cout << "runConvBremFinder:: pushing_back V0 " << newPFRecTrackRef->trackRef()->pt()
<< " eta,phi " << newPFRecTrackRef->trackRef()->eta() << ", " << newPFRecTrackRef->trackRef()->phi() << endl;
AllPFRecTracks.push_back(newPFRecTrackRef);
}
}
}
}
pfRecTrRef_vec_.clear();
for(unsigned iPF = 0; iPF < AllPFRecTracks.size(); iPF++) {
double dphi= fabs(AllPFRecTracks[iPF]->trackRef()->phi()-refGsf->phi());
if (dphi>TMath::Pi()) dphi-= TMath::TwoPi();
double deta=fabs(AllPFRecTracks[iPF]->trackRef()->eta()-refGsf->eta());
// limiting the phase space (just for saving cpu-time)
if( fabs(dphi)> 1.0 || fabs(deta) > 0.4) continue;
double minDEtaBremKF = 1000.;
double minDPhiBremKF = 1000.;
double minDRBremKF = 1000.;
double minDEtaBremKFPos = 1000.;
double minDPhiBremKFPos = 1000.;
double minDRBremKFPos = 1000.;
reco:: TrackRef trkRef = AllPFRecTracks[iPF]->trackRef();
double secEta = trkRef->innerMomentum().eta();
double secPhi = trkRef->innerMomentum().phi();
for(unsigned ipbrem = 0; ipbrem < primPFBrem.size(); ipbrem++) {
if(primPFBrem[ipbrem].indTrajPoint() == 99) continue;
const reco::PFTrajectoryPoint& atPrimECAL
= primPFBrem[ipbrem].extrapolatedPoint( reco::PFTrajectoryPoint::ECALEntrance );
if( ! atPrimECAL.isValid() ) continue;
double bremEta = atPrimECAL.momentum().Eta();
double bremPhi = atPrimECAL.momentum().Phi();
double deta = fabs(bremEta - secEta);
double dphi = fabs(bremPhi - secPhi);
if (dphi>TMath::Pi()) dphi-= TMath::TwoPi();
double DR = sqrt(deta*deta + dphi*dphi);
double detaPos = fabs(bremEta - trkRef->innerPosition().eta());
double dphiPos = fabs(bremPhi - trkRef->innerPosition().phi());
if (dphiPos>TMath::Pi()) dphiPos-= TMath::TwoPi();
double DRPos = sqrt(detaPos*detaPos + dphiPos*dphiPos);
// find the closest track tangent
if(DR < minDRBremKF) {
minDRBremKF = DR;
minDEtaBremKF = deta;
minDPhiBremKF = fabs(dphi);
}
if(DRPos < minDRBremKFPos) {
minDRBremKFPos = DR;
minDEtaBremKFPos = detaPos;
minDPhiBremKFPos = fabs(dphiPos);
}
}
//gsfR
float gsfR = sqrt(refGsf->innerPosition().x()*refGsf->innerPosition().x() +
refGsf->innerPosition().y()*refGsf->innerPosition().y() );
// secR
secR = sqrt(trkRef->innerPosition().x()*trkRef->innerPosition().x() +
trkRef->innerPosition().y()*trkRef->innerPosition().y() );
// apply loose selection (to be parallel) between the secondary track and brem-tangents.
// Moreover if the secR is internal with respect to the GSF track by two pixel layers discard it.
if( (minDPhiBremKF < 0.1 || minDPhiBremKFPos < 0.1) &&
(minDEtaBremKF < 0.02 || minDEtaBremKFPos < 0.02)&&
secR > (gsfR-8)) {
if(debug)
cout << "runConvBremFinder:: OK Find track and BREM close "
<< " MinDphi " << minDPhiBremKF << " MinDeta " << minDEtaBremKF << endl;
float MinDist = 100000.;
float EE_calib = 0.;
PFRecTrack pfrectrack = *AllPFRecTracks[iPF];
pfrectrack.calculatePositionREP();
// Find and ECAL associated cluster
for (PFClusterCollection::const_iterator clus = theEClus.begin();
clus != theEClus.end();
clus++ ) {
// Removed unusd variable, left this in case it has side effects
clus->position();
double dist = -1.;
PFCluster clust = *clus;
clust.calculatePositionREP();
dist = pfrectrack.extrapolatedPoint( reco::PFTrajectoryPoint::ECALShowerMax ).isValid() ?
LinkByRecHit::testTrackAndClusterByRecHit(pfrectrack , clust ) : -1.;
if(dist > 0.) {
bool applyCrackCorrections = false;
vector<double> ps1Ene(0);
vector<double> ps2Ene(0);
double ps1,ps2;
ps1=ps2=0.;
if(dist < MinDist) {
MinDist = dist;
EE_calib = pfcalib_->energyEm(*clus,ps1Ene,ps2Ene,ps1,ps2,applyCrackCorrections);
}
}
}
if(MinDist > 0. && MinDist < 100000.) {
// compute all the input variables for conv brem selection
secPout = sqrt(trkRef->outerMomentum().x()*trkRef->outerMomentum().x() +
trkRef->outerMomentum().y()*trkRef->outerMomentum().y() +
trkRef->outerMomentum().z()*trkRef->outerMomentum().z());
secPin = sqrt(trkRef->innerMomentum().x()*trkRef->innerMomentum().x() +
trkRef->innerMomentum().y()*trkRef->innerMomentum().y() +
trkRef->innerMomentum().z()*trkRef->innerMomentum().z());
// maybe put innter momentum pt?
ptRatioGsfKF = trkRef->pt()/(refGsf->ptMode());
Vertex dummy;
const Vertex *pv = &dummy;
edm::Ref<VertexCollection> pvRef;
if (primaryVertex->size() != 0) {
pv = &*primaryVertex->begin();
// we always use the first vertex (at the moment)
pvRef = edm::Ref<VertexCollection>(primaryVertex, 0);
} else { // create a dummy PV
Vertex::Error e;
e(0, 0) = 0.0015 * 0.0015;
e(1, 1) = 0.0015 * 0.0015;
e(2, 2) = 15. * 15.;
Vertex::Point p(0, 0, 0);
dummy = Vertex(p, e, 0, 0, 0);
}
// direction of the Gsf track
GlobalVector direction(refGsf->innerMomentum().x(),
refGsf->innerMomentum().y(),
refGsf->innerMomentum().z());
TransientTrack transientTrack = builder_.build(*trkRef);
sTIP = IPTools::signedTransverseImpactParameter(transientTrack, direction, *pv).second.significance();
Epout = EE_calib/secPout;
// eta distance brem-secondary kf track
detaBremKF = minDEtaBremKF;
// Number of commont hits
trackingRecHit_iterator nhit=refGsf->recHitsBegin();
trackingRecHit_iterator nhit_end=refGsf->recHitsEnd();
unsigned int tmp_sh = 0;
//uint ish=0;
for (;nhit!=nhit_end;++nhit){
if ((*nhit)->isValid()){
trackingRecHit_iterator ihit=trkRef->recHitsBegin();
trackingRecHit_iterator ihit_end=trkRef->recHitsEnd();
for (;ihit!=ihit_end;++ihit){
if ((*ihit)->isValid()) {
// method 1
if((*nhit)->sharesInput(&*(*ihit),TrackingRecHit::all)) tmp_sh++;
// method 2 to switch in case of problem with rechit collections
// if(((*ihit)->geographicalId()==(*nhit)->geographicalId())&&
// (((*nhit)->localPosition()-(*ihit)->localPosition()).mag()<0.01)) ish++;
}
}
}
}
nHITS1 = tmp_sh;
double mvaValue = tmvaReader_->EvaluateMVA("BDT");
if(debug)
cout << " The imput variables for conv brem tracks identification " << endl
<< " secR " << secR << " gsfR " << gsfR << endl
<< " N shared hits " << nHITS1 << endl
<< " sTIP " << sTIP << endl
<< " detaBremKF " << detaBremKF << endl
<< " E/pout " << Epout << endl
<< " pin " << secPin << endl
<< " ptRatioKFGsf " << ptRatioGsfKF << endl
<< " ***** MVA ***** " << mvaValue << endl;
if(mvaValue > mvaBremConvCut_) {
found_ = true;
pfRecTrRef_vec_.push_back(AllPFRecTracks[iPF]);
}
} // end MinDIST
} // end selection kf - brem tangents
} // loop on the kf tracks
}
Definition at line 74 of file ConvBremPFTrackFinder.h.
Referenced by runConvBremFinder().
float ConvBremPFTrackFinder::detaBremKF [private] |
Definition at line 79 of file ConvBremPFTrackFinder.h.
Referenced by ConvBremPFTrackFinder(), and runConvBremFinder().
float ConvBremPFTrackFinder::Epout [private] |
Definition at line 79 of file ConvBremPFTrackFinder.h.
Referenced by ConvBremPFTrackFinder(), and runConvBremFinder().
bool ConvBremPFTrackFinder::found_ [private] |
Definition at line 73 of file ConvBremPFTrackFinder.h.
Referenced by foundConvBremPFRecTrack(), and runConvBremFinder().
double ConvBremPFTrackFinder::mvaBremConvCut_ [private] |
Definition at line 75 of file ConvBremPFTrackFinder.h.
Referenced by runConvBremFinder().
std::string ConvBremPFTrackFinder::mvaWeightFileConvBrem_ [private] |
Definition at line 76 of file ConvBremPFTrackFinder.h.
float ConvBremPFTrackFinder::nHITS1 [private] |
Definition at line 81 of file ConvBremPFTrackFinder.h.
Referenced by ConvBremPFTrackFinder(), and runConvBremFinder().
Definition at line 83 of file ConvBremPFTrackFinder.h.
Referenced by ConvBremPFTrackFinder(), runConvBremFinder(), and ~ConvBremPFTrackFinder().
std::vector<reco::PFRecTrackRef> ConvBremPFTrackFinder::pfRecTrRef_vec_ [private] |
Definition at line 78 of file ConvBremPFTrackFinder.h.
Referenced by getConvBremPFRecTracks(), and runConvBremFinder().
float ConvBremPFTrackFinder::ptRatioGsfKF [private] |
Definition at line 79 of file ConvBremPFTrackFinder.h.
Referenced by ConvBremPFTrackFinder(), and runConvBremFinder().
float ConvBremPFTrackFinder::secPin [private] |
Definition at line 79 of file ConvBremPFTrackFinder.h.
Referenced by ConvBremPFTrackFinder(), and runConvBremFinder().
float ConvBremPFTrackFinder::secPout [private] |
Definition at line 79 of file ConvBremPFTrackFinder.h.
Referenced by runConvBremFinder().
float ConvBremPFTrackFinder::secR [private] |
Definition at line 79 of file ConvBremPFTrackFinder.h.
Referenced by ConvBremPFTrackFinder(), and runConvBremFinder().
float ConvBremPFTrackFinder::sTIP [private] |
Definition at line 79 of file ConvBremPFTrackFinder.h.
Referenced by ConvBremPFTrackFinder(), and runConvBremFinder().
TMVA::Reader* ConvBremPFTrackFinder::tmvaReader_ [private] |
Definition at line 77 of file ConvBremPFTrackFinder.h.
Referenced by ConvBremPFTrackFinder(), runConvBremFinder(), and ~ConvBremPFTrackFinder().
1.7.3