Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

MTD reconstruction validation: tracks and vertices #35989

Merged
merged 14 commits into from Nov 11, 2021
Merged

MTD reconstruction validation: tracks and vertices #35989

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
ff0ae9d
Add gloabl resolution and pull plots
fabiocos Oct 18, 2021
bc15708
Update deltaT histograms
fabiocos Oct 19, 2021
0516900
Fix deltaT histograms
fabiocos Oct 19, 2021
85f684d
Update dt plots, fill input for efficiency calculation
fabiocos Oct 20, 2021
a795159
add matched track selection
fabiocos Oct 22, 2021
8b8e254
Fill matched tracks histograms
fabiocos Oct 22, 2021
f70b990
Add matching efficiency ratio
fabiocos Oct 25, 2021
cb4a04a
Fixes to the track selection
fabiocos Oct 25, 2021
28c4eb8
Add harvesting of 4D vertex analysis efficiencies
fabiocos Oct 28, 2021
cb109a3
Updates to selection for test
fabiocos Oct 28, 2021
3a9cc66
Cleaning of the code
fabiocos Oct 29, 2021
a03c199
Slim down size and range of space resolution histograms
fabiocos Oct 29, 2021
89c4e1f
Remove pull plots for mass hypotheses
fabiocos Oct 29, 2021
22b242c
Move vertex histograms of track validation into vertex validation class
fabiocos Nov 4, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
2 changes: 2 additions & 0 deletions Validation/MtdValidation/plugins/BuildFile.xml
View file
Open in desktop
Expand Up @@ -8,6 +8,8 @@
<use name="DQMServices/Core"/>
<use name="SimDataFormats/Track"/>
<use name="SimDataFormats/PileupSummaryInfo"/>
<use name="SimGeneral/HepPDTESSource"/>
<use name="SimGeneral/HepPDTRecord"/>
<use name="Geometry/Records"/>
<use name="SimDataFormats/TrackingAnalysis"/>
<use name="SimTracker/VertexAssociation"/>
Expand Down
93 changes: 92 additions & 1 deletion Validation/MtdValidation/plugins/MtdTracksHarvester.cc
View file
Open in desktop
Expand Up @@ -30,6 +30,10 @@ class MtdTracksHarvester : public DQMEDHarvester {
MonitorElement* meEtlEtaEff_[2];
MonitorElement* meEtlPhiEff_[2];
MonitorElement* meEtlPtEff_[2];
MonitorElement* meMVAPtSelEff_;
MonitorElement* meMVAEtaSelEff_;
MonitorElement* meMVAPtMatchEff_;
MonitorElement* meMVAEtaMatchEff_;
};

