move OCDB manipulation from local train (run.C) to the task level

[u/mrichter/AliRoot.git] / PWG1 / TRD / AliTRDinfoGen.h

diff --git a/PWG1/TRD/AliTRDinfoGen.h b/PWG1/TRD/AliTRDinfoGen.h
index ebd9e8949b0e167533715e4c6930805d76056051..2550bdb8aa8a97b1b85d462c02d91fe45a0657af 100644 (file)
--- a/PWG1/TRD/AliTRDinfoGen.h
+++ b/PWG1/TRD/AliTRDinfoGen.h
@@ -34,6 +34,7 @@ public:
    ,kUseLocalEvSelection  = BIT(19)
    ,kUseLocalTrkSelection = BIT(20)
    ,kCollision            = BIT(21)
+   ,kOCDB                 = BIT(22)
   };
 
   AliTRDinfoGen();
@@ -45,20 +46,23 @@ public:
   static Float_t GetEndTPC() { return fgkTPC;}
   static Float_t GetEndTRD() { return fgkTRD;}
   Int_t   GetNRefFigures() const  { return 1;} 
+  const char* GetOCDB() const {return fOCDB.Data();}
   Bool_t  GetRefFigure(Int_t ifig);
   Bool_t  Load(const Char_t *fn="AnalysisResults.root", const Char_t *dir="TRD_Performance", const Char_t *name=NULL);
 
   Bool_t  HasMCdata() const       { return TestBit(kMCdata);};
   // temporary until check with AliAnalysisTaskSE collision selection mechanism
+  Bool_t  IsInitOCDB() const {return TestBit(kOCDB);}
   Bool_t  IsCollision() const {return TestBit(kCollision);}
+  void    SetInitOCDB(Bool_t set=kTRUE) {SetBit(kOCDB, set);}
   void    SetCollision(Bool_t set=kTRUE) {SetBit(kCollision, set);}
-
   void    SetLocalEvSelection(AliTRDeventCuts */*cut*/){;} 
   void    SetLocalEvSelection(Bool_t use=kTRUE) {SetBit(kUseLocalEvSelection, use);}
   void    SetLocalTrkSelection(AliESDtrackCuts */*cut*/){;} 
   void    SetLocalTrkSelection(Bool_t use=kTRUE) {SetBit(kUseLocalTrkSelection, use);}
   void    SetLocalV0Selection(AliTRDv0Info *v0);
   void    SetMCdata(Bool_t mc = kTRUE) {SetBit(kMCdata, mc);}
+  void    SetOCDB(const char *ocdb) {fOCDB=ocdb;}
   void    SetTrigger(const Char_t *trigger);
 
   Bool_t  UseLocalEvSelection() const {return TestBit(kUseLocalEvSelection);}
@@ -94,6 +98,7 @@ private:
   AliTRDeventCuts  *fEventCut;       // event cut
   AliESDtrackCuts  *fTrackCut;       // track cut
   AliTRDv0Info     *fV0Cut;          // v0 cut
+  TString           fOCDB;           // OCDB location
   AliTRDtrackInfo  *fTrackInfo;      //! Track info
   AliTRDeventInfo  *fEventInfo;             //! Event info
   AliTRDv0Info     *fV0Info;         //! V0 info