add option to make a summary tree; get collision geometry info for Hijing events...

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

#ifndef ALI_TEST_SHUTTLE_H
#define ALI_TEST_SHUTTLE_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

//
// test implementation of the AliShuttleInterface, to be used for local tests of preprocessors
//

#include <AliShuttleInterface.h>
#include <TString.h>

class TMap;
class TList;
class AliCDBMetaData;
class AliCDBPath;
class AliCDBPath;

class AliTestShuttle : public AliShuttleInterface
{
  public:
    AliTestShuttle(Int_t run, UInt_t startTime, UInt_t endTime);
    virtual ~AliTestShuttle();

    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 SetInputRunType(const char* runType) { fRunType = runType; }
    void SetInputHLTStatus(Bool_t status) { fHLTStatus = status; }
    void SetInputTriggerConfiguration(const char* configuration) { fTriggerConfiguration = configuration; }
    void SetInputCTPTimeParams(const char* ctptiming) { fCTPtiming = ctptiming; }
    void SetInputTriggerDetectorMask(const char* triggerDetectorMask) { fTriggerDetectorMask = triggerDetectorMask; }
    Bool_t AddInputCDBEntry(AliCDBEntry* entry);
    void SetTimeCreated(UInt_t timeCreated) { fTimeCreated = timeCreated;}
    void SetDCSQueryOffset(UInt_t dcsQueryOffset) { fDCSQueryOffset = dcsQueryOffset;}
    void Process();

    // AliShuttleInterface functions
    virtual Bool_t Store(const AliCDBPath& path, TObject* object, AliCDBMetaData* metaData,
        				Int_t validityStart = 0, Bool_t validityInfinite = kFALSE);
    virtual Bool_t StoreReferenceData(const AliCDBPath& path, TObject* object, AliCDBMetaData* metaData);
    virtual Bool_t StoreReferenceFile(const char* detector, const char* localFile, const char* gridFileName);
    virtual Bool_t StoreRunMetadataFile(const char* localFile, const char* gridFileName);
    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 = 0);
    virtual TList* GetFileIDs(Int_t system, const char* detector, const char* source);
    virtual const char* GetRunParameter(const char* key);
    virtual UInt_t GetStartTimeDCSQuery();
    virtual UInt_t GetEndTimeDCSQuery();
    virtual AliCDBEntry* GetFromOCDB(const char* detector, const AliCDBPath& path);
    virtual const char* GetRunType();
    virtual Bool_t GetHLTStatus() {return fHLTStatus;}
    virtual const char* GetTriggerConfiguration();
    virtual const char* GetCTPTimeParams();
    virtual const char* GetTriggerDetectorMask(); 
    virtual void Log(const char* detector, const char* message, UInt_t level=3);

    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);

    virtual void SendMLFromDet(const char* value);

  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
    UInt_t fTimeCreated;    // timecreated that is simulated with the AliTestShuttle
    UInt_t fDCSQueryOffset; // DCS query offset

    TMap* fInputFiles;      // files for GetFile, GetFileSources
    TMap* fRunParameters;   // run parameters
    TString fRunType; 	    // run type
    Bool_t fHLTStatus; 	    // HLT status for current run (on=1/off=0)
    TObjArray* fPreprocessors; // list of preprocessors that are to be tested
    TMap* fDcsAliasMap; // DCS data for testing
    TString fTriggerConfiguration;  // trigger configuration for testing
    TString fTriggerDetectorMask;   // trigger detector mask for testing
    TString fCTPtiming;         // CTP time paramters for testing 

  private:
    Bool_t CopyFileLocally(TString& targetDir, const char* localFile, const char* gridFileName);
    const char* GetRefFilePrefix(const char* base, const char* detector);
    
    ClassDef(AliTestShuttle, 0);
};

#endif