TFCSParametrizationChain.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ISF_FASTCALOSIMEVENT_TFCSParametrizationChain_h
#define ISF_FASTCALOSIMEVENT_TFCSParametrizationChain_h
 
#include "ISF_FastCaloSimEvent/TFCSParametrization.h"
 
class TFCSParametrizationChain : public TFCSParametrization {
public:
  TFCSParametrizationChain(const char *name = nullptr,
                           const char *title = nullptr)
      : TFCSParametrization(name, title){};
  TFCSParametrizationChain(const TFCSParametrizationChain &ref)
      : TFCSParametrization(ref.GetName(), ref.GetTitle()),
        m_chain(ref.chain()){};
 
  enum FCSSplitChainObjects {
    kSplitChainObjects =
        BIT(16), 
    kRetryChainFromStart = BIT(17)
  };
 
  bool SplitChainObjects() const { return TestBit(kSplitChainObjects); };
  void set_SplitChainObjects() { SetBit(kSplitChainObjects); };
  void reset_SplitChainObjects() { ResetBit(kSplitChainObjects); };
 
  bool RetryChainFromStart() const { return TestBit(kRetryChainFromStart); };
  void set_RetryChainFromStart() { SetBit(kRetryChainFromStart); };
  void reset_RetryChainFromStart() { ResetBit(kRetryChainFromStart); };
 
  typedef std::vector<TFCSParametrizationBase *> Chain_t;
  virtual unsigned int size() const override { return m_chain.size(); };
  virtual const TFCSParametrizationBase *
  operator[](unsigned int ind) const override {
    return m_chain[ind];
  };
  virtual TFCSParametrizationBase *operator[](unsigned int ind) override {
    return m_chain[ind];
  };
  virtual void set_daughter(unsigned int ind,
                            TFCSParametrizationBase *param) override {
    m_chain.at(ind) = param;
  };
  const Chain_t &chain() const { return m_chain; };
  Chain_t &chain() { return m_chain; };
  void push_back(const Chain_t::value_type &param) {
    m_chain.push_back(param);
    recalc();
  };
 
  virtual bool is_match_Ekin_bin(int Ekin_bin) const override;
  virtual bool is_match_calosample(int calosample) const override;
 
  virtual FCSReturnCode
  simulate(TFCSSimulationState &simulstate, const TFCSTruthState *truth,
           const TFCSExtrapolationState *extrapol) const override;
 
  void Print(Option_t *option = "") const override;
 
  static void unit_test(TFCSSimulationState *simulstate = nullptr,
                        const TFCSTruthState *truth = nullptr,
                        const TFCSExtrapolationState *extrapol = nullptr);
 
  // THIS CLASS HAS A CUSTOM STREAMER! CHANGES IN THE VERSIONING OR DATA TYPES
  // NEED TO BE IMPLEMENTED BY HAND! void
  // TFCSParametrizationChain::Streamer(TBuffer &R__b)
protected:
  void recalc_pdgid_intersect();
  void recalc_pdgid_union();
 
  void recalc_Ekin_intersect();
  void recalc_eta_intersect();
  void recalc_Ekin_eta_intersect();
 
  void recalc_Ekin_union();
  void recalc_eta_union();
  void recalc_Ekin_eta_union();
 
  virtual void recalc();
 
  FCSReturnCode
  simulate_and_retry(TFCSParametrizationBase *parametrization,
                     TFCSSimulationState &simulstate,
                     const TFCSTruthState *truth,
                     const TFCSExtrapolationState *extrapol) const;
 
private:
  Chain_t m_chain;
 
  // Ensure all objects to be written by the streamer live long enough.
  std::vector<std::unique_ptr<TFCSParametrizationBase>> m_writtenBases;
 
  ClassDefOverride(TFCSParametrizationChain, 2) // TFCSParametrizationChain
};
 
#include "ISF_FastCaloSimEvent/TFCSParametrizationChain.icc"
 
#endif