PHSartreParticleTrigger.h

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

#ifndef PHSARTRE_PHSARTREPARTICLETRIGGER_H
#define PHSARTRE_PHSARTREPARTICLETRIGGER_H

#include "PHSartreGenTrigger.h"

#include <string>
#include <vector>                // for vector

class Event;

class PHSartreParticleTrigger : public PHSartreGenTrigger
{
 public:
  PHSartreParticleTrigger(const std::string &name = "PHSartreParticleTrigger");
  virtual ~PHSartreParticleTrigger();

  bool Apply(Event *event);

  void AddParticles(const std::string &particles);
  void AddParticles(int particle);
  void AddParticles(std::vector<int> particles);

  void SetPtHigh(double pt);
  void SetPtLow(double pt);
  void SetPtHighLow(double ptHigh, double ptLow);

  void SetPHigh(double p);
  void SetPLow(double p);
  void SetPHighLow(double pHigh, double pLow);

  void SetEtaHigh(double eta);
  void SetEtaLow(double eta);
  void SetEtaHighLow(double etaHigh, double etaLow);

  void SetAbsEtaHigh(double eta);
  void SetAbsEtaLow(double eta);
  void SetAbsEtaHighLow(double etaHigh, double etaLow);

  void SetPzHigh(double pz);
  void SetPzLow(double pz);
  void SetPzHighLow(double pzHigh, double pzLow);

  void PrintConfig();

 private:
  std::vector<int> _theParticles;

  double _theEtaHigh, _theEtaLow;
  double _thePtHigh, _thePtLow;
  double _thePHigh, _thePLow;
  double _thePzHigh, _thePzLow;

  bool _doEtaHighCut, _doEtaLowCut, _doBothEtaCut;
  bool _doAbsEtaHighCut, _doAbsEtaLowCut, _doBothAbsEtaCut;
  bool _doPtHighCut, _doPtLowCut, _doBothPtCut;
  bool _doPHighCut, _doPLowCut, _doBothPCut;
  bool _doPzHighCut, _doPzLowCut, _doBothPzCut;
};

#endif