[u/mrichter/AliRoot.git] / PWG0 / dNdPt / AlidNdPtCorrection.h

#ifndef ALIDNDPTCORRECTION_H
#define ALIDNDPTCORRECTION_H

//------------------------------------------------------------------------------
// AlidNdPtCorrection class to correct and
// normalised dNdPt spectra. 
//
// Author: J.Otwinowski 04/11/2008 
//------------------------------------------------------------------------------

class iostream;
class TFile;
class TCint;
class TFolder;
class TObjArray;
class TString;
class THnSparse;

class AliESDtrackCuts;
class AliVertexerTracks;
class AliESD;
class AliESDfriend;
class AliESDfriendTrack;

class AlidNdPtEventCuts;
class AlidNdPtAcceptanceCuts;
class AlidNdPtCorrection;
class AlidNdPt;
class AlidNdPtHelper;

#include "AlidNdPt.h"

class AlidNdPtCorrection : public AlidNdPt {
public :
  AlidNdPtCorrection(); 
  AlidNdPtCorrection(Char_t* name, Char_t* title, TString corrMatrixFileName="ala.root");
  ~AlidNdPtCorrection();

  // Init data members
  virtual void Init();

  // Process events
  virtual void Process(AliESDEvent *const esdEvent=0, AliMCEvent *const mcEvent=0);

  // Merge output objects (needed by PROOF) 
  virtual Long64_t Merge(TCollection* const list);

  // Analyse output histograms 
  virtual void Analyse();

  // Export objects to folder
  virtual TFolder *ExportToFolder(TObjArray * const array=0);

  // Get analysis folder
  TFolder* GetCorrectionFolder() const {return fCorrectionFolder;}

  // Create folder for analysed histograms
  TFolder *CreateFolder(TString folder = "folderdNdPtCorrection",TString title = "Analysed dNdPt histograms");

  // Fill histograms
  void FillHistograms(AliESDtrack *const esdTrack, AliStack *const stack, AlidNdPtHelper::TrackObject trackObj, Double_t zv, Int_t multRec, Int_t trueMult) const;
  void FillHistograms(AliStack *const stack, Int_t label, AlidNdPtHelper::TrackObject trackObj, Int_t multRec) const;
  void FillHistograms(AlidNdPtHelper::EventObject eventObj, Double_t zv, Int_t multMBRec) const;

  // Get correction factors
  Double_t GetCorrFactZvPtEta(THnSparse *const hist=0, Double_t zv =0, Double_t pt=0, Double_t eta=0) const;
  Double_t GetContFactZvPtEta(THnSparse *const hist=0, Double_t zv =0, Double_t pt=0, Double_t eta=0) const;
  Double_t GetCorrFactZvMult(THnSparse *const hist=0, Double_t zv =0, Int_t mult=0) const;
  Double_t GetContFactZvMult(THnSparse *const hist=0, Double_t zv =0, Int_t mult=0) const;

  // Getters
  THnSparseF *GetMCAllEventMultHist1() const {return fMCAllEventMultHist1;}; 
  THnSparseF *GetMCAllNDEventMultHist1() const {return fMCAllNDEventMultHist1;}; 
  THnSparseF *GetMCAllNSDEventMultHist1() const {return fMCAllNSDEventMultHist1;}; 
  THnSparseF *GetMCTriggerMultHist1() const {return fMCTriggerMultHist1;}; 
  THnSparseF *GetMCEventMultHist1() const {return fMCEventMultHist1;}; 

  THnSparseF *GetMCAllPrimTrackMultHist1() const {return fMCAllPrimTrackMultHist1;}; 
  THnSparseF *GetMCNDEventAllPrimTrackMultHist1() const {return fMCNDEventAllPrimTrackMultHist1;}; 
  THnSparseF *GetMCNSDEventAllPrimTrackMultHist1() const {return fMCNSDEventAllPrimTrackMultHist1;}; 
  THnSparseF *GetMCTriggerPrimTrackMultHist1() const {return fMCTriggerPrimTrackMultHist1;}; 
  THnSparseF *GetMCEventPrimTrackMultHist1() const {return fMCEventPrimTrackMultHist1;}; 

  THnSparseF *GetMCAllPrimTrackTrueMultHist1() const {return fMCAllPrimTrackTrueMultHist1;}; 
  THnSparseF *GetMCNDEventAllPrimTrackTrueMultHist1() const {return fMCNDEventAllPrimTrackTrueMultHist1;}; 
  THnSparseF *GetMCNSDEventAllPrimTrackTrueMultHist1() const {return fMCNSDEventAllPrimTrackTrueMultHist1;}; 
  THnSparseF *GetMCTriggerPrimTrackTrueMultHist1() const {return fMCTriggerPrimTrackTrueMultHist1;}; 
  THnSparseF *GetMCEventPrimTrackTrueMultHist1() const {return fMCEventPrimTrackTrueMultHist1;}; 


  THnSparseF *GetMCAllPrimTrackMeanPtMult1() const {return fMCAllPrimTrackMeanPtMult1;}; 
  THnSparseF *GetMCNDEventAllPrimTrackMeanPtMult1() const {return fMCNDEventAllPrimTrackMeanPtMult1;}; 
  THnSparseF *GetMCNSDEventAllPrimTrackMeanPtMult1() const {return fMCNSDEventAllPrimTrackMeanPtMult1;}; 
  THnSparseF *GetMCTriggerPrimTrackMeanPtMult1() const {return fMCTriggerPrimTrackMeanPtMult1;}; 
  THnSparseF *GetMCEventPrimTrackMeanPtMult1() const {return fMCEventPrimTrackMeanPtMult1;}; 