// ------------ constructor and destructor --------------
Expand Down Expand Up @@ -59,12 +63,20 @@ void MtdTracksHarvester::dqmEndJob(DQMStore::IBooker& ibook, DQMStore::IGetter&
MonitorElement* meETLTrackEffEtaMtdZpos = igetter.get(folder_ + "TrackETLEffEtaMtdZpos");
MonitorElement* meETLTrackEffPhiMtdZpos = igetter.get(folder_ + "TrackETLEffPhiMtdZpos");
MonitorElement* meETLTrackEffPtMtdZpos = igetter.get(folder_ + "TrackETLEffPtMtdZpos");
MonitorElement* meMVATrackEffPtTot = igetter.get(folder_ + "MVAEffPtTot");
MonitorElement* meMVATrackMatchedEffPtTot = igetter.get(folder_ + "MVAMatchedEffPtTot");
MonitorElement* meMVATrackMatchedEffPtMtd = igetter.get(folder_ + "MVAMatchedEffPtMtd");
MonitorElement* meMVATrackEffEtaTot = igetter.get(folder_ + "MVAEffEtaTot");
MonitorElement* meMVATrackMatchedEffEtaTot = igetter.get(folder_ + "MVAMatchedEffEtaTot");
MonitorElement* meMVATrackMatchedEffEtaMtd = igetter.get(folder_ + "MVAMatchedEffEtaMtd");

if (!meBTLTrackEffEtaTot || !meBTLTrackEffPhiTot || !meBTLTrackEffPtTot || !meBTLTrackEffEtaMtd ||
!meBTLTrackEffPhiMtd || !meBTLTrackEffPtMtd || !meETLTrackEffEtaTotZneg || !meETLTrackEffPhiTotZneg ||
!meETLTrackEffPtTotZneg || !meETLTrackEffEtaMtdZneg || !meETLTrackEffPhiMtdZneg || !meETLTrackEffPtMtdZneg ||
!meETLTrackEffEtaTotZpos || !meETLTrackEffPhiTotZpos || !meETLTrackEffPtTotZpos || !meETLTrackEffEtaMtdZpos ||
!meETLTrackEffPhiMtdZpos || !meETLTrackEffPtMtdZpos) {
!meETLTrackEffPhiMtdZpos || !meETLTrackEffPtMtdZpos || !meMVATrackEffPtTot || !meMVATrackMatchedEffPtTot ||
!meMVATrackMatchedEffPtMtd || !meMVATrackEffEtaTot || !meMVATrackMatchedEffEtaTot ||
!meMVATrackMatchedEffEtaMtd) {
edm::LogError("MtdTracksHarvester") << "Monitoring histograms not found!" << std::endl;
return;
}
Expand Down Expand Up @@ -116,6 +128,26 @@ void MtdTracksHarvester::dqmEndJob(DQMStore::IBooker& ibook, DQMStore::IGetter&
meETLTrackEffPtTotZpos->getNbinsX(),
meETLTrackEffPtTotZpos->getTH1()->GetXaxis()->GetXmin(),
meETLTrackEffPtTotZpos->getTH1()->GetXaxis()->GetXmax());
meMVAPtSelEff_ = ibook.book1D("MVAPtSelEff",
"Track selected efficiency VS Pt;Pt [GeV];Efficiency",
meMVATrackEffPtTot->getNbinsX(),
meMVATrackEffPtTot->getTH1()->GetXaxis()->GetXmin(),
meMVATrackEffPtTot->getTH1()->GetXaxis()->GetXmax());
meMVAEtaSelEff_ = ibook.book1D("MVAEtaSelEff",
"Track selected efficiency VS Eta;Eta;Efficiency",
meMVATrackEffEtaTot->getNbinsX(),
meMVATrackEffEtaTot->getTH1()->GetXaxis()->GetXmin(),
meMVATrackEffEtaTot->getTH1()->GetXaxis()->GetXmax());
meMVAPtMatchEff_ = ibook.book1D("MVAPtMatchEff",
"Track matched to GEN efficiency VS Pt;Pt [GeV];Efficiency",
meMVATrackMatchedEffPtTot->getNbinsX(),
meMVATrackMatchedEffPtTot->getTH1()->GetXaxis()->GetXmin(),
meMVATrackMatchedEffPtTot->getTH1()->GetXaxis()->GetXmax());
meMVAEtaMatchEff_ = ibook.book1D("MVAEtaMatchEff",
"Track matched to GEN efficiency VS Eta;Eta;Efficiency",
meMVATrackMatchedEffEtaTot->getNbinsX(),
meMVATrackMatchedEffEtaTot->getTH1()->GetXaxis()->GetXmin(),
meMVATrackMatchedEffEtaTot->getTH1()->GetXaxis()->GetXmax());

meBtlEtaEff_->getTH1()->SetMinimum(0.);
meBtlPhiEff_->getTH1()->SetMinimum(0.);
Expand All @@ -125,6 +157,10 @@ void MtdTracksHarvester::dqmEndJob(DQMStore::IBooker& ibook, DQMStore::IGetter&
meEtlPhiEff_[i]->getTH1()->SetMinimum(0.);
meEtlPtEff_[i]->getTH1()->SetMinimum(0.);
}
meMVAPtSelEff_->getTH1()->SetMinimum(0.);
meMVAEtaSelEff_->getTH1()->SetMinimum(0.);
meMVAPtMatchEff_->getTH1()->SetMinimum(0.);
meMVAEtaMatchEff_->getTH1()->SetMinimum(0.);

// --- Calculate efficiency BTL
for (int ibin = 1; ibin <= meBTLTrackEffEtaTot->getNbinsX(); ibin++) {
Expand Down Expand Up @@ -247,6 +283,61 @@ void MtdTracksHarvester::dqmEndJob(DQMStore::IBooker& ibook, DQMStore::IGetter&
meEtlPtEff_[1]->setBinContent(ibin, eff);
meEtlPtEff_[1]->setBinError(ibin, bin_err);
}

for (int ibin = 1; ibin <= meMVATrackEffPtTot->getNbinsX(); ibin++) {
double eff = meMVATrackMatchedEffPtTot->getBinContent(ibin) / meMVATrackEffPtTot->getBinContent(ibin);
double bin_err = sqrt((meMVATrackMatchedEffPtTot->getBinContent(ibin) *
(meMVATrackEffPtTot->getBinContent(ibin) - meMVATrackMatchedEffPtTot->getBinContent(ibin))) /
pow(meMVATrackEffPtTot->getBinContent(ibin), 3));
if (meMVATrackEffPtTot->getBinContent(ibin) == 0) {
eff = 0;
bin_err = 0;
}
meMVAPtSelEff_->setBinContent(ibin, eff);
meMVAPtSelEff_->setBinError(ibin, bin_err);
}

for (int ibin = 1; ibin <= meMVATrackEffEtaTot->getNbinsX(); ibin++) {
double eff = meMVATrackMatchedEffEtaTot->getBinContent(ibin) / meMVATrackEffEtaTot->getBinContent(ibin);
double bin_err =
sqrt((meMVATrackMatchedEffEtaTot->getBinContent(ibin) *
(meMVATrackEffEtaTot->getBinContent(ibin) - meMVATrackMatchedEffEtaTot->getBinContent(ibin))) /
pow(meMVATrackEffEtaTot->getBinContent(ibin), 3));
if (meMVATrackEffEtaTot->getBinContent(ibin) == 0) {
eff = 0;
bin_err = 0;
}
meMVAEtaSelEff_->setBinContent(ibin, eff);
meMVAEtaSelEff_->setBinError(ibin, bin_err);
}

for (int ibin = 1; ibin <= meMVATrackMatchedEffPtTot->getNbinsX(); ibin++) {
double eff = meMVATrackMatchedEffPtMtd->getBinContent(ibin) / meMVATrackMatchedEffPtTot->getBinContent(ibin);
double bin_err =
sqrt((meMVATrackMatchedEffPtMtd->getBinContent(ibin) *
(meMVATrackMatchedEffPtTot->getBinContent(ibin) - meMVATrackMatchedEffPtMtd->getBinContent(ibin))) /
pow(meMVATrackMatchedEffPtTot->getBinContent(ibin), 3));
if (meMVATrackMatchedEffPtTot->getBinContent(ibin) == 0) {
eff = 0;
bin_err = 0;
}
meMVAPtMatchEff_->setBinContent(ibin, eff);
meMVAPtMatchEff_->setBinError(ibin, bin_err);
}

for (int ibin = 1; ibin <= meMVATrackMatchedEffEtaTot->getNbinsX(); ibin++) {
double eff = meMVATrackMatchedEffEtaMtd->getBinContent(ibin) / meMVATrackMatchedEffEtaTot->getBinContent(ibin);
double bin_err =
sqrt((meMVATrackMatchedEffEtaMtd->getBinContent(ibin) *
(meMVATrackMatchedEffEtaTot->getBinContent(ibin) - meMVATrackMatchedEffEtaMtd->getBinContent(ibin))) /
pow(meMVATrackMatchedEffEtaTot->getBinContent(ibin), 3));
if (meMVATrackMatchedEffEtaTot->getBinContent(ibin) == 0) {
eff = 0;
bin_err = 0;
}
meMVAEtaMatchEff_->setBinContent(ibin, eff);
meMVAEtaMatchEff_->setBinError(ibin, bin_err);
}
}

// ------------ method fills 'descriptions' with the allowed parameters for the module ----------
Expand Down
154 changes: 138 additions & 16 deletions Validation/MtdValidation/plugins/MtdTracksValidation.cc
View file
Open in desktop
Expand Up @@ -10,6 +10,7 @@
#include "DQMServices/Core/interface/DQMStore.h"

#include "DataFormats/Common/interface/ValidHandle.h"
#include "DataFormats/Math/interface/deltaR.h"
#include "DataFormats/Math/interface/GeantUnits.h"
#include "DataFormats/ForwardDetId/interface/ETLDetId.h"
#include "DataFormats/ForwardDetId/interface/BTLDetId.h"
Expand All @@ -32,6 +33,11 @@
#include "Geometry/MTDGeometryBuilder/interface/ProxyMTDTopology.h"
#include "Geometry/MTDGeometryBuilder/interface/RectangularMTDTopology.h"

#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
#include "SimGeneral/HepPDTRecord/interface/ParticleDataTable.h"
#include "HepMC/GenRanges.h"
#include "CLHEP/Units/PhysicalConstants.h"

class MtdTracksValidation : public DQMEDAnalyzer {
public:
explicit MtdTracksValidation(const edm::ParameterSet&);
Expand All @@ -44,17 +50,30 @@ class MtdTracksValidation : public DQMEDAnalyzer {

void analyze(const edm::Event&, const edm::EventSetup&) override;

const bool mvaGenSel(const HepMC::GenParticle&, const float&);
const bool mvaRecSel(const reco::TrackBase&, const reco::Vertex&, const double&, const double&);
const bool mvaGenRecMatch(const HepMC::GenParticle&, const double&, const reco::TrackBase&);

// ------------ member data ------------

const std::string folder_;
const float trackMinPt_;
const float trackMinEta_;
const float trackMaxEta_;

static constexpr double etacutGEN_ = 4.; // |eta| < 4;
static constexpr double etacutREC_ = 3.; // |eta| < 3;
static constexpr double pTcut_ = 0.7; // PT > 0.7 GeV
static constexpr double deltaZcut_ = 0.1; // dz separation 1 mm
static constexpr double deltaPTcut_ = 0.05; // dPT < 5%
static constexpr double deltaDRcut_ = 0.03; // DeltaR separation

edm::EDGetTokenT<reco::TrackCollection> GenRecTrackToken_;
edm::EDGetTokenT<reco::TrackCollection> RecTrackToken_;
edm::EDGetTokenT<std::vector<reco::Vertex>> RecVertexToken_;

edm::EDGetTokenT<edm::HepMCProduct> HepMCProductToken_;

edm::EDGetTokenT<edm::ValueMap<int>> trackAssocToken_;
edm::EDGetTokenT<edm::ValueMap<float>> pathLengthToken_;

Expand All @@ -69,6 +88,7 @@ class MtdTracksValidation : public DQMEDAnalyzer {
edm::EDGetTokenT<edm::ValueMap<float>> trackMVAQualToken_;

edm::ESGetToken<MTDTopology, MTDTopologyRcd> mtdtopoToken_;
edm::ESGetToken<HepPDT::ParticleDataTable, edm::DefaultRecord> particleTableToken_;

MonitorElement* meBTLTrackRPTime_;
MonitorElement* meBTLTrackEffEtaTot_;
Expand Down Expand Up @@ -99,9 +119,15 @@ class MtdTracksValidation : public DQMEDAnalyzer {
MonitorElement* meTrackMVAQual_;
MonitorElement* meTrackPathLenghtvsEta_;

MonitorElement* meVerNumber_;
MonitorElement* meVerZ_;
MonitorElement* meVerTime_;
MonitorElement* meMVATrackEffPtTot_;
MonitorElement* meMVATrackMatchedEffPtTot_;
MonitorElement* meMVATrackMatchedEffPtMtd_;
MonitorElement* meMVATrackEffEtaTot_;
MonitorElement* meMVATrackMatchedEffEtaTot_;
MonitorElement* meMVATrackMatchedEffEtaMtd_;
MonitorElement* meMVATrackResTot_;
MonitorElement* meMVATrackPullTot_;
MonitorElement* meMVATrackZposResTot_;
};

// ------------ constructor and destructor --------------
Expand All @@ -113,6 +139,7 @@ MtdTracksValidation::MtdTracksValidation(const edm::ParameterSet& iConfig)
GenRecTrackToken_ = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("inputTagG"));
RecTrackToken_ = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("inputTagT"));
RecVertexToken_ = consumes<std::vector<reco::Vertex>>(iConfig.getParameter<edm::InputTag>("inputTagV"));
HepMCProductToken_ = consumes<edm::HepMCProduct>(iConfig.getParameter<edm::InputTag>("inputTagH"));
trackAssocToken_ = consumes<edm::ValueMap<int>>(iConfig.getParameter<edm::InputTag>("trackAssocSrc"));
pathLengthToken_ = consumes<edm::ValueMap<float>>(iConfig.getParameter<edm::InputTag>("pathLengthSrc"));
tmtdToken_ = consumes<edm::ValueMap<float>>(iConfig.getParameter<edm::InputTag>("tmtd"));
Expand All @@ -125,6 +152,7 @@ MtdTracksValidation::MtdTracksValidation(const edm::ParameterSet& iConfig)
Sigmat0SafePidToken_ = consumes<edm::ValueMap<float>>(iConfig.getParameter<edm::InputTag>("sigmat0SafePID"));
trackMVAQualToken_ = consumes<edm::ValueMap<float>>(iConfig.getParameter<edm::InputTag>("trackMVAQual"));
mtdtopoToken_ = esConsumes<MTDTopology, MTDTopologyRcd>();
particleTableToken_ = esConsumes<HepPDT::ParticleDataTable, edm::DefaultRecord>();
}

MtdTracksValidation::~MtdTracksValidation() {}
Expand Down Expand Up @@ -312,17 +340,65 @@ void MtdTracksValidation::analyze(const edm::Event& iEvent, const edm::EventSetu
}
} //RECO tracks loop

