SoftLeptonTaggingTruth.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file SoftLeptonTaggingTruth.h

#ifndef __SoftLeptonTaggingTruth_H__
#define __SoftLeptonTaggingTruth_H__

#include <fun4all/SubsysReco.h>
#include <phool/PHCompositeNode.h>

#include <string>
#include <set>
#include <map>
#include <utility>      // std::pair, std::make_pair
#include <stdint.h>
#ifndef __CINT__
#include <memory>
#endif

#include <TString.h>

class PHCompositeNode;
class Fun4AllHistoManager;
class TH1F;
class JetEvalStack;
class JetTruthEval;
class Jet;
class TAxis;
class PHG4TruthInfoContainer;

class SoftLeptonTaggingTruth : public SubsysReco
{

public:

  enum enu_flags
  {
    kProcessTruthSpectrum = 1 << 1,

    kDefaultFlag = kProcessTruthSpectrum
  };

  SoftLeptonTaggingTruth(const std::string & truth_jet, enu_flags flags =
      kDefaultFlag);
  virtual
  ~SoftLeptonTaggingTruth();

  int
  add_reco_jet(const std::string & reco_jet)
  {
    _reco_jets.insert(reco_jet);
    return _reco_jets.size();
  }

  uint32_t
  get_flags() const
  {
    return _flags;
  }

  void
  set_flags(enu_flags flags)
  {
    _flags = (uint32_t) flags;
  }

  void
  set_flag(enu_flags flag)
  {
    _flags |= (uint32_t) flag;
  }

  bool
  flag(enu_flags flag)
  {
    return _flags & flag;
  }

  void
  reset_flag(enu_flags flag)
  {
    _flags &= ~(uint32_t) flag;
  }

  double
  get_jet_match_E_Ratio() const
  {
    return _jet_match_E_Ratio;
  }

  void
  set_jet_match_E_Ratio(double jetMatchDERatio)
  {
    _jet_match_E_Ratio = jetMatchDERatio;
  }

  double
  get_jet_match_dR() const
  {
    return _jet_match_dR;
  }

  void
  set_jet_match_dR(double jetMatchDEta)
  {
    _jet_match_dR = jetMatchDEta;
  }

  double
  get_jet_match_dca() const
  {
    return _jet_match_dca;
  }

  void
  set_jet_match_dca(double jetMatchDPhi)
  {
    _jet_match_dca = jetMatchDPhi;
  }

  void
  set_eta_range(double low, double high);

  int
  Init(PHCompositeNode *topNode);
  int
  InitRun(PHCompositeNode *topNode);
  int
  process_event(PHCompositeNode *topNode);
  int
  End(PHCompositeNode *topNode);

  static Fun4AllHistoManager *
  getHistoManager();

private:

  void
  useLogBins(TAxis * axis);

  int
  Init_Spectrum(PHCompositeNode *topNode, const std::string & jet_name);
  int
  process_Spectrum(PHCompositeNode *topNode, const std::string & jet_name,
      const bool is_reco_jet);

  std::string
  get_histo_prefix(const std::string & src_jet_name = "",
      const std::string & reco_jet_name = "");

#ifndef __CINT__

  typedef std::map<std::string, std::shared_ptr<JetEvalStack>> jetevalstacks_map;
  jetevalstacks_map _jetevalstacks;
  std::shared_ptr<JetTruthEval> _jettrutheval;
#endif

  std::string _truth_jet;

  std::set<std::string> _reco_jets;

  PHG4TruthInfoContainer* _truth_container;

  uint32_t _flags;

  std::pair<double, double> eta_range;

  TString
  get_eta_range_str(const char * eta_name = "#eta_{Jet}") const;

  bool
  jet_acceptance_cut(const Jet * jet) const;

  double _jet_match_dR;

  double _jet_match_dca;

  double _jet_match_E_Ratio;
};

#endif // __CALOEVALUATOR_H__