[u/mrichter/AliRoot.git] / PWGGA / CaloTrackCorrelations / AliAnaPi0EbE.h

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

//_________________________________________________________________________
//
// Class for the analysis of high pT pi0 event by event
// Pi0/Eta identified by one of the following:
//  -Invariant mass of 2 cluster in calorimeter
//  -Shower shape analysis in calorimeter
//  -Invariant mass of one cluster in calorimeter and one photon reconstructed in TPC (in near future)
//
//-- Author: Gustavo Conesa (INFN-LNF)  &  Raphaelle Ichou (SUBATECH)
//_________________________________________________________________________


// --- ROOT system ---
class TList ;
class TObjString;

// --- ANALYSIS system ---
#include "AliAnaCaloTrackCorrBaseClass.h"

class AliAnaPi0EbE : public AliAnaCaloTrackCorrBaseClass {

 public: 
  AliAnaPi0EbE() ; // default ctor
  virtual ~AliAnaPi0EbE() { ; } //virtual dtor
	  
  TObjString *   GetAnalysisCuts();
  
  TList      *   GetCreateOutputObjects();
  
  void           Init();
  
  void           InitParameters();

  void           MakeAnalysisFillAOD()  ;
   
  void           MakeAnalysisFillHistograms() ; 
  
  void           Print(const Option_t * opt) const;
  
  // Main
  
  void           FillSelectedClusterHistograms(AliVCluster* cluster, 
                                               const Int_t nLocMax,
                                               const Int_t tag);
    
  void           FillWeightHistograms(AliVCluster *clus);
    
  void           HasPairSameMCMother(AliAODPWG4Particle * photon1, 
                                     AliAODPWG4Particle * photon2, 
                                     Int_t & label, Int_t & tag);
  
  void           MakeInvMassInCalorimeter() ;
  
  void           MakeInvMassInCalorimeterAndCTS() ;
  
  void           MakeShowerShapeIdentification() ;
          
  //Setters Getters
  
  //Analysis types
  enum anaTypes  {kIMCalo, kSSCalo, kIMCaloTracks};  
  anaTypes       GetAnalysisType()                     const { return fAnaType                ; }
  void           SetAnalysisType(anaTypes ana)               { fAnaType = ana                 ; }
  
  TString        GetInputAODGammaConvName()            const { return fInputAODGammaConvName  ; }
  void           SetInputAODGammaConvName(TString name)      { fInputAODGammaConvName = name  ; }	
  
  //Only for pi0 SS identification case
  void           SetCalorimeter(TString & det)               { fCalorimeter = det             ; }
  
  void           SetMinDistanceToBadChannel(Float_t m1, Float_t m2, Float_t m3) {
                  fMinDist = m1; fMinDist2 = m2; fMinDist3 = m3                               ; }

  void           SwitchOnFillWeightHistograms()              { fFillWeightHistograms = kTRUE  ; }
  void           SwitchOffFillWeightHistograms()             { fFillWeightHistograms = kFALSE ; }  
  
  void           SwitchOnTMHistoFill()                       { fFillTMHisto          = kTRUE  ; }
  void           SwitchOffTMHistoFill()                      { fFillTMHisto          = kFALSE ; }

  void           SwitchOnSelectedClusterHistoFill()          { fFillSelectClHisto    = kTRUE  ; }
  void           SwitchOffSelectedClusterHistoFill()         { fFillSelectClHisto    = kFALSE ; }
  
  //For histograms
  enum mcTypes   { kmcPhoton = 0, kmcConversion = 1, kmcPi0    = 2,  
                   kmcEta    = 3, kmcElectron   = 4, kmcHadron = 5 };

 private:
  
  anaTypes       fAnaType;                 // Select analysis type
  
