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 "JetInput.h"

#include "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)
  {
    _embed_id.push_back(embed_stream_id);
  }

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

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

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

  void set_eta_range(float eta_min, float eta_max)
  {
    _eta_min = eta_min;
    _eta_max = eta_max;
  }

 private:
  Jet::SRC _input;
  float _eta_min;
  float _eta_max;

  std::vector<int> _embed_id;

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

#endif