TFCSEnergyBinParametrization.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ISF_FASTCALOSIMEVENT_TFCSEnergyBinParametrization_h
#define ISF_FASTCALOSIMEVENT_TFCSEnergyBinParametrization_h
 
#include "ISF_FastCaloSimEvent/TFCSEnergyParametrization.h"
#include "TFile.h"
#include <map>
#include <vector>
 
class TFCSEnergyBinParametrization : public TFCSEnergyParametrization {
public:
  TFCSEnergyBinParametrization(const char *name = nullptr,
                               const char *title = nullptr);
 
  virtual void set_pdgid(int id) override;
  virtual void set_pdgid(const std::set<int> &ids) override;
  virtual void add_pdgid(int id) override;
  virtual void clear_pdgid() override;
 
  virtual int n_bins() const override { return m_number_of_Ekin_bins; };

  void set_number_of_Ekin_bins(int n_Ekin_bin) {
    m_number_of_Ekin_bins = n_Ekin_bin;
    resize();
  };

  virtual void set_pdgid_Ekin_bin_probability(int id, std::vector<float> prob);
  virtual bool
  load_pdgid_Ekin_bin_probability_from_file(int id, TFile *file,
                                            std::string prob_object_name);
 
  virtual FCSReturnCode
  simulate(TFCSSimulationState &simulstate, const TFCSTruthState *truth,
           const TFCSExtrapolationState *extrapol) const override;
 
  virtual bool is_match_Ekin_bin(int Ekin_bin) const override;
 
  void Print(Option_t *option = "") const override;
 
private:
  int m_number_of_Ekin_bins;
  std::map<int, std::vector<float>> m_pdgid_Ebin_probability;
 
  void resize();
 
  ClassDefOverride(TFCSEnergyBinParametrization,
                   1) // TFCSEnergyBinParametrization
};
 
#endif