[u/mrichter/AliRoot.git] / EMCAL / AliEMCALRecPoint.h

#ifndef ALIEMCALRECPOINT_H
#define ALIEMCALRECPOINT_H
/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
//_________________________________________________________________________
//  Base Class for EMCAL Reconstructed Points  
//  A recpoint being equivalent to a cluster in encal terminology                 
//*-- Author: Yves Schutz (SUBATECH)
//*-- Author: Dmitri Peressounko (RRC KI & SUBATECH)
//*-- Author: Heather Gray (LBL): merged AliEMCALRecPoint and AliEMCALTowerRecPoint 02/04

// --- ROOT system ---
#include <TVector3.h>

// --- Standard library ---

// --- AliRoot header files ---

#include "AliCluster.h"
class AliEMCALDigit;
class AliDigitNew;
class AliEMCALGeometry;

class AliEMCALRecPoint : public AliCluster {

 public:
  
  typedef TObjArray RecPointsList ; 

  AliEMCALRecPoint() ;                   // ctor         
  AliEMCALRecPoint(const char * opt) ;   // ctor 
  AliEMCALRecPoint(const AliEMCALRecPoint & rp);

  AliEMCALRecPoint& operator= (const AliEMCALRecPoint &rp);

  virtual ~AliEMCALRecPoint();

  virtual void    AddDigit(AliDigitNew &){ 
    Fatal("AddDigit", "use AddDigit(AliEMCALDigit & digit, Float_t Energy )") ; 
  }
  virtual void    AddDigit(AliEMCALDigit & digit, Float_t Energy); 
  virtual Int_t   Compare(const TObject * obj) const;   
  virtual Int_t   DistancetoPrimitive(Int_t px, Int_t py);
  virtual void    Draw(Option_t * option="") ;
  virtual void    ExecuteEvent(Int_t event, Int_t, Int_t) ;

  virtual void    SetClusterType(Int_t ver) { fClusterType = ver; }
  virtual Int_t   GetClusterType() const { return fClusterType; }

  virtual void    EvalAll(Float_t logWeight, TClonesArray * digits);
  virtual void    EvalLocalPosition(Float_t logWeight, TClonesArray * digits);
  virtual void    EvalPrimaries(TClonesArray * digits) ;
  virtual void    EvalParents(TClonesArray * digits) ;

  virtual int *   GetDigitsList(void) const { return fDigitsList ; }
  virtual Float_t GetEnergy() const {return fAmp; }

  void   EvalLocal2TrackingCSTransform();
  void   EvalLocalPositionFit(Double_t deff, Double_t w0, Double_t phiSlope,TClonesArray * digits);
  Bool_t EvalLocalPosition2(TClonesArray *digits, TArrayD &ed);

  static  Bool_t  EvalLocalPositionFromDigits(const Double_t esum, const Double_t deff, const Double_t w0, 
                                              TClonesArray *digits, TArrayD &ed, TVector3 &locPos);
  static  Bool_t  EvalLocalPositionFromDigits(TClonesArray *digits, TArrayD &ed, TVector3 &locPos);
  static  void    GetDeffW0(const Double_t esum, Double_t &deff,  Double_t &w0);

  virtual void    GetGlobalPosition(TVector3 & gpos, TMatrixF & gmat) const; // return global position (x, y, z) in ALICE
  virtual void    GetGlobalPosition(TVector3 & gpos) const; // return global position (x, y, z) in ALICE
  virtual void    GetLocalPosition(TVector3 & lpos) const;  // return local position  (x, y, z) in EMCAL SM
  virtual Int_t * GetPrimaries(Int_t & number) const {number = fMulTrack ; 
                                                      return fTracksList ; }
  virtual Int_t * GetParents(Int_t & number) const {number = fMulParent ; 
                                                      return fParentsList ; }

  virtual Int_t   GetDigitsMultiplicity(void) const { return fMulDigit ; }
  Int_t           GetIndexInList() const { return fIndexInList ; }

  Float_t         GetCoreEnergy()const {return fCoreEnergy ;}
  virtual Float_t GetDispersion()const {return fDispersion ;}
  virtual void    GetElipsAxis(Float_t * lambda)const {lambda[0] = fLambda[0]; lambda[1] = fLambda[1];};
  
