SusyNtuple/SusyNtAna.h

#ifndef SusyNtuple_SusyNtAna_h
#define SusyNtuple_SusyNtAna_h

#include "TSelector.h"
#include "TTree.h"
#include "TStopwatch.h"

#include "SusyNtuple/SusyNtObject.h"
#include "SusyNtuple/SusyNtTools.h"
#include "SusyNtuple/MCWeighter.h"
#include "SusyNtuple/SusyNtSys.h"
#include "SusyNtuple/TauId.h"

#include <fstream>
#include <map>
#include <set>


// To debug events in input file 
typedef std::map< unsigned int, std::set<unsigned int>* > RunEventMap;


class SusyNtAna : public TSelector
{

  public:

    SusyNtAna();
    virtual ~SusyNtAna(){};

    Susy::SusyNtObject nt;
    SusyNtTools m_nttools;
    inline void setAnaType(AnalysisType v){ m_nttools.setAnaType(v); }
    /*const*/ SusyNtTools& nttools() /*const*/ { return m_nttools; } // DG should be const

    //
    // TSelector methods
    //

    virtual void    Init(TTree *tree);
    virtual void    Begin(TTree *tree);
    virtual Bool_t  Notify() { return kTRUE; }
    virtual void    Terminate();
    virtual Int_t   Version() const {
      return 2;
    }

    virtual Bool_t  Process(Long64_t entry);

    virtual Int_t   GetEntry(Long64_t e, Int_t getall = 0) {
      m_entry=e;
      return kTRUE;
    }

    // Object selection
    void clearObjects();
    void selectObjects(Susy::NtSys::SusyNtSys sys = Susy::NtSys::NOM,
                       TauId signalTauID=TauId::Medium);

    // Cleaning cuts
    int cleaningCutFlags();

    // Event and object dumps
    void dumpEvent();
    void dumpNtupleObjects();
    void dumpBaselineObjects();
    void dumpBaselineLeptons();
    void dumpBaselineJets();
    void dumpSignalObjects();
    void dumpSignalLeptons();
    void dumpSignalJets();

    void setSelectTaus(bool doIt) { m_selectTaus = doIt; }

    void setPrintFreq(int freq) { m_printFreq = freq; }

    void setDebug(int dbg) { m_dbg = dbg; m_mcWeighter.setVerbose(dbg); }
    int dbg() { return m_dbg; }

    void toggleCheckDuplicates(bool b=true) { m_duplicate = b; }
    bool checkDuplicate() { return m_duplicate; }
    
    void setEvtDebug() { m_dbgEvt = true; }
    bool dbgEvt() const { return m_dbgEvt; }
    void loadEventList(const std::string filename="debugEvents.txt");
    bool processThisEvent(unsigned int run, unsigned int event);
    bool checkRunEvent(const RunEventMap &runEventMap, unsigned int run, unsigned int event);
    bool checkAndAddRunEvent(RunEventMap &runEventMap, unsigned int run, unsigned int event);
    void addRunEvent(RunEventMap &runEventMap, unsigned int run, unsigned int event) 
    { checkAndAddRunEvent(runEventMap, run, event); }

    bool isDuplicate(unsigned int run, unsigned int event);

    // Sample name - can be used however you like
    std::string sampleName() const { return m_sample; }
    void setSampleName(std::string s) { m_sample = s; }

    MCWeighter& mcWeighter() { return m_mcWeighter; }

    void dumpTimer();

    TTree* getTree() { return m_tree; }

    ClassDef(SusyNtAna, 1);

  protected:

    //
    // General
    //

    TTree* m_tree;              

    Long64_t m_entry;           
    Long64_t m_chainEntry;      

    bool m_selectTaus;          

    int   m_printFreq;          
    int   m_dbg;                
    bool  m_dbgEvt;             
    bool  m_duplicate;          
    
    std::string m_sample;       

    // To debug events in input file 
    RunEventMap m_eventList;          
    RunEventMap m_eventListDuplicate; 

    MCWeighter m_mcWeighter;   // provides MC normalization and event weight


    //
    // Object collections
    //

    ElectronVector      m_preElectrons;         
    MuonVector          m_preMuons;             
    LeptonVector        m_preLeptons;       
    JetVector           m_preJets;              
    TauVector           m_preTaus;

    ElectronVector      m_baseElectrons;        
    MuonVector          m_baseMuons;            
    LeptonVector        m_baseLeptons;          
    TauVector           m_baseTaus;             
    JetVector           m_baseJets;             

    ElectronVector      m_signalElectrons;      
    MuonVector          m_signalMuons;          
    LeptonVector        m_signalLeptons;        
    TauVector           m_signalTaus;           
    JetVector           m_signalJets;           
    JetVector           m_signalJets2Lep;       

    // New organization of tau selections
    TauVector           m_mediumTaus;           
    TauVector           m_tightTaus;            

    const Susy::Met*    m_met;                  
    const Susy::TrackMet*    m_trackMet;        

    TStopwatch          m_timer;

};



#endif