ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/LJMet/MultivariateAnalysis/root/make_training_sample.C
Revision: 1.4
Committed: Fri Apr 10 19:54:02 2009 UTC (16 years, 1 month ago) by kukartse
Content type: text/plain
Branch: MAIN
Changes since 1.3: +15 -17 lines
Log Message:
*** empty log message ***

File Contents

# User Rev Content
1 kukartse 1.1 {
2 kukartse 1.4 TChain chain("mva/ttljets");
3     chain . Add("tjets_muon-08apr2009.root");
4 kukartse 1.1
5 kukartse 1.4 TChain chain2("mva/ttljets");
6     chain2 . Add("wjets_muon-08apr2009.root");
7 kukartse 1.1
8 kukartse 1.4 TChain chain21("mva/ttljets");
9     chain21 . Add("zjets_muon-08apr2009.root");
10 kukartse 1.1
11 kukartse 1.4 TChain chain3("mva/ttljets");
12     chain3 . Add("qcd_muon-08apr2009.root");
13 kukartse 1.1
14     // selection
15 kukartse 1.2 #include "cuts.C"
16 kukartse 1.4 the_cut = cut3;
17     TCut the_qcd_cut = cut3_rev;
18 kukartse 1.1
19     // when copying, include all branches
20     chain . SetBranchStatus("*",1);
21     chain2 . SetBranchStatus("*",1);
22     chain21 . SetBranchStatus("*",1);
23 kukartse 1.3 chain3 . SetBranchStatus("*",1);
24 kukartse 1.1
25     // create new TTrees from the input
26     TTree * t_sig = chain . CopyTree(the_cut);
27     TTree * t_wjets = chain2 . CopyTree(the_cut);
28     TTree * t_zjets = chain21 . CopyTree(the_cut);
29 kukartse 1.3 TTree * t_qcd = chain3 . CopyTree(the_qcd_cut);
30 kukartse 1.4 TTree * _sig = t_sig->CopyTree("","",52630,330);
31     TTree * _wjets = t_wjets->CopyTree("","",1728,151);
32     TTree * _zjets = t_zjets->CopyTree("","",394,27);
33 kukartse 1.1 _wjets -> CopyAddresses(_zjets);
34     _wjets -> CopyEntries(_zjets);
35 kukartse 1.4 TTree * _qcd = t_qcd->CopyTree("","",36497,0);
36 kukartse 1.1
37    
38     // define new TTree names
39     _sig.SetName("ttbar");
40     _wjets.SetName("wjets");
41 kukartse 1.3 _qcd.SetName("qcd");
42 kukartse 1.1 //_sig->Print();
43     //_wjets->Print();
44     //_qcd->Print();
45    
46     // output file
47 kukartse 1.4 TFile * out_file = new TFile("/uscms_data/d1/lpcljm/MVA/Summer08/training/muon_jets_training-summer08-08apr2009.root", "RECREATE");
48 kukartse 1.3 //TFile * out_file = new TFile("./tmva_training-summer08-25feb2009.root", "RECREATE");
49 kukartse 1.1 out_file->cd();
50     _sig->Write();
51     _wjets->Write();
52 kukartse 1.3 _qcd->Write();
53 kukartse 1.1
54     out_file->Write();
55     delete out_file;
56     }
57