LiteCaloEval.h

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

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

#include <fun4all/SubsysReco.h>

#include <string>

class PHCompositeNode;
class TFile;
class TH1;
class TH2;
class TH3;
class TGraph;
class TNtuple;
class TF1;

double LCE_fitf(double *f, double *p);
TGraph *LCE_grff = nullptr;

class LiteCaloEval : public SubsysReco
{
 public:


  int m_myminbin = -1;
  int m_mymaxbin = -3;

  enum Calo
  {
    NONE = 0,
    CEMC = 1,
    HCALIN = 2,
    HCALOUT = 3
  };

  LiteCaloEval(const std::string &name = "LiteCaloEval", const std::string &caloNm = "CEMC", const std::string &fnm = "outJF");

  // to distinguish when we want to implement input decal
  void set_mode(int modeset)
  {
    mode = modeset;
  }

  void set_UseTowerInfo(int setTowerInfo)
  {
    m_UseTowerInfo = setTowerInfo;
  }

  virtual ~LiteCaloEval() {}

  int InitRun(PHCompositeNode *topNode) override;

  int process_event(PHCompositeNode *topNode) override;

  int End(PHCompositeNode *topNode) override;

  void CaloType(const Calo i)
  {
    calotype = i;
  }

  // TNtuple -> to store fit parameters

  TFile *f_temp = nullptr;
  /*
  TNtuple *nt_corrVals;
  TF1 *fit_func;
  TF1 *fit_result;
  float fit_value_mean;
  float corr_val;
  */

  //  TF1 *mygaus;
  void Get_Histos(const char *infile, const char *fun4all_file = "");
  // void Fit_Histos();
  void FitRelativeShifts(LiteCaloEval *ref_lce, int modeFitShifts);

  void setFitMax(float fitMax);
  void setFitMin(float fitMin);

  float getFitMax();
  float getFitMin();


  void setInputTowerNodeName(const char * inpNodenm)
  {
    _inputnodename = inpNodenm;
  }


 private:
  TFile *cal_output = nullptr;

  TH1 *hcal_out_eta_phi[24][64] = {};
  TH1 *hcalout_eta[25] = {};
  TH2 *hcalout_energy_eta = nullptr;
  TH3 *hcalout_e_eta_phi = {};

  TH1 *hcal_in_eta_phi[24][64] = {};
  TH1 *hcalin_eta[25] = {};
  TH2 *hcalin_energy_eta = nullptr;
  TH3 *hcalin_e_eta_phi = nullptr;

  TH1 *cemc_hist_eta_phi[96][258] = {};
  TH1 *eta_hist[97] = {};
  TH2 *energy_eta_hist = nullptr;
  TH3 *e_eta_phi = nullptr;

  //  TH2 *evtCentrality = nullptr;

  Calo calotype = NONE;
  int _ievent = 0;
  float fitmin = 0.;
  float fitmax = 0.;
  std::string _caloname;
  std::string _filename;
  std::string _inputnodename;


  int mode = 0;

  // flag for using tower info
  int m_UseTowerInfo = 0;
};

#endif  // LITECALOEVAL_H