PWG1/TRD/AliTRDinfoGen.h

   1 #ifndef ALITRDINFOGEN_H\r
   2 #define ALITRDINFOGEN_H\r
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
   4  * See cxx source for full Copyright notice                               */\r
   5 \r
   6 /* $Id: AliTRDinfoGen.h 27496 2008-07-22 08:35:45Z cblume $ */\r
   7 \r
   8 ////////////////////////////////////////////////////////////////////////////\r
   9 //                                                                        //\r
  10 //  TRD Performance tender wagon                                          //\r
  11 //                                                                        //\r
  12 ////////////////////////////////////////////////////////////////////////////\r
  13 \r
  14 #ifndef ALIANALYSISTASKSE_H\r
  15 #include "AliAnalysisTaskSE.h"\r
  16 #endif\r
  17 \r
  18 class AliESDEvent;\r
  19 class AliMCEvent;\r
  20 class AliESDfriend;\r
  21 class AliTRDtrackInfo;\r
  22 class AliTRDeventInfo;\r
  23 class AliTRDv0Info;\r
  24 class AliTRDeventCuts;\r
  25 class AliESDtrackCuts;\r
  26 class TObjArray;\r
  27 class TString;\r
  28 class TTreeSRedirector;\r
  29 class AliTRDReconstructor;\r
  30 class AliTRDgeometry;\r
  31 class AliTRDinfoGen : public AliAnalysisTaskSE\r
  32 {\r
  33 public:\r
  34   enum AliTRDinfoGenSteeringBits{\r
  35     kMCdata               = BIT(18)\r
  36    ,kUseLocalTrkSelection = BIT(19)\r
  37    ,kCollision            = BIT(20)\r
  38    ,kOCDB                 = BIT(21)\r
  39   };\r
  40   enum AliTRDinfoGenObjects{\r
  41      kTracksESD =  0\r
  42     ,kTracksMC\r
  43     ,kV0\r
  44     ,kTPC\r
  45     ,kTRDin\r
  46     ,kTRDout\r
  47     ,kBarrel\r
  48     ,kBarrelMC\r
  49     ,kSA\r
  50     ,kSAMC\r
  51     ,kKink\r
  52     ,kKinkMC\r
  53     ,kBarrelFriend\r
  54     ,kSAFriend\r
  55     ,kNObjects\r
  56   };\r
  57   enum AliTRDinfoGenClasses{\r
  58      kStatTrk = 0\r
  59     ,kEvType\r
  60     ,kBC\r
  61     ,kTrigger\r
  62     ,kNclasses\r
  63   };\r
  64   AliTRDinfoGen();\r
  65   AliTRDinfoGen(char* name);\r
  66   virtual ~AliTRDinfoGen();\r
  67   \r
  68   static Float_t GetEndITS() { return fgkITS;}\r
  69   static Float_t GetEndTPC() { return fgkTPC;}\r
  70   static Float_t GetEndTRD() { return fgkTRD;}\r
  71   Int_t   GetNRefFigures() const  { return 1;} \r
  72   const char* GetOCDB() const {return fOCDB.Data();}\r
  73   Bool_t  GetRefFigure(Int_t ifig);\r
  74   Bool_t  Load(const Char_t *fn="AnalysisResults.root", const Char_t *dir="TRD_Performance", const Char_t *name=NULL);\r
  75 \r
  76   Bool_t  HasMCdata() const       { return TestBit(kMCdata);};\r
  77   // temporary until check with AliAnalysisTaskSE collision selection mechanism\r
  78   Bool_t  IsInitOCDB() const {return TestBit(kOCDB);}\r
  79   Bool_t  IsCollision() const {return TestBit(kCollision);}\r
  80   static const AliTRDReconstructor* Reconstructor() {return fgReconstructor;}\r
  81   static AliTRDgeometry*      Geometry() {return fgGeo;}\r
  82   void    SetInitOCDB(Bool_t set=kTRUE) {SetBit(kOCDB, set);}\r
  83   void    SetCollision(Bool_t set=kTRUE) {SetBit(kCollision, set);}\r
  84   //void    SetLocalEvSelection(const AliTRDeventCuts */*cut*/){;} \r
  85   void    SetLocalEvSelection(const AliTRDeventCuts &ec);\r
  86   //void    SetLocalTrkSelection(const AliESDtrackCuts */*cut*/){;} \r
  87   void    SetLocalTrkSelection(Bool_t use=kTRUE) {SetBit(kUseLocalTrkSelection, use);}\r
  88   void    SetLocalV0Selection(const AliTRDv0Info &v0);\r
  89   void    SetMCdata(Bool_t mc = kTRUE) {SetBit(kMCdata, mc);}\r
  90   void    SetOCDB(const char *ocdb) {fOCDB=ocdb;}\r
  91 \r
  92   Bool_t  UseLocalEvSelection() const {return Bool_t(fEventCut);}\r
  93   Bool_t  UseLocalTrkSelection() const {return TestBit(kUseLocalTrkSelection);}\r
  94   void    UserCreateOutputObjects();\r
  95   void    UserExec(Option_t *);\r
  96   void    Terminate(Option_t* option = "");\r
  97 private:\r
  98   // rough radial limits for TRD\r
  99   static const Float_t fgkITS;      // end ITS\r
 100   static const Float_t fgkTPC;      // end TPC\r
 101   static const Float_t fgkTRD;      // end TRD\r
 102 \r
 103   // Track selection\r
 104   static const Float_t fgkTrkDCAxy; // cm\r
 105   static const Float_t fgkTrkDCAz;  // cm\r
 106   static const Int_t   fgkNclTPC;   // N clusters TPC\r
 107   static const Float_t fgkPt;       // min. pt\r
 108   static const Float_t fgkEta;      // eta range\r
 109   static AliTRDReconstructor   *fgReconstructor; // single instance of TRD reconstructor used by all tasks\r
 110   static AliTRDgeometry        *fgGeo;           // single instance of TRD geometry used by all tasks\r
 111 \r
 112   AliTRDinfoGen(const AliTRDinfoGen&);\r
 113   AliTRDinfoGen& operator=(const AliTRDinfoGen&);\r
 114   TTreeSRedirector* DebugStream();\r
 115 \r
 116   AliESDEvent      *fESDev;          //! ESD event\r
 117   AliMCEvent       *fMCev;           //! MC event\r
 118   // event/track cuts OO - to be used\r
 119   AliTRDeventCuts  *fEventCut;       // event cut\r
 120   AliESDtrackCuts  *fTrackCut;       // track cut\r
 121   AliTRDv0Info     *fV0Cut;          // v0 cut\r
 122   TString           fOCDB;           // OCDB location\r
 123   AliTRDtrackInfo  *fTrackInfo;      //! Track info\r
 124   AliTRDeventInfo  *fEventInfo;      //! Event info\r
 125   AliTRDv0Info     *fV0Info;         //! V0 info\r
 126   TObjArray        *fTracksBarrel;   //! Array of barrel tracks\r
 127   TObjArray        *fTracksSA;       //! Array of stand alone tracks\r
 128   TObjArray        *fTracksKink;     //! Array of kink tracks\r
 129   TObjArray        *fV0List;         //! V0 container\r
 130   TObjArray        *fContainer;      //! container to store results\r
 131   TObjArray        *fRecos;          //! array of reco params\r
 132   TTreeSRedirector *fDebugStream;    //! debug stream\r
 133 \r
 134   ClassDef(AliTRDinfoGen, 7)         // entry to TRD analysis train\r
 135 };\r
 136 #endif\r