Improving event printout
[u/mrichter/AliRoot.git] / STEER / AliESDCaloCluster.h
CommitLineData
85c60a8e 1#ifndef ALIESDCALOCLUSTER_H
2#define ALIESDCALOCLUSTER_H
3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
85c60a8e 5/* $Id$ */
6/* $Log $ */
7
8//-------------------------------------------------------------------------
9// Class AliESDCaloCluster
10// This is the class to deal with during the physics analysis of data
11//
12// New container for calorimeter clusters, which are the effective
13// "tracks" for calorimeter detectors. Can be used by PHOS and EMCAL
14//
15// J.L. Klay (LLNL)
16//-------------------------------------------------------------------------
17
18#include <TObject.h>
19#include "AliPID.h"
5efdec54 20#include "TArrayS.h"
4dd59c4a 21#include "TArrayI.h"
85c60a8e 22
bab0b5f0 23class TLorentzVector;
85c60a8e 24
e649177a 25class AliESDCaloCluster : public TObject
26{
85c60a8e 27
28public:
29
30 AliESDCaloCluster();
31 AliESDCaloCluster(const AliESDCaloCluster& clus);
fe12e09c 32 AliESDCaloCluster & operator=(const AliESDCaloCluster& source);
85c60a8e 33 virtual ~AliESDCaloCluster();
732a24fe 34 virtual void Copy(TObject &) const;
85c60a8e 35
36 void SetID(Int_t id) {fID = id;}
37 Int_t GetID() const {return fID;}
38
8ada0ffe 39 //similar to AliAODCluster but offset by one for
40 // backward comp. -1 was undefined, which only applied
41 // for PHOS clusters before
42 enum ESDClu_t {kUndef = -2,
e649177a 43 kPHOSCluster,
44 kEMCALPseudoCluster, //Not any more in use, keep for backward comp.
8ada0ffe 45 kEMCALClusterv1};
e649177a 46
85c60a8e 47 void SetClusterType(Int_t type) { fClusterType = type; }
8ada0ffe 48 Char_t GetClusterType() const {return fClusterType; }
85c60a8e 49
e649177a 50 Bool_t IsEMCAL() const {return (fClusterType == kEMCALClusterv1);}
8ada0ffe 51 Bool_t IsPHOS() const {return (fClusterType == kPHOSCluster);}
85c60a8e 52
5efdec54 53 void SetPosition(const Float_t *pos) {
85c60a8e 54 fGlobalPos[0] = pos[0]; fGlobalPos[1] = pos[1]; fGlobalPos[2] = pos[2];
55 }
5efdec54 56 void GetPosition(Float_t *pos) const {
85c60a8e 57 pos[0] = fGlobalPos[0]; pos[1] = fGlobalPos[1]; pos[2] = fGlobalPos[2];
58 }
59
5efdec54 60 void SetE(Float_t ene) { fEnergy = ene;}
8ada0ffe 61 Double_t E() const { return fEnergy;}
85c60a8e 62
63 void SetClusterDisp(Float_t disp) { fDispersion = disp; }
8ada0ffe 64 Double_t GetClusterDisp() const { return fDispersion; }
85c60a8e 65
66 void SetClusterChi2(Float_t chi2) { fChi2 = chi2; }
8ada0ffe 67 Double_t GetClusterChi2() const { return fChi2; }
85c60a8e 68
69 void SetPid(const Float_t *p);
8ada0ffe 70 Double_t *GetPid() {return fPID;}
85c60a8e 71
85c60a8e 72 void SetM20(Float_t m20) { fM20 = m20; }
8ada0ffe 73 Double_t GetM20() const { return fM20; }
85c60a8e 74
75 void SetM02(Float_t m02) { fM02 = m02; }
8ada0ffe 76 Double_t GetM02() const { return fM02; }
85c60a8e 77
8ada0ffe 78 void SetNExMax(UChar_t nExMax) { fNExMax = nExMax; }
79 UChar_t GetNExMax() const { return fNExMax; }
85c60a8e 80
81 void SetEmcCpvDistance(Float_t dEmcCpv) { fEmcCpvDistance = dEmcCpv; }
8ada0ffe 82 Double_t GetEmcCpvDistance() const { return fEmcCpvDistance; }
85c60a8e 83
45636e1b 84 void SetDistanceToBadChannel(Float_t dist) {fDistToBadChannel=dist;}
8ada0ffe 85 Double_t GetDistanceToBadChannel() const {return fDistToBadChannel;}
45636e1b 86
78902954 87 void SetTOF(Double_t tof) { fTOF = tof; }
88 Double_t GetTOF() const { return fTOF; }
89
85005d58 90 void AddTracksMatched(TArrayI & array) {
91 if(!fTracksMatched)fTracksMatched = new TArrayI(array);
92 else *fTracksMatched = array;
93 }
94 void AddLabels(TArrayI & array) {
95 if(!fLabels)fLabels = new TArrayI(array) ;
96 else *fLabels = array;
97}
78902954 98
4dd59c4a 99 TArrayI * GetTracksMatched() const {return fTracksMatched;}
100 TArrayI * GetLabels() const {return fLabels;}
78902954 101
5efdec54 102 Int_t GetTrackMatched() const
103 {if( fTracksMatched && fTracksMatched->GetSize() >0) return fTracksMatched->At(0);
104 else return -1;} //Most likely the track associated to the cluster
105 Int_t GetLabel() const
106 {if( fLabels && fLabels->GetSize() >0) return fLabels->At(0);
107 else return -1;} //Most likely the track associated to the cluster
78902954 108
5efdec54 109 Int_t GetNTracksMatched() const {if (fTracksMatched) return fTracksMatched->GetSize();
110 else return -1;}
111 Int_t GetNLabels() const { if (fLabels) return fLabels->GetSize();
112 else return -1;}
e649177a 113
5efdec54 114 void GetMomentum(TLorentzVector& p, Double_t * vertexPosition );
e649177a 115
116 // --- NEW ---
117 void SetNCells(Int_t n) { fNCells = n;}
0fdeb120 118 Int_t GetNCells() const { return fNCells;}
e649177a 119
ed712271 120 void SetCellsAbsId(UShort_t *array) ;
e649177a 121 UShort_t *GetCellsAbsId() {return fCellsAbsId;}
122
ed712271 123 void SetCellsAmplitudeFraction(Double32_t *array) ;
e649177a 124 Double32_t *GetCellsAmplitudeFraction() {return fCellsAmpFraction;}
125
126 Int_t GetCellAbsId(Int_t i) const {
127 if (fCellsAbsId && i >=0 && i < fNCells ) return fCellsAbsId[i];
128 else return -1;}
129
130 Double_t GetCellAmplitudeFraction(Int_t i) const {
131 if (fCellsAmpFraction && i >=0 && i < fNCells ) return fCellsAmpFraction[i];
132 else return -1;}
133
134 //_____________________________________________________
135 //Not used anymore, kept to avoid backward incompatibility
136 void AddDigitIndex(TArrayS & array) { fDigitIndex = new TArrayS(array) ; Warning("AddDigitAmplitude","This method is no more in use") ;}
137 void AddDigitAmplitude(TArrayS & array) { fDigitAmplitude = new TArrayS(array) ; Warning("AddDigitAmplitude","This method is no more in use") ;}
138 void AddDigitTime(TArrayS & array) { fDigitTime = new TArrayS(array) ;Warning("AddDigitTime","This method is no more in use") ;}
139 TArrayS * GetDigitAmplitude() const {return fDigitAmplitude;}
140 TArrayS * GetDigitTime() const {return fDigitTime;}
141 TArrayS * GetDigitIndex() const {return fDigitIndex;}
142 Int_t GetNumberOfDigits() const { return -1;}
143 //_____________________________________________________
bab0b5f0 144
85c60a8e 145protected:
146
4dd59c4a 147 TArrayI * fTracksMatched; //Index of tracks close to cluster. First entry is the most likely match.
148 TArrayI * fLabels; //list of primaries that generated the cluster, ordered in deposited energy.
e649177a 149
150 //NEW
151 Int_t fNCells ;
152 UShort_t *fCellsAbsId; //[fNCells] array of cell absId numbers
153 Double32_t *fCellsAmpFraction; //[fNCells][0.,1.,16] array with cell amplitudes fraction.
154
155 //__________________________________________________________
156 //Not in use
5efdec54 157 TArrayS * fDigitAmplitude; //digit energy (integer units)
158 TArrayS * fDigitTime; //time of this digit (integer units)
159 TArrayS * fDigitIndex; //calorimeter digit index
e649177a 160 //_________________________________________________________
8ada0ffe 161
162 Double32_t fGlobalPos[3]; // position in global coordinate systemD
163 Double32_t fEnergy; // energy measured by calorimeter
164 Double32_t fDispersion; // cluster dispersion, for shape analysis
165 Double32_t fChi2; // chi2 of cluster fi
166 Double32_t fM20; // 2-nd moment along the main eigen axis
167 Double32_t fM02; // 2-nd moment along the second eigen axis
78902954 168
8ada0ffe 169 Double32_t fEmcCpvDistance; // the distance from PHOS EMC rec.point to the closest CPV rec.point
170 Double32_t fDistToBadChannel; // Distance to nearest bad channel
171 Double32_t fPID[AliPID::kSPECIESN]; //[0,1,8]"detector response probabilities" (for the PID)
172 Int_t fID; // Unique Id of the cluster
173 UChar_t fNExMax ; // number of (Ex-)maxima before unfolding
174 Char_t fClusterType; // Flag for different cluster type/versions
78902954 175 Double32_t fTOF; //[0,0,12] time-of-flight
8ada0ffe 176
78902954 177 ClassDef(AliESDCaloCluster,8) //ESDCaloCluster
85c60a8e 178};
179
180#endif
181