benhoob/HWW/BDT.C

#include <iostream>
#include <iomanip>
#include <fstream>

#include "tmvaglob.C"

#include "RQ_OBJECT.h"

#include "TROOT.h"
#include "TStyle.h"
#include "TPad.h"
#include "TCanvas.h"
#include "TLine.h"
#include "TFile.h"
#include "TColor.h"
#include "TPaveText.h"
#include "TObjString.h"
#include "TControlBar.h"

#include "TGWindow.h"
#include "TGButton.h"
#include "TGLabel.h"
#include "TGNumberEntry.h"

#include "TMVA/DecisionTree.h"
#include "TMVA/Tools.h"
#include "TXMLEngine.h"

// Uncomment this only if the link problem is solved. The include statement tends
// to use the ROOT classes rather than the local TMVA release
// #include "TMVA/DecisionTree.h"
// #include "TMVA/DecisionTreeNode.h"

// this macro displays a decision tree read in from the weight file

static const Int_t kSigColorF = TColor::GetColor( "#2244a5" );  // novel blue 
static const Int_t kBkgColorF = TColor::GetColor( "#dd0033" );  // novel red  
static const Int_t kIntColorF = TColor::GetColor( "#33aa77" );  // novel green

static const Int_t kSigColorT = 10;
static const Int_t kBkgColorT = 10;
static const Int_t kIntColorT = 10;

enum PlotType { EffPurity = 0 };

class StatDialogBDT {  

   RQ_OBJECT("StatDialogBDT")

 public:

   StatDialogBDT( const TGWindow* p, TString wfile = "weights/TMVAClassification_BDT.weights.txt", 
                  TString methName = "BDT", Int_t itree = 0 );
   virtual ~StatDialogBDT() {
      TMVA::DecisionTreeNode::fgIsTraining=false;
      fThis = 0;
      fMain->CloseWindow();
      fMain->Cleanup();
      if(gROOT->GetListOfCanvases()->FindObject(fCanvas))
        delete fCanvas; 
   }
   
   // draw method
   void DrawTree( Int_t itree );

   void RaiseDialog() { if (fMain) { fMain->RaiseWindow(); fMain->Layout(); fMain->MapWindow(); } }
   
 private:
   
   TGMainFrame *fMain;
   Int_t        fItree;
   Int_t        fNtrees;
   TCanvas*     fCanvas;

   TGNumberEntry* fInput;

   TGHorizontalFrame* fButtons;
   TGTextButton* fDrawButton;
   TGTextButton* fCloseButton;

   void UpdateCanvases();

   // draw methods
   TMVA::DecisionTree* ReadTree( TString * &vars, Int_t itree );
   void                DrawNode( TMVA::DecisionTreeNode *n, 
                                 Double_t x, Double_t y, Double_t xscale,  Double_t yscale, TString* vars );
   void GetNtrees();

   TString fWfile;
   TString fMethName;

 public:

   // static function for external deletion
   static void Delete() { if (fThis != 0) { delete fThis; fThis = 0; } }

   // slots
   void SetItree(); //*SIGNAL*
   void Redraw(); //*SIGNAL*
   void Close(); //*SIGNAL*

 private:

   static StatDialogBDT* fThis;

};

StatDialogBDT* StatDialogBDT::fThis = 0;

void StatDialogBDT::SetItree() 
{
   fItree = Int_t(fInput->GetNumber());
}

void StatDialogBDT::Redraw() 
{
   UpdateCanvases();
}

void StatDialogBDT::Close() 
{
   delete this;
}

StatDialogBDT::StatDialogBDT( const TGWindow* p, TString wfile, TString methName, Int_t itree )
   : fMain( 0 ),
     fItree(itree),
     fNtrees(0),
     fCanvas(0),
     fInput(0),
     fButtons(0),
     fDrawButton(0),
     fCloseButton(0),
     fWfile( wfile ),
     fMethName( methName )
{
   UInt_t totalWidth  = 500;
   UInt_t totalHeight = 200;

   fThis = this;

   TMVA::DecisionTreeNode::fgIsTraining=true;

   // read number of decision trees from weight file
   GetNtrees();

   // main frame
   fMain = new TGMainFrame(p, totalWidth, totalHeight, kMainFrame | kVerticalFrame);

   TGLabel *sigLab = new TGLabel( fMain, Form( "Decision tree [%i-%i]",0,fNtrees-1 ) );
   fMain->AddFrame(sigLab, new TGLayoutHints(kLHintsLeft | kLHintsTop,5,5,5,5));

   fInput = new TGNumberEntry(fMain, (Double_t) fItree,5,-1,(TGNumberFormat::EStyle) 5);
   fMain->AddFrame(fInput, new TGLayoutHints(kLHintsLeft | kLHintsTop,5,5,5,5));
   fInput->Resize(100,24);
   fInput->SetLimits(TGNumberFormat::kNELLimitMinMax,0,fNtrees-1);

   fButtons = new TGHorizontalFrame(fMain, totalWidth,30);

   fCloseButton = new TGTextButton(fButtons,"&Close");
   fButtons->AddFrame(fCloseButton, new TGLayoutHints(kLHintsLeft | kLHintsTop));

   fDrawButton = new TGTextButton(fButtons,"&Draw");
   fButtons->AddFrame(fDrawButton, new TGLayoutHints(kLHintsRight | kLHintsTop,15));
  
   fMain->AddFrame(fButtons,new TGLayoutHints(kLHintsLeft | kLHintsBottom,5,5,5,5));

   fMain->SetWindowName("Decision tree");
   fMain->SetWMPosition(0,0);
   fMain->MapSubwindows();
   fMain->Resize(fMain->GetDefaultSize());
   fMain->MapWindow();

   fInput->Connect("ValueSet(Long_t)","StatDialogBDT",this, "SetItree()");

   fDrawButton->Connect("Clicked()","TGNumberEntry",fInput, "ValueSet(Long_t)");
   fDrawButton->Connect("Clicked()", "StatDialogBDT", this, "Redraw()");   

   fCloseButton->Connect("Clicked()", "StatDialogBDT", this, "Close()");
}

