TrigTools/TrigVrtSecInclusive/TrigVrtSecInclusive/Coordinate.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
#ifndef TRIGTOOLS_TRIG_VSI_COORD
#define TRIGTOOLS_TRIG_VSI_COORD

#include "TrigVrtSecInclusive/KDPoint.h"
 
#include "TMath.h"
#include "TH3D.h"
#include "TVector3.h"
 
#include <cmath>
 
namespace TrigVSI{

namespace Coordinate{

class Cartesian {
   public :
      static inline TVector3 X123toXYZ( double x1, double x2, double x3) {
         return TVector3(x1, x2, x3);
      };
 
      static inline TVector3 X123toXYZ( const KDPoint<double,3>& p ) {
         return TVector3(p.at(0),p.at(1),p.at(2));
      };
 
      static inline KDPoint<double,3> XYZtoX123( const TVector3& v ) {
         return KDPoint<double,3>({v.x(), v.y(), v.z()});
      };


      static inline KDPoint<double,3> Proj(const KDPoint<double,3>& p) {
         return p;
      };


      static inline KDPoint<int,3> ProjBin(const KDPoint<int,3>& ibin, const std::unique_ptr<TH3D>&) {
         return ibin;
      };

};
 
 

class Pseudo {
   public :
      static inline TVector3 X123toXYZ( double x1, double x2, double x3) {
         TVector3 tmp;
         tmp.SetPtEtaPhi(x1,x2,x3);
         return tmp;
      };
 
      static inline TVector3 X123toXYZ( const KDPoint<double,3>& p ) {
         TVector3 tmp;
         tmp.SetPtEtaPhi(p.at(0),p.at(1),p.at(2));
         return tmp;
      };
 
      static inline KDPoint<double,3> XYZtoX123( const TVector3& v ) {
         return KDPoint<double,3>({v.Perp(), v.Eta(), v.Phi()});
      };


      static inline KDPoint<double,3> Proj(const KDPoint<double,3>& p) {
         KDPoint<double,3> tmp = p;
         tmp[2] = std::fmod( p[2] + TMath::Pi(), 2. * TMath::Pi() ) - TMath::Pi();
         return tmp;
      };


      static inline KDPoint<int,3> ProjBin(const KDPoint<int,3>& ibin, const std::unique_ptr<TH3D>& map) {
         KDPoint<int,3> tmp = ibin;
         tmp[2] = (ibin[2] - 1) % (map->GetNbinsZ() - 1 + 1) + 1;
         return tmp;
      };

};
 
 

template<typename T>
struct is_coord {
   static auto check(...) -> std::false_type;
 
   template<typename _Coord>
   static auto check(_Coord*) ->
               decltype(   static_cast<TVector3(*)(const KDPoint<double,3>&)>(&_Coord::X123toXYZ),
                           static_cast<KDPoint<double,3>(*)(const TVector3&)>(&_Coord::XYZtoX123),
                           static_cast<KDPoint<double,3>(*)(const KDPoint<double,3>&)>(&_Coord::Proj),
                           static_cast<KDPoint<int,3>(*)(const KDPoint<int,3>&, const std::unique_ptr<TH3D>&)>(&_Coord::ProjBin),
                           std::true_type() );
 
   static constexpr T*   insPolicy = nullptr;
   static constexpr bool value = decltype(check(insPolicy))::value;
};
 
} // end of namespace Coordinate
 
} // end of namespace TrigVSI
 
#endif