  Float_t *   GetEnergiesList() const {return fEnergyList ;}       // gets the list of energies making this recpoint
  Double_t    GetPointEnergy() const;                              // gets point energy (sum of energy list)
  Float_t *   GetTimeList() const {return fTimeList ;}       // gets the list of digit times in this recpoint
  Float_t     GetMaximalEnergy(void) const ;                       // get the highest energy in the cluster
  Int_t       GetMaximumMultiplicity() const {return fMaxDigit ;}  // gets the maximum number of digits allowed
  Int_t       GetMultiplicity(void) const { return fMulDigit ; }   // gets the number of digits making this recpoint
  Int_t       GetMultiplicityAtLevel(Float_t level) const ;  // computes multiplicity of digits with 
  Int_t *     GetAbsId() const {return fAbsIdList;}
  Int_t       GetAbsId(int i) const {if(i>=0 && i<fMulDigit)return fAbsIdList[i]; else return -1;}
  Int_t       GetAbsIdMaxDigit() {return GetAbsId(fDigitIndMax);}
  Int_t       GetIndMaxDigit() {return fDigitIndMax;}
  void        SetIndMaxDigit(const Int_t ind) {fDigitIndMax = ind;}
  void            SetIndexInList(Int_t val) { fIndexInList = val ; }

  virtual Int_t GetSuperModuleNumber(void) const { return fSuperModuleNumber;}

  // energy above relative level
  virtual Int_t GetNumberOfLocalMax(AliEMCALDigit **  maxAt, Float_t * maxAtEnergy,
                                    Float_t locMaxCut,TClonesArray * digits ) const ; 
                                                                   // searches for the local maxima 
  
  Int_t       GetPrimaryIndex() const  ;
  Float_t     GetTime(void) const{return  fTime ; }
 
  virtual Bool_t  IsEmc(void)const { return kTRUE ;  }
  virtual Bool_t  IsSortable() const { 
    // tells that this is a sortable object
    return kTRUE ; 
  }  
  virtual void    Paint(Option_t * option="");
  virtual void    Print(Option_t * option="") const ; 
  
  static Double_t TmaxInCm(const Double_t e=0.0, const Int_t key=0);

protected:
          void  EvalCoreEnergy(Float_t logWeight,TClonesArray * digits) ;             
	  virtual void  EvalDispersion(Float_t logWeight,TClonesArray * digits) ;   // computes the dispersion of the shower
	  virtual void  EvalElipsAxis(Float_t logWeight, TClonesArray * digits );   // computes the axis of shower ellipsoide
          void  EvalTime( TClonesArray * digits );
	  virtual Bool_t AreNeighbours(AliEMCALDigit * digit1, AliEMCALDigit * digit2 ) const;
	  Float_t ThetaToEta(Float_t arg) const;  //Converts Theta (Radians) to Eta(Radians)
	  Float_t EtaToTheta(Float_t arg) const;  //Converts Eta (Radians) to Theta(Radians)

private:

  //JLK do we need this?
          AliEMCALGeometry* fGeomPtr;  //! Pointer to geometry for utilities

          Float_t fAmp ;            // summed amplitude of digits   
          Int_t   fIndexInList ;    // the index of this RecPoint in the
			            // list stored in TreeR (to be set by analysis)
          TVector3    fLocPos ;     // local position in the sub-detector coordinate
          TMatrixF *  fLocPosM ;    // covariance matrix ;
          Int_t       fMaxDigit ;   //! max initial size of digits array (not saved)
          Int_t       fMulDigit ;   // total multiplicity of digits       
          Int_t       fMaxTrack ;   //! max initial size of tracks array (not saved)
          Int_t       fMulTrack ;   // total multiplicity of tracks
          Int_t *     fDigitsList ; //[fMulDigit] list of digit's indexes from which the point was reconstructed
          Int_t *     fTracksList ; //[fMulTrack] list of tracks to which the point was assigned

