TruthJetInput.h

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

#ifndef G4JET_TRUTHJETINPUT_H
#define G4JET_TRUTHJETINPUT_H

#include <jetbase/JetInput.h>
#include <jetbase/Jet.h>

#include <iostream>  // for cout, ostream
#include <vector>

class PHCompositeNode;

class TruthJetInput : public JetInput
{
 public:
  TruthJetInput(Jet::SRC input);
  ~TruthJetInput() override {}

  void add_embedding_flag(const int embed_stream_id)
  {
    m_EmbedID.push_back(embed_stream_id);
  }

  void identify(std::ostream& os = std::cout) override;

  Jet::SRC get_src() override { return m_Input; }

  std::vector<Jet*> get_input(PHCompositeNode* topNode) override;

  void set_eta_range(float eta_min, float eta_max)
  {
    m_EtaMin = eta_min;
    m_EtaMax = eta_max;
  }

 private:
  Jet::SRC m_Input = Jet::VOID;
  float m_EtaMin = -4.;
  float m_EtaMax = 4.;

  std::vector<int> m_EmbedID;

  bool use_embed_stream() { return m_EmbedID.size() > 0; }
};

#endif