  THnSparseF *GetMCAllPrimTrackMeanPtTrueMult1() const {return fMCAllPrimTrackMeanPtTrueMult1;}; 
  THnSparseF *GetMCNDEventAllPrimTrackMeanPtTrueMult1() const {return fMCNDEventAllPrimTrackMeanPtTrueMult1;}; 
  THnSparseF *GetMCNSDEventAllPrimTrackMeanPtTrueMult1() const {return fMCNSDEventAllPrimTrackMeanPtTrueMult1;}; 
  THnSparseF *GetMCTriggerPrimTrackMeanPtTrueMult1() const {return fMCTriggerPrimTrackMeanPtTrueMult1;}; 
  THnSparseF *GetMCEventPrimTrackMeanPtTrueMult1() const {return fMCEventPrimTrackMeanPtTrueMult1;};

  THnSparseF *GetCorrRecTrackMultHist1(Int_t i) const {return fCorrRecTrackMultHist1[i];}
  THnSparseF *GetCorrRecTrackTrueMultHist1(Int_t i) const {return fCorrRecTrackTrueMultHist1[i];}
  THnSparseF *GetCorrRecTrackMeanPtMultHist1(Int_t i) const {return fCorrRecTrackMeanPtMultHist1[i];}
  THnSparseF *GetCorrRecTrackMeanPtTrueMultHist1(Int_t i) const {return fCorrRecTrackMeanPtTrueMultHist1[i];}
  THnSparseF *GetCorrRecTrackPt1(Int_t i) const {return fCorrRecTrackPt1[i];}

  THnSparseF *GetCorrRecEventHist1(Int_t i) const {return fCorrRecEventHist1[i];}
  THnSparseF *GetCorrRecEventHist2(Int_t i) const {return fCorrRecEventHist2[i];}


  // correlation matrix
  void SetEventMultCorrelationMatrix(THnSparseF *const matrix=0) {fEventMultCorrelationMatrix = matrix;}
  THnSparseF *GetEventMultCorrelationMatrix() const {return fEventMultCorrelationMatrix;}

  //
  void SetZvNorm(TH1D *const matrix=0) {fZvNorm = matrix;}
  TH1D *GetZvNorm() const {return fZvNorm;}

  void SetZvEmptyEventsNorm(TH1D *const matrix=0) {fZvEmptyEventsNorm = matrix;}
  TH1D *GetZvEmptyEventsNorm() const {return fZvEmptyEventsNorm;}

  // 
  void SetCorrEventMatrix(THnSparseF *const matrix=0) {fCorrEventMatrix = matrix;}
  THnSparseF *GetCorrEventMatrix() const {return fCorrEventMatrix;}

  void SetCorrTriggerMBtoInelEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoInelEventMatrix = matrix;}
  THnSparseF *GetCorrTriggerMBtoInelEventMatrix() const {return fCorrTriggerMBtoInelEventMatrix;}

  void SetCorrTriggerMBtoNDEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoNDEventMatrix = matrix;}
  THnSparseF *GetCorrTriggerMBtoNDEventMatrix() const {return fCorrTriggerMBtoNDEventMatrix;}

  void SetCorrTriggerMBtoNSDEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoNSDEventMatrix = matrix;}
  THnSparseF *GetCorrTriggerMBtoNSDEventMatrix() const {return fCorrTriggerMBtoNSDEventMatrix;}

  // 
  void SetCorrTrackEventMatrix(THnSparseF *const matrix=0) {fCorrTrackEventMatrix = matrix;}
  THnSparseF *GetCorrTrackEventMatrix() const {return fCorrTrackEventMatrix;}

  void SetCorrTriggerMBtoInelTrackEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoInelTrackEventMatrix = matrix;}
  THnSparseF *GetCorrTriggerMBtoInelTrackEventMatrix() const {return fCorrTriggerMBtoInelTrackEventMatrix;}

  void SetCorrTriggerMBtoNDTrackEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoNDTrackEventMatrix = matrix;}
  THnSparseF *GetCorrTriggerMBtoNDTrackEventMatrix() const {return fCorrTriggerMBtoNDTrackEventMatrix;}

  void SetCorrTriggerMBtoNSDTrackEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoNSDTrackEventMatrix = matrix;}
  THnSparseF *GetCorrTriggerMBtoNSDTrackEventMatrix() const {return fCorrTriggerMBtoNSDTrackEventMatrix;}

  void SetCorrTrackMatrix(THnSparseF *const matrix=0) {fCorrTrackMatrix = matrix;}
  THnSparseF *GetCorrTrackMatrix() const {return fCorrTrackMatrix;}

  void SetCorrHighPtTrackMatrix(THnSparseF *const matrix=0) {fCorrHighPtTrackMatrix = matrix;}
  THnSparseF *GetCorrHighPtTrackMatrix() const {return fCorrHighPtTrackMatrix;}

  void SetContTrackMatrix(THnSparseF *const matrix=0) {fContTrackMatrix = matrix;}
  THnSparseF *GetContTrackMatrix() const {return fContTrackMatrix;}

  void SetContMultTrackMatrix(THnSparseF *const matrix=0) {fContMultTrackMatrix = matrix;}
  THnSparseF *GetContMultTrackMatrix() const {return fContMultTrackMatrix;}

  void SetCorrMatrixFileName(TString name="")    { fCorrMatrixFileName = name; }

  Int_t GetTrueMult(THnSparse *const hist=0, Int_t mult=0) const;

