Add Reaction Plane aware analysis
[u/mrichter/AliRoot.git] / PWG2 / FEMTOSCOPY / AliFemtoUser / AliFemtoModelCorrFctnSource.h
1 ////////////////////////////////////////////////////////////////////////////////
2 ///                                                                          ///
3 /// AliFemtoModelCorrFctnSource - the class for correlation function which   ///
4 /// uses the model framework and weight generation and saves the generated   ///
5 /// emission source                                                          ///
6 /// Authors: Adam Kisiel, kisiel@mps.ohio-state.edu                          ///
7 ///                                                                          ///
8 ////////////////////////////////////////////////////////////////////////////////
9 #ifndef ALIFEMTOMODELCORRFCTNSOURCE_H
10 #define ALIFEMTOMODELCORRFCTNSOURCE_H
11
12 #include "TH2D.h"
13 #include "AliFemtoCorrFctn.h"
14 #include "AliFemtoPair.h"
15 #include "AliFemtoModelManager.h"
16 #include "AliFemtoModelCorrFctn.h"
17
18 class AliFemtoModelCorrFctnSource: public AliFemtoModelCorrFctn {
19
20 public:
21   AliFemtoModelCorrFctnSource();
22   AliFemtoModelCorrFctnSource(const char *title, Int_t aNbins, Double_t aQinvLo, Double_t aQinvHi);
23   AliFemtoModelCorrFctnSource(const AliFemtoModelCorrFctnSource& aCorrFctn);
24   virtual ~AliFemtoModelCorrFctnSource();
25   
26   AliFemtoModelCorrFctnSource& operator=(const AliFemtoModelCorrFctnSource& aCorrFctn);
27
28   virtual AliFemtoString Report();
29
30   virtual void AddRealPair(AliFemtoPair* aPair);
31   virtual void AddMixedPair(AliFemtoPair* aPir);
32
33   virtual void Write();
34   virtual TList* GetOutputList();
35
36   virtual AliFemtoModelCorrFctn* Clone();
37
38   void SetUseRPSelection(unsigned short aRPSel);
39 protected:
40
41   TH1D *fHistROut;     // Distribution of Rout
42   TH1D *fHistRSide;    // Distribution of Rside
43   TH1D *fHistRLong;    // Distribution of Rlong
44   TH1D *fHistRStar;    // Distribution of RStar
45   TH1D *fHistdNdR;     // Distribution of RStar weighted by Jacobian 
46   TH2D *fHistNumWS;    // Weight spread for numerator
47   TH2D *fHistDenWS;    // Weight spread for denominator
48
49 private:
50
51   unsigned short fUseRPSelection;  // The pair cut uses RP selection
52
53 #ifdef __ROOT__
54   ClassDef(AliFemtoModelCorrFctnSource, 1)
55 #endif
56 };
57
58 #endif