BmnRamIterator.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
 
    BmnRamIterator class declaration
    Iterator for tracks and hits stored in RAM
*/
 
#ifndef BMNRAMITERATOR_H
#define BMNRAMITERATOR_H
 
#include "BmnDataIterator.h"
#include "BmnHitRecord.h"
#include "BmnSimpleProgressBar.h"
 
#include <stdexcept>
 
// different types of hits to be used in test tasks and real tasks
template<typename HitType = BmnHitRecord>
class BmnRamIterator : public BmnDataIterator
{
  public:
    BmnRamIterator() = delete;
    inline BmnRamIterator(const HitType* pHits,
                          const Int_t* pFirstHits,
                          Int_t nTracks,
                          BmnSimpleProgressBar* pb = nullptr,
                          Float_t progressPart = 1.0f);
    virtual ~BmnRamIterator() = default;
    BmnRamIterator(const BmnRamIterator& other) = default;
    BmnRamIterator Slice(Int_t firstTrack, Int_t tracksCount) const;
 
    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;
 
  private:
    const HitType* fpHits;
    const Int_t *fpFirstHits, fTracksCount;
    Int_t fTrackIdx, fHitIdx, fHitsInTrack, fNextFirstHit;
 
    const Float_t fProgressPart;
 
    void CountHitsInTrack() noexcept;
};
 
// inline implementation
 
template<typename HitType>
inline BmnRamIterator<HitType>::BmnRamIterator(const HitType* pHits,
                                               const Int_t* pFirstHits,
                                               Int_t nTracks,
                                               BmnSimpleProgressBar* pb,
                                               Float_t progressPart)
    : BmnDataIterator(pb)
    , fpHits{pHits}
    , fpFirstHits{pFirstHits}
    , fTracksCount{nTracks}
    , fProgressPart{progressPart}
{
    fInitialized = kTRUE;
    SetTickProgress(fProgressPart / static_cast<Float_t>(fTracksCount));
};
 
template<typename HitType>
inline BmnRamIterator<HitType> BmnRamIterator<HitType>::Slice(Int_t firstTrack, Int_t tracksCount) const
{
    if (firstTrack >= fTracksCount)
        throw std::invalid_argument("BmnRamIterator::Slice: invalid track index");
    if (tracksCount > fTracksCount - firstTrack)
        throw std::invalid_argument("BmnRamIterator::Slice: invalid tracks count");
    return BmnRamIterator(fpHits, fpFirstHits + firstTrack, tracksCount, GetProgressBar(),
                          static_cast<Float_t>(tracksCount) / static_cast<Float_t>(fTracksCount) * fProgressPart);
}
 
#endif