AliHFMultiTrials.h

Go to the documentation of this file.

#ifndef ALIHFMULTITRIALS_H
#define ALIHFMULTITRIALS_H
/* Copyright(c) 2008-2019, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#include <TNamed.h>
#include <TString.h>
#include <TPad.h>


class AliHFMultiTrials : public TNamed {

 public:
  AliHFMultiTrials();
  virtual ~AliHFMultiTrials();

  void ConfigureRebinSteps(Int_t nSteps, Int_t*  values){
    if(fRebinSteps) delete [] fRebinSteps;
    fNumOfRebinSteps=nSteps;
    fRebinSteps = new Int_t[fNumOfRebinSteps];
    for(Int_t ib=0; ib<fNumOfRebinSteps; ib++) fRebinSteps[ib]=values[ib];
  }

  void ConfigureLowLimFitSteps(Int_t nSteps, Double_t*  values){
    if(fLowLimFitSteps) delete [] fLowLimFitSteps;
    fNumOfLowLimFitSteps=nSteps;
    fLowLimFitSteps = new Double_t[fNumOfLowLimFitSteps];
    for(Int_t ib=0; ib<fNumOfLowLimFitSteps; ib++) fLowLimFitSteps[ib]=values[ib];
  }

  void ConfigureUpLimFitSteps(Int_t nSteps, Double_t*  values){
    if(fUpLimFitSteps) delete [] fUpLimFitSteps;
    fNumOfUpLimFitSteps=nSteps;
    fUpLimFitSteps = new Double_t[fNumOfUpLimFitSteps];
    for(Int_t ib=0; ib<fNumOfUpLimFitSteps; ib++) fUpLimFitSteps[ib]=values[ib];
  }

  void ConfigurenSigmaBinCSteps(Int_t nSteps, Double_t*  values){
    if(fnSigmaBinCSteps) delete [] fnSigmaBinCSteps;
    fNumOfnSigmaBinCSteps=nSteps;
    fnSigmaBinCSteps = new Double_t[fNumOfnSigmaBinCSteps];
    for(Int_t ib=0; ib<fNumOfnSigmaBinCSteps; ib++) fnSigmaBinCSteps[ib]=values[ib];
  }

  void SetMass(Double_t mass){fMassD=mass;}
  void SetSigmaGaussMC(Double_t sig){fSigmaGausMC=sig;}
  void SetSigmaMCVariation(Double_t var=0.15){fSigmaMCVariation=var;}

  void SetSuffixForHistoNames(const Char_t* name){
    fSuffix=name;
  }
  void SetUseChi2Fit(){fFitOption=1;}
  void SetUseLogLikelihoodFit(){fFitOption=0;}
  void SetUseExpoBackground(Bool_t opt=kTRUE){fUseExpoBkg=opt;}
  void SetUseLinBackground(Bool_t opt=kTRUE){fUseLinBkg=opt;}
  void SetUsePol2Background(Bool_t opt=kTRUE){fUsePol2Bkg=opt;}
  void SetUsePol3Background(Bool_t opt=kTRUE){fUsePol3Bkg=opt;}
  void SetUsePol4Background(Bool_t opt=kTRUE){fUsePol4Bkg=opt;}
  void SetUsePol5Background(Bool_t opt=kTRUE){fUsePol5Bkg=opt;}

  void SetDrawIndividualFits(Bool_t opt=kTRUE){fDrawIndividualFits=opt;}

  Bool_t DoMultiTrials(TH1D* hInvMassHisto, TPad* thePad=0x0);
  void SaveToRoot(TString fileName, TString option="recreate") const;
  void DrawHistos(TCanvas* cry) const;

  enum EBkgFuncCases{ kExpoBkg, kLinBkg, kPol2Bkg, kPol3Bkg, kPol4Bkg, kPol5Bkg, kNBkgFuncCases };
  enum EFitParamCases{ kFixSigFreeMean, kFixSigUpFreeMean, kFixSigDownFreeMean, kFreeSigFreeMean, kFixSigFixMean, kFreeSigFixMean, kNFitConfCases};

 private:

  Bool_t CreateHistos();
  TH1F* RebinHisto(TH1D* hOrig, Int_t reb, Int_t firstUse) const;
  void BinCount(TH1F* h, TF1* fB, Int_t rebin, Double_t minMass, Double_t maxMass, Double_t& count, Double_t& ecount) const;
  Bool_t DoFitWithPol3Bkg(TH1F* histoToFit, Double_t  hmin, Double_t  hmax,
        Int_t theCase);

  AliHFMultiTrials(const AliHFMultiTrials &source);
  AliHFMultiTrials& operator=(const AliHFMultiTrials& source); 

  Int_t fNumOfRebinSteps; 
  Int_t* fRebinSteps;     //[fNumOfRebinSteps] values of rebin
  Int_t fNumOfLowLimFitSteps; 
  Double_t* fLowLimFitSteps; //[fNumOfLowLimFitSteps] values of low limits for fit
  Int_t fNumOfUpLimFitSteps; 
  Double_t* fUpLimFitSteps; //[fNumOfUpLimFitSteps] values of up limits for fit
  Int_t fNumOfnSigmaBinCSteps; 
  Double_t* fnSigmaBinCSteps; //[fNumOfnSigmaBinCSteps] values of nsigma for bin count

  Double_t fSigmaGausMC; 
  Double_t fSigmaMCVariation; 
  Double_t fMassD;       
  TString fSuffix;       
  Int_t fFitOption;      
  Bool_t fUseExpoBkg;    
  Bool_t fUseLinBkg;    
  Bool_t fUsePol2Bkg;    
  Bool_t fUsePol3Bkg;    
  Bool_t fUsePol4Bkg;    
  Bool_t fUsePol5Bkg;    

  Bool_t fDrawIndividualFits; 
 
  TH1F* fHistoRawYieldDistAll;  
  TH1F* fHistoRawYieldTrialAll; 
  TH1F* fHistoSigmaTrialAll;    
  TH1F* fHistoMeanTrialAll;     
  TH1F* fHistoChi2TrialAll;     

  TH1F* fHistoRawYieldDistBinCAll; 
  TH2F* fHistoRawYieldTrialBinCAll; 

  TH1F** fHistoRawYieldDist;  
  TH1F** fHistoRawYieldTrial; 
  TH1F** fHistoSigmaTrial;    
  TH1F** fHistoMeanTrial;     
  TH1F** fHistoChi2Trial;     

  TH1F** fHistoRawYieldDistBinC;  
  TH2F** fHistoRawYieldTrialBinC; 

  TNtuple* fNtupleMultiTrials; 
 
  Double_t fMinYieldGlob;   
  Double_t fMaxYieldGlob;   

  ClassDef(AliHFMultiTrials,1); 
};

#endif