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

#ifndef ALIDNDPTANALYSIS_H
#define ALIDNDPTANALYSIS_H

//------------------------------------------------------------------------------
// AlidNdPtAnalysis class used for dNdPt analysis. 
// 
// Author: J.Otwinowski 04/11/2008 
// last change: 2011-04-04 by M.Knichel
//------------------------------------------------------------------------------

class iostream;

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

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

#include "AlidNdPt.h"

class AlidNdPtAnalysis : public AlidNdPt {
public :
  AlidNdPtAnalysis(); 
  AlidNdPtAnalysis(Char_t* name, Char_t* title);
  ~AlidNdPtAnalysis();

  // 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* GetAnalysisFolder() const {return fAnalysisFolder;}

  // Fill control histograms
  void SetHistogramsOn(const Bool_t histOn=kTRUE) {fHistogramsOn = histOn;}
  Bool_t IsHistogramsOn() const {return fHistogramsOn;}

  // Define 0-multiplicity bin for LHC
  // background calculation
  static Bool_t IsBinZeroSPDvtx(const AliESDEvent* esdEvent);
  static Bool_t IsBinZeroTrackSPDvtx(const AliESDEvent* esdEvent);
    
  // Create folder for analysed histograms
  TFolder *CreateFolder(TString folder = "folderdNdPtAnalysis",TString title = "Analysed dNdPt histograms");

  // Set binning for Histograms (if not set default binning is used)
  void SetBinsMult(Int_t nbins, Double_t* edges) { fMultNbins = nbins; fBinsMult = CloneArray(nbins+1,edges); }
  void SetBinsPt(Int_t nbins, Double_t* edges) { fPtNbins = nbins; fBinsPt = CloneArray(nbins+1,edges); }
  void SetBinsPtCorr(Int_t nbins, Double_t* edges) { fPtCorrNbins = nbins; fBinsPtCorr = CloneArray(nbins+1,edges); }
  void SetBinsEta(Int_t nbins, Double_t* edges) { fEtaNbins = nbins; fBinsEta = CloneArray(nbins+1,edges); }
  void SetBinsZv(Int_t nbins, Double_t* edges) { fZvNbins = nbins; fBinsZv = CloneArray(nbins+1,edges); }


  // Fill histograms
  void FillHistograms(AliESDtrack *const esdTrack, AliStack *const stack, const Double_t zv, AlidNdPtHelper::TrackObject trackObj, Int_t multMB);
  void FillHistograms(AliStack *const stack, Int_t label, AlidNdPtHelper::TrackObject trackObj);
  void FillHistograms(TObjArray *const allChargedTracks,Int_t *const labelsAll,Int_t multAll,Int_t *const labelsAcc,Int_t multAcc,Int_t *const labelsRec,Int_t multRec);

  // Getters
  THnSparseF *GetEventMultCorrelationMatrix() const {return fEventMultCorrelationMatrix;}
  THnSparseF *GetTrackPtCorrelationMatrix()   const {return fTrackPtCorrelationMatrix;}

  //
  THnSparseF *GetGenEventMatrix() const {return fGenEventMatrix;}
  THnSparseF *GetGenSDEventMatrix() const {return fGenSDEventMatrix;}
  THnSparseF *GetGenDDEventMatrix() const {return fGenDDEventMatrix;}
  THnSparseF *GetGenNDEventMatrix() const {return fGenNDEventMatrix;}
  THnSparseF *GetGenNSDEventMatrix() const {return fGenNSDEventMatrix;}

  THnSparseF *GetTriggerEventMatrix() const {return fTriggerEventMatrix;}
  THnSparseF *GetTriggerSDEventMatrix() const {return fTriggerSDEventMatrix;}
  THnSparseF *GetTriggerDDEventMatrix() const {return fTriggerDDEventMatrix;}
  THnSparseF *GetTriggerNDEventMatrix() const {return fTriggerNDEventMatrix;}
  THnSparseF *GetTriggerNSDEventMatrix() const {return fTriggerNSDEventMatrix;}

  THnSparseF *GetRecEventMatrix() const {return fRecEventMatrix;}
  THnSparseF *GetRecSDEventMatrix() const {return fRecSDEventMatrix;}
  THnSparseF *GetRecDDEventMatrix() const {return fRecDDEventMatrix;}
  THnSparseF *GetRecNDEventMatrix() const {return fRecNDEventMatrix;}
  THnSparseF *GetRecNSDEventMatrix() const {return fRecNSDEventMatrix;}

  //
  THnSparseF *GetRecCandleEventMatrix() const {return fRecCandleEventMatrix;}

