]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWG2/FEMTOSCOPY/AliFemtoUser/AliFemtoModelGausRinvFreezeOutGenerator.h
Give the ability to guess if particle is primary
[u/mrichter/AliRoot.git] / PWG2 / FEMTOSCOPY / AliFemtoUser / AliFemtoModelGausRinvFreezeOutGenerator.h
1 ////////////////////////////////////////////////////////////////////////////////
2 ///                                                                          ///
3 /// AliFemtoModelGausRinvFreezeOutGenerator - freeze-out                     ///
4 /// coordinates generator, generating a 3D gaussian spheroid in PRF          ///
5 /// Authors: Adam Kisiel kisiel@mps.ohio-state.edu                           ///
6 ///                                                                          ///
7 ////////////////////////////////////////////////////////////////////////////////
8 #ifndef AliFemtoModelGausRinvFreezeOutGenerator_hh
9 #define AliFemtoModelGausRinvFreezeOutGenerator_hh
10
11 #include "AliFemtoModelFreezeOutGenerator.h"
12
13 #include "TRandom.h"
14
15 class AliFemtoModelGausRinvFreezeOutGenerator : public AliFemtoModelFreezeOutGenerator
16 {
17  public:
18   AliFemtoModelGausRinvFreezeOutGenerator();
19   AliFemtoModelGausRinvFreezeOutGenerator(const AliFemtoModelGausRinvFreezeOutGenerator &aModel);
20   virtual ~AliFemtoModelGausRinvFreezeOutGenerator();
21   virtual void GenerateFreezeOut(AliFemtoPair *aPair);
22
23   void SetSelectPrimaryFromHidden(bool aUse);
24   Bool_t GetSelectPrimaryFromHidden();
25
26   void SetSizeInv(Double_t aSizeInv);
27   
28   Double_t GetSizeInv() const;
29
30   virtual AliFemtoModelFreezeOutGenerator* Clone() const;
31
32  protected:
33   Double_t fSizeInv;        // Size of the source
34   Bool_t fSelectPrimary;    // If set to true, the existing hidden info is assumed
35                             // to contain the particle creation point (in cm)
36                             // and the model will try to guess whether the particle
37                             // is primary based on that and assign creation point
38                             // only for primary particles
39
40  private:
41   AliFemtoModelFreezeOutGenerator* GetGenerator() const;
42                 
43 #ifdef __ROOT__
44   ClassDef(AliFemtoModelGausRinvFreezeOutGenerator, 1)
45 #endif
46
47     };
48   
49 #endif
50
51