SusyNtuple/OverlapTools.h

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

#include "SusyNtuple/SusyDefs.h"
#include "SusyNtuple/SusyNt.h"
#include "SusyNtuple/AnalysisType.h"
#include "SusyNtuple/Isolation.h"

namespace Susy {


class OverlapTools
{
public :

    static OverlapTools* build(const AnalysisType &a, bool verbose);
    OverlapTools(); 
    virtual ~OverlapTools(){};  
    // main overlap removal function, performs all removals
    virtual void performOverlap(ElectronVector& electrons, MuonVector& muons,
                                TauVector& taus, JetVector& jets);
    virtual void j_e_overlap(ElectronVector& electrons, JetVector& jets, double dR); 
    virtual void e_j_overlap(ElectronVector& electrons, JetVector& jets, double dR); 
    virtual void m_j_overlap(MuonVector& muons, JetVector& jets, double dR); 
    virtual void e_m_overlap(ElectronVector& electrons, MuonVector& muons, double dR); 
    virtual void e_e_overlap(ElectronVector& electrons, double dR); 
    virtual void m_m_overlap(MuonVector& muons, double dR); 
    virtual void t_e_overlap(TauVector& taus, ElectronVector& electrons, double dR); 
    virtual void t_m_overlap(TauVector& taus, MuonVector& muons, double dR); 
    virtual void j_t_overlap(TauVector& taus, JetVector& jets, double dR); 


    virtual void removeNonisolatedLeptons(ElectronVector& electrons, MuonVector& muons);
    OverlapTools& setElectronIsolation( Isolation eleIso ) { m_electronIsolation = eleIso; return *this; }
    OverlapTools& setMuonIsolation( Isolation muIso ) { m_muonIsolation = muIso; return *this; }
    bool leptonPassesIsolation(const Lepton* lep, const Isolation &iso);

protected :
    Isolation m_electronIsolation;
    Isolation m_muonIsolation;
    bool m_verbose;
}; // class OverlapTools

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

class OverlapTools_2Lep : public OverlapTools
{
};
class OverlapTools_3Lep : public OverlapTools
{
};

class OverlapTools_2LepWH : public OverlapTools
{
};

class OverlapTools_SS3L: public OverlapTools {
    virtual void j_e_overlap(ElectronVector& electrons, JetVector& jets, double dR);
};

class OverlapTools_Stop2L : public OverlapTools
{
};



}







#endif