Function GetFromOCDB added to AliShuttle and AliTestShuttle.
[u/mrichter/AliRoot.git] / SHUTTLE / TestShuttle / AliTestShuttle.h
1 #ifndef ALI_TEST_SHUTTLE_H
2 #define ALI_TEST_SHUTTLE_H
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 /* $Id$ */
8
9 //
10 // test implementation of the AliShuttleInterface, to be used for local tests of preprocessors
11 //
12
13 #include <AliShuttleInterface.h>
14 #include <TString.h>
15
16 class TMap;
17 class TList;
18 class AliCDBMetaData;
19 class AliCDBPath;
20 class AliCDBPath;
21
22 class AliTestShuttle : public AliShuttleInterface
23 {
24   public:
25     AliTestShuttle(Int_t run, UInt_t startTime, UInt_t endTime);
26     virtual ~AliTestShuttle();
27
28     void AddInputFile(Int_t system, const char* detector, const char* id, const char* source, const char* fileName);
29     void SetDCSInput(TMap* dcsAliasMap) { fDcsAliasMap = dcsAliasMap; }
30     void AddInputRunParameter(const char* key, const char* value);
31     Bool_t AddInputCDBEntry(AliCDBEntry* entry);
32
33     void Process();
34
35     // AliShuttleInterface functions
36     virtual UInt_t Store(const AliCDBPath& path, TObject* object, AliCDBMetaData* metaData,
37                                         Int_t validityStart = 0, Bool_t validityInfinite = kFALSE);
38     virtual UInt_t StoreReferenceData(const AliCDBPath& path, TObject* object, AliCDBMetaData* metaData);
39     virtual const char* GetFile(Int_t system, const char* detector, const char* id, const char* source);
40     virtual TList* GetFileSources(Int_t system, const char* detector, const char* id);
41     virtual const char* GetRunParameter(const char* key);
42     virtual AliCDBEntry* GetFromOCDB(const AliCDBPath& path);
43     virtual void Log(const char* detector, const char* message);
44
45     virtual void RegisterPreprocessor(AliPreprocessor* preprocessor);
46
47     static void SetMainCDB (TString mainCDB) {fgkMainCDB = mainCDB;}
48     static void SetLocalCDB (TString localCDB) {fgkLocalCDB = localCDB;}
49
50     static void SetMainRefStorage (TString mainRefStorage) {fgkMainRefStorage = mainRefStorage;}
51     static void SetLocalRefStorage (TString localRefStorage) {fgkLocalRefStorage = localRefStorage;}
52
53     static void SetShuttleTempDir (const char* tmpDir);
54     static void SetShuttleLogDir (const char* logDir);
55
56   protected:
57
58     Int_t fRun;         // run that is simulated with the AliTestShuttle
59     UInt_t fStartTime;  // starttime that is simulated with the AliTestShuttle
60     UInt_t fEndTime;    // endtime that is simulated with the AliTestShuttle
61
62     TMap* fInputFiles;      // files for GetFile, GetFileSources
63     TMap* fRunParameters;    // run parameters
64     TObjArray* fPreprocessors; // list of preprocessors that are to be tested
65     TMap* fDcsAliasMap; // DCS data for testing
66
67   private:
68     ClassDef(AliTestShuttle, 0);
69 };
70
71 #endif