/[PAMELA software]/calo/flight/CaloPreSampler/src/CaloPreSampler.cpp
ViewVC logotype

Contents of /calo/flight/CaloPreSampler/src/CaloPreSampler.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.11 - (show annotations) (download)
Thu Dec 18 20:57:58 2008 UTC (16 years, 1 month ago) by mocchiut
Branch: MAIN
Changes since 1.10: +12 -0 lines
Method SplitInto(int,int) added

1 /**
2 * \file CaloPreSampler.cpp
3 * \author Emiliano Mocchiutti (2007/07/18)
4 */
5 //
6 // headers
7 //
8 #include <CaloPreSampler.h>
9 //--------------------------------------
10 /**
11 * Default constructor
12 */
13 CaloPreSampler::CaloPreSampler(){
14 Clear();
15 }
16
17 CaloPreSampler::CaloPreSampler(PamLevel2 *l2p){
18 //
19 L2 = l2p;
20 //
21 if ( !L2->IsORB() ) printf(" WARNING: OrbitalInfo Tree is needed, the plugin could not work properly without it \n");
22 //
23 OBT = 0;
24 PKT = 0;
25 atime = 0;
26 //
27 // Default variables
28 //
29 event = new CaloLevel0();
30 cstrip = new CaloStrip(false);
31 // c1 = new CaloLevel1();
32 pcalo = new CaloLevel2();
33 N = 4;
34 NC = 22-N;
35 debug = false;
36 sel = true;
37 cont = false;
38 emulate18 = true;
39 simulation = false;
40 withtrk = true;
41 rigdefault = 50.;
42 nox = false;
43 noy = false;
44 forcecalo = false;
45 memset(mask,0,2*22*sizeof(Int_t));
46 //
47 Clear();
48 //
49 // loading magnetic field...
50 //
51 TrkLevel2 *trk = new TrkLevel2();
52 GL_PARAM *q4 = new GL_PARAM();
53 TSQLServer *dbc = 0;
54 TString host = "mysql://localhost/pamelaprod";
55 TString user = "anonymous";
56 TString psw = "";
57 const char *pamdbhost=gSystem->Getenv("PAM_DBHOST");
58 const char *pamdbuser=gSystem->Getenv("PAM_DBUSER");
59 const char *pamdbpsw=gSystem->Getenv("PAM_DBPSW");
60 if ( !pamdbhost ) pamdbhost = "";
61 if ( !pamdbuser ) pamdbuser = "";
62 if ( !pamdbpsw ) pamdbpsw = "";
63 if ( strcmp(pamdbhost,"") ) host = pamdbhost;
64 if ( strcmp(pamdbuser,"") ) user = pamdbuser;
65 if ( strcmp(pamdbpsw,"") ) psw = pamdbpsw;
66 dbc = TSQLServer::Connect(host.Data(),user.Data(),psw.Data());
67 //
68 q4->Query_GL_PARAM(1,1,dbc);
69 printf(" Reading magnetic field maps at %s\n",(q4->PATH+q4->NAME).Data());
70 trk->LoadField(q4->PATH+q4->NAME);
71 //
72 }
73
74 void CaloPreSampler::SetNoWpreSampler(Int_t n){
75 if ( NC+n < 23 ){
76 N = n;
77 } else {
78 printf(" ERROR! Calorimeter is made of 22 W planes\n");
79 printf(" you are giving N presampler = %i and N calo = %i \n",n,NC);
80 printf(" WARNING: using default values NWpre = 4, NWcalo = 18\n");
81 NC = 18;
82 N = 4;
83 };
84 }
85
86 void CaloPreSampler::SetNoWcalo(Int_t n){
87 if ( N+n < 23 ){
88 NC = n;
89 } else {
90 printf(" ERROR! Calorimeter is made of 22 W planes\n");
91 printf(" you are giving N W presampler = %i and N W calo = %i \n",N,n);
92 printf(" WARNING: using default values NWpre = 4, NWcalo = 18\n");
93 NC = 18;
94 N = 4;
95 };
96 }
97
98 void CaloPreSampler::SplitInto(Int_t NoWpreSampler, Int_t NoWcalo){
99 this->SetNoWcalo(0);
100 this->SetNoWpreSampler(0);
101 if ( NoWpreSampler < NoWcalo ){
102 this->SetNoWpreSampler(NoWpreSampler);
103 this->SetNoWcalo(NoWcalo);
104 } else {
105 this->SetNoWcalo(NoWcalo);
106 this->SetNoWpreSampler(NoWpreSampler);
107 };
108 }
109
110 void CaloPreSampler::Clear(){
111 //
112 pcalo->Clear();
113 //
114 }
115
116 void CaloPreSampler::Print(){
117 //
118 Process();
119 //
120 printf("========================================================================\n");
121 printf(" OBT: %u PKT: %u ATIME: %u \n",OBT,PKT,atime);
122 printf(" debug [debug flag]:.. %i\n",debug);
123 printf(" simulation [simulation flag]:.. %i\n",simulation);
124 printf(" emulate18 [emulate dead plane 18]:.. %i\n",emulate18);
125 printf(" selection mode :.. %i\n",sel);
126 printf(" contamination mode :.. %i\n",cont);
127 printf(" pre-sampler planes :.. %i\n",N);
128 printf(" pcalo->qtot :.. %f\n",pcalo->qtot);
129 printf(" pcalo->nstrip :.. %i\n",pcalo->nstrip);
130 if ( pcalo->ntrk() > 0 ){
131 printf(" pcalo->track0->qtrack :.. %f\n",pcalo->GetCaloTrkVar(0)->qtrack);
132 printf(" pcalo->track0->dX0l :.. %f\n",pcalo->GetCaloTrkVar(0)->dX0l);
133 };
134 printf("========================================================================\n");
135 //
136 }
137
138 void CaloPreSampler::Delete(){
139 Clear();
140 delete pcalo;
141 //delete this;
142 }
143
144
145 void CaloPreSampler::Process(){
146 //
147 if ( !L2 ){
148 printf(" ERROR: cannot find PamLevel2 object, use the correct constructor or check your program!\n");
149 printf(" ERROR: CaloPreSampler variables _NOT_ filled \n");
150 return;
151 };
152 //
153 Bool_t newentry = false;
154 //
155 if ( L2->IsORB() ){
156 if ( L2->GetOrbitalInfo()->pkt_num != PKT || L2->GetOrbitalInfo()->OBT != OBT || L2->GetOrbitalInfo()->absTime != atime || sel != ssel ){
157 newentry = true;
158 OBT = L2->GetOrbitalInfo()->OBT;
159 PKT = L2->GetOrbitalInfo()->pkt_num;
160 atime = L2->GetOrbitalInfo()->absTime;
161 ssel = sel;
162 };
163 } else {
164 newentry = true;
165 };
166 //
167 if ( !newentry ) return;
168 //
169 // Some variables
170 //
171 Int_t S3 = 0;
172 Int_t S2 = 0;
173 Int_t S12 = 0;
174 Int_t S11 = 0;
175 Float_t tmptrigty = -1.;
176 Bool_t trackanyway = true;
177 // Float_t rigdefault = 50.;
178 Bool_t hZn = true;
179 // Bool_t withtrk = true;
180 Bool_t st = true;
181 Int_t ntrkentry = 0;
182 TrkLevel2 *trk = L2->GetTrkLevel2();
183 Bool_t filled = false;
184 //
185 if ( debug ) printf(" Processing event at OBT %u PKT %u time %u \n",OBT,PKT,atime);
186 //
187 this->Clear();
188 //
189 // find out if we have trkseqno = -1, -2 or -3
190 //
191 Bool_t m1 = false;
192 Bool_t m2 = false;
193 Bool_t m3 = false;
194 for (Int_t mm=0; mm < L2->GetCaloLevel2()->ntrk(); mm++ ){
195 if ( L2->GetCaloLevel2()->GetCaloTrkVar(mm)->trkseqno == -1 ) m1 = true;
196 if ( L2->GetCaloLevel2()->GetCaloTrkVar(mm)->trkseqno == -2 ) m2 = true;
197 if ( L2->GetCaloLevel2()->GetCaloTrkVar(mm)->trkseqno == -3 ) m3 = true;
198 };
199 if ( !withtrk ) m3 = true;
200 //
201 if ( debug ) printf(" Fill estrip matrix needed to calculate variables \n");
202 //
203 // Fill the estrip matrix
204 //
205 memset(event->clevel1->estrip, 0, 2*22*96*sizeof(Float_t));
206 Int_t view = 0;
207 Int_t plane = 0;
208 Int_t strip = 0;
209 Float_t mip = 0.;
210 for ( Int_t i=0; i<L2->GetCaloLevel1()->istrip; i++ ){
211 //
212 mip = L2->GetCaloLevel1()->DecodeEstrip(i,view,plane,strip);
213 //
214 // Mask x or y view if nox and/or noy are true (default false)
215 //
216 if ( mask[view][plane] ) mip = 0.;
217 if ( nox && view == 0 ) mip = 0.;
218 if ( noy && view == 1 ) mip = 0.;
219 //
220 if ( emulate18 && view == 0 && plane == 18 ) mip = 0.;
221 //
222 // Selection mode: fill the matrix only for plane < (22 - N) REMEMBER N = number of W planes to be used as presampler, ie if N = 2 then we want to use planes from 0 to 19
223 // included so plane < (22 - 2)
224 //
225 if ( sel ){
226 if ( plane < (22 - N) ){
227 //
228 if ( emulate18 && plane == (18 - N) && view == 0 ) mip = 0.;
229 if ( plane >= NC ) mip = 0.;
230 event->clevel1->estrip[strip][plane][view] = mip;
231 //
232 };
233 };
234 //
235 // Contamination mode: fill the matrix only for planes from N to 22 but shift all planes up to the first one
236 //
237 if ( cont ){
238 if ( plane >= N ){
239 //
240 if ( emulate18 && plane == (18 + N) && view == 0 ) mip = 0.;
241 if ( (plane-N) >= NC ) mip = 0.;
242 event->clevel1->estrip[strip][(plane-N)][view] = mip;
243 //
244 };
245 };
246 //
247 };
248 //
249 // if data comes from the simulation we must use mechanical alignment parameters (default is flight parameters)
250 //
251 if ( simulation ){
252 cstrip->UseMechanicalAlig();
253 };
254 //
255 // Set alignment parameter
256 //
257 event->clevel1->xalig = cstrip->GetXalig();
258 event->clevel1->yalig = cstrip->GetYalig();
259 event->clevel1->zalig = cstrip->GetZalig();
260 //
261 event->clevel1->emin = 0.7;
262 //
263 // in case of the contamination mode we must play with the Z alignment in order to have the correct track in the calo since we have moved the planes up...
264 //
265 if ( cont ){
266 if ( !(N%2) ){
267 event->clevel1->reverse = 0; // if the number of planes is even we have taken away a full module no need to do anything strange...
268 event->clevel1->zalig -= (N/2) * (8.09 + 10.09);
269 } else {
270 event->clevel1->reverse = 1; // if the number of planes is odd we have taken away half a module, we need to reverse silicon planes shifting
271 event->clevel1->zalig -= ((N+1)/2) * 8.09 + ((N-1)/2) * 10.09;
272 };
273 };
274 if ( debug ) printf(" xalig = %f \n",event->clevel1->xalig);
275 if ( debug ) printf(" yalig = %f \n",event->clevel1->yalig);
276 if ( debug ) printf(" zalig = %f \n",event->clevel1->zalig);
277 //
278 if ( debug ) printf(" Calculate variables as done in CaloCore, N = %i \n",N);
279 //
280 // Calculate variables
281 //
282 //
283 // use only N W planes
284 //
285 // event->clevel1->npla = 22-N;
286 event->clevel1->npla = NC;
287 //
288 S3 = 0;
289 S2 = 0;
290 S12 = 0;
291 S11 = 0;
292 S3 = L2->GetTrigLevel2()->patterntrig[2];
293 S2 = L2->GetTrigLevel2()->patterntrig[3];
294 S12 = L2->GetTrigLevel2()->patterntrig[4];
295 S11 = L2->GetTrigLevel2()->patterntrig[5];
296 if ( L2->GetTrigLevel2()->patterntrig[1] & (1<<0) ) tmptrigty = 1.;
297 if ( L2->GetTrigLevel2()->patterntrig[0] ) tmptrigty = 2.;
298 if ( S3 || S2 || S12 || S11 ) tmptrigty = 0.;
299 if ( !(L2->GetTrigLevel2()->patterntrig[1] & (1<<0)) && !L2->GetTrigLevel2()->patterntrig[0] && !S3 && !S2 && !S12 && !S11 ) tmptrigty = 1.;
300 event->clevel2->trigty = tmptrigty;
301 //
302 // do we have at least one track from the tracker? this check has been disabled
303 //
304 event->clevel1->good2 = 1;
305 //
306 // copy variables calculated during calibration process which is skipped here...
307 //
308 event->clevel2->good = L2->GetCaloLevel2()->good;
309 memcpy(event->clevel2->perr,L2->GetCaloLevel2()->perr,sizeof(L2->GetCaloLevel2()->perr));
310 memcpy(event->clevel2->swerr,L2->GetCaloLevel2()->swerr,sizeof(L2->GetCaloLevel2()->swerr));
311 memcpy(event->clevel2->crc,L2->GetCaloLevel2()->crc,sizeof(L2->GetCaloLevel2()->crc));
312 event->clevel2->selftrigger = L2->GetCaloLevel2()->selftrigger;
313 //
314 // Calculate variables common to all tracks (qtot, nstrip, etc.)
315 //
316 if ( debug ) printf("1 Call GetCommonVar() \n");
317 event->GetCommonVar();
318 //
319 // Fill common variables
320 //
321 if ( debug ) printf("1 Call FillCommonVar() \n");
322 event->FillCommonVar(NULL,pcalo);
323 //
324 // Calculate variables related to tracks only if we have at least one track (from selftrigger and/or tracker)
325 //
326 ntrkentry = 0;
327 //
328 filled = false;
329 //
330 // Run over tracks (tracker or calorimeter )
331 //
332 if ( withtrk ){
333 //
334 for (Int_t nt=0; nt < trk->ntrk(); nt++){
335 //
336 event->clevel1->good2 = 1;
337 //
338 TrkTrack *ptt = trk->GetStoredTrack(nt);
339 //
340 event->clevel1->trkchi2 = 0;
341 //
342 // Copy the alpha vector in the input structure
343 //
344 for (Int_t e = 0; e < 5 ; e++){
345 event->clevel1->al_p[e][0] = ptt->al[e];
346 };
347 //
348 // Get tracker related variables for this track
349 //
350 if ( debug ) printf("track %i Call GetTrkVar() \n",nt);
351 event->GetTrkVar();
352 if ( debug ) printf(" event->clevel2->dX0l %f \n",event->clevel2->dX0l);
353 //
354 // Save tracker track sequence number
355 //
356 event->trkseqno = nt;
357 //
358 // Copy values in the class ca from the structure clevel2
359 //
360 if ( debug ) printf("track %i Call FillTrkVar() \n",nt);
361 event->FillTrkVar(pcalo,ntrkentry);
362
363
364 ntrkentry++;
365 filled = true;
366 //
367 }; // loop on all the tracks
368 };
369 //
370 // if no tracks found but there is the possibility to have a good track we should try to calculate anyway the track related variables using the calorimeter
371 // fit of the track (to be used for example when TRK is off due to any reason like IPM3/5 off).
372 // here we make an event selection so it must be done very carefully...
373 //
374 // conditions are: 0) no track from the tracker 1) we have a track fit both in x and y 2) no problems with calo for this event 3) no selftrigger event
375 //
376 // if ( trackanyway && !filled && event->clevel2->npcfit[0] >= 2 && event->clevel2->npcfit[1] >= 2 && event->clevel2->good != 0 && event->clevel2->trigty < 2. ){
377 if ( (trackanyway && m3) || forcecalo ){
378 if ( debug ) printf(" Event with a track not fitted by the tracker \n");
379 //
380 // Disable "track mode" in the fortran routine
381 //
382 event->clevel1->good2 = 0;
383 event->clevel1->riginput = rigdefault;
384 if ( debug ) printf(" Using as default rigidity: %f \n",event->clevel1->riginput);
385 //
386 // We have a selftrigger event to analyze.
387 //
388 for (Int_t e = 0; e < 5 ; e++){
389 event->clevel1->al_p[e][0] = 0.;
390 event->clevel1->al_p[e][1] = 0.;
391 };
392 event->clevel1->trkchi2 = 0;
393 //
394 if ( debug ) printf("-3 a Call GetTrkVar() \n");
395 event->GetTrkVar();
396 //
397 // if we had no problem (clevel1->good2 = 0, NOTICE zero, not one in this mode!), fill and go on
398 //
399 if ( event->clevel1->good2 == 0 ) {
400 //
401 // In selftrigger mode the trkentry variable is set to -1
402 //
403 event->trkseqno = -3;
404 //
405 // Copy values in the class ca from the structure clevel2
406 //
407 if ( debug ) printf("-3 a Call FillTrkVar() \n");
408 event->FillTrkVar(pcalo,ntrkentry);
409 ntrkentry++;
410 filled = true;
411 //
412 } else {
413 if ( debug ) printf(" Selftrigger: problems with event \n");
414 };
415 //
416 };
417 //
418 // Call high energy nuclei routine
419 //
420 // if ( hZn && event->clevel2->trigty >= 2. ){
421 if ( hZn && m2 ){
422 if ( debug ) printf(" Calling selftrigger high energy nuclei routine \n");
423 //
424 // Disable "track mode" in the fortran routine
425 //
426 event->clevel1->good2 = 0;
427 //
428 // Set high energy nuclei flag to one
429 //
430 event->clevel1->hzn = 1;
431 event->clevel1->riginput = rigdefault;
432 //
433 // We have a selftrigger event to analyze.
434 //
435 for (Int_t e = 0; e < 5 ; e++){
436 event->clevel1->al_p[e][0] = 0.;
437 event->clevel1->al_p[e][1] = 0.;
438 };
439 event->clevel1->trkchi2 = 0;
440 //
441 if ( debug ) printf("-2 a Call GetTrkVar() \n");
442 event->GetTrkVar();
443 //
444 // if we had no problem (clevel1->good2 = 0, NOTICE zero, not one in this mode!), fill and go on
445 //
446 if ( event->clevel1->good2 == 0 ) {
447 //
448 // In selftrigger mode the trkentry variable is set to -1
449 //
450 event->trkseqno = -2;
451 //
452 // Copy values in the class ca from the structure clevel2
453 //
454 if ( debug ) printf("-2 a Call FillTrkVar() \n");
455 event->FillTrkVar(pcalo,ntrkentry);
456 ntrkentry++;
457 filled = true;
458 //
459 } else {
460 if ( debug ) printf(" Selftrigger: problems with event \n");
461 };
462 //
463 };
464 //
465 // self trigger event
466 //
467 // if ( st && event->clevel2->trigty >= 2. ){
468 if ( st && m1 ){
469 if ( debug ) printf(" Selftrigger event \n");
470 //
471 // Disable "track mode" in the fortran routine
472 //
473 event->clevel1->good2 = 0;
474 //
475 // disable high enery nuclei flag;
476 //
477 event->clevel1->hzn = 0;
478 //
479 // We have a selftrigger event to analyze.
480 //
481 for (Int_t e = 0; e < 5 ; e++){
482 event->clevel1->al_p[e][0] = 0.;
483 event->clevel1->al_p[e][1] = 0.;
484 };
485 event->clevel1->trkchi2 = 0;
486 //
487 if ( debug ) printf("-1 a Call GetTrkVar() \n");
488 event->GetTrkVar();
489 //
490 // if we had no problem (clevel2->good = 0, NOTICE zero, not one in selftrigger mode!), fill and go on
491 //
492 if ( event->clevel1->good2 == 0 ) {
493 //
494 // In selftrigger mode the trkentry variable is set to -1
495 //
496 event->trkseqno = -1;
497 //
498 // Copy values in the class ca from the structure clevel2
499 //
500 if ( debug ) printf("-1 a Call FillTrkVar() \n");
501 event->FillTrkVar(pcalo,ntrkentry);
502 ntrkentry++;
503 filled = true;
504 //
505 } else {
506 if ( debug ) printf(" Selftrigger: problems with event \n");
507 };
508 };
509 //
510 // Clear structures used to communicate with fortran
511 //
512 event->ClearStructs();
513 //
514 //
515 //
516 if ( debug ) this->Print();
517 if ( debug ) printf(" exit \n");
518 //
519 }

  ViewVC Help
Powered by ViewVC 1.1.23