void StatDialogBDT::UpdateCanvases() 
{
   DrawTree( fItree );
}

void StatDialogBDT::GetNtrees()
{
   if(!fWfile.EndsWith(".xml") ){
      ifstream fin( fWfile );
      if (!fin.good( )) { // file not found --> Error
         cout << "*** ERROR: Weight file: " << fWfile << " does not exist" << endl;
         return;
      }
   
      TString dummy = "";
      
      // read total number of trees, and check whether requested tree is in range
      Int_t nc = 0;
      while (!dummy.Contains("NTrees")) { 
         fin >> dummy; 
         nc++; 
         if (nc > 200) {
            cout << endl;
            cout << "*** Huge problem: could not locate term \"NTrees\" in BDT weight file: " 
                 << fWfile << endl;
            cout << "==> panic abort (please contact the TMVA authors)" << endl;
            cout << endl;
            exit(1);
         }
      }
      fin >> dummy; 
      fNtrees = dummy.ReplaceAll("\"","").Atoi();
      fin.close();
   }
   else{
      void* doc = TMVA::gTools().xmlengine().ParseFile(fWfile);
      void* rootnode = TMVA::gTools().xmlengine().DocGetRootElement(doc);
      void* ch = TMVA::gTools().xmlengine().GetChild(rootnode);
      while(ch){
         TString nodeName = TString( TMVA::gTools().xmlengine().GetNodeName(ch) );
         if(nodeName=="Weights") {
            TMVA::gTools().ReadAttr( ch, "NTrees", fNtrees );
            break;
         }
         ch = TMVA::gTools().xmlengine().GetNext(ch);
      }
   }
   cout << "--- Found " << fNtrees << " decision trees in weight file" << endl;

}

//_______________________________________________________________________
void StatDialogBDT::DrawNode( TMVA::DecisionTreeNode *n, 
                               Double_t x, Double_t y, 
                               Double_t xscale,  Double_t yscale, TString * vars) 
{
   // recursively puts an entries in the histogram for the node and its daughters
   //
   Float_t xsize=xscale*1.5;
   Float_t ysize=yscale/3;
   if (xsize>0.15) xsize=0.1; //xscale/2;
   if (n->GetLeft() != NULL){
      TLine *a1 = new TLine(x-xscale/4,y-ysize,x-xscale,y-ysize*2);
      a1->SetLineWidth(2);
      a1->Draw();
      DrawNode((TMVA::DecisionTreeNode*) n->GetLeft(), x-xscale, y-yscale, xscale/2, yscale, vars);
   }
   if (n->GetRight() != NULL){
      TLine *a1 = new TLine(x+xscale/4,y-ysize,x+xscale,y-ysize*2);
      a1->SetLineWidth(2);
      a1->Draw();
      DrawNode((TMVA::DecisionTreeNode*) n->GetRight(), x+xscale, y-yscale, xscale/2, yscale, vars  );
   }

   //   TPaveText *t = new TPaveText(x-xscale/2,y-yscale/2,x+xscale/2,y+yscale/2, "NDC");
   TPaveText *t = new TPaveText(x-xsize,y-ysize,x+xsize,y+ysize, "NDC");

   t->SetBorderSize(1);

   t->SetFillStyle(1);
   if      (n->GetNodeType() ==  1) { t->SetFillColor( kSigColorF ); t->SetTextColor( kSigColorT ); }
   else if (n->GetNodeType() == -1) { t->SetFillColor( kBkgColorF ); t->SetTextColor( kBkgColorT ); }
   else if (n->GetNodeType() ==  0) { t->SetFillColor( kIntColorF ); t->SetTextColor( kIntColorT ); }

   char buffer[25];
   sprintf( buffer, "N=%f", n->GetNEvents() );
   if (n->GetNEvents()>0) t->AddText(buffer);
   sprintf( buffer, "S/(S+B)=%4.3f", n->GetPurity() );
   t->AddText(buffer);

   if (n->GetNodeType() == 0){
      if (n->GetCutType()){
         t->AddText(TString(vars[n->GetSelector()]+">"+=::Form("%5.3g",n->GetCutValue())));
      }else{
         t->AddText(TString(vars[n->GetSelector()]+"<"+=::Form("%5.3g",n->GetCutValue())));
      }
   }

   t->Draw();

   return;
}
TMVA::DecisionTree* StatDialogBDT::ReadTree( TString* &vars, Int_t itree )
{
   cout << "--- Reading Tree " << itree << " from weight file: " << fWfile << endl;
   TMVA::DecisionTree *d = new TMVA::DecisionTree();
   if(!fWfile.EndsWith(".xml") ){
      ifstream fin( fWfile );
      if (!fin.good( )) { // file not found --> Error
         cout << "*** ERROR: Weight file: " << fWfile << " does not exist" << endl;
         return 0;
      }
      
      TString dummy = "";
      
      if (itree >= fNtrees) {
         cout << "*** ERROR: requested decision tree: " << itree 
              << ", but number of trained trees only: " << fNtrees << endl;
         return 0;
      }
      
      // file header with name
      while (!dummy.Contains("#VAR")) fin >> dummy;
      fin >> dummy >> dummy >> dummy; // the rest of header line
      
      // number of variables
      Int_t nVars;
      fin >> dummy >> nVars;
      
      // variable mins and maxes
      vars = new TString[nVars+1]; // last one is if "fisher cut criterium"
      for (Int_t i = 0; i < nVars; i++) fin >> vars[i] >> dummy >> dummy >> dummy >> dummy;
      vars[nVars]="FisherCrit";

      char buffer[20];
      char line[256];
      sprintf(buffer,"Tree %d",itree);

      while (!dummy.Contains(buffer)) {
         fin.getline(line,256);
         dummy = TString(line);
      }

      d->Read(fin);
      
      fin.close();
   }
   else{
     if (itree >= fNtrees) {
         cout << "*** ERROR: requested decision tree: " << itree 
              << ", but number of trained trees only: " << fNtrees << endl;
         return 0;
      }
     Int_t nVars;
      void* doc = TMVA::gTools().xmlengine().ParseFile(fWfile);
      void* rootnode = TMVA::gTools().xmlengine().DocGetRootElement(doc);
      void* ch = TMVA::gTools().xmlengine().GetChild(rootnode);
      while(ch){
         TString nodeName = TString( TMVA::gTools().xmlengine().GetNodeName(ch) );
         if(nodeName=="Variables"){
            TMVA::gTools().ReadAttr( ch, "NVar", nVars);
            vars = new TString[nVars+1]; 
            void* varnode =  TMVA::gTools().xmlengine().GetChild(ch);
            for (Int_t i = 0; i < nVars; i++){
               TMVA::gTools().ReadAttr( varnode, "Expression", vars[i]);
               varnode =  TMVA::gTools().xmlengine().GetNext(varnode);
            }
            vars[nVars]="FisherCrit";
         }
         if(nodeName=="Weights") break;
         ch = TMVA::gTools().xmlengine().GetNext(ch);
      }
      ch = TMVA::gTools().xmlengine().GetChild(ch);
      for (int i=0; i<itree; i++) ch = TMVA::gTools().xmlengine().GetNext(ch);
      d->ReadXML(ch);
   }
   return d;
}

