FancyTTreeInterpolator.h

// -*- C++ -*-.
/**************************************************************************************************************
 Author: Ben Strutt
 Email: b.strutt.12@ucl.ac.uk

 Description:
             Give this class a TTree plus a branch name to serve as an 'x-axis', normally this is a time.
             FancyTTreeInterpolator generates TGraphs of any TTree branch variable as a function of your x-axis.
             That allows us to do two slightly clever things.
             The first slightly clever thing is that it sorts the data so it is x-axis (time) ordered.
             The second slightly clever thing is that it can then interpolate the data to values between entries.
             This interpolation is equivalent to TGraph::Eval.
**************************************************************************************************************/

#ifndef FANCYTTREEINTERPOLATOR_H
#define FANCYTTREEINTERPOLATOR_H

#include "TObject.h"
#include "TTree.h"
#include "TChain.h"
#include "TGraph.h"
#include "TROOT.h"
#include "TMath.h"
#include "TAxis.h"
#include "TCanvas.h"

#include <iostream>
#include <exception>
#include <stdexcept>

namespace Acclaim
{

  class FancyTTreeInterpolator{

  public:
    FancyTTreeInterpolator(TTree* t, TString xAxisText);
    ~FancyTTreeInterpolator();

    void add(TString yAxisText);
    void add(TString yAxisText, TString cut);
    void add(TString yAxisText, Double_t wrapValue);
    void add(TString yAxisText, TString cut, Double_t wrapValue);
    Double_t interp(TString yAxisText, Double_t xAxisValue);
    TGraph* get(TString yAxisText);
    TGraph* makeSortedTGraph(TString yAxisText);
    TGraph* makeSortedTGraph(TString yAxisText, TString cutString);
    TGraph* makeSortedTGraph(TString yAxisText, Double_t wrapValue);
    TGraph* makeSortedTGraph(TString yAxisText, TString cutString, Double_t wrapValue);

    TTree* fTree; 
    TString fXAxisText; 
    std::map<TString,TGraph*> fStringToGraph; 
    std::map<TString, Double_t> fStringToWrapValue; 
    Double_t fXmin; 
    Double_t fXmax; 
  
  };
}

#endif //FANCYTTREESORTERANDINTERPOLATOR_H