Update TPCCEda to write output file in parts (to avoid too big files produced in...
[u/mrichter/AliRoot.git] / TPC / TPCCEda.cxx
1 /*
2 TPC DA for online calibration
3
4 Contact: Haavard.Helstrup@cern.ch
5 Link: 
6 Run Type: PHYSICS STANDALONE DAQ
7 DA Type: MON
8 Number of events needed: 500
9 Input Files: 
10 Output Files: tpcCE.root, to be exported to the DAQ FXS
11 fileId:   CE
12 Trigger types used: PHYSICS_EVENT
13
14 */
15
16 /*
17
18 TPCCEda.cxx - calibration algorithm for TPC Central Electrode events
19
20 10/06/2007  sylvain.chapeland@cern.ch :  first version - clean skeleton based on DAQ DA case1
21 06/12/2007  haavard.helstrup@cern.ch  :  created CE DA based on pulser code
22 19/09/2008  J.Wiechula@gsi.de:           Added support for configuration files.
23
24 contact: marian.ivanov@cern.ch
25
26
27 This process reads RAW data from the files provided as command line arguments
28 and save results in a file (named from RESULT_FILE define - see below).
29
30 */
31
32 #define RESULT_FILE "tpcCE.root"
33 #define FILE_ID "CE"
34 #define MAPPING_FILE "tpcMapping.root"
35 #define CONFIG_FILE "TPCCEda.conf"
36
37 #include <daqDA.h>
38 #include "event.h"
39 #include "monitor.h"
40 #include <stdio.h>
41 #include <stdlib.h>
42 #include <vector>
43 //
44 //Root includes
45 //
46 #include <TFile.h>
47 #include "TROOT.h"
48 #include "TPluginManager.h"
49 #include "TSystem.h"
50 #include "TString.h"
51 #include "TObjString.h"
52 #include "TDatime.h"
53 #include "TStopwatch.h"
54 #include "TMap.h"
55 #include "TGraph.h"
56 #include "TMath.h"
57 //
58 //AliRoot includes
59 //
60 #include "AliRawReader.h"
61 #include "AliRawReaderDate.h"
62 #include "AliTPCmapper.h"
63 #include "AliTPCRawStream.h"
64 #include "AliTPCROC.h"
65 #include "AliTPCCalROC.h"
66 #include "AliTPCCalPad.h"
67 #include "AliMathBase.h"
68 #include "TTreeStream.h"
69 #include "AliLog.h"
70 #include "AliTPCConfigDA.h"
71 //
72 //AMORE
73 //
74 #include <AmoreDA.h>
75 //
76 // TPC calibration algorithm includes
77 //
78 #include "AliTPCCalibCE.h"
79
80
81 //functios, implementation below
82 void SendToAmoreDB(AliTPCCalibCE *calibCE, unsigned long32 runNb);
83 //for threaded processing
84
85
86 /* Main routine
87       Arguments: list of DATE raw data files
88 */
89 int main(int argc, char **argv) {
90   /* log start of process */
91   printf("TPCCEda: DA started - %s\n",__FILE__);
92   
93   if (argc<2) {
94     printf("TPCCEda: Wrong number of arguments\n");
95     return -1;
96   }
97   
98   AliLog::SetClassDebugLevel("AliTPCRawStream",-5);
99   AliLog::SetClassDebugLevel("AliRawReaderDate",-5);
100   AliLog::SetClassDebugLevel("AliTPCAltroMapping",-5);
101   AliLog::SetModuleDebugLevel("RAW",-5);
102   
103   gROOT->GetPluginManager()->AddHandler("TVirtualStreamerInfo",
104                                         "*",
105                                         "TStreamerInfo",
106                                         "RIO",
107                                         "TStreamerInfo()");
108   
109  /* declare monitoring program */
110   int i,status;
111   status=monitorDeclareMp( __FILE__ );
112   if (status!=0) {
113     printf("TPCCEda: monitorDeclareMp() failed : %s\n",monitorDecodeError(status));
114     return -1;
115   }
116   monitorSetNowait();
117   monitorSetNoWaitNetworkTimeout(1000);
118
119   //variables
120   AliTPCmapper *mapping = 0;   // The TPC mapping
121   char localfile[255];
122   unsigned long32 runNb=0;      //run number
123   
124  
125   //
126   // DA configuration from configuration file
127   //
128   //retrieve configuration file
129   sprintf(localfile,"./%s",CONFIG_FILE);
130   status = daqDA_DB_getFile(CONFIG_FILE,localfile);
131   if (status) {
132     printf("TPCCEda: Failed to get configuration file (%s) from DAQdetDB, status=%d\n", CONFIG_FILE, status);
133     return -1;
134   }
135   AliTPCConfigDA config(CONFIG_FILE);
136   // check configuration options
137   TString laserTriggerName("C0LSR-ABCE-NOPF-CENT");
138   TString monitoringType("YES");
139   Int_t   forceTriggerId=-1;
140   Int_t   saveOption=2; // how to store the object. See AliTPCCalibCE::DumpToFile
141   
142   if ( config.GetConfigurationMap()->GetValue("LaserTriggerName") ) {
143     laserTriggerName=config.GetConfigurationMap()->GetValue("LaserTriggerName")->GetName();
144     printf("TPCCEda: Laser trigger class name set to: %s.\n",laserTriggerName.Data());
145   }
146
147   if ( config.GetConfigurationMap()->GetValue("MonitoringType") ) {
148     monitoringType=config.GetConfigurationMap()->GetValue("MonitoringType")->GetName();
149     printf("TPCCEda: Monitoring type set to: %s.\n",monitoringType.Data());
150   }
151
152   if ( config.GetConfigurationMap()->GetValue("ForceLaserTriggerId") ) {
153     forceTriggerId=TMath::Nint(config.GetValue("ForceLaserTriggerId"));
154     printf("TPCCEda: Only processing triggers with Id: %d.\n",forceTriggerId);
155   }
156   
157   if ( config.GetConfigurationMap()->GetValue("SaveOption") ) {
158     saveOption=TMath::Nint(config.GetValue("SaveOption"));
159     printf("TPCCEda: Saving option set to: %d.\n",saveOption);
160   }
161   
162   //subsribe to laser triggers only in physics partition
163   //if the trigger class is not available the return value is -1
164   //in this case we are most probably running as a standalone
165   //  laser run and should request all events
166   unsigned char classIdptr=0;
167   int retClassId=daqDA_getClassIdFromName(laserTriggerName.Data(),&classIdptr);
168   if (retClassId==0){
169     //interleaved laser in physics runs
170     //select proper trigger class id
171     char c[5];
172     snprintf(c,sizeof(c),"%u",(unsigned int)classIdptr);
173     char *table[5] = {"PHY",const_cast<char*>(monitoringType.Data()),"*",c,NULL};
174     monitorDeclareTableExtended(table);
175     printf("TPCCEda: Using monitoring table: (PHY, %s, *, %s)\n",monitoringType.Data(),c);
176   } else if (retClassId==-1){
177     //global partition without laser triggered events
178     //the DA should exit properly without processing
179     printf("TPCCEda: Laser trigger class '%s' was not found among trigger class names. Will stop processing.\n",laserTriggerName.Data());
180     return 0;
181   } else if (retClassId==-2){
182     //standalone case, accept all physics events
183     char *table[5] = {"PHY","Y","*","*",NULL};
184     monitorDeclareTableExtended(table);
185     printf("TPCCEda: Using all trigger class Ids\n");
186   } else {
187     printf("TPCCEda: Unknown return value of 'daqDA_getClassIdFromName': %d\n",retClassId);
188     return -2;
189   }
190
191   //see if we should force the trigger id
192   if (forceTriggerId>-1){
193     char c[5];
194     sprintf(c,"%d",forceTriggerId);
195     char *table[5] = {"PHY","Y","*",c,NULL};
196     monitorDeclareTableExtended(table);
197   }
198   
199   
200   // if  test setup get parameters from $DAQDA_TEST_DIR
201   if (!mapping){
202     /* copy locally the mapping file from daq detector config db */
203     sprintf(localfile,"./%s",MAPPING_FILE);
204     status = daqDA_DB_getFile(MAPPING_FILE,localfile);
205     if (status) {
206       printf("TPCCEda: Failed to get mapping file (%s) from DAQdetDB, status=%d\n", MAPPING_FILE, status);
207       return -1;
208     }
209     
210     /* open the mapping file and retrieve mapping object */
211     TFile *fileMapping = new TFile(MAPPING_FILE, "read");
212     mapping = (AliTPCmapper*) fileMapping->Get("tpcMapping");
213     delete fileMapping;
214   }
215   
216   if (mapping == 0) {
217     printf("TPCCEda: Failed to get mapping object from %s.  ...\n", MAPPING_FILE);
218     return -1;
219   } else {
220     printf("TPCCEda: Got mapping object from %s\n", MAPPING_FILE);
221   }
222   
223     
224   //create calibration object
225   AliTPCCalibCE *calibCE=new AliTPCCalibCE(config.GetConfigurationMap());   // central electrode calibration
226   calibCE->SetAltroMapping(mapping->GetAltroMapping()); // Use altro mapping we got from daqDetDb
227
228   //amore update interval
229   Double_t updateInterval=300; //seconds
230   Double_t valConf=config.GetValue("AmoreUpdateInterval");
231   if ( valConf>0 ) updateInterval=valConf;
232   //timer
233   TStopwatch stopWatch;
234   
235   //===========================//
236   // loop over RAW data files //
237   //==========================//
238   int nevents=0;
239   int neventsOld=0;
240   size_t counter=0;
241   for ( i=1; i<argc; i++) {
242     
243     /* define data source : this is argument i */
244     printf("TPCCEda: Processing file %s\n", argv[i]);
245     status=monitorSetDataSource( argv[i] );
246     if (status!=0) {
247       printf("TPCCEda: monitorSetDataSource() failed : %s\n",monitorDecodeError(status));
248       return -1;
249     }
250
251     
252     /* read until EOF */
253     while (true) {
254       struct eventHeaderStruct *event;
255       
256       /* check shutdown condition */
257       if (daqDA_checkShutdown()) {break;}
258         
259       /* get next event (blocking call until timeout) */
260       status=monitorGetEventDynamic((void **)&event);
261       if (status==MON_ERR_EOF) {
262         printf ("TPCCEda: End of File %d detected\n",i);
263         break; /* end of monitoring file has been reached */
264       }
265       
266       if (status!=0) {
267         printf("TPCCEda: monitorGetEventDynamic() failed : %s\n",monitorDecodeError(status));
268         break;
269       }
270       
271         /* retry if got no event */
272       if (event==NULL){
273         //use time in between bursts to
274         // send the data to AMOREdb
275         if (stopWatch.RealTime()>updateInterval){
276           calibCE->Analyse();
277           SendToAmoreDB(calibCE,runNb);
278           stopWatch.Start();
279         } else {
280           stopWatch.Continue();
281         }
282         //debug output
283         if (nevents>neventsOld){
284           printf ("TPCCEda: %d events processed, %d used\n",nevents,calibCE->GetNeventsProcessed());
285           neventsOld=nevents;
286         }
287         
288         continue;
289       }
290       
291       /* skip start/end of run events */
292       if ( (event->eventType != physicsEvent) && (event->eventType != calibrationEvent) ){
293         free(event);
294         continue;
295       }
296       
297       
298       // get the run number
299       runNb = event->eventRunNb;
300       
301       // CE calibration
302       calibCE->ProcessEvent(event);
303       
304       /* free resources */
305       free(event);
306       ++nevents;
307     }
308   }
309   
310   //
311   // Analyse CE data and write them to rootfile
312   //
313 //   calibCE->Analyse();
314   printf ("TPCCEda: %d events processed, %d used\n",nevents,calibCE->GetNeventsProcessed());
315   
316 //   TFile * fileTPC = new TFile (RESULT_FILE,"recreate");
317 //   calibCE->Write("tpcCalibCE");
318 //   delete fileTPC;
319   calibCE->DumpToFile(RESULT_FILE,Form("name=tpcCalibCE,type=%d",saveOption));
320   printf("TPCCEda: Wrote %s\n",RESULT_FILE);
321   
322   /* store the result file on FES */
323   
324   status=daqDA_FES_storeFile(RESULT_FILE,FILE_ID);
325   if (status) {
326     status = -2;
327   }
328   
329   SendToAmoreDB(calibCE,runNb);
330
331   delete calibCE;
332   return status;
333 }
334
335
336 void SendToAmoreDB(AliTPCCalibCE *calibCE, unsigned long32 runNb)
337 {
338   //AMORE
339 //   printf ("AMORE part\n");
340   const char *amoreDANameorig=gSystem->Getenv("AMORE_DA_NAME");
341   //cheet a little -- temporary solution (hopefully)
342   //
343   //currently amoreDA uses the environment variable AMORE_DA_NAME to create the mysql
344   //table in which the calib objects are stored. This table is dropped each time AmoreDA
345   //is initialised. This of course makes a problem if we would like to store different
346   //calibration entries in the AMORE DB. Therefore in each DA which writes to the AMORE DB
347   //the AMORE_DA_NAME env variable is overwritten.
348   gSystem->Setenv("AMORE_DA_NAME",Form("TPC-%s",FILE_ID));
349   //
350   // end cheet
351   TGraph *grA=calibCE->MakeGraphTimeCE(-1,0,2);
352   TGraph *grC=calibCE->MakeGraphTimeCE(-2,0,2);
353   TDatime time;
354   TObjString info(Form("Run: %u; Date: %s",runNb,time.AsSQLString()));
355   amore::da::AmoreDA amoreDA(amore::da::AmoreDA::kSender);
356   Int_t statusDA=0;
357   statusDA+=amoreDA.Send("CET0",calibCE->GetCalPadT0());
358   statusDA+=amoreDA.Send("CEQ",calibCE->GetCalPadQ());
359   statusDA+=amoreDA.Send("CERMS",calibCE->GetCalPadRMS());
360   statusDA+=amoreDA.Send("DriftA",grA);
361   statusDA+=amoreDA.Send("DriftC",grC);
362   statusDA+=amoreDA.Send("Info",&info);
363   if ( statusDA!=0 )
364     printf("TPCCEda: Waring: Failed to write one of the calib objects to the AMORE database\n");
365   // reset env var
366   if (amoreDANameorig) gSystem->Setenv("AMORE_DA_NAME",amoreDANameorig);
367   if (grA) delete grA;
368   if (grC) delete grC;
369 }
370
371