[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEBeautySpectrum.h

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/
//
// Class for spectrum correction
// Subtraction of hadronic background, Unfolding of the data and
// Renormalization done here
// For more information see the implementation file
//
#ifndef ALIHFEBEAUTYSPECTRUM_H
#define ALIHFEBEAUTYSPECTRUM_H

#include "AliHFECorrectSpectrumBase.h"


class TGraphErrors;
class TObject;
class TH1;
class TF1;
class TList;
class TObjArray;
class AliCFContainer;
class AliHFEcontainer;
class AliCFDataGrid;
class AliCFEffGrid;
class AliHFEInclusiveSpectrumQA;
class AliHFEBeautySpectrumQA;

class AliHFEBeautySpectrum : public AliHFECorrectSpectrumBase{
 public:
  
  enum{
    kElecBgSources = 9,
    kBgLevels = 3,
    kBgPtBins = 44,
    kSignalPtBins = 35,
    kCentrality = 12
  };
  
  AliHFEBeautySpectrum(const char* name);
  ~AliHFEBeautySpectrum();
  
  
  virtual Bool_t Init(const AliHFEcontainer *datahfecontainer, const AliHFEcontainer *mchfecontainer, const AliHFEcontainer *bghfecontainer=0x0, const AliHFEcontainer */*v0hfecontainer*/ = 0x0,AliCFContainer */*photoniccontainerD*/ = 0x0);
  virtual Bool_t Correct(Bool_t subtractcontamination=kTRUE, Bool_t /*subtractphotonic*/=kFALSE);
  
  AliCFDataGrid *SubtractBackground(Bool_t setBackground = kFALSE);
  
  AliCFDataGrid *CorrectParametrizedEfficiency(AliCFDataGrid* const bgsubpectrum = 0x0);
  
  THnSparse *Unfold(AliCFDataGrid* const bgsubpectrum = 0x0);
  void UnfoldBG(AliCFDataGrid* const bgsubpectrum);
  AliCFDataGrid *CorrectForEfficiency(AliCFDataGrid* const bgsubpectrum = 0x0);
  
  void SetPbPbAnalysis(Bool_t isPbPb = kFALSE) { fBeamType=(Char_t) isPbPb; };
  void SetEtaSyst(Bool_t etaSyst = kTRUE) { fEtaSyst = etaSyst; };
  
  void SetParameterizedEff(AliCFContainer *container, AliCFContainer *containermb, AliCFContainer *containeresd, AliCFContainer *containeresdmb, Int_t *dimensions);
  
  void SetNumberOfMCEvents(Int_t nEvents) { fNMCEvents = nEvents; };
  void SetNumberOfMC2Events(Int_t nEvents) { fNMCbgEvents = nEvents; };
  void SetUnSetCorrelatedErrors(Bool_t unsetcorrelatederrors) {fUnSetCorrelatedErrors = unsetcorrelatederrors;};
  void SetFillMoreCorrelationMatrix(Bool_t fillMoreCorrelationMatrix) { fFillMoreCorrelationMatrix = fillMoreCorrelationMatrix;}
  
  void SetNCentralityBinAtTheEnd(Int_t nCentralityBinAtTheEnd) {fNCentralityBinAtTheEnd = nCentralityBinAtTheEnd; };
  void SetLowHighBoundaryCentralityBinAtTheEnd(Int_t low, Int_t high, Int_t i) { fLowBoundaryCentralityBinAtTheEnd[i] = low; fHighBoundaryCentralityBinAtTheEnd[i] = high;};
  
  void CallInputFileForBeauty2ndMethod();
  void SetInputFileForBeauty2ndMethod(const char *filenameb = "BSpectrum2ndmethod.root"){fkBeauty2ndMethodfilename = filenameb; };
  void SetBeautyAnalysis2ndMethod(Bool_t beauty2ndmethod) { fBeauty2ndMethod = beauty2ndmethod; }
  void SetIPEffCombinedSamples(Bool_t ipEffCombinedSamples) { fIPEffCombinedSamples = ipEffCombinedSamples; }
  void SetHadronEffbyIPcut(THnSparseF* hsHadronEffbyIPcut) { fHadronEffbyIPcut = hsHadronEffbyIPcut;};
  void SetNonHFEsyst(Bool_t syst){ fNonHFEsyst = syst; };
  
  void SetDumpToFile(Bool_t dumpToFile) { fDumpToFile=dumpToFile; }; 
  
  void SetDebugLevel(Int_t debugLevel, Bool_t writeToFile = kFALSE) { fDebugLevel = debugLevel; fWriteToFile = writeToFile; };
  void SetUnfoldBG() { fUnfoldBG = kTRUE; };
  
  
  AliCFDataGrid* GetRawBspectra2ndMethod();
  AliCFDataGrid* GetCharmBackground();
  AliCFDataGrid* GetNonHFEBackground(Int_t decay = 0, Int_t source = 0);
  THnSparse* GetCharmWeights(Int_t centBin);
  THnSparse* GetBeautyIPEff(Bool_t isMCpt);
  THnSparse* GetPIDxIPEff(Int_t source);
  void CalculateNonHFEsyst();
  
  void EnableIPanaHadronBgSubtract() { fIPanaHadronBgSubtract = kTRUE; };
  void EnableIPanaCharmBgSubtract() { fIPanaCharmBgSubtract = kTRUE; };
  void EnableIPanaNonHFEBgSubtract() { fIPanaNonHFEBgSubtract = kTRUE; };
  void EnableIPParameterizedEff() { fIPParameterizedEff = kTRUE; };
  
 protected:
  void AddTemporaryObject(TObject *cont);
  void ClearObject(TObject *o);
  
 private:
  AliHFEBeautySpectrum(const AliHFEBeautySpectrum &ref);
  AliHFEBeautySpectrum &operator=(const AliHFEBeautySpectrum &ref);
  virtual void Copy(TObject &o) const;
  
  AliHFEBeautySpectrumQA *fQA; // Beauty QA
  
  //shift some of the following to QA or Base class:
  TList *fTemporaryObjects;     // Emulate garbage collection
  AliCFDataGrid *fBackground;   // Background Grid
  TF1 *fEfficiencyTOFPIDD;       // TOF PID efficiency parameterized
  TF1 *fEfficiencyesdTOFPIDD;    // TOF PID efficiency parameterized
  TF1 *fEfficiencyIPCharmD;      // IP efficiency parameterized for charm
  TF1 *fEfficiencyIPBeautyD;     // IP efficiency parameterized for beauty 
  TF1 *fEfficiencyIPBeautyesdD;  // IP efficiency parameterized for beauty for esd
  TF1 *fEfficiencyIPConversionD; // IP efficiency parameterized for conversion
  TF1 *fEfficiencyIPNonhfeD;     // IP efficiency parameterized for nonhfe 
  TF1 *fNonHFEbg;     // Efficiency Function
  THnSparseF *fWeightCharm;     // Weight for charm bg
  
  AliCFContainer *fConvSourceContainer[kElecBgSources][kBgLevels]; //container for conversion electrons, divided into different photon sources
  AliCFContainer *fNonHFESourceContainer[kElecBgSources][kBgLevels]; //container for non-HF electrons, divided into different sources
  
  Bool_t fInclusiveSpectrum;     // Inclusive Spectrum
  Bool_t fDumpToFile;           // Write Result in a file
  
  Bool_t fUnSetCorrelatedErrors;    // Unset correlated errors
  
  Bool_t fIPanaHadronBgSubtract;     // Hadron background subtraction
  Bool_t fIPanaCharmBgSubtract;      // Charm background subtraction 
  Bool_t fIPanaNonHFEBgSubtract;     // nonHFE background subtraction
  Bool_t fIPParameterizedEff;        // switch to use parameterized efficiency for ip analysis
  Bool_t fNonHFEsyst;            // choose NonHFE background level (upper, lower, central)
  Bool_t fBeauty2ndMethod;      // 2nd method to get beauty spectrum
  Bool_t fIPEffCombinedSamples; // flag to combine two different samples
  
  Int_t fNMCEvents;             // Number of MC Events
  Int_t fNMCbgEvents;       // Number of BG MC Events
  
  Int_t fNCentralityBinAtTheEnd;// Number of centrality class at the end
  Int_t fLowBoundaryCentralityBinAtTheEnd[20];  // Boundary of the bins
  Int_t fHighBoundaryCentralityBinAtTheEnd[20];  // Boundary of the bins
  Bool_t fFillMoreCorrelationMatrix;             // For low stats to have reasonable errors
  
  THnSparseF *fHadronEffbyIPcut;// container for hadron efficiency by IP cut
  TH1D *fEfficiencyCharmSigD; // charm IP cut eff from signal enhanced MC
  TH1D *fEfficiencyBeautySigD; // beauty IP cut eff from signal enhanced MC
  TH1D *fEfficiencyBeautySigesdD; // beauty IP cut eff from signal enhanced MC for esd
  TH1D *fConversionEff;     // conversion IP cut eff
  TH1D *fNonHFEEff;         // nonhfe IP cut eff
  TH1D *fCharmEff;          // charm IP cut eff
  TH1D *fBeautyEff;         // beauty IP cut eff
  TH1D *fConversionEffbgc;      // conversion IP cut eff
  TH1D *fNonHFEEffbgc;          // nonhfe IP cut eff
  TH1D *fBSpectrum2ndMethod;             // beauty spectrum for 2nd method
  const char *fkBeauty2ndMethodfilename;      // name of file, which contains beauty spectrum for 2ndmethod
  Char_t fBeamType;             // beamtype; default -1; pp =0; PbPb=1
  Bool_t fEtaSyst;              // pp 2.76 TeV (= kTRUE) or 7 TeV (= kFALSE)
  
  
  Int_t fDebugLevel;            // Debug Level
  Bool_t fWriteToFile;           // Write plots to eps files
  Bool_t fUnfoldBG;             // flag to unfold backgroud
  
  ClassDef(AliHFEBeautySpectrum, 1) 
    };
#endif
Commit	Line	Data
4bd4fc13	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15	//
	16	// Class for spectrum correction
	17	// Subtraction of hadronic background, Unfolding of the data and
	18	// Renormalization done here
	19	// For more information see the implementation file
	20	//
	21	#ifndef ALIHFEBEAUTYSPECTRUM_H
	22	#define ALIHFEBEAUTYSPECTRUM_H
	23
	24	#include "AliHFECorrectSpectrumBase.h"
	25
	26
	27
	28	class TGraphErrors;
	29	class TObject;
	30	class TH1;
	31	class TF1;
	32	class TList;
	33	class TObjArray;
	34	class AliCFContainer;
	35	class AliHFEcontainer;
	36	class AliCFDataGrid;
	37	class AliCFEffGrid;
	38	class AliHFEInclusiveSpectrumQA;
	39	class AliHFEBeautySpectrumQA;
	40
	41	class AliHFEBeautySpectrum : public AliHFECorrectSpectrumBase{
	42	public:
	43
	44	enum{
	45	kElecBgSources = 9,
	46	kBgLevels = 3,
	47	kBgPtBins = 44,
	48	kSignalPtBins = 35,
	49	kCentrality = 12
	50	};
	51
	52	AliHFEBeautySpectrum(const char* name);
	53	~AliHFEBeautySpectrum();
	54
	55
	56	virtual Bool_t Init(const AliHFEcontainer datahfecontainer, const AliHFEcontainer mchfecontainer, const AliHFEcontainer bghfecontainer=0x0, const AliHFEcontainer /v0hfecontainer/ = 0x0,AliCFContainer /photoniccontainerD*/ = 0x0);
	57	virtual Bool_t Correct(Bool_t subtractcontamination=kTRUE, Bool_t /subtractphotonic/=kFALSE);
	58
	59	AliCFDataGrid *SubtractBackground(Bool_t setBackground = kFALSE);
	60
	61	AliCFDataGrid CorrectParametrizedEfficiency(AliCFDataGrid const bgsubpectrum = 0x0);
	62
	63	THnSparse Unfold(AliCFDataGrid const bgsubpectrum = 0x0);
	64	void UnfoldBG(AliCFDataGrid* const bgsubpectrum);
65	AliCFDataGrid CorrectForEfficiency(AliCFDataGrid const bgsubpectrum = 0x0);
66
67	void SetPbPbAnalysis(Bool_t isPbPb = kFALSE) { fBeamType=(Char_t) isPbPb; };
68	void SetEtaSyst(Bool_t etaSyst = kTRUE) { fEtaSyst = etaSyst; };
69
70	void SetParameterizedEff(AliCFContainer container, AliCFContainer containermb, AliCFContainer containeresd, AliCFContainer containeresdmb, Int_t *dimensions);
71
72	void SetNumberOfMCEvents(Int_t nEvents) { fNMCEvents = nEvents; };
73	void SetNumberOfMC2Events(Int_t nEvents) { fNMCbgEvents = nEvents; };
74	void SetUnSetCorrelatedErrors(Bool_t unsetcorrelatederrors) {fUnSetCorrelatedErrors = unsetcorrelatederrors;};
75	void SetFillMoreCorrelationMatrix(Bool_t fillMoreCorrelationMatrix) { fFillMoreCorrelationMatrix = fillMoreCorrelationMatrix;}
76
77	void SetNCentralityBinAtTheEnd(Int_t nCentralityBinAtTheEnd) {fNCentralityBinAtTheEnd = nCentralityBinAtTheEnd; };
78	void SetLowHighBoundaryCentralityBinAtTheEnd(Int_t low, Int_t high, Int_t i) { fLowBoundaryCentralityBinAtTheEnd[i] = low; fHighBoundaryCentralityBinAtTheEnd[i] = high;};
79
80	void CallInputFileForBeauty2ndMethod();
81	void SetInputFileForBeauty2ndMethod(const char *filenameb = "BSpectrum2ndmethod.root"){fkBeauty2ndMethodfilename = filenameb; };
82	void SetBeautyAnalysis2ndMethod(Bool_t beauty2ndmethod) { fBeauty2ndMethod = beauty2ndmethod; }
83	void SetIPEffCombinedSamples(Bool_t ipEffCombinedSamples) { fIPEffCombinedSamples = ipEffCombinedSamples; }
84	void SetHadronEffbyIPcut(THnSparseF* hsHadronEffbyIPcut) { fHadronEffbyIPcut = hsHadronEffbyIPcut;};
85	void SetNonHFEsyst(Bool_t syst){ fNonHFEsyst = syst; };
86
87	void SetDumpToFile(Bool_t dumpToFile) { fDumpToFile=dumpToFile; };
88
89	void SetDebugLevel(Int_t debugLevel, Bool_t writeToFile = kFALSE) { fDebugLevel = debugLevel; fWriteToFile = writeToFile; };
90	void SetUnfoldBG() { fUnfoldBG = kTRUE; };
91
92
93	AliCFDataGrid* GetRawBspectra2ndMethod();
94	AliCFDataGrid* GetCharmBackground();
95	AliCFDataGrid* GetNonHFEBackground(Int_t decay = 0, Int_t source = 0);
96	THnSparse* GetCharmWeights(Int_t centBin);
97	THnSparse* GetBeautyIPEff(Bool_t isMCpt);
98	THnSparse* GetPIDxIPEff(Int_t source);
99	void CalculateNonHFEsyst();
100
101	void EnableIPanaHadronBgSubtract() { fIPanaHadronBgSubtract = kTRUE; };
102	void EnableIPanaCharmBgSubtract() { fIPanaCharmBgSubtract = kTRUE; };
103	void EnableIPanaNonHFEBgSubtract() { fIPanaNonHFEBgSubtract = kTRUE; };
104	void EnableIPParameterizedEff() { fIPParameterizedEff = kTRUE; };
105
106	protected:
107	void AddTemporaryObject(TObject *cont);
108	void ClearObject(TObject *o);
109
110	private:
111	AliHFEBeautySpectrum(const AliHFEBeautySpectrum &ref);
112	AliHFEBeautySpectrum &operator=(const AliHFEBeautySpectrum &ref);
113	virtual void Copy(TObject &o) const;
114
115	AliHFEBeautySpectrumQA *fQA; // Beauty QA
116
117	//shift some of the following to QA or Base class:
118	TList *fTemporaryObjects; // Emulate garbage collection
119	AliCFDataGrid *fBackground; // Background Grid
120	TF1 *fEfficiencyTOFPIDD; // TOF PID efficiency parameterized
121	TF1 *fEfficiencyesdTOFPIDD; // TOF PID efficiency parameterized
122	TF1 *fEfficiencyIPCharmD; // IP efficiency parameterized for charm
123	TF1 *fEfficiencyIPBeautyD; // IP efficiency parameterized for beauty
124	TF1 *fEfficiencyIPBeautyesdD; // IP efficiency parameterized for beauty for esd
125	TF1 *fEfficiencyIPConversionD; // IP efficiency parameterized for conversion
126	TF1 *fEfficiencyIPNonhfeD; // IP efficiency parameterized for nonhfe
127	TF1 *fNonHFEbg; // Efficiency Function
128	THnSparseF *fWeightCharm; // Weight for charm bg
129
130	AliCFContainer *fConvSourceContainer[kElecBgSources][kBgLevels]; //container for conversion electrons, divided into different photon sources
131	AliCFContainer *fNonHFESourceContainer[kElecBgSources][kBgLevels]; //container for non-HF electrons, divided into different sources
132
133	Bool_t fInclusiveSpectrum; // Inclusive Spectrum
134	Bool_t fDumpToFile; // Write Result in a file
135
136	Bool_t fUnSetCorrelatedErrors; // Unset correlated errors
137
138	Bool_t fIPanaHadronBgSubtract; // Hadron background subtraction
139	Bool_t fIPanaCharmBgSubtract; // Charm background subtraction
140	Bool_t fIPanaNonHFEBgSubtract; // nonHFE background subtraction
141	Bool_t fIPParameterizedEff; // switch to use parameterized efficiency for ip analysis
142	Bool_t fNonHFEsyst; // choose NonHFE background level (upper, lower, central)
143	Bool_t fBeauty2ndMethod; // 2nd method to get beauty spectrum
144	Bool_t fIPEffCombinedSamples; // flag to combine two different samples
145
146	Int_t fNMCEvents; // Number of MC Events
147	Int_t fNMCbgEvents; // Number of BG MC Events
148
149	Int_t fNCentralityBinAtTheEnd;// Number of centrality class at the end
150	Int_t fLowBoundaryCentralityBinAtTheEnd[20]; // Boundary of the bins
151	Int_t fHighBoundaryCentralityBinAtTheEnd[20]; // Boundary of the bins
152	Bool_t fFillMoreCorrelationMatrix; // For low stats to have reasonable errors
153
154	THnSparseF *fHadronEffbyIPcut;// container for hadron efficiency by IP cut
155	TH1D *fEfficiencyCharmSigD; // charm IP cut eff from signal enhanced MC
156	TH1D *fEfficiencyBeautySigD; // beauty IP cut eff from signal enhanced MC
157	TH1D *fEfficiencyBeautySigesdD; // beauty IP cut eff from signal enhanced MC for esd
158	TH1D *fConversionEff; // conversion IP cut eff
159	TH1D *fNonHFEEff; // nonhfe IP cut eff
160	TH1D *fCharmEff; // charm IP cut eff
161	TH1D *fBeautyEff; // beauty IP cut eff
162	TH1D *fConversionEffbgc; // conversion IP cut eff
163	TH1D *fNonHFEEffbgc; // nonhfe IP cut eff
164	TH1D *fBSpectrum2ndMethod; // beauty spectrum for 2nd method
165	const char *fkBeauty2ndMethodfilename; // name of file, which contains beauty spectrum for 2ndmethod
166	Char_t fBeamType; // beamtype; default -1; pp =0; PbPb=1
167	Bool_t fEtaSyst; // pp 2.76 TeV (= kTRUE) or 7 TeV (= kFALSE)
168
169
170	Int_t fDebugLevel; // Debug Level
171	Bool_t fWriteToFile; // Write plots to eps files
172	Bool_t fUnfoldBG; // flag to unfold backgroud
173
174	ClassDef(AliHFEBeautySpectrum, 1)
3513afb7	175	};
4bd4fc13	176	#endif
4bd4fc13	177