          Int_t   fClusterType;    // type of cluster stored: pseudocluster or v1
	  Float_t fCoreEnergy ;       // energy in a shower core 
	  Float_t fLambda[2] ;        // shower ellipse axes
	  Float_t fDispersion ;       // shower dispersion
	  Float_t *fEnergyList ;      //[fMulDigit] energy of digits
	  Float_t *fTimeList ;        //[fMulDigit] time of digits
          Int_t   *fAbsIdList;        //[fMulDigit] absId  of digits
	  Float_t fTime ;             // Time of the digit with maximal energy deposition
	  Float_t fCoreRadius;        // The radius in which the core energy is evaluated
	  Float_t *fDETracksList ;    //[fMulTrack] list of tracks to which the point was assigned
	  Int_t fMulParent;           // Multiplicity of the parents
          Int_t fMaxParent;           // Maximum number of parents allowed
          Int_t * fParentsList;       // [fMulParent] list of the parents of the digits
          Float_t * fDEParentsList;   // [fMulParent] list of the parents of the digits
          Int_t   fSuperModuleNumber; // number identifying supermodule containing recpoint
          // Aug 16, 2007
          Int_t   fDigitIndMax;       // Index of digit with max energy in array fAbsIdList

  ClassDef(AliEMCALRecPoint,10) // RecPoint for EMCAL (Base Class)
 
};

#endif // AliEMCALRECPOINT_H
Commit	Line	Data
ab48128d	1	#ifndef ALIEMCALRECPOINT_H
ab48128d	2	#define ALIEMCALRECPOINT_H
801f6c04	3	/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
ab48128d	4	* See cxx source for full Copyright notice */
	5	//_________________________________________________________________________
	6	// Base Class for EMCAL Reconstructed Points
	7	// A recpoint being equivalent to a cluster in encal terminology
d64c959b	8	//*-- Author: Yves Schutz (SUBATECH)
70a93198	9	//*-- Author: Dmitri Peressounko (RRC KI & SUBATECH)
70a93198	10	//*-- Author: Heather Gray (LBL): merged AliEMCALRecPoint and AliEMCALTowerRecPoint 02/04
ab48128d	11
ab48128d	12	// --- ROOT system ---
9aa6a5f6	13	#include <TVector3.h>
ab48128d	14
ab48128d	15	// --- Standard library ---
	16
	17	// --- AliRoot header files ---
	18
