yangyong/zSelection/testSelection.h

int dataflag; 
int evtRange; 
int dataOrMC; 
int totalEntries; 
int entry; 

TChain *fChain; 

TTree *newtree; 
string dataversion;
string datasetname;

float detasctrkvtx;
float dphisctrkvtx;

float e1x5;


float photonet[nPhotonMAX];
float photonetvtx0[nPhotonMAX];
float photoncaloeta[nPhotonMAX];
float photoncalophi[nPhotonMAX];
int photoninbarrel[nPhotonMAX];

float electronecalEnergy[nElectronMAX];
float electronecalEnergyCorr[nElectronMAX];


TTree *evtInfo;

int totalNumberofEvents ; 
int preselectedNumberofEvents; 
vector<string> *datasettag; 

float EnergyScaleOffPhotonCat[4];
float EnergySmearingPhotonCat[4];

TRandom3 *rgen_;

std::map< int,std::vector<TH1*> > kFactors_;


///ID efficieny correction

std::map<std::string,TGraphAsymmErrors*> IDCorrection_; 

/* std::map<std::string, RooRealVar> m_real_var_; */
/* std::map<std::string, double> m_var_min_; */
/* std::map<std::string, double> m_var_max_; */
/* std::map<std::string,RooRealVar*> m_data_var_ptr_; */
/* std::map<std::string,RooRealVar*> m_weight_var_ptr_; */
/* std::map<std::string,RooDataSet> data_; */
/* std::map<std::string,std::string> data_obs_names_; */


 std::vector<float> ptunbal_;
std::vector<float> angleptunbaldipht_;
std::vector<float> asym_ptunbal_;

std::vector<float> ncha_;
std::vector<float> nchapt1_;


 std::vector<float> ptbal_;
  std::vector<float> thrust_;
  std::vector<float> sumpt_;
  std::vector<float> sumpt2_;
  std::vector<float> logsumpt2_;
  std::vector<float> sumawy_;
  std::vector<float> sumtwd_;
  std::vector<float> sumtrv_;
  std::vector<float> sumweight_;
  std::vector<float> ptmax_;
  std::vector<float> nchthr_;
  std::vector<float> nch_;
  std::vector<std::vector<float> > tksPt_;
  std::vector<TMatrixDSym> sphers_;
  std::vector<float> sumpr_;
  std::vector<float> spher_;
  std::vector<float> tspher_;
  std::vector<float> aplan_;
  std::vector<float> threejetC_;
  std::vector<float> fourjetD_;
  
  std::vector<TVector3> vtxP_;
  std::vector<TVector2> vtxPt_;
  std::vector<float> ptvtx_;
  std::vector<float> diphopt_; 
  std::vector<float> diPhotonPz_;
  std::vector<float> acosA_;
  std::vector<float> ptasym_;
  
  std::vector<float> ptmax3_;
  std::vector<float> ptratio_;
  std::vector<float> pzasym_;
  std::vector<float> awytwdasym_;
  std::vector<TVector2> diPhotonPt_;

  vector<int> indpresel; 
  vector<int> indaccept; 

std::vector<int> preselConv;


std::vector<int> rankprodvtx;

std::vector<int> tmvavtx;

  float ptbal;
float tspher;
float spher;
float aplan;
float threejetC;
float fourjetD;
float diphopt;
float acosA;
float ptasym;
float ptmax;
float ptmax3;
float thrust;
float ptratio;
float pzasym;
float sumwt;
float vtxpt;

float sumpt2;
float sumpt;
float awytwdasym;
 int preselectedbyconv; 
 int isgenmatched; 
 float ptunbal; 
 float angleptunbaldipht;
 float asym_ptunbal;
float ncha; 
float nchapt1; 
 
 float nchapt1f;
 
 float logsumpt2; 

float ptdiphtfirstvertex;


TMVA::Reader *reader; 
TMVA::Reader *reader2; 

TMVA::Reader *reader_eb_unconv;
TMVA::Reader *reader_eb_conv;
TMVA::Reader *reader_ee_unconv;
TMVA::Reader *reader_ee_conv;


TString methodName;

float tmva; 
float tmva2; 


int isbestvtx;
int isworstvtx;
float distvtxgen;

int testTMVA;


