DeltaP -> DeltaPvector
[u/mrichter/AliRoot.git] / HBTAN / AliHBTTwoTrackEffFctn.h
index 857ef62899033cee5e9ac2ec8d2d524bb9749bdb..81ba3ccd1f734ea68f53750d4ff7e10608bfc350 100644 (file)
@@ -15,7 +15,7 @@
 #include "AliHBTPair.h"
 #include "AliHBTFunction.h"
 
-class AliHBTTwoTrackEffFctn: public AliHBTOnePairFctn1D
+class AliHBTTwoTrackEffFctn: public AliHBTOnePairFctn1D, public AliHBTCorrelFunction
  {
   public:
     AliHBTTwoTrackEffFctn();
@@ -23,23 +23,39 @@ class AliHBTTwoTrackEffFctn: public AliHBTOnePairFctn1D
     virtual ~AliHBTTwoTrackEffFctn(){}
     TH1* GetResult();
   protected:
-    Double_t GetValue(AliHBTPair* pair){return pair->GetDeltaP();}
+    Double_t GetValue(AliHBTPair* pair) const {return pair->GetDeltaPvector();}
   private:
-    ClassDef(AliHBTTwoTrackEffFctn,1)
+    ClassDef(AliHBTTwoTrackEffFctn,2)
  };
+/******************************************************************/
 
-class AliHBTTwoTrackEffFctn3D: public AliHBTOnePairFctn3D
+class AliHBTTwoTrackEffFctnPxPyPz: public AliHBTOnePairFctn3D, public AliHBTCorrelFunction
  {
   public:
-    AliHBTTwoTrackEffFctn3D();
-    virtual ~AliHBTTwoTrackEffFctn3D(){}
-
-    void ProcessSameEventParticles(AliHBTPair* /*pair*/){}
-    void ProcessDiffEventParticles(AliHBTPair* /*pair*/){}
+    AliHBTTwoTrackEffFctnPxPyPz(Int_t nXbins = 100, Double_t maxXval = 0.15, Double_t minXval = 0.0,
+                                Int_t nYbins = 100, Double_t maxYval = 0.15, Double_t minYval = 0.0,
+                   Int_t nZbins = 100, Double_t maxZval = 0.15, Double_t minZval = 0.0);
+    virtual ~AliHBTTwoTrackEffFctnPxPyPz(){}
+    TH1* GetResult();
+  protected:
+    void GetValues(AliHBTPair* pair,Double_t& x, Double_t& y,Double_t& z) const;
+  private:
+    ClassDef(AliHBTTwoTrackEffFctnPxPyPz,2)
+ };
+/******************************************************************/
 
+class AliHBTTwoTrackEffFctnPtThetaPhi: public AliHBTOnePairFctn3D, public AliHBTCorrelFunction
+ {
+  public:
+    AliHBTTwoTrackEffFctnPtThetaPhi(Int_t nXbins = 100, Double_t maxXval = 0.15, Double_t minXval = 0.0,
+                                    Int_t nYbins = 100, Double_t maxYval = 0.3, Double_t minYval = 0.0,
+                       Int_t nZbins = 100, Double_t maxZval = 0.3, Double_t minZval = 0.0);
+    virtual ~AliHBTTwoTrackEffFctnPtThetaPhi(){}
+    TH1* GetResult();
   protected:
-    void GetValues(AliHBTPair*,Double_t&, Double_t&,Double_t&);
+    void GetValues(AliHBTPair* pair,Double_t& x, Double_t& y,Double_t& z) const;
   private:
-    ClassDef(AliHBTTwoTrackEffFctn3D,1)
+    ClassDef(AliHBTTwoTrackEffFctnPtThetaPhi,1)
  };
+
 #endif