Copy arrays in assignment instead of the pointer; avoid double delete.
[u/mrichter/AliRoot.git] / STEER / AliAODCaloCluster.h
1 #ifndef AliAODCaloCluster_H
2 #define AliAODCaloCluster_H
3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 //-------------------------------------------------------------------------
9 //     AOD calorimeter cluster class (for PHOS and EMCAL)
10 //     Author: Markus Oldenburg, CERN, 
11 //             Gustavo Conesa, INFN
12 //-------------------------------------------------------------------------
13
14 #include "AliAODCluster.h"
15
16 #include <TRefArray.h>
17 #include <TArrayS.h>
18
19 class TLorentzVector;
20
21 class AliAODCaloCluster : public AliAODCluster {
22
23  public:
24   
25   AliAODCaloCluster();
26   AliAODCaloCluster(Int_t id,
27                     UInt_t nLabel,
28                     Int_t *label,
29                     Double_t energy,
30                     Double_t x[3],
31                     Double_t pid[9],
32                     Char_t ttype=kUndef,
33                     UInt_t selectInfo=0);
34   
35   AliAODCaloCluster(Int_t id,
36                     UInt_t nLabel,
37                     Int_t *label,
38                     Float_t energy,
39                     Float_t x[3],
40                     Float_t pid[9],
41                     Char_t ttype=kUndef,
42                     UInt_t selectInfo=0);
43   
44   virtual ~AliAODCaloCluster();
45   AliAODCaloCluster(const AliAODCaloCluster& clus); 
46   AliAODCaloCluster& operator=(const AliAODCaloCluster& clus);
47
48   
49   // getters
50   Double_t GetDistToBadChannel() const { return fDistToBadChannel; }
51   Double_t GetDispersion() const { return fDispersion; }
52   Double_t GetM20() const { return fM20; }
53   Double_t GetM02() const { return fM02; }
54   Double_t GetEmcCpvDistance() const { return fEmcCpvDistance; }
55   UShort_t GetNExMax() const { return fNExMax; }
56   Double_t GetTOF() const { return fTOF; }
57
58   Int_t    GetNTracksMatched() const { return fTracksMatched.GetEntriesFast(); }
59   TObject *GetTrackMatched(Int_t i) const { return fTracksMatched.At(i); }
60  
61   void SetNCells(Int_t n) { fNCells = n;}
62   Double_t GetNCells() const   { return fNCells;}
63   
64   void SetCellsAbsId(UShort_t *array);
65   UShort_t *GetCellsAbsId() {return  fCellsAbsId;}
66   
67   void SetCellsAmplitudeFraction(Double32_t *array);
68   Double32_t *GetCellsAmplitudeFraction() {return  fCellsAmpFraction;}
69   
70   Int_t GetCellAbsId(Int_t i) const {  
71     if (fCellsAbsId && i >=0 && i < fNCells ) return fCellsAbsId[i];    
72     else return -1;}
73   
74   Double_t GetCellAmplitudeFraction(Int_t i) const {  
75     if (fCellsAmpFraction && i >=0 && i < fNCells ) return fCellsAmpFraction[i];    
76     else return -1;}
77
78   // setters
79   void SetDistToBadChannel(Double_t dist) { fDistToBadChannel = dist; }
80   void SetDispersion(Double_t disp) { fDispersion = disp; }
81   void SetM20(Double_t m20) { fM20 = m20; }
82   void SetM02(Double_t m02) { fM02 = m02; }
83   void SetEmcCpvDistance(Double_t emcCpvDist) { fEmcCpvDistance = emcCpvDist; }
84   void SetNExMax(UShort_t nExMax) { fNExMax = nExMax; }
85   void SetTOF(Double_t tof) { fTOF = tof; }
86
87   void SetCaloCluster(Double_t dist = -999., 
88                       Double_t disp = -1., 
89                       Double_t m20 = 0., 
90                       Double_t m02 = 0., 
91                       Double_t emcCpvDist = -999., 
92                       UShort_t nExMax = 0, 
93                       Double_t tof = 0.) 
94   {
95     fDistToBadChannel = dist;
96     fDispersion = disp;
97     fM20 = m20;
98     fM02 = m02;
99     fEmcCpvDistance = emcCpvDist;
100     fNExMax = nExMax;
101     fTOF = tof ;
102   }
103   
104   void GetMomentum(TLorentzVector& p, Double_t * vertexPosition );
105
106   void AddTrackMatched(TObject *trk) { fTracksMatched.Add(trk); }
107   void RemoveTrackMatched(TObject *trk) { fTracksMatched.Remove(trk); }
108   Bool_t HasTrackMatched(TObject *trk) const;
109
110  private :
111
112   Double32_t   fDistToBadChannel; // Distance to nearest bad channel
113   Double32_t   fDispersion;       // cluster dispersion, for shape analysis
114   Double32_t   fM20;              // 2-nd moment along the main eigen axis
115   Double32_t   fM02;              // 2-nd moment along the second eigen axis
116   Double32_t   fEmcCpvDistance;   // the distance from PHOS EMC rec.point to the closest CPV rec.point
117   UShort_t     fNExMax;           // number of (Ex-)maxima before unfolding
118   Double32_t fTOF;        ////[0,0,12] time-of-flight
119
120   TRefArray    fTracksMatched;    // references to tracks close to cluster. First entry is the most likely match.
121
122   Int_t       fNCells ;
123   UShort_t   *fCellsAbsId;        //[fNCells] array of cell absId numbers
124   Double32_t *fCellsAmpFraction;  //[fNCells][0.,1.,16] array with cell amplitudes fraction.
125
126   ClassDef(AliAODCaloCluster,5);
127 };
128
129 #endif