MeasurementDefs.h

Go to the documentation of this file.

 
/*
   Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
#ifndef XAODMEASUREMENTBASE_MEASUREMENTDEFS_H
#define XAODMEASUREMENTBASE_MEASUREMENTDEFS_H
// EDM include(s):
#include <array>
 
#include "AthContainers/AuxElement.h"
#include "EventPrimitives/EventPrimitives.h"
 
#define AUX_MEASUREMENTVAR(VAR, DIM)                         \
    do {                                                     \
        static const std::string varName =                   \
            std::string{#VAR} + "Dim" + std::to_string(DIM); \
        static const auxid_t auxId = getAuxID(varName, VAR); \
        regAuxVar(auxId, varName, VAR);                      \
    } while (false);
 
namespace xAOD {
 
enum class UncalibMeasType {
    Other = 0,
    // InDet
    PixelClusterType = 1,
    StripClusterType = 2,
    // Muon
    MdtDriftCircleType = 3,
    RpcStripType = 4,
    TgcStripType = 5,
    MMClusterType = 6,
    sTgcStripType = 7,
    // HGTD    
    HGTDClusterType = 8,
    // Do not add anything after nTypes
    nTypes
};
 
using DetectorIDHashType = unsigned int;
using DetectorIdentType = long unsigned int;
template <size_t N>
using PosAccessor = SG::AuxElement::Accessor<std::array<float, N>>;
template <size_t N>
using CovAccessor = SG::AuxElement::Accessor<std::array<float, N * N>>;
template <size_t N>
using MeasVector = Eigen::Matrix<float, N, 1>;
template <size_t N>
using MeasMatrix = Eigen::Matrix<float, N, N>;
 
template <size_t N>
using VectorMap = Eigen::Map<MeasVector<N>>;
template <size_t N>
using ConstVectorMap = Eigen::Map<const MeasVector<N>>;
 
template <size_t N>
using MatrixMap = Eigen::Map<MeasMatrix<N>>;
template <size_t N>
using ConstMatrixMap = Eigen::Map<const MeasMatrix<N>>;
 
template <int N> MeasVector<N> toStorage(const AmgVector(N)& amgVec){
        MeasVector<N> vec{MeasVector<N>::Zero()};
        for (int i =0 ; i < N ; ++i) vec[i] = amgVec[i];
        return vec;
}
 
template <int N> MeasMatrix<N> toStorage(const AmgSymMatrix(N)& amgMat) {
     MeasMatrix<N> mat{MeasMatrix<N>::Zero()};
     for (int i =0 ; i < N; ++i){
        for (int j =0 ; j < N; ++j) {
            mat(i,j) = amgMat(i, j);
        }
     }
     return mat;
}
 
template <int N> AmgSymMatrix(N) toEigen(const ConstMatrixMap<N>& xAODmat) {
    AmgSymMatrix(N) mat{AmgSymMatrix(N)::Zero()};
    for (int i=0; i < N; ++i){
        for (int j =0; j < N; ++j){
            mat(i, j) = xAODmat(i, j);
        }
    }
    return mat;
}
 
template <int N> AmgVector(N) toEigen(const ConstVectorMap<N>& xAODvec) {
    AmgVector(N) v{AmgVector(N)::Zero()};
    for (int i = 0 ; i < N; ++i) {
        v[i] = xAODvec[i];
    }
    return v;
}
 
 
}  // namespace xAOD
#endif