HcalRawTowerBuilder.h

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

#ifndef G4CALO_HCALRAWTOWERBUILDER_H
#define G4CALO_HCALRAWTOWERBUILDER_H

#include <fun4all/SubsysReco.h>

#include <phparameter/PHParameterInterface.h>

#include <cmath>
#include <map>  // for map
#include <string>
#include <utility>  // for pair
#include <vector>

class PHCompositeNode;
class RawTowerContainer;
class RawTowerGeomContainer;

class HcalRawTowerBuilder : public SubsysReco, public PHParameterInterface
{
 public:
  HcalRawTowerBuilder(const std::string &name = "HcalRawTowerBuilder");
  ~HcalRawTowerBuilder() override {}

  int InitRun(PHCompositeNode *topNode) override;
  int process_event(PHCompositeNode *topNode) override;
  void Detector(const std::string &d)
  {
    m_InputDetector = d;
    m_OutputDetector = d;
  }
  void InDetector(const std::string &d) { m_InputDetector = d; }
  void OutDetector(const std::string &d) { m_OutputDetector = d; }
  void EminCut(const double e) { m_Emin = e; }
  void checkenergy(const int i = 1) { m_ChkEnergyConservationFlag = i; }

  enum enu_tower_energy_src
  {
    kEnergyDeposition,

    kLightYield,

    kIonizationEnergy,

    kRawLightYield,

    unknown = -1
  };
  enum ProcessTowerType
  {
    kRawTowerOnly= 0,
    kTowerInfoOnly = 1,
    kBothTowers =2
  };


  int get_tower_energy_src() const
  {
    return m_TowerEnergySrc;
  }

  std::string
  get_sim_tower_node_prefix() const
  {
    return m_SimTowerNodePrefix;
  }

  void
  set_sim_tower_node_prefix(const std::string &simTowerNodePrefix)
  {
    m_SimTowerNodePrefix = simTowerNodePrefix;
  }

  short get_tower_row(const short cellrow) const;

  void set_decal_filename(const std::string &fname) { m_DeCalibrationFileName = fname; }

  void SetDefaultParameters() override;

  void set_cell_decal_factor(const int etabin, const int phibin, const double d);
  void set_tower_decal_factor(const int etabin, const int phibin, const double d);
  void Print(const std::string &what = "ALL") const override;

  void set_towerinfo(HcalRawTowerBuilder::ProcessTowerType UseTowerInfo )
  {
    m_UseTowerInfo = UseTowerInfo;
  }

 private:
  void CreateNodes(PHCompositeNode *topNode);
  void ReadParamsFromNodeTree(PHCompositeNode *topNode);
  void SetTowerDecalFactors();
  void set_tower_decal_factor_real(const int etabin, const int phibin, const double d);

  RawTowerContainer *m_Towers = nullptr;
  RawTowerGeomContainer *m_RawTowerGeom = nullptr;

  double m_Emin = NAN;
  int m_ChkEnergyConservationFlag = 0;
  int m_TowerEnergySrc = enu_tower_energy_src::unknown;
  int m_NcellToTower = -1;
  HcalRawTowerBuilder::ProcessTowerType m_UseTowerInfo = HcalRawTowerBuilder::ProcessTowerType::kBothTowers;  // 0 just produce RawTowers, 1 just produce TowerInfo objects, and 2 produce both



  std::string m_OutputDetector;
  std::string m_InputDetector;
  std::string m_TowerNodeName;
  std::string m_TowerInfoNodeName;
  std::string m_TowerGeomNodeName;
  std::string m_SimTowerNodePrefix;
  std::string m_DeCalibrationFileName;
  std::vector<std::vector<double> > m_DecalArray;
  std::map<std::pair<int, int>, double> m_TowerDecalFactors;
};

#endif /* G4CALO_HCALRAWTOWERBUILDER_H */