std::vector<std::vector<float> >* photontrkiso_recvtx_030_002_0000_10_01;
float photontrkiso_badvtx_040_002_0000_10_01[nPhotonMAX];
float photondrtotrk[nPhotonMAX];
int indvertexSelected;


vector<short> *indvertexSelected_allpairpresel;
vector<short> *indvertexSelectedtmva_allpairpresel;

////vector<vector<float> >* photontrkisoworstdr04;
///the aboved updated, for each photon, there is one number 
float photontrkisoworstdr04[nPhotonMAX];
int photonindvtxtrkisoworstdr04[nPhotonMAX];
vector<vector<float> >* photontrkisoselvtxdr03;
vector<vector<float> >* photontrkisoselvtxtmvadr03;
//vector<vector<short> >* photonindvtxtrkisoworstdr04;


int ngenphoton;
float etagenphoton[nPhotonMAX];
float phigenphoton[nPhotonMAX];
float ptgenphoton[nPhotonMAX];
float vxgenphoton[nPhotonMAX];
float vygenphoton[nPhotonMAX];
float vzgenphoton[nPhotonMAX];
int pidmomgenphoton[nPhotonMAX];
int pidmom2genphoton[nPhotonMAX];
int pidmom3genphoton[nPhotonMAX];
int statusgenphoton[nPhotonMAX];


int ngenelectron;
float etagenelectron[nElectronMAX];
float phigenelectron[nElectronMAX];
float ptgenelectron[nElectronMAX];
float vxgenelectron[nElectronMAX];
float vygenelectron[nElectronMAX];
float vzgenelectron[nElectronMAX];
int pidmomgenelectron[nElectronMAX];
int pidmom2genelectron[nElectronMAX];
int pidmom3genelectron[nElectronMAX];
int statusgenelectron[nElectronMAX];


float photon_convp[nPhotonMAX];
float photon_convpt[nPhotonMAX];
float photon_convpttrk1[nPhotonMAX];
float photon_convpttrk2[nPhotonMAX];
float photon_deltaphiconvsc[nPhotonMAX];
float photon_deltaetaconvsc[nPhotonMAX];
float photon_convrho[nPhotonMAX];
float photon_convz[nPhotonMAX];


  float isosumoet; 
  float isosumoetbad; 
  float trkisooet; 
  //float sieie;
  float hoe; 
  float r9; 
  float drtotrk; 
  int haspromptele; 
  
   ///new tree for singlePhotonTree
  float etrue; 
float etatrue; 
float phitrue;
float etaecaltrue;
float phiecaltrue;

float vxtrue;
float vytrue;
float vztrue;

float vx;
float vy;
float vz;

float ecalisodr03;
float ecalisodr04;
float hcalisodr03;
float hcalisodr04;


  float escraw; 
  float e5x5; 
  
  float e2x2;
  float e3x3;
  float e1x3;
  float e3x1;
  float e4x4;
  float e2x5;
  float e2x5right;
  float e2x5left;
  float e2x5top;
  float e2x5bottom;
  float e2x5max;
    
  float eps_escraw;

  float e3x3_e5x5;
  float eps_e5x5;
  float e5x5_escraw; 
  float emax_escraw; 
  float eleft_escraw; 
  float eright_escraw; 
  float etop_escraw; 
  float ebottom_escraw; 

float e2x5left_escraw;
float e2x5right_escraw;
float e2x5top_escraw;
float e2x5bottom_escraw;


  float emax_e5x5; 
  float eleft_e5x5; 
  float eright_e5x5; 
  float etop_e5x5; 
  float ebottom_e5x5; 

float etaele; 
float phiele;

  float eps; 
  float esc; 
float eele;

  float epht;
  float etasc; 
  float phisc; 
  float etapht;
  float phipht;
  float scetawidth; 
  float scphiwidth; 
  float sigietaieta; 
  float sigiphiiphi;
  float sigcovietaiphi;
  float emax; 
  int scnbc; 
  int scncrystal;
  float scbcfmax;
  int seedieta; 
  int seediphi; 
  float eleft; 
  float eright; 
float e2max; 
  float etop;
  float ebottom;
  float convp; 
  float convpt; 
  float convpttrk1; 
  float convpttrk2; 
  float deltaphi_convsc;
  float deltaeta_convsc; 
  
  float convrho;
  float convz; 
  
  float escrawoverconvp;
  float convleadingptoverpt;

