Simulation/ISF/ISF_FastCaloSim/ISF_FastCaloGpu/ISF_FastCaloGpu/Hit.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ISF_FASTCALOGPU_HIT_G_H
#define ISF_FASTCALOGPU_HIT_G_H
 
#ifndef CUDA_HOSTDEV
#  ifdef __CUDACC__
#    define CUDA_HOSTDEV __host__ __device__
#  else
#    define CUDA_HOSTDEV
#  endif
#endif
 
class Hit {
public:
  CUDA_HOSTDEV Hit()
      : m_eta_x( 0. )
      , m_phi_y( 0. )
      , m_z( 0. )
      , m_E( 0. )
      , m_useXYZ( false )
      , m_center_r( 0. )
      , m_center_z( 0. )
      , m_center_eta( 0. )
      , m_center_phi( 0. ){}; // for hits with the same energy, m_E should normalized to E(layer)/nhit
  CUDA_HOSTDEV Hit( float eta, float phi, float E )
      : m_eta_x( eta )
      , m_phi_y( phi )
      , m_E( E )
      , m_useXYZ( false )
      , m_center_r( 0. )
      , m_center_z( 0. )
      , m_center_eta( 0. )
      , m_center_phi( 0. ){};
  CUDA_HOSTDEV Hit( float x, float y, float z, float E )
      : m_eta_x( x )
      , m_phi_y( y )
      , m_z( z )
      , m_E( E )
      , m_useXYZ( true )
      , m_center_r( 0. )
      , m_center_z( 0. )
      , m_center_eta( 0. )
      , m_center_phi( 0. ){};
 
  CUDA_HOSTDEV inline void setEtaPhiZE( float eta, float phi, float z, float E ) {
    m_eta_x  = eta;
    m_phi_y  = phi;
    m_z      = z;
    m_E      = E;
    m_useXYZ = false;
  }
  CUDA_HOSTDEV inline void setXYZE( float x, float y, float z, float E ) {
    m_eta_x  = x;
    m_phi_y  = y;
    m_z      = z;
    m_E      = E;
    m_useXYZ = true;
  }
 
  CUDA_HOSTDEV inline void reset() {
    m_eta_x  = 0.;
    m_phi_y  = 0.;
    m_z      = 0.;
    m_E      = 0.;
    m_useXYZ = false;
  }
 
  CUDA_HOSTDEV inline float& eta() { return m_eta_x; };
  CUDA_HOSTDEV inline float& phi() { return m_phi_y; };
  CUDA_HOSTDEV inline float& x() { return m_eta_x; };
  CUDA_HOSTDEV inline float& y() { return m_phi_y; };
  CUDA_HOSTDEV inline float& E() { return m_E; };
  CUDA_HOSTDEV inline float& z() { return m_z; }
  CUDA_HOSTDEV inline float  r() {
    if ( m_useXYZ )
      return sqrt( m_eta_x * m_eta_x + m_phi_y * m_phi_y );
    else
      return m_z / sinh( m_eta_x );
  }
  CUDA_HOSTDEV inline float& center_r() { return m_center_r; }
  CUDA_HOSTDEV inline float& center_z() { return m_center_z; }
  CUDA_HOSTDEV inline float& center_eta() { return m_center_eta; }
  CUDA_HOSTDEV inline float& center_phi() { return m_center_phi; }
  CUDA_HOSTDEV inline void   setCenter_r( float r ) { m_center_r = r; }
  CUDA_HOSTDEV inline void   setCenter_z( float z ) { m_center_z = z; }
  CUDA_HOSTDEV inline void   setCenter_eta( float eta ) { m_center_eta = eta; }
  CUDA_HOSTDEV inline void   setCenter_phi( float phi ) { m_center_phi = phi; }
 
private:
  float m_eta_x; // eta for barrel and end-cap, x for FCal
  float m_phi_y; // phi for barrel and end-cap, y for FCal
  float m_z;
  float m_E;
  bool  m_useXYZ;
  // Variables used to store extrapolated position
  float m_center_r;
  float m_center_z;
  float m_center_eta;
  float m_center_phi;
};
 
#endif