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.int
 
    BmnAlignerApply class declaration
    Interface for alignment correction application
    To be used in reconstruction macros
*/
 
#ifndef BMNALIGNERAPPLY_H
#define BMNALIGNERAPPLY_H
 
#include "BmnAlignDefines.h"
#include "BmnAlignHierarchy.h"
#include "BmnDetectorModel.h"
#include "BmnMeasureModel.h"
#include "BmnTask.h"
 
#include <TClonesArray.h>
#include <TH2D.h>
#include <map>
 
/*
 *  Applies a stored alignment solution to hits during reconstruction.
 *
 *  It reads a solution file and shifts hits by what it says; it does not solve
 *  anything, so it holds no aligner. The corrections a solution file lists per
 *  element are the *effective* ones - whatever the hierarchy that produced them
 *  looked like, its levels are already summed into these - hence the flat
 *  parametrization below is all that is needed to read the file.
 */
 
class BmnAlignerApply : public BmnTask
{
  private:
    const BmnMeasureModel* fpMeasureModel;
    const BmnDetectorModel* fpDetModel;
    BmnAlignHierarchy fHierarchy;   // flat, just to address the file
    VecSVectGL_t fCorrections;      // per element, as loaded
 
    const char* fAlignFilePath;
    const Bool_t fLorentzCorrectionApply;
    TClonesArray* fpStsHitsArray{nullptr};
    Int_t fHitsAligned{0}, fHitsIgnored{0}, fHitsFailed{0};
 
  public:
    BmnAlignerApply() = delete;
    BmnAlignerApply(const BmnMeasureModel* measureModel,
                    const BmnDetectorModel* modelDetector,
                    const char* alignFilePath,
                    Bool_t applyLorentz = kTRUE)
        : fpMeasureModel(measureModel)
        , fpDetModel(modelDetector)
        , fAlignFilePath(alignFilePath)
        , fLorentzCorrectionApply(applyLorentz) {};
    virtual ~BmnAlignerApply();
 
    virtual InitStatus Init();
    virtual void Exec(Option_t* opt);
    virtual void Finish();
 
  private:
    InitStatus LorentzCorrectionInit();
    /*
     * Lorentz corrections section-----------------------------------------------------------------
     *
     * The members below are taken from BmnToCbmHitConverter. The histograms are detached from
     * ROOT directories (AZ-010926), so this class owns them and deletes them in the destructor.
     */
    Bool_t fieldFlag{kFALSE};
    std::map<int, TH2D*> fhLorCorX[17], fhLorCorY[17];   // AZ-220826
 
    /*
     * End of Lorentz corrections section ---------------------------------------------------------
     */
  public:
    ClassDef(BmnAlignerApply, 0);   // don't write to ROOT file
};
 
#endif   // BMNALIGNERAPPLY_H