private:

  // analysis folder 
  TFolder *fCorrectionFolder; // folder for analysed histograms

  //
  // event histograms
  //
  THnSparseF *fMCEventHist1;  //-> mcXv:mcYv:mcZv
  THnSparseF *fRecEventHist1; //-> Xv:Yv:Zv

  // all MC events
  THnSparseF *fMCAllEventMultHist1; //-> mcZv:multiplicity

  // all ND MC events
  THnSparseF *fMCAllNDEventMultHist1; //-> mcZv:multiplicity

  // all NSD MC events
  THnSparseF *fMCAllNSDEventMultHist1; //-> mcZv:multiplicity

  // MC triggered events
  THnSparseF *fMCTriggerMultHist1; //-> mcZv:multiplicity

  // MC triggered and event vertex
  THnSparseF *fMCEventMultHist1; //-> mcZv:multiplicity

  //
  // track histograms
  //
  
  // all mc primary tracks in acceptance (INEL)
  THnSparseF *fMCAllPrimTrackMultHist1; //-> mcPt:mcEta:multiplicity

  // all mc primary tracks in acceptance (ND events)
  THnSparseF *fMCNDEventAllPrimTrackMultHist1; //-> mcPt:mcEta:multiplicity

  // all mc primary tracks in acceptance (NSD events)
  THnSparseF *fMCNSDEventAllPrimTrackMultHist1; //-> mcPt:mcEta:multiplicity

  // all mc primary tracks in acceptance (triggered events)
  THnSparseF *fMCTriggerPrimTrackMultHist1; //-> mcPt:mcEta:multiplicity

  // mc primary tracks in acceptance (triggered and event vertex reconstructed)
  THnSparseF *fMCEventPrimTrackMultHist1; //-> mcPt:mcEta:multiplicity

  // true multiplicity 

  // all mc primary tracks in acceptance (INEL)
  THnSparseF *fMCAllPrimTrackTrueMultHist1; //-> mcPt:mcEta:true multiplicity

  // all mc primary tracks in acceptance (ND events)
  THnSparseF *fMCNDEventAllPrimTrackTrueMultHist1; //-> mcPt:mcEta:true multiplicity

  // all mc primary tracks in acceptance (NSD events)
  THnSparseF *fMCNSDEventAllPrimTrackTrueMultHist1; //-> mcPt:mcEta:true multiplicity

  // all mc primary tracks in acceptance (triggered events)
  THnSparseF *fMCTriggerPrimTrackTrueMultHist1; //-> mcPt:mcEta:true multiplicity

  // mc primary tracks in acceptance (triggered and event vertex reconstructed)
  THnSparseF *fMCEventPrimTrackTrueMultHist1; //-> mcPt:mcEta:true multiplicity

  //
  // mc <pt> from the event
  //

  THnSparseF *fMCAllPrimTrackMeanPtMult1; //-> <mcPt>:multiplicity
  THnSparseF *fMCNDEventAllPrimTrackMeanPtMult1; //-> <mcPt>:multiplicity
  THnSparseF *fMCNSDEventAllPrimTrackMeanPtMult1; //-> <mcPt>:multiplicity
  THnSparseF *fMCTriggerPrimTrackMeanPtMult1; //-> <mcPt>:multiplicity
  THnSparseF *fMCEventPrimTrackMeanPtMult1; //-> <mcPt>:multiplicity

  // true multiplicity 

  THnSparseF *fMCAllPrimTrackMeanPtTrueMult1; //-> <mcPt>:true multiplicity
  THnSparseF *fMCNDEventAllPrimTrackMeanPtTrueMult1; //-> <mcPt>:true multiplicity
  THnSparseF *fMCNSDEventAllPrimTrackMeanPtTrueMult1; //-> <mcPt>:true multiplicity
  THnSparseF *fMCTriggerPrimTrackMeanPtTrueMult1; //-> <mcPt>:true multiplicity
  THnSparseF *fMCEventPrimTrackMeanPtTrueMult1; //-> <mcPt>:true multiplicity


  // track histograms 
  // [0]=all charged tracks, 
  // [1]=[0]+fiducual volume, 
  // [2]=[1]+after esd track cuts 
  THnSparseF *fRecTrackHist1[AlidNdPtHelper::kCutSteps]; //-> Pt:Eta:Phi

  // corrected track histograms 
  // ([0]-not corrected,
  // [1]=track cont.,
  // [2]=[1]+track eff.,
  // [3]=[2]+multple track,
  // [4]=[3]+event vertex,
  // [5]=[4]+trigger MBtoInel,
  // [6]=[4]+trigger MBtoND, 
  // [7]=[4]+trigger MBToNSD)
  THnSparseF *fCorrRecTrackMultHist1[8]; //-> Pt:Eta:mult corrected histograms 
  THnSparseF *fCorrRecTrackTrueMultHist1[8]; //-> Pt:Eta:trueMult corrected histograms

  // <pt> vs multiplicity from the event
  THnSparseF *fCorrRecTrackMeanPtMultHist1[8]; //-> <Pt>:mult corrected histograms
  THnSparseF *fCorrRecTrackMeanPtTrueMultHist1[8]; //-> <Pt>:trueMult corrected histograms

  THnSparseF *fCorrRecTrackPt1[8]; //-> Pt corrected histograms in the event (helper histograms)

  // corrected event histograms
  // [0]-not corrected, 
  // [1]=event vertex, 
  // [2]=[1]+trigger MBtoInel, 
  // [3]=[1]+trigger MBtoND, 
  // [4]=[1]+trigger MBtoNSD 
  THnSparseF *fCorrRecEventHist1[5]; //-> mcZv:multMB 


  // corrected event histograms (empty events)
  // [0]=not corrected,
  // [1]=trigger/trigger+vertex correction,
  // [2]=[1]+trigger MBtoInel,
  // [3]=[1]+trigger MBtoND,
  // [4]=[1]+trigger MBtoNSD
  THnSparseF *fCorrRecEventHist2[5]; //-> mcZv:multMB


  // 
  // correction matrices (histograms)
  // must be loaded
  //

  // event rec. track vs true track multiplicity correlation matrix 
  THnSparseF *fEventMultCorrelationMatrix; //-> mult:mult_true_tracks

  // histograms needed for empty events corrections
  TH1D *fZvNorm; //-> normalised reconstructed zv distribution
  TH1D *fZvEmptyEventsNorm; //-> trigger/trigger+vertex empty event correction

  // trigger bias corrections
  THnSparseF *fCorrTriggerMBtoInelEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerEventMatrix / fGenEventMatrix)
  THnSparseF *fCorrTriggerMBtoNDEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerEventMatrix / fGenNDEventMatrix)
  THnSparseF *fCorrTriggerMBtoNSDEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerEventMatrix / fGenNSDEventMatrix)

  // event vertex corrections
  THnSparseF *fCorrEventMatrix;  //-> mcVz:mcPt:mcEta   (fRecEventMatrix/fTriggerEventMatrix)

  // track-event trigger bias corrections
  THnSparseF *fCorrTriggerMBtoInelTrackEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerTrackEventMatrix / fGenTrackEventMatrix)
  THnSparseF *fCorrTriggerMBtoNDTrackEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerTrackEventMatrix / fGenTrackEventNDMatrix)
  THnSparseF *fCorrTriggerMBtoNSDTrackEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerTrackEventMatrix / fGenTrackEventNSDMatrix)

  // track-event vertex corrections
  THnSparseF *fCorrTrackEventMatrix;  //-> mcVz:mcPt:mcEta   (fRecTrackEventMatrix/fTriggerTrackEventMatrix)

  // track-level corrections
  THnSparseF *fCorrTrackMatrix;  //-> mcVz:mcPt:mcEta (fRecPrimTrackMatrix / fGenPrimTrackMatrix)
  THnSparseF *fCorrHighPtTrackMatrix;  //-> mcVz:mcPt:mcEta (fRecPrimTrackMatrix / fGenPrimTrackMatrix high pt tracks)
  THnSparseF *fContTrackMatrix;  //-> mcVz:mcPt:mcEta (fRecSecTrackMatrix / fRecTrackMatrix)
  THnSparseF *fContMultTrackMatrix; //-> mcVz:mcPt:mcEta (fRecMultTrackMatrix / fRecTrackMatrix)
  
  TString fCorrMatrixFileName; // file with efficiency correction matrices

  //  deta, dphi, pt1 for cosmics
  THnSparseF *fCosmicsHisto; //-> deta:dphi:pt
  
  AlidNdPtCorrection(const AlidNdPtCorrection&); // not implemented
  AlidNdPtCorrection& operator=(const AlidNdPtCorrection&); // not implemented

  ClassDef(AlidNdPtCorrection,1);
};

