Read digits via TBranch::GetEntry() instead of TTree::GetEntry() as SPD/SDD/SSD entri...

[u/mrichter/AliRoot.git] / HBTAN / AliHBTQDistributionFctns.h

diff --git a/HBTAN/AliHBTQDistributionFctns.h b/HBTAN/AliHBTQDistributionFctns.h
index 7abda369fd218c8d15ab3d46ce5af1cf225786ab..cc803fb7ce56ad4efb51b5b224d502b60a1cd911 100644 (file)
--- a/HBTAN/AliHBTQDistributionFctns.h
+++ b/HBTAN/AliHBTQDistributionFctns.h
@@ -1,18 +1,33 @@
 #ifndef ALIHBTQOUTDISTRIBUTIONVSKTLFCTN_H
 #define ALIHBTQOUTDISTRIBUTIONVSKTLFCTN_H
 
+/////////////////////////////////////////////////////////////////////////////
+// 
+// class AliHBTQInvDistributionVsKtFctn;    //QInvLCMS   Distribution Vs   Kt
+// class AliHBTQOutDistributionVsKtFctn;    //QOutLCMS   Distribution Vs   Kt
+// class AliHBTQSideDistributionVsKtFctn;   //QSideLCMS  Distribution Vs   Kt
+// class AliHBTQLongDistributionVsKtFctn;   //QLongLCMS  Distribution Vs   Kt
+
+// class AliHBTQOutDistributionVsQInvFctn;    //QOutLCMS   Distribution Vs   QInv
+// class AliHBTQSideDistributionVsQInvFctn;   //QSideLCMS  Distribution Vs   QInv
+// class AliHBTQLongDistributionVsQInvFctn;   //QLongLCMS  Distribution Vs   QInv
+// class AliHBTPtDiffDistributionVsQInvFctn;
+//
 // added by Zbigniew.Chajecki@cern.ch
 // this classes create distribution functions of pair momentum 
+//
+/////////////////////////////////////////////////////////////////////////////
 
-class AliHBTQInvDistributionVsKtFctn;    //QInvCMSLC   Distribution Vs   Kt
-class AliHBTQOutDistributionVsKtFctn;    //QOutCMSLC   Distribution Vs   Kt
-class AliHBTQSideDistributionVsKtFctn;   //QSideCMSLC  Distribution Vs   Kt
-class AliHBTQLongDistributionVsKtFctn;   //QLongCMSLC  Distribution Vs   Kt
-
-class AliHBTQOutDistributionVsQInvFctn;    //QOutCMSLC   Distribution Vs   QInv
-class AliHBTQSideDistributionVsQInvFctn;   //QSideCMSLC  Distribution Vs   QInv
-class AliHBTQLongDistributionVsQInvFctn;   //QLongCMSLC  Distribution Vs   QInv
+class AliHBTQInvDistributionVsKtFctn;    //QInvLCMS   Distribution Vs   Kt
+class AliHBTQOutDistributionVsKtFctn;    //QOutLCMS   Distribution Vs   Kt
+class AliHBTQSideDistributionVsKtFctn;   //QSideLCMS  Distribution Vs   Kt
+class AliHBTQLongDistributionVsKtFctn;   //QLongLCMS  Distribution Vs   Kt
 
+class AliHBTQOutDistributionVsQInvFctn;    //QOutLCMS   Distribution Vs   QInv
+class AliHBTQSideDistributionVsQInvFctn;   //QSideLCMS  Distribution Vs   QInv
+class AliHBTQLongDistributionVsQInvFctn;   //QLongLCMS  Distribution Vs   QInv
+class AliHBTPtDiffDistributionVsQInvFctn;
+class AliHBTRStarDistribution;
 #include "AliHBTFunction.h"
 
 /***********************************************************************/
