MadChain.h

Go to the documentation of this file.

#ifndef madchain_h
#define madchain_h
#include <iostream>
#include <string>
#include <map>
#include <utility>
#include "TROOT.h"
#include "TChain.h"
#include "TFile.h"
#include "CandNtupleSR/NtpSRRecord.h"
#include "MCNtuple/NtpMCRecord.h"
#include "TruthHelperNtuple/NtpTHRecord.h"
#include "CandNtupleEM/NtpEMRecord.h"
#include "StandardNtuple/NtpStRecord.h"

class MadChain
{
  protected :

  TChain *fChainSR;
  Int_t fCurrentSR; //current SR chain entry number

  TChain *fChainMC;
  Int_t fCurrentMC; //current MC chain entry number

  TChain *fChainTH;
  Int_t fCurrentTH; //current TH chain entry number

  TChain *fChainEM;
  Int_t fCurrentEM; //current EM chain entry number

  void   Zero();

 public:
  MadChain(TChain *chainSR=0,TChain *chainMC=0,
           TChain *chainTH=0,TChain *chainEM=0, bool build_index=false);
  ~MadChain();
  Int_t  GetEntryNumber(Int_t run,Int_t snarl);
  Int_t  Get(Int_t entry);
  void   Init(TChain *chainSR,TChain *chainMC,
              TChain *chainTH,TChain *chainEM);
  void   Show(Int_t entry = -1);

  Bool_t isMC;
  Bool_t isTH;
  Bool_t isEM;
  Bool_t isST;
  Int_t Nentries; //number of entries for fChainSR

  NtpSRRecord *Record;
  NtpStRecord *stRecord;
  NtpMCRecord *mcRecord;
  NtpTHRecord *thRecord;
  NtpEMRecord *emRecord;

 private:

  void BuildLookup();
  bool fLookupBuilt;


#ifndef __MAKECINT__
#ifndef __CINT__
  // run,snarl lookup table... 
  // TTreeIndex doesn't work properly w/ chains in root v4
  struct first_then_second
  {
    bool operator()(const std::pair<Int_t,Int_t>& s1, 
                    const std::pair<Int_t,Int_t>& s2) const
    {
      if(s1.first < s2.first) return true;
      else if(s1.first > s2.first) return false;
      else {
        if(s1.second<s2.second) return true;
        else return false;
      }
    }
  };
  std::map< std::pair<Int_t,Int_t> , Long64_t, first_then_second > fLookUp; 

#endif
#endif
  

  //  ClassDef(MadChain,0)
};

#endif // #ifdef madchain_h

---