SusyNtuple/SusyNtTruthAna.h

#ifndef SusyNtuple_SusyNtTruthAna_h
#define SusyNtuple_SusyNtTruthAna_h

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

#include "SusyNtuple/SusyNtObject.h"

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

class SusyNtTruthAna : public TSelector
{

  public:

    SusyNtTruthAna();
    virtual ~SusyNtTruthAna(){};

    Susy::SusyNtObject nt;

    //
    // 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 clearTruthObjects();
    void selectTruthObjects();

    //
    // Methods to grab objects
    //
    TruthParticleVector getPreTruthLeptons   (Susy::SusyNtObject* susyNt, int pdgId);
    TruthJetVector      getPreTruthJets      (Susy::SusyNtObject* susyNt           );
    TruthParticleVector getSignalTruthLeptons(TruthParticleVector& truthLeptons    );
    TruthJetVector      getSignalTruthJets   (TruthJetVector& truthJets            );

    // Get Baseline Objects
    void getBaselineTruthObjects(Susy::SusyNtObject*  susyNt, 
                                 TruthParticleVector& truthElectrons, 
                                 TruthParticleVector& truthMuons, 
                                 TruthParticleVector& truthTaus, 
                                 TruthJetVector&      truthJets);
    // Get Signal Objects
    void getSignalTruthObjects (TruthParticleVector& truthBaseElectrons, 
                                TruthParticleVector& truthBaseMuons, 
                                TruthParticleVector& truthBaseTaus, 
                                TruthJetVector&      truthBaseJets,
                                TruthParticleVector& truthSignalElectrons, 
                                TruthParticleVector& truthSignalMuons, 
                                TruthParticleVector& truthSignalTaus, 
                                TruthJetVector&      truthSignalJets);

    // Get Met
    Susy::TruthMet* getTruthMet(Susy::SusyNtObject* susyNt);

    // Misc
    void removeSFOSPair        (TruthParticleVector& leptons, 
                                float MllCut = 20.); // 20 GeV

    // Event and object dumps
    void dumpEvent();
    void dumpBaselineTruthObjects();
    void dumpSignalTruthObjects();

    // Set-Get options
    void setDebug(bool flag) { m_dbg = flag; }
 
    // Dump timer
    void dumpTimer();

    // Access tree
    TTree* getTree() { return m_tree; }

    ClassDef(SusyNtTruthAna, 1);

  protected:

    //
    // General
    //

    TTree* m_tree;              

    Long64_t m_entry;           
    Long64_t m_chainEntry;      

    int   m_dbg;                
 
    //
    // Object collections
    //
    TruthParticleVector         m_baseTruthElectrons;
    TruthParticleVector         m_baseTruthMuons;
    TruthParticleVector         m_baseTruthTaus;
    TruthJetVector              m_baseTruthJets;
    
    TruthParticleVector         m_signalTruthElectrons;
    TruthParticleVector         m_signalTruthMuons;
    TruthParticleVector         m_signalTruthTaus;
    TruthJetVector              m_signalTruthJets;
    const Susy::TruthMet*       m_truthMet;

    TStopwatch          m_timer;
};

#endif