code cleaning

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

diff --git a/HBTAN/AliHBTPair.h b/HBTAN/AliHBTPair.h
index 1d87ec9b80cdfdca7ba48f1be09f1148791b4e52..ebd73b1eba3d8b4b6e07514184430fd47426ffe4 100644 (file)
--- a/HBTAN/AliHBTPair.h
+++ b/HBTAN/AliHBTPair.h
@@ -13,17 +13,21 @@ class AliHBTPair: public TObject
 {
  public:
    AliHBTPair(Bool_t rev = kFALSE); //contructor
-   ~AliHBTPair(){}
+   AliHBTPair(AliHBTParticle* part1, AliHBTParticle* part2, Bool_t rev = kFALSE); //contructor
+   virtual ~AliHBTPair(){}
    void SetParticles(AliHBTParticle*,AliHBTParticle*); //sets particles in the pair
    AliHBTPair* GetSwapedPair() {return fSwapedPair;} //returns pair with swapped particles
    
    AliHBTParticle* Particle1() const {return fPart1;} //returns pointer to first particle
    AliHBTParticle* Particle2() const {return fPart2;} //returns pointer to decond particle
    
+   void Changed();
    //Center Mass System - Longitudinally Comoving
    
    Double_t GetInvMass(); //returns invariant mass of the pair
-   
+    
+  
+   Double_t GetQInv(); //returns Q invariant
    Double_t GetQSideCMSLC(); //returns Q Side CMS longitudionally co-moving
    Double_t GetQOutCMSLC(); //returns Q out CMS longitudionally co-moving
    Double_t GetQLongCMSLC(); //returns Q Long CMS longitudionally co-moving
@@ -32,11 +36,10 @@ class AliHBTPair: public TObject
    Double_t GetKt();  //returns K transverse
    Double_t GetKStar();
    
-   Double_t GetQInv(); //returns Q invariant
-   Double_t GetQSide(); //returns Q side
-   Double_t GetQLong(); //returns Q long 
-   Double_t GetQOut(); //returns Q out
-   
+   Double_t GetDeltaP(); //return difference of momenta
+   Double_t GetDeltaPx();
+   Double_t GetDeltaPy();
+   Double_t GetDeltaPz();
    
  protected:
    AliHBTParticle* fPart1;  //pointer to first particle
@@ -68,7 +71,7 @@ class AliHBTPair: public TObject
    Double_t fKt; //K == sum vector of particle's momenta. Kt transverse component
    Bool_t   fKtNotCalc;//flag indicating if fKt is already calculated for this pair
    
-   Double_t fKStar;
+   Double_t fKStar; //
    Bool_t   fKStarNotCalc;
    
    Double_t fPInv;  //invariant momentum
@@ -119,9 +122,15 @@ void AliHBTPair::SetParticles(AliHBTParticle* p1,AliHBTParticle* p2)
  fPart2 = p2;
  if (fSwapedPair) //if we have Swaped (so we are not)
    fSwapedPair->SetParticles(p2,p1); //set particles for him too
- 
- // Resel all calculations (flags)
+ Changed();
+ //and do nothing until will be asked for
+} 
+/****************************************************************/
 
+inline
+void AliHBTPair::Changed()
+{
+ // Resel all calculations (flags)
  fChanged           = kTRUE;
  fSumsNotCalc       = kTRUE;
  fDiffsNotCalc      = kTRUE;
@@ -134,14 +143,6 @@ void AliHBTPair::SetParticles(AliHBTParticle* p1,AliHBTParticle* p2)
  fKtNotCalc         = kTRUE;
  fKStarNotCalc      = kTRUE;
  fQInvLNotCalc      = kTRUE;
- 
- //and do nothing until will be asked for
-}
-/****************************************************************/
-inline
-Double_t AliHBTPair::GetQSide()
-{
- return 0.0;
 }
 /****************************************************************/
 inline 
@@ -162,6 +163,7 @@ void AliHBTPair::CalculateInvMassSqr()
 inline 
 void AliHBTPair::CalculateQInvL()
  {
+ //Calculates square root of Qinv
   if (fQInvLNotCalc)
   {
    CalculateDiffs();
@@ -196,4 +198,35 @@ void AliHBTPair::CalculateDiffs()
     }
  }
 
+/****************************************************************/
+inline 
+Double_t AliHBTPair::GetDeltaP() //return difference of momenta
+{
+ CalculateDiffs();
+ return TMath::Sqrt(fPxDiff*fPxDiff + fPyDiff*fPyDiff + fPzDiff*fPzDiff);
+}
+/****************************************************************/
+inline 
+Double_t AliHBTPair::GetDeltaPx()
+ {
+   CalculateDiffs();
+   return fPxDiff;
+ }
+/****************************************************************/
+inline 
+Double_t AliHBTPair::GetDeltaPy()
+ {
+   CalculateDiffs();
+   return fPyDiff;
+ }
+
+/****************************************************************/
+inline 
+Double_t AliHBTPair::GetDeltaPz()
+ {
+   CalculateDiffs();
+   return fPzDiff;
+ }
+
+
 #endif