MultivariateAnalysis/root/make_training_sample_electron.C

{
  TChain chain("ttljets");
  chain . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Mar_29_2009/PAT_Layer1/TTJets-madgraph_Fall08_IDEAL_V9_v2_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");

  TChain chain2("ttljets");
  chain2 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Mar_29_2009/PAT_Layer1/Wjets-madgraph_Winter09_IDEAL_V11_FastSim_v1_GEN-SIM-DIGI-RECO/secondary/May_14/ejets_dr0.3/all.root");

  TChain chain21("ttljets");
  chain21 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Mar_29_2009/PAT_Layer1/Zjets-madgraph_Winter09_IDEAL_V11_FastSim_v1_GEN-SIM-DIGI-RECO/secondary/May_14/ejets_dr0.3/all.root");

  TChain chain_qcd1("ttljets");
  chain_qcd1 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_BCtoE_Pt20to30_Summer08_IDEAL_V11_redigi_v1_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
  TChain chain_qcd2("ttljets");
  chain_qcd2 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_BCtoE_Pt30to80_Summer08_IDEAL_V11_redigi_v1_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
  TChain chain_qcd3("ttljets");
  chain_qcd3 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_BCtoE_Pt80to170_Summer08_IDEAL_V11_redigi_v1_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
  TChain chain_qcd4("ttljets");
  chain_qcd4 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_EMenriched_Pt20to30_Summer08_IDEAL_V11_redigi_v2_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
  TChain chain_qcd5("ttljets");
  chain_qcd5 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_EMenriched_Pt30to80_Summer08_IDEAL_V11_redigi_v2_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
  TChain chain_qcd6("ttljets");
  chain_qcd6 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_EMenriched_Pt80to170_Summer08_IDEAL_V11_redigi_v1_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
  TChain chain_s("ttljets");
   chain_s . Add("/uscms_data/d2/lpcljm/MVA/Summer08/May_14_2009/PAT_Layer1/SingleTop_sChannel_Summer08_IDEAL_V9_v1_GEN-SIM-RECO/secondary/ejets_14may2009/all.root");
  TChain chain_t("ttljets");
  chain_t . Add("/uscms_data/d2/lpcljm/MVA/Summer08/May_14_2009/PAT_Layer1/SingleTop_tChannel_Summer08_IDEAL_V9_v1_GEN-SIM-RECO/secondary/ejets_14may2009/all.root");
  TChain chain_tW("ttljets");
  chain_tW . Add("/uscms_data/d2/lpcljm/MVA/Summer08/May_14_2009/PAT_Layer1/SingleTop_tWChannel_Summer08_IDEAL_V9_v1_GEN-SIM-RECO/secondary/ejets_14may2009/all.root");


// selection
  #include "cuts.C"
  the_cut = cut4_electrons;
  TCut the_qcd_cut = cut4_rev_electrons;

  // when copying, include all branches
  chain . SetBranchStatus("*",1);
  chain2 . SetBranchStatus("*",1);
  chain21 . SetBranchStatus("*",1);
  chain_qcd1 . SetBranchStatus("*",1);
  chain_qcd2 . SetBranchStatus("*",1);
  chain_qcd3 . SetBranchStatus("*",1);
  chain_qcd4 . SetBranchStatus("*",1);
  chain_qcd5 . SetBranchStatus("*",1);
  chain_qcd6 . SetBranchStatus("*",1);
  chain_s . SetBranchStatus("*",1);
  chain_t . SetBranchStatus("*",1);
  chain_tW . SetBranchStatus("*",1);
  // create new TTrees from the input
  TTree * t_sig = chain . CopyTree(the_cut);
  TTree * t_wjets = chain2 . CopyTree(the_cut);
  TTree * t_zjets = chain21 . CopyTree(the_cut);
  TTree * t_qcd1 = chain_qcd1 . CopyTree(the_qcd_cut);
  TTree * t_qcd2 = chain_qcd2 . CopyTree(the_qcd_cut);
  TTree * t_qcd3 = chain_qcd3 . CopyTree(the_qcd_cut);
  TTree * t_qcd4 = chain_qcd4 . CopyTree(the_qcd_cut);
  TTree * t_qcd5 = chain_qcd5 . CopyTree(the_qcd_cut);
  TTree * t_qcd6 = chain_qcd6 . CopyTree(the_qcd_cut);
  TTree * t_t = chain_t . CopyTree(the_cut);
  TTree * t_s = chain_s . CopyTree(the_cut);
  TTree * t_tW = chain_tW . CopyTree(the_cut);
  //
  TTree * _sig = t_sig->CopyTree("","",34161,281);
  TTree * _wjets = t_wjets->CopyTree("","",19299,0);
  TTree * _zjets = t_zjets->CopyTree("","",2913,0);
  TTree * _t  = t_t->CopyTree("","",98,2);
  TTree * _s  = t_s->CopyTree("","",6,0);
  TTree * _tW = t_tW->CopyTree("","",63,0);
  _wjets -> CopyAddresses(_zjets);
  _wjets -> CopyEntries(_zjets);
  _wjets -> CopyAddresses(_s);
  _wjets -> CopyEntries(_s);
  _wjets -> CopyAddresses(_t);
  _wjets -> CopyEntries(_t);
  _wjets -> CopyAddresses(_tW);
  _wjets -> CopyEntries(_tW);
  TTree * _qcd1 = t_qcd1->CopyTree("","",3,0);
  TTree * _qcd2 = t_qcd2->CopyTree("","",855,0);
  TTree * _qcd3 = t_qcd3->CopyTree("","",1287,0);
  TTree * _qcd4 = t_qcd4->CopyTree("","",65,0);
  TTree * _qcd5 = t_qcd5->CopyTree("","",7832,0);
  TTree * _qcd6 = t_qcd6->CopyTree("","",10092,0);
  _qcd1 -> CopyAddresses(_qcd2);
  _qcd1 ->   CopyEntries(_qcd2);
  _qcd1 -> CopyAddresses(_qcd3);
  _qcd1 ->   CopyEntries(_qcd3);
  _qcd1 -> CopyAddresses(_qcd4);
  _qcd1 ->   CopyEntries(_qcd4);
  _qcd1 -> CopyAddresses(_qcd5);
  _qcd1 ->   CopyEntries(_qcd5);
  _qcd1 -> CopyAddresses(_qcd6);
  _qcd1 ->   CopyEntries(_qcd6);


  // define new TTree names
  _sig.SetName("ttbar");
  _wjets.SetName("wzjets");
  _qcd1.SetName("qcd");
  //_sig->Print();
  //_wjets->Print();
  //_qcd->Print();

  // output file
  TFile * out_file = new TFile("/uscms_data/d1/lpcljm/MVA/Summer08/training/electron_jets_training-wzfastsim-summer08-19may2009.root", "RECREATE");
  out_file->cd();
  _sig->Write();
  _wjets->Write();
  _qcd1->Write();
  

  out_file->Write();
  delete out_file;
}
 
