PWGCF/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