- Update to PYQUEN v1.3 (http://lokhtin.web.cern.ch/lokhtin/pyquen/pyquen.txt)
[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 */
5
6/* $Id$ */
7/* $Log $ */
8
9//-------------------------------------------------------------------------
10// Class AliESDCaloCluster
11// This is the class to deal with during the physics analysis of data
12//
13// New container for calorimeter clusters, which are the effective
14// "tracks" for calorimeter detectors. Can be used by PHOS and EMCAL
15//
16// J.L. Klay (LLNL)
17//-------------------------------------------------------------------------
18
19#include <TObject.h>
20#include "AliPID.h"
5efdec54 21#include "TArrayS.h"
85c60a8e 22
bab0b5f0 23class TLorentzVector;
85c60a8e 24
25class AliESDCaloCluster : public TObject {
26
27public:
28
29 AliESDCaloCluster();
30 AliESDCaloCluster(const AliESDCaloCluster& clus);
fe12e09c 31 AliESDCaloCluster & operator=(const AliESDCaloCluster& source);
85c60a8e 32 virtual ~AliESDCaloCluster();
33
34 void SetID(Int_t id) {fID = id;}
35 Int_t GetID() const {return fID;}
36
90319d49 37 enum ClusterType {kPseudoCluster, kClusterv1};//Two types of clusters stored
38 //in EMCAL.
85c60a8e 39 void SetClusterType(Int_t type) { fClusterType = type; }
40 Int_t GetClusterType() const {return fClusterType; }
41
42 void SetEMCAL(Bool_t emc) { fEMCALCluster = emc;}
43 Bool_t IsEMCAL() const {return fEMCALCluster;}
44
45 void SetPHOS(Bool_t phos) { fPHOSCluster = phos;}
46 Bool_t IsPHOS() const {return fPHOSCluster;}
47
5efdec54 48 void SetPosition(const Float_t *pos) {
85c60a8e 49 fGlobalPos[0] = pos[0]; fGlobalPos[1] = pos[1]; fGlobalPos[2] = pos[2];
50 }
5efdec54 51 void GetPosition(Float_t *pos) const {
85c60a8e 52 pos[0] = fGlobalPos[0]; pos[1] = fGlobalPos[1]; pos[2] = fGlobalPos[2];
53 }
54
5efdec54 55 void SetE(Float_t ene) { fEnergy = ene;}
56 Float_t E() const { return fEnergy;}
85c60a8e 57
58 void SetClusterDisp(Float_t disp) { fDispersion = disp; }
59 Float_t GetClusterDisp() const { return fDispersion; }
60
61 void SetClusterChi2(Float_t chi2) { fChi2 = chi2; }
62 Float_t GetClusterChi2() const { return fChi2; }
63
64 void SetPid(const Float_t *p);
65 Float_t *GetPid() {return fPID;}
66
85c60a8e 67 void SetM20(Float_t m20) { fM20 = m20; }
68 Float_t GetM20() const { return fM20; }
69
70 void SetM02(Float_t m02) { fM02 = m02; }
71 Float_t GetM02() const { return fM02; }
72
73 void SetM11(Float_t m11) { fM11 = m11; }
74 Float_t GetM11() const { return fM11; }
75
76 void SetNExMax(UShort_t nExMax) { fNExMax = nExMax; }
77 UShort_t GetNExMax() const { return fNExMax; }
78
79 void SetEmcCpvDistance(Float_t dEmcCpv) { fEmcCpvDistance = dEmcCpv; }
80 Float_t GetEmcCpvDistance() const { return fEmcCpvDistance; }
81
45636e1b 82 void SetDistanceToBadChannel(Float_t dist) {fDistToBadChannel=dist;}
83 Float_t GetDistanceToBadChannel() const {return fDistToBadChannel;}
84
5efdec54 85 void AddTracksMatched(TArrayS & array) { fTracksMatched = new TArrayS(array) ; }
86 void AddLabels(TArrayS & array) { fLabels = new TArrayS(array) ; }
87 void AddDigitAmplitude(TArrayS & array) { fDigitAmplitude = new TArrayS(array) ; }
88 void AddDigitTime(TArrayS & array) { fDigitTime = new TArrayS(array) ; }
89 void AddDigitIndex(TArrayS & array) { fDigitIndex = new TArrayS(array) ; }
90
91 TArrayS * GetTracksMatched() const {return fTracksMatched;}
92 TArrayS * GetLabels() const {return fLabels;}
93 TArrayS * GetDigitAmplitude() const {return fDigitAmplitude;}
94 TArrayS * GetDigitTime() const {return fDigitTime;}
95 TArrayS * GetDigitIndex() const {return fDigitIndex;}
96
97 Int_t GetTrackMatched() const
98 {if( fTracksMatched && fTracksMatched->GetSize() >0) return fTracksMatched->At(0);
99 else return -1;} //Most likely the track associated to the cluster
100 Int_t GetLabel() const
101 {if( fLabels && fLabels->GetSize() >0) return fLabels->At(0);
102 else return -1;} //Most likely the track associated to the cluster
103
104
105 Int_t GetNTracksMatched() const {if (fTracksMatched) return fTracksMatched->GetSize();
106 else return -1;}
107 Int_t GetNLabels() const { if (fLabels) return fLabels->GetSize();
108 else return -1;}
109 Int_t GetNumberOfDigits() const { if (fDigitAmplitude) return fDigitAmplitude->GetSize();
110 else return -1;}
111
112 void GetMomentum(TLorentzVector& p, Double_t * vertexPosition );
cb8cf003 113 // Sep 7, 2007
114 Int_t GetTrueDigitAmplitude(Int_t i, Double_t cc);
115 Double_t GetTrueDigitEnergy(Int_t i, Double_t cc);
116 Double_t GetRecalibratedDigitEnergy(Int_t i, Double_t ccOld, Double_t ccNew);
bab0b5f0 117
85c60a8e 118protected:
119
120 Int_t fID; // Unique Id of the cluster
121 Int_t fClusterType; // Flag for different clustering versions
122 Bool_t fEMCALCluster; // Is this is an EMCAL cluster?
123 Bool_t fPHOSCluster; // Is this is a PHOS cluster?
124 Float_t fGlobalPos[3]; // position in global coordinate system
125 Float_t fEnergy; // energy measured by calorimeter
126 Float_t fDispersion; // cluster dispersion, for shape analysis
127 Float_t fChi2; // chi2 of cluster fit
128 Float_t fPID[AliPID::kSPECIESN]; //"detector response probabilities" (for the PID)
85c60a8e 129 Float_t fM20; // 2-nd moment along the main eigen axis
130 Float_t fM02; // 2-nd moment along the second eigen axis
131 Float_t fM11; // 2-nd mixed moment Mxy
132 UShort_t fNExMax ; // number of (Ex-)maxima before unfolding
133 Float_t fEmcCpvDistance; // the distance from PHOS EMC rec.point to the closest CPV rec.point
45636e1b 134 Float_t fDistToBadChannel; // Distance to nearest bad channel
85c60a8e 135
5efdec54 136 TArrayS * fTracksMatched; //Index of tracks close to cluster. First entry is the most likely match.
137 TArrayS * fLabels; //list of primaries that generated the cluster, ordered in deposited energy.
138 TArrayS * fDigitAmplitude; //digit energy (integer units)
139 TArrayS * fDigitTime; //time of this digit (integer units)
140 TArrayS * fDigitIndex; //calorimeter digit index
85c60a8e 141
b9a3b656 142 ClassDef(AliESDCaloCluster,4) //ESDCaloCluster
85c60a8e 143};
144
145#endif
146