// --- Loop over the RECO vertices ---
int nv = 0;
for (const auto& v : *RecVertexHandle) {
if (v.isValid()) {
meVerZ_->Fill(v.z());
meVerTime_->Fill(v.t());
nv++;
} else
cout << "The vertex is not valid" << endl;
// reco-gen matching used for MVA quality flag
const auto& primRecoVtx = *(RecVertexHandle.product()->begin());

auto GenEventHandle = makeValid(iEvent.getHandle(HepMCProductToken_));
const HepMC::GenEvent* mc = GenEventHandle->GetEvent();
double zsim = convertMmToCm((*(mc->vertices_begin()))->position().z());
double tsim = (*(mc->vertices_begin()))->position().t() * CLHEP::mm / CLHEP::c_light;

auto pdt = iSetup.getHandle(particleTableToken_);
const HepPDT::ParticleDataTable* pdTable = pdt.product();

// select events with reco vertex close to true simulated primary vertex
if (std::abs(primRecoVtx.z() - zsim) < deltaZcut_) {
index = 0;
for (const auto& trackGen : *GenRecTrackHandle) {
const reco::TrackRef trackref(iEvent.getHandle(GenRecTrackToken_), index);
index++;

// select the reconstructed track

if (trackAssoc[trackref] == -1) {
continue;
}

if (mvaRecSel(trackGen, primRecoVtx, t0Safe[trackref], Sigmat0Safe[trackref])) {
meMVATrackEffPtTot_->Fill(trackGen.pt());
meMVATrackEffEtaTot_->Fill(std::abs(trackGen.eta()));

double dZ = trackGen.vz() - zsim;
double dT(-9999.);
double pullT(-9999.);
if (Sigmat0Safe[trackref] != -1.) {
dT = t0Safe[trackref] - tsim;
pullT = dT / Sigmat0Safe[trackref];
}
for (const auto& genP : mc->particle_range()) {
// select status 1 genParticles and match them to the reconstructed track

float charge = pdTable->particle(HepPDT::ParticleID(genP->pdg_id())) != nullptr
? pdTable->particle(HepPDT::ParticleID(genP->pdg_id()))->charge()
: 0.f;
if (mvaGenSel(*genP, charge)) {
if (mvaGenRecMatch(*genP, zsim, trackGen)) {
meMVATrackZposResTot_->Fill(dZ);
meMVATrackMatchedEffPtTot_->Fill(trackGen.pt());
meMVATrackMatchedEffEtaTot_->Fill(std::abs(trackGen.eta()));
if (pullT > -9999.) {
meMVATrackResTot_->Fill(dT);
meMVATrackPullTot_->Fill(pullT);
meMVATrackMatchedEffPtMtd_->Fill(trackGen.pt());
meMVATrackMatchedEffEtaMtd_->Fill(std::abs(trackGen.eta()));
}
break;
}
}
}
}
}
}
meVerNumber_->Fill(nv);
}

