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
#	User	Rev	Content
1	yangyong	1.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];