[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 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 ; }

  //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

  //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         * fhdEdx  ;                 //! matched track dEdx vs cluster E 
  TH2F         * fhEOverP;                 //! matched track E cluster over P track vs cluster E
  TH2F         * fhTrackMatchedMCParticle; //! Trace origin of matched particle
  TH2F         * fhEOverPNoTRD;                 //! matched track E cluster over P track vs cluster E, not behind TRD 

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


#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
42d47cb7	47	void FillSelectedClusterHistograms(AliVCluster* cluster, const Int_t tag);
	48
	49	void FillWeightHistograms(AliVCluster *clus);
0a14e9ae	50
b5dbb99b	51	void HasPairSameMCMother(AliAODPWG4Particle * photon1,
	52	AliAODPWG4Particle * photon2,
	53	Int_t & label, Int_t & tag);
	54
521636d2	55	void MakeInvMassInCalorimeter() ;
57b97dc6	56
521636d2	57	void MakeInvMassInCalorimeterAndCTS() ;
57b97dc6	58
521636d2	59	void MakeShowerShapeIdentification() ;
dbba06ca	60
521636d2	61	//Setters Getters
	62
	63	//Analysis types
	64	enum anaTypes {kIMCalo, kSSCalo, kIMCaloTracks};
78a28af3	65	anaTypes GetAnalysisType() const { return fAnaType ; }
78a28af3	66	void SetAnalysisType(anaTypes ana) { fAnaType = ana ; }
ddc0a8a5	67
78a28af3	68	TString GetInputAODGammaConvName() const { return fInputAODGammaConvName ; }
78a28af3	69	void SetInputAODGammaConvName(TString name) { fInputAODGammaConvName = name ; }
ddc0a8a5	70
521636d2	71	//Only for pi0 SS identification case
78a28af3	72	void SetCalorimeter(TString & det) { fCalorimeter = det ; }
521636d2	73
521636d2	74	void SetMinDistanceToBadChannel(Float_t m1, Float_t m2, Float_t m3) {
78a28af3	75	fMinDist = m1; fMinDist2 = m2; fMinDist3 = m3 ; }
521636d2	76
c5693f62	77	void SwitchOnFillWeightHistograms() { fFillWeightHistograms = kTRUE ; }
	78	void SwitchOffFillWeightHistograms() { fFillWeightHistograms = kFALSE ; }
	79
09273901	80	void SwitchOnTMHistoFill() { fFillTMHisto = kTRUE ; }
	81	void SwitchOffTMHistoFill() { fFillTMHisto = kFALSE ; }
	82
521636d2	83	//For histograms
c5693f62	84	enum mcTypes { kmcPhoton = 0, kmcConversion = 1, kmcPi0 = 2,
c5693f62	85	kmcEta = 3, kmcElectron = 4, kmcHadron = 5 };
521636d2	86
477d6cee	87	private:
477d6cee	88
c5693f62	89	anaTypes fAnaType; // Select analysis type
477d6cee	90
477d6cee	91	//Only for pi0 SS identification case, kSSCalo
521636d2	92	TString fCalorimeter ; // Calorimeter where the gamma is searched;
	93	Float_t fMinDist ; // Minimal distance to bad channel to accept cluster
	94	Float_t fMinDist2; // Cuts on Minimal distance to study acceptance evaluation
	95	Float_t fMinDist3; // One more cut on distance used for acceptance-efficiency study
477d6cee	96
78a28af3	97	Bool_t fFillWeightHistograms ; // Fill weigth histograms
09273901	98	Bool_t fFillTMHisto; // Fill track matching plots
09273901	99
477d6cee	100	//Only for combination of calorimeter and conversion photons, kIMCaloTracks
521636d2	101	TString fInputAODGammaConvName; // Name of AOD branch with conversion photons
477d6cee	102
477d6cee	103	//Histograms
521636d2	104
09273901	105	TH1F * fhPt ; //! Number of identified pi0/eta vs pT
	106	TH1F * fhE ; //! Number of identified pi0/eta vs E
	107	TH2F * fhEEta ; //! E vs eta of identified pi0/eta
	108	TH2F * fhEPhi ; //! E vs phi of identified pi0/eta
	109	TH2F * fhEtaPhi ; //! eta vs phi of identified pi0/eta
7a972c0c	110
09273901	111	TH1F * fhPtDecay ; //! Number of identified pi0/eta decay photons vs pT
	112	TH1F * fhEDecay ; //! Number of identified pi0/eta decay photons vs E
	113
