CbmBbaAlignmentBody.h

Go to the documentation of this file.

/* Copyright (C) 2025 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergey Gorbunov [committer] */

 
#pragma once
 
#include <TGeoMatrix.h>
#include <TString.h>
 
#include <array>
 
namespace cbm::bba
{
  class AlignmentBody {
   public:
    AlignmentBody() = default;

    explicit AlignmentBody(double shiftX, double shiftY, double shiftZ, double rotX, double rotY, double rotZ);

    AlignmentBody(const std::string nodePath);

    ~AlignmentBody()                   = default;
    static constexpr int NofParameters = 6;  
 
    bool IsValid() const { return fIsValid; }
 
    bool CreateFromGeoNode(const std::string nodePath);
 
    void ActivateParameters(const std::array<bool, NofParameters>& isParActive);
 
    bool IsParameterActive(int i) const { return fIsParActive[i]; }
 
    void SetParameters(double shiftX = 0., double shiftY = 0., double shiftZ = 0., double rotX = 0., double rotY = 0.,
                       double rotZ = 0.);

    const std::string& GetNodePath() const { return fNodePath; }

    const TGeoHMatrix& GetGlobalTransform() const { return fGlobal; }

    std::array<double, 3> ApplyAlignmentToHit(const std::array<double, 3> hit) const;

    std::array<double, 3> ApplyInverseAlignmentToHit(const std::array<double, 3> hit) const;

    const std::array<double, NofParameters>& GetParameters() const { return fPar; }
 
    void GetParameters(double* par) const;

    const std::array<double, NofParameters>& GetInitialParameters() const { return fPar0; }

    const TGeoHMatrix& GetAlignmentMatrix() const { return fAlignment; }

    const TGeoHMatrix& GetHitTransform() const { return fHitTransform; }

    void SetStatistics(long statistics) { fStatistics = statistics; }

    void IncrementStatistics(int increment) { fStatistics += increment; }

    long GetStatistics() const { return fStatistics; }

    int GetTrackingStation() const { return fTrackingStation; }

    void SetTrackingStation(int station) { fTrackingStation = station; }
 
 
   private:
    int fTrackingStation{-1};
    bool fIsValid{false};  
 
    std::string fNodePath{};  
 
    std::array<double, NofParameters> fPar{0.};  
 
    std::array<bool, NofParameters> fIsParActive{false};  
 
 
    std::array<double, NofParameters> fPar0{0.};  
 
    TGeoHMatrix fAlignment;  
 
    TGeoHMatrix fGlobal;         
    TGeoHMatrix fGlobalInv;      
    TGeoHMatrix fGlobalNominal;  
 
    TGeoHMatrix fHitTransform;  

 
    long fStatistics{0};  
 
    ClassDefNV(AlignmentBody, 0);
  };
}  // namespace cbm::bba