Run type field added in SHUTTLE framework. Run type is read from "run type" logbook...

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

diff --git a/SHUTTLE/TestShuttle/AliTestShuttle.h b/SHUTTLE/TestShuttle/AliTestShuttle.h
index 5a7ab2af502de139ff021147dc78f0723670d0b4..db6adba1f734bbcb27ebc8caaaeb3d29fe61210c 100644 (file)
--- a/SHUTTLE/TestShuttle/AliTestShuttle.h
+++ b/SHUTTLE/TestShuttle/AliTestShuttle.h
@@ -11,10 +11,13 @@
 //
 
 #include <AliShuttleInterface.h>
+#include <TString.h>
 
 class TMap;
 class TList;
 class AliCDBMetaData;
+class AliCDBPath;
+class AliCDBPath;
 
 class AliTestShuttle : public AliShuttleInterface
 {
@@ -24,23 +27,43 @@ class AliTestShuttle : public AliShuttleInterface
 
     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 AddInputRunType(const char* detCode, const char* runType);
+    Bool_t AddInputCDBEntry(AliCDBEntry* entry);
 
     void Process();
 
     // AliShuttleInterface functions
-    virtual UInt_t Store(const char* detector, TObject* object, AliCDBMetaData* metaData, Int_t validityStart = 0, Bool_t validityInfinite = kFALSE);
+    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);
     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 AliCDBEntry* GetFromOCDB(const AliCDBPath& path);
+    virtual const char* GetRunType(const char* detCode);
     virtual void Log(const char* detector, const char* message);
 
     virtual void RegisterPreprocessor(AliPreprocessor* preprocessor);
 
+    static void SetMainCDB (TString mainCDB) {fgkMainCDB = mainCDB;}
+    static void SetLocalCDB (TString localCDB) {fgkLocalCDB = localCDB;}
+
+    static void SetMainRefStorage (TString mainRefStorage) {fgkMainRefStorage = mainRefStorage;}
+    static void SetLocalRefStorage (TString localRefStorage) {fgkLocalRefStorage = localRefStorage;}
+
+    static void SetShuttleTempDir (const char* tmpDir);
+    static void SetShuttleLogDir (const char* logDir);
+
   protected:
+
     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
+    TMap* fRunTypeMap;             // run type
     TObjArray* fPreprocessors; // list of preprocessors that are to be tested
     TMap* fDcsAliasMap; // DCS data for testing