  // 
  THnSparseF *GetGenTrackEventMatrix() const {return fGenTrackEventMatrix;}
  THnSparseF *GetGenTrackSDEventMatrix() const {return fGenTrackSDEventMatrix;}
  THnSparseF *GetGenTrackDDEventMatrix() const {return fGenTrackDDEventMatrix;}
  THnSparseF *GetGenTrackNDEventMatrix() const {return fGenTrackNDEventMatrix;}
  THnSparseF *GetGenTrackNSDEventMatrix() const {return fGenTrackNSDEventMatrix;}

  THnSparseF *GetTriggerTrackEventMatrix() const {return fTriggerTrackEventMatrix;}
  THnSparseF *GetTriggerTrackSDEventMatrix() const {return fTriggerTrackSDEventMatrix;}
  THnSparseF *GetTriggerTrackDDEventMatrix() const {return fTriggerTrackDDEventMatrix;}
  THnSparseF *GetTriggerTrackNDEventMatrix() const {return fTriggerTrackNDEventMatrix;}
  THnSparseF *GetTriggerTrackNSDEventMatrix() const {return fTriggerTrackNSDEventMatrix;}

  THnSparseF *GetRecTrackEventMatrix() const {return fRecTrackEventMatrix;}
  THnSparseF *GetRecTrackSDEventMatrix() const {return fRecTrackSDEventMatrix;}
  THnSparseF *GetRecTrackDDEventMatrix() const {return fRecTrackDDEventMatrix;}
  THnSparseF *GetRecTrackNDEventMatrix() const {return fRecTrackNDEventMatrix;}
  THnSparseF *GetRecTrackNSDEventMatrix() const {return fRecTrackNSDEventMatrix;}

  //
  THnSparseF *GetGenTrackMatrix() const {return fGenTrackMatrix;}
  THnSparseF *GetGenPrimTrackMatrix() const {return fGenPrimTrackMatrix;}
  THnSparseF *GetRecPrimTrackMatrix() const {return fRecPrimTrackMatrix;}

  THnSparseF *GetRecTrackMatrix() const {return fRecTrackMatrix;}
  THnSparseF *GetRecSecTrackMatrix() const {return fRecSecTrackMatrix;}
  THnSparseF *GetRecMultTrackMatrix() const {return fRecMultTrackMatrix;}

  //
  // control histograms
  //
  THnSparseF *GetMCEventHist1() const {return fMCEventHist1;}
  THnSparseF *GetRecEventHist1() const {return fRecEventHist1;}
  THnSparseF *GetRecEventHist2() const {return fRecEventHist2;}
  THnSparseF *GetRecMCEventHist1() const {return fRecMCEventHist1;}
  THnSparseF *GetRecMCEventHist2() const {return fRecMCEventHist2;}
  THnSparseF *GetRecMCEventHist3() const {return fRecMCEventHist3;}

  THnSparseF *GetMCTrackHist1(Int_t i) const {return fMCTrackHist1[i];}
  THnSparseF *GetMCPrimTrackHist1(Int_t i) const {return fMCPrimTrackHist1[i];}
  THnSparseF *GetMCPrimTrackHist2(Int_t i) const {return fMCPrimTrackHist2[i];}
  THnSparseF *GetMCSecTrackHist1(Int_t i) const {return fMCSecTrackHist1[i];}

  THnSparseF *GetRecTrackHist1(Int_t i) const {return fRecTrackHist1[i];}
  THnSparseF *GetRecTrackMultHist1(Int_t i) const {return fRecTrackMultHist1[i];}

  THnSparseF *GetRecMCTrackHist1() const {return fRecMCTrackHist1;}
  THnSparseF *GetMCMultRecTrackHist1() const {return fMCMultRecTrackHist1;}

  THnSparseF *GetRecTrackHist2() const {return fRecTrackHist2;}
  THnSparseF *GetEventCount() const {return fEventCount;}  

  //
  // Generic histograms to be corrected
  //
  THnSparseF *GetRecEventHist() const {return fRecEventHist;} 
  THnSparseF *GetRecTrackHist() const {return fRecTrackHist;} 

private:

  // analysis folder 
  TFolder *fAnalysisFolder; // folder for analysed histograms
  Bool_t fHistogramsOn; // switch on/off filling of control histograms 

  // 
  // correlation matrices (histograms)
  //

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

  // rec. track pt vs true track pt correlation matrix for given eta
  THnSparseF *fTrackPtCorrelationMatrix; //-> Pt:mcPt:mcEta

  //
  // event level correction 
  //

