BmnRootIterator.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.ru
 
    BmnRootIterator class declaration
    Iterator for tracks and hits stored in ROOT files
*/
 
#ifndef BMNROOTITERATOR_H
#define BMNROOTITERATOR_H
 
#include "BmnAlignDefines.h"
#include "BmnDataIterator.h"
#include "BmnDataReader.h"
#include "BmnDetectorModel.h"
#include "BmnSimpleProgressBar.h"
#include "CbmStsHit.h"
#include "CbmStsTrack.h"
#include "TTreeReaderValue.h"
 
#define BMN_TRACKS_BRACH_NAME "StsVector"
#define BMN_HITS_BRACH_NAME "StsHit"
#define BMN_HIT_CLASS_NAME "CbmStsHit"
 
class BmnRootIterator : public BmnDataIterator
{
  private:
    static constexpr Double_t MAX_WEIGHT{1.0 / (BMN_CLOSE_TO_ZERO * BMN_CLOSE_TO_ZERO)};
 
  public:
    typedef CbmStsTrack Track_t;
    typedef CbmStsHit Hit_t;
 
    BmnRootIterator() = delete;
    BmnRootIterator(const char* filesPathList, const BmnDetectorModel& detModel, BmnSimpleProgressBar* pb = nullptr);
    virtual ~BmnRootIterator();
 
    virtual void ResetHits() override;
    virtual void ResetAll() override;
    virtual void NextTrack() override;
    virtual void NextHit() override;
    virtual Double_t HitX() const override;
    virtual Double_t HitY() const override;
    virtual Double_t HitZ() const override;
    virtual Double_t HitWx() const override;
    virtual Double_t HitWy() const override;
    virtual Int_t HitDetectorID() const override;
    virtual Int_t HitsInTrack() const override;
 
    void CountElements();
    Long64_t GetTotalEvents() const noexcept { return fTotalEvents; }
    Int_t GetTotalTracks() const noexcept { return fTotalTracks; }
    Int_t GetTotalHits() const noexcept { return fTotalHits; }
    const Hit_t* GetHit() const noexcept { return fpCurrentHit; }
 
  private:
    const BmnDetectorModel* fpDetModel;
    // the class is involved into ROOT dictionary schema which utilizes forward
    // declarations - only pointers (or refs) to classes are possible as a field
    BmnDataReader* fpData{nullptr};
    TTreeReader* fpReader{nullptr};
    TTreeReaderValue<TClonesArray>*fpRootTracks{nullptr}, *fpRootHits{nullptr};
 
    Track_t* fpCurrentTrack{nullptr};
    Hit_t* fpCurrentHit{nullptr};
    Long64_t fTotalEvents{0};
    Int_t fTotalTracks{0}, fTotalHits{0}, fTracksInEvent{0}, fHitsInTrack{0}, fTrackIdx, fHitIdx;
 
    Bool_t fProgressByTracks{kTRUE};
 
    void UnsetProperties();
};
 
#endif