SHUTTLE/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     void SetInputRunType(const char* runType) { fRunType = runType; }
  32     void SetInputHLTStatus(Bool_t status) { fHLTStatus = status; }
  33     void SetInputTriggerConfiguration(const char* configuration) { fTriggerConfiguration = configuration; }
  34     void SetInputTriggerDetectorMask(const char* triggerDetectorMask) { fTriggerDetectorMask = triggerDetectorMask; }
  35     Bool_t AddInputCDBEntry(AliCDBEntry* entry);
  36     void SetTimeCreated(UInt_t timeCreated) { fTimeCreated = timeCreated;}
  37     void SetDCSQueryOffset(UInt_t dcsQueryOffset) { fDCSQueryOffset = dcsQueryOffset;}
  38     void Process();
  39
  40     // AliShuttleInterface functions
  41     virtual Bool_t Store(const AliCDBPath& path, TObject* object, AliCDBMetaData* metaData,
  42                                         Int_t validityStart = 0, Bool_t validityInfinite = kFALSE);
  43     virtual Bool_t StoreReferenceData(const AliCDBPath& path, TObject* object, AliCDBMetaData* metaData);
  44     virtual Bool_t StoreReferenceFile(const char* detector, const char* localFile, const char* gridFileName);
  45     virtual Bool_t StoreRunMetadataFile(const char* localFile, const char* gridFileName);
  46     virtual const char* GetFile(Int_t system, const char* detector, const char* id, const char* source);
  47     virtual TList* GetFileSources(Int_t system, const char* detector, const char* id = 0);
  48     virtual TList* GetFileIDs(Int_t system, const char* detector, const char* source);
  49     virtual const char* GetRunParameter(const char* key);
  50     virtual UInt_t GetStartTimeDCSQuery();
  51     virtual UInt_t GetEndTimeDCSQuery();
  52     virtual AliCDBEntry* GetFromOCDB(const char* detector, const AliCDBPath& path);
  53     virtual const char* GetRunType();
  54     virtual Bool_t GetHLTStatus() {return fHLTStatus;}
  55     virtual const char* GetTriggerConfiguration();
  56     virtual const char* GetTriggerDetectorMask();
  57     virtual void Log(const char* detector, const char* message);
  58
  59     virtual void RegisterPreprocessor(AliPreprocessor* preprocessor);
  60
  61     static void SetMainCDB (TString mainCDB) {fgkMainCDB = mainCDB;}
  62     static void SetLocalCDB (TString localCDB) {fgkLocalCDB = localCDB;}
  63
  64     static void SetMainRefStorage (TString mainRefStorage) {fgkMainRefStorage = mainRefStorage;}
  65     static void SetLocalRefStorage (TString localRefStorage) {fgkLocalRefStorage = localRefStorage;}
  66
  67     static void SetShuttleTempDir (const char* tmpDir);
  68     static void SetShuttleLogDir (const char* logDir);
  69
  70   protected:
  71
  72     Int_t fRun;             // run that is simulated with the AliTestShuttle
  73     UInt_t fStartTime;      // starttime that is simulated with the AliTestShuttle
  74     UInt_t fEndTime;        // endtime that is simulated with the AliTestShuttle
  75     UInt_t fTimeCreated;    // timecreated that is simulated with the AliTestShuttle
  76     UInt_t fDCSQueryOffset; // DCS query offset
  77
  78     TMap* fInputFiles;      // files for GetFile, GetFileSources
  79     TMap* fRunParameters;   // run parameters
  80     TString fRunType;       // run type
  81     Bool_t fHLTStatus;      // HLT status for current run (on=1/off=0)
  82     TObjArray* fPreprocessors; // list of preprocessors that are to be tested
  83     TMap* fDcsAliasMap; // DCS data for testing
  84     TString fTriggerConfiguration;  // trigger configuration for testing
  85     TString fTriggerDetectorMask;   // trigger detector mask for testing
  86
  87   private:
  88     Bool_t CopyFileLocally(TString& targetDir, const char* localFile, const char* gridFileName);
  89     const char* GetRefFilePrefix(const char* base, const char* detector);
  90
  91     ClassDef(AliTestShuttle, 0);
  92 };
  93
  94 #endif