Add information about digits in neighborhood of the cluster (M.Ivanov)
[u/mrichter/AliRoot.git] / TRD / AliTRDcluster.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 /* $Id$ */
17
18  
19 ///////////////////////////////////////////////////////////////////////////////
20 //                                                                           //
21 //  TRD cluster                                                              //
22 //                                                                           //
23 /////////////////////////////////////////////////////////////////////////////// 
24
25 #include "AliTRDcluster.h"
26 #include "AliTRDrecPoint.h"
27
28 ClassImp(AliTRDcluster)
29
30
31   //___________________________________________________________________________
32
33   AliTRDcluster::AliTRDcluster() : AliCluster() { 
34   //
35   // default constructor
36   //
37   fQ=0; 
38   fTimeBin=0; 
39   fDetector=0; 
40   fNPads=0; 
41   for (Int_t i = 0;i<7; i++) fSignals[i]=0;
42 }
43 //_____________________________________________________________________________
44   AliTRDcluster::AliTRDcluster(const AliTRDrecPoint &p):AliCluster()
45 {
46   //
47   // Constructor from AliTRDrecPoint
48   //
49
50   fDetector   = p.GetDetector();
51   fTimeBin    = p.GetLocalTimeBin();
52
53   fTracks[0]  = p.GetTrackIndex(0);
54   fTracks[1]  = p.GetTrackIndex(1);
55   fTracks[2]  = p.GetTrackIndex(2);
56
57   fQ          = p.GetEnergy();
58
59   fY          = p.GetY();
60   fZ          = p.GetZ();
61   fSigmaY2    = p.GetSigmaY2();
62   fSigmaZ2    = p.GetSigmaZ2();  
63
64   fSigmaY2    = 0.2;
65   fSigmaZ2    = 5.;  
66   fNPads      =0;
67   fCenter     = 0;
68 }
69
70 //_____________________________________________________________________________
71 AliTRDcluster::AliTRDcluster(const AliTRDcluster &c):AliCluster()
72 {
73   //
74   // Copy constructor 
75   //
76
77   fTracks[0]  = c.GetLabel(0);
78   fTracks[1]  = c.GetLabel(1);
79   fTracks[2]  = c.GetLabel(2);
80
81   fY          = c.GetY();
82   fZ          = c.GetZ();
83   fSigmaY2    = c.GetSigmaY2();
84   fSigmaZ2    = c.GetSigmaZ2();  
85
86   fDetector   = c.GetDetector();
87   fTimeBin    = c.GetLocalTimeBin();
88   fQ          = c.GetQ();
89   fNPads      = c.fNPads;
90   fCenter     = c.fCenter;
91   for (Int_t i=0;i<7;i++) fSignals[i] = c.fSignals[i];
92 }
93
94 //_____________________________________________________________________________
95 void AliTRDcluster::AddTrackIndex(Int_t *track)
96 {
97   //
98   // Adds track index. Currently assumed that track is an array of
99   // size 9, and up to 3 track indexes are stored in fTracks[3].
100   // Indexes are sorted according to:
101   //  1) index of max number of appearances is stored first
102   //  2) if two or more indexes appear equal number of times, the lowest
103   //     ones are stored first;
104   //
105
106   const Int_t kSize = 9;
107
108   Int_t entries[kSize][2], i, j, index;
109
110   Bool_t indexAdded;
111
112   for (i=0; i<kSize; i++) {
113     entries[i][0]=-1;
114     entries[i][1]=0;
115   }                                 
116
117   for (Int_t k=0; k<kSize; k++) {
118     index=track[k];
119     indexAdded=kFALSE; 
120     j=0;
121     if (index >= 0) {
122       while ( (!indexAdded) && ( j < kSize ) ) {
123         if ((entries[j][0]==index) || (entries[j][1]==0)) {
124           entries[j][0]=index;
125           entries[j][1]=entries[j][1]+1;
126           indexAdded=kTRUE;
127         }
128         j++;
129       }
130     }
131   }             
132
133   // sort by number of appearances and index value
134   Int_t swap=1, tmp0, tmp1;
135   while ( swap > 0) {
136     swap=0;
137     for(i=0; i<(kSize-1); i++) {
138       if ((entries[i][0] >= 0) && (entries[i+1][0] >= 0)) {
139         if ((entries[i][1] < entries[i+1][1]) ||
140             ((entries[i][1] == entries[i+1][1]) &&
141              (entries[i][0] > entries[i+1][0]))) {
142                tmp0=entries[i][0];
143                tmp1=entries[i][1];
144                entries[i][0]=entries[i+1][0];
145                entries[i][1]=entries[i+1][1];
146                entries[i+1][0]=tmp0;
147                entries[i+1][1]=tmp1;
148                swap++;
149         }
150       }
151     }
152   }               
153
154   // set track indexes
155   for(i=0; i<3; i++) SetLabel(entries[i][0],i);
156
157   return;
158
159 }          
160
161 void AliTRDcluster::SetSignals(Short_t*signals){
162   //
163   // write signals in the cluster
164   //
165   for (Int_t i = 0;i<7;i++) fSignals[i]=signals[i];
166 }
167
168 Float_t AliTRDcluster::GetSumS() const
169 {
170   //
171   // return total charge in non unfolded cluster
172   //
173   Float_t sum=0;
174   for (Int_t i = 0;i<7;i++) sum+=fSignals[i];
175   return sum;
176 }
177 Float_t AliTRDcluster::GetCenterS() const
178 {
179   //
180   //
181   //
182   Float_t sum=0;
183   Float_t sum2=0;
184   for (Int_t i = 0;i<7;i++) {    
185     sum+=fSignals[i];
186     sum2+=i*fSignals[i];
187   }
188   if (sum>0) return sum2/sum-2;
189   return 0;
190
191 }