SusyNtuple/MuonSelector.h

// -*- c++ -*-
#ifndef SUSYNTUPLE_MUONSELECTOR_H
#define SUSYNTUPLE_MUONSELECTOR_H

#include "SusyNtuple/SusyNtSys.h"
#include "SusyNtuple/AnalysisType.h"
#include "SusyNtuple/Isolation.h"
#include "SusyNtuple/MuonId.h"


namespace Susy {

class Muon;


class MuonSelector
{
public :

    static MuonSelector* build(const AnalysisType &a, bool verbose);
    MuonSelector();  
    virtual ~MuonSelector() {}; 
    virtual bool isBaseline(const Muon* mu); 


    virtual bool isSignal(const Muon* mu);
    virtual bool passIpCut(const Muon* mu);
    virtual float effSF(const Muon& mu);
    virtual float effSF(const Muon* mu) { return effSF(*mu); }
    virtual float errEffSF(const Muon& mu, const NtSys::SusyNtSys sys);
    virtual float errEffSF(const Muon* mu,
                           const NtSys::SusyNtSys sys)
        { return errEffSF(*mu, sys); }
    Isolation signalIsolation() const { return m_signalIsolation; }

    MuonSelector& setSignalIsolation(const Isolation &v) { m_signalIsolation = v; return *this; }
    MuonId signalId() const { return m_signalId; }
    MuonSelector& setSignalId(const MuonId &v) { m_signalId = v; return *this; }
    MuonSelector& setVerbose(const bool &v) { m_verbose = v; return *this; }
    bool verbose() const { return m_verbose; }
protected :
    MuonId m_signalId; 
    Isolation m_signalIsolation; 
    bool m_verbose;

}; // end MuonSelector

//----------------------------------------------------------
//
// End generic selector, begin analysis-specific ones
//
//----------------------------------------------------------

class MuonSelector_2Lep : public MuonSelector
{
};

class MuonSelector_3Lep : public MuonSelector
{
    virtual bool isSignal(const Muon* mu);
};

class MuonSelector_2LepWH : public MuonSelector
{
    virtual bool isSignal(const Muon* mu);
};

class MuonSelector_SS3L : public MuonSelector
{
    virtual bool passIpCut(const Muon* mu);
    virtual bool isBaseline(const Muon* mu);
    virtual bool isSignal(const Muon* mu);
};

class MuonSelector_Stop2L : public MuonSelector
{
    virtual bool passIpCut(const Muon* mu);
    virtual bool isBaseline(const Muon* mu);
    virtual bool isSignal(const Muon* mu);
};

} // end namespace

#endif