BmnDataFilters.h

Go to the documentation of this file.

/*
    BM@N alignment routine
    BM@N experiment at NICA complex, JINR, 2025
 
    Department: Math & Soft Group of HEP lab
    Author:     Igor Polev, polev@jinr.int
 
    Data selection policy for the ROOT file iterator
*/
 
#ifndef BMNDATAFILTERS_H
#define BMNDATAFILTERS_H
 
#include "BmnDetectorModel.h"
#include "CbmStsHit.h"
#include "CbmStsTrack.h"
 
#include <functional>
 
// USER DEFINED:
// Definition of trakcs and hits selection criteria
// ================================================
 
inline constexpr Int_t BMN_MIN_HITS_PER_TRACK{5};      // minimum allowed number of hits in a track
inline constexpr Double_t BMN_MAX_CHI2_PER_NDF{5.0};   // maximum allowed chi2 per NDF in a track
 
inline constexpr const char* BMN_TRACKS_CUT_STR{"at least 5 hits, max Chi2/NDF = 5.0"};
inline constexpr const char* BMN_HITS_CUT_STR{"no hit cuts applied"};
inline constexpr const char* BMN_VERTEX_CUT_STR{"one SiBT hit per station, then the residual test in Align()"};
 
// CAUTION: BmnAlignTestTask rewrites hit positions, and hit errors when it is
// given its own StdDev. A cut on either would let a test task keep other tracks
// than its source data, and so give its vertex hits other weights.
inline Bool_t BmnAcceptHit(const CbmStsHit& hit, const BmnDetectorModel& model)
{
    return !model.IgnoreHit(hit.GetDetectorID());
}
 
inline Bool_t BmnAcceptTrack(const CbmStsTrack& track, Int_t usableHits, Bool_t wholeTrackOnly = kTRUE)
{
    if (wholeTrackOnly && usableHits != track.GetNStsHits())
        return kFALSE;
    if (usableHits < BMN_MIN_HITS_PER_TRACK)
        return kFALSE;
    if (track.GetNDF() <= 0)
        return kFALSE;
    if (track.GetChi2() / track.GetNDF() > BMN_MAX_CHI2_PER_NDF)
        return kFALSE;
 
    return kTRUE;
}
 
// ROOT iterator filter functions
// ==============================
 
typedef std::function<Bool_t(const CbmStsHit&)> HitFilter_t;
typedef std::function<Bool_t(const CbmStsTrack&, Int_t usableHits)> TrackFilter_t;
 
inline HitFilter_t BmnDefaultHitFilter(const BmnDetectorModel& model)
{
    return [&model](const CbmStsHit& hit) { return BmnAcceptHit(hit, model); };
}
 
inline TrackFilter_t BmnDefaultTrackFilter(Bool_t wholeTrackOnly = kTRUE)
{
    return [wholeTrackOnly](const CbmStsTrack& track, Int_t usableHits) {
        return BmnAcceptTrack(track, usableHits, wholeTrackOnly);
    };
}
 
#endif   // BMNDATAFILTERS_H