#endif
Commit	Line	Data
00b81913	1	#ifndef ALIDNDPTCORRECTION_H
00b81913	2	#define ALIDNDPTCORRECTION_H
0aaa8b91	3
0aaa8b91	4	//------------------------------------------------------------------------------
00b81913	5	// AlidNdPtCorrection class to correct and
00b81913	6	// normalised dNdPt spectra.
0aaa8b91	7	//
0aaa8b91	8	// Author: J.Otwinowski 04/11/2008
	9	//------------------------------------------------------------------------------
	10
00b81913	11	class iostream;
	12	class TFile;
	13	class TCint;
0aaa8b91	14	class TFolder;
	15	class TObjArray;
	16	class TString;
00b81913	17	class THnSparse;
0aaa8b91	18
	19	class AliESDtrackCuts;
	20	class AliVertexerTracks;
	21	class AliESD;
	22	class AliESDfriend;
	23	class AliESDfriendTrack;
	24
	25	class AlidNdPtEventCuts;
	26	class AlidNdPtAcceptanceCuts;
	27	class AlidNdPtCorrection;
	28	class AlidNdPt;
bad4ba69	29	class AlidNdPtHelper;
0aaa8b91	30
0aaa8b91	31	#include "AlidNdPt.h"
0aaa8b91	32
	33	class AlidNdPtCorrection : public AlidNdPt {
	34	public :
	35	AlidNdPtCorrection();
	36	AlidNdPtCorrection(Char_t* name, Char_t* title, TString corrMatrixFileName="ala.root");
	37	~AlidNdPtCorrection();
	38
	39	// Init data members
	40	virtual void Init();
	41
	42	// Process events
	43	virtual void Process(AliESDEvent const esdEvent=0, AliMCEvent const mcEvent=0);
	44
	45	// Merge output objects (needed by PROOF)
00b81913	46	virtual Long64_t Merge(TCollection* const list);
0aaa8b91	47
	48	// Analyse output histograms
	49	virtual void Analyse();
	50
	51	// Export objects to folder
c931948b	52	virtual TFolder ExportToFolder(TObjArray const array=0);
0aaa8b91	53
0aaa8b91	54	// Get analysis folder
00b81913	55	TFolder* GetCorrectionFolder() const {return fCorrectionFolder;}
0aaa8b91	56
	57	// Create folder for analysed histograms
	58	TFolder *CreateFolder(TString folder = "folderdNdPtCorrection",TString title = "Analysed dNdPt histograms");
	59
	60	// Fill histograms
15b7cc05	61	void FillHistograms(AliESDtrack const esdTrack, AliStack const stack, AlidNdPtHelper::TrackObject trackObj, Double_t zv, Int_t multRec, Int_t trueMult) const;
00b81913	62	void FillHistograms(AliStack *const stack, Int_t label, AlidNdPtHelper::TrackObject trackObj, Int_t multRec) const;
00b81913	63	void FillHistograms(AlidNdPtHelper::EventObject eventObj, Double_t zv, Int_t multMBRec) const;
0aaa8b91	64
	65	// Get correction factors
	66	Double_t GetCorrFactZvPtEta(THnSparse *const hist=0, Double_t zv =0, Double_t pt=0, Double_t eta=0) const;
	67	Double_t GetContFactZvPtEta(THnSparse *const hist=0, Double_t zv =0, Double_t pt=0, Double_t eta=0) const;
	68	Double_t GetCorrFactZvMult(THnSparse *const hist=0, Double_t zv =0, Int_t mult=0) const;
	69	Double_t GetContFactZvMult(THnSparse *const hist=0, Double_t zv =0, Int_t mult=0) const;
	70
	71	// Getters
00b81913	72	THnSparseF *GetMCAllEventMultHist1() const {return fMCAllEventMultHist1;};
	73	THnSparseF *GetMCAllNDEventMultHist1() const {return fMCAllNDEventMultHist1;};
	74	THnSparseF *GetMCAllNSDEventMultHist1() const {return fMCAllNSDEventMultHist1;};
	75	THnSparseF *GetMCTriggerMultHist1() const {return fMCTriggerMultHist1;};
	76	THnSparseF *GetMCEventMultHist1() const {return fMCEventMultHist1;};
0aaa8b91	77
00b81913	78	THnSparseF *GetMCAllPrimTrackMultHist1() const {return fMCAllPrimTrackMultHist1;};
	79	THnSparseF *GetMCNDEventAllPrimTrackMultHist1() const {return fMCNDEventAllPrimTrackMultHist1;};
	80	THnSparseF *GetMCNSDEventAllPrimTrackMultHist1() const {return fMCNSDEventAllPrimTrackMultHist1;};
	81	THnSparseF *GetMCTriggerPrimTrackMultHist1() const {return fMCTriggerPrimTrackMultHist1;};
	82	THnSparseF *GetMCEventPrimTrackMultHist1() const {return fMCEventPrimTrackMultHist1;};
0aaa8b91	83
15b7cc05	84	THnSparseF *GetMCAllPrimTrackTrueMultHist1() const {return fMCAllPrimTrackTrueMultHist1;};
	85	THnSparseF *GetMCNDEventAllPrimTrackTrueMultHist1() const {return fMCNDEventAllPrimTrackTrueMultHist1;};
	86	THnSparseF *GetMCNSDEventAllPrimTrackTrueMultHist1() const {return fMCNSDEventAllPrimTrackTrueMultHist1;};
	87	THnSparseF *GetMCTriggerPrimTrackTrueMultHist1() const {return fMCTriggerPrimTrackTrueMultHist1;};
	88	THnSparseF *GetMCEventPrimTrackTrueMultHist1() const {return fMCEventPrimTrackTrueMultHist1;};
	89
	90
	91	THnSparseF *GetMCAllPrimTrackMeanPtMult1() const {return fMCAllPrimTrackMeanPtMult1;};
	92	THnSparseF *GetMCNDEventAllPrimTrackMeanPtMult1() const {return fMCNDEventAllPrimTrackMeanPtMult1;};
	93	THnSparseF *GetMCNSDEventAllPrimTrackMeanPtMult1() const {return fMCNSDEventAllPrimTrackMeanPtMult1;};
	94	THnSparseF *GetMCTriggerPrimTrackMeanPtMult1() const {return fMCTriggerPrimTrackMeanPtMult1;};
	95	THnSparseF *GetMCEventPrimTrackMeanPtMult1() const {return fMCEventPrimTrackMeanPtMult1;};
	96
	97	THnSparseF *GetMCAllPrimTrackMeanPtTrueMult1() const {return fMCAllPrimTrackMeanPtTrueMult1;};
	98	THnSparseF *GetMCNDEventAllPrimTrackMeanPtTrueMult1() const {return fMCNDEventAllPrimTrackMeanPtTrueMult1;};
	99	THnSparseF *GetMCNSDEventAllPrimTrackMeanPtTrueMult1() const {return fMCNSDEventAllPrimTrackMeanPtTrueMult1;};
	100	THnSparseF *GetMCTriggerPrimTrackMeanPtTrueMult1() const {return fMCTriggerPrimTrackMeanPtTrueMult1;};
	101	THnSparseF *GetMCEventPrimTrackMeanPtTrueMult1() const {return fMCEventPrimTrackMeanPtTrueMult1;};
	102
00b81913	103	THnSparseF *GetCorrRecTrackMultHist1(Int_t i) const {return fCorrRecTrackMultHist1[i];}
15b7cc05	104	THnSparseF *GetCorrRecTrackTrueMultHist1(Int_t i) const {return fCorrRecTrackTrueMultHist1[i];}
	105	THnSparseF *GetCorrRecTrackMeanPtMultHist1(Int_t i) const {return fCorrRecTrackMeanPtMultHist1[i];}
	106	THnSparseF *GetCorrRecTrackMeanPtTrueMultHist1(Int_t i) const {return fCorrRecTrackMeanPtTrueMultHist1[i];}
	107	THnSparseF *GetCorrRecTrackPt1(Int_t i) const {return fCorrRecTrackPt1[i];}
	108
00b81913	109	THnSparseF *GetCorrRecEventHist1(Int_t i) const {return fCorrRecEventHist1[i];}
00b81913	110	THnSparseF *GetCorrRecEventHist2(Int_t i) const {return fCorrRecEventHist2[i];}
0aaa8b91	111
0aaa8b91	112
	113	// correlation matrix
	114	void SetEventMultCorrelationMatrix(THnSparseF *const matrix=0) {fEventMultCorrelationMatrix = matrix;}
00b81913	115	THnSparseF *GetEventMultCorrelationMatrix() const {return fEventMultCorrelationMatrix;}
0aaa8b91	116
	117	//
	118	void SetZvNorm(TH1D *const matrix=0) {fZvNorm = matrix;}
00b81913	119	TH1D *GetZvNorm() const {return fZvNorm;}
0aaa8b91	120
0aaa8b91	121	void SetZvEmptyEventsNorm(TH1D *const matrix=0) {fZvEmptyEventsNorm = matrix;}
00b81913	122	TH1D *GetZvEmptyEventsNorm() const {return fZvEmptyEventsNorm;}
0aaa8b91	123
	124	//
	125	void SetCorrEventMatrix(THnSparseF *const matrix=0) {fCorrEventMatrix = matrix;}
00b81913	126	THnSparseF *GetCorrEventMatrix() const {return fCorrEventMatrix;}
0aaa8b91	127
0aaa8b91	128	void SetCorrTriggerMBtoInelEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoInelEventMatrix = matrix;}
00b81913	129	THnSparseF *GetCorrTriggerMBtoInelEventMatrix() const {return fCorrTriggerMBtoInelEventMatrix;}
0aaa8b91	130
0aaa8b91	131	void SetCorrTriggerMBtoNDEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoNDEventMatrix = matrix;}
00b81913	132	THnSparseF *GetCorrTriggerMBtoNDEventMatrix() const {return fCorrTriggerMBtoNDEventMatrix;}
0aaa8b91	133
0aaa8b91	134	void SetCorrTriggerMBtoNSDEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoNSDEventMatrix = matrix;}
00b81913	135	THnSparseF *GetCorrTriggerMBtoNSDEventMatrix() const {return fCorrTriggerMBtoNSDEventMatrix;}
0aaa8b91	136
	137	//
	138	void SetCorrTrackEventMatrix(THnSparseF *const matrix=0) {fCorrTrackEventMatrix = matrix;}