  //Only for pi0 SS identification case, kSSCalo
  TString        fCalorimeter ;            // Calorimeter where the gamma is searched;
  Float_t        fMinDist ;                // Minimal distance to bad channel to accept cluster
  Float_t        fMinDist2;                // Cuts on Minimal distance to study acceptance evaluation
  Float_t        fMinDist3;                // One more cut on distance used for acceptance-efficiency study
  
  Bool_t         fFillWeightHistograms ;   // Fill weigth histograms
  Bool_t         fFillTMHisto;             // Fill track matching plots
  Bool_t         fFillSelectClHisto;       // Fill selected cluster histograms

  //Only for combination of calorimeter and conversion photons, kIMCaloTracks
  TString        fInputAODGammaConvName;   //  Name of AOD branch with conversion photons
  
  //Histograms
  
  TH1F         * fhPt  ;                   //! Number of identified  pi0/eta vs pT
  TH1F         * fhE   ;                   //! Number of identified  pi0/eta vs E
  TH2F         * fhEEta  ;                 //! E vs eta of identified  pi0/eta 
  TH2F         * fhEPhi  ;                 //! E vs phi of identified  pi0/eta 
  TH2F         * fhEtaPhi  ;               //! eta vs phi of identified  pi0/eta 

  TH1F         * fhPtDecay  ;              //! Number of identified  pi0/eta decay photons vs pT
  TH1F         * fhEDecay   ;              //! Number of identified  pi0/eta decay photons vs E
  
  TH2F         * fhEDispersion ;           //! E vs disp of selected cluster
  TH2F         * fhELambda0 ;              //! E vs lambda0 of selected cluster 
  TH2F         * fhELambda1 ;              //! E vs lambda1 of selected cluster 
  TH2F         * fhELambda0NoTRD ;         //! E vs lambda0 of selected cluster, not behind TRD 
  TH2F         * fhELambda0FracMaxCellCut ;//! E vs lambda0 of selected cluster, fraction of cluster energy in max cell cut 
  TH2F         * fhEFracMaxCell ;          //! E vs frac max cell of selected cluster 
  TH2F         * fhEFracMaxCellNoTRD ;     //! E vs frac max cell of selected cluster, not behind TRD  
  TH2F         * fhENCells;                //! E vs N cells in selected cluster
  TH2F         * fhETime;                  //! E vs Time of selected cluster 
  TH2F         * fhEPairDiffTime;          //! E vs Pair of clusters time difference vs E
  
  //MC histograms
  
  TH2F         * fhEMCLambda0[6] ;         //! E vs lambda0 of pi0 pairs but really from MC particle
  TH2F         * fhEMCLambda1[6] ;         //! E vs lambda1 of pi0 pairs but really from MC particle
  TH2F         * fhEMCDispersion[6] ;      //! E vs dispersion of pi0 pairs but really from MC particle
  TH2F         * fhEMCLambda0NoTRD[6] ;         //! E vs lambda0 of pi0 pairs but really from MC particle, not behind TRD
  TH2F         * fhEMCLambda0FracMaxCellCut[6] ;//! E vs lambda0 of pi0 pairs but really from MC particle, fraction of cluster energy in max cell cut
  TH2F         * fhEMCFracMaxCell[6] ;     //! E vs fraction of max cell 
  
  TH1F         * fhPtMCNo;                 //! Number of identified pi0, not coming from pi0/eta
  TH2F         * fhPhiMCNo;                //! Phi of identified pi0, not coming from pi0/eta
  TH2F         * fhEtaMCNo;                //! eta of identified  pi0, not coming from pi0/eta
  TH1F         * fhPtMC;                   //! Number of identified pi0, coming from pi0/eta
  TH2F         * fhPhiMC;                  //! Phi of identified pi0, coming from pi0/eta
  TH2F         * fhEtaMC;                  //! eta of identified pi0, coming from pi0/eta

