PixelDistortionData.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#ifndef PIXELDISTORTIONDATA_H
#define PIXELDISTORTIONDATA_H
 
#include "AthenaKernel/CLASS_DEF.h"
#include <map>
 
#include "GeoPrimitives/GeoPrimitives.h"
#include "AthenaKernel/CondCont.h"
 
class PixelDistortionData {
  public:
    void setVersion(int version);
    void setIds(const std::unordered_map<uint32_t,unsigned long long>& ids);
    void setDistortionMap(const std::unordered_map<uint32_t,std::vector<float>>& distortionMap);
 
    int getVersion() const;
    unsigned long long getId(uint32_t hashID) const;
    std::vector<float> getDistortionMap(uint32_t id) const;
 
    Amg::Vector2D correction(uint32_t hashID, const Amg::Vector2D & locpos, const Amg::Vector3D & direction) const;
    Amg::Vector2D correctReconstruction(uint32_t hashID, const Amg::Vector2D & locpos, const Amg::Vector3D & direction) const;
    Amg::Vector2D correctSimulation(uint32_t hashID, const Amg::Vector2D & locpos, const Amg::Vector3D & direction) const;
    static double getInSituZ(const double localeta, const double eta_size, const double localphi, const double phi_size, const float *disto) ;
    static double getSurveyZ(const double localeta, const double localphi, const float *disto) ;
    static bool isOldParam(const unsigned long long ull_id) ;
    static bool isIBL3D(const unsigned long long ull_id) ;
 
    void clear();
 
  private:
    int m_version{};
    std::unordered_map<uint32_t,unsigned long long> m_ids;
    std::unordered_map<uint32_t,std::vector<float>> m_distortionMap;
};
 
inline void PixelDistortionData::setVersion(int version) { m_version=version; }
inline void PixelDistortionData::setIds(const std::unordered_map<uint32_t,unsigned long long>& ids) { m_ids=ids; }
inline void PixelDistortionData::setDistortionMap(const std::unordered_map<uint32_t,std::vector<float>>& distortionMap) { m_distortionMap=distortionMap; }
 
inline int PixelDistortionData::getVersion() const { return m_version; }
 
CLASS_DEF( PixelDistortionData , 127567278 , 1 )
 
CONDCONT_DEF( PixelDistortionData, 127567679 );
 
#endif