/[PAMELA software]/DarthVader/AnticounterLevel2/src/AcCore.cpp
ViewVC logotype

Contents of /DarthVader/AnticounterLevel2/src/AcCore.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.17 - (show annotations) (download)
Sun Sep 9 18:57:23 2007 UTC (17 years, 5 months ago) by mocchiut
Branch: MAIN
CVS Tags: v4r00
Changes since 1.16: +7 -3 lines
Changed core routines to open themselves the DB connection

1 //
2 // C/C++ headers
3 //
4 #include <fstream>
5 #include <string.h>
6 //
7 // ROOT headers
8 //
9 #include <TTree.h>
10 #include <TClassEdit.h>
11 #include <TObject.h>
12 #include <TList.h>
13 #include <TArrayI.h>
14 #include <TSystem.h>
15 #include <TSystemDirectory.h>
16 #include <TString.h>
17 #include <TFile.h>
18 #include <TClass.h>
19 #include <TCanvas.h>
20 #include <TH1.h>
21 #include <TH1F.h>
22 #include <TH2D.h>
23 #include <TLatex.h>
24 #include <TPad.h>
25 #include <TSQLServer.h>
26 #include <TSQLRow.h>
27 #include <TSQLResult.h>
28 //#include <TClonesArray.h>
29 //
30 // RunInfo header
31 //
32 #include <RunInfo.h>
33 //
34 // YODA headers
35 //
36 #include <PamelaRun.h>
37 #include <physics/trigger/TriggerEvent.h>
38 #include <physics/anticounter/AnticounterEvent.h>
39 //
40 // RunInfo header
41 //
42 //#include <RunInfo.h>
43 //
44 // This program headers
45 //
46 #include <AcCore.h>
47 #include <AcLevel2.h>
48 #include <AcVerl2.h>
49 //
50 using namespace std;
51 //
52 // CORE ROUTINE
53 //
54 //
55 int AcCore(UInt_t run, TFile *file, GL_TABLES *glt, Int_t Acargc, char *Acargv[]){
56 Int_t i = 0;
57 Bool_t verbose = false;
58 Bool_t debug = false;
59 //
60 TString processFolder = Form("AcFolder_%u",run);
61 if ( Acargc > 0 ){
62 i = 0;
63 while ( i < Acargc ){
64 if ( !strcmp(Acargv[i],"-processFolder") ) {
65 if ( Acargc < i+1 ){
66 throw -3;
67 };
68 processFolder = (TString)Acargv[i+1];
69 i++;
70 };
71 if ( !strcmp(Acargv[i],"-v") || !strcmp(Acargv[i],"--verbose") ) {
72 verbose = true;
73 };
74 if ( !strcmp(Acargv[i],"-g") || !strcmp(Acargv[i],"--debug") ) {
75 verbose = true;
76 debug = true;
77 };
78 i++;
79 };
80 };
81 //
82 // Set these to true to have a very verbose output.
83 //
84 //
85 // Output directory is the working directoy.
86 //
87 const char* outdir = gSystem->DirName(gSystem->DirName(file->GetPath()));
88 //
89 // Variables for level2
90 //
91 TTree *ac = 0;
92 UInt_t nevents = 0;
93 //
94 // variables needed to reprocess data
95 //
96 TString acversion;
97 ItoRunInfo *runinfo = 0;
98 TArrayI *runlist = 0;
99 TTree *acclone = 0;
100 Bool_t reproc = false;
101 Bool_t reprocall = false;
102 UInt_t nobefrun = 0;
103 UInt_t noaftrun = 0;
104 UInt_t numbofrun = 0;
105 stringstream ftmpname;
106 TString fname;
107 UInt_t totfileentries = 0;
108 UInt_t idRun = 0;
109 //
110 // variables needed to handle error signals
111 //
112 Int_t code = 0;
113 Int_t sgnl;
114 //
115 // anticounter level2 classes
116 //
117 AcLevel2 *acl2 = new AcLevel2();
118 AcLevel2 *acl2clone = new AcLevel2();
119 // note: the string "mydect" is now "ac", "mydetector" is "anticounter" (mydectversion -> acversion, mydectclone -> acclone);
120 // the string "mydec" (without t) -> acl2 (acl2, acl2clone)
121 //
122 // define variables for opening and reading level0 file
123 //
124 TFile *l0File = 0;
125 TTree *l0tr = 0;
126 TBranch *l0head = 0;
127 TBranch *l0ac = 0;
128 pamela::EventHeader *eh = 0;
129 pamela::PscuHeader *ph = 0;
130 pamela::anticounter::AnticounterEvent *acc = 0;
131 //
132 // Define other basic variables
133 //
134 UInt_t procev = 0;
135 stringstream file2;
136 stringstream file3;
137 stringstream qy;
138 Int_t totevent = 0;
139 UInt_t atime = 0;
140 UInt_t re = 0;
141 //
142 // Working filename
143 //
144 TString outputfile;
145 stringstream name;
146 name.str("");
147 name << outdir << "/";
148 //
149 // temporary file and folder
150 //
151 TFile *tempfile = 0;
152 TTree *tempac = 0;
153 stringstream tempname;
154 stringstream acfolder;
155 Bool_t myfold = false;
156 tempname.str("");
157 tempname << outdir;
158 tempname << "/" << processFolder.Data();
159 acfolder.str("");
160 acfolder << tempname.str().c_str();
161 tempname << "/ac2tree_run";
162 tempname << run << ".root";
163 //
164 // DB classes
165 //
166 GL_ROOT *glroot = new GL_ROOT();
167 GL_TIMESYNC *dbtime = 0;
168 //
169 UShort_t CRCcheck[2]= {0,0};
170 UShort_t Status[2]= {0,0};
171 UShort_t Hitmap[2]= {0,0};
172 UShort_t Hitstatus[2]= {0,0};
173 UShort_t Trigger[2] = {0,0};
174 UShort_t Trigg_old[2] = {0,0};
175 UShort_t Counter[2][16];
176 UShort_t Counter_old[2][16];
177 UShort_t Shift[2][16];
178 Int_t vec[2][16][16];
179 Int_t tmp = 0;
180 memset(vec, 0, 2*16*16*sizeof(UShort_t));
181 memset(Shift, 0, 2*16*sizeof(UShort_t));
182 memset(Counter, 0, 2*16*sizeof(UShort_t));
183 memset(Counter_old, 0, 2*16*sizeof(UShort_t));
184 //
185 Long64_t maxsize = 10000000000LL;
186 TTree::SetMaxTreeSize(maxsize);
187 //
188 //
189 // Let's start!
190 //
191 //
192 // As a first thing we must check what we have to do: if run = 0 we must process all events in the file has been passed
193 // if run != 0 we must process only that run but first we have to check if the tree Anticounter already exist in the file
194 // if it exists we are reprocessing data and we must delete that entries, if not we must create it.
195 //
196 if ( run == 0 ) reproc = true;
197 //
198 // Output file is "outputfile"
199 //
200 outputfile = name.str().c_str();
201 if ( verbose ) printf("\n Output filename is: \n %s \n\n",outputfile.Data());
202 //
203 //
204 if ( !file->IsOpen() ){
205 if ( verbose ) printf(" ANTICOUNTER - ERROR: cannot open file for writing\n");
206 throw -701;
207 };
208 //
209 // Retrieve GL_RUN variables from the level2 file
210 //
211 acversion = AcInfo(false); // we should decide how to handle versioning system
212 //
213 // create an interface to RunInfo called "runinfo"
214 //
215 runinfo = new ItoRunInfo(file);
216 //
217 // open "Run" tree in level2 file, if not existing return an error (sngl != 0)
218 //
219 sgnl = 0;
220 sgnl = runinfo->Update(run,"AC",acversion);
221 if ( sgnl ){
222 if ( verbose ) printf(" ANTICOUNTER - ERROR: RunInfo exited with non-zero status\n");
223 code = sgnl;
224 goto closeandexit;
225 } else {
226 sgnl = 0;
227 };
228 //
229 // number of events in the file BEFORE the first event of our run
230 //
231 nobefrun = runinfo->GetFirstEntry();
232 //
233 // total number of events in the file
234 //
235 totfileentries = runinfo->GetFileEntries();
236 //
237 // first file entry AFTER the last event of our run
238 //
239 noaftrun = runinfo->GetLastEntry() + 1;
240 //
241 // number of run to be processed
242 //
243 numbofrun = runinfo->GetNoRun();
244 UInt_t totnorun = runinfo->GetRunEntries();
245 //
246 // Try to access the Anticounter tree in the file, if it exists we are reprocessing data if not we are processing a new run
247 //
248 acclone = (TTree*)file->Get("Anticounter");
249 //
250 if ( !acclone ){
251 //
252 // tree does not exist, we are not reprocessing
253 //
254 reproc = false;
255 if ( run == 0 && verbose ) printf(" ANTICOUNTER - WARNING: you are reprocessing data but AC tree does not exist!\n");
256 if ( verbose && runinfo->IsReprocessing() && run != 0 ) printf(" ANTICOUNTER - WARNING: it seems you are not reprocessing data but Anticounter\n versioning information already exists in RunInfo.\n");
257
258 } else {
259 //
260 // tree exists, we are reprocessing data. Are we reprocessing a single run or all the file?
261 //
262 acclone->SetAutoSave(900000000000000LL);
263 reproc = true;
264 //
265 //
266 if ( verbose ) printf("\n Preparing the pre-processing...\n");
267 //
268 if ( run == 0 || totnorun == 1 ){
269 //
270 // we are reprocessing all the file
271 // if we are reprocessing everything we don't need to copy any old event and we can just work with the new tree and delete the old one immediately
272 //
273 reprocall = true;
274 //
275 if ( verbose ) printf("\n ANTICOUNTER - WARNING: Reprocessing all runs\n");
276 //
277 } else {
278 //
279 // we are reprocessing a single run, we must copy to the new tree the events in the file which preceed the first event of the run
280 //
281 reprocall = false;
282 //
283 if ( verbose ) printf("\n ANTICOUNTER - WARNING: Reprocessing run number %u \n",run);
284 //
285 // copying old tree to a new file
286 //
287 gSystem->MakeDirectory(acfolder.str().c_str());
288 myfold = true;
289 tempfile = new TFile(tempname.str().c_str(),"RECREATE");
290 tempac = acclone->CloneTree(-1,"fast");
291 tempac->SetName("Anticounter-old");
292 tempfile->Write();
293 tempfile->Close();
294 }
295 //
296 // Delete the old tree from old file and memory
297 //
298 acclone->Delete("all");
299 //
300 if ( verbose ) printf(" ...done!\n");
301 //
302 };
303 //
304 // create anticounter tree ac
305 //
306 file->cd();
307 ac = new TTree("Anticounter-new","PAMELA Level2 Anticounter data");
308 ac->SetAutoSave(900000000000000LL);
309 ac->Branch("AcLevel2","AcLevel2",&acl2);
310 //
311 if ( reproc && !reprocall ){
312 //
313 // open new file and retrieve alo tree informations
314 //
315 tempfile = new TFile(tempname.str().c_str(),"READ");
316 acclone = (TTree*)tempfile->Get("Anticounter-old");
317 acclone->SetAutoSave(900000000000000LL);
318 acclone->SetBranchAddress("AcLevel2",&acl2clone);
319 //
320 if ( nobefrun > 0 ){
321 if ( verbose ) {
322 printf("\n Pre-processing: copying events from the old tree before the processed run\n");
323 printf(" Copying %u events in the file which are before the beginning of the run %u \n",nobefrun,run);
324 printf(" Start copying at event number 0, end copying at event number %u \n",nobefrun);
325 };
326 for (UInt_t j = 0; j < nobefrun; j++){
327 //
328 acclone->GetEntry(j);
329 //
330 // copy acl2clone to acl2
331 //
332 acl2->Clear();
333 //
334 memcpy(&acl2,&acl2clone,sizeof(acl2clone));
335 //
336 // Fill entry in the new tree
337 //
338 ac->Fill();
339 //
340 };
341 if ( verbose ) printf(" Finished successful copying!\n");
342 };
343 };
344 //
345 // Get the list of run to be processed, if only one run has to be processed the list will contain one entry only.
346 //
347 runlist = runinfo->GetRunList();
348 //
349 // Loop over the run to be processed
350 //
351 for (UInt_t irun=0; irun < numbofrun; irun++){
352 //
353 // retrieve the first run ID to be processed using the RunInfo list
354 //
355 idRun = runlist->At(irun);
356 if ( verbose ){
357 printf("\n\n\n ####################################################################### \n");
358 printf(" PROCESSING RUN NUMBER %u \n",idRun);
359 printf(" ####################################################################### \n\n\n");
360 };
361 //
362 runinfo->ID_ROOT_L0 = 0;
363 //
364 // store in the runinfo class the GL_RUN variables for our run
365 //
366 sgnl = 0;
367 sgnl = runinfo->GetRunInfo(idRun);
368 if ( sgnl ){
369 if ( verbose ) printf(" ANTICOUNTER - ERROR: RunInfo exited with non-zero status\n");
370 code = sgnl;
371 goto closeandexit;
372 } else {
373 sgnl = 0;
374 };
375 //
376 // now you can access that variables using the RunInfo class this way runinfo->ID_ROOT_L0
377 //
378 if ( runinfo->ID_ROOT_L0 == 0 ){
379 if ( verbose ) printf("\n ANTICOUNTER - ERROR: no run with ID_RUN = %u \n\n Exiting... \n\n",idRun);
380 code = -5;
381 goto closeandexit;
382 };
383 //
384 // prepare the timesync for the db
385 //
386 TString host = glt->CGetHost();
387 TString user = glt->CGetUser();
388 TString psw = glt->CGetPsw();
389 TSQLServer *dbc = TSQLServer::Connect(host.Data(),user.Data(),psw.Data());
390 if ( !dbc->IsConnected() ) throw -705;
391 dbtime = new GL_TIMESYNC(runinfo->ID_ROOT_L0,"ID",dbc);
392 //
393 // Search in the DB the path and name of the LEVEL0 file to be processed.
394 //
395 // if ( !dbc->IsConnected() ) throw -705;
396 glroot->Query_GL_ROOT(runinfo->ID_ROOT_L0,dbc);
397 //
398 ftmpname.str("");
399 ftmpname << glroot->PATH.Data() << "/";
400 ftmpname << glroot->NAME.Data();
401 fname = ftmpname.str().c_str();
402 //
403 // print out informations
404 //
405 totevent = runinfo->NEVENTS;
406 if ( verbose ) {
407 printf("\n LEVEL0 data file: %s \n",fname.Data());
408 printf(" RUN HEADER absolute time is: %u \n",runinfo->RUNHEADER_TIME);
409 printf(" RUN TRAILER absolute time is: %u \n",runinfo->RUNTRAILER_TIME);
410 printf(" %i events to be processed for run %u: from %i to %i \n\n",totevent,idRun,runinfo->EV_FROM,runinfo->EV_FROM+totevent);
411 };
412 //
413 // Open Level0 file
414 //
415 l0File = new TFile(fname.Data());
416 if ( !l0File ) {
417 if ( verbose ) printf(" ANTICOUNTER - ERROR: problems opening Level0 file\n");
418 code = -6;
419 goto closeandexit;
420 };
421 l0tr = (TTree*)l0File->Get("Physics");
422 if ( !l0tr ) {
423 if ( verbose ) printf(" ANTICOUNTER - ERROR: no Physics tree in Level0 file\n");
424 l0File->Close();
425 code = -7;
426 goto closeandexit;
427 };
428 l0head = l0tr->GetBranch("Header");
429 if ( !l0head ) {
430 if ( verbose ) printf(" ANTICOUNTER - ERROR: no Header branch in Level0 tree\n");
431 l0File->Close();
432 code = -8;
433 goto closeandexit;
434 };
435 l0ac = l0tr->GetBranch("Anticounter");
436 if ( !l0ac ) {
437 if ( verbose ) printf(" ANTICOUNTER - ERROR: no Anticounter branch in Level0 tree\n");
438 l0File->Close();
439 code = -704;
440 goto closeandexit;
441 };
442 //
443 l0tr->SetBranchAddress("Anticounter", &acc);
444 l0tr->SetBranchAddress("Header", &eh);
445 //
446 nevents = l0ac->GetEntries();
447 //
448 if ( nevents < 1 ) {
449 if ( verbose ) printf(" ANTICOUNTER - ERROR: Level0 file is empty\n\n");
450 l0File->Close();
451 code = -11;
452 goto closeandexit;
453 };
454 //
455 if ( runinfo->EV_TO > nevents-1 ) {
456 if ( verbose ) printf(" ANTICOUNTER - ERROR: too few entries in the tree\n");
457 l0File->Close();
458 code = -12;
459 goto closeandexit;
460 };
461 //
462 // run over all the events of the run
463 //
464 if ( dbc ){
465 dbc->Close();
466 delete dbc;
467 };
468 //
469 if ( verbose ) printf("\n Ready to start! \n\n Processed events: \n\n");
470 //
471 for ( re = runinfo->EV_FROM; re < (runinfo->EV_FROM+runinfo->NEVENTS); re++){
472 //
473 if ( procev%1000 == 0 && procev > 0 && verbose ) printf(" %iK \n",procev/1000);
474 //
475 l0head->GetEntry(re);
476 //
477 // absolute time of this event
478 //
479 ph = eh->GetPscuHeader();
480 atime = dbtime->DBabsTime(ph->GetOrbitalTime());
481 //
482 // paranoid check
483 //
484 if ( atime > runinfo->RUNTRAILER_TIME || atime < runinfo->RUNHEADER_TIME ) {
485 if ( verbose ) printf(" ANTICOUNTER - WARNING: event at time outside the run time window, skipping it\n");
486 goto jumpev;
487 };
488 //
489 procev++;
490 //
491 // start processing
492 //
493 acl2->Clear();
494 //
495 l0ac->GetEntry(re);
496 //
497 /***********************************************/
498 // Here starts Anticounter specific code (Silvio)
499 //
500 memset(vec, 0, 2*16*16*sizeof(UShort_t));
501 memset(Shift, 0, 2*16*sizeof(UShort_t));
502 memset(Counter, 0, 2*16*sizeof(UShort_t));
503 memset(Counter_old, 0, 2*16*sizeof(UShort_t));
504 memset(CRCcheck, 0, 2*sizeof(UShort_t));
505 memset(Status, 0, 2*sizeof(UShort_t));
506 memset(Hitmap, 0, 2*sizeof(UShort_t));
507 memset(Hitstatus, 0, 2*sizeof(UShort_t));
508 memset(Trigger, 0, 2*sizeof(UShort_t));
509 memset(Trigg_old, 0, 2*sizeof(UShort_t));
510 tmp = 0;
511 //
512 for(Int_t gg = 0; gg < 2; gg++)
513 {
514 Hitstatus[gg] = 0;
515 }
516 //
517 //fetch data
518 //
519 for(Int_t kk = 0; kk < 2; kk++)
520 {
521 if(re>0)
522 {
523 Trigg_old[kk] = Trigger[kk];
524 for(Int_t jj = 0; jj < 16; jj++)
525 Counter_old[kk][jj] = Counter[kk][jj];
526 }
527
528 for(Int_t jj = 0; jj < 16; jj++)
529 {
530 Counter[kk][jj] = acc->counters[kk][jj];
531 Shift[kk][jj] = acc->shift[kk][jj];
532 }
533 Status[kk] = acc->status[kk];
534 Hitmap[kk] = acc->hitmap[kk];
535 Trigger[kk] = acc->trigg[kk];
536 CRCcheck[kk] = acc->CRCcheck[kk];
537 }
538
539 /***********************************************/
540 //process data
541 /***********************************************/
542 //shiftregisters
543 for(Int_t b = 0; b < 2; b++){ //card
544 for(Int_t k = 0; k < 16; k++){ //shift register
545 Int_t cnt=1;
546 for(Int_t l = 0; l < 16; l++){ //bin
547 tmp = ((Shift[b][k] & cnt) > 0 ? 1 : 0);
548 vec[b][k][l]=tmp;
549 cnt=cnt<<1;
550 }
551 //cnt=1;
552 }
553 }
554 //
555 //fill Level1 file
556 /************************************************/
557 for(Int_t s = 0; s < 2; s++) {
558 acl2->hitmap[s] = Hitmap[s];
559 acl2->trigger[s] = Trigger[s];
560 Int_t cnt=1;
561 for(Int_t k = 0; k < 16; k++) {
562 for(Int_t bin = 5; bin < 9; bin++) { //acceptance window
563 if(vec[s][bin][k]==1)
564 Hitstatus[s] = Hitstatus[s] | cnt;
565 }
566 cnt=cnt<<1;
567 }
568 acl2->hitstatus[s] = Hitstatus[s];
569 //
570 //Status
571 /****************************************/
572 if(s==0){
573 if(Trigger[0] != (Trigg_old[0]+1))
574 acl2->status[0] = acl2->status[0] | 0x1;
575 if(Status[0] & 0x001F < 0x001F)
576 acl2->status[0] = acl2->status[0] | 0x2;
577 if(CRCcheck[0] == 0)
578 acl2->status[0] = acl2->status[0] | 0x4;
579 for(Int_t gg = 0; gg < 16; gg++){
580 if((Counter[0][gg] == Counter_old[0][gg]) && re>0){
581 acl2->status[0] = acl2->status[0] | 0x8;
582 }
583 }
584 }
585 else {
586 if(Trigger[1] != (Trigg_old[1]+2))
587 acl2->status[1] = acl2->status[1] | 0x1;
588 if(Status[1] & 0x001F < 0x001F)
589 acl2->status[1] = acl2->status[1] | 0x2;
590 if(CRCcheck[1] == 0)
591 acl2->status[1] = acl2->status[1] | 0x4;
592 for(Int_t gg = 0; gg < 16; gg++){
593 if((Counter[1][gg] == Counter_old[1][gg]) && re>0)
594 acl2->status[1] = acl2->status[1] | 0x8;
595 }
596 }
597 }
598 //
599 // End Anticounter specific code
600 //
601 /************************************************/
602 acl2->unpackError = acc->unpackError;
603 //
604 // Fill the rootple
605 //
606 ac->Fill();
607 //
608 //
609 jumpev:
610 debug = false;
611 //
612 };
613 //
614 // Here you may want to clear some variables before processing another run
615 //
616 delete dbtime;
617 }; // process all the runs
618 //
619 if ( verbose ) printf("\n Finished processing data \n");
620 //
621 closeandexit:
622 //
623 // we have finished processing the run(s). If we processed a single run now we must copy all the events after our run from the old tree to the new one and delete the old tree.
624 //
625 if ( !reprocall && reproc && code >= 0 ){
626 if ( totfileentries > noaftrun ){
627 if ( verbose ){
628 printf("\n Post-processing: copying events from the old tree after the processed run\n");
629 printf(" Copying %i events in the file which are after the end of the run %i \n",(int)(totfileentries-noaftrun),(int)run);
630 printf(" Start copying at event number %i end copying at event number %i \n",(int)noaftrun,(int)totfileentries);
631 };
632 for (UInt_t j = noaftrun; j < totfileentries; j++ ){
633 //
634 // Get entry from old tree
635 //
636 acclone->GetEntry(j);
637 //
638 // copy acl2clone to acl2
639 //
640 acl2->Clear();
641 memcpy(&acl2,&acl2clone,sizeof(acl2clone));
642 //
643 // Fill entry in the new tree
644 //
645 ac->Fill();
646 };
647 if ( verbose ) printf(" Finished successful copying!\n");
648 };
649 };
650 //
651 // Close files, delete old tree(s), write and close level2 file
652 //
653 if ( l0File ) l0File->Close();
654 if ( tempfile ) tempfile->Close();
655 if ( myfold ) gSystem->Unlink(tempname.str().c_str());
656 //
657 if ( code < 0 && verbose ) printf("\n ANTICOUNTER - ERROR: an error occurred, try to save anyway...\n");
658 if ( verbose ) printf("\n Writing and closing rootple\n");
659 if ( runinfo ) runinfo->Close();
660 if ( ac ) ac->SetName("Anticounter");
661 if ( file ){
662 file->cd();
663 file->Write();
664 };
665 //
666 if ( myfold ) gSystem->Unlink(acfolder.str().c_str());
667 //
668 // the end
669 //
670 if ( verbose ) printf("\n Exiting...\n");
671 //
672 if( ac ) ac->Delete();
673 if ( acl2 ) delete acl2;
674 if ( acl2clone ) delete acl2clone;
675 if ( glroot ) delete glroot;
676 if ( runinfo ) delete runinfo;
677 //
678 if(code < 0) throw code;
679 return(code);
680 }
681
682

  ViewVC Help
Powered by ViewVC 1.1.23