Bug fix (Renu)

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliBtoJPSItoEleCDFfitFCN.h

#ifndef ALIBTOJPSITOELECDFFITFCN_H\r
#define ALIBTOJPSITOELECDFFITFCN_H\r
/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *\r
 * See cxx source for full Copyright notice                               */\r
\r
//_________________________________________________________________________\r
//                      Class AliBtoJPSItoEleCDFfitFCN\r
//                    Definition of main function used in \r
//                     unbinned log-likelihood fit for\r
//                 the channel B -> JPsi + X -> e+e- + X\r
//      \r
//                          Origin: C.Di Giglio\r
//     Contact: Carmelo.Digiglio@ba.infn.it , Giuseppe.Bruno@ba.infn.it\r
//_________________________________________________________________________\r
\r
#include <TNamed.h>\r
#include <TDatabasePDG.h>\r
#include "TH1F.h"\r
#include "TMath.h"\r
\r
enum IntegralType {kBkg, \r
                   kBkgNorm, \r
                   kSig, \r
                   kSigNorm};\r
\r
enum PtBins       {kallpt, \r
                   kptbin1,kptbin2,kptbin3,\r
                   kptbin4,kptbin5,kptbin6,\r
                   kptbin7,kptbin8,kptbin9};\r
//_________________________________________________________________________________________________\r
class AliBtoJPSItoEleCDFfitFCN : public TNamed {\r
 public:\r
  //\r
  AliBtoJPSItoEleCDFfitFCN();\r
  AliBtoJPSItoEleCDFfitFCN(const AliBtoJPSItoEleCDFfitFCN& source); \r
  AliBtoJPSItoEleCDFfitFCN& operator=(const AliBtoJPSItoEleCDFfitFCN& source);  \r
  virtual ~AliBtoJPSItoEleCDFfitFCN();\r
\r
  Double_t EvaluateLikelihood(const Double_t* pseudoproperdecaytime,\r
                              const Double_t* invariantmass, const Int_t ncand);\r
 \r
  Double_t GetFPlus() const {return fFPlus;}\r
  Double_t GetFMinus() const {return fFMinus;}\r
  Double_t GetFSym() const {return fFSym;}\r
  Double_t GetRadius() const {return fParameters[0];}\r
  Double_t GetTheta() const {return fParameters[1];}\r
  Double_t GetPhi() const {return fParameters[2];}\r
  Double_t GetLamPlus() const {return fParameters[3];}\r
  Double_t GetLamMinus() const {return fParameters[4];}\r
  Double_t GetLamSym() const {return fParameters[5];}\r
  Double_t GetMassSlope() const {return fParameters[6];}\r
  Double_t GetFractionJpsiFromBeauty() const {return fParameters[7];}\r
  Double_t GetFsig() const {return fParameters[8];}\r
  Double_t GetCrystalBallMmean() const {return fParameters[9];}\r
  Double_t GetCrystalBallNexp() const {return fParameters[10];}\r
  Double_t GetCrystalBallSigma() const {return fParameters[11];}\r
  Double_t GetCrystalBallAlpha() const {return fParameters[12];}\r
  Double_t GetIntegral() const {return fIntegral;}\r
  Bool_t GetCrystalBallParam() const {return fCrystalBallParam;}\r
\r
  void SetFPlus(Double_t plus) {fFPlus = plus;}\r
  void SetFMinus(Double_t minus) {fFMinus = minus;}\r
  void SetFSym(Double_t sym) {fFSym = sym;}\r
  void SetRadius(Double_t radius) {fParameters[0] = radius;}\r
  void SetTheta(Double_t theta) {fParameters[1] = theta;}\r
  void SetPhi(Double_t phi) {fParameters[2] = phi;}\r
  void SetLamPlus(Double_t lamplus) {fParameters[3] = lamplus;}\r
  void SetLamMinus(Double_t lamminus) {fParameters[4] = lamminus;}\r
  void SetLamSym(Double_t lamsym) {fParameters[5] = lamsym;}\r
  void SetMassSlope(Double_t slope) {fParameters[6] = slope;}\r
  void SetFractionJpsiFromBeauty(Double_t B) {fParameters[7] = B;}\r
  void SetFsig(Double_t Fsig) {fParameters[8] = Fsig;}\r
  void SetCrystalBallMmean(Double_t CrystalBallMmean) {fParameters[9] = CrystalBallMmean;}\r
  void SetCrystalBallNexp(Double_t CrystalBallNexp) {fParameters[10] = CrystalBallNexp;}\r
  void SetCrystalBallSigma(Double_t CrystalBallSigma) {fParameters[11] = CrystalBallSigma;}\r
  void SetCrystalBallAlpha(Double_t CrystalBallAlpha) {fParameters[12] = CrystalBallAlpha;}\r
\r
  void SetAllParameters(const Double_t* parameters);\r
  void SetIntegral(Double_t integral) {fIntegral = integral;}\r
  void SetCsiMC(const TH1F* MCtemplate) {fhCsiMC = (TH1F*)MCtemplate->Clone("fhCsiMC");}\r
  void SetResolutionConstants(Int_t BinNum);\r
  void SetMassWndHigh(Double_t limit) { fMassWndHigh = TDatabasePDG::Instance()->GetParticle(443)->Mass() + limit ;}//here use pdg code instead\r
  void SetMassWndLow(Double_t limit) { fMassWndLow = TDatabasePDG::Instance()->GetParticle(443)->Mass() - limit ;}//here use pdg code instead\r
  void SetCrystalBallParam(Bool_t okCB) {fCrystalBallParam = okCB;}\r
\r
  void ConvertFromSpherical() { fFPlus  = TMath::Power((fParameters[0]*TMath::Cos(fParameters[1])),2.);\r
                                fFMinus = TMath::Power((fParameters[0]*TMath::Sin(fParameters[1])*TMath::Sin(fParameters[2])),2.);\r
                                fFSym   = TMath::Power((fParameters[0]*TMath::Sin(fParameters[1])*TMath::Cos(fParameters[2])),2.);} \r
\r
  void ComputeIntegral(); \r
\r
  void ReadMCtemplates(Int_t BinNum);\r
\r
  void PrintStatus();\r
\r
 private:  \r
  //\r
  Double_t fParameters[13];        /*  par[0]  = fRadius;                \r
                                       par[1]  = fTheta;\r
                                       par[2]  = fPhi;\r
                                       par[3]  = fOneOvLamPlus;\r
                                       par[4]  = fOneOvLamMinus;\r
                                       par[5]  = fOneOvLamSym;\r
                                       par[6]  = fMassBkgSlope;\r
                                       par[7]  = fFractionJpsiFromBeauty;\r
                                       par[8]  = fFsig;\r
                                       par[9]  = fCrystalBallMmean;\r
                                       par[10] = fCrystalBallNexp;\r
                                       par[11] = fCrystalBallSigma;\r
                                       par[12] = fCrystalBallAlpha;*/\r
\r
  Double_t fFPlus;                  // parameters of the log-likelihood function\r
  Double_t fFMinus;                 // Slopes of the x distributions of the background \r
  Double_t fFSym;                   // functions \r
\r
  Double_t fIntegral;               // integral values of log-likelihood function terms\r
  TH1F *fhCsiMC;                    // X distribution used as MC template for JPSI from B\r
  Double_t fResolutionConstants[7]; // constants for the parametrized resolution function R(X)\r
  Double_t fMassWndHigh;            // JPSI Mass window higher limit\r
  Double_t fMassWndLow;             // JPSI Mass window lower limit\r
  Bool_t fCrystalBallParam;         // Boolean to switch to Crystall Ball parameterisation\r
\r
  ////\r
\r
  Double_t EvaluateCDFfunc(Double_t x, Double_t m) const ;\r
\r
  Double_t EvaluateCDFfuncNorm(Double_t x, Double_t m) const ;\r
\r
  ////\r
\r
  Double_t EvaluateCDFfuncSignalPart(Double_t x, Double_t m) const ;      // Signal part \r
\r
  Double_t EvaluateCDFDecayTimeSigDistr(Double_t x) const ;\r
\r
  Double_t EvaluateCDFInvMassSigDistr(Double_t m) const ;\r
\r
  Double_t EvaluateCDFfuncBkgPart(Double_t x,Double_t m) const ;          // Background part\r
\r
  Double_t EvaluateCDFDecayTimeBkgDistr(Double_t x) const ;\r
 \r
  Double_t EvaluateCDFInvMassBkgDistr(Double_t m) const ;\r
\r
  ////\r
\r
  Double_t FunB(Double_t x) const ;\r
\r
  Double_t FunP(Double_t x) const ;\r
\r
  Double_t CsiMC(Double_t x) const ;\r
\r
  Double_t FunBkgPos(Double_t x) const ;\r
\r
  Double_t FunBkgNeg(Double_t x) const ;\r
\r
  Double_t FunBkgSym(Double_t x) const ;\r
\r
  Double_t ResolutionFunc(Double_t x) const ;\r
  //\r
  ClassDef (AliBtoJPSItoEleCDFfitFCN,0);         // Unbinned log-likelihood fit \r
\r
};\r
\r
#endif\r