BmnAlignerTest.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
 
    BmnAlignerTest class declaration.
    Implements alignment algorithm test with predefined task with known solution.
    Each detector is defined by z-coordinate and is considered to be infinite plane.
    Set of random linear tracks are generated.
*/
 
#ifndef BMNALIGNERTEST_H
#define BMNALIGNERTEST_H
 
#include "BmnATestDetector.h"
#include "BmnATestHit.h"
#include "BmnATestTrack.h"
#include "BmnAligner.h"
 
class BmnAlignerTest : public BmnAligner<BmnATestHit>
{
  public:
    BmnAlignerTest() = delete;
    BmnAlignerTest(const BmnMeasureModel* measureModel,
                   const BmnATestDetector* modelDetector,
                   Int_t nTracks,
                   const char* constraintsPath = nullptr,
                   Int_t nThreads = 1);
    virtual ~BmnAlignerTest();
    virtual Bool_t PrepareData(const char* tracksPath = nullptr) override;
    virtual Bool_t IterateAlignment() override;
    virtual Double_t CalculateMSE(Bool_t withResiduals) override;
    virtual void ReportResults() const override;
    virtual void Draw() override;
 
    Bool_t SaveTracks(const char* path) const;
    Bool_t LoadTracks(const char* path);
 
    const BmnATestTrack& GetTestTrack(Int_t trackID) const noexcept { return fpTestTracks[trackID]; }
    const BmnATestHit& GetTestHit(Int_t hitID) const noexcept { return fpHits[hitID]; }
 
  private:
    const BmnATestDetector* fpDetectors;
    Byte_t* fpTestTracksData{nullptr};
    BmnATestTrack* fpTestTracks{nullptr};
};
 
#endif