42d47cb7	114	TH2F * fhEDispersion ; //! E vs disp of selected cluster
	115	TH2F * fhELambda0 ; //! E vs lambda0 of selected cluster
	116	TH2F * fhELambda1 ; //! E vs lambda1 of selected cluster
	117	TH2F * fhELambda0NoTRD ; //! E vs lambda0 of selected cluster, not behind TRD
	118	TH2F * fhELambda0FracMaxCellCut ;//! E vs lambda0 of selected cluster, fraction of cluster energy in max cell cut
	119	TH2F * fhEFracMaxCell ; //! E vs frac max cell of selected cluster
	120	TH2F * fhEFracMaxCellNoTRD ; //! E vs frac max cell of selected cluster, not behind TRD
	121	TH2F * fhENCells; //! E vs N cells in selected cluster
	122	TH2F * fhETime; //! E vs Time of selected cluster
	123	TH2F * fhEPairDiffTime; //! E vs Pair of clusters time difference vs E
3bfcb597	124
c4a7d28a	125	//MC histograms
c4a7d28a	126
6db946bd	127	TH2F * fhEMCLambda0[6] ; //! E vs lambda0 of pi0 pairs but really from MC particle
	128	TH2F * fhEMCLambda1[6] ; //! E vs lambda1 of pi0 pairs but really from MC particle
	129	TH2F * fhEMCDispersion[6] ; //! E vs dispersion of pi0 pairs but really from MC particle
3bfcb597	130	TH2F * fhEMCLambda0NoTRD[6] ; //! E vs lambda0 of pi0 pairs but really from MC particle, not behind TRD
	131	TH2F * fhEMCLambda0FracMaxCellCut[6] ;//! E vs lambda0 of pi0 pairs but really from MC particle, fraction of cluster energy in max cell cut
	132	TH2F * fhEMCFracMaxCell[6] ; //! E vs fraction of max cell
521636d2	133
09273901	134	TH1F * fhPtMCNo; //! Number of identified pi0, not coming from pi0/eta
	135	TH2F * fhPhiMCNo; //! Phi of identified pi0, not coming from pi0/eta
	136	TH2F * fhEtaMCNo; //! eta of identified pi0, not coming from pi0/eta
	137	TH1F * fhPtMC; //! Number of identified pi0, coming from pi0/eta
	138	TH2F * fhPhiMC; //! Phi of identified pi0, coming from pi0/eta
	139	TH2F * fhEtaMC; //! eta of identified pi0, coming from pi0/eta
b5dbb99b	140
	141	TH2F * fhMassPairMCPi0; //! pair mass, origin is same pi0
	142	TH2F * fhMassPairMCEta; //! pair mass, origin is same eta
	143	TH2F * fhAnglePairMCPi0; //! pair opening angle, origin is same pi0
	144	TH2F * fhAnglePairMCEta; //! pair opening angle, origin is same eta
521636d2	145
78a28af3	146	// Weight studies
	147
	148	TH2F * fhECellClusterRatio; //! e cell / e cluster vs e cluster for selected photons
	149	TH2F * fhECellClusterLogRatio; //! log (e cell / e cluster) vs e cluster for selected photons
	150	TH2F * fhEMaxCellClusterRatio; //! e max cell / e cluster vs e cluster for selected photons
	151	TH2F * fhEMaxCellClusterLogRatio;//! log (e max cell / e cluster) vs e cluster for selected photons
c5693f62	152	TH2F * fhLambda0ForW0[14]; //! L0 for 7 defined w0= 3, 3.5 ... 6 for selected photons
1a72f6c5	153	//TH2F * fhLambda1ForW0[7]; //! L1 for 7 defined w0= 3, 3.5 ... 6 for selected photons
78a28af3	154
09273901	155	// Track Matching
	156	TH2F * fhTrackMatchedDEta ; //! Eta distance between track and cluster vs cluster E
	157	TH2F * fhTrackMatchedDPhi ; //! Phi distance between track and cluster vs cluster E
	158	TH2F * fhTrackMatchedDEtaDPhi ; //! Eta vs Phi distance between track and cluster, E cluster > 0.5 GeV
31ae6d59	159	TH2F * fhdEdx ; //! matched track dEdx vs cluster E
b5dbb99b	160	TH2F * fhEOverP; //! matched track E cluster over P track vs cluster E
31ae6d59	161	TH2F * fhTrackMatchedMCParticle; //! Trace origin of matched particle
b5dbb99b	162	TH2F * fhEOverPNoTRD; //! matched track E cluster over P track vs cluster E, not behind TRD
b5dbb99b	163
09273901	164	AliAnaPi0EbE( const AliAnaPi0EbE & g) ; // cpy ctor
c5693f62	165	AliAnaPi0EbE & operator = (const AliAnaPi0EbE & g) ; // cpy assignment
c5693f62	166
b5dbb99b	167	ClassDef(AliAnaPi0EbE,12)
c4a7d28a	168	} ;
477d6cee	169
	170
	171	#endif //ALIANAPI0EBE_H
	172
	173
	174