// ------------ method for histogram booking ------------
Expand Down Expand Up @@ -383,9 +459,22 @@ void MtdTracksValidation::bookHistograms(DQMStore::IBooker& ibook, edm::Run cons
meTrackMVAQual_ = ibook.book1D("TrackMVAQual", "Track MVA Quality as stored in Value Map ; MVAQual", 100, 0, 1);
meTrackPathLenghtvsEta_ = ibook.bookProfile(
"TrackPathLenghtvsEta", "MTD Track pathlength vs MTD track Eta;|#eta|;Pathlength", 100, 0, 3.2, 100.0, 400.0, "S");
meVerZ_ = ibook.book1D("VerZ", "RECO Vertex Z;Z_{RECO} [cm]", 180, -18, 18);
meVerTime_ = ibook.book1D("VerTime", "RECO Vertex Time;t0 [ns]", 100, -1, 1);
meVerNumber_ = ibook.book1D("VerNumber", "RECO Vertex Number: Number of vertices", 100, 0, 500);
meMVATrackEffPtTot_ = ibook.book1D("MVAEffPtTot", "Pt of tracks associated to LV; track pt [GeV] ", 110, 0., 11.);
meMVATrackMatchedEffPtTot_ =
ibook.book1D("MVAMatchedEffPtTot", "Pt of tracks associated to LV matched to GEN; track pt [GeV] ", 110, 0., 11.);
meMVATrackMatchedEffPtMtd_ = ibook.book1D(
"MVAMatchedEffPtMtd", "Pt of tracks associated to LV matched to GEN with time; track pt [GeV] ", 110, 0., 11.);
meMVATrackEffEtaTot_ = ibook.book1D("MVAEffEtaTot", "Pt of tracks associated to LV; track eta ", 66, 0., 3.3);
meMVATrackMatchedEffEtaTot_ =
ibook.book1D("MVAMatchedEffEtaTot", "Pt of tracks associated to LV matched to GEN; track eta ", 66, 0., 3.3);
meMVATrackMatchedEffEtaMtd_ = ibook.book1D(
"MVAMatchedEffEtaMtd", "Pt of tracks associated to LV matched to GEN with time; track eta ", 66, 0., 3.3);
meMVATrackResTot_ = ibook.book1D(
"MVATrackRes", "t_{rec} - t_{sim} for LV associated tracks; t_{rec} - t_{sim} [ns] ", 120, -0.15, 0.15);
meMVATrackPullTot_ =
ibook.book1D("MVATrackPull", "Pull for associated tracks; (t_{rec}-t_{sim})/#sigma_{t}", 50, -5., 5.);
meMVATrackZposResTot_ = ibook.book1D(
"MVATrackZposResTot", "Z_{PCA} - Z_{sim} for associated tracks;Z_{PCA} - Z_{sim} [cm] ", 100, -0.1, 0.1);
}

