Major dielectron framework update; includes "alignment" to updates in
[u/mrichter/AliRoot.git] / PWG3 / dielectron / AliDielectron.h
1 #ifndef ALIDIELECTRON_H
2 #define ALIDIELECTRON_H
3 /* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 //#####################################################
7 //#                                                   # 
8 //#             Class AliDielectron                   #
9 //#         Main Class for e+e- analysis              #
10 //#                                                   #
11 //#  by WooJin J. Park, GSI / W.J.Park@gsi.de         #
12 //#     Ionut C. Arsene, GSI / I.C.Arsene@gsi.de      #
13 //#     Magnus Mager, CERN / Magnus.Mager@cern.ch     #
14 //#     Jens Wiechula, Uni HD / Jens.Wiechula@cern.ch #
15 //#                                                   #
16 //#####################################################
17
18
19 #include <TNamed.h>
20 #include <TObjArray.h>
21
22 #include <AliAnalysisFilter.h>
23 #include <AliKFParticle.h>
24
25 #include "AliDielectronHistos.h"
26
27 class AliVEvent;
28 class THashList;
29 class AliDielectronCF;
30 class AliDielectronDebugTree;
31 class AliDielectronTrackRotator;
32 class AliDielectronPair;
33
34 //________________________________________________________________
35 class AliDielectron : public TNamed {
36   
37 public:
38   enum EPairType { kEv1PP=0, kEv1PM, kEv1MM,
39       kEv1PEv2P, kEv1MEv2P, kEv2PP,
40       kEv1PEv2M, kEv1MEv2M, kEv2PM,
41       kEv2MM, kEv1PMRot };
42   enum ELegType  { kEv1P, kEv1M, kEv2P, kEv2M };
43   
44   AliDielectron();
45   AliDielectron(const char* name, const char* title);
46   virtual ~AliDielectron();
47
48   void Init();
49   
50   void Process(AliVEvent *ev1, AliVEvent *ev2=0);
51
52   const AliAnalysisFilter& GetEventFilter() const { return fEventFilter; }
53   const AliAnalysisFilter& GetTrackFilter() const { return fTrackFilter; }
54   const AliAnalysisFilter& GetPairFilter()  const { return fPairFilter;  }
55
56   AliAnalysisFilter& GetEventFilter()       { return fEventFilter;       }
57   AliAnalysisFilter& GetTrackFilter()       { return fTrackFilter;       }
58   AliAnalysisFilter& GetPairFilter()        { return fPairFilter;        }
59   AliAnalysisFilter& GetPairPreFilter()     { return fPairPreFilter;     }
60   AliAnalysisFilter& GetPairPreFilterLegs() { return fPairPreFilterLegs; }
61   
62   void  SetMotherPdg( Int_t pdgMother ) { fPdgMother=pdgMother; }
63   void  SetLegPdg(Int_t pdgLeg1, Int_t pdgLeg2) { fPdgLeg1=pdgLeg1; fPdgLeg2=pdgLeg2; }
64   Int_t GetMotherPdg() const { return fPdgMother; }
65   Int_t GetLeg1Pdg()   const { return fPdgLeg1;   }
66   Int_t GetLeg2Pdg()   const { return fPdgLeg2;   }
67
68   void SetNoPairing(Bool_t noPairing=kTRUE) { fNoPairing=noPairing; }
69   const TObjArray* GetTrackArray(Int_t i) const {return (i>=0&&i<4)?&fTracks[i]:0;}
70   const TObjArray* GetPairArray(Int_t i)  const {return (i>=0&&i<10)?
71       static_cast<TObjArray*>(fPairCandidates->UncheckedAt(i)):0;}
72
73   TObjArray** GetPairArraysPointer() { return &fPairCandidates; }
74   
75   void SetHistogramManager(AliDielectronHistos * const histos) { fHistos=histos; }
76   const THashList * GetHistogramList() const { return fHistos?fHistos->GetHistogramList():0x0; }
77
78   Bool_t HasCandidates() const { return GetPairArray(1)?GetPairArray(1)->GetEntriesFast()>0:0; }
79
80   void SetCFManagerPair(AliDielectronCF * const cf) { fCfManagerPair=cf; }
81   AliDielectronCF* GetCFManagerPair() const { return fCfManagerPair; }
82
83   void SetPreFilterUnlikeOnly(Bool_t setValue=kTRUE){fPreFilterUnlikeOnly=setValue;};
84
85   void SetTrackRotator(AliDielectronTrackRotator * const rot) { fTrackRotator=rot; }
86   AliDielectronTrackRotator* GetTrackRotator() const { return fTrackRotator; }
87   
88   void SetDebugTree(AliDielectronDebugTree * const tree) { fDebugTree=tree; }
89   
90   static const char* TrackClassName(Int_t i) { return (i>=0&&i<4)?fgkTrackClassNames[i]:""; }
91   static const char* PairClassName(Int_t i)  { return (i>=0&&i<11)?fgkPairClassNames[i]:""; }
92
93   void SaveDebugTree();
94   
95 private:
96
97   
98   AliAnalysisFilter fEventFilter;    // Event cuts
99   AliAnalysisFilter fTrackFilter;    // leg cuts
100   AliAnalysisFilter fPairPreFilter;  // pair prefilter cuts
101   AliAnalysisFilter fPairPreFilterLegs; // Leg filter after the pair prefilter cuts
102   AliAnalysisFilter fPairFilter;     // pair cuts
103   
104   Int_t fPdgMother;     // pdg code of mother tracks
105   Int_t fPdgLeg1;       // pdg code leg1
106   Int_t fPdgLeg2;       // pdg code leg2
107
108   Bool_t fNoPairing;    // if to skip pairing, can be used for track QA only
109     
110   AliDielectronHistos *fHistos;   // Histogram manager
111                                   //  Streaming and merging should be handled
112                                   //  by the analysis framework
113  
114   TObjArray fTracks[4];           //! Selected track candidates
115                                   //  0: Event1, positive particles
116                                   //  1: Event1, negative particles
117                                   //  2: Event2, positive particles
118                                   //  3: Event2, negative particles
119
120   TObjArray *fPairCandidates;     //! Pair candidate arrays
121                                   //TODO: better way to store it? TClonesArray?
122
123   AliDielectronCF *fCfManagerPair;//Correction Framework Manager for the Pair
124   AliDielectronTrackRotator *fTrackRotator; //Track rotator
125   AliDielectronDebugTree *fDebugTree;  // Debug tree output
126
127   Bool_t fPreFilterUnlikeOnly;  //Apply PreFilter either in +- or to ++/--/+- individually
128
129   
130   void FillTrackArrays(AliVEvent * const ev, Int_t eventNr=0);
131   void PairPreFilter(Int_t arr1, Int_t arr2, TObjArray &arrTracks1, TObjArray &arrTracks2);
132   void FillPairArrays(Int_t arr1, Int_t arr2);
133   void FillPairArrayTR();
134   
135   Int_t GetPairIndex(Int_t arr1, Int_t arr2) const {return arr1>=arr2?arr1*(arr1+1)/2+arr2:arr2*(arr2+1)/2+arr1;}
136
137   void InitPairCandidateArrays();
138   void ClearArrays();
139   
140   TObjArray* PairArray(Int_t i);
141   
142   static const char* fgkTrackClassNames[4];   //Names for track arrays
143   static const char* fgkPairClassNames[11];   //Names for pair arrays
144
145   void ProcessMC();
146   
147   void  FillHistograms(const AliVEvent *ev);
148   void  FillHistogramsPair(AliDielectronPair *pair,Bool_t fromPreFilter=kFALSE);
149   void  FillHistogramsTracks(TObjArray **tracks);
150
151   void  FillDebugTree();
152   
153   AliDielectron(const AliDielectron &c);
154   AliDielectron &operator=(const AliDielectron &c);
155   
156   ClassDef(AliDielectron,4);
157 };
158
159 inline void AliDielectron::InitPairCandidateArrays()
160 {
161   //
162   // initialise all pair candidate arrays
163   //
164   fPairCandidates->SetOwner();
165   for (Int_t i=0;i<10;++i){
166     TObjArray *arr=new TObjArray;
167     fPairCandidates->AddAt(arr,i);
168     arr->SetOwner();
169   }
170 }
171
172 inline TObjArray* AliDielectron::PairArray(Int_t i)
173 {
174   //
175   // for internal use only: unchecked return of track array for fast access
176   //
177   return static_cast<TObjArray*>(fPairCandidates->UncheckedAt(i));
178 }
179
180 inline void AliDielectron::ClearArrays()
181 {
182   //
183   // Reset the Arrays
184   //
185   for (Int_t i=0;i<4;++i){
186     fTracks[i].Clear();
187   }
188   for (Int_t i=0;i<10;++i){
189     PairArray(i)->Delete();
190   }
191 }
192
193 #endif