//_______________________________________________________________________
void StatDialogBDT::DrawTree( Int_t itree )
{
   TString *vars;   
   TMVA::DecisionTree* d = ReadTree( vars, itree );
   if (d == 0) return;

   UInt_t   depth = d->GetTotalTreeDepth();
   Double_t ystep = 1.0/(depth + 1.0);

   cout << "--- Tree depth: " << depth << endl;

   TStyle* TMVAStyle   = gROOT->GetStyle("Plain"); // our style is based on Plain
   Int_t   canvasColor = TMVAStyle->GetCanvasColor(); // backup

   TString cbuffer = Form( "Reading weight file: %s", fWfile.Data() );
   TString tbuffer = Form( "Decision Tree no.: %d", itree );
   if (!fCanvas) fCanvas = new TCanvas( "c1", cbuffer, 200, 0, 1000, 600 ); 
   else          fCanvas->Clear();
   fCanvas->Draw();   

   DrawNode( (TMVA::DecisionTreeNode*)d->GetRoot(), 0.5, 1.-0.5*ystep, 0.25, ystep ,vars);
  
   // make the legend
   Double_t yup=0.99;
   Double_t ydown=yup-ystep/2.5;
   Double_t dy= ystep/2.5 * 0.2;
 
   TPaveText *whichTree = new TPaveText(0.85,ydown,0.98,yup, "NDC");
   whichTree->SetBorderSize(1);
   whichTree->SetFillStyle(1);
   whichTree->SetFillColor( TColor::GetColor( "#ffff33" ) );
   whichTree->AddText( tbuffer );
   whichTree->Draw();

   TPaveText *intermediate = new TPaveText(0.02,ydown,0.15,yup, "NDC");
   intermediate->SetBorderSize(1);
   intermediate->SetFillStyle(1);
   intermediate->SetFillColor( kIntColorF );
   intermediate->AddText("Intermediate Nodes");
   intermediate->SetTextColor( kIntColorT );
   intermediate->Draw();

   ydown = ydown - ystep/2.5 -dy;
   yup   = yup - ystep/2.5 -dy;
   TPaveText *signalleaf = new TPaveText(0.02,ydown ,0.15,yup, "NDC");
   signalleaf->SetBorderSize(1);
   signalleaf->SetFillStyle(1);
   signalleaf->SetFillColor( kSigColorF );
   signalleaf->AddText("Signal Leaf Nodes");
   signalleaf->SetTextColor( kSigColorT );
   signalleaf->Draw();

   ydown = ydown - ystep/2.5 -dy;
   yup   = yup - ystep/2.5 -dy;
   TPaveText *backgroundleaf = new TPaveText(0.02,ydown,0.15,yup, "NDC");
   backgroundleaf->SetBorderSize(1);
   backgroundleaf->SetFillStyle(1);
   backgroundleaf->SetFillColor( kBkgColorF );

   backgroundleaf->AddText("Backgr. Leaf Nodes");
   backgroundleaf->SetTextColor( kBkgColorT );
   backgroundleaf->Draw();

   fCanvas->Update();
   TString fname = Form("plots/%s_%i", fMethName.Data(), itree );
   cout << "--- Creating image: " << fname << endl;
   TMVAGlob::imgconv( fCanvas, fname );   

   TMVAStyle->SetCanvasColor( canvasColor );
}   
      