// ------------ method fills 'descriptions' with the allowed parameters for the module ------------
Expand All @@ -397,6 +486,7 @@ void MtdTracksValidation::fillDescriptions(edm::ConfigurationDescriptions& descr
desc.add<edm::InputTag>("inputTagG", edm::InputTag("generalTracks"));
desc.add<edm::InputTag>("inputTagT", edm::InputTag("trackExtenderWithMTD"));
desc.add<edm::InputTag>("inputTagV", edm::InputTag("offlinePrimaryVertices4D"));
desc.add<edm::InputTag>("inputTagH", edm::InputTag("generatorSmeared"));
desc.add<edm::InputTag>("tmtd", edm::InputTag("trackExtenderWithMTD:generalTracktmtd"));
desc.add<edm::InputTag>("sigmatmtd", edm::InputTag("trackExtenderWithMTD:generalTracksigmatmtd"));
desc.add<edm::InputTag>("t0Src", edm::InputTag("trackExtenderWithMTD:generalTrackt0"));
Expand All @@ -416,4 +506,36 @@ void MtdTracksValidation::fillDescriptions(edm::ConfigurationDescriptions& descr
descriptions.add("mtdTracks", desc);
}

const bool MtdTracksValidation::mvaGenSel(const HepMC::GenParticle& gp, const float& charge) {
bool match = false;
if (gp.status() != 1) {
return match;
}
match = charge != 0.f && gp.momentum().perp() > pTcut_ && std::abs(gp.momentum().eta()) < etacutGEN_;
return match;
}

const bool MtdTracksValidation::mvaRecSel(const reco::TrackBase& trk,
const reco::Vertex& vtx,
const double& t0,
const double& st0) {
bool match = false;
match = trk.pt() > pTcut_ && std::abs(trk.eta()) < etacutREC_ && std::abs(trk.vz() - vtx.z()) <= deltaZcut_;
if (st0 > 0.) {
match = match && std::abs(t0 - vtx.t()) < 3. * st0;
}
return match;
}

const bool MtdTracksValidation::mvaGenRecMatch(const HepMC::GenParticle& genP,
const double& zsim,
const reco::TrackBase& trk) {
bool match = false;
double dR = reco::deltaR(genP.momentum(), trk.momentum());
double genPT = genP.momentum().perp();
match =
std::abs(genPT - trk.pt()) < trk.pt() * deltaPTcut_ && dR < deltaDRcut_ && std::abs(trk.vz() - zsim) < deltaZcut_;
return match;
}

DEFINE_FWK_MODULE(MtdTracksValidation);