PWG1/TPC/AliPerfAnalyzeInvPt.h

   1
   2 #ifndef ALIPERFANALYZEINVPT_H
   3 #define ALIPERFANALYZEINVPT_H
   4
   5 //----------------------------------------------------------------------------------------------------
   6 // Class to analyse output of AliPerformancePtCalib.cxx and AliPerformancePtCalibMC.cxx:
   7 // Projection of charge/pt vs theta and vs phi resp. histoprams will be fitted with either
   8 // polynomial or gaussian fit function to extract minimum position of charge/pt.
   9 // Fit options and theta, phi bins can be set by user.
  10 // Attention: use the Set* functions of AliPerformancePtCalib.h and AliPerformancePtCalibMC.h
  11 // when running AliPerformancePtCalib*::Analyse().
  12 //
  13 // Author: S. Schuchmann 11/13/2009
  14 //----------------------------------------------------------------------------------------------------
  15
  16 class TNamed;
  17 class TString;
  18 class TList;
  19 class TF1;
  20 class TH1F;
  21 class TH2F;
  22 class TH1D;
  23 class TGraphErrors;
  24 //class TMath;
  25 class TCanvas;
  26 class TObjArray;
  27
  28
  29 class AliPerfAnalyzeInvPt : public TNamed {
  30
  31 public:
  32    AliPerfAnalyzeInvPt();
  33    AliPerfAnalyzeInvPt(const Char_t* name, const Char_t* title);
  34    virtual ~AliPerfAnalyzeInvPt(){;}
  35    void InitGraphs(Double_t *binsXTheta,Double_t *fitParamTheta,Double_t *errFitParamTheta,Double_t *binsXPhi,Double_t *fitParamPhi,Double_t *errFitParamPhi);
  36    void InitFitFcn();
  37    void StartAnalysis(const TH2F *histThetaInvPt, const TH2F *histPhiInvPt, TObjArray* aFolderObj);
  38    void SetProjBinsPhi(const Double_t *pBins,const Int_t sizep);
  39    void SetProjBinsTheta(const Double_t *tBins, const Int_t sizet);
  40    void SetMakeFitOption(const Bool_t setGausFit, const Double_t exclusionR,const Double_t fitR );
  41    void SetDoRebin(const Int_t rebin){if(rebin) {fDoRebin = kTRUE; fRebin = rebin;}}
  42
  43 protected:
  44    Double_t fThetaBins[100];// array of theta bins for projection and fit
  45    Double_t fPhiBins[100];// array of phi bins for projection and fit
  46
  47    Int_t fNThetaBins;// number of theta bins for projection and fit
  48    Int_t fNPhiBins; // number of phi bins for projection and fit
  49    Double_t fRange;// fit range of 1/pt spectrum
  50    Double_t fExclRange ;// range of exlusion of points around 0 when fitting 1/pt
  51    Bool_t fFitGaus;// set this flag for usage of gaussian fit function instead of polynomial (default)
  52    Bool_t  fDoRebin;
  53    Int_t  fRebin;
  54    // projection histos
  55    TH1D *fHistFitTheta[100];// projection histos for analysis in theta bins
  56    TH1D *fHistFitPhi[100];// projection histos for analysis in phi bins
  57
  58    //histos for copy of input histos
  59    TH2F *fHistH2InvPtTheta; // for copy of input histos
  60    TH2F *fHistH2InvPtPhi; // for copy of input histos
  61
  62    //graphs for displaying the fit parameters (minimum position of 1/pt and error)
  63    TGraphErrors *fGrMinPosTheta;
  64    TGraphErrors *fGrMinPosPhi;
  65
  66
  67
  68 private:
  69    //fit functions
  70    TF1 *fFitMinPos;//fit function for first fit with polynomial
  71    TF1 *fFitMinPosRejP;//fit function for second fit with polynomial to improve result
  72    TF1 *fFitInvGauss;//fit functionfor first fit with gaussian
  73    TF1 *fFitInvGaussRejP;//fit function for second fit with gaussian to improve result
  74
  75    // infput for fit functions
  76    static Double_t Polynomial(Double_t *x, const Double_t *par);
  77    static Double_t PolynomialRejP(Double_t *x, const Double_t *par);
  78    static Double_t InvGauss(Double_t *x, const Double_t *par);
  79    static Double_t InvGaussRejP(Double_t *x, const Double_t *par);
  80
  81    //functions for fitting procedure
  82    void MakeFit(TH1D *dmproy, TF1 * fitpb, Double_t &mean, Double_t &ErrMean,  Double_t &excl,Double_t &range);
  83    void MakeFitBetter(TH1D *dmproy, TF1 * fitpb2, Double_t &mean, Double_t &ErrMean, Double_t &f, Double_t &excl, Double_t &range);
  84    void MakeFitInvGauss(TH1D *dmproy, TF1 * fitpb2, Double_t &mean, Double_t &ErrMean,Double_t &excl , Double_t &range);
  85    void MakeFitInvGaussBetter(TH1D *dmproy, TF1 * fitpb2, Double_t &mean, Double_t &ErrMean, Double_t &f,Double_t &excl,  Double_t &range);
  86
  87
  88
  89
  90    AliPerfAnalyzeInvPt(const AliPerfAnalyzeInvPt&);            // not implemented
  91    AliPerfAnalyzeInvPt& operator=(const AliPerfAnalyzeInvPt&); // not implemented
  92
  93    ClassDef(AliPerfAnalyzeInvPt, 1);
  94 };
  95
  96 #endif