BmnAlignerApply.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
 
    BmnAlignerApply class declaration
    Interface for alignment correction application
    To be used in reconstruction macros
*/
 
#ifndef BMNALIGNERAPPLY_H
#define BMNALIGNERAPPLY_H
 
#include "BmnAligner.h"
#include "BmnTask.h"
#include "TClonesArray.h"
#include "TH2D.h"
 
#include <map>
#include <vector>
 
class BmnAlignerApply
    : public BmnAligner<>
    , public BmnTask
{
  private:
    TClonesArray* fpStsHitsArray{nullptr};
    const char* fAlignFilePath;
    const Bool_t fLorentzCorrectionApply;
 
  public:
    BmnAlignerApply() = delete;
    BmnAlignerApply(const BmnMeasureModel* measureModel,
                    const BmnDetectorModel* modelDetector,
                    const char* alignFilePath,
                    Bool_t applyLorentz = kTRUE)
        : BmnAligner<>(measureModel, modelDetector)
        , fAlignFilePath(alignFilePath)
        , fLorentzCorrectionApply(applyLorentz) {};
    virtual ~BmnAlignerApply() = default;
 
    virtual InitStatus Init();
    virtual void Exec(Option_t* opt);
    virtual void Finish();
 
  private:
    /*
     * Lorentz corrections section-----------------------------------------------------------------
     *
     * The code below is directly taken from BmnToCbmHitConverter::ApplyAlignment(CbmStsHit*).
     * It implements the Lorentz correction of hit positions and was not revised, except few
     * lines marked with tag PLV. Indentation is the kept as in the original code.
     */
 
    // PLV : begin
    // hLorCorX and hLorCorY are moved from local variables of method ApplyAlignment()
    // static attribute was removed and indentation was changed
    // original code:
    // static TH2D *hLorCorX[2][17] = {{nullptr}, {nullptr}}, *hLorCorY[2][17] = {{nullptr}, {nullptr}};
    // modified code:
    TH2D *hLorCorX[2][17] = {{nullptr}, {nullptr}}, *hLorCorY[2][17] = {{nullptr}, {nullptr}};
    // PLV : end
    void ReadCorrections(int irun,
                         int igem,
                         std::map<int, std::vector<Double_t>>* locCorX,
                         std::map<int, std::vector<Double_t>>* locCorY,
                         int& nx,
                         int& ny);
    Bool_t fieldFlag;   // AZ-310823
 
    /*
     * End of Lorentz corrections section ---------------------------------------------------------
     */
  public:
    ClassDef(BmnAlignerApply, 0);   // don't write to ROOT file
};
 
#endif   // BMNALIGNERAPPLY_H