]>
Commit | Line | Data |
---|---|---|
1 | /************************************************************************** | |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | ||
16 | ///////////////////////////////////////////////////////////////////////////////////////////////// | |
17 | // // | |
18 | // AliCDBLocal // | |
19 | // access class to a DataBase in a local storage // | |
20 | // // | |
21 | ///////////////////////////////////////////////////////////////////////////////////////////////// | |
22 | ||
23 | #include <cstdlib> | |
24 | #include <TSystem.h> | |
25 | #include <TObjString.h> | |
26 | #include <TRegexp.h> | |
27 | #include <TFile.h> | |
28 | #include <TKey.h> | |
29 | ||
30 | #include "AliCDBLocal.h" | |
31 | #include "AliCDBEntry.h" | |
32 | #include "AliLog.h" | |
33 | ||
34 | ClassImp(AliCDBLocal) | |
35 | ||
36 | //_____________________________________________________________________________ | |
37 | AliCDBLocal::AliCDBLocal(const char* baseDir): | |
38 | fBaseDirectory(baseDir) | |
39 | { | |
40 | // constructor | |
41 | ||
42 | // check baseDire: trying to cd to baseDir; if it does not exist, create it | |
43 | void* dir = gSystem->OpenDirectory(baseDir); | |
44 | if (dir == NULL) { | |
45 | if (gSystem->mkdir(baseDir, kTRUE)) { | |
46 | AliError(Form("Can't open directory <%s>!", baseDir)); | |
47 | } | |
48 | ||
49 | } else { | |
50 | AliDebug(2,Form("Folder <%s> found",fBaseDirectory.Data())); | |
51 | gSystem->FreeDirectory(dir); | |
52 | } | |
53 | fType="local"; | |
54 | fBaseFolder = fBaseDirectory; | |
55 | } | |
56 | ||
57 | //_____________________________________________________________________________ | |
58 | AliCDBLocal::~AliCDBLocal() { | |
59 | // destructor | |
60 | ||
61 | } | |
62 | ||
63 | ||
64 | //_____________________________________________________________________________ | |
65 | Bool_t AliCDBLocal::FilenameToId(const char* filename, AliCDBRunRange& runRange, | |
66 | Int_t& version, Int_t& subVersion) { | |
67 | // build AliCDBId from filename numbers | |
68 | ||
69 | ||
70 | Ssiz_t mSize; | |
71 | ||
72 | // valid filename: Run#firstRun_#lastRun_v#version_s#subVersion.root | |
73 | TRegexp keyPattern("^Run[0-9]+_[0-9]+_v[0-9]+_s[0-9]+.root$"); | |
74 | keyPattern.Index(filename, &mSize); | |
75 | if (!mSize) { | |
76 | AliDebug(2, Form("Bad filename <%s>.", filename)); | |
77 | return kFALSE; | |
78 | } | |
79 | ||
80 | TString idString(filename); | |
81 | idString.Resize(idString.Length() - sizeof(".root") + 1); | |
82 | ||
83 | TObjArray* strArray = (TObjArray*) idString.Tokenize("_"); | |
84 | ||
85 | TString firstRunString(((TObjString*) strArray->At(0))->GetString()); | |
86 | runRange.SetFirstRun(atoi(firstRunString.Data() + 3)); | |
87 | runRange.SetLastRun(atoi(((TObjString*) strArray->At(1))->GetString())); | |
88 | ||
89 | TString verString(((TObjString*) strArray->At(2))->GetString()); | |
90 | version = atoi(verString.Data() + 1); | |
91 | ||
92 | TString subVerString(((TObjString*) strArray->At(3))->GetString()); | |
93 | subVersion = atoi(subVerString.Data() + 1); | |
94 | ||
95 | delete strArray; | |
96 | ||
97 | return kTRUE; | |
98 | } | |
99 | ||
100 | ||
101 | //_____________________________________________________________________________ | |
102 | Bool_t AliCDBLocal::IdToFilename(const AliCDBId& id, TString& filename) const { | |
103 | // build file name from AliCDBId data (run range, version, subVersion) | |
104 | ||
105 | if (!id.GetAliCDBRunRange().IsValid()) { | |
106 | AliDebug(2,Form("Invalid run range <%d, %d>.", | |
107 | id.GetFirstRun(), id.GetLastRun())); | |
108 | return kFALSE; | |
109 | } | |
110 | ||
111 | if (id.GetVersion() < 0) { | |
112 | AliDebug(2,Form("Invalid version <%d>.", id.GetVersion())); | |
113 | return kFALSE; | |
114 | } | |
115 | ||
116 | if (id.GetSubVersion() < 0) { | |
117 | AliDebug(2,Form("Invalid subversion <%d>.", id.GetSubVersion())); | |
118 | return kFALSE; | |
119 | } | |
120 | ||
121 | filename = Form("Run%d_%d_v%d_s%d.root", id.GetFirstRun(), id.GetLastRun(), | |
122 | id.GetVersion(), id.GetSubVersion()); | |
123 | ||
124 | filename.Prepend(fBaseDirectory +'/' + id.GetPath() + '/'); | |
125 | ||
126 | return kTRUE; | |
127 | } | |
128 | ||
129 | //_____________________________________________________________________________ | |
130 | Bool_t AliCDBLocal::PrepareId(AliCDBId& id) { | |
131 | // prepare id (version, subVersion) of the object that will be stored (called by PutEntry) | |
132 | ||
133 | TString dirName = Form("%s/%s", fBaseDirectory.Data(), id.GetPath().Data()); | |
134 | ||
135 | // go to the path; if directory does not exist, create it | |
136 | void* dirPtr = gSystem->OpenDirectory(dirName); | |
137 | if (!dirPtr) { | |
138 | gSystem->mkdir(dirName, kTRUE); | |
139 | dirPtr = gSystem->OpenDirectory(dirName); | |
140 | ||
141 | if (!dirPtr) { | |
142 | AliError(Form("Can't create directory <%s>!", | |
143 | dirName.Data())); | |
144 | return kFALSE; | |
145 | } | |
146 | } | |
147 | ||
148 | const char* filename; | |
149 | AliCDBRunRange aRunRange; // the runRange got from filename | |
150 | AliCDBRunRange lastRunRange(-1,-1); // highest runRange found | |
151 | Int_t aVersion, aSubVersion; // the version subVersion got from filename | |
152 | Int_t lastVersion = 0, lastSubVersion = -1; // highest version and subVersion found | |
153 | ||
154 | if (!id.HasVersion()) { // version not specified: look for highest version & subVersion | |
155 | ||
156 | while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on the files | |
157 | ||
158 | TString aString(filename); | |
159 | if (aString == "." || aString == "..") continue; | |
160 | ||
161 | if (!FilenameToId(filename, aRunRange, aVersion, | |
162 | aSubVersion)) { | |
163 | AliDebug(2,Form( | |
164 | "Bad filename <%s>! I'll skip it.", | |
165 | filename)); | |
166 | continue; | |
167 | } | |
168 | ||
169 | if (!aRunRange.Overlaps(id.GetAliCDBRunRange())) continue; | |
170 | if(aVersion < lastVersion) continue; | |
171 | if(aVersion > lastVersion) lastSubVersion = -1; | |
172 | if(aSubVersion < lastSubVersion) continue; | |
173 | lastVersion = aVersion; | |
174 | lastSubVersion = aSubVersion; | |
175 | lastRunRange = aRunRange; | |
176 | } | |
177 | ||
178 | id.SetVersion(lastVersion); | |
179 | id.SetSubVersion(lastSubVersion + 1); | |
180 | ||
181 | } else { // version specified, look for highest subVersion only | |
182 | ||
183 | while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on the files | |
184 | ||
185 | TString aString(filename); | |
186 | if (aString == "." || aString == "..") { | |
187 | continue; | |
188 | } | |
189 | ||
190 | if (!FilenameToId(filename, aRunRange, aVersion, | |
191 | aSubVersion)) { | |
192 | AliDebug(2,Form( | |
193 | "Bad filename <%s>!I'll skip it.", | |
194 | filename)); | |
195 | continue; | |
196 | } | |
197 | ||
198 | if (aRunRange.Overlaps(id.GetAliCDBRunRange()) | |
199 | && aVersion == id.GetVersion() | |
200 | && aSubVersion > lastSubVersion) { | |
201 | lastSubVersion = aSubVersion; | |
202 | lastRunRange = aRunRange; | |
203 | } | |
204 | ||
205 | } | |
206 | ||
207 | id.SetSubVersion(lastSubVersion + 1); | |
208 | } | |
209 | ||
210 | gSystem->FreeDirectory(dirPtr); | |
211 | ||
212 | TString lastStorage = id.GetLastStorage(); | |
213 | if(lastStorage.Contains(TString("grid"), TString::kIgnoreCase) && | |
214 | id.GetSubVersion() > 0 ){ | |
215 | AliError(Form("Grid to Local Storage error! local object with version v%d_s%d found:",id.GetVersion(), id.GetSubVersion()-1)); | |
216 | AliError(Form("This object has been already transferred from Grid (check v%d_s0)!",id.GetVersion())); | |
217 | return kFALSE; | |
218 | } | |
219 | ||
220 | if(lastStorage.Contains(TString("new"), TString::kIgnoreCase) && | |
221 | id.GetSubVersion() > 0 ){ | |
222 | AliDebug(2, Form("A NEW object is being stored with version v%d_s%d", | |
223 | id.GetVersion(),id.GetSubVersion())); | |
224 | AliDebug(2, Form("and it will hide previously stored object with v%d_s%d!", | |
225 | id.GetVersion(),id.GetSubVersion()-1)); | |
226 | } | |
227 | ||
228 | if(!lastRunRange.IsAnyRange() && !(lastRunRange.IsEqual(& id.GetAliCDBRunRange()))) | |
229 | AliWarning(Form("Run range modified w.r.t. previous version (Run%d_%d_v%d_s%d)", | |
230 | lastRunRange.GetFirstRun(), lastRunRange.GetLastRun(), | |
231 | id.GetVersion(), id.GetSubVersion()-1)); | |
232 | ||
233 | return kTRUE; | |
234 | } | |
235 | ||
236 | // //_____________________________________________________________________________ | |
237 | // Bool_t AliCDBLocal::GetId(const AliCDBId& query, AliCDBId& result) { | |
238 | // // look for filename matching query (called by GetEntry) | |
239 | // | |
240 | // TString dirName = Form("%s/%s", fBaseDirectory.Data(), query.GetPath().Data()); | |
241 | // | |
242 | // void* dirPtr = gSystem->OpenDirectory(dirName); | |
243 | // if (!dirPtr) { | |
244 | // AliDebug(2,Form("Directory <%s> not found", (query.GetPath()).Data())); | |
245 | // AliDebug(2,Form("in DB folder %s", fBaseDirectory.Data())); | |
246 | // return kFALSE; | |
247 | // } | |
248 | // | |
249 | // const char* filename; | |
250 | // | |
251 | // AliCDBRunRange aRunRange; // the runRange got from filename | |
252 | // Int_t aVersion, aSubVersion; // the version and subVersion got from filename | |
253 | // | |
254 | // if (!query.HasVersion()) { // neither version and subversion specified -> look for highest version and subVersion | |
255 | // | |
256 | // while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on files | |
257 | // | |
258 | // TString aString(filename); | |
259 | // if (aString == "." || aString == "..") continue; | |
260 | // | |
261 | // if (!FilenameToId(filename, aRunRange, aVersion, aSubVersion)) continue; | |
262 | // // aRunRange, aVersion, aSubVersion filled from filename | |
263 | // | |
264 | // if (!aRunRange.Comprises(query.GetAliCDBRunRange())) continue; | |
265 | // // aRunRange contains requested run! | |
266 | // | |
267 | // if (result.GetVersion() < aVersion) { | |
268 | // result.SetVersion(aVersion); | |
269 | // result.SetSubVersion(aSubVersion); | |
270 | // | |
271 | // result.SetFirstRun( | |
272 | // aRunRange.GetFirstRun()); | |
273 | // result.SetLastRun( | |
274 | // aRunRange.GetLastRun()); | |
275 | // | |
276 | // } else if (result.GetVersion() == aVersion | |
277 | // && result.GetSubVersion() | |
278 | // < aSubVersion) { | |
279 | // | |
280 | // result.SetSubVersion(aSubVersion); | |
281 | // | |
282 | // result.SetFirstRun( | |
283 | // aRunRange.GetFirstRun()); | |
284 | // result.SetLastRun( | |
285 | // aRunRange.GetLastRun()); | |
286 | // } else if (result.GetVersion() == aVersion | |
287 | // && result.GetSubVersion() == aSubVersion){ | |
288 | // AliDebug(2,Form("More than one object valid for run %d, version %d_%d!", | |
289 | // query.GetFirstRun(), aVersion, aSubVersion)); | |
290 | // gSystem->FreeDirectory(dirPtr); | |
291 | // return kFALSE; | |
292 | // } | |
293 | // } | |
294 | // | |
295 | // } else if (!query.HasSubVersion()) { // version specified but not subversion -> look for highest subVersion | |
296 | // | |
297 | // result.SetVersion(query.GetVersion()); | |
298 | // | |
299 | // while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on files | |
300 | // | |
301 | // TString aString(filename); | |
302 | // if (aString == "." || aString == "..") continue; | |
303 | // | |
304 | // if (!FilenameToId(filename, aRunRange, aVersion, aSubVersion)) continue; | |
305 | // // aRunRange, aVersion, aSubVersion filled from filename | |
306 | // | |
307 | // if (!aRunRange.Comprises(query.GetAliCDBRunRange())) continue; | |
308 | // // aRunRange contains requested run! | |
309 | // | |
310 | // if(query.GetVersion() != aVersion) continue; | |
311 | // // aVersion is requested version! | |
312 | // | |
313 | // if(result.GetSubVersion() == aSubVersion){ | |
314 | // AliDebug(2,Form("More than one object valid for run %d, version %d_%d!", | |
315 | // query.GetFirstRun(), aVersion, aSubVersion)); | |
316 | // gSystem->FreeDirectory(dirPtr); | |
317 | // return kFALSE; | |
318 | // } | |
319 | // if( result.GetSubVersion() < aSubVersion) { | |
320 | // | |
321 | // result.SetSubVersion(aSubVersion); | |
322 | // | |
323 | // result.SetFirstRun( | |
324 | // aRunRange.GetFirstRun()); | |
325 | // result.SetLastRun( | |
326 | // aRunRange.GetLastRun()); | |
327 | // } | |
328 | // } | |
329 | // | |
330 | // } else { // both version and subversion specified | |
331 | // | |
332 | // while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on files | |
333 | // | |
334 | // TString aString(filename); | |
335 | // if (aString == "." || aString == "..") continue; | |
336 | // | |
337 | // if (!FilenameToId(filename, aRunRange, aVersion, aSubVersion)) continue; | |
338 | // // aRunRange, aVersion, aSubVersion filled from filename | |
339 | // | |
340 | // if (!aRunRange.Comprises(query.GetAliCDBRunRange())) continue; | |
341 | // // aRunRange contains requested run! | |
342 | // | |
343 | // if(query.GetVersion() != aVersion || query.GetSubVersion() != aSubVersion) continue; | |
344 | // // aVersion and aSubVersion are requested version and subVersion! | |
345 | // | |
346 | // if(result.GetVersion() == aVersion && result.GetSubVersion() == aSubVersion){ | |
347 | // AliDebug(2,Form("More than one object valid for run %d, version %d_%d!", | |
348 | // query.GetFirstRun(), aVersion, aSubVersion)); | |
349 | // gSystem->FreeDirectory(dirPtr); | |
350 | // return kFALSE; | |
351 | // } | |
352 | // result.SetVersion(aVersion); | |
353 | // result.SetSubVersion(aSubVersion); | |
354 | // result.SetFirstRun(aRunRange.GetFirstRun()); | |
355 | // result.SetLastRun(aRunRange.GetLastRun()); | |
356 | // | |
357 | // } | |
358 | // } | |
359 | // | |
360 | // gSystem->FreeDirectory(dirPtr); | |
361 | // | |
362 | // return kTRUE; | |
363 | // } | |
364 | ||
365 | //_____________________________________________________________________________ | |
366 | AliCDBId* AliCDBLocal::GetId(const AliCDBId& query) { | |
367 | // look for filename matching query (called by GetEntryId) | |
368 | ||
369 | TString dirName = Form("%s/%s", fBaseDirectory.Data(), query.GetPath().Data()); | |
370 | ||
371 | void* dirPtr = gSystem->OpenDirectory(dirName); | |
372 | if (!dirPtr) { | |
373 | AliDebug(2,Form("Directory <%s> not found", (query.GetPath()).Data())); | |
374 | AliDebug(2,Form("in DB folder %s", fBaseDirectory.Data())); | |
375 | return NULL; | |
376 | } | |
377 | ||
378 | const char* filename; | |
379 | AliCDBId *result = new AliCDBId(); | |
380 | result->SetPath(query.GetPath()); | |
381 | ||
382 | AliCDBRunRange aRunRange; // the runRange got from filename | |
383 | Int_t aVersion, aSubVersion; // the version and subVersion got from filename | |
384 | ||
385 | if (!query.HasVersion()) { // neither version and subversion specified -> look for highest version and subVersion | |
386 | ||
387 | while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on files | |
388 | ||
389 | TString aString(filename); | |
390 | if (aString == "." || aString == "..") continue; | |
391 | ||
392 | if (!FilenameToId(filename, aRunRange, aVersion, aSubVersion)) continue; | |
393 | // aRunRange, aVersion, aSubVersion filled from filename | |
394 | ||
395 | if (!aRunRange.Comprises(query.GetAliCDBRunRange())) continue; | |
396 | // aRunRange contains requested run! | |
397 | ||
398 | AliDebug(1,Form("Filename %s matches\n",filename)); | |
399 | ||
400 | if (result->GetVersion() < aVersion) { | |
401 | result->SetVersion(aVersion); | |
402 | result->SetSubVersion(aSubVersion); | |
403 | ||
404 | result->SetFirstRun( | |
405 | aRunRange.GetFirstRun()); | |
406 | result->SetLastRun( | |
407 | aRunRange.GetLastRun()); | |
408 | ||
409 | } else if (result->GetVersion() == aVersion | |
410 | && result->GetSubVersion() | |
411 | < aSubVersion) { | |
412 | ||
413 | result->SetSubVersion(aSubVersion); | |
414 | ||
415 | result->SetFirstRun( | |
416 | aRunRange.GetFirstRun()); | |
417 | result->SetLastRun( | |
418 | aRunRange.GetLastRun()); | |
419 | } else if (result->GetVersion() == aVersion | |
420 | && result->GetSubVersion() == aSubVersion){ | |
421 | AliError(Form("More than one object valid for run %d, version %d_%d!", | |
422 | query.GetFirstRun(), aVersion, aSubVersion)); | |
423 | gSystem->FreeDirectory(dirPtr); | |
424 | delete result; | |
425 | return NULL; | |
426 | } | |
427 | } | |
428 | ||
429 | } else if (!query.HasSubVersion()) { // version specified but not subversion -> look for highest subVersion | |
430 | ||
431 | result->SetVersion(query.GetVersion()); | |
432 | ||
433 | while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on files | |
434 | ||
435 | TString aString(filename); | |
436 | if (aString == "." || aString == "..") continue; | |
437 | ||
438 | if (!FilenameToId(filename, aRunRange, aVersion, aSubVersion)) continue; | |
439 | // aRunRange, aVersion, aSubVersion filled from filename | |
440 | ||
441 | if (!aRunRange.Comprises(query.GetAliCDBRunRange())) continue; | |
442 | // aRunRange contains requested run! | |
443 | ||
444 | if(query.GetVersion() != aVersion) continue; | |
445 | // aVersion is requested version! | |
446 | ||
447 | if(result->GetSubVersion() == aSubVersion){ | |
448 | AliError(Form("More than one object valid for run %d, version %d_%d!", | |
449 | query.GetFirstRun(), aVersion, aSubVersion)); | |
450 | gSystem->FreeDirectory(dirPtr); | |
451 | delete result; | |
452 | return NULL; | |
453 | } | |
454 | if( result->GetSubVersion() < aSubVersion) { | |
455 | ||
456 | result->SetSubVersion(aSubVersion); | |
457 | ||
458 | result->SetFirstRun( | |
459 | aRunRange.GetFirstRun()); | |
460 | result->SetLastRun( | |
461 | aRunRange.GetLastRun()); | |
462 | } | |
463 | } | |
464 | ||
465 | } else { // both version and subversion specified | |
466 | ||
467 | //AliCDBId dataId(queryId.GetAliCDBPath(), -1, -1, -1, -1); | |
468 | //Bool_t result; | |
469 | while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on files | |
470 | ||
471 | TString aString(filename); | |
472 | if (aString == "." || aString == "..") continue; | |
473 | ||
474 | if (!FilenameToId(filename, aRunRange, aVersion, aSubVersion)) continue; | |
475 | // aRunRange, aVersion, aSubVersion filled from filename | |
476 | ||
477 | if (!aRunRange.Comprises(query.GetAliCDBRunRange())) continue; | |
478 | // aRunRange contains requested run! | |
479 | ||
480 | if(query.GetVersion() != aVersion || query.GetSubVersion() != aSubVersion) continue; | |
481 | // aVersion and aSubVersion are requested version and subVersion! | |
482 | ||
483 | if(result->GetVersion() == aVersion && result->GetSubVersion() == aSubVersion){ | |
484 | AliError(Form("More than one object valid for run %d, version %d_%d!", | |
485 | query.GetFirstRun(), aVersion, aSubVersion)); | |
486 | gSystem->FreeDirectory(dirPtr); | |
487 | delete result; | |
488 | return NULL; | |
489 | } | |
490 | result->SetVersion(aVersion); | |
491 | result->SetSubVersion(aSubVersion); | |
492 | result->SetFirstRun(aRunRange.GetFirstRun()); | |
493 | result->SetLastRun(aRunRange.GetLastRun()); | |
494 | ||
495 | } | |
496 | } | |
497 | ||
498 | gSystem->FreeDirectory(dirPtr); | |
499 | ||
500 | return result; | |
501 | } | |
502 | ||
503 | //_____________________________________________________________________________ | |
504 | AliCDBEntry* AliCDBLocal::GetEntry(const AliCDBId& queryId) { | |
505 | // get AliCDBEntry from the database | |
506 | ||
507 | AliCDBId* dataId = GetEntryId(queryId); | |
508 | ||
509 | if (!dataId || !dataId->IsSpecified()) return NULL; | |
510 | ||
511 | TString filename; | |
512 | if (!IdToFilename(*dataId, filename)) { | |
513 | ||
514 | AliDebug(2,Form("Bad data ID encountered! Subnormal error!")); | |
515 | delete dataId; | |
516 | return NULL; | |
517 | } | |
518 | ||
519 | TFile file(filename, "READ"); // open file | |
520 | if (!file.IsOpen()) { | |
521 | AliDebug(2,Form("Can't open file <%s>!", filename.Data())); | |
522 | delete dataId; | |
523 | return NULL; | |
524 | } | |
525 | ||
526 | // get the only AliCDBEntry object from the file | |
527 | // the object in the file is an AliCDBEntry entry named "AliCDBEntry" | |
528 | ||
529 | AliCDBEntry* anEntry = dynamic_cast<AliCDBEntry*> (file.Get("AliCDBEntry")); | |
530 | if (!anEntry) { | |
531 | AliDebug(2,Form("Bad storage data: No AliCDBEntry in file!")); | |
532 | file.Close(); | |
533 | delete dataId; | |
534 | return NULL; | |
535 | } | |
536 | ||
537 | AliCDBId& entryId = anEntry->GetId(); | |
538 | ||
539 | // The object's Id are not reset during storage | |
540 | // If object's Id runRange or version do not match with filename, | |
541 | // it means that someone renamed file by hand. In this case a warning msg is issued. | |
542 | ||
543 | anEntry-> SetLastStorage("local"); | |
544 | ||
545 | if(!entryId.IsEqual(dataId)){ | |
546 | AliWarning(Form("Mismatch between file name and object's Id!")); | |
547 | AliWarning(Form("File name: %s", dataId->ToString().Data())); | |
548 | AliWarning(Form("Object's Id: %s", entryId.ToString().Data())); | |
549 | } | |
550 | ||
551 | // Check whether entry contains a TTree. In case load the tree in memory! | |
552 | LoadTreeFromFile(anEntry); | |
553 | ||
554 | // close file, return retieved entry | |
555 | file.Close(); | |
556 | delete dataId; | |
557 | return anEntry; | |
558 | } | |
559 | ||
560 | //_____________________________________________________________________________ | |
561 | AliCDBId* AliCDBLocal::GetEntryId(const AliCDBId& queryId) { | |
562 | // get AliCDBId from the database | |
563 | ||
564 | AliCDBId* dataId = 0; | |
565 | ||
566 | // look for a filename matching query requests (path, runRange, version, subVersion) | |
567 | if (!queryId.HasVersion()) { | |
568 | // if version is not specified, first check the selection criteria list | |
569 | AliCDBId selectedId(queryId); | |
570 | GetSelection(&selectedId); | |
571 | dataId = GetId(selectedId); | |
572 | } else { | |
573 | dataId = GetId(queryId); | |
574 | } | |
575 | ||
576 | if (dataId && !dataId->IsSpecified()) { | |
577 | delete dataId; | |
578 | return NULL; | |
579 | } | |
580 | ||
581 | return dataId; | |
582 | } | |
583 | ||
584 | //_____________________________________________________________________________ | |
585 | void AliCDBLocal::GetEntriesForLevel0(const char* level0, | |
586 | const AliCDBId& queryId, TList* result) { | |
587 | // multiple request (AliCDBStorage::GetAll) | |
588 | ||
589 | TString level0Dir = Form("%s/%s", fBaseDirectory.Data(), level0); | |
590 | ||
591 | void* level0DirPtr = gSystem->OpenDirectory(level0Dir); | |
592 | if (!level0DirPtr) { | |
593 | AliDebug(2,Form("Can't open level0 directory <%s>!", | |
594 | level0Dir.Data())); | |
595 | return; | |
596 | } | |
597 | ||
598 | const char* level1; | |
599 | Long_t flag=0; | |
600 | while ((level1 = gSystem->GetDirEntry(level0DirPtr))) { | |
601 | ||
602 | TString level1Str(level1); | |
603 | if (level1Str == "." || level1Str == "..") { | |
604 | continue; | |
605 | } | |
606 | ||
607 | TString fullPath = Form("%s/%s",level0Dir.Data(), level1); | |
608 | ||
609 | Int_t res=gSystem->GetPathInfo(fullPath.Data(), 0, (Long64_t*) 0, &flag, 0); | |
610 | ||
611 | if(res){ | |
612 | AliDebug(2, Form("Error reading entry %s !",level1Str.Data())); | |
613 | continue; | |
614 | } | |
615 | if(!(flag&2)) continue; // bit 1 of flag = directory! | |
616 | ||
617 | if (queryId.GetAliCDBPath().Level1Comprises(level1)) { | |
618 | GetEntriesForLevel1(level0, level1, queryId, result); | |
619 | } | |
620 | } | |
621 | ||
622 | gSystem->FreeDirectory(level0DirPtr); | |
623 | } | |
624 | ||
625 | //_____________________________________________________________________________ | |
626 | void AliCDBLocal::GetEntriesForLevel1(const char* level0, const char* level1, | |
627 | const AliCDBId& queryId, TList* result) { | |
628 | // multiple request (AliCDBStorage::GetAll) | |
629 | ||
630 | TString level1Dir = Form("%s/%s/%s", fBaseDirectory.Data(), level0,level1); | |
631 | ||
632 | void* level1DirPtr = gSystem->OpenDirectory(level1Dir); | |
633 | if (!level1DirPtr) { | |
634 | AliDebug(2,Form("Can't open level1 directory <%s>!", | |
635 | level1Dir.Data())); | |
636 | return; | |
637 | } | |
638 | ||
639 | const char* level2; | |
640 | Long_t flag=0; | |
641 | while ((level2 = gSystem->GetDirEntry(level1DirPtr))) { | |
642 | ||
643 | TString level2Str(level2); | |
644 | if (level2Str == "." || level2Str == "..") { | |
645 | continue; | |
646 | } | |
647 | ||
648 | TString fullPath = Form("%s/%s",level1Dir.Data(), level2); | |
649 | ||
650 | Int_t res=gSystem->GetPathInfo(fullPath.Data(), 0, (Long64_t*) 0, &flag, 0); | |
651 | ||
652 | if(res){ | |
653 | AliDebug(2, Form("Error reading entry %s !",level2Str.Data())); | |
654 | continue; | |
655 | } | |
656 | if(!(flag&2)) continue; // bit 1 of flag = directory! | |
657 | ||
658 | if (queryId.GetAliCDBPath().Level2Comprises(level2)) { | |
659 | ||
660 | AliCDBPath entryPath(level0, level1, level2); | |
661 | AliCDBId entryId(entryPath, queryId.GetAliCDBRunRange(), | |
662 | queryId.GetVersion(), queryId.GetSubVersion()); | |
663 | ||
664 | AliCDBEntry* anEntry = GetEntry(entryId); | |
665 | if (anEntry) { | |
666 | result->Add(anEntry); | |
667 | } | |
668 | } | |
669 | } | |
670 | ||
671 | gSystem->FreeDirectory(level1DirPtr); | |
672 | } | |
673 | ||
674 | //_____________________________________________________________________________ | |
675 | TList* AliCDBLocal::GetEntries(const AliCDBId& queryId) { | |
676 | // multiple request (AliCDBStorage::GetAll) | |
677 | ||
678 | void* storageDirPtr = gSystem->OpenDirectory(fBaseDirectory); | |
679 | if (!storageDirPtr) { | |
680 | AliDebug(2,Form("Can't open storage directory <%s>", | |
681 | fBaseDirectory.Data())); | |
682 | return NULL; | |
683 | } | |
684 | ||
685 | TList* result = new TList(); | |
686 | result->SetOwner(); | |
687 | ||
688 | const char* level0; | |
689 | Long_t flag=0; | |
690 | while ((level0 = gSystem->GetDirEntry(storageDirPtr))) { | |
691 | ||
692 | TString level0Str(level0); | |
693 | if (level0Str == "." || level0Str == "..") { | |
694 | continue; | |
695 | } | |
696 | ||
697 | TString fullPath = Form("%s/%s",fBaseDirectory.Data(), level0); | |
698 | ||
699 | Int_t res=gSystem->GetPathInfo(fullPath.Data(), 0, (Long64_t*) 0, &flag, 0); | |
700 | ||
701 | if(res){ | |
702 | AliDebug(2, Form("Error reading entry %s !",level0Str.Data())); | |
703 | continue; | |
704 | } | |
705 | ||
706 | if(!(flag&2)) continue; // bit 1 of flag = directory! | |
707 | ||
708 | if (queryId.GetAliCDBPath().Level0Comprises(level0)) { | |
709 | GetEntriesForLevel0(level0, queryId, result); | |
710 | } | |
711 | } | |
712 | ||
713 | gSystem->FreeDirectory(storageDirPtr); | |
714 | ||
715 | return result; | |
716 | } | |
717 | ||
718 | //_____________________________________________________________________________ | |
719 | Bool_t AliCDBLocal::PutEntry(AliCDBEntry* entry) { | |
720 | // put an AliCDBEntry object into the database | |
721 | ||
722 | AliCDBId& id = entry->GetId(); | |
723 | ||
724 | // set version and subVersion for the entry to be stored | |
725 | if (!PrepareId(id)) return kFALSE; | |
726 | ||
727 | ||
728 | // build filename from entry's id | |
729 | TString filename=""; | |
730 | if (!IdToFilename(id, filename)) { | |
731 | ||
732 | AliDebug(2,Form("Bad ID encountered! Subnormal error!")); | |
733 | return kFALSE; | |
734 | } | |
735 | ||
736 | // open file | |
737 | TFile file(filename, "CREATE"); | |
738 | if (!file.IsOpen()) { | |
739 | AliError(Form("Can't open file <%s>!", filename.Data())); | |
740 | return kFALSE; | |
741 | } | |
742 | ||
743 | //SetTreeToFile(entry, &file); | |
744 | ||
745 | entry->SetVersion(id.GetVersion()); | |
746 | entry->SetSubVersion(id.GetSubVersion()); | |
747 | ||
748 | // write object (key name: "AliCDBEntry") | |
749 | Bool_t result = file.WriteTObject(entry, "AliCDBEntry"); | |
750 | if (!result) AliDebug(2,Form("Can't write entry to file: %s", filename.Data())); | |
751 | ||
752 | file.Close(); | |
753 | if(result) { | |
754 | if(!(id.GetPath().Contains("SHUTTLE/STATUS"))) | |
755 | AliInfo(Form("CDB object stored into file %s",filename.Data())); | |
756 | } | |
757 | ||
758 | return result; | |
759 | } | |
760 | ||
761 | //_____________________________________________________________________________ | |
762 | TList* AliCDBLocal::GetIdListFromFile(const char* fileName){ | |
763 | ||
764 | TString fullFileName(fileName); | |
765 | fullFileName.Prepend(fBaseDirectory+'/'); | |
766 | TFile *file = TFile::Open(fullFileName); | |
767 | if (!file) { | |
768 | AliError(Form("Can't open selection file <%s>!", fullFileName.Data())); | |
769 | return NULL; | |
770 | } | |
771 | file->cd(); | |
772 | ||
773 | TList *list = new TList(); | |
774 | list->SetOwner(); | |
775 | int i=0; | |
776 | TString keycycle; | |
777 | ||
778 | AliCDBId *id; | |
779 | while(1){ | |
780 | i++; | |
781 | keycycle = "AliCDBId;"; | |
782 | keycycle+=i; | |
783 | ||
784 | id = (AliCDBId*) file->Get(keycycle); | |
785 | if(!id) break; | |
786 | list->AddFirst(id); | |
787 | } | |
788 | file->Close(); delete file; file=0; | |
789 | return list; | |
790 | } | |
791 | ||
792 | //_____________________________________________________________________________ | |
793 | Bool_t AliCDBLocal::Contains(const char* path) const{ | |
794 | // check for path in storage's fBaseDirectory | |
795 | ||
796 | TString dirName = Form("%s/%s", fBaseDirectory.Data(), path); | |
797 | Bool_t result=kFALSE; | |
798 | ||
799 | void* dirPtr = gSystem->OpenDirectory(dirName); | |
800 | if (dirPtr) result=kTRUE; | |
801 | gSystem->FreeDirectory(dirPtr); | |
802 | ||
803 | return result; | |
804 | } | |
805 | ||
806 | //_____________________________________________________________________________ | |
807 | void AliCDBLocal::QueryValidFiles() | |
808 | { | |
809 | // Query the CDB for files valid for AliCDBStorage::fRun | |
810 | // fills list fValidFileIds with AliCDBId objects created from file name | |
811 | ||
812 | if(fVersion != -1) AliWarning ("Version parameter is not used by local storage query!"); | |
813 | if(fMetaDataFilter) { | |
814 | AliWarning ("CDB meta data parameters are not used by local storage query!"); | |
815 | delete fMetaDataFilter; fMetaDataFilter=0; | |
816 | } | |
817 | ||
818 | void* storageDirPtr = gSystem->OpenDirectory(fBaseDirectory); | |
819 | ||
820 | const char* level0; | |
821 | while ((level0 = gSystem->GetDirEntry(storageDirPtr))) { | |
822 | ||
823 | TString level0Str(level0); | |
824 | if (level0Str == "." || level0Str == "..") { | |
825 | continue; | |
826 | } | |
827 | ||
828 | if (fPathFilter.Level0Comprises(level0)) { | |
829 | TString level0Dir = Form("%s/%s",fBaseDirectory.Data(),level0); | |
830 | void* level0DirPtr = gSystem->OpenDirectory(level0Dir); | |
831 | const char* level1; | |
832 | while ((level1 = gSystem->GetDirEntry(level0DirPtr))) { | |
833 | ||
834 | TString level1Str(level1); | |
835 | if (level1Str == "." || level1Str == "..") { | |
836 | continue; | |
837 | } | |
838 | ||
839 | if (fPathFilter.Level1Comprises(level1)) { | |
840 | TString level1Dir = Form("%s/%s/%s", | |
841 | fBaseDirectory.Data(),level0,level1); | |
842 | ||
843 | void* level1DirPtr = gSystem->OpenDirectory(level1Dir); | |
844 | const char* level2; | |
845 | while ((level2 = gSystem->GetDirEntry(level1DirPtr))) { | |
846 | ||
847 | TString level2Str(level2); | |
848 | if (level2Str == "." || level2Str == "..") { | |
849 | continue; | |
850 | } | |
851 | ||
852 | if (fPathFilter.Level2Comprises(level2)) { | |
853 | TString dirName = Form("%s/%s/%s/%s", | |
854 | fBaseDirectory.Data(),level0,level1,level2); | |
855 | ||
856 | void* dirPtr = gSystem->OpenDirectory(dirName); | |
857 | ||
858 | const char* filename; | |
859 | ||
860 | AliCDBRunRange aRunRange; // the runRange got from filename | |
861 | Int_t aVersion, aSubVersion; // the version and subVersion got from filename | |
862 | ||
863 | while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on files | |
864 | ||
865 | TString aString(filename); | |
866 | if (aString == "." || aString == "..") continue; | |
867 | ||
868 | if (!FilenameToId(filename, aRunRange, aVersion, aSubVersion)) continue; | |
869 | AliCDBRunRange runrg(fRun,fRun); | |
870 | if (!aRunRange.Comprises(runrg)) continue; | |
871 | // aRunRange contains requested run! | |
872 | AliCDBPath validPath(level0,level1,level2); | |
873 | AliCDBId validId(validPath,aRunRange,aVersion,aSubVersion); | |
874 | fValidFileIds.AddLast(validId.Clone()); | |
875 | } | |
876 | gSystem->FreeDirectory(dirPtr); | |
877 | } | |
878 | } | |
879 | gSystem->FreeDirectory(level1DirPtr); | |
880 | } | |
881 | } | |
882 | gSystem->FreeDirectory(level0DirPtr); | |
883 | } | |
884 | } | |
885 | gSystem->FreeDirectory(storageDirPtr); | |
886 | ||
887 | } | |
888 | ||
889 | //_____________________________________________________________________________ | |
890 | Int_t AliCDBLocal::GetLatestVersion(const char* path, Int_t run){ | |
891 | // get last version found in the database valid for run and path | |
892 | ||
893 | AliCDBPath aCDBPath(path); | |
894 | if(!aCDBPath.IsValid() || aCDBPath.IsWildcard()) { | |
895 | AliError(Form("Invalid path in request: %s", path)); | |
896 | return -1; | |
897 | } | |
898 | ||
899 | AliCDBId query(path, run, run, -1, -1); | |
900 | AliCDBId* dataId = GetId(query); | |
901 | ||
902 | if(!dataId) return -1; | |
903 | ||
904 | Int_t version = dataId->GetVersion(); | |
905 | delete dataId; | |
906 | ||
907 | return version; | |
908 | ||
909 | } | |
910 | ||
911 | //_____________________________________________________________________________ | |
912 | Int_t AliCDBLocal::GetLatestSubVersion(const char* path, Int_t run, Int_t version){ | |
913 | // get last version found in the database valid for run and path | |
914 | ||
915 | AliCDBPath aCDBPath(path); | |
916 | if(!aCDBPath.IsValid() || aCDBPath.IsWildcard()) { | |
917 | AliError(Form("Invalid path in request: %s", path)); | |
918 | return -1; | |
919 | } | |
920 | ||
921 | AliCDBId query(path, run, run, version, -1); | |
922 | AliCDBId *dataId = GetId(query); | |
923 | ||
924 | if(!dataId) return -1; | |
925 | ||
926 | Int_t subVersion = dataId->GetSubVersion(); | |
927 | ||
928 | delete dataId; | |
929 | return subVersion; | |
930 | ||
931 | } | |
932 | ||
933 | ///////////////////////////////////////////////////////////////////////////////////////////////// | |
934 | // // | |
935 | // AliCDBLocal factory // | |
936 | // // | |
937 | ///////////////////////////////////////////////////////////////////////////////////////////////// | |
938 | ||
939 | ClassImp(AliCDBLocalFactory) | |
940 | ||
941 | //_____________________________________________________________________________ | |
942 | Bool_t AliCDBLocalFactory::Validate(const char* dbString) { | |
943 | // check if the string is valid local URI | |
944 | ||
945 | TRegexp dbPattern("^local://.+$"); | |
946 | ||
947 | return TString(dbString).Contains(dbPattern); | |
948 | } | |
949 | ||
950 | //_____________________________________________________________________________ | |
951 | AliCDBParam* AliCDBLocalFactory::CreateParameter(const char* dbString) { | |
952 | // create AliCDBLocalParam class from the URI string | |
953 | ||
954 | if (!Validate(dbString)) { | |
955 | return NULL; | |
956 | } | |
957 | ||
958 | TString pathname(dbString + sizeof("local://") - 1); | |
959 | ||
960 | gSystem->ExpandPathName(pathname); | |
961 | ||
962 | if (pathname[0] != '/') { | |
963 | pathname.Prepend(TString(gSystem->WorkingDirectory()) + '/'); | |
964 | } | |
965 | ||
966 | return new AliCDBLocalParam(pathname); | |
967 | } | |
968 | ||
969 | //_____________________________________________________________________________ | |
970 | AliCDBStorage* AliCDBLocalFactory::Create(const AliCDBParam* param) { | |
971 | // create AliCDBLocal storage instance from parameters | |
972 | ||
973 | if (AliCDBLocalParam::Class() == param->IsA()) { | |
974 | ||
975 | const AliCDBLocalParam* localParam = | |
976 | (const AliCDBLocalParam*) param; | |
977 | ||
978 | return new AliCDBLocal(localParam->GetPath()); | |
979 | } | |
980 | ||
981 | return NULL; | |
982 | } | |
983 | ||
984 | ///////////////////////////////////////////////////////////////////////////////////////////////// | |
985 | // // | |
986 | // AliCDBLocal Parameter class // // | |
987 | // // | |
988 | ///////////////////////////////////////////////////////////////////////////////////////////////// | |
989 | ||
990 | ClassImp(AliCDBLocalParam) | |
991 | ||
992 | //_____________________________________________________________________________ | |
993 | AliCDBLocalParam::AliCDBLocalParam(): | |
994 | AliCDBParam(), | |
995 | fDBPath() | |
996 | { | |
997 | // default constructor | |
998 | ||
999 | } | |
1000 | ||
1001 | //_____________________________________________________________________________ | |
1002 | AliCDBLocalParam::AliCDBLocalParam(const char* dbPath): | |
1003 | AliCDBParam(), | |
1004 | fDBPath(dbPath) | |
1005 | { | |
1006 | // constructor | |
1007 | ||
1008 | SetType("local"); | |
1009 | SetURI(TString("local://") + dbPath); | |
1010 | } | |
1011 | ||
1012 | //_____________________________________________________________________________ | |
1013 | AliCDBLocalParam::~AliCDBLocalParam() { | |
1014 | // destructor | |
1015 | ||
1016 | } | |
1017 | ||
1018 | //_____________________________________________________________________________ | |
1019 | AliCDBParam* AliCDBLocalParam::CloneParam() const { | |
1020 | // clone parameter | |
1021 | ||
1022 | return new AliCDBLocalParam(fDBPath); | |
1023 | } | |
1024 | ||
1025 | //_____________________________________________________________________________ | |
1026 | ULong_t AliCDBLocalParam::Hash() const { | |
1027 | // return Hash function | |
1028 | ||
1029 | return fDBPath.Hash(); | |
1030 | } | |
1031 | ||
1032 | //_____________________________________________________________________________ | |
1033 | Bool_t AliCDBLocalParam::IsEqual(const TObject* obj) const { | |
1034 | // check if this object is equal to AliCDBParam obj | |
1035 | ||
1036 | if (this == obj) { | |
1037 | return kTRUE; | |
1038 | } | |
1039 | ||
1040 | if (AliCDBLocalParam::Class() != obj->IsA()) { | |
1041 | return kFALSE; | |
1042 | } | |
1043 | ||
1044 | AliCDBLocalParam* other = (AliCDBLocalParam*) obj; | |
1045 | ||
1046 | return fDBPath == other->fDBPath; | |
1047 | } | |
1048 |