00b81913	139	THnSparseF *GetCorrTrackEventMatrix() const {return fCorrTrackEventMatrix;}
0aaa8b91	140
0aaa8b91	141	void SetCorrTriggerMBtoInelTrackEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoInelTrackEventMatrix = matrix;}
00b81913	142	THnSparseF *GetCorrTriggerMBtoInelTrackEventMatrix() const {return fCorrTriggerMBtoInelTrackEventMatrix;}
0aaa8b91	143
0aaa8b91	144	void SetCorrTriggerMBtoNDTrackEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoNDTrackEventMatrix = matrix;}
00b81913	145	THnSparseF *GetCorrTriggerMBtoNDTrackEventMatrix() const {return fCorrTriggerMBtoNDTrackEventMatrix;}
0aaa8b91	146
0aaa8b91	147	void SetCorrTriggerMBtoNSDTrackEventMatrix(THnSparseF *const matrix=0) {fCorrTriggerMBtoNSDTrackEventMatrix = matrix;}
00b81913	148	THnSparseF *GetCorrTriggerMBtoNSDTrackEventMatrix() const {return fCorrTriggerMBtoNSDTrackEventMatrix;}
0aaa8b91	149
0aaa8b91	150	void SetCorrTrackMatrix(THnSparseF *const matrix=0) {fCorrTrackMatrix = matrix;}
00b81913	151	THnSparseF *GetCorrTrackMatrix() const {return fCorrTrackMatrix;}
0aaa8b91	152
15b7cc05	153	void SetCorrHighPtTrackMatrix(THnSparseF *const matrix=0) {fCorrHighPtTrackMatrix = matrix;}
	154	THnSparseF *GetCorrHighPtTrackMatrix() const {return fCorrHighPtTrackMatrix;}
	155
