set reco param on an event by event basis
[u/mrichter/AliRoot.git] / 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    ,kUseLocalEvSelection  = BIT(19)\r
37    ,kUseLocalTrkSelection = BIT(20)\r
38    ,kCollision            = BIT(21)\r
39    ,kOCDB                 = BIT(22)\r
40   };\r
41   enum AliTRDinfoGenObjects{\r
42     kTracksESD     =  0\r
43     ,kTracksMC     =  1\r
44     ,kV0           =  2\r
45     ,kTPC          =  3\r
46     ,kTRDin        =  4\r
47     ,kTRDout       =  5\r
48     ,kBarrel       =  6\r
49     ,kBarrelMC     =  7\r
50     ,kSA           =  8\r
51     ,kSAMC         =  9\r
52     ,kKink         = 10\r
53     ,kKinkMC       = 11\r
54     ,kBarrelFriend = 12\r
55     ,kSAFriend     = 13\r
56     ,kNObjects     = 14\r
57   };\r
58 \r
59   AliTRDinfoGen();\r
60   AliTRDinfoGen(char* name);\r
61   virtual ~AliTRDinfoGen();\r
62   \r
63   static Float_t GetEndITS() { return fgkITS;}\r
64   static Float_t GetEndTPC() { return fgkTPC;}\r
65   static Float_t GetEndTRD() { return fgkTRD;}\r
66   Int_t   GetNRefFigures() const  { return 1;} \r
67   const char* GetOCDB() const {return fOCDB.Data();}\r
68   Bool_t  GetRefFigure(Int_t ifig);\r
69   Bool_t  Load(const Char_t *fn="AnalysisResults.root", const Char_t *dir="TRD_Performance", const Char_t *name=NULL);\r
70 \r
71   Bool_t  HasMCdata() const       { return TestBit(kMCdata);};\r
72   // temporary until check with AliAnalysisTaskSE collision selection mechanism\r
73   Bool_t  IsInitOCDB() const {return TestBit(kOCDB);}\r
74   Bool_t  IsCollision() const {return TestBit(kCollision);}\r
75   static const AliTRDReconstructor* Reconstructor() {return fgReconstructor;}\r
76   static AliTRDgeometry*      Geometry() {return fgGeo;}\r
77   void    SetInitOCDB(Bool_t set=kTRUE) {SetBit(kOCDB, set);}\r
78   void    SetCollision(Bool_t set=kTRUE) {SetBit(kCollision, set);}\r
79   //void    SetLocalEvSelection(const AliTRDeventCuts */*cut*/){;} \r
80   void    SetLocalEvSelection(Bool_t use=kTRUE) {SetBit(kUseLocalEvSelection, use);}\r
81   //void    SetLocalTrkSelection(const AliESDtrackCuts */*cut*/){;} \r
82   void    SetLocalTrkSelection(Bool_t use=kTRUE) {SetBit(kUseLocalTrkSelection, use);}\r
83   void    SetLocalV0Selection(const AliTRDv0Info *v0);\r
84   void    SetMCdata(Bool_t mc = kTRUE) {SetBit(kMCdata, mc);}\r
85   void    SetOCDB(const char *ocdb) {fOCDB=ocdb;}\r
86   void    SetTrigger(const Char_t *trigger);\r
87 \r
88   Bool_t  UseLocalEvSelection() const {return TestBit(kUseLocalEvSelection);}\r
89   Bool_t  UseLocalTrkSelection() const {return TestBit(kUseLocalTrkSelection);}\r
90   void    UserCreateOutputObjects();\r
91   void    UserExec(Option_t *);\r
92 \r
93 private:\r
94   // rough radial limits for TRD\r
95   static const Float_t fgkITS;      // end ITS\r
96   static const Float_t fgkTPC;      // end TPC\r
97   static const Float_t fgkTRD;      // end TRD\r
98 \r
99   // Trigger selection\r
100   TString              *fEvTrigger; // list of trigger classes separated by space\r
101   // Vertex selection\r
102   static const Float_t fgkEvVertexZ;// cm\r
103   static const Int_t   fgkEvVertexN;// cm\r
104   // Track selection\r
105   static const Float_t fgkTrkDCAxy; // cm\r
106   static const Float_t fgkTrkDCAz;  // cm\r
107   static const Int_t   fgkNclTPC;   // N clusters TPC\r
108   static const Float_t fgkPt;       // min. pt\r
109   static const Float_t fgkEta;      // eta range\r
110   static AliTRDReconstructor   *fgReconstructor; // single instance of TRD reconstructor used by all tasks\r
111   static AliTRDgeometry        *fgGeo;           // single instance of TRD geometry used by all tasks\r
112 \r
113   AliTRDinfoGen(const AliTRDinfoGen&);\r
114   AliTRDinfoGen& operator=(const AliTRDinfoGen&);\r
115   TTreeSRedirector* DebugStream();\r
116 \r
117   AliESDEvent      *fESDev;          //! ESD event\r
118   AliMCEvent       *fMCev;           //! MC event\r
119   // event/track cuts OO - to be used\r
120   AliTRDeventCuts  *fEventCut;       // event cut\r
121   AliESDtrackCuts  *fTrackCut;       // track cut\r
122   AliTRDv0Info     *fV0Cut;          // v0 cut\r
123   TString           fOCDB;           // OCDB location\r
124   AliTRDtrackInfo  *fTrackInfo;      //! Track info\r
125   AliTRDeventInfo  *fEventInfo;      //! Event info\r
126   AliTRDv0Info     *fV0Info;         //! V0 info\r
127   TObjArray        *fTracksBarrel;   //! Array of barrel tracks\r
128   TObjArray        *fTracksSA;       //! Array of stand alone tracks\r
129   TObjArray        *fTracksKink;     //! Array of kink tracks\r
130   TObjArray        *fV0List;         //! V0 container\r
131   TObjArray        *fContainer;      //! container to store results\r
132   TObjArray        *fRecos;          //! array of reco params\r
133   TTreeSRedirector *fDebugStream;    //! debug stream\r
134 \r
135   ClassDef(AliTRDinfoGen, 7)         // entry to TRD analysis train\r
136 };\r
137 #endif\r