Skip to content

Add Track Filter Variables #33

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

Merged
gianelle merged 2 commits into from
Dec 11, 2025
Merged

Add Track Filter Variables #33

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
9bc8c84
LP: new track filters and BDT variables update
LeonardoPalombini Sep 17, 2025
24015f2
LP: add reduced chi2 track filter
LeonardoPalombini Oct 3, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions source/Utils/include/FilterTracks.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -89,6 +89,7 @@ class FilterTracks : public marlin::Processor

//! Cut off for the value ndf
int _MinNdf = 1;
float _MaxReducedChi2 = 1.e20;

//! Cut off for outliers number
int _MaxOutl = 10;
Expand All @@ -106,6 +107,9 @@ class FilterTracks : public marlin::Processor
float _MaxSigPhi = 999;
float _MaxSigQoverP = 999;

float _MaxD0 = 999.;
float _MaxZ0 = 999.;

//! NN parameters
std::string _NNmethod = ""; // if defined apply the NN
std::string _NNweights = ""; // xml file with weights
Expand Down
62 changes: 45 additions & 17 deletions source/Utils/src/FilterTracks.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -140,6 +140,24 @@ FilterTracks::FilterTracks()
_MaxSigQoverP
);

registerProcessorParameter("MaxReducedChi2",
"Max reduced Chi2",
_MaxReducedChi2,
_MaxReducedChi2
);

registerProcessorParameter("MaxPromptD0",
"Max d0",
_MaxD0,
_MaxD0
);

registerProcessorParameter("MaxPromptZ0",
"Max z0",
_MaxZ0,
_MaxZ0
);

registerProcessorParameter("Bz",
"Magnetic field in Tesla (default: 5)",
_Bz,
Expand Down Expand Up @@ -245,11 +263,15 @@ void FilterTracks::processEvent( LCEvent * evt )
{"trtnh", 0},
{"trch2", 0},
{"trndf", 0},
{"trnoh", 0},
{"tr_sigd0",0},
{"tr_sigz0",0},
{"tr_sigtheta",0},
{"tr_sigphi",0},
{"tr_sigqoverp",0}
{"tr_sigqoverp",0},
{"tr_redchi2",0},
{"tr_d0",0},
{"tr_z0",0}
};

if ( not _NNmethod.empty() ) {
Expand Down Expand Up @@ -279,13 +301,17 @@ void FilterTracks::processEvent( LCEvent * evt )
vars["trch2"] = trk->getChi2();

vars["trndf"] = trk->getNdf();

vars["tr_redchi2"] = vars["trch2"] / vars["trndf"];
vars["trnoh"] = (vars["trtnh"]-vars["trndf"]/2) / vars["trtnh"];
vars["tr_sigd0"] = sqrt(trk->getCovMatrix()[0]);
vars["tr_sigz0"] = sqrt(trk->getCovMatrix()[2]);
vars["tr_sigtheta"] = sqrt(trk->getCovMatrix()[9]);
vars["tr_sigphi"] = sqrt(trk->getCovMatrix()[5]);
vars["tr_sigqoverp"] = sqrt(trk->getCovMatrix()[14]);

vars["tr_d0"] = trk->getD0();
vars["tr_z0"] = trk->getZ0();

if(_BarrelOnly == true) {
bool endcaphits = false;
for(int j=0; j<vars["trtnh"]; ++j) {
Expand All @@ -301,14 +327,8 @@ void FilterTracks::processEvent( LCEvent * evt )
OutputTrackCollection->addElement(new IMPL::TrackImpl(*itrk));
}
} else { // track property cuts
if ( not _NNmethod.empty() ) { // NN cuts
auto prediction = reader->EvaluateMVA(_NNmethod);
if ( prediction > _NNthr ) {
auto itrk = dynamic_cast<IMPL::TrackImpl*>(trk);
OutputTrackCollection->addElement(new IMPL::TrackImpl(*itrk));
}
} else { // user cuts
if (vars["trtnh"] > _NHitsTotal &&

if (!(vars["trtnh"] > _NHitsTotal &&
vars["trtvhn"] > _NHitsVertex &&
vars["trtihn"] > _NHitsInner &&
vars["trtohn"] > _NHitsOuter &&
Expand All @@ -318,19 +338,27 @@ void FilterTracks::processEvent( LCEvent * evt )
theta < _MaxTheta &&
vars["trch2"] > _Chi2Spatial &&
vars["trndf"] > _MinNdf &&
vars["tr_redchi2"] < _MaxReducedChi2 &&
vars["trtnh"]-vars["trndf"]/2 < _MaxOutl &&
(vars["trtnh"]-vars["trndf"]/2) / vars["trtnh"] < _MaxOutlOverHits &&
vars["trnoh"] < _MaxOutlOverHits &&
vars["trthn"] < _MaxHoles &&
vars["tr_sigd0"] < _MaxSigD0 &&
vars["tr_sigz0"] < _MaxSigZ0 &&
vars["tr_sigtheta"] < _MaxSigTheta &&
vars["tr_sigphi"] < _MaxSigPhi &&
vars["tr_sigqoverp"] < _MaxSigQoverP)
{
auto itrk = dynamic_cast<IMPL::TrackImpl*>(trk);
OutputTrackCollection->addElement(new IMPL::TrackImpl(*itrk));
}
}
vars["tr_sigqoverp"] < _MaxSigQoverP &&
std::fabs(vars["tr_d0"]) < _MaxD0 &&
std::fabs(vars["tr_z0"]) < _MaxZ0)) continue;

if ( not _NNmethod.empty() ) { // NN cuts

auto prediction = reader->EvaluateMVA(_NNmethod);
if ( prediction <= _NNthr ) continue;
}

auto itrk = dynamic_cast<IMPL::TrackImpl*>(trk);
OutputTrackCollection->addElement(new IMPL::TrackImpl(*itrk));

}
}

Expand Down
Loading