float pileupwt;
  
  float seedietaf; 
  float seediphif; 
  float scnbcf; 
  float scncrystalf; 
float nVertexf; 


  int truephtmatched; 
  float mpair; 
  

  float etaCGap; 
  float phiCGap;
  float etaSGap; 
  float phiSGap;
  float etaMGap; 
  float phiMGap;

float xCGap; 
float xMGap; 
float xSGap; 
float yCGap; 
float yMGap; 
float ySGap; 


int preselcut;

int testPhotonCorr;
int testVtxID;
int testPhotonID;
int trueelematched;
int npvbx0;
float npvbxav;


int nWarning;
int eleclass;
float elefbrem;
int elenbrem;
float sigiphi_sigieta;
float scsigphi_sigeta;
int trainTarget;
int pidphtmom; 
float trkisodr03vtxbestGenMatched;
float trkisodr04vtxWorst;

bool  doEscaleSyst, doEresolSyst, doPhotonIdEffSyst, doVtxEffSyst, doR9Syst, doTriggerEffSyst, doKFactorSyst;
bool doPhotonEffSys;


std::map<std::string,int>::iterator it_sys;

int nSystSteps; 
float systRange;

float photonenergy0[nPhotonMAX];

float et1cut;
float et2cut;
int nPhotonPreSelected;
float add_sigmaEoverE[4];
float addErr_sigmaEoverE[4];
Revision:	1.1
Committed:	Fri Sep 30 15:23:49 2011 UTC (13 years, 7 months ago) by yangyong
Content type:	text/plain
Branch:	MAIN
CVS Tags:	V01-00-01, V01-00-00, HEAD
Log Message:	z selection code
#	Content
1	int dataflag;
2	int evtRange;
3	int dataOrMC;
4	int totalEntries;
5	int entry;
6
7	TChain *fChain;
8
9	TTree *newtree;
10	string dataversion;
11	string datasetname;
12
13	float detasctrkvtx;
14	float dphisctrkvtx;
15
16	float e1x5;
17
18
19
20	float photonet[nPhotonMAX];
21	float photonetvtx0[nPhotonMAX];
22	float photoncaloeta[nPhotonMAX];
23	float photoncalophi[nPhotonMAX];
24	int photoninbarrel[nPhotonMAX];
25
26	float electronecalEnergy[nElectronMAX];
27	float electronecalEnergyCorr[nElectronMAX];
28
29
30	TTree *evtInfo;
31
32	int totalNumberofEvents ;
33	int preselectedNumberofEvents;
34	vector<string> *datasettag;
35
36	float EnergyScaleOffPhotonCat[4];
37	float EnergySmearingPhotonCat[4];
38
39	TRandom3 *rgen_;
40
41	std::map< int,std::vector<TH1*> > kFactors_;
42
43
44	///ID efficieny correction
45
46	std::map<std::string,TGraphAsymmErrors*> IDCorrection_;
47
48	/* std::map<std::string, RooRealVar> m_real_var_; */
49	/* std::map<std::string, double> m_var_min_; */
50	/* std::map<std::string, double> m_var_max_; */
51	/* std::map<std::string,RooRealVar> m_data_var_ptr_; /
52	/* std::map<std::string,RooRealVar> m_weight_var_ptr_; /
53	/* std::map<std::string,RooDataSet> data_; */
54	/* std::map<std::string,std::string> data_obs_names_; */
55
56
57	std::vector<float> ptunbal_;
58	std::vector<float> angleptunbaldipht_;
59	std::vector<float> asym_ptunbal_;
60
61	std::vector<float> ncha_;
62	std::vector<float> nchapt1_;
63
64
65	std::vector<float> ptbal_;
66	std::vector<float> thrust_;
67	std::vector<float> sumpt_;
68	std::vector<float> sumpt2_;
69	std::vector<float> logsumpt2_;
70	std::vector<float> sumawy_;
71	std::vector<float> sumtwd_;
72	std::vector<float> sumtrv_;
73	std::vector<float> sumweight_;
74	std::vector<float> ptmax_;
75	std::vector<float> nchthr_;
76	std::vector<float> nch_;
77	std::vector<std::vector<float> > tksPt_;
78	std::vector<TMatrixDSym> sphers_;
79	std::vector<float> sumpr_;
80	std::vector<float> spher_;
81	std::vector<float> tspher_;
82	std::vector<float> aplan_;
83	std::vector<float> threejetC_;
84	std::vector<float> fourjetD_;
85
86	std::vector<TVector3> vtxP_;
87	std::vector<TVector2> vtxPt_;
88	std::vector<float> ptvtx_;
89	std::vector<float> diphopt_;
90	std::vector<float> diPhotonPz_;
91	std::vector<float> acosA_;
92	std::vector<float> ptasym_;
93
94	std::vector<float> ptmax3_;
95	std::vector<float> ptratio_;
96	std::vector<float> pzasym_;
97	std::vector<float> awytwdasym_;
98	std::vector<TVector2> diPhotonPt_;
99
100	vector<int> indpresel;
101	vector<int> indaccept;
102
103	std::vector<int> preselConv;
104
105
106	std::vector<int> rankprodvtx;
107
108	std::vector<int> tmvavtx;
109
110	float ptbal;
111	float tspher;
112	float spher;
113	float aplan;
114	float threejetC;
115	float fourjetD;
116	float diphopt;
117	float acosA;
118	float ptasym;
119	float ptmax;
120	float ptmax3;
121	float thrust;
122	float ptratio;
123	float pzasym;
124	float sumwt;
125	float vtxpt;
126
127	float sumpt2;
128	float sumpt;
129	float awytwdasym;
130	int preselectedbyconv;
131	int isgenmatched;
132	float ptunbal;
133	float angleptunbaldipht;
134	float asym_ptunbal;
135	float ncha;
136	float nchapt1;
137
138	float nchapt1f;
139
140	float logsumpt2;
141
142	float ptdiphtfirstvertex;
143
144
145	TMVA::Reader *reader;
146	TMVA::Reader *reader2;
147
148	TMVA::Reader *reader_eb_unconv;
149	TMVA::Reader *reader_eb_conv;
150	TMVA::Reader *reader_ee_unconv;
151	TMVA::Reader *reader_ee_conv;
152
153
154	TString methodName;
155
156	float tmva;
157	float tmva2;
158
159
160
161	int isbestvtx;
162	int isworstvtx;
163	float distvtxgen;
164
165	int testTMVA;
166
167
168	std::vector<std::vector<float> >* photontrkiso_recvtx_030_002_0000_10_01;
169	float photontrkiso_badvtx_040_002_0000_10_01[nPhotonMAX];
170	float photondrtotrk[nPhotonMAX];
171	int indvertexSelected;
172
173
174	vector<short> *indvertexSelected_allpairpresel;
175	vector<short> *indvertexSelectedtmva_allpairpresel;
176
177	////vector<vector<float> >* photontrkisoworstdr04;
178	///the aboved updated, for each photon, there is one number
179	float photontrkisoworstdr04[nPhotonMAX];
180	int photonindvtxtrkisoworstdr04[nPhotonMAX];
181	vector<vector<float> >* photontrkisoselvtxdr03;
182	vector<vector<float> >* photontrkisoselvtxtmvadr03;
183	//vector<vector<short> >* photonindvtxtrkisoworstdr04;
184
185
186	int ngenphoton;
187	float etagenphoton[nPhotonMAX];
188	float phigenphoton[nPhotonMAX];
189	float ptgenphoton[nPhotonMAX];
190	float vxgenphoton[nPhotonMAX];
191	float vygenphoton[nPhotonMAX];
192	float vzgenphoton[nPhotonMAX];
193	int pidmomgenphoton[nPhotonMAX];
194	int pidmom2genphoton[nPhotonMAX];
195	int pidmom3genphoton[nPhotonMAX];
196	int statusgenphoton[nPhotonMAX];
197
198
199	int ngenelectron;
200	float etagenelectron[nElectronMAX];
201	float phigenelectron[nElectronMAX];
202	float ptgenelectron[nElectronMAX];
203	float vxgenelectron[nElectronMAX];
204	float vygenelectron[nElectronMAX];
205	float vzgenelectron[nElectronMAX];
206	int pidmomgenelectron[nElectronMAX];
207	int pidmom2genelectron[nElectronMAX];
208	int pidmom3genelectron[nElectronMAX];
209	int statusgenelectron[nElectronMAX];
210
211
212	float photon_convp[nPhotonMAX];
213	float photon_convpt[nPhotonMAX];
214	float photon_convpttrk1[nPhotonMAX];
215	float photon_convpttrk2[nPhotonMAX];
216	float photon_deltaphiconvsc[nPhotonMAX];
217	float photon_deltaetaconvsc[nPhotonMAX];
218	float photon_convrho[nPhotonMAX];
219	float photon_convz[nPhotonMAX];
220
221
222
223
224	float isosumoet;
225	float isosumoetbad;
226	float trkisooet;
227	//float sieie;
228	float hoe;
229	float r9;
230	float drtotrk;
231	int haspromptele;
232
233	///new tree for singlePhotonTree
234	float etrue;
235	float etatrue;
236	float phitrue;
237	float etaecaltrue;
238	float phiecaltrue;
239
240	float vxtrue;
241	float vytrue;
242	float vztrue;
243
244	float vx;
245	float vy;
246	float vz;
247
248	float ecalisodr03;
249	float ecalisodr04;
250	float hcalisodr03;
251	float hcalisodr04;
252
253
254	float escraw;
255	float e5x5;
256
257	float e2x2;
258	float e3x3;
259	float e1x3;
260	float e3x1;
261	float e4x4;
262	float e2x5;
263	float e2x5right;
264	float e2x5left;
265	float e2x5top;
266	float e2x5bottom;
267	float e2x5max;
268
269	float eps_escraw;
270
271	float e3x3_e5x5;
272	float eps_e5x5;
273	float e5x5_escraw;
274	float emax_escraw;
275	float eleft_escraw;
276	float eright_escraw;
277	float etop_escraw;
278	float ebottom_escraw;
279
280	float e2x5left_escraw;
281	float e2x5right_escraw;
282	float e2x5top_escraw;
283	float e2x5bottom_escraw;
284
285
286	float emax_e5x5;
287	float eleft_e5x5;
288	float eright_e5x5;
289	float etop_e5x5;
290	float ebottom_e5x5;
291
292	float etaele;
293	float phiele;
294
295	float eps;
296	float esc;
297	float eele;
298
299	float epht;
300	float etasc;
301	float phisc;
302	float etapht;
303	float phipht;
304	float scetawidth;
305	float scphiwidth;
306	float sigietaieta;
307	float sigiphiiphi;
308	float sigcovietaiphi;
309	float emax;
310	int scnbc;
311	int scncrystal;
312	float scbcfmax;
313	int seedieta;
314	int seediphi;
315	float eleft;
316	float eright;
317	float e2max;
318	float etop;
319	float ebottom;
320	float convp;
321	float convpt;
322	float convpttrk1;
323	float convpttrk2;
324	float deltaphi_convsc;
325	float deltaeta_convsc;
326
327	float convrho;
328	float convz;
329
330	float escrawoverconvp;
331	float convleadingptoverpt;
332
333	float pileupwt;
334
335	float seedietaf;
336	float seediphif;
337	float scnbcf;
338	float scncrystalf;
339	float nVertexf;
340
341
342	int truephtmatched;
343	float mpair;
344
345
346	float etaCGap;
347	float phiCGap;
348	float etaSGap;
349	float phiSGap;
350	float etaMGap;
351	float phiMGap;
352
353	float xCGap;
354	float xMGap;
355	float xSGap;
356	float yCGap;
357	float yMGap;
358	float ySGap;
359
360
361	int preselcut;
362
363	int testPhotonCorr;
364	int testVtxID;
365	int testPhotonID;
366	int trueelematched;
367	int npvbx0;
368	float npvbxav;
369
370
371
372	int nWarning;
373	int eleclass;
374	float elefbrem;
375	int elenbrem;
376	float sigiphi_sigieta;
377	float scsigphi_sigeta;
378	int trainTarget;
379	int pidphtmom;
380	float trkisodr03vtxbestGenMatched;
381	float trkisodr04vtxWorst;
382
383	bool doEscaleSyst, doEresolSyst, doPhotonIdEffSyst, doVtxEffSyst, doR9Syst, doTriggerEffSyst, doKFactorSyst;
384	bool doPhotonEffSys;
385
386
387	std::map<std::string,int>::iterator it_sys;
388
389	int nSystSteps;
390	float systRange;
391
392	float photonenergy0[nPhotonMAX];
393
394	float et1cut;
395	float et2cut;
396	int nPhotonPreSelected;
397	float add_sigmaEoverE[4];
398	float addErr_sigmaEoverE[4];