HFFastSim.h

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

#ifndef __HFFastSim_H__
#define __HFFastSim_H__

// --- need to check all these includes...
#include <fun4all/SubsysReco.h>

#include <TObject.h>
#include <TString.h>

#include <limits.h>
#include <cmath>
#include <string>
#include <vector>
#include <utility>

class TTree;
class TFile;
class TH2D;
class TH2F;
class TH1D;

class PHCompositeNode;
class Jet;
namespace HepMC
{
class GenParticle;
class GenEvent;
}  // namespace HepMC

class HFFastSim : public SubsysReco
{
 public:
  HFFastSim(std::string filename, int flavor = 5, int maxevent = INT_MAX);

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

  double
  get_eta_max() const
  {
    return _eta_max;
  }

  void
  set_eta_max(double etaMax)
  {
    _eta_max = etaMax;
  }

  double
  get_eta_min() const
  {
    return _eta_min;
  }

  void
  set_eta_min(double etaMin)
  {
    _eta_min = etaMin;
  }

  double
  get_pt_max() const
  {
    return _pt_max;
  }

  void
  set_pt_max(double ptMax)
  {
    _pt_max = ptMax;
  }

  double
  get_pt_min() const
  {
    return _pt_min;
  }

  void
  set_pt_min(double ptMin)
  {
    _pt_min = ptMin;
  }

  void
  set_rejection_action(int action)
  {
    _rejection_action = action;
  }

  int get_embedding_id() const { return _embedding_id; }
  //
  void set_embedding_id(int id) { _embedding_id = id; }

  static std::pair<int, TString> quark_trace(const HepMC::GenParticle *p, HepMC::GenEvent *theEvent);

  bool process_D02PiK(HepMC::GenEvent *theEvent);

  class D02PiK : public TObject
  {
   public:
    D02PiK() { Clear(); }

    int pid;
    int hadron_origion_qpid;
    float y;
    float pt;
    float prodL;
    float decayL;
    TString decayChain;

    void Clear(Option_t * /*option*/ = "")
    {
      pid = 0;
      hadron_origion_qpid = 0;
      y = NAN;
      pt = NAN;
      prodL = NAN;
      decayL = NAN;
      decayChain = "";
    }

    ClassDef(HFFastSim::D02PiK, 1)  //Event structure
  };

  bool process_Lc2pPiK(HepMC::GenEvent *theEvent);

  class Lc2pPiK : public TObject
  {
   public:
    Lc2pPiK() { Clear(); }

    int pid;
    int hadron_origion_qpid;
    float y;
    float pt;
    float prodL;
    float decayL;
    TString decayChain;

    void Clear(Option_t * /*option*/ = "")
    {
      pid = 0;
      hadron_origion_qpid = 0;
      y = NAN;
      pt = NAN;
      prodL = NAN;
      decayL = NAN;
      decayChain = "";
    }

    ClassDef(HFFastSim::Lc2pPiK, 1)  //Event structure
  };

 private:
  bool _verbose;

  int _ievent;
  int _total_pass;

  TFile *_f;

  TH2D *_h2;
  TH2D *_h2all;
  TH2D *_h2_b;
  TH2D *_h2_c;

  TH1D *m_hNorm;
  TH2F *m_DRapidity;

  TTree *m_tSingleD;
  D02PiK *m_singleD;

  TTree *m_tSingleLc;
  Lc2pPiK *m_singleLc;

  std::string _foutname;

  int _flavor;
  int _maxevent;

  double _pt_min;
  double _pt_max;

  double _eta_min;
  double _eta_max;

  std::string _jet_name;

  int _rejection_action;

  int _embedding_id;
};

#endif  // __HFFastSim_H__