/data/git/CMSSW_5_3_11_patch5/src/TopQuarkAnalysis/TopTools/interface/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