New version including TOF

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

#ifndef ALIHBTWeightTHEORFCTN_H
#define ALIHBTWeightTHEORFCTN_H
/* $Id$ */
//_____________________________________________________________________________
///////////////////////////////////////////////////////////////////////////////
//
// class AliHBTWeightTheorQInvFctn
//
// This function allows to obtain Q_inv correlation function with weights
// calculated by Lednicky's alghorithm.
// Numerator is filled with weighted events. Weights are attributed to simulated particles.
// Weights are calculated with corresponding simulated particles momenta.
// Denominator is filled with mixing unweighted simulated particles.
// One needs only simulated pairs, so 
// this function is of class AliHBTOnePairFctn1D.
// Author Ludmila Malinina JINR (malinina@sunhe.jinr.ru)
//
///////////////////////////////////////////////////////////////////////////////

#include "AliHBTFunction.h"

class AliHBTWeights;

class AliHBTWeightTheorQInvFctn: public AliHBTOnePairFctn1D, public AliHBTCorrelFunction
{
  public:
    AliHBTWeightTheorQInvFctn(Int_t nbins = 100, Double_t maxXval = 0.15, Double_t minXval = 0.0);
    virtual  ~AliHBTWeightTheorQInvFctn(){}

    TH1* GetResult(); 
    void   ProcessSameEventParticles(AliHBTPair* partpair);

  protected:
    Double_t GetValue(AliHBTPair* partpair)
      { return partpair->GetQInv();} 

    ClassDef(AliHBTWeightTheorQInvFctn,1)
};
/*************************************************************/

class AliHBTWeightTheorQOutFctn: public AliHBTOnePairFctn1D, public AliHBTCorrelFunction
{

  public:
    AliHBTWeightTheorQOutFctn(Int_t nbins = 100, Double_t maxXval = 0.15, Double_t minXval = 0.0);
    virtual  ~AliHBTWeightTheorQOutFctn(){}

    TH1* GetResult(); 
    void   ProcessSameEventParticles(AliHBTPair* partpair);

  protected:
    Double_t GetValue(AliHBTPair* partpair)
      { return partpair->GetQOutCMSLC();}

    ClassDef(AliHBTWeightTheorQOutFctn,1)
};
/*************************************************************/

class AliHBTWeightTheorQSideFctn: public AliHBTOnePairFctn1D, public AliHBTCorrelFunction
{
  public:
    AliHBTWeightTheorQSideFctn(Int_t nbins = 100, Double_t maxXval = 0.15, Double_t minXval = 0.0);
    virtual  ~AliHBTWeightTheorQSideFctn(){}

    TH1* GetResult(); 
    void   ProcessSameEventParticles(AliHBTPair* partpair);
    
  protected:
    Double_t GetValue(AliHBTPair* partpair)
      { return partpair->GetQSideCMSLC();} 

    ClassDef(AliHBTWeightTheorQSideFctn,1)
};
/*************************************************************/

class AliHBTWeightTheorQLongFctn: public AliHBTOnePairFctn1D, public AliHBTCorrelFunction
{
  public:
    AliHBTWeightTheorQLongFctn(Int_t nbins = 100, Double_t maxXval = 0.15, Double_t minXval = 0.0);
    virtual  ~AliHBTWeightTheorQLongFctn(){}

    TH1* GetResult(); 
    void   ProcessSameEventParticles(AliHBTPair* partpair);
  
  protected:
    Double_t GetValue(AliHBTPair* partpair)
      { return partpair->GetQLongCMSLC();} 

    ClassDef(AliHBTWeightTheorQLongFctn,1)
};
/*************************************************************/

class AliHBTWeightTheorOSLFctn: public AliHBTOnePairFctn3D, public AliHBTCorrelFunction
{

  public:
    AliHBTWeightTheorOSLFctn(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  ~AliHBTWeightTheorOSLFctn(){}

    TH1* GetResult();
    void   ProcessSameEventParticles(AliHBTPair* partpair);
  
  protected:
    void GetValues(AliHBTPair* pair, Double_t& x, Double_t& y, Double_t& z)
      { x=TMath::Abs(pair->GetQOutCMSLC()); y=TMath::Abs(pair->GetQSideCMSLC()); z=TMath::Abs(pair->GetQLongCMSLC());} 

    ClassDef(AliHBTWeightTheorOSLFctn,1)
};
    
#endif