major update (Alberto)

[u/mrichter/AliRoot.git] / SHUTTLE / TestShuttle / AliTestShuttle.h

diff --git a/SHUTTLE/TestShuttle/AliTestShuttle.h b/SHUTTLE/TestShuttle/AliTestShuttle.h
index 7586c65bcc1994ecb3cc0f83afe4b7de6ba62ca0..2e7d5fe43f56d9c34a30d0f0ebe8e30fbaf6115f 100644 (file)
--- a/SHUTTLE/TestShuttle/AliTestShuttle.h
+++ b/SHUTTLE/TestShuttle/AliTestShuttle.h
@@ -11,6 +11,7 @@
 //
 
 #include <AliShuttleInterface.h>
+#include <TString.h>
 
 class TMap;
 class TList;
@@ -23,28 +24,38 @@ class AliTestShuttle : public AliShuttleInterface
     AliTestShuttle(Int_t run, UInt_t startTime, UInt_t endTime);
     virtual ~AliTestShuttle();
 
+    static TString GetOCDBStorage () {return fgkOCDBUri;}
+    static void SetOCDBStorage (TString ocdbUri) {fgkOCDBUri = ocdbUri;}
+    static TString GetReferenceStorage () {return fgkRefUri;}
+    static void SetReferenceStorage (TString refUri) {fgkRefUri = refUri;}
+
     void AddInputFile(Int_t system, const char* detector, const char* id, const char* source, const char* fileName);
     void SetDCSInput(TMap* dcsAliasMap) { fDcsAliasMap = dcsAliasMap; }
+    void AddInputRunParameter(const char* key, const char* value);
 
     void Process();
 
     // AliShuttleInterface functions
     virtual UInt_t Store(const AliCDBPath& path, TObject* object, AliCDBMetaData* metaData,
                                        Int_t validityStart = 0, Bool_t validityInfinite = kFALSE);
-    virtual UInt_t StoreReferenceData(const AliCDBPath& path, TObject* object, AliCDBMetaData* metaData,
-                                       Int_t validityStart = 0, Bool_t validityInfinite = kFALSE);
+    virtual UInt_t StoreReferenceData(const AliCDBPath& path, TObject* object, AliCDBMetaData* metaData);
     virtual const char* GetFile(Int_t system, const char* detector, const char* id, const char* source);
     virtual TList* GetFileSources(Int_t system, const char* detector, const char* id);
+    virtual const char* GetRunParameter(const char* key);
     virtual void Log(const char* detector, const char* message);
 
     virtual void RegisterPreprocessor(AliPreprocessor* preprocessor);
 
   protected:
+    static TString         fgkOCDBUri;         // URI of the test OCDB storage
+    static TString         fgkRefUri;          // URI of the test Reference storage
+
     Int_t fRun;         // run that is simulated with the AliTestShuttle
     UInt_t fStartTime;  // starttime that is simulated with the AliTestShuttle
     UInt_t fEndTime;    // endtime that is simulated with the AliTestShuttle
 
-    TMap* fInputFiles;   // files for GetFile, GetFileSources
+    TMap* fInputFiles;      // files for GetFile, GetFileSources
+    TMap* fRunParameters;    // run parameters
     TObjArray* fPreprocessors; // list of preprocessors that are to be tested
     TMap* fDcsAliasMap; // DCS data for testing