@@ -23,15 +38,13 @@ class AliHBTQOutDistributionVsKtFctn: public AliHBTOnePairFctn2D
    AliHBTQOutDistributionVsKtFctn(Int_t nXbins = 200, Double_t maxXval = 1., Double_t minXval = 0.0, 
                              Int_t nYbins = 500, Double_t maxYval = .15, Double_t minYval =-0.15);
    virtual ~AliHBTQOutDistributionVsKtFctn(){}
-   TH1* GetResult(){return GetNumerator();}
-   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y)
+   TH1* GetResult();
+   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y) const
     {
-     y = partpair->GetQOutCMSLC();
+     y = partpair->GetQOutLCMS();
      x = partpair->GetKt();
     }
   protected:
-  private:
-  public:
     ClassDef(AliHBTQOutDistributionVsKtFctn,1)
  };
 /***********************************************************************/
@@ -42,15 +55,13 @@ class AliHBTQSideDistributionVsKtFctn: public AliHBTOnePairFctn2D
    AliHBTQSideDistributionVsKtFctn(Int_t nXbins = 200, Double_t maxXval = 1.2, Double_t minXval = -0.1, 
                              Int_t nYbins = 500, Double_t maxYval = 1.2, Double_t minYval =-1.2);
    virtual ~AliHBTQSideDistributionVsKtFctn(){}
-   TH1* GetResult(){return GetNumerator();}
-   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y)
+   TH1* GetResult();
+   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y) const
     {
-     y = partpair->GetQSideCMSLC();
+     y = partpair->GetQSideLCMS();
      x = partpair->GetKt();
     }
   protected:
-  private:
-  public:
     ClassDef(AliHBTQSideDistributionVsKtFctn,1)
  };
 /***********************************************************************/
@@ -62,15 +73,13 @@ class AliHBTQLongDistributionVsKtFctn: public AliHBTOnePairFctn2D
    AliHBTQLongDistributionVsKtFctn(Int_t nXbins = 200, Double_t maxXval = 1.2, Double_t minXval = -0.1, 
                              Int_t nYbins = 500, Double_t maxYval = 1.2, Double_t minYval =-1.2);
    virtual ~AliHBTQLongDistributionVsKtFctn(){}
-   TH1* GetResult(){return GetNumerator();}
-   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y)
+   TH1* GetResult();
+   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y) const
     {
-     y = partpair->GetQLongCMSLC();
+     y = partpair->GetQLongLCMS();
      x = partpair->GetKt();
     }
   protected:
-  private:
-  public:
     ClassDef(AliHBTQLongDistributionVsKtFctn,1)
  };
 /***********************************************************************/
@@ -82,15 +91,13 @@ class AliHBTQInvDistributionVsKtFctn: public AliHBTOnePairFctn2D
    AliHBTQInvDistributionVsKtFctn(Int_t nXbins = 200, Double_t maxXval = 1.2, Double_t minXval = -0.1, 
                              Int_t nYbins = 500, Double_t maxYval = 1.2, Double_t minYval =-1.2);
    virtual ~AliHBTQInvDistributionVsKtFctn(){}
-   TH1* GetResult(){return GetNumerator();}
-   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y)
+   TH1* GetResult();
+   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y) const
     {
      y = partpair->GetQInv();
      x = partpair->GetKt();
     }
   protected:
-  private:
-  public:
     ClassDef(AliHBTQInvDistributionVsKtFctn,1)
  };
 
@@ -102,15 +109,13 @@ class AliHBTQOutDistributionVsQInvFctn: public AliHBTOnePairFctn2D
    AliHBTQOutDistributionVsQInvFctn(Int_t nXbins = 200, Double_t maxXval = 1., Double_t minXval = 0.0, 
                              Int_t nYbins = 500, Double_t maxYval = .15, Double_t minYval =-0.15);
    virtual ~AliHBTQOutDistributionVsQInvFctn(){}
-   TH1* GetResult(){return GetNumerator();}
-   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y)
+   TH1* GetResult();
+   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y) const
     {
-     y = partpair->GetQOutCMSLC();
+     y = partpair->GetQOutLCMS();
      x = partpair->GetQInv();
     }
   protected:
