ElementModelScaleSag.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONNSWASBUILT_ELEMENTMODELSCALESAG_H
#define MUONNSWASBUILT_ELEMENTMODELSCALESAG_H
 
#include "MuonNSWAsBuilt/ElementModel.h"
 
namespace NswAsBuilt {
  class ElementModelScaleSag : public ElementModel {
    public:
      enum parameter_t {
        X=0,
        Y=1,
        Z=2,
        THX=3,
        THY=4,
        THZ=5,
        EGX=6,
        EGY=7,
        EGZ=8,
        SAGX=9,
        SAGY=10,
        DEGX=11,
        DEGY=12,
        PGX=13,
        PGY=14,
        DSAGX=15,
        DSAGY=16
      };
 
      ElementModelScaleSag(double lenX, double lenY, Amg::Vector3D defo0);
      ElementModelScaleSag() = delete;
 
      // Number of parameters used
      virtual int nParameters() const override { return 17; }
 
      // Transform a set of vectors expressed in local frame, stored in a matrix
      virtual void transform(const ParameterVector& parvec, VectorSetRef local) const override;
 
      // Cache the rigid component of this deformation model
      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: applyDeformation makes use of the Eigen
      // optimizations.
      void applyDeformation2(const ParameterVector& parvec, VectorSetRef local) const;
 
      // Basic ingredients of the calculation, called by method applyDeformation
      static Amg::Vector3D DEg( double egx, double egy, double egz, const Amg::Vector3D& d0) ;
      Amg::Vector3D DSagX(double sagx, const Amg::Vector3D& d0) const;
      Amg::Vector3D DSagY(double sagy, const Amg::Vector3D& d0) const;
      Amg::Vector3D DDSagX(double dsagx, const Amg::Vector3D& d0) const;
      Amg::Vector3D DDSagY(double dsagy, const Amg::Vector3D& d0) const;
      Amg::Vector3D DDegX(double degx, const Amg::Vector3D& d0) const;
      Amg::Vector3D DDegY(double degy, const Amg::Vector3D& d0) const;
      Amg::Vector3D DPgX(double pgx, const Amg::Vector3D& d0) const;
      Amg::Vector3D DPgY(double pgy, const Amg::Vector3D& d0) const;
 
      // Properties and scales needed for the calculation
      double m_lenX {0.};
      double m_lenY {0.};
      Amg::Vector3D m_defo0 {Amg::Vector3D::Zero()};
  };
}
 
#endif