Bug correction
[u/mrichter/AliRoot.git] / HBTAN / AliHBTPIDPurityFctns.h
1 #ifndef ALIHBTPIDPURITYFCTNS_H
2 #define ALIHBTPIDPURITYFCTNS_H
3 //_______________________________________________________________________________
4 /////////////////////////////////////////////////////////////////////////////////
5 //
6 // class AliHBTMonPIDPurityVsPtFctn;
7 // class AliHBTMonThetaResolutionVsPtFctn;
8 //
9 // file: AliHBTPIDPurityFctns.cxx AliHBTPIDPurityFctns.h
10 //
11 // Caution: On 2D plots on X axis in simulated values
12 // That is contrary to two-particle resolutions where it is reconstructed one
13 //
14 // added by Piotr.Skowronski@cern.ch
15 // 
16 //
17 //////////////////////////////////////////////////////////////////////////////////
18
19 #include "AliHBTFunction.h"
20 #include "AliHBTMonitorFunction.h"
21
22 class AliHBTMonPIDPurityVsPtFctn: public AliHBTMonTwoParticleFctn1D, public AliHBTCorrelFunction
23 {
24   public: 
25     AliHBTMonPIDPurityVsPtFctn(Int_t nbins = 20, Double_t maxXval = 2.0, Double_t minXval = 0.0);
26     virtual ~AliHBTMonPIDPurityVsPtFctn();
27     void Init();
28     void Write();
29     void Rename(const Char_t * name);
30     void Rename(const Char_t * name, const Char_t * title);
31     TH1* GetResult();
32     Double_t GetValue(AliHBTParticle * /*track*/,AliHBTParticle * /*part*/) { return 0.0; }
33     void Process(AliHBTParticle * track,AliHBTParticle * part);
34   protected:
35     TH1D* fGood;
36     TH1D* fAll;
37     ClassDef(AliHBTMonPIDPurityVsPtFctn,1)
38 };
39 /***********************************************************************/
40
41 class AliHBTMonPIDContaminationVsPtFctn: public AliHBTMonTwoParticleFctn1D, public AliHBTCorrelFunction
42 {
43   public: 
44     AliHBTMonPIDContaminationVsPtFctn(Int_t nbins = 20, Double_t maxXval = 2.0, Double_t minXval = 0.0);
45     virtual ~AliHBTMonPIDContaminationVsPtFctn();
46     void Init();
47     void Write();
48     void Rename(const Char_t * name);
49     void Rename(const Char_t * name, const Char_t * title);
50     TH1* GetResult();
51     Double_t GetValue(AliHBTParticle * /*track*/,AliHBTParticle * /*part*/) { return 0.0; }
52     void Process(AliHBTParticle * track,AliHBTParticle * part);
53   protected:
54     TH1D* fWrong;  
55     TH1D* fAll;
56     ClassDef(AliHBTMonPIDContaminationVsPtFctn,1)
57 };
58 /*************************************************************************************/
59
60 class AliHBTPairPIDProbVsQInvFctn: public AliHBTOnePairFctn1D, public AliHBTCorrelFunction
61 {
62 //Q Invaraint Correlation Function
63 //1D two particle function
64  public:
65    AliHBTPairPIDProbVsQInvFctn(Int_t nbins = 100, Double_t maxXval = 0.15, Double_t minXval = 0.0);
66    virtual ~AliHBTPairPIDProbVsQInvFctn(){};
67    void ProcessSameEventParticles(AliHBTPair* pair);
68    void ProcessDiffEventParticles(AliHBTPair* pair);
69    TH1* GetResult();
70  protected:
71    Double_t GetValue(AliHBTPair * pair){return pair->GetQInv();}
72  private:
73    ClassDef(AliHBTPairPIDProbVsQInvFctn,1)
74 };
75 /*************************************************************************************/
76 class AliHBTPairPIDProbVsQOutSQideQLongFctn: public AliHBTOnePairFctn3D, public AliHBTCorrelFunction
77 {
78
79   public:
80     AliHBTPairPIDProbVsQOutSQideQLongFctn(Int_t nXbins = 100, Double_t maxXval = 0.15, Double_t minXval = 0.0,
81                              Int_t nYbins = 100, Double_t maxYval = 0.15, Double_t minYval = 0.0,
82                  Int_t nZbins = 100, Double_t maxZval = 0.15, Double_t minZval = 0.0);
83     virtual  ~AliHBTPairPIDProbVsQOutSQideQLongFctn(){}
84
85     TH1* GetResult();
86     void ProcessSameEventParticles(AliHBTPair* part);
87     void ProcessDiffEventParticles(AliHBTPair* pair);
88
89   protected:
90     void GetValues(AliHBTPair* /*pair*/, Double_t& /*x*/, Double_t& /*y*/, Double_t& /*z*/){}
91
92     ClassDef(AliHBTPairPIDProbVsQOutSQideQLongFctn,1)
93 };
94
95
96 #endif