]>
Commit | Line | Data |
---|---|---|
fe913d8f | 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 | ||
fe913d8f | 16 | ///////////////////////////////////////////////////////////////////////////////////////////////// |
17 | // // | |
9e1ceb13 | 18 | // AliCDBLocal // |
19 | // access class to a DataBase in a local storage // | |
fe913d8f | 20 | // // |
21 | ///////////////////////////////////////////////////////////////////////////////////////////////// | |
22 | ||
fe913d8f | 23 | #include <TSystem.h> |
fe913d8f | 24 | #include <TObjString.h> |
25 | #include <TRegexp.h> | |
9e1ceb13 | 26 | #include <TFile.h> |
27 | #include <TKey.h> | |
fe913d8f | 28 | |
fe913d8f | 29 | #include "AliCDBLocal.h" |
9e1ceb13 | 30 | #include "AliCDBEntry.h" |
31 | #include "AliLog.h" | |
fe913d8f | 32 | |
33 | ClassImp(AliCDBLocal) | |
34 | ||
35 | //_____________________________________________________________________________ | |
9e1ceb13 | 36 | AliCDBLocal::AliCDBLocal(const char* baseDir): |
37 | fBaseDirectory(baseDir) | |
fe913d8f | 38 | { |
39 | // constructor | |
9e1ceb13 | 40 | |
41 | // check baseDire: trying to cd to baseDir; if it does not exist, create it | |
42 | void* dir = gSystem->OpenDirectory(baseDir); | |
43 | if (dir == NULL) { | |
44 | if (gSystem->mkdir(baseDir, kTRUE)) { | |
45 | AliError(Form("Can't open directory <%s>!", baseDir)); | |
46 | } | |
47 | ||
48 | } else { | |
49 | AliDebug(2,Form("Folder <%s> found",fBaseDirectory.Data())); | |
50 | gSystem->FreeDirectory(dir); | |
51 | } | |
fe913d8f | 52 | } |
53 | ||
54 | //_____________________________________________________________________________ | |
9e1ceb13 | 55 | AliCDBLocal::~AliCDBLocal() { |
56 | // destructor | |
fe913d8f | 57 | |
58 | } | |
59 | ||
9e1ceb13 | 60 | |
fe913d8f | 61 | //_____________________________________________________________________________ |
9e1ceb13 | 62 | Bool_t AliCDBLocal::FilenameToId(const char* filename, AliCDBRunRange& runRange, |
4b5e0dce | 63 | Int_t& version, Int_t& subVersion) { |
9e1ceb13 | 64 | // build AliCDBId from filename numbers |
65 | ||
9e1ceb13 | 66 | |
67 | Ssiz_t mSize; | |
fe913d8f | 68 | |
9e1ceb13 | 69 | // valid filename: Run#firstRun_#lastRun_v#version_s#subVersion.root |
70 | TRegexp keyPattern("^Run[0-9]+_[0-9]+_v[0-9]+_s[0-9]+.root$"); | |
71 | keyPattern.Index(filename, &mSize); | |
72 | if (!mSize) { | |
73 | AliDebug(2, Form("Bad filename <%s>.", filename)); | |
74 | return kFALSE; | |
75 | } | |
76 | ||
77 | TString idString(filename); | |
78 | idString.Resize(idString.Length() - sizeof(".root") + 1); | |
79 | ||
80 | TObjArray* strArray = (TObjArray*) idString.Tokenize("_"); | |
81 | ||
82 | TString firstRunString(((TObjString*) strArray->At(0))->GetString()); | |
83 | runRange.SetFirstRun(atoi(firstRunString.Data() + 3)); | |
84 | runRange.SetLastRun(atoi(((TObjString*) strArray->At(1))->GetString())); | |
85 | ||
86 | TString verString(((TObjString*) strArray->At(2))->GetString()); | |
87 | version = atoi(verString.Data() + 1); | |
88 | ||
89 | TString subVerString(((TObjString*) strArray->At(3))->GetString()); | |
90 | subVersion = atoi(subVerString.Data() + 1); | |
91 | ||
92 | delete strArray; | |
93 | ||
94 | return kTRUE; | |
fe913d8f | 95 | } |
96 | ||
9e1ceb13 | 97 | |
fe913d8f | 98 | //_____________________________________________________________________________ |
9e1ceb13 | 99 | Bool_t AliCDBLocal::IdToFilename(const AliCDBRunRange& runRange, Int_t version, |
100 | Int_t subVersion, TString& filename) { | |
101 | // build file name from AliCDBId data (run range, version, subVersion) | |
102 | ||
103 | if (!runRange.IsValid()) { | |
b05400be | 104 | AliDebug(2,Form("Invalid run range <%d, %d>.", |
9e1ceb13 | 105 | runRange.GetFirstRun(), runRange.GetLastRun())); |
106 | return kFALSE; | |
107 | } | |
108 | ||
109 | if (version < 0) { | |
b05400be | 110 | AliDebug(2,Form("Invalid version <%d>.", version)); |
9e1ceb13 | 111 | return kFALSE; |
112 | } | |
fe913d8f | 113 | |
9e1ceb13 | 114 | if (subVersion < 0) { |
b05400be | 115 | AliDebug(2,Form("Invalid subversion <%s>.", subVersion)); |
9e1ceb13 | 116 | return kFALSE; |
117 | } | |
118 | ||
119 | filename += "Run"; | |
120 | filename += runRange.GetFirstRun(); | |
121 | filename += "_"; | |
122 | filename += runRange.GetLastRun(); | |
123 | filename += "_v"; | |
124 | filename += version; | |
125 | filename += "_s"; | |
126 | filename += subVersion; | |
127 | filename += ".root"; | |
128 | ||
129 | return kTRUE; | |
fe913d8f | 130 | } |
131 | ||
132 | //_____________________________________________________________________________ | |
9e1ceb13 | 133 | Bool_t AliCDBLocal::PrepareId(AliCDBId& id) { |
134 | // prepare id (version, subVersion) of the object that will be stored (called by PutEntry) | |
135 | ||
136 | TString dirName; | |
137 | dirName += fBaseDirectory; | |
138 | dirName += '/'; | |
139 | dirName += id.GetPath(); | |
140 | ||
141 | // go to the path; if directory does not exist, create it | |
142 | void* dirPtr = gSystem->OpenDirectory(dirName); | |
143 | if (!dirPtr) { | |
144 | gSystem->mkdir(dirName, kTRUE); | |
145 | dirPtr = gSystem->OpenDirectory(dirName); | |
146 | ||
147 | if (!dirPtr) { | |
148 | AliError(Form("Can't create directory <%s>!", | |
149 | dirName.Data())); | |
150 | return kFALSE; | |
151 | } | |
152 | } | |
fe913d8f | 153 | |
9e1ceb13 | 154 | const char* filename; |
155 | AliCDBRunRange aRunRange; // the runRange got from filename | |
156 | AliCDBRunRange lastRunRange(-1,-1); // highest runRange found | |
157 | Int_t aVersion, aSubVersion; // the version subVersion got from filename | |
158 | Int_t lastVersion = 0, lastSubVersion = -1; // highest version and subVersion found | |
159 | ||
160 | if (!id.HasVersion()) { // version not specified: look for highest version & subVersion | |
161 | ||
162 | while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on the files | |
163 | ||
164 | TString aString(filename); | |
165 | if (aString == "." || aString == "..") continue; | |
166 | ||
167 | if (!FilenameToId(filename, aRunRange, aVersion, | |
168 | aSubVersion)) { | |
b05400be | 169 | AliDebug(2,Form( |
9e1ceb13 | 170 | "Bad filename <%s>! I'll skip it.", |
171 | filename)); | |
172 | continue; | |
173 | } | |
174 | ||
175 | if (!aRunRange.Overlaps(id.GetAliCDBRunRange())) continue; | |
176 | if(aVersion < lastVersion) continue; | |
177 | if(aVersion > lastVersion) lastSubVersion = -1; | |
178 | if(aSubVersion < lastSubVersion) continue; | |
179 | lastVersion = aVersion; | |
180 | lastSubVersion = aSubVersion; | |
181 | lastRunRange = aRunRange; | |
182 | } | |
183 | ||
184 | id.SetVersion(lastVersion); | |
185 | id.SetSubVersion(lastSubVersion + 1); | |
186 | ||
187 | } else { // version specified, look for highest subVersion only | |
188 | ||
189 | while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on the files | |
190 | ||
191 | TString aString(filename); | |
192 | if (aString == "." || aString == "..") { | |
193 | continue; | |
194 | } | |
195 | ||
196 | if (!FilenameToId(filename, aRunRange, aVersion, | |
197 | aSubVersion)) { | |
b05400be | 198 | AliDebug(2,Form( |
9e1ceb13 | 199 | "Bad filename <%s>!I'll skip it.", |
200 | filename)); | |
201 | continue; | |
202 | } | |
203 | ||
204 | if (aRunRange.Overlaps(id.GetAliCDBRunRange()) | |
205 | && aVersion == id.GetVersion() | |
206 | && aSubVersion > lastSubVersion) { | |
207 | lastSubVersion = aSubVersion; | |
208 | lastRunRange = aRunRange; | |
209 | } | |
210 | ||
211 | } | |
212 | ||
213 | id.SetSubVersion(lastSubVersion + 1); | |
214 | } | |
fe913d8f | 215 | |
9e1ceb13 | 216 | gSystem->FreeDirectory(dirPtr); |
fe913d8f | 217 | |
9e1ceb13 | 218 | TString lastStorage = id.GetLastStorage(); |
219 | if(lastStorage.Contains(TString("grid"), TString::kIgnoreCase) && | |
220 | id.GetSubVersion() > 0 ){ | |
221 | AliError(Form("Grid to Local Storage error! local object with version v%d_s%d found:",id.GetVersion(), id.GetSubVersion()-1)); | |
222 | AliError(Form("This object has been already transferred from Grid (check v%d_s0)!",id.GetVersion())); | |
223 | return kFALSE; | |
224 | } | |
225 | ||
226 | if(lastStorage.Contains(TString("new"), TString::kIgnoreCase) && | |
227 | id.GetSubVersion() > 0 ){ | |
b05400be | 228 | AliWarning(Form("A NEW object is being stored with version v%d_s%d", |
9e1ceb13 | 229 | id.GetVersion(),id.GetSubVersion())); |
230 | AliWarning(Form("and it will hide previously stored object with v%d_s%d!", | |
231 | id.GetVersion(),id.GetSubVersion()-1)); | |
232 | } | |
233 | ||
234 | if(!lastRunRange.IsAnyRange() && !(lastRunRange.IsEqual(& id.GetAliCDBRunRange()))) | |
235 | AliWarning(Form("Run range modified w.r.t. previous version (Run%d_%d_v%d_s%d)", | |
236 | lastRunRange.GetFirstRun(), lastRunRange.GetLastRun(), | |
237 | id.GetVersion(), id.GetSubVersion()-1)); | |
238 | ||
239 | return kTRUE; | |
fe913d8f | 240 | } |
241 | ||
9e1ceb13 | 242 | //_____________________________________________________________________________ |
4b5e0dce | 243 | Bool_t AliCDBLocal::GetId(const AliCDBId& query, AliCDBId& result) { |
9e1ceb13 | 244 | // look for filename matching query (called by GetEntry) |
245 | ||
9e1ceb13 | 246 | TString dirName; |
247 | dirName += fBaseDirectory; | |
248 | dirName += '/'; | |
249 | dirName += query.GetPath(); // dirName = fDBPath/idPath | |
250 | ||
251 | void* dirPtr = gSystem->OpenDirectory(dirName); | |
252 | if (!dirPtr) { | |
024cf675 | 253 | AliDebug(2,Form("Directory <%s> not found", (query.GetPath()).Data())); |
254 | AliDebug(2,Form("in DB folder %s", fBaseDirectory.Data())); | |
4b5e0dce | 255 | return kFALSE; |
9e1ceb13 | 256 | } |
257 | ||
258 | const char* filename; | |
259 | ||
260 | AliCDBRunRange aRunRange; // the runRange got from filename | |
261 | Int_t aVersion, aSubVersion; // the version and subVersion got from filename | |
262 | ||
263 | if (!query.HasVersion()) { // neither version and subversion specified -> look for highest version and subVersion | |
264 | ||
265 | while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on files | |
266 | ||
267 | TString aString(filename); | |
268 | if (aString == "." || aString == "..") continue; | |
269 | ||
270 | if (!FilenameToId(filename, aRunRange, aVersion, aSubVersion)) continue; | |
271 | // aRunRange, aVersion, aSubVersion filled from filename | |
272 | ||
273 | if (!aRunRange.Comprises(query.GetAliCDBRunRange())) continue; | |
274 | // aRunRange contains requested run! | |
275 | ||
276 | if (result.GetVersion() < aVersion) { | |
277 | result.SetVersion(aVersion); | |
278 | result.SetSubVersion(aSubVersion); | |
279 | ||
280 | result.SetFirstRun( | |
281 | aRunRange.GetFirstRun()); | |
282 | result.SetLastRun( | |
283 | aRunRange.GetLastRun()); | |
284 | ||
285 | } else if (result.GetVersion() == aVersion | |
286 | && result.GetSubVersion() | |
287 | < aSubVersion) { | |
288 | ||
289 | result.SetSubVersion(aSubVersion); | |
290 | ||
291 | result.SetFirstRun( | |
292 | aRunRange.GetFirstRun()); | |
293 | result.SetLastRun( | |
294 | aRunRange.GetLastRun()); | |
295 | } else if (result.GetVersion() == aVersion | |
296 | && result.GetSubVersion() == aSubVersion){ | |
024cf675 | 297 | AliDebug(2,Form("More than one object valid for run %d, version %d_%d!", |
9e1ceb13 | 298 | query.GetFirstRun(), aVersion, aSubVersion)); |
4b5e0dce | 299 | gSystem->FreeDirectory(dirPtr); |
300 | return kFALSE; | |
9e1ceb13 | 301 | } |
302 | } | |
303 | ||
304 | } else if (!query.HasSubVersion()) { // version specified but not subversion -> look for highest subVersion | |
305 | ||
306 | result.SetVersion(query.GetVersion()); | |
307 | ||
308 | while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on files | |
309 | ||
310 | TString aString(filename); | |
311 | if (aString == "." || aString == "..") continue; | |
312 | ||
313 | if (!FilenameToId(filename, aRunRange, aVersion, aSubVersion)) continue; | |
314 | // aRunRange, aVersion, aSubVersion filled from filename | |
315 | ||
316 | if (!aRunRange.Comprises(query.GetAliCDBRunRange())) continue; | |
317 | // aRunRange contains requested run! | |
318 | ||
319 | if(query.GetVersion() != aVersion) continue; | |
320 | // aVersion is requested version! | |
321 | ||
322 | if(result.GetSubVersion() == aSubVersion){ | |
024cf675 | 323 | AliDebug(2,Form("More than one object valid for run %d, version %d_%d!", |
9e1ceb13 | 324 | query.GetFirstRun(), aVersion, aSubVersion)); |
4b5e0dce | 325 | gSystem->FreeDirectory(dirPtr); |
326 | return kFALSE; | |
9e1ceb13 | 327 | } |
328 | if( result.GetSubVersion() < aSubVersion) { | |
329 | ||
330 | result.SetSubVersion(aSubVersion); | |
331 | ||
332 | result.SetFirstRun( | |
333 | aRunRange.GetFirstRun()); | |
334 | result.SetLastRun( | |
335 | aRunRange.GetLastRun()); | |
336 | } | |
337 | } | |
338 | ||
339 | } else { // both version and subversion specified | |
340 | ||
341 | while ((filename = gSystem->GetDirEntry(dirPtr))) { // loop on files | |
342 | ||
343 | TString aString(filename); | |
344 | if (aString == "." || aString == "..") continue; | |
345 | ||
346 | if (!FilenameToId(filename, aRunRange, aVersion, aSubVersion)) continue; | |
347 | // aRunRange, aVersion, aSubVersion filled from filename | |
348 | ||
349 | if (!aRunRange.Comprises(query.GetAliCDBRunRange())) continue; | |
350 | // aRunRange contains requested run! | |
351 | ||
352 | if(query.GetVersion() != aVersion || query.GetSubVersion() != aSubVersion) continue; | |
353 | // aVersion and aSubVersion are requested version and subVersion! | |
354 | ||
355 | if(result.GetVersion() == aVersion && result.GetSubVersion() == aSubVersion){ | |
024cf675 | 356 | AliDebug(2,Form("More than one object valid for run %d, version %d_%d!", |
9e1ceb13 | 357 | query.GetFirstRun(), aVersion, aSubVersion)); |
4b5e0dce | 358 | gSystem->FreeDirectory(dirPtr); |
359 | return kFALSE; | |
9e1ceb13 | 360 | } |
361 | result.SetVersion(aVersion); | |
362 | result.SetSubVersion(aSubVersion); | |
363 | result.SetFirstRun(aRunRange.GetFirstRun()); | |
364 | result.SetLastRun(aRunRange.GetLastRun()); | |
365 | ||
366 | } | |
367 | } | |
368 | ||
369 | gSystem->FreeDirectory(dirPtr); | |
370 | ||
4b5e0dce | 371 | return kTRUE; |
9e1ceb13 | 372 | } |
fe913d8f | 373 | |
374 | //_____________________________________________________________________________ | |
9e1ceb13 | 375 | AliCDBEntry* AliCDBLocal::GetEntry(const AliCDBId& queryId) { |
376 | // get AliCDBEntry from the database | |
377 | ||
4b5e0dce | 378 | AliCDBId dataId(queryId.GetAliCDBPath(), -1, -1, -1, -1); |
379 | Bool_t result; | |
380 | ||
9e1ceb13 | 381 | // look for a filename matching query requests (path, runRange, version, subVersion) |
382 | if (!queryId.HasVersion()) { | |
383 | // if version is not specified, first check the selection criteria list | |
4b5e0dce | 384 | AliCDBId selectedId(queryId); |
385 | GetSelection(&selectedId); | |
386 | result = GetId(selectedId, dataId); | |
9e1ceb13 | 387 | } else { |
4b5e0dce | 388 | result = GetId(queryId, dataId); |
fe913d8f | 389 | } |
9e1ceb13 | 390 | |
4b5e0dce | 391 | if (!result || !dataId.IsSpecified()) return NULL; |
9e1ceb13 | 392 | |
393 | TString filename; | |
394 | if (!IdToFilename(dataId.GetAliCDBRunRange(), dataId.GetVersion(), | |
395 | dataId.GetSubVersion(), filename)) { | |
396 | ||
024cf675 | 397 | AliDebug(2,Form("Bad data ID encountered! Subnormal error!")); |
9e1ceb13 | 398 | return NULL; |
399 | } | |
400 | ||
401 | filename.Prepend((fBaseDirectory +'/' + queryId.GetPath() + '/')); | |
402 | ||
403 | TFile file(filename, "READ"); // open file | |
404 | if (!file.IsOpen()) { | |
024cf675 | 405 | AliDebug(2,Form("Can't open file <%s>!", filename.Data())); |
9e1ceb13 | 406 | return NULL; |
407 | } | |
408 | ||
409 | // get the only AliCDBEntry object from the file | |
410 | // the object in the file is an AliCDBEntry entry named "AliCDBEntry" | |
411 | ||
412 | TObject* anObject = file.Get("AliCDBEntry"); | |
413 | if (!anObject) { | |
024cf675 | 414 | AliDebug(2,Form("Bad storage data: NULL entry object!")); |
9e1ceb13 | 415 | return NULL; |
416 | } | |
417 | ||
418 | if (AliCDBEntry::Class() != anObject->IsA()) { | |
024cf675 | 419 | AliDebug(2,Form("Bad storage data: Invalid entry object!")); |
9e1ceb13 | 420 | return NULL; |
421 | } | |
422 | ||
423 | AliCDBId entryId = ((AliCDBEntry* ) anObject)->GetId(); | |
fe913d8f | 424 | |
9e1ceb13 | 425 | // The object's Id are not reset during storage |
426 | // If object's Id runRange or version do not match with filename, | |
427 | // it means that someone renamed file by hand. In this case a warning msg is issued. | |
428 | ||
429 | ((AliCDBEntry*) anObject)-> SetLastStorage("local"); | |
fe913d8f | 430 | |
9e1ceb13 | 431 | if(!((entryId.GetAliCDBRunRange()).IsEqual(& dataId.GetAliCDBRunRange())) || |
432 | (entryId.GetVersion() != dataId.GetVersion()) || (entryId.GetSubVersion() != dataId.GetSubVersion())){ | |
433 | AliWarning(Form("Either object Id's RunRange or version do noth match with file name:")); | |
434 | AliWarning("someone renamed file by hand!"); | |
435 | } | |
436 | ||
437 | // close file, return retieved entry | |
438 | file.Close(); | |
439 | return (AliCDBEntry*) anObject; | |
440 | } | |
fe913d8f | 441 | |
9e1ceb13 | 442 | //_____________________________________________________________________________ |
443 | void AliCDBLocal::GetEntriesForLevel0(const char* level0, | |
444 | const AliCDBId& queryId, TList* result) { | |
445 | // multiple request (AliCDBStorage::GetAll) | |
446 | ||
447 | TString level0Dir; | |
448 | level0Dir += fBaseDirectory; | |
449 | level0Dir += '/'; | |
450 | level0Dir += level0; | |
451 | ||
452 | void* level0DirPtr = gSystem->OpenDirectory(level0Dir); | |
453 | if (!level0DirPtr) { | |
024cf675 | 454 | AliDebug(2,Form("Can't open level0 directory <%s>!", |
9e1ceb13 | 455 | level0Dir.Data())); |
456 | return; | |
457 | } | |
458 | ||
459 | const char* level1; | |
460 | while ((level1 = gSystem->GetDirEntry(level0DirPtr))) { | |
461 | ||
462 | TString level1Str(level1); | |
463 | if (level1Str == "." || level1Str == "..") { | |
464 | continue; | |
465 | } | |
466 | ||
467 | if (queryId.GetAliCDBPath().Level1Comprises(level1)) { | |
468 | GetEntriesForLevel1(level0, level1, queryId, result); | |
469 | } | |
470 | } | |
fe913d8f | 471 | |
9e1ceb13 | 472 | gSystem->FreeDirectory(level0DirPtr); |
fe913d8f | 473 | } |
474 | ||
9e1ceb13 | 475 | //_____________________________________________________________________________ |
476 | void AliCDBLocal::GetEntriesForLevel1(const char* level0, const char* level1, | |
477 | const AliCDBId& queryId, TList* result) { | |
478 | // multiple request (AliCDBStorage::GetAll) | |
479 | ||
480 | TString level1Dir; | |
481 | level1Dir += fBaseDirectory; | |
482 | level1Dir += '/'; | |
483 | level1Dir += level0; | |
484 | level1Dir += '/'; | |
485 | level1Dir += level1; | |
486 | ||
487 | void* level1DirPtr = gSystem->OpenDirectory(level1Dir); | |
488 | if (!level1DirPtr) { | |
024cf675 | 489 | AliDebug(2,Form("Can't open level1 directory <%s>!", |
9e1ceb13 | 490 | level1Dir.Data())); |
491 | return; | |
492 | } | |
fe913d8f | 493 | |
9e1ceb13 | 494 | const char* level2; |
495 | while ((level2 = gSystem->GetDirEntry(level1DirPtr))) { | |
496 | ||
497 | TString level2Str(level2); | |
498 | if (level2Str == "." || level2Str == "..") { | |
499 | continue; | |
500 | } | |
501 | ||
502 | if (queryId.GetAliCDBPath().Level2Comprises(level2)) { | |
503 | ||
504 | AliCDBPath entryPath(level0, level1, level2); | |
505 | AliCDBId entryId(entryPath, queryId.GetAliCDBRunRange(), | |
506 | queryId.GetVersion(), queryId.GetSubVersion()); | |
507 | ||
508 | AliCDBEntry* anEntry = GetEntry(entryId); | |
509 | if (anEntry) { | |
510 | result->Add(anEntry); | |
511 | } | |
512 | } | |
513 | } | |
fe913d8f | 514 | |
9e1ceb13 | 515 | gSystem->FreeDirectory(level1DirPtr); |
516 | } | |
fe913d8f | 517 | |
9e1ceb13 | 518 | //_____________________________________________________________________________ |
519 | TList* AliCDBLocal::GetEntries(const AliCDBId& queryId) { | |
520 | // multiple request (AliCDBStorage::GetAll) | |
521 | ||
522 | void* storageDirPtr = gSystem->OpenDirectory(fBaseDirectory); | |
523 | if (!storageDirPtr) { | |
024cf675 | 524 | AliDebug(2,Form("Can't open storage directory <%s>", |
9e1ceb13 | 525 | fBaseDirectory.Data())); |
526 | return NULL; | |
527 | } | |
528 | ||
529 | TList* result = new TList(); | |
530 | result->SetOwner(); | |
531 | ||
532 | const char* level0; | |
533 | while ((level0 = gSystem->GetDirEntry(storageDirPtr))) { | |
534 | ||
535 | TString level0Str(level0); | |
536 | if (level0Str == "." || level0Str == "..") { | |
537 | continue; | |
538 | } | |
539 | ||
540 | if (queryId.GetAliCDBPath().Level0Comprises(level0)) { | |
541 | GetEntriesForLevel0(level0, queryId, result); | |
542 | } | |
543 | } | |
544 | ||
545 | gSystem->FreeDirectory(storageDirPtr); | |
546 | ||
547 | return result; | |
fe913d8f | 548 | } |
549 | ||
550 | //_____________________________________________________________________________ | |
9e1ceb13 | 551 | Bool_t AliCDBLocal::PutEntry(AliCDBEntry* entry) { |
552 | // put an AliCDBEntry object into the database | |
553 | ||
554 | AliCDBId& id = entry->GetId(); | |
555 | ||
556 | // set version and subVersion for the entry to be stored | |
557 | if (!PrepareId(id)) return kFALSE; | |
558 | ||
559 | ||
560 | // build filename from entry's id | |
561 | TString filename; | |
562 | if (!IdToFilename(id.GetAliCDBRunRange(), id.GetVersion(), | |
563 | id.GetSubVersion(), filename)) { | |
564 | ||
024cf675 | 565 | AliDebug(2,Form("Bad ID encountered! Subnormal error!")); |
9e1ceb13 | 566 | return kFALSE; |
567 | } | |
568 | ||
569 | filename.Prepend(fBaseDirectory +'/' + id.GetPath() + '/'); | |
570 | ||
571 | // open file | |
572 | TFile file(filename, "CREATE"); | |
573 | if (!file.IsOpen()) { | |
574 | AliError(Form("Can't open file <%s>!", filename.Data())); | |
575 | return kFALSE; | |
576 | } | |
577 | ||
578 | entry->SetVersion(id.GetVersion()); | |
579 | entry->SetSubVersion(id.GetSubVersion()); | |
580 | ||
581 | // write object (key name: "AliCDBEntry") | |
582 | Bool_t result = file.WriteTObject(entry, "AliCDBEntry"); | |
024cf675 | 583 | if (!result) AliDebug(2,Form("Can't write entry to file: %s", filename.Data())); |
9e1ceb13 | 584 | |
585 | file.Close(); | |
b05400be | 586 | if(result) AliInfo(Form("CDB object stored into file %s",filename.Data())); |
587 | ||
588 | return result; | |
589 | } | |
590 | ||
591 | //_____________________________________________________________________________ | |
592 | TList* AliCDBLocal::GetIdListFromFile(const char* fileName){ | |
593 | ||
594 | TString fullFileName(fileName); | |
595 | fullFileName.Prepend(fBaseDirectory+'/'); | |
596 | TFile *file = TFile::Open(fullFileName); | |
597 | if (!file) { | |
598 | AliError(Form("Can't open selection file <%s>!", fullFileName.Data())); | |
599 | return NULL; | |
600 | } | |
601 | file->cd(); | |
602 | ||
603 | TList *list = new TList(); | |
604 | list->SetOwner(); | |
605 | int i=0; | |
606 | TString keycycle; | |
607 | ||
608 | AliCDBId *id; | |
609 | while(1){ | |
610 | i++; | |
611 | keycycle = "AliCDBId;"; | |
612 | keycycle+=i; | |
613 | ||
614 | id = (AliCDBId*) file->Get(keycycle); | |
615 | if(!id) break; | |
616 | list->AddFirst(id); | |
617 | } | |
618 | file->Close(); delete file; file=0; | |
619 | return list; | |
620 | } | |
621 | ||
622 | //_____________________________________________________________________________ | |
623 | Bool_t AliCDBLocal::Contains(const char* path) const{ | |
624 | // check for path in storage's fBaseDirectory | |
625 | ||
626 | TString dirName; | |
627 | dirName += fBaseDirectory; | |
628 | dirName += '/'; | |
629 | dirName += path; // dirName = fDBPath/path | |
630 | Bool_t result=kFALSE; | |
631 | ||
632 | void* dirPtr = gSystem->OpenDirectory(dirName); | |
633 | if (dirPtr) result=kTRUE; | |
634 | gSystem->FreeDirectory(dirPtr); | |
9e1ceb13 | 635 | |
636 | return result; | |
fe913d8f | 637 | } |
638 | ||
9e1ceb13 | 639 | ///////////////////////////////////////////////////////////////////////////////////////////////// |
640 | // // | |
641 | // AliCDBLocal factory // | |
642 | // // | |
643 | ///////////////////////////////////////////////////////////////////////////////////////////////// | |
644 | ||
645 | ClassImp(AliCDBLocalFactory) | |
646 | ||
647 | //_____________________________________________________________________________ | |
648 | Bool_t AliCDBLocalFactory::Validate(const char* dbString) { | |
649 | // check if the string is valid local URI | |
650 | ||
651 | TRegexp dbPattern("^local://.+$"); | |
652 | ||
653 | return TString(dbString).Contains(dbPattern); | |
fe913d8f | 654 | } |
655 | ||
9e1ceb13 | 656 | //_____________________________________________________________________________ |
657 | AliCDBParam* AliCDBLocalFactory::CreateParameter(const char* dbString) { | |
658 | // create AliCDBLocalParam class from the URI string | |
659 | ||
660 | if (!Validate(dbString)) { | |
661 | return NULL; | |
662 | } | |
663 | ||
664 | TString pathname(dbString + sizeof("local://") - 1); | |
665 | ||
666 | gSystem->ExpandPathName(pathname); | |
667 | ||
668 | if (pathname[0] != '/') { | |
669 | pathname.Prepend(TString(gSystem->WorkingDirectory()) + '/'); | |
670 | } | |
671 | ||
9e1ceb13 | 672 | return new AliCDBLocalParam(pathname); |
fe913d8f | 673 | } |
674 | ||
9e1ceb13 | 675 | //_____________________________________________________________________________ |
676 | AliCDBStorage* AliCDBLocalFactory::Create(const AliCDBParam* param) { | |
677 | // create AliCDBLocal storage instance from parameters | |
678 | ||
679 | if (AliCDBLocalParam::Class() == param->IsA()) { | |
680 | ||
681 | const AliCDBLocalParam* localParam = | |
682 | (const AliCDBLocalParam*) param; | |
683 | ||
684 | return new AliCDBLocal(localParam->GetPath()); | |
685 | } | |
fe913d8f | 686 | |
9e1ceb13 | 687 | return NULL; |
fe913d8f | 688 | } |
689 | ||
9e1ceb13 | 690 | ///////////////////////////////////////////////////////////////////////////////////////////////// |
691 | // // | |
692 | // AliCDBLocal Parameter class // // | |
693 | // // | |
694 | ///////////////////////////////////////////////////////////////////////////////////////////////// | |
695 | ||
696 | ClassImp(AliCDBLocalParam) | |
697 | ||
fe913d8f | 698 | //_____________________________________________________________________________ |
9e1ceb13 | 699 | AliCDBLocalParam::AliCDBLocalParam() { |
700 | // default constructor | |
fe913d8f | 701 | |
fe913d8f | 702 | } |
703 | ||
9e1ceb13 | 704 | //_____________________________________________________________________________ |
705 | AliCDBLocalParam::AliCDBLocalParam(const char* dbPath): | |
706 | fDBPath(dbPath) | |
707 | { | |
708 | // constructor | |
709 | ||
710 | SetType("local"); | |
711 | SetURI(TString("local://") + dbPath); | |
712 | } | |
713 | ||
714 | //_____________________________________________________________________________ | |
715 | AliCDBLocalParam::~AliCDBLocalParam() { | |
716 | // destructor | |
717 | ||
718 | } | |
fe913d8f | 719 | |
720 | //_____________________________________________________________________________ | |
9e1ceb13 | 721 | AliCDBParam* AliCDBLocalParam::CloneParam() const { |
722 | // clone parameter | |
723 | ||
724 | return new AliCDBLocalParam(fDBPath); | |
fe913d8f | 725 | } |
fe913d8f | 726 | |
9e1ceb13 | 727 | //_____________________________________________________________________________ |
728 | ULong_t AliCDBLocalParam::Hash() const { | |
729 | // return Hash function | |
730 | ||
731 | return fDBPath.Hash(); | |
fe913d8f | 732 | } |
9e1ceb13 | 733 | |
734 | //_____________________________________________________________________________ | |
735 | Bool_t AliCDBLocalParam::IsEqual(const TObject* obj) const { | |
736 | // check if this object is equal to AliCDBParam obj | |
737 | ||
738 | if (this == obj) { | |
739 | return kTRUE; | |
740 | } | |
741 | ||
742 | if (AliCDBLocalParam::Class() != obj->IsA()) { | |
743 | return kFALSE; | |
744 | } | |
745 | ||
746 | AliCDBLocalParam* other = (AliCDBLocalParam*) obj; | |
747 | ||
748 | return fDBPath == other->fDBPath; | |
749 | } | |
750 |