  TH2F         * fhMassPairMCPi0;          //! pair mass, origin is same pi0
  TH2F         * fhMassPairMCEta;          //! pair mass, origin is same eta
  TH2F         * fhAnglePairMCPi0;         //! pair opening angle, origin is same pi0
  TH2F         * fhAnglePairMCEta;         //! pair opening angle, origin is same eta
  
  // Weight studies
  
  TH2F         * fhECellClusterRatio;      //! e cell / e cluster vs e cluster for selected photons
  TH2F         * fhECellClusterLogRatio;   //! log (e cell / e cluster)  vs e cluster for selected photons
  TH2F         * fhEMaxCellClusterRatio;   //! e max cell / e cluster vs e cluster for selected photons
  TH2F         * fhEMaxCellClusterLogRatio;//! log (e max cell / e cluster) vs e cluster for selected photons
  TH2F         * fhLambda0ForW0[14];       //! L0 for 7 defined w0= 3, 3.5 ... 6 for selected photons
  //TH2F         * fhLambda1ForW0[7];        //! L1 for 7 defined w0= 3, 3.5 ... 6 for selected photons  
  
  // Track Matching
  TH2F         * fhTrackMatchedDEta     ;  //! Eta distance between track and cluster vs cluster E
  TH2F         * fhTrackMatchedDPhi     ;  //! Phi distance between track and cluster vs cluster E
  TH2F         * fhTrackMatchedDEtaDPhi ;  //! Eta vs Phi distance between track and cluster, E cluster > 0.5 GeV
  TH2F         * fhTrackMatchedMCParticle; //! Trace origin of matched particle
  TH2F         * fhdEdx  ;                 //! matched track dEdx vs cluster E 
  TH2F         * fhEOverP;                 //! matched track E cluster over P track vs cluster E
  TH2F         * fhEOverPNoTRD;                 //! matched track E cluster over P track vs cluster E, not behind TRD 

  // Local maxima
  TH2F         * fhNLocMax;                //! number of maxima in selected clusters
  TH2F         * fhELambda0LocMax1 ;       //! E vs lambda0 of selected cluster, 1 local maxima in cluster 
  TH2F         * fhELambda1LocMax1 ;       //! E vs lambda1 of selected cluster, 1 local maxima in cluster 
  TH2F         * fhELambda0LocMax2 ;       //! E vs lambda0 of selected cluster, 2 local maxima in cluster 
  TH2F         * fhELambda1LocMax2 ;       //! E vs lambda1 of selected cluster, 2 local maxima in cluster
  TH2F         * fhELambda0LocMaxN ;       //! E vs lambda0 of selected cluster, N>2 local maxima in cluster 
  TH2F         * fhELambda1LocMaxN ;       //! E vs lambda1 of selected cluster, N>2 local maxima in cluster 
  TH2F         * fhMassPairLocMax[8];      //! pair mass, origin is same pi0, combine clusters depending on number of maxima

  AliAnaPi0EbE(              const AliAnaPi0EbE & g) ; // cpy ctor
  AliAnaPi0EbE & operator = (const AliAnaPi0EbE & g) ; // cpy assignment
  
  ClassDef(AliAnaPi0EbE,15)
} ;


#endif //ALIANAPI0EBE_H
Commit	Line	Data
477d6cee	1	#ifndef ALIANAPI0EBE_H
	2	#define ALIANAPI0EBE_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
477d6cee	5
	6	//_________________________________________________________________________
	7	//
	8	// Class for the analysis of high pT pi0 event by event
09273901	9	// Pi0/Eta identified by one of the following:
477d6cee	10	// -Invariant mass of 2 cluster in calorimeter
	11	// -Shower shape analysis in calorimeter
	12	// -Invariant mass of one cluster in calorimeter and one photon reconstructed in TPC (in near future)
	13	//
	14	//-- Author: Gustavo Conesa (INFN-LNF) & Raphaelle Ichou (SUBATECH)
	15	//_________________________________________________________________________
	16
	17
	18	// --- ROOT system ---
