SusyNtuple/Muon.h

// Dear emacs, this is -*- c++ -*-
#ifndef SUSYNTUPLE_MUON_H
#define SUSYNTUPLE_MUON_H

#include "SusyNtuple/Lepton.h"
#include "SusyNtuple/SusyNtSys.h"
#include "SusyNtuple/MuonId.h"

#include <vector>

namespace Susy
{
class Muon : public Lepton
{
public:
    Muon(){ clear(); }
    virtual ~Muon(){};
    Muon(const Muon &);
    Muon& operator=(const Muon &);

    bool isCombined;          

    float idTrackPt;          
    float idTrackEta;         
    float idTrackPhi;         
    int   idTrackQ;           
    float idTrackTheta;       
    float idTrackQoverP;      

    float msTrackPt;          
    float msTrackEta;         
    float msTrackPhi;         
    int   msTrackQ;           
    float msTrackTheta;       
    float msTrackQoverP;      

    // Muon quality flags
    bool veryLoose;           
    bool loose;               
    bool medium;              
    bool tight;               

    // efficiency SF per muon quality WP
    std::vector<float> muoEffSF;

    bool isBadMuon;           
    bool isCosmic;            

    // Systematic sf
    float ms_up;              
    float ms_dn;              
    float id_up;              
    float id_dn;              
    float scale_up;           
    float scale_dn;           

    // SF uncertainties
    std::vector<float> errEffSF_stat_up;
    std::vector<float> errEffSF_stat_dn;
    std::vector<float> errEffSF_syst_up;
    std::vector<float> errEffSF_syst_dn;

    // Polymorphism, baby!!
    bool isEle() const { return false; }
    bool isMu()  const { return true; }
    void setState(int sys);

    void print() const;

    void clear(){
      isCombined = 0;
      idTrackPt = idTrackEta = idTrackPhi = 0;
      msTrackPt = msTrackEta = msTrackPhi = 0;
      idTrackQ  = msTrackQ = 0;
      idTrackTheta = idTrackQoverP = 0;
      msTrackTheta = msTrackQoverP = 0;
      veryLoose = loose = medium = tight = false;

      muoEffSF.assign(MuonId::MuonIdInvalid, 1);
      errEffSF_stat_up.assign(MuonId::MuonIdInvalid, 0);
      errEffSF_stat_dn.assign(MuonId::MuonIdInvalid, 0);
      errEffSF_syst_up.assign(MuonId::MuonIdInvalid, 0);
      errEffSF_syst_dn.assign(MuonId::MuonIdInvalid, 0);

      isBadMuon = isCosmic = false;
      ms_up = ms_dn = id_up = id_dn = scale_up = scale_dn  = 0;

      Lepton::clear();
    }

    ClassDef(Muon, 14);
};
} // Susy
#endif