Revision:	1.3
Committed:	Thu May 21 16:23:37 2009 UTC (15 years, 11 months ago) by jindal
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
Changes since 1.2:	+41 -20 lines
Log Message:	make_fake_data_electron.C
#	User	Rev	Content
1	kukartse	1.1	{
2			TChain chain("ttljets");
3	jindal	1.3	chain . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Mar_29_2009/PAT_Layer1/TTJets-madgraph_Fall08_IDEAL_V9_v2_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
4	kukartse	1.1
5			TChain chain2("ttljets");
6	jindal	1.3	chain2 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Mar_29_2009/PAT_Layer1/Wjets-madgraph_Winter09_IDEAL_V11_FastSim_v1_GEN-SIM-DIGI-RECO/secondary/May_14/ejets_dr0.3/all.root");
7	kukartse	1.1
8			TChain chain21("ttljets");
9	jindal	1.3	chain21 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Mar_29_2009/PAT_Layer1/Zjets-madgraph_Winter09_IDEAL_V11_FastSim_v1_GEN-SIM-DIGI-RECO/secondary/May_14/ejets_dr0.3/all.root");
10	kukartse	1.1
11			TChain chain_qcd1("ttljets");
12	jindal	1.3	chain_qcd1 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_BCtoE_Pt20to30_Summer08_IDEAL_V11_redigi_v1_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
13	kukartse	1.1	TChain chain_qcd2("ttljets");
14	jindal	1.3	chain_qcd2 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_BCtoE_Pt30to80_Summer08_IDEAL_V11_redigi_v1_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
15	kukartse	1.1	TChain chain_qcd3("ttljets");
16	jindal	1.3	chain_qcd3 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_BCtoE_Pt80to170_Summer08_IDEAL_V11_redigi_v1_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
17	kukartse	1.1	TChain chain_qcd4("ttljets");
18	jindal	1.3	chain_qcd4 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_EMenriched_Pt20to30_Summer08_IDEAL_V11_redigi_v2_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
19	kukartse	1.1	TChain chain_qcd5("ttljets");
20	jindal	1.3	chain_qcd5 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_EMenriched_Pt30to80_Summer08_IDEAL_V11_redigi_v2_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
21	kukartse	1.1	TChain chain_qcd6("ttljets");
22	jindal	1.3	chain_qcd6 . Add("/uscms_data/d2/lpcljm/MVA/Summer08/Apr_17_2009/QCD_EMenriched_Pt80to170_Summer08_IDEAL_V11_redigi_v1_GEN-SIM-RECO/secondary/May_14/ejets_dr0.3/all.root");
23			TChain chain_s("ttljets");
24			chain_s . Add("/uscms_data/d2/lpcljm/MVA/Summer08/May_14_2009/PAT_Layer1/SingleTop_sChannel_Summer08_IDEAL_V9_v1_GEN-SIM-RECO/secondary/ejets_14may2009/all.root");
25			TChain chain_t("ttljets");
26			chain_t . Add("/uscms_data/d2/lpcljm/MVA/Summer08/May_14_2009/PAT_Layer1/SingleTop_tChannel_Summer08_IDEAL_V9_v1_GEN-SIM-RECO/secondary/ejets_14may2009/all.root");
27			TChain chain_tW("ttljets");
28			chain_tW . Add("/uscms_data/d2/lpcljm/MVA/Summer08/May_14_2009/PAT_Layer1/SingleTop_tWChannel_Summer08_IDEAL_V9_v1_GEN-SIM-RECO/secondary/ejets_14may2009/all.root");
29	kukartse	1.1
30	jindal	1.3
31			// selection
32	kukartse	1.1	#include "cuts.C"
33	kukartse	1.2	the_cut = cut4_electrons;
34			TCut the_qcd_cut = cut4_rev_electrons;
35	kukartse	1.1
36			// when copying, include all branches
37			chain . SetBranchStatus("*",1);
38			chain2 . SetBranchStatus("*",1);
39			chain21 . SetBranchStatus("*",1);
40			chain_qcd1 . SetBranchStatus("*",1);
41			chain_qcd2 . SetBranchStatus("*",1);
42			chain_qcd3 . SetBranchStatus("*",1);
43			chain_qcd4 . SetBranchStatus("*",1);
44			chain_qcd5 . SetBranchStatus("*",1);
45			chain_qcd6 . SetBranchStatus("*",1);
46	jindal	1.3	chain_s . SetBranchStatus("*",1);
47			chain_t . SetBranchStatus("*",1);
48			chain_tW . SetBranchStatus("*",1);
49	kukartse	1.1	// create new TTrees from the input
50			TTree * t_sig = chain . CopyTree(the_cut);
51			TTree * t_wjets = chain2 . CopyTree(the_cut);
52			TTree * t_zjets = chain21 . CopyTree(the_cut);
53			TTree * t_qcd1 = chain_qcd1 . CopyTree(the_qcd_cut);
54			TTree * t_qcd2 = chain_qcd2 . CopyTree(the_qcd_cut);
55			TTree * t_qcd3 = chain_qcd3 . CopyTree(the_qcd_cut);
56			TTree * t_qcd4 = chain_qcd4 . CopyTree(the_qcd_cut);
57			TTree * t_qcd5 = chain_qcd5 . CopyTree(the_qcd_cut);
58			TTree * t_qcd6 = chain_qcd6 . CopyTree(the_qcd_cut);
59	jindal	1.3	TTree * t_t = chain_t . CopyTree(the_cut);
60			TTree * t_s = chain_s . CopyTree(the_cut);
61			TTree * t_tW = chain_tW . CopyTree(the_cut);
62	kukartse	1.1	//
63	jindal	1.3	TTree * _sig = t_sig->CopyTree("","",34161,281);
64			TTree * _wjets = t_wjets->CopyTree("","",19299,0);
65			TTree * _zjets = t_zjets->CopyTree("","",2913,0);
66			TTree * _t = t_t->CopyTree("","",98,2);
67			TTree * _s = t_s->CopyTree("","",6,0);
68			TTree * _tW = t_tW->CopyTree("","",63,0);
69	kukartse	1.1	_wjets -> CopyAddresses(_zjets);
70			_wjets -> CopyEntries(_zjets);
71	jindal	1.3	_wjets -> CopyAddresses(_s);
72			_wjets -> CopyEntries(_s);
73			_wjets -> CopyAddresses(_t);
74			_wjets -> CopyEntries(_t);
75			_wjets -> CopyAddresses(_tW);
76			_wjets -> CopyEntries(_tW);
77	kukartse	1.1	TTree * _qcd1 = t_qcd1->CopyTree("","",3,0);
78	jindal	1.3	TTree * _qcd2 = t_qcd2->CopyTree("","",855,0);
79			TTree * _qcd3 = t_qcd3->CopyTree("","",1287,0);
80			TTree * _qcd4 = t_qcd4->CopyTree("","",65,0);
81			TTree * _qcd5 = t_qcd5->CopyTree("","",7832,0);
82			TTree * _qcd6 = t_qcd6->CopyTree("","",10092,0);
83	kukartse	1.1	_qcd1 -> CopyAddresses(_qcd2);
84			_qcd1 -> CopyEntries(_qcd2);
85			_qcd1 -> CopyAddresses(_qcd3);
86			_qcd1 -> CopyEntries(_qcd3);
87			_qcd1 -> CopyAddresses(_qcd4);
88			_qcd1 -> CopyEntries(_qcd4);
89			_qcd1 -> CopyAddresses(_qcd5);
90			_qcd1 -> CopyEntries(_qcd5);
91			_qcd1 -> CopyAddresses(_qcd6);
92			_qcd1 -> CopyEntries(_qcd6);
93
94
95			// define new TTree names
96			_sig.SetName("ttbar");
97			_wjets.SetName("wzjets");
98			_qcd1.SetName("qcd");
99			//_sig->Print();
100			//_wjets->Print();
101			//_qcd->Print();
102
103			// output file
104	jindal	1.3	TFile * out_file = new TFile("/uscms_data/d1/lpcljm/MVA/Summer08/training/electron_jets_training-wzfastsim-summer08-19may2009.root", "RECREATE");
105	kukartse	1.1	out_file->cd();
106			_sig->Write();
107			_wjets->Write();
108			_qcd1->Write();
109
110
111			out_file->Write();
112			delete out_file;
113			}
114