PHG4ParticleGenerator_flat_pT.h

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

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

#include <g4main/PHG4ParticleGeneratorBase.h>

#include <cmath>
#include <string>  // for string

class PHCompositeNode;

class PHG4ParticleGenerator_flat_pT : public PHG4ParticleGeneratorBase
{
  public:
    PHG4ParticleGenerator_flat_pT(const std::string &name = "PGENERATOR");
    virtual ~PHG4ParticleGenerator_flat_pT() {}

    int process_event(PHCompositeNode *topNode);
    void set_z_range(const double z_min, const double z_max);
    void set_eta_range(const double eta_min, const double eta_max);
    void set_phi_range(const double phi_min, const double phi_max);
    void set_pT_range(const double mom_min, const double mom_max);
    void Print(const std::string &what = "ALL") const;

  protected:
    double m_ZMin = -10.;
    double m_ZMax = 10.;
    double m_EtaMin = -1.;
    double m_EtaMax = 1.;
    double m_PhiMin = -M_PI;
    double m_PhiMax = M_PI;
    double m_MomMin = 0.;
    double m_MomMax = 10.;
};

#endif