JetHepMCLoader.h

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

// $Id: $

#ifndef G4JET_JETHEPMCLOADER_H
#define G4JET_JETHEPMCLOADER_H

#include <jetbase/Jet.h>

#include <fun4all/SubsysReco.h>
#include <string>
#include <vector>

class Fun4AllHistoManager;
class PHCompositeNode;

class JetHepMCLoader : public SubsysReco
{
 public:
  JetHepMCLoader(const std::string &jetInputCategory);
  ~JetHepMCLoader() override {}

  int InitRun(PHCompositeNode *topNode) override;
  int process_event(PHCompositeNode *topNode) override;
  int End(PHCompositeNode *topNode) override;

  void addJet(
      const std::string &name,
      int embeddingID,
      Jet::ALGO algorithm,
      double parameter,
      int tagPID,
      int tagStatus);

  void saveQAPlots(bool b = true) { m_saveQAPlots = b; }

 private:
  int CreateNodes(PHCompositeNode *topNode);
  Fun4AllHistoManager *getHistoManager();

  std::string m_jetInputCategory;

  bool m_saveQAPlots = false;

  struct hepmc_jet_src
  {
    std::string m_name;

    int m_embeddingID;

    std::string m_algorithmName;

    Jet::ALGO m_algorithmID;

    double m_parameter;

    int m_tagPID;

    int m_tagStatus;
  };

  std::vector<hepmc_jet_src> m_jetSrc;
};

#endif /* JETHEPMCLOADER_H_ */