ParentChildFilter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration 
*/
#ifndef GENERATORFILTERS_PARENTCHILDFILTER_H
#define GENERATORFILTERS_PARENTCHILDFILTER_H
 
#include "GeneratorModules/GenFilter.h"
 
class ParentChildFilter : public GenFilter {
public:
 
  ParentChildFilter(const std::string& name, ISvcLocator* pSvcLocator);
  virtual StatusCode filterInitialize();
  virtual StatusCode filterEvent();
 
private:
 
  std::vector<int> m_PDGParent;
  double m_PtMinParent;
  double m_PtMaxParent;
  double m_MassMinParent;
  double m_MassMaxParent;
  double m_EtaRangeParent;
  double m_EtaMinParent;
  double m_RapidityRangeParent;
  double m_RapidityMinParent;
  std::vector<int> m_PDGChild;
  double m_PtMinChild;
  double m_EtaRangeChild;
 
#ifdef HEPMC3
 
#else
 
  inline double getRapidity(const HepMC::GenParticle* p) const {
    double e = (p)->momentum().e();
    double pz = (p)->momentum().pz();
    if (e == pz) return 9999.;
    if (e == -pz) return -9999.;
    double rapidity = 0.5 * log((e + pz) / (e - pz));
    return rapidity;
  };
#endif
};
 
#endif