GeneratorInterface/HiGenCommon/interface/HadronDecayGenEvtSelector.h - master - githubmirror/cmssw

GeneratorInterface/HiGenCommon/interface/HadronDecayGenEvtSelector.h

45 строк1 KB

Cms Build

Clang-Format

16 май 2019, 15:48

16 май 2019, 15:480847e19

#ifndef _HI_HadronDecayGenEvtSelector_h__
#define _HI_HadronDecayGenEvtSelector_h__

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

class HadronDecayGenEvtSelector : public BaseHiGenEvtSelector {
public:
  HadronDecayGenEvtSelector(const edm::ParameterSet& pset);
  ~HadronDecayGenEvtSelector() override { ; }

  bool filter(HepMC::GenEvent*) override;
  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> 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