IsolationTools.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef ISOLATIONTOOLS_H_
#define ISOLATIONTOOLS_H_
 
#include "xAODBase/IParticle.h"
#include "xAODPrimitives/IsolationType.h"
 
namespace top {
  class IsolationBase {
  public:
    IsolationBase();
 
    virtual ~IsolationBase();
 
    virtual bool passSelection(const xAOD::IParticle& p) const = 0;
 
    virtual bool passSelectionLoose(const xAOD::IParticle& p) const = 0;
 
    virtual void print(std::ostream& os) const = 0;
  };
 
  class AbsoluteIsolationDC14: public IsolationBase {
  public:
    AbsoluteIsolationDC14(const xAOD::Iso::IsolationType type, double value);
    virtual bool passSelection(const xAOD::IParticle& p) const override;
    virtual bool passSelectionLoose(const xAOD::IParticle& p) const override;
    virtual void print(std::ostream& os) const override;
  protected:
    xAOD::Iso::IsolationType m_type;
 
    double m_cutvalue;
  };
 
  class RelativePTVarCone: public IsolationBase {
  public:
    RelativePTVarCone(unsigned int size, double fraction, unsigned int sizeLoose, double fractionLoose);
 
    virtual bool passSelection(const xAOD::IParticle& p) const override;
 
    virtual bool passSelectionLoose(const xAOD::IParticle& p) const override;
 
    virtual void print(std::ostream& os) const override;
  protected:
    std::string m_size;
 
    double m_fraction;
 
    std::string m_sizeLoose;
 
    double m_fractionLoose;
  };
 
  class ApproxPTVarCone: public IsolationBase {
  public:
    ApproxPTVarCone(double fraction, double fractionLoose);
 
    virtual bool passSelection(const xAOD::IParticle& p) const override;
 
    virtual bool passSelectionLoose(const xAOD::IParticle& p) const override;
 
    virtual void print(std::ostream& os) const override;
  protected:
    double m_fraction;
 
    double m_fractionLoose;
  };
 
  class AntiMuonIsolation: public IsolationBase {
  public:
    AntiMuonIsolation(const std::string& workingPoint = "AntiMuon_Nominal");
 
    virtual bool passSelection(const xAOD::IParticle& p) const override;
 
    virtual bool passSelectionLoose(const xAOD::IParticle& /*p*/) const override {return true;};
 
    virtual void print(std::ostream& os) const override;
  protected:
    std::string m_workingPoint;
  };
 
  class StandardIsolation: public IsolationBase {
  public:
    StandardIsolation(const std::string& tightLeptonIsolation,
                      const std::string& looseLeptonIsolation);
 
    virtual bool passSelection(const xAOD::IParticle& p) const override;
 
    virtual bool passSelectionLoose(const xAOD::IParticle& p) const override;
 
    std::string tightLeptonIsolation() {return m_tightLeptonIsolation;};
    std::string looseLeptonIsolation() {return m_looseLeptonIsolation;};
 
    virtual void print(std::ostream& os) const override;
  protected:
    std::string m_tightLeptonDecoration;
    std::string m_looseLeptonDecoration;
    std::string m_tightLeptonIsolation;
    std::string m_looseLeptonIsolation;
    bool m_doTightIsolation;
    bool m_doLooseIsolation;
    bool m_doTightPromptLeptonIso;
    bool m_doLoosePromptLeptonIso;
    bool m_skipUnavailable; // skip unavailable working points (instead of aborting)
  };
}
 
#endif