TFCSLateralShapeParametrizationHitChain.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TFCSLateralShapeParametrizationHitChain_h
#define TFCSLateralShapeParametrizationHitChain_h
 
#include "ISF_FastCaloSimEvent/TFCSLateralShapeParametrization.h"
#include "ISF_FastCaloSimEvent/TFCSLateralShapeParametrizationHitBase.h"
#include <vector>
 
class ICaloGeometry;
 
class TFCSLateralShapeParametrizationHitChain
    : public TFCSLateralShapeParametrization {
public:
#ifdef USE_GPU
  // modify the construction function in order to get the geometry for FCS-GPU
  TFCSLateralShapeParametrizationHitChain(const char *name = nullptr,
                                          const char *title = nullptr,
                                          ICaloGeometry *geo = nullptr);
  // override the set_geometry function for FCS-GPU
  // the geometry is needed to assign the deposit energy to cell
  // will not compile by default
  virtual void set_geometry(ICaloGeometry *geo) override;
#else
  TFCSLateralShapeParametrizationHitChain(const char *name = nullptr,
                                          const char *title = nullptr);
#endif
 
  TFCSLateralShapeParametrizationHitChain(
      TFCSLateralShapeParametrizationHitBase *hitsim);
 
  virtual FCSReturnCode
  init_hit(TFCSLateralShapeParametrizationHitBase::Hit &hit,
           TFCSSimulationState &simulstate, const TFCSTruthState *truth,
           const TFCSExtrapolationState *extrapol) const;
 
  virtual FCSReturnCode
  simulate(TFCSSimulationState &simulstate, const TFCSTruthState *truth,
           const TFCSExtrapolationState *extrapol) const override;
 
  typedef std::vector<TFCSLateralShapeParametrizationHitBase *> Chain_t;
  virtual unsigned int size() const override;
  virtual const TFCSParametrizationBase *
  operator[](unsigned int ind) const override;
  virtual TFCSParametrizationBase *operator[](unsigned int ind) override;
  virtual void set_daughter(unsigned int ind,
                            TFCSParametrizationBase *param) override;
  const Chain_t &chain() const { return m_chain; };
  Chain_t &chain() { return m_chain; };
  void push_back(const Chain_t::value_type &value) {
    m_chain.push_back(value);
  };
  void push_back_init(const Chain_t::value_type &value);
 
  unsigned int get_nr_of_init() const { return m_ninit; };
  void set_nr_of_init(unsigned int ninit) { m_ninit = ninit; };

  virtual void
  set_number_of_hits_simul(TFCSLateralShapeParametrizationHitBase *sim) {
    m_number_of_hits_simul = sim;
  };

  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
  get_sigma2_fluctuation(TFCSSimulationState &simulstate,
                         const TFCSTruthState *truth,
                         const TFCSExtrapolationState *extrapol) const;

  virtual float getMinWeight() const;
  virtual float getMaxWeight() const;
 
  static constexpr float s_max_sigma2_fluctuation = 1000; 
 
  void Print(Option_t *option = "") const override;
 
protected:
  void PropagateMSGLevel(MSG::Level level) const;
 
  virtual bool check_all_hits_simulated(
      TFCSLateralShapeParametrizationHitBase::Hit &hit,
      TFCSSimulationState &simulstate, const TFCSTruthState *truth,
      const TFCSExtrapolationState *extrapol, bool success) const;
 
  Chain_t m_chain;
 
private:
#ifdef USE_GPU
  // the FCS-GPU will use this geometry to get the correct cell
  // will not compile by default
  ICaloGeometry *m_geo;
#endif
 
  TFCSLateralShapeParametrizationHitBase *m_number_of_hits_simul;
  unsigned int m_ninit = 0;
 
  ClassDefOverride(TFCSLateralShapeParametrizationHitChain,
                   2) // TFCSLateralShapeParametrizationHitChain
};
 
#endif