QAG4SimulationTruthDecay.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef GEANTTESTER_H
#define GEANTTESTER_H

#include <fun4all/SubsysReco.h>

#include <decayfinder/DecayFinderContainer_v1.h>  // for DecayFinderContainer_v1

#include <g4main/PHG4Particle.h>
#include <g4main/PHG4TruthInfoContainer.h>
#include <g4main/PHG4VtxPoint.h>

#include <TBranch.h>
#include <TFile.h>
#include <TTree.h>

#include <string>

class PHCompositeNode;
class PHG4TruthInfoContainer;
class PHG4Particle;
class PHG4VtxPoint;

class QAG4SimulationTruthDecay : public SubsysReco
{
 public:
  QAG4SimulationTruthDecay(const std::string &name = "QAG4SimulationTruthDecay");

  virtual ~QAG4SimulationTruthDecay();

  int Init(PHCompositeNode *topNode) override;

  int process_event(PHCompositeNode *topNode) override;

  std::string get_histo_prefix();

  int End(PHCompositeNode *topNode) override;

  void setMotherPDG(int PDGID) { m_decay_pdg_id = PDGID; }

  void setMotherPDGRange(int min, int max)
  {
    m_mother_PDG_ID_min = min;
    m_mother_PDG_ID_max = max;
  };
  void setDaughterPDGRange(int min, int max)
  {
    m_daughter_PDG_ID_min = min;
    m_daughter_PDG_ID_max = max;
  };

  void setMinPT(float value) { m_pt_min = value; }

  void setEtaRange(float min, float max)
  {
    m_eta_min = min;
    m_eta_max = max;
  }

  void setNBinsMass(int nBins) { m_mass_nBins = nBins; }
  void setMassRange(float min, float max)
  {
    m_mass_min = min;
    m_mass_max = max;
  }

  void setNBinsDecayLength(int nBins) { m_decayLength_nBins = nBins; }
  void setDecayLengthRange(float min, float max)
  {
    m_decayLength_min = min;
    m_decayLength_max = max;
  }

  void setNBinsDecayTime(int nBins) { m_decayTime_nBins = nBins; }
  void setDecayTimeRange(float min, float max)
  {
    m_decayTime_min = min;
    m_decayTime_max = max;
  }

  void setDFNodeName(const std::string &name) { m_df_module_name = name; }
  void setOutputName(const std::string &name) { m_outfile_name = name; }
  void writeTuple(bool write) { m_write_nTuple = write; }

 private:
  typedef std::vector<std::pair<std::pair<int, int>, int>> Decay;

  unsigned int m_nTracks;
  float m_pt_min;
  float m_eta_min;
  float m_eta_max;
  int m_decay_pdg_id;

  PHG4TruthInfoContainer *m_truth_info;
  PHG4Particle *m_g4particle;
  std::string m_df_module_name;
  std::string m_outfile_name;
  TFile *m_outfile;
  TTree *m_tree;
  bool m_write_nTuple;
  bool m_write_QAHists;
  DecayFinderContainer_v1 *m_decayMap = nullptr;

  void initializeBranches();
  void getMotherPDG(PHCompositeNode *topNode);
  std::vector<int> getDecayFinderMothers(PHCompositeNode *topNode);
  bool isInRange(float min, float value, float max);
  void resetValues();

  int m_mother_PDG_ID_min = -421;
  int m_mother_PDG_ID_max = 421;
  int m_daughter_PDG_ID_min = -321;
  int m_daughter_PDG_ID_max = 321;

  int m_mass_nBins = 100;
  float m_mass_min = 0;
  float m_mass_max = 2;

  int m_decayLength_nBins = 100;
  float m_decayLength_min = 0;
  float m_decayLength_max = 1;

  int m_decayTime_nBins = 100;
  float m_decayTime_min = 0;
  float m_decayTime_max = 0.1;

  static const int max_tracks = 20;
  unsigned int m_event_number = 0;
  float m_mother_mass = -99;
  float m_daughter_sum_mass = 0;
  float m_mother_decayLength = -99;
  float m_mother_decayTime = -99;
  int m_mother_pdg_id = -99;
  float m_mother_px = 0;
  float m_mother_py = 0;
  float m_mother_pz = 0;
  float m_mother_pE = 0;
  float m_mother_pT = 0;
  float m_mother_eta = 0;
  int m_mother_barcode = -99;
  int m_track_pdg_id[max_tracks] = {0};
  float m_track_px[max_tracks] = {0};
  float m_track_py[max_tracks] = {0};
  float m_track_pz[max_tracks] = {0};
  float m_track_pE[max_tracks] = {0};
  float m_track_pT[max_tracks] = {0};
  float m_track_eta[max_tracks] = {0};
  float m_track_mass[max_tracks] = {0};
  int m_track_mother_barcode[max_tracks] = {0};
  float m_delta_px = 0;
  float m_delta_py = 0;
  float m_delta_pz = 0;
  float m_delta_pE = 0;
  bool m_accept_px_1percent = false;
  bool m_accept_py_1percent = false;
  bool m_accept_pz_1percent = false;
  bool m_accept_pE_1percent = false;
  bool m_accept_px_5percent = false;
  bool m_accept_py_5percent = false;
  bool m_accept_pz_5percent = false;
  bool m_accept_pE_5percent = false;
  bool m_accept_px_15percent = false;
  bool m_accept_py_15percent = false;
  bool m_accept_pz_15percent = false;
  bool m_accept_pE_15percent = false;
  bool m_accept_eta = true;
  bool m_accept_pT = true;
};

#endif  // GEANTTESTER_H