LRHelpFunctions.h

Go to the documentation of this file.

//
// Author:  Jan Heyninck
// Created: Tue Apr  3 17:33:23 PDT 2007
//
// $Id: LRHelpFunctions.h,v 1.10 2008/06/19 12:28:27 rwolf Exp $
//

#ifndef LRHelpFunctions_h
#define LRHelpFunctions_h

#include <cmath>
#include <iostream>

#include "TROOT.h"
#include "TString.h"
#include "TFile.h"
#include "TKey.h"
#include "TH1F.h"
#include "TH2F.h"
#include "TF1.h"
#include "TGraph.h"
#include "TList.h"
#include "TPaveText.h"
#include "TText.h"
#include "TCanvas.h"


class LRHelpFunctions {

  public:
    LRHelpFunctions();
    LRHelpFunctions(std::vector<int>, int, std::vector<double>, std::vector<double>,std::vector<const char*>);
    LRHelpFunctions(int, double, double, const char*);
    ~LRHelpFunctions(); 
    void    recreateFitFct(std::vector<int> obsNr, std::vector<const char*> functions);
    void    initLRHistsAndFits(int, double, double, const char*);
    void    setObsFitParameters(int obs,std::vector<double>);
    void    fillToSignalHists(std::vector<double>, double weight = 1.0);
    void    fillToBackgroundHists(std::vector<double>, double weight = 1.0);
    void    fillToSignalHists(int, double, double weight = 1.0);
    void    fillToBackgroundHists(int, double, double weight = 1.0);
    void    fillToSignalCorrelation(int obsNbr1, double obsVal1, int obsNbr2,
    double obsVal2, double weight);
    void    normalizeSandBhists();
    void        makeAndFitSoverSplusBHists();
    void        readObsHistsAndFits(TString,std::vector<int>,bool);
    void        storeToROOTfile(TString);
    void        storeControlPlots(TString); 
    void        fillLRSignalHist(double, double weight = 1.0);
    void        fillLRBackgroundHist(double, double weight = 1.0);
    void        makeAndFitPurityHists(); 
    double  calcLRval(std::vector<double>);
    double  calcPtdrLRval(std::vector<double> vals, bool useCorrelation = false);
    double  calcProb(double);
    bool    obsFitIncluded(int);
    void setXlabels(const std::vector<std::string> & xLabels);
    void setYlabels(const std::vector<std::string> & yLabels);
    void singlePlot(TString fname, int obsNbr, TString extension= TString("eps"));
    void purityPlot(TString fname, TString extension= TString("eps"));


  private:
    std::vector<TH1F*> hObsS, hObsB, hObsSoverSplusB;
    std::vector<TH2F*> hObsCorr;
    std::vector<TF1*>  fObsSoverSplusB;
    TH1F           *hLRtotS, *hLRtotB, *hLRtotSoverSplusB;
    TF1            *fLRtotSoverSplusB;
    TGraph             *hEffvsPur, *hLRValvsPur, *hLRValvsEff;
    bool           constructPurity;
    std::vector<int> obsNumbers;
};

#endif