#ifndef rcshifter_h
#define rcshifter_h

#include <TROOT.h>
#include <TChain.h>
#include <TFile.h>
#include <TH1D.h>
#include <vector>

class rcshifter {
 public :
   TTree          *fChain;   //!pointer to the analyzed TTree or TChain
   Int_t           fCurrent; //!current Tree number in a TChain

   // Declaration of leaf types
   TH1D            *rchist;

   // List of branches
   TBranch        *b_rchist;   //!

   rcshifter(const char *rootfile);
   virtual ~rcshifter();
   virtual Int_t    Cut(Long64_t entry);
   virtual Int_t    GetEntry(Long64_t entry);
   virtual Long64_t LoadTree(Long64_t entry);
   virtual void     Init(TTree *tree);
   virtual void     Loop();
   virtual Bool_t   Notify();
   virtual void     Show(Long64_t entry = -1);

   void drawold(int i, int j);
   void drawnew(int i, int j);

 protected:
   TFile *rcfile;
   TTree *rctree;
   TTree *rctree_corr;
   TH1D *rchist_corr;
   Int_t rcrows;
   Int_t rccolumns;
   Int_t rcsteps;
   std::vector<TH1D*> srchist;

 public:
   void save();
};

#endif

#ifdef rcshifter_cxx
rcshifter::~rcshifter()
{
   if (!fChain) return;
   delete fChain->GetCurrentFile();
}

Int_t rcshifter::GetEntry(Long64_t entry)
{
   // Read contents of entry.

   if (!fChain) return 0;
   return fChain->GetEntry(entry);
}

Long64_t rcshifter::LoadTree(Long64_t entry)
{
   // Set the environment to read one entry

   if (!fChain) return -5;
   Long64_t centry = fChain->LoadTree(entry);
   if (centry < 0) return centry;
   if (fChain->GetTreeNumber() != fCurrent) {
      fCurrent = fChain->GetTreeNumber();
      Notify();
   }
   return centry;
}

void rcshifter::Init(TTree *tree)
{
   // The Init() function is called when the selector needs to initialize
   // a new tree or chain. Typically here the branch addresses and branch
   // pointers of the tree will be set.
   // It is normally not necessary to make changes to the generated
   // code, but the routine can be extended by the user if needed.
   // Init() will be called many times when running on PROOF
   // (once per file to be processed).

   // Set object pointer
   rchist = 0;
   // Set branch addresses and branch pointers
   if (!tree) return;
   fChain = tree;
   fCurrent = -1;
   fChain->SetMakeClass(1);
   fChain->SetBranchAddress("rchist", &rchist, &b_rchist);
   Notify();
}

Bool_t rcshifter::Notify()
{
   // The Notify() function is called when a new file is opened. This
   // can be either for a new TTree in a TChain or when when a new TTree
   // is started when using PROOF. It is normally not necessary to make changes
   // to the generated code, but the routine can be extended by the
   // user if needed. The return value is currently not used.

   return kTRUE;
}

void rcshifter::Show(Long64_t entry)
{
   // Print contents of entry.
   // If entry is not specified, print current entry

   if (!fChain) return;
   fChain->Show(entry);
}

Int_t rcshifter::Cut(Long64_t entry)
{
   // This function may be called from Loop.
   // returns  1 if entry is accepted.
   // returns -1 otherwise.
 
   return 1;
}
#endif // #ifdef rcshifter_cxx