#ifdef __MAKECINT__ 
#pragma link C++ class std::vector< int >+;
#pragma link C++ class std::vector< std::vector< int > >+;
#pragma link C++ class std::vector< std::vector< float >  >+;
#pragma link C++ class std::vector< std::vector< double >  >+;
#pragma link C++ class std::vector< float >;
#pragma link C++ class std::vector< std::string>;
#pragma link C++ class std::vector< std::vector< string > >+;
#pragma link C++ class std::vector< TLorentzVector >+;
#pragma link C++ class std::vector< TString >+;
#pragma link C++ class std::vector< std::vector< TLorentzVector > >+;
#pragma link C++ class std::vector< std::vector< double > >+;


#endif


#ifndef ZTree_h
#define ZTree_h

#include <TROOT.h>
#include <TChain.h>
#include <TFile.h>
#include <TSelector.h>
#include <TLorentzVector.h>
#include <vector>
#include <TH1.h>
#include <TH2.h>
#include <TNtuple.h>
#include <string.h>
#include <cstdlib>
#include <map>
#include <iomanip>
#include <iostream>
#include <fstream>
#include <cctype>
#include <cmath>
#include <cstdlib>

class ZTree : public TSelector {


public :
   TTree          *fChain;   //!pointer to the analyzed TTree or TChain

   TH1D * hNmuons;
   TH1D * hPTmuons;
   TH1D * hetamuons;
   
   
  
   // Declaration of leaf types
   
    std::vector<TLorentzVector>   *muon4mom;//!
    std::vector<TLorentzVector>   *electron4mom;//!
   
   // List of branches
   TBranch        *b_muon4mom;   //!
   
   TBranch        *b_electron4mom;//!
   
   
   
   ZTree(TTree * /*tree*/ =0) { }
   virtual ~ZTree() { }
   virtual Int_t   Version() const { return 2; }
   virtual void    Begin(TTree *tree);
   virtual void    SlaveBegin(TTree *tree);
   virtual void    Init(TTree *tree);
   virtual Bool_t  Notify();
   virtual Bool_t  Process(Long64_t entry);
   virtual Int_t   GetEntry(Long64_t entry, Int_t getall = 0) { return fChain ? fChain->GetTree()->GetEntry(entry, getall) : 0; }
   virtual void    SetOption(const char *option) { fOption = option; }
   virtual void    SetObject(TObject *obj) { fObject = obj; }
   virtual void    SetInputList(TList *input) { fInput = input; }
   virtual TList  *GetOutputList() const { return fOutput; }
   virtual void    SlaveTerminate();
   virtual void    Terminate();

   


   ClassDef(ZTree,0);
};

#endif

#ifdef ZTree_cxx
void ZTree::Init(TTree *tree)
{
   
   muon4mom= 0;
   electron4mom= 0;
  
   // Set branch addresses and branch pointers
   if (!tree) return;
   fChain = tree;
   fChain->SetMakeClass(1);
    fChain->SetBranchAddress("muon4mom", &muon4mom, &b_muon4mom);
   fChain->SetBranchAddress("electron4mom", &electron4mom, &b_electron4mom);
   
   
}

Bool_t ZTree::Notify()
{
   
   return kTRUE;
}

#endif // #ifdef ZTree_cxx