ReadSynRadFiles.h

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

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

#include <fun4all/SubsysReco.h>

#include <phhepmc/PHHepMCGenHelper.h>

#include <fstream>
#include <string>

class PHCompositeNode;
class TChain;

namespace erhic
{
  class EventMC;
}

class ReadSynRadFiles : public SubsysReco
{
 public:
  ReadSynRadFiles(const std::string &name = "ReadSynRadFiles");
  virtual ~ReadSynRadFiles();

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

  bool OpenInputFile(const std::string &name);

  void SetEntryPerEvent(int e) { nEntries = e; }
  void SetNodeName(const std::string &s) { _node_name = s; }
  void set_vertex_distribution_function(PHHepMCGenHelper::VTXFUNC x, PHHepMCGenHelper::VTXFUNC y, PHHepMCGenHelper::VTXFUNC z, PHHepMCGenHelper::VTXFUNC t)
  {
    hepmc_helper.set_vertex_distribution_function(x, y, z, t);
  }

  void set_vertex_distribution_mean(const double x, const double y, const double z, const double t)
  {
    hepmc_helper.set_vertex_distribution_mean(x, y, z, t);
  }

  void set_vertex_distribution_width(const double x, const double y, const double z, const double t)
  {
    hepmc_helper.set_vertex_distribution_width(x, y, z, t);
  }
  //
  void set_reuse_vertex(int src_embedding_id)
  {
    hepmc_helper.set_reuse_vertex(src_embedding_id);
  }

  int get_embedding_id() const { return hepmc_helper.get_embedding_id(); }
  //
  void set_embedding_id(int id) { hepmc_helper.set_embedding_id(id); }

  void set_reverseXZ(bool b = true) { m_reverseXZ = b; }

 protected:
  void GetTree();

  int CreateNodeTree(PHCompositeNode *topNode);

  std::string filename;

  std::ifstream m_csv_input;

  int nEntries;

  int entry;

  erhic::EventMC *GenEvent;

  // output
  std::string _node_name;

  PHHepMCGenHelper hepmc_helper;

  bool m_reverseXZ = false;
};

#endif /* ReadSynRadFiles_H__ */