///////////////////////////////////////////////////////////////////////////
//                                                                       //
// AliFemtoBPLCMS3DCorrFctn: a class to calculate 3D correlation         //
// for pairs of identical particles.                                     //
//                                                                       //
///////////////////////////////////////////////////////////////////////////

#ifndef ALIFEMTOBPLCMS3DCORRFCTN_H
#define ALIFEMTOBPLCMS3DCORRFCTN_H

#include "AliFemtoCorrFctn.h"
//#include "AliFemtoCoulomb.h"
#include "AliFemtoPairCut.h"
//#include "AliFemtoHisto.h"
#include "TH3D.h"
//#include "AliFemtoSmearPair.h"

class AliFemtoBPLCMS3DCorrFctn : public AliFemtoCorrFctn {
public:
  AliFemtoBPLCMS3DCorrFctn(char* title, const int& nbins, const float& QLo, const float& QHi);
  AliFemtoBPLCMS3DCorrFctn(const AliFemtoBPLCMS3DCorrFctn& aCorrFctn);
  virtual ~AliFemtoBPLCMS3DCorrFctn();

  AliFemtoBPLCMS3DCorrFctn& operator=(const AliFemtoBPLCMS3DCorrFctn& aCorrFctn);

  virtual AliFemtoString Report();
  virtual void AddRealPair( AliFemtoPair* aPair);
  virtual void AddMixedPair( AliFemtoPair* aPair);

  virtual void Finish();

  TH3D* Numerator();
  TH3D* Denominator();
  TH3D* Ratio();
  TH3D* QinvHisto();

  // here are get and set for the range over which the correlation function 
  // is normalized (in Qinv).  The range is set to 0.15..0.18 in the constuctor
  // by default, but the Set's below override this
  void SetNormRangeLo(float qLo);
  void SetNormRangeHi(float qHi);
  float GetNormRangeLo() const;
  float GetNormRangeHi() const;

  void WriteOutHistos();
  virtual TList* GetOutputList();

  //  void SetCoulombCorrection(AliFemtoCoulomb* Correction);

  void SetSpecificPairCut(AliFemtoPairCut* aCut);
  void SetUseRPSelection(unsigned short aRPSel);

  //  void SetSmearPair(AliFemtoSmearPair*);
  void SetRout(double guess);
  void SetRside(double guess);
  void SetRlong(double guess);
  void SetLambda(double guess);

private:
  // here are a whole bunch of histos that get filled if we do resolution correction
  TH3D* fIDNumHisto;        // true pairs numerator  
  TH3D* fIDDenHisto;        // true pairs denominator
  TH3D* fIDRatHisto;        // true pairs ratio      
  // 
  TH3D* fSMNumHisto;        // mixed pairs numerator  
  TH3D* fSMDenHisto;	    // mixed pairs denominator
  TH3D* fSMRatHisto;	    // mixed pairs ratio      
  //
  TH3D* fCorrectionHisto;   // correction histogram
  TH3D* fCorrCFHisto;       // Corrected CF

  TH3D* fNumerator;         // numerator
  TH3D* fDenominator;       // denominator
  //  TH3D* fUncorrectedDenominator;
  TH3D* fRatio;             // ratio - the correlation function
  TH3D* fQinvHisto;         // Qinv weights

  // for resolution correction
  //  AliFemtoSmearPair* fSmearPair; //!
  double fLambda;           // lambda for smearing correction
  double fRout2;            // Rout for smearing correction
  double fRside2;           // Rside for smearing correction
  double fRlong2;           // Rlong for smearing correction

  AliFemtoPairCut* fPairCut;    //! this is a PairCut specific to THIS CorrFctn, not the Analysis

  // upper and lower bounds of Qinv region where to do normalization
  float fQinvNormLo;        // Lower bound of Qinv normalization range
  float fQinvNormHi;        // Upper bound of Qinv normalization range

  // and here are the number of pairs in that region...
  unsigned long int fNumRealsNorm; // pairs in numerator in Qinv normalization range
  unsigned long int fNumMixedNorm; // pairs in denominator in Qinv normalization range

  unsigned short fUseRPSelection;  // The pair cut uses RP selection

  //  AliFemtoCoulomb* fCorrection; //!
  

#ifdef __ROOT__
  ClassDef(AliFemtoBPLCMS3DCorrFctn, 1)
#endif
};

inline  TH3D* AliFemtoBPLCMS3DCorrFctn::Numerator(){return fNumerator;}
inline  TH3D* AliFemtoBPLCMS3DCorrFctn::Denominator(){return fDenominator;}
//inline  TH3D* AliFemtoBPLCMS3DCorrFctn::UncorrectedDenominator(){return fUncorrectedDenominator;}
inline  TH3D* AliFemtoBPLCMS3DCorrFctn::Ratio(){return fRatio;}
inline  TH3D* AliFemtoBPLCMS3DCorrFctn::QinvHisto(){return fQinvHisto;}
inline  void AliFemtoBPLCMS3DCorrFctn::SetNormRangeLo(float qLo){fQinvNormLo = qLo;}
inline  void AliFemtoBPLCMS3DCorrFctn::SetNormRangeHi(float qHi){fQinvNormHi = qHi;}
inline  float AliFemtoBPLCMS3DCorrFctn::GetNormRangeLo() const{return fQinvNormLo;}
inline  float AliFemtoBPLCMS3DCorrFctn::GetNormRangeHi() const{return fQinvNormHi;}
//inline  void AliFemtoBPLCMS3DCorrFctn::SetCoulombCorrection(AliFemtoCoulomb* Correction){fCorrection = Correction;}
inline  void AliFemtoBPLCMS3DCorrFctn::SetSpecificPairCut(AliFemtoPairCut* pc){fPairCut=pc;}
//inline  void AliFemtoBPLCMS3DCorrFctn::SetSmearPair(AliFemtoSmearPair* sp){fSmearPair = sp;}

inline  void AliFemtoBPLCMS3DCorrFctn::SetRout(double r){fRout2 = r*r;}
inline  void AliFemtoBPLCMS3DCorrFctn::SetRside(double r){fRside2 = r*r;}
inline  void AliFemtoBPLCMS3DCorrFctn::SetRlong(double r){fRlong2 = r*r;}
inline  void AliFemtoBPLCMS3DCorrFctn::SetLambda(double l){fLambda = l;}

#endif