  // all genertated
  THnSparseF *fGenEventMatrix; //-> mcZv:multMB (inelastic)
  THnSparseF *fGenSDEventMatrix; //-> mcZv:multMB (single diffractive)
  THnSparseF *fGenDDEventMatrix; //-> mcZv:multMB (single diffractive)
  THnSparseF *fGenNDEventMatrix; //-> mcZv:multMB (non diffractive)
  THnSparseF *fGenNSDEventMatrix; //-> mcZv:multMB (non single diffractive)

  // trigger bias corrections (fTriggerEventMatrix / fGenEventMatrix)
  THnSparseF *fTriggerEventMatrix; //-> mcZv:multMB
  THnSparseF *fTriggerSDEventMatrix; //-> mcZv:multMB
  THnSparseF *fTriggerDDEventMatrix; //-> mcZv:multMB
  THnSparseF *fTriggerNDEventMatrix; //-> mcZv:multMB
  THnSparseF *fTriggerNSDEventMatrix; //-> mcZv:multMB

  // event vertex rec. eff correction (fRecEventMatrix / fTriggerEventMatrix)
  THnSparseF *fRecEventMatrix; //-> mcZv:multMB 
  THnSparseF *fRecSDEventMatrix; //-> mcZv:multMB
  THnSparseF *fRecDDEventMatrix; //-> mcZv:multMB
  THnSparseF *fRecNDEventMatrix; //-> mcZv:multMB
  THnSparseF *fRecNSDEventMatrix; //-> mcZv:multMB


  //
  // track-event level correction 
  //

  THnSparseF *fGenTrackEventMatrix; //-> mcZv:mcPt:mcEta
  THnSparseF *fGenTrackSDEventMatrix; //-> mcZv:mcPt:mcEta
  THnSparseF *fGenTrackDDEventMatrix; //-> mcZv:mcPt:mcEta
  THnSparseF *fGenTrackNDEventMatrix; //-> mcZv:mcPt:mcEta
  THnSparseF *fGenTrackNSDEventMatrix; //-> mcZv:mcPt:mcEta

  // trigger bias corrections (fTriggerTrackEventMatrix / fGenTrackEventMatrix)
  THnSparseF *fTriggerTrackEventMatrix; //-> mcZv:mcPt:mcEta
  THnSparseF *fTriggerTrackSDEventMatrix; //-> mcZv:mcPt:mcEta
  THnSparseF *fTriggerTrackDDEventMatrix; //-> mcZv:mcPt:mcEta
  THnSparseF *fTriggerTrackNDEventMatrix; //-> mcZv:mcPt:mcEta
  THnSparseF *fTriggerTrackNSDEventMatrix; //-> mcZv:mcPt:mcEta

  // event vertex rec. corrections (fRecTrackEventMatrix / fTriggerTrackEventMatrix)
  THnSparseF *fRecTrackEventMatrix; //-> mcZv:Pt:mcEta
  THnSparseF *fRecTrackSDEventMatrix; //-> mcZv:Pt:mcEta
  THnSparseF *fRecTrackDDEventMatrix; //-> mcZv:Pt:mcEta
  THnSparseF *fRecTrackNDEventMatrix; //-> mcZv:Pt:mcEta
  THnSparseF *fRecTrackNSDEventMatrix; //-> mcZv:Pt:mcEta

  //
  // track level correction 
  //

  // track rec. efficiency correction (fRecPrimTrackMatrix / fGenPrimTrackMatrix)
  THnSparseF *fGenTrackMatrix; //-> mcZv:mcPt:mcEta
  THnSparseF *fGenPrimTrackMatrix; //-> mcZv:mcPt:mcEta
  THnSparseF *fRecPrimTrackMatrix; //-> mcZv:mcPt:mcEta

  // secondary track contamination correction (fRecSecTrackMatrix / fRecTrackMatrix)
  THnSparseF *fRecTrackMatrix;    //-> mcZv:mcPt:mcEta
  THnSparseF *fRecSecTrackMatrix; //-> mcZv:mcPt:mcEta

  // multiple rec. track corrections (fRecMultTrackMatrix / fRecTrackMatrix)
  THnSparseF *fRecMultTrackMatrix; //-> mcZv:Pt:mcEta

  //
  // ESD and MC control analysis histograms
  //

  // THnSparse event histograms
  THnSparseF *fMCEventHist1;  //-> mcXv:mcYv:mcZv
  THnSparseF *fRecEventHist1; //-> Xv:Yv:Zv
  THnSparseF *fRecEventHist2; //-> Zv:multMB:mult
  THnSparseF *fRecMCEventHist1; //-> Xv-mcXv:Yv-mcYv:Zv-mcZv
  THnSparseF *fRecMCEventHist2; //-> Xv-mcXv:Zv-mcZv:mult
  THnSparseF *fRecMCEventHist3; //-> mult:EventType (ND, DD, SD)