// ========================================================================================

static std::vector<TControlBar*> BDT_Global__cbar;

// intermediate GUI
void BDT( const TString& fin = "TMVA.root" )
{
   // --- read the available BDT weight files

   // destroy all open cavases
   TMVAGlob::DestroyCanvases(); 

   // checks if file with name "fin" is already open, and if not opens one
   TFile* file = TMVAGlob::OpenFile( fin );  

   TDirectory* dir = file->GetDirectory( "Method_BDT" );
   if (!dir) {
      cout << "*** Error in macro \"BDT.C\": cannot find directory \"Method_BDT\" in file: " << fin << endl;
      return;
   }

   // read all directories
   TIter next( dir->GetListOfKeys() );
   TKey *key(0);   
   std::vector<TString> methname;   
   std::vector<TString> path;   
   std::vector<TString> wfile;   
   while ((key = (TKey*)next())) {
      TDirectory* mdir = dir->GetDirectory( key->GetName() );
      if (!mdir) {
         cout << "*** Error in macro \"BDT.C\": cannot find sub-directory: " << key->GetName() 
              << " in directory: " << dir->GetName() << endl;
         return;
      }

      // retrieve weight file name and path
      TObjString* strPath = (TObjString*)mdir->Get( "TrainingPath" );
      TObjString* strWFile = (TObjString*)mdir->Get( "WeightFileName" );
      if (!strPath || !strWFile) {
         cout << "*** Error in macro \"BDT.C\": could not find TObjStrings \"TrainingPath\" and/or \"WeightFileName\" *** " << endl;
         cout << "*** Maybe you are using TMVA >= 3.8.15 with an older training target file ? *** " << endl;
         return;
      }

      methname.push_back( key->GetName() );
      path    .push_back( strPath->GetString() );
      wfile   .push_back( strWFile->GetString() );
   }

   // create the control bar
   TControlBar* cbar = new TControlBar( "vertical", "Choose weight file:", 50, 50 );
   BDT_Global__cbar.push_back(cbar);

   for (UInt_t im=0; im<path.size(); im++) {
      TString fname = path[im];
      if (fname[fname.Length()-1] != '/') fname += "/";
      fname += wfile[im];
      TString macro = Form( ".x BDT.C+\(0,\"%s\",\"%s\")", fname.Data(), methname[im].Data() );
      cbar->AddButton( fname, macro, "Plot decision trees from this weight file", "button" );
   }

   // *** problems with this button in ROOT 5.19 ***
   #if ROOT_VERSION_CODE < ROOT_VERSION(5,19,0)
   cbar->AddButton( "Close", Form("BDT_DeleteTBar(%i)", BDT_Global__cbar.size()-1), "Close this control bar", "button" );
   #endif
   // **********************************************

   // set the style 
   cbar->SetTextColor("blue");

   // draw
   cbar->Show();   
}

void BDT_DeleteTBar(int i)
{
   // destroy all open canvases
   StatDialogBDT::Delete();
   TMVAGlob::DestroyCanvases();

   delete BDT_Global__cbar[i];
   BDT_Global__cbar[i] = 0;
}

// input: - No. of tree
//        - the weight file from which the tree is read
void BDT( Int_t itree, TString wfile = "weights/TMVAnalysis_test_BDT.weights.txt", TString methName = "BDT", Bool_t useTMVAStyle = kTRUE ) 
{
   // destroy possibly existing dialog windows and/or canvases
   StatDialogBDT::Delete();
   TMVAGlob::DestroyCanvases(); 

   // quick check if weight file exist
   if(!wfile.EndsWith(".xml") ){
      ifstream fin( wfile );
      if (!fin.good( )) { // file not found --> Error
         cout << "*** ERROR: Weight file: " << wfile << " does not exist" << endl;
         return;
      }
   }
   std::cout << "test1";
   // set style and remove existing canvas'
   TMVAGlob::Initialize( useTMVAStyle );

   StatDialogBDT* gGui = new StatDialogBDT( gClient->GetRoot(), wfile, methName, itree );

   gGui->DrawTree( itree );

   gGui->RaiseDialog();
}

