root/TMVA/PlotFoams.h

0001 #ifndef PlotFoams__HH
0002 #define PlotFoams__HH
0003
0004 #include "tmvaglob.h"
0005 #include "TControlBar.h"
0006 #include "TMap.h"
0007 #include "TVectorT.h"
0008 #include "TString.h"
0009 #include "TPaveText.h"
0010 #include "TMVA/PDEFoamKernelBase.h"
0011 #include "TMVA/PDEFoamKernelTrivial.h"
0012
0013 #include <cfloat>
0014
0015 #include "TMVA/PDEFoam.h"
0016
0017 namespace TMVA{
0018
0019    void PlotFoams( TString fileName = "weights/TMVAClassification_PDEFoam.weights_foams.root",
0020                    bool useTMVAStyle = kTRUE );
0021    // foam plotting macro
0022    void Plot(TString fileName, TMVA::ECellValue cv, TString cv_long, bool useTMVAStyle = kTRUE);
0023
0024    void Plot1DimFoams(TList& foam_list, TMVA::ECellValue cell_value,
0025                       const TString& cell_value_description,
0026                       TMVA::PDEFoamKernelBase* kernel);
0027    void PlotNDimFoams(TList& foam_list, TMVA::ECellValue cell_value,
0028                       const TString& cell_value_description,
0029                       TMVA::PDEFoamKernelBase* kernel);
0030    void PlotCellTree(TString fileName, TString cv_long, bool useTMVAStyle = kTRUE);
0031
0032    void DrawCell( TMVA::PDEFoamCell *cell, TMVA::PDEFoam *foam,
0033                   Double_t x, Double_t y,
0034                   Double_t xscale,  Double_t yscale );
0035 }
0036
0037 #endif