9aa6a5f6	19	#include "AliCluster.h"
1d59832c	20	class AliEMCALDigit;
9aa6a5f6	21	class AliDigitNew;
a5c60732	22	class AliEMCALGeometry;
a5c60732	23
9aa6a5f6	24	class AliEMCALRecPoint : public AliCluster {
ab48128d	25
	26	public:
	27
	28	typedef TObjArray RecPointsList ;
	29
	30	AliEMCALRecPoint() ; // ctor
	31	AliEMCALRecPoint(const char * opt) ; // ctor
0a4cb131	32	AliEMCALRecPoint(const AliEMCALRecPoint & rp);
9aa6a5f6	33
	34	AliEMCALRecPoint& operator= (const AliEMCALRecPoint &rp);
	35
70a93198	36	virtual ~AliEMCALRecPoint();
9aa6a5f6	37
	38	virtual void AddDigit(AliDigitNew &){
	39	Fatal("AddDigit", "use AddDigit(AliEMCALDigit & digit, Float_t Energy )") ;
	40	}
6e49fcaa	41	virtual void AddDigit(AliEMCALDigit & digit, Float_t Energy);
70a93198	42	virtual Int_t Compare(const TObject * obj) const;
ab48128d	43	virtual Int_t DistancetoPrimitive(Int_t px, Int_t py);
ab48128d	44	virtual void Draw(Option_t * option="") ;
9e5d2067	45	virtual void ExecuteEvent(Int_t event, Int_t, Int_t) ;
70a93198	46
85c60a8e	47	virtual void SetClusterType(Int_t ver) { fClusterType = ver; }
	48	virtual Int_t GetClusterType() const { return fClusterType; }
	49
70a93198	50	virtual void EvalAll(Float_t logWeight, TClonesArray * digits);
1ae500a2	51	virtual void EvalLocalPosition(Float_t logWeight, TClonesArray * digits);
70a93198	52	virtual void EvalPrimaries(TClonesArray * digits) ;
87cdc3be	53	virtual void EvalParents(TClonesArray * digits) ;
70a93198	54
9aa6a5f6	55	virtual int * GetDigitsList(void) const { return fDigitsList ; }
	56	virtual Float_t GetEnergy() const {return fAmp; }
	57
	58	void EvalLocal2TrackingCSTransform();
1ae500a2	59	void EvalLocalPositionFit(Double_t deff, Double_t w0, Double_t phiSlope,TClonesArray * digits);
	60	Bool_t EvalLocalPosition2(TClonesArray *digits, TArrayD &ed);
	61
	62	static Bool_t EvalLocalPositionFromDigits(const Double_t esum, const Double_t deff, const Double_t w0,
	63	TClonesArray *digits, TArrayD &ed, TVector3 &locPos);
	64	static Bool_t EvalLocalPositionFromDigits(TClonesArray *digits, TArrayD &ed, TVector3 &locPos);
	65	static void GetDeffW0(const Double_t esum, Double_t &deff, Double_t &w0);
	66
9aa6a5f6	67	virtual void GetGlobalPosition(TVector3 & gpos, TMatrixF & gmat) const; // return global position (x, y, z) in ALICE
70a93198	68	virtual void GetGlobalPosition(TVector3 & gpos) const; // return global position (x, y, z) in ALICE
e52475ed	69	virtual void GetLocalPosition(TVector3 & lpos) const; // return local position (x, y, z) in EMCAL SM
ab48128d	70	virtual Int_t * GetPrimaries(Int_t & number) const {number = fMulTrack ;
ab48128d	71	return fTracksList ; }
e52475ed	72	virtual Int_t * GetParents(Int_t & number) const {number = fMulParent ;
87cdc3be	73	return fParentsList ; }
9aa6a5f6	74
	75	virtual Int_t GetDigitsMultiplicity(void) const { return fMulDigit ; }
	76	Int_t GetIndexInList() const { return fIndexInList ; }
	77
70a93198	78	Float_t GetCoreEnergy()const {return fCoreEnergy ;}
	79	virtual Float_t GetDispersion()const {return fDispersion ;}
	80	virtual void GetElipsAxis(Float_t * lambda)const {lambda[0] = fLambda[0]; lambda[1] = fLambda[1];};
	81
	82	Float_t * GetEnergiesList() const {return fEnergyList ;} // gets the list of energies making this recpoint
1d46d1f6	83	Double_t GetPointEnergy() const; // gets point energy (sum of energy list)
85c60a8e	84	Float_t * GetTimeList() const {return fTimeList ;} // gets the list of digit times in this recpoint
70a93198	85	Float_t GetMaximalEnergy(void) const ; // get the highest energy in the cluster
	86	Int_t GetMaximumMultiplicity() const {return fMaxDigit ;} // gets the maximum number of digits allowed
	87	Int_t GetMultiplicity(void) const { return fMulDigit ; } // gets the number of digits making this recpoint
	88	Int_t GetMultiplicityAtLevel(Float_t level) const ; // computes multiplicity of digits with
e52475ed	89	Int_t * GetAbsId() const {return fAbsIdList;}
e52475ed	90	Int_t GetAbsId(int i) const {if(i>=0 && i<fMulDigit)return fAbsIdList[i]; else return -1;}
1ae500a2	91	Int_t GetAbsIdMaxDigit() {return GetAbsId(fDigitIndMax);}
	92	Int_t GetIndMaxDigit() {return fDigitIndMax;}
	93	void SetIndMaxDigit(const Int_t ind) {fDigitIndMax = ind;}
9aa6a5f6	94	void SetIndexInList(Int_t val) { fIndexInList = val ; }
	95
	96	virtual Int_t GetSuperModuleNumber(void) const { return fSuperModuleNumber;}
	97
e52475ed	98	// energy above relative level
70a93198	99	virtual Int_t GetNumberOfLocalMax(AliEMCALDigit ** maxAt, Float_t * maxAtEnergy,
	100	Float_t locMaxCut,TClonesArray * digits ) const ;
	101	// searches for the local maxima
4635df1f	102
4635df1f	103	Int_t GetPrimaryIndex() const ;
70a93198	104	Float_t GetTime(void) const{return fTime ; }
70a93198	105
7ee5c5be	106	virtual Bool_t IsEmc(void)const { return kTRUE ; }
ab48128d	107	virtual Bool_t IsSortable() const {
	108	// tells that this is a sortable object
	109	return kTRUE ;
	110	}
	111	virtual void Paint(Option_t * option="");
5dee926e	112	virtual void Print(Option_t * option="") const ;
7ee5c5be	113
1ae500a2	114	static Double_t TmaxInCm(const Double_t e=0.0, const Int_t key=0);
ab48128d	115
ab48128d	116	protected:
70a93198	117	void EvalCoreEnergy(Float_t logWeight,TClonesArray * digits) ;
	118	virtual void EvalDispersion(Float_t logWeight,TClonesArray * digits) ; // computes the dispersion of the shower
	119	virtual void EvalElipsAxis(Float_t logWeight, TClonesArray * digits ); // computes the axis of shower ellipsoide
	120	void EvalTime( TClonesArray * digits );
	121	virtual Bool_t AreNeighbours(AliEMCALDigit * digit1, AliEMCALDigit * digit2 ) const;
	122	Float_t ThetaToEta(Float_t arg) const; //Converts Theta (Radians) to Eta(Radians)
	123	Float_t EtaToTheta(Float_t arg) const; //Converts Eta (Radians) to Theta(Radians)
	124
85c60a8e	125	private:
9aa6a5f6	126
9aa6a5f6	127	//JLK do we need this?
14ce0a6e	128	AliEMCALGeometry* fGeomPtr; //! Pointer to geometry for utilities
a5c60732	129
9aa6a5f6	130	Float_t fAmp ; // summed amplitude of digits
	131	Int_t fIndexInList ; // the index of this RecPoint in the
	132	// list stored in TreeR (to be set by analysis)
	133	TVector3 fLocPos ; // local position in the sub-detector coordinate
	134	TMatrixF * fLocPosM ; // covariance matrix ;
	135	Int_t fMaxDigit ; //! max initial size of digits array (not saved)
	136	Int_t fMulDigit ; // total multiplicity of digits
	137	Int_t fMaxTrack ; //! max initial size of tracks array (not saved)
	138	Int_t fMulTrack ; // total multiplicity of tracks
	139	Int_t * fDigitsList ; //[fMulDigit] list of digit's indexes from which the point was reconstructed
	140	Int_t * fTracksList ; //[fMulTrack] list of tracks to which the point was assigned
	141
	142	Int_t fClusterType; // type of cluster stored: pseudocluster or v1
70a93198	143	Float_t fCoreEnergy ; // energy in a shower core
70a93198	144	Float_t fLambda[2] ; // shower ellipse axes
1ae500a2	145	Float_t fDispersion ; // shower dispersion
70a93198	146	Float_t *fEnergyList ; //[fMulDigit] energy of digits
85c60a8e	147	Float_t *fTimeList ; //[fMulDigit] time of digits
e52475ed	148	Int_t *fAbsIdList; //[fMulDigit] absId of digits
70a93198	149	Float_t fTime ; // Time of the digit with maximal energy deposition
70a93198	150	Float_t fCoreRadius; // The radius in which the core energy is evaluated
af5bdd85	151	Float_t *fDETracksList ; //[fMulTrack] list of tracks to which the point was assigned
af5bdd85	152	Int_t fMulParent; // Multiplicity of the parents
87cdc3be	153	Int_t fMaxParent; // Maximum number of parents allowed
73f2fed2	154	Int_t * fParentsList; // [fMulParent] list of the parents of the digits
af5bdd85	155	Float_t * fDEParentsList; // [fMulParent] list of the parents of the digits
14ce0a6e	156	Int_t fSuperModuleNumber; // number identifying supermodule containing recpoint
1ae500a2	157	// Aug 16, 2007
1ae500a2	158	Int_t fDigitIndMax; // Index of digit with max energy in array fAbsIdList
9aa6a5f6	159
9aa6a5f6	160	ClassDef(AliEMCALRecPoint,10) // RecPoint for EMCAL (Base Class)
ab48128d	161
	162	};
	163
	164	#endif // AliEMCALRECPOINT_H