TFCSLateralShapeParametrizationHitBase.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TFCSLateralShapeParametrizationHitBase_h
#define TFCSLateralShapeParametrizationHitBase_h
 
#include "ISF_FastCaloSimEvent/TFCSLateralShapeParametrization.h"
 
class CaloGeometry;
 
class TFCSLateralShapeParametrizationHitBase
    : public TFCSLateralShapeParametrization {
public:
  TFCSLateralShapeParametrizationHitBase(const char *name = nullptr,
                                         const char *title = nullptr);
 
  virtual double
  get_sigma2_fluctuation(TFCSSimulationState &simulstate,
                         const TFCSTruthState *truth,
                         const TFCSExtrapolationState *extrapol) const;
 
  virtual int get_number_of_hits(TFCSSimulationState &simulstate,
                                 const TFCSTruthState *truth,
                                 const TFCSExtrapolationState *extrapol) const;
 
  virtual float get_E_hit(TFCSSimulationState &simulstate,
                          const TFCSTruthState *truth,
                          const TFCSExtrapolationState *extrapol) const;
 
  virtual float getMinWeight() const;
  virtual float getMaxWeight() const;
 
  class Hit {
  public:
    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
    Hit(float eta, float phi, float E)
        : m_eta_x(eta), m_phi_y(phi), m_z(0.), m_E(E), m_useXYZ(false), m_center_r(0.),
          m_center_z(0.), m_center_eta(0.), m_center_phi(0.){};
    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.){};
 
    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;
    }
    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;
    }
 
    inline void reset_center() {
      m_center_r = 0;
      m_center_z = 0;
      m_center_eta = 0;
      m_center_phi = 0;
    }
 
    inline void reset() {
      m_eta_x = 0.;
      m_phi_y = 0.;
      m_z = 0.;
      m_E = 0.;
      m_useXYZ = false;
    }
 
    inline float &eta() { return m_eta_x; };
    inline float &phi() { return m_phi_y; };
    inline float &x() { return m_eta_x; };
    inline float &y() { return m_phi_y; };
    inline float &E() { return m_E; };
    inline float &z() { return m_z; };
    inline long unsigned int &idx() { return m_hit_index; };
    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);
    }
    inline float &center_r() { return m_center_r; }
    inline float &center_z() { return m_center_z; }
    inline float &center_eta() { return m_center_eta; }
    inline float &center_phi() { return m_center_phi; }
    inline void setCenter_r(float r) { m_center_r = r; }
    inline void setCenter_z(float z) { m_center_z = z; }
    inline void setCenter_eta(float eta) { m_center_eta = eta; }
    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;
    long unsigned int m_hit_index{};
    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;
  };
 
  virtual FCSReturnCode simulate_hit(Hit &hit, TFCSSimulationState &simulstate,
                                     const TFCSTruthState *truth,
                                     const TFCSExtrapolationState *extrapol);
 
private:
  ClassDefOverride(TFCSLateralShapeParametrizationHitBase,
                   1) // TFCSLateralShapeParametrizationHitBase
};
 
#endif