BmnDataIterator.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
 
    BmnDataIterator abstract class header-only implementation
    Base class for ROOT file and RAM iterators
*/
 
#ifndef BMNDATAITERATOR_H
#define BMNDATAITERATOR_H
 
#include "BmnSimpleProgressBar.h"
#include "RtypesCore.h"
 
class BmnDataIterator
{
  public:
    inline BmnDataIterator(BmnSimpleProgressBar* pb = nullptr)
        : fpProgressBar(pb)
    {}
    virtual ~BmnDataIterator() = default;
 
    virtual Double_t HitX() const = 0;
    virtual Double_t HitY() const = 0;
    virtual Double_t HitZ() const = 0;
    virtual Double_t HitWx() const = 0;   // 1 / dx^2
    virtual Double_t HitWy() const = 0;   // 1 / dy^2
    virtual Int_t HitDetectorID() const = 0;
    virtual Int_t HitsInTrack() const = 0;
    virtual void ResetHits() = 0;
    virtual void ResetAll() = 0;
    virtual void NextHit() = 0;
    virtual void NextTrack() = 0;
 
    Bool_t Initialized() const noexcept { return fInitialized; }
    Bool_t EndOfTracks() const noexcept { return fEndOfTracks; }
    Bool_t EndOfHits() const noexcept { return fEndOfHits; }
 
  protected:
    Bool_t fInitialized{kFALSE}, fEndOfTracks{kTRUE}, fEndOfHits{kTRUE};
 
    // progress bar interface
  private:
    BmnSimpleProgressBar* fpProgressBar;
    Float_t fTickProgress{0.0f}, fProgress{0.0f};
 
  protected:
    BmnSimpleProgressBar* GetProgressBar() const noexcept { return fpProgressBar; }
    Float_t GetTickProgress() const noexcept { return fTickProgress; }
    void SetTickProgress(Float_t progress) { fTickProgress = progress; }
    void ProgressTick()
    {
        if (!fpProgressBar)
            return;
        fProgress += fTickProgress;
        if (fProgress >= 0.01f)   // fixed interval in BmnSimpleProgressBar
        {
            fpProgressBar->Tick();
            fProgress = 0.0f;
        }
    }
    void ProgressReset()
    {
        if (!fpProgressBar)
            return;
        fpProgressBar->Init();
        fProgress = 0.0f;
    }
    void ProgressClear()
    {
        if (!fpProgressBar)
            return;
        fpProgressBar->Clear();
    }
};
 
#endif