0aaa8b91	156	void SetContTrackMatrix(THnSparseF *const matrix=0) {fContTrackMatrix = matrix;}
00b81913	157	THnSparseF *GetContTrackMatrix() const {return fContTrackMatrix;}
0aaa8b91	158
0aaa8b91	159	void SetContMultTrackMatrix(THnSparseF *const matrix=0) {fContMultTrackMatrix = matrix;}
00b81913	160	THnSparseF *GetContMultTrackMatrix() const {return fContMultTrackMatrix;}
0aaa8b91	161
	162	void SetCorrMatrixFileName(TString name="") { fCorrMatrixFileName = name; }
	163
00b81913	164	Int_t GetTrueMult(THnSparse *const hist=0, Int_t mult=0) const;
0aaa8b91	165
	166	private:
	167
	168	// analysis folder
	169	TFolder *fCorrectionFolder; // folder for analysed histograms
	170
	171	//
	172	// event histograms
	173	//
	174	THnSparseF *fMCEventHist1; //-> mcXv:mcYv:mcZv
	175	THnSparseF *fRecEventHist1; //-> Xv:Yv:Zv
	176
	177	// all MC events
	178	THnSparseF *fMCAllEventMultHist1; //-> mcZv:multiplicity
	179
	180	// all ND MC events
	181	THnSparseF *fMCAllNDEventMultHist1; //-> mcZv:multiplicity
	182
	183	// all NSD MC events
	184	THnSparseF *fMCAllNSDEventMultHist1; //-> mcZv:multiplicity
	185
	186	// MC triggered events
	187	THnSparseF *fMCTriggerMultHist1; //-> mcZv:multiplicity
	188
	189	// MC triggered and event vertex
	190	THnSparseF *fMCEventMultHist1; //-> mcZv:multiplicity
	191
	192	//
	193	// track histograms
	194	//
	195
	196	// all mc primary tracks in acceptance (INEL)
	197	THnSparseF *fMCAllPrimTrackMultHist1; //-> mcPt:mcEta:multiplicity
	198
	199	// all mc primary tracks in acceptance (ND events)
	200	THnSparseF *fMCNDEventAllPrimTrackMultHist1; //-> mcPt:mcEta:multiplicity
	201
	202	// all mc primary tracks in acceptance (NSD events)
	203	THnSparseF *fMCNSDEventAllPrimTrackMultHist1; //-> mcPt:mcEta:multiplicity
	204