Revision:	1.1
Committed:	Mon Feb 14 12:39:14 2011 UTC (14 years, 3 months ago) by benhoob
Content type:	text/plain
Branch:	MAIN
CVS Tags:	HEAD
Error occurred while calculating annotation data.
Log Message:	Initial commit
#	Content
1	#include <iostream>
2	#include <iomanip>
3	#include <fstream>
4
5	#include "tmvaglob.C"
6
7	#include "RQ_OBJECT.h"
8
9	#include "TROOT.h"
10	#include "TStyle.h"
11	#include "TPad.h"
12	#include "TCanvas.h"
13	#include "TLine.h"
14	#include "TFile.h"
15	#include "TColor.h"
16	#include "TPaveText.h"
17	#include "TObjString.h"
18	#include "TControlBar.h"
19
20	#include "TGWindow.h"
21	#include "TGButton.h"
22	#include "TGLabel.h"
23	#include "TGNumberEntry.h"
24
25	#include "TMVA/DecisionTree.h"
26	#include "TMVA/Tools.h"
27	#include "TXMLEngine.h"
28
29	// Uncomment this only if the link problem is solved. The include statement tends
30	// to use the ROOT classes rather than the local TMVA release
31	// #include "TMVA/DecisionTree.h"
32	// #include "TMVA/DecisionTreeNode.h"
33
34	// this macro displays a decision tree read in from the weight file
35
36	static const Int_t kSigColorF = TColor::GetColor( "#2244a5" ); // novel blue
37	static const Int_t kBkgColorF = TColor::GetColor( "#dd0033" ); // novel red
38	static const Int_t kIntColorF = TColor::GetColor( "#33aa77" ); // novel green
39
40	static const Int_t kSigColorT = 10;
41	static const Int_t kBkgColorT = 10;
42	static const Int_t kIntColorT = 10;
43
44	enum PlotType { EffPurity = 0 };
45
46	class StatDialogBDT {
47
48	RQ_OBJECT("StatDialogBDT")
49
50	public:
51
52	StatDialogBDT( const TGWindow* p, TString wfile = "weights/TMVAClassification_BDT.weights.txt",
53	TString methName = "BDT", Int_t itree = 0 );
54	virtual ~StatDialogBDT() {
55	TMVA::DecisionTreeNode::fgIsTraining=false;
56	fThis = 0;
57	fMain->CloseWindow();
58	fMain->Cleanup();
59	if(gROOT->GetListOfCanvases()->FindObject(fCanvas))
60	delete fCanvas;
61	}
62
63	// draw method
64	void DrawTree( Int_t itree );
65
66	void RaiseDialog() { if (fMain) { fMain->RaiseWindow(); fMain->Layout(); fMain->MapWindow(); } }
67
68	private:
69
70	TGMainFrame *fMain;
71	Int_t fItree;
72	Int_t fNtrees;
73	TCanvas* fCanvas;
74
75	TGNumberEntry* fInput;
76
77	TGHorizontalFrame* fButtons;
78	TGTextButton* fDrawButton;
79	TGTextButton* fCloseButton;
80
81	void UpdateCanvases();
82
83	// draw methods
84	TMVA::DecisionTree* ReadTree( TString * &vars, Int_t itree );
85	void DrawNode( TMVA::DecisionTreeNode *n,
86	Double_t x, Double_t y, Double_t xscale, Double_t yscale, TString* vars );
87	void GetNtrees();
88
89	TString fWfile;
90	TString fMethName;
91
92	public:
93
94	// static function for external deletion
95	static void Delete() { if (fThis != 0) { delete fThis; fThis = 0; } }
96
97	// slots
98	void SetItree(); //SIGNAL
99	void Redraw(); //SIGNAL
100	void Close(); //SIGNAL
101
102	private:
103
104	static StatDialogBDT* fThis;
105
106	};
107
108	StatDialogBDT* StatDialogBDT::fThis = 0;
109
110	void StatDialogBDT::SetItree()
111	{
112	fItree = Int_t(fInput->GetNumber());
113	}
114
115	void StatDialogBDT::Redraw()
116	{
117	UpdateCanvases();
118	}
119
120	void StatDialogBDT::Close()
121	{
122	delete this;
123	}
124
125	StatDialogBDT::StatDialogBDT( const TGWindow* p, TString wfile, TString methName, Int_t itree )
126	: fMain( 0 ),
127	fItree(itree),
128	fNtrees(0),
129	fCanvas(0),
130	fInput(0),
131	fButtons(0),
132	fDrawButton(0),
133	fCloseButton(0),
134	fWfile( wfile ),
135	fMethName( methName )
136	{
137	UInt_t totalWidth = 500;
138	UInt_t totalHeight = 200;
139
140	fThis = this;
141
142	TMVA::DecisionTreeNode::fgIsTraining=true;
143
144	// read number of decision trees from weight file
145	GetNtrees();
146
147	// main frame
148	fMain = new TGMainFrame(p, totalWidth, totalHeight, kMainFrame \| kVerticalFrame);
149
150	TGLabel *sigLab = new TGLabel( fMain, Form( "Decision tree [%i-%i]",0,fNtrees-1 ) );
151	fMain->AddFrame(sigLab, new TGLayoutHints(kLHintsLeft \| kLHintsTop,5,5,5,5));
152
153	fInput = new TGNumberEntry(fMain, (Double_t) fItree,5,-1,(TGNumberFormat::EStyle) 5);
154	fMain->AddFrame(fInput, new TGLayoutHints(kLHintsLeft \| kLHintsTop,5,5,5,5));
155	fInput->Resize(100,24);
156	fInput->SetLimits(TGNumberFormat::kNELLimitMinMax,0,fNtrees-1);
157
158	fButtons = new TGHorizontalFrame(fMain, totalWidth,30);
159
160	fCloseButton = new TGTextButton(fButtons,"&Close");
161	fButtons->AddFrame(fCloseButton, new TGLayoutHints(kLHintsLeft \| kLHintsTop));
162
163	fDrawButton = new TGTextButton(fButtons,"&Draw");
164	fButtons->AddFrame(fDrawButton, new TGLayoutHints(kLHintsRight \| kLHintsTop,15));
165
166	fMain->AddFrame(fButtons,new TGLayoutHints(kLHintsLeft \| kLHintsBottom,5,5,5,5));
167
168	fMain->SetWindowName("Decision tree");
169	fMain->SetWMPosition(0,0);
170	fMain->MapSubwindows();
171	fMain->Resize(fMain->GetDefaultSize());
172	fMain->MapWindow();
173
174	fInput->Connect("ValueSet(Long_t)","StatDialogBDT",this, "SetItree()");
175
176	fDrawButton->Connect("Clicked()","TGNumberEntry",fInput, "ValueSet(Long_t)");
177	fDrawButton->Connect("Clicked()", "StatDialogBDT", this, "Redraw()");
178
179	fCloseButton->Connect("Clicked()", "StatDialogBDT", this, "Close()");
180	}
181
182	void StatDialogBDT::UpdateCanvases()
183	{
184	DrawTree( fItree );
185	}
186
187	void StatDialogBDT::GetNtrees()
188	{
189	if(!fWfile.EndsWith(".xml") ){
190	ifstream fin( fWfile );
191	if (!fin.good( )) { // file not found --> Error
192	cout << "*** ERROR: Weight file: " << fWfile << " does not exist" << endl;
193	return;
194	}
195
196	TString dummy = "";
197
198	// read total number of trees, and check whether requested tree is in range
199	Int_t nc = 0;
200	while (!dummy.Contains("NTrees")) {
201	fin >> dummy;
202	nc++;
203	if (nc > 200) {
204	cout << endl;
205	cout << "*** Huge problem: could not locate term \"NTrees\" in BDT weight file: "
206	<< fWfile << endl;
207	cout << "==> panic abort (please contact the TMVA authors)" << endl;
208	cout << endl;
209	exit(1);
210	}
211	}
212	fin >> dummy;
213	fNtrees = dummy.ReplaceAll("\"","").Atoi();
214	fin.close();
215	}
216	else{
217	void* doc = TMVA::gTools().xmlengine().ParseFile(fWfile);
218	void* rootnode = TMVA::gTools().xmlengine().DocGetRootElement(doc);
219	void* ch = TMVA::gTools().xmlengine().GetChild(rootnode);
220	while(ch){
221	TString nodeName = TString( TMVA::gTools().xmlengine().GetNodeName(ch) );
222	if(nodeName=="Weights") {
223	TMVA::gTools().ReadAttr( ch, "NTrees", fNtrees );
224	break;
225	}
226	ch = TMVA::gTools().xmlengine().GetNext(ch);
227	}
228	}
229	cout << "--- Found " << fNtrees << " decision trees in weight file" << endl;
230
231	}
232
233	//_______________________________________________________________________
234	void StatDialogBDT::DrawNode( TMVA::DecisionTreeNode *n,
235	Double_t x, Double_t y,
236	Double_t xscale, Double_t yscale, TString * vars)
237	{
238	// recursively puts an entries in the histogram for the node and its daughters
239	//
240	Float_t xsize=xscale*1.5;
241	Float_t ysize=yscale/3;
242	if (xsize>0.15) xsize=0.1; //xscale/2;
243	if (n->GetLeft() != NULL){
244	TLine a1 = new TLine(x-xscale/4,y-ysize,x-xscale,y-ysize2);
245	a1->SetLineWidth(2);
246	a1->Draw();
247	DrawNode((TMVA::DecisionTreeNode*) n->GetLeft(), x-xscale, y-yscale, xscale/2, yscale, vars);
248	}
249	if (n->GetRight() != NULL){
250	TLine a1 = new TLine(x+xscale/4,y-ysize,x+xscale,y-ysize2);
251	a1->SetLineWidth(2);
252	a1->Draw();
253	DrawNode((TMVA::DecisionTreeNode*) n->GetRight(), x+xscale, y-yscale, xscale/2, yscale, vars );
254	}
255
256	// TPaveText *t = new TPaveText(x-xscale/2,y-yscale/2,x+xscale/2,y+yscale/2, "NDC");
257	TPaveText *t = new TPaveText(x-xsize,y-ysize,x+xsize,y+ysize, "NDC");
258
259	t->SetBorderSize(1);
260
261	t->SetFillStyle(1);
262	if (n->GetNodeType() == 1) { t->SetFillColor( kSigColorF ); t->SetTextColor( kSigColorT ); }
263	else if (n->GetNodeType() == -1) { t->SetFillColor( kBkgColorF ); t->SetTextColor( kBkgColorT ); }
264	else if (n->GetNodeType() == 0) { t->SetFillColor( kIntColorF ); t->SetTextColor( kIntColorT ); }
265
266	char buffer[25];
267	sprintf( buffer, "N=%f", n->GetNEvents() );
268	if (n->GetNEvents()>0) t->AddText(buffer);
269	sprintf( buffer, "S/(S+B)=%4.3f", n->GetPurity() );
270	t->AddText(buffer);
271
272	if (n->GetNodeType() == 0){
273	if (n->GetCutType()){
274	t->AddText(TString(vars[n->GetSelector()]+">"+=::Form("%5.3g",n->GetCutValue())));
275	}else{
276	t->AddText(TString(vars[n->GetSelector()]+"<"+=::Form("%5.3g",n->GetCutValue())));
277	}
278	}
279
280	t->Draw();
281
282	return;
283	}
284	TMVA::DecisionTree* StatDialogBDT::ReadTree( TString* &vars, Int_t itree )
285	{
286	cout << "--- Reading Tree " << itree << " from weight file: " << fWfile << endl;
287	TMVA::DecisionTree *d = new TMVA::DecisionTree();
288	if(!fWfile.EndsWith(".xml") ){
289	ifstream fin( fWfile );
290	if (!fin.good( )) { // file not found --> Error
291	cout << "*** ERROR: Weight file: " << fWfile << " does not exist" << endl;
292	return 0;
293	}
294
295	TString dummy = "";
296
297	if (itree >= fNtrees) {
298	cout << "*** ERROR: requested decision tree: " << itree
299	<< ", but number of trained trees only: " << fNtrees << endl;
300	return 0;
301	}
302
303	// file header with name
304	while (!dummy.Contains("#VAR")) fin >> dummy;
305	fin >> dummy >> dummy >> dummy; // the rest of header line
306
307	// number of variables
308	Int_t nVars;
309	fin >> dummy >> nVars;
310
311	// variable mins and maxes
312	vars = new TString[nVars+1]; // last one is if "fisher cut criterium"
313	for (Int_t i = 0; i < nVars; i++) fin >> vars[i] >> dummy >> dummy >> dummy >> dummy;
314	vars[nVars]="FisherCrit";
315
316	char buffer[20];
317	char line[256];
318	sprintf(buffer,"Tree %d",itree);
319
320	while (!dummy.Contains(buffer)) {
321	fin.getline(line,256);
322	dummy = TString(line);
323	}
324
325	d->Read(fin);
326
327	fin.close();
328	}
329	else{
330	if (itree >= fNtrees) {
331	cout << "*** ERROR: requested decision tree: " << itree
332	<< ", but number of trained trees only: " << fNtrees << endl;
333	return 0;
334	}
335	Int_t nVars;
336	void* doc = TMVA::gTools().xmlengine().ParseFile(fWfile);
337	void* rootnode = TMVA::gTools().xmlengine().DocGetRootElement(doc);
338	void* ch = TMVA::gTools().xmlengine().GetChild(rootnode);
339	while(ch){
340	TString nodeName = TString( TMVA::gTools().xmlengine().GetNodeName(ch) );
341	if(nodeName=="Variables"){
342	TMVA::gTools().ReadAttr( ch, "NVar", nVars);
343	vars = new TString[nVars+1];
344	void* varnode = TMVA::gTools().xmlengine().GetChild(ch);
345	for (Int_t i = 0; i < nVars; i++){
346	TMVA::gTools().ReadAttr( varnode, "Expression", vars[i]);
347	varnode = TMVA::gTools().xmlengine().GetNext(varnode);
348	}
349	vars[nVars]="FisherCrit";
350	}
351	if(nodeName=="Weights") break;
352	ch = TMVA::gTools().xmlengine().GetNext(ch);
353	}
354	ch = TMVA::gTools().xmlengine().GetChild(ch);
355	for (int i=0; i<itree; i++) ch = TMVA::gTools().xmlengine().GetNext(ch);
356	d->ReadXML(ch);
357	}
358	return d;
359	}
360
361	//_______________________________________________________________________
362	void StatDialogBDT::DrawTree( Int_t itree )
363	{
364	TString *vars;
365	TMVA::DecisionTree* d = ReadTree( vars, itree );
366	if (d == 0) return;
367
368	UInt_t depth = d->GetTotalTreeDepth();
369	Double_t ystep = 1.0/(depth + 1.0);
370
371	cout << "--- Tree depth: " << depth << endl;
372
373	TStyle* TMVAStyle = gROOT->GetStyle("Plain"); // our style is based on Plain
374	Int_t canvasColor = TMVAStyle->GetCanvasColor(); // backup
375
376	TString cbuffer = Form( "Reading weight file: %s", fWfile.Data() );
377	TString tbuffer = Form( "Decision Tree no.: %d", itree );
378	if (!fCanvas) fCanvas = new TCanvas( "c1", cbuffer, 200, 0, 1000, 600 );
379	else fCanvas->Clear();
380	fCanvas->Draw();
381
382	DrawNode( (TMVA::DecisionTreeNode)d->GetRoot(), 0.5, 1.-0.5ystep, 0.25, ystep ,vars);
383
384	// make the legend
385	Double_t yup=0.99;
386	Double_t ydown=yup-ystep/2.5;
387	Double_t dy= ystep/2.5 * 0.2;
388
389	TPaveText *whichTree = new TPaveText(0.85,ydown,0.98,yup, "NDC");
390	whichTree->SetBorderSize(1);
391	whichTree->SetFillStyle(1);
392	whichTree->SetFillColor( TColor::GetColor( "#ffff33" ) );
393	whichTree->AddText( tbuffer );
394	whichTree->Draw();
395
396	TPaveText *intermediate = new TPaveText(0.02,ydown,0.15,yup, "NDC");
397	intermediate->SetBorderSize(1);
398	intermediate->SetFillStyle(1);
399	intermediate->SetFillColor( kIntColorF );
400	intermediate->AddText("Intermediate Nodes");
401	intermediate->SetTextColor( kIntColorT );
402	intermediate->Draw();
403
404	ydown = ydown - ystep/2.5 -dy;
405	yup = yup - ystep/2.5 -dy;
406	TPaveText *signalleaf = new TPaveText(0.02,ydown ,0.15,yup, "NDC");
407	signalleaf->SetBorderSize(1);
408	signalleaf->SetFillStyle(1);
409	signalleaf->SetFillColor( kSigColorF );
410	signalleaf->AddText("Signal Leaf Nodes");
411	signalleaf->SetTextColor( kSigColorT );
412	signalleaf->Draw();
413
414	ydown = ydown - ystep/2.5 -dy;
415	yup = yup - ystep/2.5 -dy;
416	TPaveText *backgroundleaf = new TPaveText(0.02,ydown,0.15,yup, "NDC");
417	backgroundleaf->SetBorderSize(1);
418	backgroundleaf->SetFillStyle(1);
419	backgroundleaf->SetFillColor( kBkgColorF );
420
421	backgroundleaf->AddText("Backgr. Leaf Nodes");
422	backgroundleaf->SetTextColor( kBkgColorT );
423	backgroundleaf->Draw();
424
425	fCanvas->Update();
426	TString fname = Form("plots/%s_%i", fMethName.Data(), itree );
427	cout << "--- Creating image: " << fname << endl;
428	TMVAGlob::imgconv( fCanvas, fname );
429
430	TMVAStyle->SetCanvasColor( canvasColor );
431	}
432
433	// ========================================================================================
434
435	static std::vector<TControlBar*> BDT_Global__cbar;
436
437	// intermediate GUI
438	void BDT( const TString& fin = "TMVA.root" )
439	{
440	// --- read the available BDT weight files
441
442	// destroy all open cavases
443	TMVAGlob::DestroyCanvases();
444
445	// checks if file with name "fin" is already open, and if not opens one
446	TFile* file = TMVAGlob::OpenFile( fin );
447
448	TDirectory* dir = file->GetDirectory( "Method_BDT" );
449	if (!dir) {
450	cout << "*** Error in macro \"BDT.C\": cannot find directory \"Method_BDT\" in file: " << fin << endl;
451	return;
452	}
453
454	// read all directories
455	TIter next( dir->GetListOfKeys() );
456	TKey *key(0);
457	std::vector<TString> methname;
458	std::vector<TString> path;
459	std::vector<TString> wfile;
460	while ((key = (TKey*)next())) {
461	TDirectory* mdir = dir->GetDirectory( key->GetName() );
462	if (!mdir) {
463	cout << "*** Error in macro \"BDT.C\": cannot find sub-directory: " << key->GetName()
464	<< " in directory: " << dir->GetName() << endl;
465	return;
466	}
467
468	// retrieve weight file name and path
469	TObjString* strPath = (TObjString*)mdir->Get( "TrainingPath" );
470	TObjString* strWFile = (TObjString*)mdir->Get( "WeightFileName" );
471	if (!strPath \|\| !strWFile) {
472	cout << "* Error in macro \"BDT.C\": could not find TObjStrings \"TrainingPath\" and/or \"WeightFileName\" * " << endl;
473	cout << "* Maybe you are using TMVA >= 3.8.15 with an older training target file ? * " << endl;
474	return;
475	}
476
477	methname.push_back( key->GetName() );
478	path .push_back( strPath->GetString() );
479	wfile .push_back( strWFile->GetString() );
480	}
481
482	// create the control bar
483	TControlBar* cbar = new TControlBar( "vertical", "Choose weight file:", 50, 50 );
484	BDT_Global__cbar.push_back(cbar);
485
486	for (UInt_t im=0; im<path.size(); im++) {
487	TString fname = path[im];
488	if (fname[fname.Length()-1] != '/') fname += "/";
489	fname += wfile[im];
490	TString macro = Form( ".x BDT.C+\(0,\"%s\",\"%s\")", fname.Data(), methname[im].Data() );
491	cbar->AddButton( fname, macro, "Plot decision trees from this weight file", "button" );
492	}
493
494	// * problems with this button in ROOT 5.19 *
495	#if ROOT_VERSION_CODE < ROOT_VERSION(5,19,0)
496	cbar->AddButton( "Close", Form("BDT_DeleteTBar(%i)", BDT_Global__cbar.size()-1), "Close this control bar", "button" );
497	#endif
498	// **********************************************
499
500	// set the style
501	cbar->SetTextColor("blue");
502
503	// draw
504	cbar->Show();
505	}
506
507	void BDT_DeleteTBar(int i)
508	{
509	// destroy all open canvases
510	StatDialogBDT::Delete();
511	TMVAGlob::DestroyCanvases();
512
513	delete BDT_Global__cbar[i];
514	BDT_Global__cbar[i] = 0;
515	}
516
517	// input: - No. of tree
518	// - the weight file from which the tree is read
519	void BDT( Int_t itree, TString wfile = "weights/TMVAnalysis_test_BDT.weights.txt", TString methName = "BDT", Bool_t useTMVAStyle = kTRUE )
520	{
521	// destroy possibly existing dialog windows and/or canvases
522	StatDialogBDT::Delete();
523	TMVAGlob::DestroyCanvases();
524
525	// quick check if weight file exist
526	if(!wfile.EndsWith(".xml") ){
527	ifstream fin( wfile );
528	if (!fin.good( )) { // file not found --> Error
529	cout << "*** ERROR: Weight file: " << wfile << " does not exist" << endl;
530	return;
531	}
532	}
533	std::cout << "test1";
534	// set style and remove existing canvas'
535	TMVAGlob::Initialize( useTMVAStyle );
536
537	StatDialogBDT* gGui = new StatDialogBDT( gClient->GetRoot(), wfile, methName, itree );
538
539	gGui->DrawTree( itree );
540
541	gGui->RaiseDialog();
542	}
543