MultivariateAnalysis/root/add_classifier_branch.C

{
  gROOT -> Reset();
  gROOT -> Clear();
  
  TFile *file1 = new TFile("muon_jets_fake_data_1fb-summer08-13jun2009.root");
  TFile *file2 = new TFile("TMVApp-data-1fb-14jun2009.root");

  TTree *tree1 = (TTree*)file1 -> Get("data");
  TTree *tree2 = (TTree*)file2 -> Get("classifier");

  tree1->SetBranchStatus("*",1);
  tree2->SetBranchStatus("*",0);
  tree2->SetBranchStatus("MVA_BDT",1);

  Int_t nEntries = tree1 -> GetEntries();
  if (tree2 -> GetEntries() < nEntries) nEntries = tree2 -> GetEntries();
  cout << "The smallest tree has " << nEntries << " entries" << endl;

  //
  //_____ output file ___________________________________________________
  //
  TFile *newfile = new TFile("ljets-data-1fb-classifier-14jun2009.root","recreate");
  
  //TTree *interface = new TTree("interface", "dummy Title"); // empty tree to attach friends to
  //for (Int_t i=0;i<nEntries;i++) interface -> Fill();

  TTree *newtree1 = tree1 -> CloneTree();
  TTree *newtree2 = tree2 -> CloneTree();

  newtree1 -> SetName("data");
  newtree2 -> SetName("classifier");

  //===> add friends to THE SMALLEST TREE !!!
  //interface -> AddFriend( newtree1 );
  //interface -> AddFriend( newtree2 );
  newtree1 -> AddFriend( newtree2 );

  newfile->Write();

}
Revision:	1.1
Committed:	Thu Jun 18 16:46:55 2009 UTC (15 years, 10 months ago) by kukartse
Content type:	text/plain
Branch:	MAIN
CVS Tags:	V00-03-01, ZMorph_BASE_20100408, gak040610_morphing, V00-02-02, gak011410, gak010310, ejterm2010_25nov2009, V00-02-01, V00-02-00, gak112409, CMSSW_22X_branch_base, segala101609, V00-01-15, HEAD
Branch point for:	ZMorph-V00-03-01, CMSSW_22X_branch
Log Message:	* empty log message *
#	Content
1	{
2	gROOT -> Reset();
3	gROOT -> Clear();
4
5	TFile *file1 = new TFile("muon_jets_fake_data_1fb-summer08-13jun2009.root");
6	TFile *file2 = new TFile("TMVApp-data-1fb-14jun2009.root");
7
8	TTree tree1 = (TTree)file1 -> Get("data");
9	TTree tree2 = (TTree)file2 -> Get("classifier");
10
11	tree1->SetBranchStatus("*",1);
12	tree2->SetBranchStatus("*",0);
13	tree2->SetBranchStatus("MVA_BDT",1);
14
15	Int_t nEntries = tree1 -> GetEntries();
16	if (tree2 -> GetEntries() < nEntries) nEntries = tree2 -> GetEntries();
17	cout << "The smallest tree has " << nEntries << " entries" << endl;
18
19	//
20	//_____ output file ___________________________________________________
21	//
22	TFile *newfile = new TFile("ljets-data-1fb-classifier-14jun2009.root","recreate");
23
24	//TTree *interface = new TTree("interface", "dummy Title"); // empty tree to attach friends to
25	//for (Int_t i=0;i<nEntries;i++) interface -> Fill();
26
27	TTree *newtree1 = tree1 -> CloneTree();
28	TTree *newtree2 = tree2 -> CloneTree();
29
30	newtree1 -> SetName("data");
31	newtree2 -> SetName("classifier");
32
33	//===> add friends to THE SMALLEST TREE !!!
34	//interface -> AddFriend( newtree1 );
35	//interface -> AddFriend( newtree2 );
36	newtree1 -> AddFriend( newtree2 );
37
38	newfile->Write();
39
40	}