15b7cc05	205	// all mc primary tracks in acceptance (triggered events)
0aaa8b91	206	THnSparseF *fMCTriggerPrimTrackMultHist1; //-> mcPt:mcEta:multiplicity
0aaa8b91	207
15b7cc05	208	// mc primary tracks in acceptance (triggered and event vertex reconstructed)
0aaa8b91	209	THnSparseF *fMCEventPrimTrackMultHist1; //-> mcPt:mcEta:multiplicity
0aaa8b91	210
15b7cc05	211	// true multiplicity
	212
	213	// all mc primary tracks in acceptance (INEL)
	214	THnSparseF *fMCAllPrimTrackTrueMultHist1; //-> mcPt:mcEta:true multiplicity
	215
	216	// all mc primary tracks in acceptance (ND events)
	217	THnSparseF *fMCNDEventAllPrimTrackTrueMultHist1; //-> mcPt:mcEta:true multiplicity
	218
	219	// all mc primary tracks in acceptance (NSD events)
	220	THnSparseF *fMCNSDEventAllPrimTrackTrueMultHist1; //-> mcPt:mcEta:true multiplicity
	221
	222	// all mc primary tracks in acceptance (triggered events)
	223	THnSparseF *fMCTriggerPrimTrackTrueMultHist1; //-> mcPt:mcEta:true multiplicity
	224
	225	// mc primary tracks in acceptance (triggered and event vertex reconstructed)
	226	THnSparseF *fMCEventPrimTrackTrueMultHist1; //-> mcPt:mcEta:true multiplicity
	227
	228	//
	229	// mc <pt> from the event
	230	//
	231
	232	THnSparseF *fMCAllPrimTrackMeanPtMult1; //-> <mcPt>:multiplicity
	233	THnSparseF *fMCNDEventAllPrimTrackMeanPtMult1; //-> <mcPt>:multiplicity
	234	THnSparseF *fMCNSDEventAllPrimTrackMeanPtMult1; //-> <mcPt>:multiplicity
	235	THnSparseF *fMCTriggerPrimTrackMeanPtMult1; //-> <mcPt>:multiplicity
	236	THnSparseF *fMCEventPrimTrackMeanPtMult1; //-> <mcPt>:multiplicity
	237
	238	// true multiplicity
	239
	240	THnSparseF *fMCAllPrimTrackMeanPtTrueMult1; //-> <mcPt>:true multiplicity
	241	THnSparseF *fMCNDEventAllPrimTrackMeanPtTrueMult1; //-> <mcPt>:true multiplicity
	242	THnSparseF *fMCNSDEventAllPrimTrackMeanPtTrueMult1; //-> <mcPt>:true multiplicity
	243	THnSparseF *fMCTriggerPrimTrackMeanPtTrueMult1; //-> <mcPt>:true multiplicity
	244	THnSparseF *fMCEventPrimTrackMeanPtTrueMult1; //-> <mcPt>:true multiplicity
	245
