Small interface change in StoreReferenceData

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

diff --git a/SHUTTLE/TestShuttle/AliTestShuttle.h b/SHUTTLE/TestShuttle/AliTestShuttle.h
index b51567dbdfb1ae8a1af43f98ca81d140c9d9b728..e6208e86bab84b1b16c3adb25751875bb8c51cc1 100644 (file)
--- a/SHUTTLE/TestShuttle/AliTestShuttle.h
+++ b/SHUTTLE/TestShuttle/AliTestShuttle.h
@@ -15,20 +15,37 @@
 class TMap;
 class TList;
 class AliCDBMetaData;
+class AliCDBPath;
 
 class AliTestShuttle : public AliShuttleInterface
 {
   public:
-    AliTestShuttle(TMap* inputFiles);
+    AliTestShuttle(Int_t run, UInt_t startTime, UInt_t endTime);
     virtual ~AliTestShuttle();
 
-    virtual Int_t Store(const char* detector, TObject* object, AliCDBMetaData* metaData);
+    void AddInputFile(Int_t system, const char* detector, const char* id, const char* source, const char* fileName);
+    void SetDCSInput(TMap* dcsAliasMap) { fDcsAliasMap = dcsAliasMap; }
+
+    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);
     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 void Log(const char* detector, const char* message);
 
+    virtual void RegisterPreprocessor(AliPreprocessor* preprocessor);
+
   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
+    TObjArray* fPreprocessors; // list of preprocessors that are to be tested
+    TMap* fDcsAliasMap; // DCS data for testing
 
   private:
     ClassDef(AliTestShuttle, 0);