ffda7dd70201355eb7953d3bc02b5e22111805c5
[u/mrichter/AliRoot.git] / HBTAN / AliHBTAnalysis.h
1 #ifndef ALIHBTANALYSIS_H
2 #define ALIHBTANALYSIS_H
3 //_________________________________________________________
4 ////////////////////////////////////////////////////////////////////////////
5 //
6 // class AliHBTAnalysis
7 //
8 // Central Object Of HBTAnalyser: 
9 // This class performs main looping within HBT Analysis
10 // User must plug a reader of Type AliHBTReader
11 // User plugs in coorelation and monitor functions
12 // as well as monitor functions
13 //
14 // HBT Analysis Tool, which is integral part of AliRoot,
15 // ALICE Off-Line framework:
16 //
17 // Piotr.Skowronski@cern.ch
18 // more info: http://alisoft.cern.ch/people/skowron/analyzer/index.html
19 //
20 ////////////////////////////////////////////////////////////////////////////
21 //_________________________________________________________
22
23 #include <TObject.h>
24
25 class AliHBTParticleCut;
26 class AliHBTCut;
27 class AliHBTPairCut;
28 class AliHBTPair;
29
30 class AliHBTRun;
31 class AliHBTEvent;
32 class AliHBTReader;
33 class AliHBTOnePairFctn;      
34 class AliHBTTwoPairFctn;
35
36 class AliHBTMonOneParticleFctn;
37 class AliHBTMonTwoParticleFctn;
38
39 class TList;
40
41 class AliHBTAnalysis: public TObject
42  {
43    public:
44      AliHBTAnalysis();
45      AliHBTAnalysis(const AliHBTAnalysis& in);
46      const AliHBTAnalysis& operator=(const AliHBTAnalysis& /*right*/);
47      virtual ~AliHBTAnalysis();
48
49      virtual void Process(Option_t* option = "TracksAndParticles");
50      
51      void SetGlobalPairCut(AliHBTPairCut* cut);
52      
53      void AddTrackFunction(AliHBTOnePairFctn* f);
54      void AddParticleFunction(AliHBTOnePairFctn* f);
55      void AddParticleAndTrackFunction(AliHBTTwoPairFctn* f);
56      
57      void AddParticleMonitorFunction(AliHBTMonOneParticleFctn* f);    //z.ch.
58      void AddTrackMonitorFunction(AliHBTMonOneParticleFctn* f);    //z.ch.
59      void AddParticleAndTrackMonitorFunction(AliHBTMonTwoParticleFctn* f);//z.ch.
60
61      void AddResolutionFunction(AliHBTTwoPairFctn* f){AddParticleAndTrackFunction(f);}
62      
63      void SetReader(AliHBTReader* r){fReader = r;}
64      
65      void WriteFunctions();
66      
67      void SetBufferSize(Int_t buffsize){fBufferSize=buffsize;}
68      void SetOwner(Bool_t owner=kTRUE){fIsOwner=owner;}
69      Bool_t IsOwner() const {return fIsOwner;}
70      Bool_t IsNonIdentAnalysis();
71      void   Init();
72      void   ResetFunctions();
73      void   SetDisplayInfo(Int_t howoften){fDisplayMixingInfo = howoften;}//defines every each line info about mixing is displayed
74    protected:
75      
76      Bool_t RunCoherencyCheck();
77      
78      void FilterOut(AliHBTEvent* outpart1, AliHBTEvent* outpart2, AliHBTEvent* inpart,
79                     AliHBTEvent* outtrack1, AliHBTEvent* outtrack2, AliHBTEvent* intrack);
80      void FilterOut(AliHBTEvent* out1, AliHBTEvent* out2, AliHBTEvent* in);
81      void DeleteFunctions();
82      
83      virtual void ProcessTracks();
84      virtual void ProcessParticles();
85      virtual void ProcessTracksAndParticles();
86      
87      virtual void ProcessTracksAndParticlesNonIdentAnal();
88      virtual void ProcessParticlesNonIdentAnal();
89      virtual void ProcessTracksNonIdentAnal();
90
91      AliHBTReader* fReader;//! Pointer to reader
92      
93      UInt_t fNTrackFunctions; //! Number of Tracks functions 
94      UInt_t fNParticleFunctions; //! Number of particles functions
95      UInt_t fNParticleAndTrackFunctions; //! Number of resolution functions
96                 
97      UInt_t fNTrackMonitorFunctions; //! Number of Track Monitor functions 
98      UInt_t fNParticleMonitorFunctions; //! Number of Particles Monitor functions 
99      UInt_t fNParticleAndTrackMonitorFunctions; //! Number of Resolution Monitor functions 
100
101      AliHBTOnePairFctn**  fTrackFunctions; //!array of pointers to functions that analyze rekonstructed tracks
102      AliHBTOnePairFctn**  fParticleFunctions; //!array of pointers to functions that analyze generated particles
103      AliHBTTwoPairFctn**  fParticleAndTrackFunctions; //!array of pointers to functions that analyze both 
104                                         //reconstructed tracks and generated particles
105                 //i.e. - resolution analyzers
106      AliHBTMonOneParticleFctn**  fParticleMonitorFunctions; //! array of pointers to monitoring functions
107      AliHBTMonOneParticleFctn**  fTrackMonitorFunctions; //! which are used for single particle analysis,
108      AliHBTMonTwoParticleFctn**  fParticleAndTrackMonitorFunctions;  //! cut monitoring, etc.
109
110
111      /**********************************************/
112      /* Control parameters  */
113      /**********************************************/
114
115      AliHBTPairCut *fPairCut;//! Pair cut applied for all mixed particles
116       
117      Int_t  fBufferSize; //!defines the size of buffer for mixed events; -1==MIX All
118      Int_t  fDisplayMixingInfo;//!defines every which particle mixing info is displayed
119      Bool_t fIsOwner;//!defines of all functions are supposed to be deleted while by the way of analysis defaulr false
120
121    private:
122      static const UInt_t fgkFctnArraySize;//!
123      static const UInt_t fgkDefaultMixingInfo;//!
124      static const Int_t  fgkDefaultBufferSize;//!
125
126      ClassDef(AliHBTAnalysis,0)
127  };
128 #endif