PHG4FullProjSpacalCellReco.h

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

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

#include <phparameter/PHParameterInterface.h>

#include <fun4all/SubsysReco.h>

#include <map>
#include <string>

class PHCompositeNode;
class PHG4Cell;
class TH2;
class TH1;

class PHG4FullProjSpacalCellReco : public SubsysReco,  public PHParameterInterface
{
 public:

  PHG4FullProjSpacalCellReco(const std::string &name = "HCALCELLRECO");

  ~PHG4FullProjSpacalCellReco() override{}
  
  int InitRun(PHCompositeNode *topNode) override;
  
  int process_event(PHCompositeNode *topNode) override;

  int ResetEvent(PHCompositeNode *topNode) override;

  void SetDefaultParameters() override;

  void Detector(const std::string &d) {detector = d;}

  void checkenergy(const int i=1) {chkenergyconservation = i;}

  void   set_timing_window(const double tmin, const double tmax);

  class LightCollectionModel
  {
  public:
    LightCollectionModel();
    virtual ~LightCollectionModel();

    void load_data_file(const std::string & input_file, const std::string & histogram_light_guide_model, const std::string & histogram_fiber_model);

    bool use_light_guide_model() const {return data_grid_light_guide_efficiency != nullptr;}

    bool use_fiber_model() const {return data_grid_fiber_trans != nullptr;}

    double get_light_guide_efficiency(const double x_fraction, const double y_fraction);

    double get_fiber_transmission(const double z_distance);

  private:
    TH2 * data_grid_light_guide_efficiency;

    TH1 * data_grid_fiber_trans;

    TH2 * data_grid_light_guide_efficiency_verify;
    TH1 * data_grid_fiber_trans_verify;

  };

  LightCollectionModel & get_light_collection_model() {return light_collection_model;}

 protected:

  int CheckEnergy(PHCompositeNode *topNode);

  std::string detector;
  std::string hitnodename;
  std::string cellnodename;
  std::string geonodename;
  std::string seggeonodename;

  double sum_energy_g4hit;
  int chkenergyconservation;
  std::map<unsigned int, PHG4Cell *> celllist;

  double tmin;
  double tmax;

  LightCollectionModel light_collection_model;

};

#endif