  // THnSparse track histograms
  // [0] - after charged track selection, [1] - after acceptance cuts, [2] - after esd track cuts

  THnSparseF *fMCTrackHist1[AlidNdPtHelper::kCutSteps];     //-> mcPt:mcEta:mcPhi
  THnSparseF *fMCPrimTrackHist1[AlidNdPtHelper::kCutSteps]; //-> mcPt:mcEta:pid:mech:mother
  THnSparseF *fMCPrimTrackHist2[AlidNdPtHelper::kCutSteps]; //-> pdg:mech:mother
  THnSparseF *fMCSecTrackHist1[AlidNdPtHelper::kCutSteps];  //-> mcPt:mcEta:pid:mech:mother

  THnSparseF *fRecTrackHist1[AlidNdPtHelper::kCutSteps];     //-> Pt:Eta:Phi
  THnSparseF *fRecTrackMultHist1[AlidNdPtHelper::kCutSteps]; //-> Pt:mult

  THnSparseF *fRecMCTrackHist1; //-> mcPt:mcEta:(Pt-mcPt)/mcPt:(Eta-mcEta)

  //multple reconstructed tracks
  THnSparseF *fMCMultRecTrackHist1; //-> mcPt:mcEta:pid

  // track control histograms
  THnSparseF *fRecTrackHist2;  //-> nclust:chi2:Pt:Eta:Phi

  //
  // Generic histograms to be corrected
  //
  THnSparseF *fRecEventHist;    //-> Zv:multMB  
  THnSparseF *fRecTrackHist; //-> Zv:pT:eta:multRec
  THnSparseF *fEventCount; //-> trig, trig + vertex, selected event

  //
  // candle events track corrections
  //
  THnSparseF *fRecCandleEventMatrix; //-> Zv:multMB

  //binning for THNsparse
  Int_t	fMultNbins;
  Int_t fPtNbins;
  Int_t fPtCorrNbins;
  Int_t fEtaNbins;
  Int_t fZvNbins;
  Double_t* fBinsMult;
  Double_t* fBinsPt;
  Double_t* fBinsPtCorr;
  Double_t* fBinsEta;
  Double_t* fBinsZv;
  
  Bool_t fIsInit;

  AlidNdPtAnalysis(const AlidNdPtAnalysis&); // not implemented
  AlidNdPtAnalysis& operator=(const AlidNdPtAnalysis&); // not implemented

  ClassDef(AlidNdPtAnalysis,6);
};

#endif
Commit	Line	Data
00b81913	1	#ifndef ALIDNDPTANALYSIS_H
00b81913	2	#define ALIDNDPTANALYSIS_H
0aaa8b91	3
0aaa8b91	4	//------------------------------------------------------------------------------
00b81913	5	// AlidNdPtAnalysis class used for dNdPt analysis.
0aaa8b91	6	//
0aaa8b91	7	// Author: J.Otwinowski 04/11/2008
68f10917	8	// last change: 2011-04-04 by M.Knichel
0aaa8b91	9	//------------------------------------------------------------------------------
0aaa8b91	10
00b81913	11	class iostream;
	12
	13	class TFile;
	14	class TCint;
0aaa8b91	15	class TProfile;
	16	class TFolder;
	17	class TObjArray;
	18	class TString;
00b81913	19	class THnSparse;
0aaa8b91	20
	21	class AliESDtrackCuts;
	22	class AliVertexerTracks;
	23	class AliESD;
	24	class AliESDfriend;
	25	class AliESDfriendTrack;
