Lepton.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
// contact: jmaurer@cern.ch
 
#ifndef TRIGGLOBALEFFICIENCYCORRECTION_LEPTON_H
#define TRIGGLOBALEFFICIENCYCORRECTION_LEPTON_H 1
 
#include "xAODEgamma/Electron.h"
#include "xAODEgamma/Photon.h"
#include "xAODMuon/Muon.h"
 
namespace TrigGlobEffCorr
{
 
class Lepton
{
public:
    Lepton(const xAOD::IParticle* ptr, std::size_t tag=0) : m_obj(ptr), m_tag(tag) {}
    Lepton(const xAOD::Electron* ptr, std::size_t tag=0) : m_obj(ptr), m_tag(tag) {}
    Lepton(const xAOD::Photon* ptr, std::size_t tag=0) : m_obj(ptr), m_tag(tag) {}
    Lepton(const Lepton&) = default;
    Lepton& operator=(const Lepton&) = default;
    Lepton(Lepton&&) = default;
    Lepton& operator=(Lepton&&) = default;
    float pt() const { return m_obj->pt(); }
    xAOD::Type::ObjectType type() const { return m_obj->type(); }
    std::size_t tag() const { return m_tag; }
    const xAOD::Electron* electron() const { return static_cast<const xAOD::Electron*>(m_obj); }
    const xAOD::Muon* muon() const { return static_cast<const xAOD::Muon*>(m_obj); }
    const xAOD::Photon* photon() const { return static_cast<const xAOD::Photon*>(m_obj); }
    const xAOD::IParticle* particle() const { return m_obj; }
    bool operator<(const Lepton& rhs) const { return m_obj<rhs.m_obj; }
protected:
    const xAOD::IParticle* m_obj = nullptr;
    std::size_t m_tag = 0; // 0 = no tag
};
 
}
 
 
#endif