PartonHadronDecayGenEvtSelector.h

Go to the documentation of this file.

#ifndef _HI_PartonHadronDecayGenEvtSelector_h__
#define _HI_PartonHadronDecayGenEvtSelector_h__

#include <vector>
#include "GeneratorInterface/HiGenCommon/interface/BaseHiGenEvtSelector.h"

class PartonHadronDecayGenEvtSelector : public BaseHiGenEvtSelector 
{
 public:
  PartonHadronDecayGenEvtSelector(const edm::ParameterSet& pset);
  ~PartonHadronDecayGenEvtSelector() override{;}
  
  bool filter(HepMC::GenEvent *) override;
  bool selectParticle(HepMC::GenParticle* par, int status, int pdg /*Absolute*/, double ptMin, double etaMax){
    return (par->status() == status && abs(par->pdg_id()) == pdg && par->momentum().perp() > ptMin && fabs(par->momentum().eta()) < etaMax);
  }
  bool selectParticle(HepMC::GenParticle* par, int status, int pdg /*Absolute*/, double etaMax, double etaMin, double pMin, double ptMax, double ptMin){
    return (par->status() == status &&
        abs(par->pdg_id()) == pdg &&
        par->momentum().eta() < etaMax && par->momentum().eta() > etaMin &&
        par->momentum().rho() > pMin &&
        par->momentum().perp() < ptMax && par->momentum().perp() > ptMin);
  }

 private:

  std::vector<int>    partonId_;
  std::vector<int>    partonStatus_;
  std::vector<double> partonEtaMax_;
  std::vector<double> partonPtMin_;

  std::vector<int>    hadronId_;
  std::vector<int>    hadronStatus_;
  std::vector<double> hadronEtaMax_;
  std::vector<double> hadronEtaMin_;
  std::vector<double> hadronPMin_;
  std::vector<double> hadronPtMax_;
  std::vector<double> hadronPtMin_;

  int    decayId_;
  int    decayStatus_;
  double decayEtaMax_;
  double decayEtaMin_;
  double decayPMin_;
  double decayPtMax_;
  double decayPtMin_;
  int    decayNtrig_;

};

#endif