TFCS2DFunctionLateralShapeParametrization.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TFCS2DFunctionLateralShapeParametrization_h
#define TFCS2DFunctionLateralShapeParametrization_h
 
#include "ISF_FastCaloSimEvent/TFCSLateralShapeParametrizationHitBase.h"
#include "ISF_FastCaloSimEvent/TFCS2DFunction.h"
#include "ISF_FastCaloSimEvent/TFCSTruthState.h"
 
class TH2;
 
class TFCS2DFunctionLateralShapeParametrization
    : public TFCSLateralShapeParametrizationHitBase {
public:
  TFCS2DFunctionLateralShapeParametrization(const char *name = nullptr,
                                            const char *title = nullptr);
  ~TFCS2DFunctionLateralShapeParametrization();

  enum FCSStatusBits {
    k_phi_symmetric =
        BIT(15) 
  };
 
  bool is_phi_symmetric() const { return TestBit(k_phi_symmetric); };
  virtual void set_phi_symmetric() { SetBit(k_phi_symmetric); };
  virtual void reset_phi_symmetric() { ResetBit(k_phi_symmetric); };

  void set_number_of_hits(float nhits);
 
  float get_number_of_expected_hits() const { return m_nhits; };

  virtual double
  get_sigma2_fluctuation(TFCSSimulationState &simulstate,
                         const TFCSTruthState *truth,
                         const TFCSExtrapolationState *extrapol) const override;

  virtual int
  get_number_of_hits(TFCSSimulationState &simulstate,
                     const TFCSTruthState *truth,
                     const TFCSExtrapolationState *extrapol) const override;

  virtual FCSReturnCode
  simulate_hit(Hit &hit, TFCSSimulationState &simulstate,
               const TFCSTruthState *truth,
               const TFCSExtrapolationState *extrapol) override;

  bool Initialize(TFCS2DFunction *func, float nhits = -1);
 
  TFCS2DFunction *function() { return m_function; };
  const TFCS2DFunction *function() const { return m_function; };
 
  virtual void Print(Option_t *option = "") const override;
 
protected:
  TFCS2DFunction *m_function;
  float m_nhits;
 
private:
  ClassDefOverride(TFCS2DFunctionLateralShapeParametrization,
                   2) // TFCS2DFunctionLateralShapeParametrization
};
 
#endif