477d6cee	19	class TList ;
0c1383b5	20	class TObjString;
477d6cee	21
477d6cee	22	// --- ANALYSIS system ---
745913ae	23	#include "AliAnaCaloTrackCorrBaseClass.h"
477d6cee	24
745913ae	25	class AliAnaPi0EbE : public AliAnaCaloTrackCorrBaseClass {
477d6cee	26
477d6cee	27	public:
477d6cee	28	AliAnaPi0EbE() ; // default ctor
1db06135	29	virtual ~AliAnaPi0EbE() { ; } //virtual dtor
c5693f62	30
521636d2	31	TObjString * GetAnalysisCuts();
477d6cee	32
521636d2	33	TList * GetCreateOutputObjects();
477d6cee	34
521636d2	35	void Init();
477d6cee	36
521636d2	37	void InitParameters();
	38
	39	void MakeAnalysisFillAOD() ;
	40
	41	void MakeAnalysisFillHistograms() ;
477d6cee	42
521636d2	43	void Print(const Option_t * opt) const;
477d6cee	44
521636d2	45	// Main
477d6cee	46
5c46c992	47	void FillSelectedClusterHistograms(AliVCluster* cluster,
	48	const Int_t nLocMax,
	49	const Int_t tag);
42d47cb7	50
42d47cb7	51	void FillWeightHistograms(AliVCluster *clus);
0a14e9ae	52
b5dbb99b	53	void HasPairSameMCMother(AliAODPWG4Particle * photon1,
	54	AliAODPWG4Particle * photon2,
	55	Int_t & label, Int_t & tag);
	56
521636d2	57	void MakeInvMassInCalorimeter() ;
57b97dc6	58
521636d2	59	void MakeInvMassInCalorimeterAndCTS() ;
57b97dc6	60
521636d2	61	void MakeShowerShapeIdentification() ;
dbba06ca	62
521636d2	63	//Setters Getters
	64
	65	//Analysis types
	66	enum anaTypes {kIMCalo, kSSCalo, kIMCaloTracks};
78a28af3	67	anaTypes GetAnalysisType() const { return fAnaType ; }
78a28af3	68	void SetAnalysisType(anaTypes ana) { fAnaType = ana ; }
ddc0a8a5	69
78a28af3	70	TString GetInputAODGammaConvName() const { return fInputAODGammaConvName ; }
78a28af3	71	void SetInputAODGammaConvName(TString name) { fInputAODGammaConvName = name ; }
ddc0a8a5	72
521636d2	73	//Only for pi0 SS identification case
78a28af3	74	void SetCalorimeter(TString & det) { fCalorimeter = det ; }
521636d2	75
521636d2	76	void SetMinDistanceToBadChannel(Float_t m1, Float_t m2, Float_t m3) {
78a28af3	77	fMinDist = m1; fMinDist2 = m2; fMinDist3 = m3 ; }
521636d2	78
c5693f62	79	void SwitchOnFillWeightHistograms() { fFillWeightHistograms = kTRUE ; }
	80	void SwitchOffFillWeightHistograms() { fFillWeightHistograms = kFALSE ; }
	81
09273901	82	void SwitchOnTMHistoFill() { fFillTMHisto = kTRUE ; }
	83	void SwitchOffTMHistoFill() { fFillTMHisto = kFALSE ; }
	84
06e81356	85	void SwitchOnSelectedClusterHistoFill() { fFillSelectClHisto = kTRUE ; }
	86	void SwitchOffSelectedClusterHistoFill() { fFillSelectClHisto = kFALSE ; }
	87
521636d2	88	//For histograms
c5693f62	89	enum mcTypes { kmcPhoton = 0, kmcConversion = 1, kmcPi0 = 2,
c5693f62	90	kmcEta = 3, kmcElectron = 4, kmcHadron = 5 };
521636d2	91
477d6cee	92	private:
477d6cee	93
c5693f62	94	anaTypes fAnaType; // Select analysis type
477d6cee	95
477d6cee	96	//Only for pi0 SS identification case, kSSCalo
521636d2	97	TString fCalorimeter ; // Calorimeter where the gamma is searched;
	98	Float_t fMinDist ; // Minimal distance to bad channel to accept cluster
	99	Float_t fMinDist2; // Cuts on Minimal distance to study acceptance evaluation
	100	Float_t fMinDist3; // One more cut on distance used for acceptance-efficiency study
477d6cee	101
78a28af3	102	Bool_t fFillWeightHistograms ; // Fill weigth histograms
09273901	103	Bool_t fFillTMHisto; // Fill track matching plots
06e81356	104	Bool_t fFillSelectClHisto; // Fill selected cluster histograms
09273901	105
477d6cee	106	//Only for combination of calorimeter and conversion photons, kIMCaloTracks
521636d2	107	TString fInputAODGammaConvName; // Name of AOD branch with conversion photons
477d6cee	108
477d6cee	109	//Histograms
521636d2	110
09273901	111	TH1F * fhPt ; //! Number of identified pi0/eta vs pT
	112	TH1F * fhE ; //! Number of identified pi0/eta vs E
	113	TH2F * fhEEta ; //! E vs eta of identified pi0/eta
	114	TH2F * fhEPhi ; //! E vs phi of identified pi0/eta
	115	TH2F * fhEtaPhi ; //! eta vs phi of identified pi0/eta
7a972c0c	116
09273901	117	TH1F * fhPtDecay ; //! Number of identified pi0/eta decay photons vs pT
	118	TH1F * fhEDecay ; //! Number of identified pi0/eta decay photons vs E
	119
42d47cb7	120	TH2F * fhEDispersion ; //! E vs disp of selected cluster
	121	TH2F * fhELambda0 ; //! E vs lambda0 of selected cluster
	122	TH2F * fhELambda1 ; //! E vs lambda1 of selected cluster
	123	TH2F * fhELambda0NoTRD ; //! E vs lambda0 of selected cluster, not behind TRD
	124	TH2F * fhELambda0FracMaxCellCut ;//! E vs lambda0 of selected cluster, fraction of cluster energy in max cell cut
	125	TH2F * fhEFracMaxCell ; //! E vs frac max cell of selected cluster
	126	TH2F * fhEFracMaxCellNoTRD ; //! E vs frac max cell of selected cluster, not behind TRD
	127	TH2F * fhENCells; //! E vs N cells in selected cluster
	128	TH2F * fhETime; //! E vs Time of selected cluster
	129	TH2F * fhEPairDiffTime; //! E vs Pair of clusters time difference vs E
5c46c992	130
c4a7d28a	131	//MC histograms
c4a7d28a	132
6db946bd	133	TH2F * fhEMCLambda0[6] ; //! E vs lambda0 of pi0 pairs but really from MC particle
	134	TH2F * fhEMCLambda1[6] ; //! E vs lambda1 of pi0 pairs but really from MC particle
	135	TH2F * fhEMCDispersion[6] ; //! E vs dispersion of pi0 pairs but really from MC particle
3bfcb597	136	TH2F * fhEMCLambda0NoTRD[6] ; //! E vs lambda0 of pi0 pairs but really from MC particle, not behind TRD
	137	TH2F * fhEMCLambda0FracMaxCellCut[6] ;//! E vs lambda0 of pi0 pairs but really from MC particle, fraction of cluster energy in max cell cut
	138	TH2F * fhEMCFracMaxCell[6] ; //! E vs fraction of max cell
521636d2	139
09273901	140	TH1F * fhPtMCNo; //! Number of identified pi0, not coming from pi0/eta
	141	TH2F * fhPhiMCNo; //! Phi of identified pi0, not coming from pi0/eta
	142	TH2F * fhEtaMCNo; //! eta of identified pi0, not coming from pi0/eta
	143	TH1F * fhPtMC; //! Number of identified pi0, coming from pi0/eta
	144	TH2F * fhPhiMC; //! Phi of identified pi0, coming from pi0/eta
	145	TH2F * fhEtaMC; //! eta of identified pi0, coming from pi0/eta
b5dbb99b	146
	147	TH2F * fhMassPairMCPi0; //! pair mass, origin is same pi0
	148	TH2F * fhMassPairMCEta; //! pair mass, origin is same eta
	149	TH2F * fhAnglePairMCPi0; //! pair opening angle, origin is same pi0
	150	TH2F * fhAnglePairMCEta; //! pair opening angle, origin is same eta
521636d2	151
78a28af3	152	// Weight studies
	153
	154	TH2F * fhECellClusterRatio; //! e cell / e cluster vs e cluster for selected photons
	155	TH2F * fhECellClusterLogRatio; //! log (e cell / e cluster) vs e cluster for selected photons
	156	TH2F * fhEMaxCellClusterRatio; //! e max cell / e cluster vs e cluster for selected photons
	157	TH2F * fhEMaxCellClusterLogRatio;//! log (e max cell / e cluster) vs e cluster for selected photons
c5693f62	158	TH2F * fhLambda0ForW0[14]; //! L0 for 7 defined w0= 3, 3.5 ... 6 for selected photons
1a72f6c5	159	//TH2F * fhLambda1ForW0[7]; //! L1 for 7 defined w0= 3, 3.5 ... 6 for selected photons
78a28af3	160
09273901	161	// Track Matching
	162	TH2F * fhTrackMatchedDEta ; //! Eta distance between track and cluster vs cluster E
	163	TH2F * fhTrackMatchedDPhi ; //! Phi distance between track and cluster vs cluster E
	164	TH2F * fhTrackMatchedDEtaDPhi ; //! Eta vs Phi distance between track and cluster, E cluster > 0.5 GeV
5c46c992	165	TH2F * fhTrackMatchedMCParticle; //! Trace origin of matched particle
31ae6d59	166	TH2F * fhdEdx ; //! matched track dEdx vs cluster E
b5dbb99b	167	TH2F * fhEOverP; //! matched track E cluster over P track vs cluster E
b5dbb99b	168	TH2F * fhEOverPNoTRD; //! matched track E cluster over P track vs cluster E, not behind TRD
b5dbb99b	169
5c46c992	170	// Local maxima
	171	TH2F * fhNLocMax; //! number of maxima in selected clusters
	172	TH2F * fhELambda0LocMax1 ; //! E vs lambda0 of selected cluster, 1 local maxima in cluster
	173	TH2F * fhELambda1LocMax1 ; //! E vs lambda1 of selected cluster, 1 local maxima in cluster
	174	TH2F * fhELambda0LocMax2 ; //! E vs lambda0 of selected cluster, 2 local maxima in cluster
	175	TH2F * fhELambda1LocMax2 ; //! E vs lambda1 of selected cluster, 2 local maxima in cluster
	176	TH2F * fhELambda0LocMaxN ; //! E vs lambda0 of selected cluster, N>2 local maxima in cluster
	177	TH2F * fhELambda1LocMaxN ; //! E vs lambda1 of selected cluster, N>2 local maxima in cluster
3c1d9afb	178	TH2F * fhMassPairLocMax[8]; //! pair mass, origin is same pi0, combine clusters depending on number of maxima
5c46c992	179
09273901	180	AliAnaPi0EbE( const AliAnaPi0EbE & g) ; // cpy ctor
c5693f62	181	AliAnaPi0EbE & operator = (const AliAnaPi0EbE & g) ; // cpy assignment
c5693f62	182
3c1d9afb	183	ClassDef(AliAnaPi0EbE,15)
c4a7d28a	184	} ;
477d6cee	185
	186
	187	#endif //ALIANAPI0EBE_H
	188
	189
	190