mvaeffs.h

Go to the documentation of this file.

#ifndef mvaeffs__HH
#define mvaeffs__HH
 
#include "RQ_OBJECT.h"
#include "TCanvas.h"
#include "TFile.h"
#include "TGLabel.h"
#include "TGNumberEntry.h"
#include "TGWindow.h"
#include "TGaxis.h"
#include "TH1.h"
#include "TIterator.h"
#include "TLatex.h"
#include "TList.h"
 
namespace TMVA{
 
   void mvaeffs(TString dataset, TString fin = "TMVA.root",
                Float_t nSignal = 1000, Float_t nBackground = 1000,
                Bool_t useTMVAStyle = kTRUE, TString formula="S/sqrt(S+B)" );
 
   // this macro plots the signal and background efficiencies
   // as a function of the MVA cut.
 
 
   class MethodInfo : public TNamed {
   public:
   MethodInfo() :
      methodName(""),
         methodTitle(""),
         sig(nullptr),
         bgd(nullptr),
         origSigE(nullptr),
         origBgdE(nullptr),
         sigE(nullptr),
         bgdE(nullptr),
         purS(nullptr),
         sSig(nullptr),
         effpurS(nullptr),
         canvas(nullptr),
         line1(nullptr),
         line2(nullptr),
         rightAxis(nullptr),
         maxSignificance(0),
         maxSignificanceErr(0)
            {}
      virtual ~MethodInfo();
 
      TString  methodName;
      TString  methodTitle;
      TH1*     sig;
      TH1*     bgd;
      TH1*     origSigE;
      TH1*     origBgdE;
      TH1*     sigE;
      TH1*     bgdE;
      TH1*     purS;
      TH1*     sSig;
      TH1*     effpurS;
      TCanvas* canvas;
      TLatex*  line1;
      TLatex*  line2;
      TGaxis*  rightAxis;
      Double_t maxSignificance;
      Double_t maxSignificanceErr;
 
      void SetResultHists();
 
      ClassDef(MethodInfo,0);
   };
 
   class StatDialogMVAEffs {
 
      RQ_OBJECT("StatDialogMVAEffs")
 
         public:
 
      StatDialogMVAEffs(TString ds,const TGWindow* p, Float_t ns, Float_t nb);
      virtual ~StatDialogMVAEffs();
 
      void SetFormula(const TString& f) { fFormula = f; }
      TString GetFormula();
      TString GetFormulaString() { return fFormula; }
      TString GetLatexFormula();
 
      void ReadHistograms(TFile* file);
      void UpdateSignificanceHists();
      void DrawHistograms();
 
      void RaiseDialog() { if (fMain) { fMain->RaiseWindow(); fMain->Layout(); fMain->MapWindow(); } }
 
   private:
 
      TGMainFrame *fMain;
      Float_t fNSignal;
      Float_t fNBackground;
      TString fFormula;
      TString dataset;
      TList * fInfoList;
 
      TGNumberEntry* fSigInput;
      TGNumberEntry* fBkgInput;
 
      TGHorizontalFrame* fButtons;
      TGTextButton* fDrawButton;
      TGTextButton* fCloseButton;
 
      Int_t maxLenTitle;
 
      void UpdateCanvases();
 
   public:
 
      // slots
      void SetNSignal(); //*SIGNAL*
      void SetNBackground(); //*SIGNAL*
      void Redraw(); //*SIGNAL*
      void Close(); //*SIGNAL*
 
      // result printing
      void PrintResults( const MethodInfo* info );
   };
 
}
#endif