ReadEICFiles.h

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

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

#include <fun4all/SubsysReco.h>

#include <phhepmc/PHHepMCGenHelper.h>

#include <string>

class PHCompositeNode;
class TChain;

namespace erhic
{
class EventMC;
}

class ReadEICFiles : public SubsysReco, public PHHepMCGenHelper
{
 public:
  ReadEICFiles(const std::string &name = "EICReader");
  ~ReadEICFiles() override;

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

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

  void SetFirstEntry(int e) { entry = e; }
  void SetNodeName(const std::string &s) { _node_name = s; }

 private:
  void GetTree();

  int CreateNodeTree(PHCompositeNode *topNode);

  enum EvtGen
  {
    Milou = 1,
    DEMP = 2,
    Unknown = 100
  };

  std::string filename;

  TChain *Tin;

  int nEntries;

  int entry;

  int m_EvtGenId;

  erhic::EventMC *GenEvent;

  // output
  std::string _node_name;

};

#endif /* READEICFILES_H__ */