0aaa8b91	246
	247	// track histograms
	248	// [0]=all charged tracks,
	249	// [1]=[0]+fiducual volume,
	250	// [2]=[1]+after esd track cuts
	251	THnSparseF *fRecTrackHist1[AlidNdPtHelper::kCutSteps]; //-> Pt:Eta:Phi
	252
	253	// corrected track histograms
	254	// ([0]-not corrected,
	255	// [1]=track cont.,
	256	// [2]=[1]+track eff.,
	257	// [3]=[2]+multple track,
	258	// [4]=[3]+event vertex,
	259	// [5]=[4]+trigger MBtoInel,
	260	// [6]=[4]+trigger MBtoND,
	261	// [7]=[4]+trigger MBToNSD)
	262	THnSparseF *fCorrRecTrackMultHist1[8]; //-> Pt:Eta:mult corrected histograms
15b7cc05	263	THnSparseF *fCorrRecTrackTrueMultHist1[8]; //-> Pt:Eta:trueMult corrected histograms
	264
	265	// <pt> vs multiplicity from the event
	266	THnSparseF *fCorrRecTrackMeanPtMultHist1[8]; //-> <Pt>:mult corrected histograms
	267	THnSparseF *fCorrRecTrackMeanPtTrueMultHist1[8]; //-> <Pt>:trueMult corrected histograms
0aaa8b91	268
15b7cc05	269	THnSparseF *fCorrRecTrackPt1[8]; //-> Pt corrected histograms in the event (helper histograms)
bad4ba69	270
0aaa8b91	271	// corrected event histograms
	272	// [0]-not corrected,
	273	// [1]=event vertex,
	274	// [2]=[1]+trigger MBtoInel,
	275	// [3]=[1]+trigger MBtoND,
	276	// [4]=[1]+trigger MBtoNSD
	277	THnSparseF *fCorrRecEventHist1[5]; //-> mcZv:multMB
	278
bad4ba69	279
0aaa8b91	280	// corrected event histograms (empty events)
	281	// [0]=not corrected,
	282	// [1]=trigger/trigger+vertex correction,
	283	// [2]=[1]+trigger MBtoInel,
	284	// [3]=[1]+trigger MBtoND,
	285	// [4]=[1]+trigger MBtoNSD
	286	THnSparseF *fCorrRecEventHist2[5]; //-> mcZv:multMB
	287
	288
	289	//
	290	// correction matrices (histograms)
	291	// must be loaded
	292	//
	293
	294	// event rec. track vs true track multiplicity correlation matrix
	295	THnSparseF *fEventMultCorrelationMatrix; //-> mult:mult_true_tracks
	296
	297	// histograms needed for empty events corrections
	298	TH1D *fZvNorm; //-> normalised reconstructed zv distribution
	299	TH1D *fZvEmptyEventsNorm; //-> trigger/trigger+vertex empty event correction
	300
	301	// trigger bias corrections
	302	THnSparseF *fCorrTriggerMBtoInelEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerEventMatrix / fGenEventMatrix)
	303	THnSparseF *fCorrTriggerMBtoNDEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerEventMatrix / fGenNDEventMatrix)
	304	THnSparseF *fCorrTriggerMBtoNSDEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerEventMatrix / fGenNSDEventMatrix)
	305
	306	// event vertex corrections
	307	THnSparseF *fCorrEventMatrix; //-> mcVz:mcPt:mcEta (fRecEventMatrix/fTriggerEventMatrix)
	308
	309	// track-event trigger bias corrections
	310	THnSparseF *fCorrTriggerMBtoInelTrackEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerTrackEventMatrix / fGenTrackEventMatrix)
	311	THnSparseF *fCorrTriggerMBtoNDTrackEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerTrackEventMatrix / fGenTrackEventNDMatrix)
	312	THnSparseF *fCorrTriggerMBtoNSDTrackEventMatrix; //-> mcVz:mcPt:mcEta (fTriggerTrackEventMatrix / fGenTrackEventNSDMatrix)
	313
	314	// track-event vertex corrections
	315	THnSparseF *fCorrTrackEventMatrix; //-> mcVz:mcPt:mcEta (fRecTrackEventMatrix/fTriggerTrackEventMatrix)
	316
	317	// track-level corrections
	318	THnSparseF *fCorrTrackMatrix; //-> mcVz:mcPt:mcEta (fRecPrimTrackMatrix / fGenPrimTrackMatrix)
15b7cc05	319	THnSparseF *fCorrHighPtTrackMatrix; //-> mcVz:mcPt:mcEta (fRecPrimTrackMatrix / fGenPrimTrackMatrix high pt tracks)
0aaa8b91	320	THnSparseF *fContTrackMatrix; //-> mcVz:mcPt:mcEta (fRecSecTrackMatrix / fRecTrackMatrix)
	321	THnSparseF *fContMultTrackMatrix; //-> mcVz:mcPt:mcEta (fRecMultTrackMatrix / fRecTrackMatrix)
	322
	323	TString fCorrMatrixFileName; // file with efficiency correction matrices
d269b0e6	324
	325	// deta, dphi, pt1 for cosmics
	326	THnSparseF *fCosmicsHisto; //-> deta:dphi:pt
0aaa8b91	327
	328	AlidNdPtCorrection(const AlidNdPtCorrection&); // not implemented
	329	AlidNdPtCorrection& operator=(const AlidNdPtCorrection&); // not implemented
	330
	331	ClassDef(AlidNdPtCorrection,1);
	332	};
	333
	334	#endif