ElementModelSTGC.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONNSWASBUILT_ELEMENTMODELSTGC_H
#define MUONNSWASBUILT_ELEMENTMODELSTGC_H
 
#include "MuonNSWAsBuilt/ElementModel.h"
 
namespace NswAsBuilt {
  class ElementModelSTGC : public ElementModel {
    public:
      enum parameter_t {
        X=0,
        Y=1,
        Z=2,
        THX=3,
        THY=4,
        THZ=5,
        ROT=6,
        OFF=7,
        SCL=8,
        NPAR=9
      };
 
      ElementModelSTGC(double lenX, double lenY, Amg::Vector3D defo0);
      ElementModelSTGC() = delete;

      virtual int nParameters() const override { return 10; }

      virtual void transform(const ParameterVector& parvec, VectorSetRef local) const override;

      virtual void cacheTransform(ParameterVector& parvec) const override;
 
      /*
       * Helper methods to convert parameter index to string representation
       */
      virtual ipar_t getParameterIndex(const std::string& parname) const override;
      virtual std::string getParameterName(ipar_t ipar) const override;
 
    private:
      // The actual implementation: applyDeformation is the reference
      // implementation, which does not make use of Eigen optimizations
      // Not actually used, provided for comparison purposes
      void applyDeformation(const ParameterVector& parvec, Eigen::Ref<Amg::Vector3D> local) const;
  
      // The actual implementation: applyDeformation2 makes use of the Eigen
      // optimizations.
      void applyDeformation2(const ParameterVector& parvec, VectorSetRef local) const;
      
      static Amg::Vector3D stgcOffset(double off) ;
      static Amg::Vector3D stgcRotation(double rot, const Amg::Vector3D& d0) ;
      Amg::Vector3D stgcScale(double scl, const Amg::Vector3D& d0) const;
      Amg::Vector3D stgcNonPar(double npar, const Amg::Vector3D& d0) const;
      
      // Properties and scales needed for the calculation
      double m_lenY{0.};
      double m_lenX{0.};
      Amg::Vector3D m_defo0 {Amg::Vector3D::Zero()};
 
  };
}
 
#endif