[u/mrichter/AliRoot.git] / PWG4 / AliGammaMCReader.h

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

/* History of cvs commits:
 *
 * $Log$
 *
 */

//_________________________________________________________________________
// Class for reading data (Kinematics) in order to do prompt gamma correlations
//  Class created from old AliPHOSGammaJet
//  (see AliRoot versions previous Release 4-09)

//*-- Author: Gustavo Conesa (INFN-LNF)

// --- ROOT system ---
#include <TParticle.h> 
#include <TClonesArray.h> 
#include "AliStack.h"
#include "AliGammaReader.h" 
 
class TH2F ; 

class AliGammaMCReader : public AliGammaReader {

public: 

  AliGammaMCReader() ; // ctor
  AliGammaMCReader(const AliGammaMCReader & g) ; // cpy ctor
  AliGammaMCReader & operator = (const AliGammaMCReader & g) ;//cpy assignment
  virtual ~AliGammaMCReader() {;} //virtual dtor

  enum decay_t {kNoDecay, kGeantDecay, kDecay, kDecayGamma};
 
  void InitParameters();

  Bool_t  IsInEMCAL(Double_t phi, Double_t eta) ;
  Bool_t  IsInPHOS(Double_t phi, Double_t eta) ;

  Int_t    GetDecayPi0Flag() const {return fDecayPi0 ; }

  void Print(const Option_t * opt)const;
  
  void SetDecayPi0Flag(Int_t d){ fDecayPi0 = d ; }

  void SetCheckOverlapping(Bool_t check){fCheckOverlapping = check ;}
  Bool_t IsCheckOverlappingOn() {return fCheckOverlapping ;}

  private:
  
  void CaseDecayGamma(Int_t index, TParticle * particle, AliStack * stack,
		      TClonesArray * plEMCAL, Int_t &indexEMCAL,
		      TClonesArray * plPHOS, Int_t &indexPHOS);
  
  void CaseGeantDecay(TParticle * particle, AliStack * stack,
		      TClonesArray * plEMCAL, Int_t &indexEMCAL,
		      TClonesArray * plPHOS, Int_t &indexPHOS);
  
  void CasePi0Decay(TParticle * particle, TClonesArray * plEMCAL, Int_t &indexEMCAL,
		    TClonesArray * plPHOS, Int_t &indexPHOS);
  
  void CreateParticleList(TObject * stack, TObject * ,
			  TClonesArray * plCh, TClonesArray * plEMCAL, 
			  TClonesArray * plPHOS, TClonesArray * plParton,TClonesArray *,TClonesArray *);
  
  void FillListWithDecayGammaOrPi0(TParticle * pPi0, TParticle * pdaug0, TParticle * pdaug1,
				   TClonesArray * plEMCAL, Int_t &indexEMCAL,
				   TClonesArray * plPHOS, Int_t &indexPHOS);  
  void MakePi0Decay(TLorentzVector &p0, TLorentzVector &p1, TLorentzVector &p2);//, Double_t &angle);
  
  
  private:
  
  Int_t      fDecayPi0; //Decay Pi0.
  Bool_t   fCheckOverlapping; // if True, check if gammas from decay overlapp in calorimeters.
  
  ClassDef(AliGammaMCReader,1)
} ;


#endif //ALIGAMMAMCREADER_H
Commit	Line	Data
bdcfac30	1	#ifndef ALIGAMMAMCREADER_H
	2	#define ALIGAMMAMCREADER_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5	/* $Id$ */
	6
	7	/* History of cvs commits:
	8	*
	9	* $Log$
bdcfac30	10	*
	11	*/
	12
	13	//_________________________________________________________________________
	14	// Class for reading data (Kinematics) in order to do prompt gamma correlations
	15	// Class created from old AliPHOSGammaJet
	16	// (see AliRoot versions previous Release 4-09)
	17
	18	//*-- Author: Gustavo Conesa (INFN-LNF)
	19
	20	// --- ROOT system ---
	21	#include <TParticle.h>
	22	#include <TClonesArray.h>
	23	#include "AliStack.h"
	24	#include "AliGammaReader.h"
	25
	26	class TH2F ;
	27
	28	class AliGammaMCReader : public AliGammaReader {
	29
	30	public:
	31
	32	AliGammaMCReader() ; // ctor
	33	AliGammaMCReader(const AliGammaMCReader & g) ; // cpy ctor
	34	AliGammaMCReader & operator = (const AliGammaMCReader & g) ;//cpy assignment
	35	virtual ~AliGammaMCReader() {;} //virtual dtor
	36
4b707925	37	enum decay_t {kNoDecay, kGeantDecay, kDecay, kDecayGamma};
bdcfac30	38
	39	void InitParameters();
	40
	41	Bool_t IsInEMCAL(Double_t phi, Double_t eta) ;
	42	Bool_t IsInPHOS(Double_t phi, Double_t eta) ;
	43
	44	Int_t GetDecayPi0Flag() const {return fDecayPi0 ; }
bdcfac30	45
	46	void Print(const Option_t * opt)const;
	47
	48	void SetDecayPi0Flag(Int_t d){ fDecayPi0 = d ; }
bdcfac30	49
4b707925	50	void SetCheckOverlapping(Bool_t check){fCheckOverlapping = check ;}
	51	Bool_t IsCheckOverlappingOn() {return fCheckOverlapping ;}
	52
	53	private:
	54
	55	void CaseDecayGamma(Int_t index, TParticle * particle, AliStack * stack,
	56	TClonesArray * plEMCAL, Int_t &indexEMCAL,
	57	TClonesArray * plPHOS, Int_t &indexPHOS);
	58
	59	void CaseGeantDecay(TParticle * particle, AliStack * stack,
	60	TClonesArray * plEMCAL, Int_t &indexEMCAL,
	61	TClonesArray * plPHOS, Int_t &indexPHOS);
	62
	63	void CasePi0Decay(TParticle * particle, TClonesArray * plEMCAL, Int_t &indexEMCAL,
	64	TClonesArray * plPHOS, Int_t &indexPHOS);
	65
bdcfac30	66	void CreateParticleList(TObject * stack, TObject * ,
bdcfac30	67	TClonesArray * plCh, TClonesArray * plEMCAL,
4b707925	68	TClonesArray * plPHOS, TClonesArray * plParton,TClonesArray ,TClonesArray );
	69
	70	void FillListWithDecayGammaOrPi0(TParticle * pPi0, TParticle * pdaug0, TParticle * pdaug1,
	71	TClonesArray * plEMCAL, Int_t &indexEMCAL,
	72	TClonesArray * plPHOS, Int_t &indexPHOS);
	73	void MakePi0Decay(TLorentzVector &p0, TLorentzVector &p1, TLorentzVector &p2);//, Double_t &angle);
	74
	75
	76	private:
bdcfac30	77
bdcfac30	78	Int_t fDecayPi0; //Decay Pi0.
4b707925	79	Bool_t fCheckOverlapping; // if True, check if gammas from decay overlapp in calorimeters.
	80
	81	ClassDef(AliGammaMCReader,1)
bdcfac30	82	} ;
4b707925	83
bdcfac30	84
	85	#endif //ALIGAMMAMCREADER_H
	86
	87
	88