-  private:
-  public:
     ClassDef(AliHBTQOutDistributionVsQInvFctn,1)
  };
 /***********************************************************************/
@@ -121,15 +126,13 @@ class AliHBTQSideDistributionVsQInvFctn: public AliHBTOnePairFctn2D
    AliHBTQSideDistributionVsQInvFctn(Int_t nXbins = 200, Double_t maxXval = 1.2, Double_t minXval = -0.1, 
                              Int_t nYbins = 500, Double_t maxYval = 1.2, Double_t minYval =-1.2);
    virtual ~AliHBTQSideDistributionVsQInvFctn(){}
-   TH1* GetResult(){return GetNumerator();}
-   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y)
+   TH1* GetResult();
+   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y) const
     {
-     y = partpair->GetQSideCMSLC();
+     y = partpair->GetQSideLCMS();
      x = partpair->GetQInv();
     }
   protected:
-  private:
-  public:
     ClassDef(AliHBTQSideDistributionVsQInvFctn,1)
  };
 /***********************************************************************/
@@ -141,18 +144,68 @@ class AliHBTQLongDistributionVsQInvFctn: public AliHBTOnePairFctn2D
    AliHBTQLongDistributionVsQInvFctn(Int_t nXbins = 200, Double_t maxXval = 1.2, Double_t minXval = -0.1, 
                              Int_t nYbins = 500, Double_t maxYval = 1.2, Double_t minYval =-1.2);
    virtual ~AliHBTQLongDistributionVsQInvFctn(){}
-   TH1* GetResult(){return GetNumerator();}
-   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y)
+   TH1* GetResult();
+   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y) const
     {
-     y = partpair->GetQLongCMSLC();
+     y = partpair->GetQLongLCMS();
      x = partpair->GetQInv();
     }
   protected:
-  private:
-  public:
     ClassDef(AliHBTQLongDistributionVsQInvFctn,1)
  };
 /***********************************************************************/
 /***********************************************************************/
+class AliHBTPtDiffDistributionVsQInvFctn: public AliHBTOnePairFctn2D
+ {
+  public: 
+   AliHBTPtDiffDistributionVsQInvFctn(Int_t nXbins = 800, Double_t maxXval = 4.0, Double_t minXval = 0., 
+                             Int_t nYbins = 500, Double_t maxYval = 0.1, Double_t minYval =-0.1);
+   virtual ~AliHBTPtDiffDistributionVsQInvFctn(){}
+   TH1* GetResult();
+   void GetValues(AliHBTPair* partpair, Double_t& x, Double_t& y) const
+    {
+     y = partpair->Particle1()->Pt() - partpair->Particle2()->Pt();
+     x = partpair->GetQInv();
+    }
+  protected:
+    ClassDef(AliHBTPtDiffDistributionVsQInvFctn,1)
+ };
+/***********************************************************************/
+/***********************************************************************/
+
+class AliHBTRStarDistribution: public AliHBTOnePairFctn1D
+{
+  public:
+    AliHBTRStarDistribution(Int_t nXbins = 500, Double_t maxXval = 5e-11, Double_t minXval = 0.);
+    virtual ~AliHBTRStarDistribution(){}
+    TH1* GetResult();
+  protected:
+    Double_t GetValue(AliHBTPair* partpair) const
+    {
+      return partpair->GetRStar();
+    }
+   
+  private:
+   ClassDef(AliHBTRStarDistribution,1)
+};
+
+/***********************************************************************/
+/***********************************************************************/
 
+class AliHBTRDistribution: public AliHBTOnePairFctn1D
+{
+  public:
+    AliHBTRDistribution(Int_t nXbins = 500, Double_t maxXval = 5e-11, Double_t minXval = 0.);
+    virtual ~AliHBTRDistribution(){}
+    TH1* GetResult();
+  protected:
+    Double_t GetValue(AliHBTPair* partpair) const
+    {
+      return partpair->GetR();
+    }
+   
+  private:
+   ClassDef(AliHBTRDistribution,1)
+};
 #endif
+