bad4ba69	26	class AlidNdPtHelper;
bad4ba69	27
0aaa8b91	28	#include "AlidNdPt.h"
0aaa8b91	29
	30	class AlidNdPtAnalysis : public AlidNdPt {
	31	public :
	32	AlidNdPtAnalysis();
	33	AlidNdPtAnalysis(Char_t* name, Char_t* title);
	34	~AlidNdPtAnalysis();
	35
	36	// Init data members
	37	virtual void Init();
	38
	39	// Process events
	40	virtual void Process(AliESDEvent const esdEvent=0, AliMCEvent const mcEvent=0);
	41
	42	// Merge output objects (needed by PROOF)
00b81913	43	virtual Long64_t Merge(TCollection* const list);
0aaa8b91	44
	45	// Analyse output histograms
	46	virtual void Analyse();
	47
	48	// Export objects to folder
c931948b	49	virtual TFolder ExportToFolder(TObjArray const array=0);
0aaa8b91	50
0aaa8b91	51	// Get analysis folder
00b81913	52	TFolder* GetAnalysisFolder() const {return fAnalysisFolder;}
0aaa8b91	53
0aaa8b91	54	// Fill control histograms
00b81913	55	void SetHistogramsOn(const Bool_t histOn=kTRUE) {fHistogramsOn = histOn;}
00b81913	56	Bool_t IsHistogramsOn() const {return fHistogramsOn;}
0aaa8b91	57
791aaf54	58	// Define 0-multiplicity bin for LHC
	59	// background calculation
	60	static Bool_t IsBinZeroSPDvtx(const AliESDEvent* esdEvent);
	61	static Bool_t IsBinZeroTrackSPDvtx(const AliESDEvent* esdEvent);
	62
0aaa8b91	63	// Create folder for analysed histograms
	64	TFolder *CreateFolder(TString folder = "folderdNdPtAnalysis",TString title = "Analysed dNdPt histograms");
	65
68f10917	66	// Set binning for Histograms (if not set default binning is used)
fc98fbb5	67	void SetBinsMult(Int_t nbins, Double_t* edges) { fMultNbins = nbins; fBinsMult = CloneArray(nbins+1,edges); }
	68	void SetBinsPt(Int_t nbins, Double_t* edges) { fPtNbins = nbins; fBinsPt = CloneArray(nbins+1,edges); }
	69	void SetBinsPtCorr(Int_t nbins, Double_t* edges) { fPtCorrNbins = nbins; fBinsPtCorr = CloneArray(nbins+1,edges); }
	70	void SetBinsEta(Int_t nbins, Double_t* edges) { fEtaNbins = nbins; fBinsEta = CloneArray(nbins+1,edges); }
	71	void SetBinsZv(Int_t nbins, Double_t* edges) { fZvNbins = nbins; fBinsZv = CloneArray(nbins+1,edges); }
68f10917	72
68f10917	73
0aaa8b91	74	// Fill histograms
fff1caf5	75	void FillHistograms(AliESDtrack const esdTrack, AliStack const stack, const Double_t zv, AlidNdPtHelper::TrackObject trackObj, Int_t multMB);
0aaa8b91	76	void FillHistograms(AliStack *const stack, Int_t label, AlidNdPtHelper::TrackObject trackObj);
	77	void FillHistograms(TObjArray const allChargedTracks,Int_t const labelsAll,Int_t multAll,Int_t const labelsAcc,Int_t multAcc,Int_t const labelsRec,Int_t multRec);
	78
	79	// Getters
00b81913	80	THnSparseF *GetEventMultCorrelationMatrix() const {return fEventMultCorrelationMatrix;}
00b81913	81	THnSparseF *GetTrackPtCorrelationMatrix() const {return fTrackPtCorrelationMatrix;}
0aaa8b91	82
0aaa8b91	83	//
00b81913	84	THnSparseF *GetGenEventMatrix() const {return fGenEventMatrix;}
	85	THnSparseF *GetGenSDEventMatrix() const {return fGenSDEventMatrix;}
	86	THnSparseF *GetGenDDEventMatrix() const {return fGenDDEventMatrix;}
	87	THnSparseF *GetGenNDEventMatrix() const {return fGenNDEventMatrix;}
	88	THnSparseF *GetGenNSDEventMatrix() const {return fGenNSDEventMatrix;}
	89
	90	THnSparseF *GetTriggerEventMatrix() const {return fTriggerEventMatrix;}
	91	THnSparseF *GetTriggerSDEventMatrix() const {return fTriggerSDEventMatrix;}
	92	THnSparseF *GetTriggerDDEventMatrix() const {return fTriggerDDEventMatrix;}
	93	THnSparseF *GetTriggerNDEventMatrix() const {return fTriggerNDEventMatrix;}
	94	THnSparseF *GetTriggerNSDEventMatrix() const {return fTriggerNSDEventMatrix;}
	95
	96	THnSparseF *GetRecEventMatrix() const {return fRecEventMatrix;}
	97	THnSparseF *GetRecSDEventMatrix() const {return fRecSDEventMatrix;}
	98	THnSparseF *GetRecDDEventMatrix() const {return fRecDDEventMatrix;}
	99	THnSparseF *GetRecNDEventMatrix() const {return fRecNDEventMatrix;}
	100	THnSparseF *GetRecNSDEventMatrix() const {return fRecNSDEventMatrix;}
0aaa8b91	101
55468faf	102	//
	103	THnSparseF *GetRecCandleEventMatrix() const {return fRecCandleEventMatrix;}
	104
0aaa8b91	105	//
00b81913	106	THnSparseF *GetGenTrackEventMatrix() const {return fGenTrackEventMatrix;}
	107	THnSparseF *GetGenTrackSDEventMatrix() const {return fGenTrackSDEventMatrix;}
	108	THnSparseF *GetGenTrackDDEventMatrix() const {return fGenTrackDDEventMatrix;}
	109	THnSparseF *GetGenTrackNDEventMatrix() const {return fGenTrackNDEventMatrix;}
	110	THnSparseF *GetGenTrackNSDEventMatrix() const {return fGenTrackNSDEventMatrix;}
	111
	112	THnSparseF *GetTriggerTrackEventMatrix() const {return fTriggerTrackEventMatrix;}
	113	THnSparseF *GetTriggerTrackSDEventMatrix() const {return fTriggerTrackSDEventMatrix;}
	114	THnSparseF *GetTriggerTrackDDEventMatrix() const {return fTriggerTrackDDEventMatrix;}
	115	THnSparseF *GetTriggerTrackNDEventMatrix() const {return fTriggerTrackNDEventMatrix;}
	116	THnSparseF *GetTriggerTrackNSDEventMatrix() const {return fTriggerTrackNSDEventMatrix;}
	117
	118	THnSparseF *GetRecTrackEventMatrix() const {return fRecTrackEventMatrix;}
	119	THnSparseF *GetRecTrackSDEventMatrix() const {return fRecTrackSDEventMatrix;}
	120	THnSparseF *GetRecTrackDDEventMatrix() const {return fRecTrackDDEventMatrix;}
	121	THnSparseF *GetRecTrackNDEventMatrix() const {return fRecTrackNDEventMatrix;}
	122	THnSparseF *GetRecTrackNSDEventMatrix() const {return fRecTrackNSDEventMatrix;}
0aaa8b91	123
0aaa8b91	124	//
00b81913	125	THnSparseF *GetGenTrackMatrix() const {return fGenTrackMatrix;}
	126	THnSparseF *GetGenPrimTrackMatrix() const {return fGenPrimTrackMatrix;}
	127	THnSparseF *GetRecPrimTrackMatrix() const {return fRecPrimTrackMatrix;}
0aaa8b91	128
00b81913	129	THnSparseF *GetRecTrackMatrix() const {return fRecTrackMatrix;}
	130	THnSparseF *GetRecSecTrackMatrix() const {return fRecSecTrackMatrix;}
	131	THnSparseF *GetRecMultTrackMatrix() const {return fRecMultTrackMatrix;}
0aaa8b91	132
	133	//
	134	// control histograms
	135	//
00b81913	136	THnSparseF *GetMCEventHist1() const {return fMCEventHist1;}
	137	THnSparseF *GetRecEventHist1() const {return fRecEventHist1;}
	138	THnSparseF *GetRecEventHist2() const {return fRecEventHist2;}
	139	THnSparseF *GetRecMCEventHist1() const {return fRecMCEventHist1;}
	140	THnSparseF *GetRecMCEventHist2() const {return fRecMCEventHist2;}
	141	THnSparseF *GetRecMCEventHist3() const {return fRecMCEventHist3;}
0aaa8b91	142
00b81913	143	THnSparseF *GetMCTrackHist1(Int_t i) const {return fMCTrackHist1[i];}
00b81913	144	THnSparseF *GetMCPrimTrackHist1(Int_t i) const {return fMCPrimTrackHist1[i];}
7f7dd416	145	THnSparseF *GetMCPrimTrackHist2(Int_t i) const {return fMCPrimTrackHist2[i];}
00b81913	146	THnSparseF *GetMCSecTrackHist1(Int_t i) const {return fMCSecTrackHist1[i];}
0aaa8b91	147
00b81913	148	THnSparseF *GetRecTrackHist1(Int_t i) const {return fRecTrackHist1[i];}
00b81913	149	THnSparseF *GetRecTrackMultHist1(Int_t i) const {return fRecTrackMultHist1[i];}
0aaa8b91	150
00b81913	151	THnSparseF *GetRecMCTrackHist1() const {return fRecMCTrackHist1;}
00b81913	152	THnSparseF *GetMCMultRecTrackHist1() const {return fMCMultRecTrackHist1;}
0aaa8b91	153
00b81913	154	THnSparseF *GetRecTrackHist2() const {return fRecTrackHist2;}
24c88fc4	155	THnSparseF *GetEventCount() const {return fEventCount;}
f4f8cae4	156
	157	//
	158	// Generic histograms to be corrected
	159	//
	160	THnSparseF *GetRecEventHist() const {return fRecEventHist;}
	161	THnSparseF *GetRecTrackHist() const {return fRecTrackHist;}
	162
0aaa8b91	163	private:
	164
	165	// analysis folder
	166	TFolder *fAnalysisFolder; // folder for analysed histograms
00b81913	167	Bool_t fHistogramsOn; // switch on/off filling of control histograms
0aaa8b91	168
	169	//
	170	// correlation matrices (histograms)
	171	//
	172
	173	// event rec. track vs true track multiplicity correlation matrix
fff1caf5	174	THnSparseF *fEventMultCorrelationMatrix; //-> mult:mult_true_tracks:multMB
0aaa8b91	175
	176	// rec. track pt vs true track pt correlation matrix for given eta
	177	THnSparseF *fTrackPtCorrelationMatrix; //-> Pt:mcPt:mcEta
	178
	179	//
	180	// event level correction
	181	//
	182
	183	// all genertated
bad4ba69	184	THnSparseF *fGenEventMatrix; //-> mcZv:multMB (inelastic)
	185	THnSparseF *fGenSDEventMatrix; //-> mcZv:multMB (single diffractive)
	186	THnSparseF *fGenDDEventMatrix; //-> mcZv:multMB (single diffractive)
	187	THnSparseF *fGenNDEventMatrix; //-> mcZv:multMB (non diffractive)
	188	THnSparseF *fGenNSDEventMatrix; //-> mcZv:multMB (non single diffractive)
0aaa8b91	189
0aaa8b91	190	// trigger bias corrections (fTriggerEventMatrix / fGenEventMatrix)
bad4ba69	191	THnSparseF *fTriggerEventMatrix; //-> mcZv:multMB
	192	THnSparseF *fTriggerSDEventMatrix; //-> mcZv:multMB
	193	THnSparseF *fTriggerDDEventMatrix; //-> mcZv:multMB
	194	THnSparseF *fTriggerNDEventMatrix; //-> mcZv:multMB
	195	THnSparseF *fTriggerNSDEventMatrix; //-> mcZv:multMB
0aaa8b91	196
0aaa8b91	197	// event vertex rec. eff correction (fRecEventMatrix / fTriggerEventMatrix)
bad4ba69	198	THnSparseF *fRecEventMatrix; //-> mcZv:multMB
	199	THnSparseF *fRecSDEventMatrix; //-> mcZv:multMB
	200	THnSparseF *fRecDDEventMatrix; //-> mcZv:multMB
	201	THnSparseF *fRecNDEventMatrix; //-> mcZv:multMB
	202	THnSparseF *fRecNSDEventMatrix; //-> mcZv:multMB
0aaa8b91	203
55468faf	204
0aaa8b91	205	//
	206	// track-event level correction
	207	//
	208
	209	THnSparseF *fGenTrackEventMatrix; //-> mcZv:mcPt:mcEta
	210	THnSparseF *fGenTrackSDEventMatrix; //-> mcZv:mcPt:mcEta
	211	THnSparseF *fGenTrackDDEventMatrix; //-> mcZv:mcPt:mcEta
	212	THnSparseF *fGenTrackNDEventMatrix; //-> mcZv:mcPt:mcEta
	213	THnSparseF *fGenTrackNSDEventMatrix; //-> mcZv:mcPt:mcEta
	214
	215	// trigger bias corrections (fTriggerTrackEventMatrix / fGenTrackEventMatrix)
	216	THnSparseF *fTriggerTrackEventMatrix; //-> mcZv:mcPt:mcEta
	217	THnSparseF *fTriggerTrackSDEventMatrix; //-> mcZv:mcPt:mcEta
	218	THnSparseF *fTriggerTrackDDEventMatrix; //-> mcZv:mcPt:mcEta
	219	THnSparseF *fTriggerTrackNDEventMatrix; //-> mcZv:mcPt:mcEta
	220	THnSparseF *fTriggerTrackNSDEventMatrix; //-> mcZv:mcPt:mcEta
	221
	222	// event vertex rec. corrections (fRecTrackEventMatrix / fTriggerTrackEventMatrix)
	223	THnSparseF *fRecTrackEventMatrix; //-> mcZv:Pt:mcEta
	224	THnSparseF *fRecTrackSDEventMatrix; //-> mcZv:Pt:mcEta
	225	THnSparseF *fRecTrackDDEventMatrix; //-> mcZv:Pt:mcEta
	226	THnSparseF *fRecTrackNDEventMatrix; //-> mcZv:Pt:mcEta
	227	THnSparseF *fRecTrackNSDEventMatrix; //-> mcZv:Pt:mcEta
	228
	229	//
	230	// track level correction
	231	//
	232
	233	// track rec. efficiency correction (fRecPrimTrackMatrix / fGenPrimTrackMatrix)
985bdfd5	234	THnSparseF *fGenTrackMatrix; //-> mcZv:mcPt:mcEta
0aaa8b91	235	THnSparseF *fGenPrimTrackMatrix; //-> mcZv:mcPt:mcEta
	236	THnSparseF *fRecPrimTrackMatrix; //-> mcZv:mcPt:mcEta
	237
	238	// secondary track contamination correction (fRecSecTrackMatrix / fRecTrackMatrix)
	239	THnSparseF *fRecTrackMatrix; //-> mcZv:mcPt:mcEta
	240	THnSparseF *fRecSecTrackMatrix; //-> mcZv:mcPt:mcEta
	241
	242	// multiple rec. track corrections (fRecMultTrackMatrix / fRecTrackMatrix)
	243	THnSparseF *fRecMultTrackMatrix; //-> mcZv:Pt:mcEta
	244
	245	//
	246	// ESD and MC control analysis histograms
	247	//
	248
	249	// THnSparse event histograms
	250	THnSparseF *fMCEventHist1; //-> mcXv:mcYv:mcZv
	251	THnSparseF *fRecEventHist1; //-> Xv:Yv:Zv
bad4ba69	252	THnSparseF *fRecEventHist2; //-> Zv:multMB:mult
0aaa8b91	253	THnSparseF *fRecMCEventHist1; //-> Xv-mcXv:Yv-mcYv:Zv-mcZv
bad4ba69	254	THnSparseF *fRecMCEventHist2; //-> Xv-mcXv:Zv-mcZv:mult
bad4ba69	255	THnSparseF *fRecMCEventHist3; //-> mult:EventType (ND, DD, SD)
0aaa8b91	256
	257	// THnSparse track histograms
	258	// [0] - after charged track selection, [1] - after acceptance cuts, [2] - after esd track cuts
	259
	260	THnSparseF *fMCTrackHist1[AlidNdPtHelper::kCutSteps]; //-> mcPt:mcEta:mcPhi
	261	THnSparseF *fMCPrimTrackHist1[AlidNdPtHelper::kCutSteps]; //-> mcPt:mcEta:pid:mech:mother
7f7dd416	262	THnSparseF *fMCPrimTrackHist2[AlidNdPtHelper::kCutSteps]; //-> pdg:mech:mother
0aaa8b91	263	THnSparseF *fMCSecTrackHist1[AlidNdPtHelper::kCutSteps]; //-> mcPt:mcEta:pid:mech:mother
	264
	265	THnSparseF *fRecTrackHist1[AlidNdPtHelper::kCutSteps]; //-> Pt:Eta:Phi
bad4ba69	266	THnSparseF *fRecTrackMultHist1[AlidNdPtHelper::kCutSteps]; //-> Pt:mult
0aaa8b91	267
	268	THnSparseF *fRecMCTrackHist1; //-> mcPt:mcEta:(Pt-mcPt)/mcPt:(Eta-mcEta)
	269
	270	//multple reconstructed tracks
	271	THnSparseF *fMCMultRecTrackHist1; //-> mcPt:mcEta:pid
	272
985bdfd5	273	// track control histograms
	274	THnSparseF *fRecTrackHist2; //-> nclust:chi2:Pt:Eta:Phi
	275
f4f8cae4	276	//
	277	// Generic histograms to be corrected
	278	//
24c88fc4	279	THnSparseF *fRecEventHist; //-> Zv:multMB
fff1caf5	280	THnSparseF *fRecTrackHist; //-> Zv:pT:eta:multRec
24c88fc4	281	THnSparseF *fEventCount; //-> trig, trig + vertex, selected event
f4f8cae4	282
55468faf	283	//
	284	// candle events track corrections
	285	//
	286	THnSparseF *fRecCandleEventMatrix; //-> Zv:multMB
	287
68f10917	288	//binning for THNsparse
	289	Int_t fMultNbins;
	290	Int_t fPtNbins;
	291	Int_t fPtCorrNbins;
	292	Int_t fEtaNbins;
	293	Int_t fZvNbins;
	294	Double_t* fBinsMult;
	295	Double_t* fBinsPt;
	296	Double_t* fBinsPtCorr;
	297	Double_t* fBinsEta;
	298	Double_t* fBinsZv;
	299
	300	Bool_t fIsInit;
	301
0aaa8b91	302	AlidNdPtAnalysis(const AlidNdPtAnalysis&); // not implemented
	303	AlidNdPtAnalysis& operator=(const AlidNdPtAnalysis&); // not implemented
	304
24c88fc4	305	ClassDef(AlidNdPtAnalysis,6);
0aaa8b91	306	};
	307
	308	#endif