Preprocessor updated
[u/mrichter/AliRoot.git] / ZDC / TestZDCPreprocessor.C
1 /* $Id$ */
2
3 // This class runs the test preprocessor
4 // It uses AliTestShuttle to simulate a full Shuttle process
5
6 // The input data is created in the functions
7 //   CreateDCSAliasMap() creates input that would in the same way come from DCS
8 //   ReadDCSAliasMap() reads from a file
9 //   CreateInputFilesMap() creates a list of local files, that can be accessed by the shuttle
10
11 void TestZDCPreprocessor()
12 {
13   // load library
14   gSystem->Load("libTestShuttle.so");
15
16   // create AliTestShuttle instance
17   // The parameters are run, startTime, endTime
18   AliTestShuttle* shuttle = new AliTestShuttle(7, 0, 1);
19
20   // TODO if needed, change location of OCDB and Reference test folders
21   // by default they are set to $ALICE_ROOT/SHUTTLE/TestShuttle/TestCDB and TestReference
22   AliTestShuttle::SetMainCDB("local://$ALICE_ROOT/SHUTTLE/TestShuttle/TestCDB");
23   AliTestShuttle::SetMainRefStorage("local://$ALICE_ROOT/SHUTTLE/TestShuttle/TestReference");
24
25   printf("Test OCDB storage Uri: %s\n", AliShuttleInterface::GetMainCDB().Data());
26   printf("Test Reference storage Uri: %s\n", AliShuttleInterface::GetMainRefStorage().Data());
27
28
29   // TODO(1)
30   //
31   // The shuttle can read DCS data, if the preprocessor should be tested to process DCS data,
32   // some fake data has to be created.
33   //
34   // The "fake" input data can be taken using either (a) or (b):
35   // (a) data from a file: Use ReadDCSAliasMap()
36   //     the format of the file is explained in ReadDCSAliasMap()
37   //     To use it uncomment the following line:
38   //
39   //TMap* dcsAliasMap = ReadDCSAliasMap();
40   //
41   // (b) generated in this macro: Use CreateDCSAliasMap() and its documentation
42   //     To use it uncomment the following line:
43   //
44   TMap* dcsAliasMap = CreateDCSAliasMap();
45   //WriteDCSAliasMap();
46
47   // now give the alias map to the shuttle
48   shuttle->SetDCSInput(dcsAliasMap);
49
50   // TODO(2)
51   //
52   // The shuttle can also process files that originate from DCS, DAQ and HLT.
53   // To test it, we provide some local files and locations where these would be found when
54   // the online machinery would be there.
55   // In real life this functions would be produces by the sub-detectors
56   // calibration programs in DCS, DAQ or HLT. These files can then be retrieved using the Shuttle.
57   //
58   // Files are added with the function AliTestShuttle::AddInputFile. The syntax is:
59   // AddInputFile(<system>, <detector>, <id>, <source>, <local-file>)
60   // In this example we add a file originating from the GDC with the id PEDESTALS
61   // Three files originating from different LDCs but with the same id are also added
62   // Note that the test preprocessor name is TPC. The name of the detector's preprocessor must follow
63   // the "online" naming convention ALICE-INT-2003-039.
64   shuttle->AddInputFile(AliTestShuttle::kDAQ, "ZDC", "PEDESTALS", "LDC0", "ZDCPedestal.dat");
65   shuttle->AddInputFile(AliTestShuttle::kDAQ, "ZDC", "EMDCALIB",  "LDC0", "ZDCEMDCalib.dat");
66   shuttle->AddInputFile(AliTestShuttle::kDAQ, "ZDC", "PHYSICS",   "LDC0", "ZDCRecParam.dat");
67
68   // TODO(3)
69   //
70   // The shuttle can read run type stored in the DAQ logbook.
71   // To test it, we must provide the run type manually. They will be retrieved in the preprocessor
72   // using GetRunType function.
73  // shuttle->SetInputRunType("PEDESTAL_RUN");
74   shuttle->SetInputRunType("PULSER_RUN");
75 //  shuttle->SetInputRunType("PHYSICS");
76
77   // TODO(4)
78   //
79   // The shuttle can read run parameters stored in the DAQ run logbook.
80   // To test it, we must provide the run parameters manually. They will be retrieved in the preprocessor
81   // using GetRunParameter function.
82   // In real life the parameters will be retrieved automatically from the run logbook;
83   shuttle->AddInputRunParameter("totalEvents", "1000");
84   shuttle->AddInputRunParameter("NumberOfGDCs", "1");
85
86   // TODO(5) NEW!
87   //
88   // This is for preprocessor that require data from HLT.
89   // Since HLT may be switched off, the preprocessor should first query the Run logbook where
90   // the HLT status is stored. SHUTTLE implements a shortcut function (GetHLTStatus) that returns
91   // a bool directly. 1 = HLT ON, 0 = HLT OFF
92   //
93   Bool_t hltStatus=kFALSE;
94   //shuttle->SetInputHLTStatus(hltStatus);
95
96   // TODO(6)
97   //
98   // The shuttle can query condition parameters valid from the current run from the OCDB
99   // To test it, we must first store the object into the OCDB. It will be retrieved in the preprocessor
100   // using GetFromOCDB function.
101 /*
102   TObjString obj("This is a condition parameter stored in OCDB");
103   AliCDBId id("ZDC/Calib/Data", 0, AliCDBRunRange::Infinity());
104   AliCDBMetaData md;
105   AliCDBEntry entry(&obj, id, &md);
106
107   shuttle->AddInputCDBEntry(&entry);
108 */
109   // TODO(6)
110   // Create the preprocessor that should be tested, it registers itself automatically to the shuttle
111   AliPreprocessor* test = new AliZDCPreprocessor(shuttle);
112
113   // Test the preprocessor
114   shuttle->Process();
115
116   // TODO(7)
117   // In the preprocessor AliShuttleInterface::Store should be called to put the final
118   // data to the CDB. To check if all went fine have a look at the files produced in
119   // $ALICE_ROOT/SHUTTLE/TestShuttle/TestCDB/<detector>/SHUTTLE/Data
120   //
121   // Check the file which should have been created
122   /*AliCDBEntry* chkEntry = AliCDBManager::Instance()->GetStorage(AliShuttleInterface::GetMainCDB())
123                         ->Get("ZDC/Calib/Pedestals", 7);
124   */
125   AliCDBEntry* chkEntry = AliCDBManager::Instance()->GetStorage(AliShuttleInterface::GetMainCDB())
126                         ->Get("ZDC/Calib/Calib", 7);
127   
128   /*AliCDBEntry* chkEntry = AliCDBManager::Instance()->GetStorage(AliShuttleInterface::GetMainCDB())
129                         ->Get("ZDC/Calib/RecParam", 7);
130   */
131   
132   if (!chkEntry)
133   {
134     printf("The file is not there. Something went wrong.\n");
135     return;
136   }
137
138   AliTestDataDCS* output = dynamic_cast<AliTestDataDCS*> (chkEntry->GetObject());
139   // If everything went fine, draw the result
140   if (output)
141     output->Draw();
142 }
143
144 TMap* CreateDCSAliasMap()
145 {
146   // Creates a DCS structure
147   // The structure is the following:
148   //   TMap (key --> value)
149   //     <DCSAlias> --> <valueList>
150   //     <DCSAlias> is a string
151   //     <valueList> is a TObjArray of AliDCSValue
152   //     An AliDCSValue consists of timestamp and a value in form of a AliSimpleValue
153
154   // In this example 6 aliases exists: DCSAlias1 ... DCSAlias6
155   // Each contains 1000 values randomly generated by TRandom::Gaus + 5*nAlias
156
157   TMap* aliasMap = new TMap;
158   aliasMap->SetOwner(1);
159
160   TRandom random;
161   TString aliasNames[28];
162
163   // ******************************** alignment values
164   aliasNames[0] = "ZDC_ZNA_POS.actual.position";
165   aliasNames[1] = "ZDC_ZPA_POS.actual.position";
166   aliasNames[2] = "ZDC_ZNC_POS.actual.position";
167   aliasNames[3] = "ZDC_ZPC_POS.actual.position";
168   //
169   for(int nAlias=0; nAlias<4; nAlias++)
170   {
171     TObjArray* valueSet = new TObjArray;
172     valueSet->SetOwner(1);
173
174     TString aliasName = aliasNames[nAlias];
175     printf("\n\n alias: %s\n\n",aliasName.Data());
176
177     Float_t simVal = (Float_t) (random.Rndm()*0.025+random.Rndm()*0.1);
178     for(int i=0;i<3;i++)
179     {
180       int timeStamp1[3] = {0,500,1000};
181       AliDCSValue* dcsVal = new AliDCSValue(simVal, timeStamp1[i]);
182       printf("%s\n",dcsVal->ToString());
183       valueSet->Add(dcsVal);
184     }
185     aliasMap->Add(new TObjString(aliasName), valueSet);
186   }
187   // ******************************** HV values
188   /*TString ZNAAlias = "ZNA_HV.actual.vMon";
189   TString ZPAAlias = "ZPA_HV.actual.vMon";
190   TString ZNCAlias = "ZNC_HV.actual.vMon";
191   TString ZPCAlias = "ZPC_HV.actual.vMon";
192   TString idat[5];
193   for(int i=0;i<5;i++)
194   {
195     idat[i] = i;
196     aliasNames[i+3]  = ZNAAlias.Insert(6,idat[i]);
197     aliasNames[i+7]  = ZPAAlias.Insert(6,idat[i]);
198     aliasNames[i+11] = ZNCAlias.Insert(6,idat[i]);
199     aliasNames[i+15] = ZPCAlias.Insert(6,idat[i]);
200   }*/
201   aliasNames[4]  = "ZDC_ZNA_HV0.actual.vMon";
202   aliasNames[5]  = "ZDC_ZNA_HV1.actual.vMon";
203   aliasNames[6]  = "ZDC_ZNA_HV2.actual.vMon";
204   aliasNames[7]  = "ZDC_ZNA_HV3.actual.vMon";
205   aliasNames[8]  = "ZDC_ZNA_HV4.actual.vMon";
206   //
207   aliasNames[9]   = "ZDC_ZPA_HV0.actual.vMon";
208   aliasNames[10]  = "ZDC_ZPA_HV1.actual.vMon";
209   aliasNames[11]  = "ZDC_ZPA_HV2.actual.vMon";
210   aliasNames[12]  = "ZDC_ZPA_HV3.actual.vMon";
211   aliasNames[13]  = "ZDC_ZPA_HV4.actual.vMon";
212   //
213   aliasNames[14]  = "ZDC_ZNC_HV0.actual.vMon";
214   aliasNames[15]  = "ZDC_ZNC_HV1.actual.vMon";
215   aliasNames[16]  = "ZDC_ZNC_HV2.actual.vMon";
216   aliasNames[17]  = "ZDC_ZNC_HV3.actual.vMon";
217   aliasNames[18]  = "ZDC_ZNC_HV4.actual.vMon";
218   //
219   aliasNames[19]  = "ZDC_ZPC_HV0.actual.vMon";
220   aliasNames[20]  = "ZDC_ZPC_HV1.actual.vMon";
221   aliasNames[21]  = "ZDC_ZPC_HV2.actual.vMon";
222   aliasNames[22]  = "ZDC_ZPC_HV3.actual.vMon";
223   aliasNames[23]  = "ZDC_ZPC_HV4.actual.vMon";
224   //
225   aliasNames[24]  = "ZDC_ZEM_HV0.actual.vMon";
226   aliasNames[25]  = "ZDC_ZEM_HV1.actual.vMon";
227   //
228   aliasNames[26]  = "ZDC_REFA_HV0.actual.vMon";
229   aliasNames[27]  = "ZDC_REFC_HV1.actual.vMon";
230   //
231   for(int nAlias=4;nAlias<28;nAlias++)
232   {
233     TObjArray* valueSet = new TObjArray;
234     valueSet->SetOwner(1);
235    
236     TString aliasName = aliasNames[nAlias];
237     printf("\n\n alias: %s\n\n",aliasName.Data());
238
239     for(int timeStamp=0;timeStamp<=1000;timeStamp+=500)
240     {
241       Float_t simVal = (Float_t) (random.Gaus()*600.+1800.);
242       AliDCSValue* dcsVal = new AliDCSValue(simVal, timeStamp);
243       printf("%s\n",dcsVal->ToString());
244       valueSet->Add(dcsVal);
245     }
246     aliasMap->Add(new TObjString(aliasName), valueSet);
247   }
248
249   return aliasMap;
250 }
251
252 TMap* ReadDCSAliasMap()
253 {
254   // Open a file that contains DCS input data
255   // The CDB framework is used to open the file, this means the file is located
256   // in $ALICE_ROOT/SHUTTLE/TestShuttle/TestCDB/<detector>/DCS/Data
257   // The file contains an AliCDBEntry that contains a TMap with the DCS structure.
258   // An explanation of the structure can be found in CreateDCSAliasMap()
259
260   AliCDBEntry *entry = AliCDBManager::Instance()->Get("ZDC/DCS/Data", 0);
261   return dynamic_cast<TMap*> (entry->GetObject());
262 }
263
264 void WriteDCSAliasMap()
265 {
266   // This writes the output from CreateDCSAliasMap to a CDB file
267
268   TMap* dcsAliasMap = CreateDCSAliasMap();
269
270   AliCDBMetaData metaData;
271         metaData.SetBeamPeriod(0);
272         metaData.SetResponsible("Responsible person");
273         metaData.SetComment("Test object for TestPreprocessor.C");
274
275   AliCDBId id("ZDC/DCS/Data", 0, 0);
276
277   // initialize location of CDB
278   AliCDBManager::Instance()->SetDefaultStorage("local://$ALICE_ROOT/SHUTTLE/TestShuttle/TestCDB");
279
280   AliCDBManager::Instance()->Put(dcsAliasMap, id, &metaData);
281 }