Added two missing includes to allow macro compilation (thanks to Laurent for remarkin...
[u/mrichter/AliRoot.git] / ITS / ITSSPDVertexDiamondda.cxx
1 /*
2 Contact: cvetan.cheshkov@cern.ch
3 Link: http://alisoft.cern.ch/viewvc/trunk/ITS/ITSSPDVertexDiamondda.cxx?root=AliRoot&view=log , /afs/cern.ch/user/c/cheshkov/public/08000058338016.30.root.date.gz
4 Reference Run: 58338
5 Run Type: PHYSICS
6 DA Type: MON
7 Number of events needed: 100
8 Input Files: GRP/Geometry/Data , ITS/Align/Data , spd_noisy_ocdb , spd_dead_ocdb
9 Output Files: SPDVertexDiamondDA.root
10 Trigger types used: PHYSICS
11 */
12
13 #define OUTPUT_FILE "SPDVertexDiamondDA.root"
14 #define N_EVENTS_AUTOSAVE 50
15
16 extern "C" {
17 #include "daqDA.h"
18 }
19
20 #include "event.h"
21 #include "monitor.h"
22
23 #ifdef ALI_AMORE
24 #include <AmoreDA.h>
25 //int amore::da::Updated(char const*) {}
26 #endif
27
28 #include <TPluginManager.h>
29 #include <TROOT.h>
30 #include <TH1.h>
31 #include <TH2.h>
32 #include <TSystem.h>
33
34 #include "AliLog.h"
35 #include "AliRawReaderDate.h"
36 #include "AliCDBManager.h"
37 #include "AliITSMeanVertexer.h"
38
39 int main(int argc, char **argv) {
40
41   gROOT->GetPluginManager()->AddHandler("TVirtualStreamerInfo",
42                                         "*",
43                                         "TStreamerInfo",
44                                         "RIO",
45                                         "TStreamerInfo()"); 
46
47   int status;
48   if (argc<2) {
49     printf("Wrong number of arguments\n");
50     return -1;
51   }
52
53   /* define data source : this is argument 1 */  
54   status=monitorSetDataSource( argv[1] );
55   if (status!=0) {
56     printf("monitorSetDataSource() failed : %s\n",monitorDecodeError(status));
57     return -1;
58   }
59
60   /* declare monitoring program */
61   status=monitorDeclareMp( __FILE__ );
62   if (status!=0) {
63     printf("monitorDeclareMp() failed : %s\n",monitorDecodeError(status));
64     return -1;
65   }
66
67   /* define wait event timeout - 1s max */
68   monitorSetNowait();
69   monitorSetNoWaitNetworkTimeout(1000);
70   
71   /* log start of process */
72   printf("Vertex-Diamond SPD DA started\n");  
73
74   /* init some counters */
75   int nevents_with_vertex = 0;
76   int nevents_physics=0;
77   int nevents_total=0;
78
79   struct eventHeaderStruct *event;
80   eventTypeType eventT;
81
82   // Get run number
83   if (getenv("DATE_RUN_NUMBER")==0) {
84     printf("DATE_RUN_NUMBER not properly set.\n");
85     return -1;
86   }
87   int runNr = atoi(getenv("DATE_RUN_NUMBER"));
88
89   // Get the necessary OCDB files from the DAQ detector DB
90   if (gSystem->AccessPathName("localOCDB/GRP/Geometry/Data",kFileExists)) {
91     if (gSystem->mkdir("localOCDB/GRP/Geometry/Data",kTRUE) != 0) {
92       printf("Failed to create directory: localOCDB/GRP/Geometry/Data");
93       return -1;
94     }
95   }
96   status = daqDA_DB_getFile("GRP/Geometry/Data","localOCDB/GRP/Geometry/Data/Run0_999999999_v0_s0.root");
97   if (status) {
98     printf("Failed to get geometry file (GRP/Geometry/Data) from DAQdetDB, status=%d\n", status);
99     return -1;
100   }
101
102   if (gSystem->AccessPathName("localOCDB/ITS/Align/Data",kFileExists)) {
103     if (gSystem->mkdir("localOCDB/ITS/Align/Data",kTRUE) != 0) {
104       printf("Failed to create directory: localOCDB/ITS/Align/Data");
105       return -1;
106     }
107   }
108   status = daqDA_DB_getFile("ITS/Align/Data","localOCDB/ITS/Align/Data/Run0_999999999_v0_s0.root");
109   if (status) {
110     printf("Failed to get its-alignment file (ITS/Align/Data) from DAQdetDB, status=%d\n", status);
111     return -1;
112   }
113
114   if (gSystem->AccessPathName("localOCDB/ITS/Calib/SPDNoisy",kFileExists)) {
115     if (gSystem->mkdir("localOCDB/ITS/Calib/SPDNoisy",kTRUE) != 0) {
116       printf("Failed to create directory: localOCDB/ITS/Calib/SPDNoisy");
117       return -1;
118     }
119   }
120   status = daqDA_DB_getFile("spd_noisy_ocdb","localOCDB/ITS/Calib/SPDNoisy/Run0_999999999_v0_s0.root");
121   if (status) {
122     printf("Failed to get spd file (spd_noisy_ocdb) from DAQdetDB, status=%d\n", status);
123     return -1;
124   }
125
126   if (gSystem->AccessPathName("localOCDB/ITS/Calib/SPDDead",kFileExists)) {
127     if (gSystem->mkdir("localOCDB/ITS/Calib/SPDDead",kTRUE) != 0) {
128       printf("Failed to create directory: localOCDB/ITS/Calib/SPDDead");
129       return -1;
130     }
131   }
132   status = daqDA_DB_getFile("spd_dead_ocdb","localOCDB/ITS/Calib/SPDDead/Run0_999999999_v0_s0.root");
133   if (status) {
134     printf("Failed to get spd file (spd_dead_ocdb) from DAQdetDB, status=%d\n", status);
135     return -1;
136   }
137
138   // Global initializations
139   AliLog::SetGlobalLogLevel(AliLog::kError);
140   AliCDBManager *man = AliCDBManager::Instance();
141   man->SetDefaultStorage("local://localOCDB");
142   man->SetRun(runNr);
143
144   // Init mean vertexer
145   AliITSMeanVertexer *mv = new AliITSMeanVertexer();
146   if (!mv->Init()) {
147     printf("Initialization of mean vertexer object failed ! Check the log for details");
148     return -1;
149   }
150
151   // Initialization of AMORE sender
152 #ifdef ALI_AMORE
153   amore::da::AmoreDA vtxAmore(amore::da::AmoreDA::kSender);
154 #endif
155   /* main loop (infinite) */
156   for(;;) {
157     
158     /* check shutdown condition */
159     if (daqDA_checkShutdown()) {break;}
160     
161     /* get next event (blocking call until timeout) */
162     status=monitorGetEventDynamic((void **)&event);
163     if (status==MON_ERR_EOF) {
164       printf ("End of File detected\n");
165       break; /* end of monitoring file has been reached */
166     }
167     
168     if (status!=0) {
169       printf("monitorGetEventDynamic() failed : %s\n",monitorDecodeError(status));
170       break;
171     }
172     
173     /* retry if got no event */
174     if (event==NULL) {
175       continue;
176     }
177
178     nevents_total++;
179     eventT=event->eventType;
180     switch (event->eventType){
181       
182       /* START OF RUN */
183     case START_OF_RUN:
184       break;
185       /* END START OF RUN */
186       
187     /* END OF RUN */
188     case END_OF_RUN:
189       break;
190       
191     case PHYSICS_EVENT:
192       nevents_physics++;
193       AliRawReader *rawReader = new AliRawReaderDate((void*)event);
194
195       // Run mean-vertexer reco
196       if (mv->Reconstruct(rawReader)) nevents_with_vertex++;
197
198       // Auto save
199       if ((nevents_physics%N_EVENTS_AUTOSAVE) == 0) {
200         mv->WriteVertices(OUTPUT_FILE);
201
202 #ifdef ALI_AMORE
203       // send the histos to AMORE pool
204         printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexXY()->GetName(),mv->GetVertexXY()));
205         printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexZ()->GetName(),mv->GetVertexZ()));
206 #endif
207       }
208
209       delete rawReader;
210     }
211     
212     /* free resources */
213     free(event);
214     
215     /* exit when last event received, no need to wait for TERM signal */
216     if (eventT==END_OF_RUN) {
217       printf("EOR event detected\n");
218       break;
219     }
220   }
221
222   mv->WriteVertices(OUTPUT_FILE);
223
224 #ifdef ALI_AMORE
225   // send the histos to AMORE pool
226   printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexXY()->GetName(),mv->GetVertexXY()));
227   printf("AMORE send status: %d\n",vtxAmore.Send(mv->GetVertexZ()->GetName(),mv->GetVertexZ()));
228 #endif
229
230   delete mv;
231
232   /* write report */
233   printf("Run #%s, received %d events with vertex, out of %d physics and out of %d total events\n",getenv("DATE_RUN_NUMBER"),nevents_with_vertex,nevents_physics,nevents_total);
234
235   status=0;
236
237   /* export file to FXS */
238   if (daqDA_FES_storeFile(OUTPUT_FILE, "VertexDiamond")) {
239     status=-2;
240   }
241   
242   return status;
243 }