cbb7fa28ca549ab21e32166fc37384442ae8550a
[u/mrichter/AliRoot.git] / ZDC / ZDCPEDESTALda.cxx
1 /*
2
3 This program reads the DAQ data files passed as argument using the monitoring library.
4
5 It computes the average event size and populates local "./result.txt" file with the 
6 result.
7
8 The program reports about its processing progress.
9
10 Messages on stdout are exported to DAQ log system.
11
12 DA for ZDC standalone pedestal runs
13
14 Contact: Chiara.Oppedisano@to.infn.it
15 Link: 
16 Run Type: STANDALONE_PEDESTAL_RUN
17 DA Type: LDC
18 Number of events needed: no constraint (tipically ~10^3)
19 Input Files:  
20 Output Files: ZDCPedestal.dat, ZDCChMapping.dat
21 Trigger Types Used: Standalone Trigger
22
23 */
24 #define PEDDATA_FILE  "ZDCPedestal.dat"
25 #define MAPDATA_FILE  "ZDCChMapping.dat"
26
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <Riostream.h>
30
31 // DATE
32 #include <event.h>
33 #include <monitor.h>
34 #include <daqDA.h>
35
36 //ROOT
37 #include <TRandom.h>
38 #include <TH1F.h>
39 #include <TH2F.h>
40 #include <TProfile.h>
41 #include <TF1.h>
42 #include <TFile.h>
43 #include <TFitter.h>
44
45 //AliRoot
46 #include <AliRawReaderDate.h>
47 #include <AliRawEventHeaderBase.h>
48 #include <AliZDCRawStream.h>
49
50
51 /* Main routine
52       Arguments: list of DATE raw data files
53 */
54 int main(int argc, char **argv) {
55   
56   TFitter *minuitFit = new TFitter(4);
57   TVirtualFitter::SetFitter(minuitFit);
58
59   int status = 0;
60
61   /* log start of process */
62   printf("\n ZDC PEDESTAL program started\n");  
63
64   /* check that we got some arguments = list of files */
65   if (argc<2) {
66     printf("Wrong number of arguments\n");
67     return -1;
68   }
69
70   // --- Histograms for ADC pedestals 
71   //     [22 signal channels + 2 reference PTMs]  x 2 gain chains
72   //
73   TH1F::AddDirectory(0);
74   int const kNChannels = 24;
75   TH1F *hPedhg[kNChannels], *hPedOutOfTimehg[kNChannels];
76   TH2F *hPedCorrhg[kNChannels];
77   TH1F *hPedlg[kNChannels], *hPedOutOfTimelg[kNChannels];
78   TH2F *hPedCorrlg[kNChannels];
79   //
80   char namhist1hg[50], namhist2hg[50], namhist3hg[50];
81   char namhist1lg[50], namhist2lg[50], namhist3lg[50];
82   for(Int_t j=0; j<kNChannels; j++){
83      if(j<5){ // ZN1
84        sprintf(namhist1hg,"PedZN1hg_%d",j);
85        sprintf(namhist2hg,"PedZN1hgOutOfTime_%d",j);
86        sprintf(namhist3hg,"PedCorrZN1hg_%d",j);
87        //
88        sprintf(namhist1lg,"PedZN1lg_%d",j);
89        sprintf(namhist2lg,"PedZN1lgOutOfTime_%d",j);
90        sprintf(namhist3lg,"PedCorrZN1lg_%d",j);
91      }
92      else if(j>=5 && j<10){ // ZP1
93        sprintf(namhist1hg,"PedZP1hg_%d",j-5);
94        sprintf(namhist2hg,"PedZP1hgOutOfTime_%d",j-5);
95        sprintf(namhist3hg,"PedCorrZP1hg_%d",j-5);
96        //
97        sprintf(namhist1lg,"PedZP1lg_%d",j-5);
98        sprintf(namhist2lg,"PedZP1lgOutOfTime_%d",j-5);
99        sprintf(namhist3lg,"PedCorrZP1lg_%d",j-5);       
100      }
101      else if(j>=10 && j<12){ // ZEM
102        sprintf(namhist1hg,"PedZEMhg_%d",j-10);
103        sprintf(namhist2hg,"PedZEMhgOutOfTime_%d",j-10);
104        sprintf(namhist3hg,"PedCorrZEMhg_%d",j-10);
105        //
106        sprintf(namhist1lg,"PedZEMlg_%d",j-10);
107        sprintf(namhist2lg,"PedZEMlgOutOfTime_%d",j-10);
108        sprintf(namhist3lg,"PedCorrZEMlg_%d",j-10);
109      }
110      else if(j>=12 && j<17){ // ZN2
111        sprintf(namhist1hg,"PedZN2hg_%d",j-12);
112        sprintf(namhist2hg,"PedZN2hgOutOfTime_%d",j-12);
113        sprintf(namhist3hg,"PedCorrZN2hg_%d",j-12);
114        //
115        sprintf(namhist1lg,"PedZN2lg_%d",j-12);
116        sprintf(namhist2lg,"PedZN2lgOutOfTime_%d",j-12);
117        sprintf(namhist3lg,"PedCorrZN2lg_%d",j-12);
118      }
119      else if(j>=17 && j<22){ // ZP2
120        sprintf(namhist1hg,"PedZP2hg_%d",j-17);
121        sprintf(namhist2hg,"PedZP2hgOutOfTime_%d",j-17);
122        sprintf(namhist3hg,"PedCorrZP2hg_%d",j-17);
123        //
124        sprintf(namhist1lg,"PedZP2lg_%d",j-17);
125        sprintf(namhist2lg,"PedZP2lgOutOfTime_%d",j-17);
126        sprintf(namhist3lg,"PedCorrZP2lg_%d",j-17);
127      }
128      else if(j>=22 && j<24){ //Reference PMs
129        sprintf(namhist1hg,"PedRefhg_%d",j-22);
130        sprintf(namhist2hg,"PedRefhgOutOfTime_%d",j-22);
131        sprintf(namhist3hg,"PedCorrRefhg_%d",j-22);
132        //
133        sprintf(namhist1lg,"PedReflg_%d",j-22);
134        sprintf(namhist2lg,"PedReflgOutOfTime_%d",j-22);
135        sprintf(namhist3lg,"PedCorrReflg_%d",j-22);
136      }
137      // --- High gain chain histos
138      hPedhg[j] = new TH1F(namhist1hg, namhist1hg, 200,0., 200.);
139      hPedOutOfTimehg[j] = new TH1F(namhist2hg, namhist2hg, 200,0., 200.);
140      hPedCorrhg[j] = new TH2F(namhist3hg,namhist3hg,100,0.,200.,100,0.,200.);
141      // --- Low gain chain histos
142      hPedlg[j] = new TH1F(namhist1lg, namhist1lg, 100,0., 600.);
143      hPedOutOfTimelg[j] = new TH1F(namhist2lg, namhist2lg, 100,0., 600.);
144      hPedCorrlg[j] = new TH2F(namhist3lg,namhist3lg,100,0.,600.,100,0.,600.);
145   }
146
147
148   /* open result file */
149   FILE *fp=NULL;
150   fp=fopen("./result.txt","a");
151   if (fp==NULL) {
152     printf("Failed to open file\n");
153     return -1;
154   }
155   
156   FILE *mapFile4Shuttle;
157
158   /* report progress */
159   daqDA_progressReport(10);
160
161
162   /* init some counters */
163   int nevents_physics=0;
164   int nevents_total=0;
165
166   /* read the data files */
167   int n;
168   for(n=1;n<argc;n++){
169    
170     status=monitorSetDataSource( argv[n] );
171     if (status!=0) {
172       printf("monitorSetDataSource() failed : %s\n",monitorDecodeError(status));
173       return -1;
174     }
175
176     /* report progress */
177     /* in this example, indexed on the number of files */
178     daqDA_progressReport(10+80*n/argc);
179
180     /* read the file */
181     for(;;) {
182       struct eventHeaderStruct *event;
183       eventTypeType eventT;
184
185       /* get next event */
186       status=monitorGetEventDynamic((void **)&event);
187       if(status==MON_ERR_EOF) break; /* end of monitoring file has been reached */
188       if(status!=0) {
189         printf("monitorGetEventDynamic() failed : %s\n",monitorDecodeError(status));
190         return -1;
191       }
192
193       /* retry if got no event */
194       if(event==NULL) {
195         break;
196       }
197       
198       // Initalize raw-data reading and decoding
199       AliRawReader *reader = new AliRawReaderDate((void*)event);
200       reader->Select("ZDC");
201       // --- Reading event header
202       //UInt_t evtype = reader->GetType();
203       //printf("\n\t ZDCPEDESTALda -> ev. type %d\n",evtype);
204       //printf("\t ZDCPEDESTALda -> run # %d\n",reader->GetRunNumber());
205       //
206       AliZDCRawStream *rawStreamZDC = new AliZDCRawStream(reader);
207         
208
209       /* use event - here, just write event id to result file */
210       eventT=event->eventType;
211       
212       Int_t ich=0, adcMod[48], adcCh[48], sigCode[48], det[48], sec[48];
213       if(eventT==START_OF_DATA){
214                 
215         if(!rawStreamZDC->Next()) printf(" \t No raw data found!! \n");
216         else{
217           while(rawStreamZDC->Next()){
218             if(rawStreamZDC->IsChMapping()){
219               adcMod[ich] = rawStreamZDC->GetADCModFromMap(ich);
220               adcCh[ich] = rawStreamZDC->GetADCChFromMap(ich);
221               sigCode[ich] = rawStreamZDC->GetADCSignFromMap(ich);
222               det[ich] = rawStreamZDC->GetDetectorFromMap(ich);
223               sec[ich] = rawStreamZDC->GetTowerFromMap(ich);
224               ich++;
225             }
226           }
227         }
228         // --------------------------------------------------------
229         // --- Writing ascii data file for the Shuttle preprocessor
230         mapFile4Shuttle = fopen(MAPDATA_FILE,"w");
231         for(Int_t i=0; i<ich; i++){
232            fprintf(mapFile4Shuttle,"\t%d\t%d\t%d\t%d\t%d\t%d\n",i,
233              adcMod[i],adcCh[i],sigCode[i],det[i],sec[i]);
234            //
235            //printf("ZDCPEDESTALDA.cxx ->  ch.%d mod %d, ch %d, code %d det %d, sec %d\n",
236            //      i,adcMod[i],adcCh[i],sigCode[i],det[i],sec[i]);
237         }
238         fclose(mapFile4Shuttle);
239       }
240       
241       if(eventT==PHYSICS_EVENT){
242         fprintf(fp,"Run #%lu, event size: %lu, BC:%u, Orbit:%u, Period:%u\n",
243           (unsigned long)event->eventRunNb,
244           (unsigned long)event->eventSize,
245           EVENT_ID_GET_BUNCH_CROSSING(event->eventId),
246           EVENT_ID_GET_ORBIT(event->eventId),
247           EVENT_ID_GET_PERIOD(event->eventId));
248
249         // --- Reading data header
250         reader->ReadHeader();
251         const AliRawDataHeader* header = reader->GetDataHeader();
252         if(header){
253          UChar_t message = header->GetAttributes();
254          if(message & 0x20){ // PEDESTAL RUN
255             //printf("\t STANDALONE_PEDESTAL RUN raw data found\n");
256          }
257          else{
258             printf("\t NO STANDALONE_PEDESTAL RUN raw data found\n");
259             return -1;
260          }
261         }
262         else{
263            printf("\t ATTENTION! No Raw Data Header found!!!\n");
264            return -1;
265         }
266
267         if(!rawStreamZDC->Next()) printf(" \t No raw data found!! \n"); 
268         //
269         // ----- Setting ch. mapping -----
270         for(Int_t jk=0; jk<48; jk++){
271           rawStreamZDC->SetMapADCMod(jk, adcMod[jk]);
272           rawStreamZDC->SetMapADCCh(jk, adcCh[jk]);
273           rawStreamZDC->SetMapADCSig(jk, sigCode[jk]);
274           rawStreamZDC->SetMapDet(jk, det[jk]);
275           rawStreamZDC->SetMapTow(jk, sec[jk]);
276         }
277         //
278         Int_t iraw=0;
279         Int_t RawADChg[kNChannels], RawADCoothg[kNChannels];
280         Int_t RawADClg[kNChannels], RawADCootlg[kNChannels];
281         for(Int_t j=0; j<kNChannels; j++){
282            RawADChg[j]=0; RawADCoothg[j]=0;
283            RawADClg[j]=0; RawADCootlg[j]=0;
284         }
285         //
286         while(rawStreamZDC->Next()){
287          Int_t index=-1;
288          if(rawStreamZDC->IsADCDataWord()){
289           if(rawStreamZDC->GetSector(1)!=5){ // Physics signals
290             if(rawStreamZDC->GetSector(0)==1) index = rawStreamZDC->GetSector(1); // *** ZNC
291             else if(rawStreamZDC->GetSector(0)==2) index = rawStreamZDC->GetSector(1)+5; // *** ZPC
292             else if(rawStreamZDC->GetSector(0)==3) index = rawStreamZDC->GetSector(1)+9; // *** ZEM
293             else if(rawStreamZDC->GetSector(0)==4) index = rawStreamZDC->GetSector(1)+12; // *** ZNA
294             else if(rawStreamZDC->GetSector(0)==5) index = rawStreamZDC->GetSector(1)+17; // *** ZPA
295           }
296           else{ // Reference PMs
297             index = (rawStreamZDC->GetSector(0)-1)/3+22;
298           }
299           //
300           /*printf("\t iraw %d index %d det %d quad %d res %d ADC %d\n", iraw, index,
301                 rawStreamZDC->GetSector(0), rawStreamZDC->GetSector(1), 
302                 rawStreamZDC->GetADCGain(), rawStreamZDC->GetADCValue());
303           */
304            //
305            if(iraw<2*kNChannels){ // --- In-time pedestals (1st 48 raw data)
306             if(rawStreamZDC->GetADCGain()==0){ 
307               hPedhg[index]->Fill(rawStreamZDC->GetADCValue()); 
308               RawADChg[index] = rawStreamZDC->GetADCValue();
309               //
310               //printf("\t filling histo hPedhg[%d]\n",index);
311             }
312             else{
313               hPedlg[index]->Fill(rawStreamZDC->GetADCValue()); 
314               RawADClg[index] = rawStreamZDC->GetADCValue();
315               //
316               //printf("\t filling histo hPedlg[%d]\n",index);
317             }
318            }
319            else{  // --- Out-of-time pedestals
320             if(rawStreamZDC->GetADCGain()==0){
321               hPedOutOfTimehg[index]->Fill(rawStreamZDC->GetADCValue());
322               RawADCoothg[index] = rawStreamZDC->GetADCValue();
323               //
324               //printf("\t filling histo hPedOutOfTimehg[%d]\n",index);
325             }
326             else{
327               hPedOutOfTimelg[index]->Fill(rawStreamZDC->GetADCValue());
328               RawADCootlg[index] = rawStreamZDC->GetADCValue();
329               //
330               //printf("\t filling histo hPedOutOfTimelg[%d]\n",index);
331             }
332            }
333             iraw++;
334           }//IsADCDataWord()
335           //
336           if(iraw == 4*kNChannels){ // Last ADC channel -> Filling correlation histos
337             for(Int_t k=0; k<kNChannels; k++){
338               hPedCorrhg[k]->Fill(RawADCoothg[k], RawADChg[k]);
339               hPedCorrlg[k]->Fill(RawADCootlg[k], RawADClg[k]);
340             }
341           }
342          }
343          //
344          nevents_physics++;
345          //
346          delete reader;
347          delete rawStreamZDC;
348
349       }//(if PHYSICS_EVENT) 
350       nevents_total++;
351
352       /* free resources */
353       free(event);
354     
355     }
356   }  
357   
358   /* Analysis of the histograms */
359   //
360   FILE *fileShuttle;
361   fileShuttle = fopen(PEDDATA_FILE,"w");
362   //
363   Float_t MeanPed[2*kNChannels], MeanPedWidth[2*kNChannels], 
364         MeanPedOOT[2*kNChannels], MeanPedWidthOOT[2*kNChannels];
365   // --- In-time pedestals
366   TF1 *ADCfunchg[kNChannels];
367   for(Int_t i=0; i<kNChannels; i++){
368      hPedhg[i]->Fit("gaus","Q");
369      ADCfunchg[i] = hPedhg[i]->GetFunction("gaus");
370      MeanPed[i] = (Double_t) ADCfunchg[i]->GetParameter(1);
371      MeanPedWidth[i] = (Double_t)  ADCfunchg[i]->GetParameter(2);
372      fprintf(fileShuttle,"\t%f\t%f\n",MeanPed[i],MeanPedWidth[i]);
373      //printf("\t MeanPed[%d] = %f\n",i, MeanPed[i]);
374   }
375   TF1 *ADCfunclg[kNChannels];
376   for(Int_t i=0; i<kNChannels; i++){
377      hPedlg[i]->Fit("gaus","Q");
378      ADCfunclg[i] = hPedlg[i]->GetFunction("gaus");
379      MeanPed[i+kNChannels] = (Double_t)  ADCfunclg[i]->GetParameter(1);
380      MeanPedWidth[i+kNChannels] = (Double_t)  ADCfunclg[i]->GetParameter(2);
381      fprintf(fileShuttle,"\t%f\t%f\n",MeanPed[i+kNChannels],MeanPedWidth[i+kNChannels]);
382      //printf("\t MeanPed[%d] = %f\n",i+kNChannels, MeanPed[i+kNChannels]);
383   }
384   // --- Out-of-time pedestals
385   TF1 *ADCootfunchg[kNChannels];
386   for(Int_t i=0; i<kNChannels; i++){
387      hPedOutOfTimehg[i]->Fit("gaus","Q");
388      ADCootfunchg[i] = hPedOutOfTimehg[i]->GetFunction("gaus");
389      MeanPedOOT[i] = (Double_t)  ADCootfunchg[i]->GetParameter(1);
390      MeanPedWidthOOT[i] = (Double_t)  ADCootfunchg[i]->GetParameter(2);
391      fprintf(fileShuttle,"\t%f\t%f\n",MeanPedOOT[i],MeanPedWidthOOT[i]);
392      //printf("\t MeanPedOOT[%d] = %f\n",i, MeanPedOOT[i]);
393   }
394   TF1 *ADCootfunclg[kNChannels];
395   for(Int_t i=0; i<kNChannels; i++){
396      hPedOutOfTimelg[i]->Fit("gaus","Q");
397      ADCootfunclg[i] = hPedOutOfTimelg[i]->GetFunction("gaus");
398      MeanPedOOT[i+kNChannels] = (Double_t)  ADCootfunclg[i]->GetParameter(1);
399      MeanPedWidthOOT[i+kNChannels] = (Double_t)  ADCootfunclg[i]->GetParameter(2);
400      fprintf(fileShuttle,"\t%f\t%f\n",MeanPedOOT[i+kNChannels],MeanPedWidthOOT[i+kNChannels]);
401      //printf("\t MeanPedOOT[%d] = %f\n",i+kNChannels, MeanPedOOT[i+kNChannels]);
402   }
403   
404   // ***************************************************
405   //   Unless we have a narrow correlation to fit we
406   //    don't fit and store in-time vs. out-of-time
407   //    histograms -> mean pedstal subtracted!!!!!!
408   // ***************************************************
409   // --- Correlations
410 /*
411   Float_t CorrCoeff0[2*kNChannels], CorrCoeff1[2*kNChannels];
412   TProfile *hPedCorrProfhg[kNChannels], *hPedCorrProflg[kNChannels];
413   TF1 *ffunchg[kNChannels], *ffunclg[kNChannels];
414   char namhist4[50];
415   for(int i=0;i<kNChannels;i++) {
416      sprintf(namhist4,"ADCHRvsOOT%d_Prof",i);
417      hPedCorrProfhg[i] = hPedCorrhg[i]->ProfileX(namhist4,-1,-1,"S");
418      hPedCorrProfhg[i]->SetName(namhist4);
419      hPedCorrProfhg[i]->Fit("pol1","Q");
420      ffunchg[i] = hPedCorrProfhg[i]->GetFunction("pol1");
421      CorrCoeff0[i] = (Double_t)  ffunchg[i]->GetParameter(0);
422      CorrCoeff1[i] = (Double_t) ffunchg[i]->GetParameter(1);
423      fprintf(fileShuttle,"\t%d\t%f\t%f\n",i,CorrCoeff0[i],CorrCoeff1[i]);
424      //printf("\t CorrCoeff0[%d] = %f, CorrCoeff1[%d] = %f\n",i, CorrCoeff0[i], i, CorrCoeff1[i]);
425   }    
426   for(int i=0;i<kNChannels;i++) {
427      sprintf(namhist4,"ADCLRvsOOT%d_Prof",i);
428      hPedCorrProflg[i] = hPedCorrlg[i]->ProfileX(namhist4,-1,-1,"S");
429      hPedCorrProflg[i]->SetName(namhist4);
430      hPedCorrProflg[i]->Fit("pol1","Q");
431      ffunclg[i] = hPedCorrProflg[i]->GetFunction("pol1");
432      CorrCoeff0[i+kNChannels] =  (Double_t) ffunclg[i]->GetParameter(0);
433      CorrCoeff1[i+kNChannels] =  (Double_t) ffunclg[i]->GetParameter(1);
434      fprintf(fileShuttle,"\t%d\t%f\t%f\n",i+kNChannels,CorrCoeff0[i+kNChannels],CorrCoeff1[i+kNChannels]);
435      //printf("\t CorrCoeff0[%d] = %f, CorrCoeff1[%d] = %f\n",
436      //         i+kNChannels, CorrCoeff0[i+kNChannels], i+kNChannels, CorrCoeff1[i+kNChannels]);
437   }    
438 */
439   //                                                   
440   fclose(fileShuttle);
441   //
442   for(Int_t j=0; j<kNChannels; j++){
443      delete hPedhg[j];
444      delete hPedOutOfTimehg[j];
445      delete hPedCorrhg[j];
446      delete hPedlg[j];
447      delete hPedOutOfTimelg[j];
448      delete hPedCorrlg[j];
449   }
450
451   //delete minuitFit;
452   TVirtualFitter::SetFitter(0);
453
454   /* write report */
455   fprintf(fp,"Run #%s, received %d physics events out of %d\n",getenv("DATE_RUN_NUMBER"),nevents_physics,nevents_total);
456
457   /* close result file */
458   fclose(fp);
459   
460   /* report progress */
461   daqDA_progressReport(90);
462
463   /* store the result files on FES */
464   // [1] File with mapping
465   status = daqDA_FES_storeFile(MAPDATA_FILE,MAPDATA_FILE);
466   if(status){
467     printf("Failed to export file %d to DAQ FES\n",status);
468     return -1;
469   }
470   // [2] File with pedestal data
471   status = daqDA_FES_storeFile(PEDDATA_FILE,PEDDATA_FILE);
472   if(status){
473     printf("Failed to export file %d to DAQ FES\n",status);
474     return -1;
475   }
476   
477   /* store the result files on DB */
478   status = daqDA_DB_storeFile(PEDDATA_FILE,PEDDATA_FILE);  
479   if(status){
480     printf("Failed to export file %d to DAQ DB\n",status);
481     return -1;
482   }
483
484   /* report progress */
485   daqDA_progressReport(100);
486
487   return status;
488 }