4b046a3bcbc337bc1768b33dc218fce73ca85f29
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALJetFinderInput.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 // Initial JetFinder input object
21 //
22 //*-- Author: Mark Horner (LBL/UCT)
23
24 #include "AliEMCALJetFinderInput.h"
25 #include "AliEMCALDigit.h"
26 #include "AliEMCALParton.h"
27 #include "TClonesArray.h"
28
29
30 ClassImp(AliEMCALJetFinderInput)
31
32 AliEMCALJetFinderInput::AliEMCALJetFinderInput()
33 {
34 // Default constructor
35 // Defines all TClonesArrays  
36 // Creates full array of Digits - all other arrays will have objects created as necessary 
37
38 if (fDebug>0) Info("AliEMCALJetFinderInput","Beginning Constructor");   
39
40 fInitialised = kFALSE;
41 fNDigits = 96*144;     // This is the number of digits
42 fNMaxDigits = fNDigits;
43 fNMaxTracks = 3000;
44 fNMaxParticles = 2000;
45 fNMaxPartons = 4;
46 fNTracks        = 0;
47 fNPartons       = 0;
48 fNParticles     = 0;
49
50 }
51
52 void AliEMCALJetFinderInput::InitArrays()
53 {
54 if (fDebug>0) Info("AliEMCALJetFinderInput","Beginning InitArrays");
55 fNDigits = 96*144;     // This is the number of digits
56 fNMaxDigits = fNDigits;
57 fNMaxTracks = 3000;
58 fNMaxParticles = 2000;
59 fNMaxPartons = 4;
60 for (Int_t i = 0; i < 96*144; i++)
61 {
62         AliEMCALDigit tempdigit(0,0,i+1,0,1.0,-1);
63         tempdigit.SetAmp(0);
64 //       AliEMCALDigit(0,0,i+1,0,1.0,-1)
65         new(&fDigitsArray[i]) AliEMCALDigit(tempdigit); // Default digit amplitude is zero
66 }
67 fNTracks        = 0;
68 fNPartons       = 0;
69 fNParticles     = 0;
70 fInitialised = kTRUE;
71 }
72
73
74 AliEMCALJetFinderInput::~AliEMCALJetFinderInput()
75 {
76 // Destructor -
77
78 if (fDebug>0) Info("~AliEMCALJetFinderInput","Beginning Destructor");   
79 }
80
81 void AliEMCALJetFinderInput::Reset(AliEMCALJetFinderResetType_t resettype)
82 {
83 // Clears the contents of all the arrays and returns to the state we were in 
84 // after the constructor
85 if (fDebug>1) Info("Reset","Beginning Reset");  
86   
87   if (!fInitialised) InitArrays();      
88
89   if (  resettype == kResetData   ||    
90         resettype == kResetDigits ||
91         resettype == kResetAll   ){
92           for (Int_t i = 0; i < 96*144; i++)
93                   {
94                         AliEMCALDigit tempdigit(0,0,i+1,0,1.0,-1);
95                         tempdigit.SetAmp(0);
96                         new(&fDigitsArray[i]) AliEMCALDigit(tempdigit); // Default digit amplitude is zero
97                   }
98   }
99   if (  resettype == kResetData   ||    
100         resettype == kResetTracks ||
101         resettype == kResetAll   ){
102           fNTracks = 0;
103   }
104  if (  resettype == kResetData    ||
105        resettype == kResetPartons ||
106        resettype == kResetAll   ){
107          fNPartons=0;
108  }
109  if (  resettype == kResetData      ||
110        resettype == kResetParticles ||
111        resettype == kResetAll   ){
112          fNParticles = 0;
113  }
114
115 }
116 void AliEMCALJetFinderInput::AddEnergyToDigit(Int_t digitID,Int_t denergy)
117 {
118 // Adds energy to the digit specified - Note these are tower IDs and
119 // so start at 1!
120                 
121 if (fDebug>15) Info("AddEnergyToDigit","Beginning AddEnergyToDigit");
122  if (!fInitialised) InitArrays();       
123
124  Int_t amp = 0;
125  AliEMCALDigit *mydigit = &fDigitsArray[digitID];
126  amp = mydigit->GetAmp();
127  mydigit->SetAmp(amp+denergy);
128 }       
129
130 void AliEMCALJetFinderInput::AddTrack(TParticle track)
131 {
132 // Adds a TParticle to the track array
133                 
134 if (fDebug>5) Info("AddTrack","Beginning AddTrack");
135  
136         if (!fInitialised) InitArrays();        
137         if (fNTracks < fNMaxTracks){  
138                 new(&fTracksArray[fNTracks]) TParticle(track);
139                 fNTracks++;
140                 if (fDebug>5) Info("AddTracks","Added Tracks %i",fNTracks);     
141         }else
142         {
143                 Error("AddTracks","Cannot AddTrack - maximum exceeded");
144         }
145
146 }
147 void AliEMCALJetFinderInput::AddParton(AliEMCALParton *parton)
148 {
149 // Adds an AliEMCALParton to the parton array 
150         
151 if (fDebug>5) Info("AddParton","Beginning AddParton");
152
153         if (!fInitialised) InitArrays();        
154         if (fNPartons < fNMaxPartons){  
155                 new(&fPartonsArray[fNPartons]) AliEMCALParton(*parton);
156                 fNPartons++;
157                 if (fDebug>5) Info("AddParton","Added Parton %i",fNPartons);    
158         }else
159         {
160                 Error("AddParton","Cannot AddParton - maximum exceeded");
161         }
162 }
163
164 void AliEMCALJetFinderInput::AddParticle(TParticle *particle)
165 {
166 // Adds a TParticle to the particle array
167         
168 if (fDebug>5) Info("AddParticle","Beginning AddParticle");      
169
170         if (!fInitialised) InitArrays();        
171         if (fNParticles < fNMaxParticles){  
172                 new(&fParticlesArray[fNParticles]) TParticle(*particle);
173                 fNParticles++;
174                 if (fDebug>5) Info("AddParticle","Added Particle %i",fNParticles);      
175         }else
176         {
177                 Error("AddParticle","Cannot AddParticle - maximum exceeded");
178         }
179
180 }
181
182
183 AliEMCALDigit* AliEMCALJetFinderInput::GetDigit(Int_t digitID)
184 {
185 // Returns a pointer to an AliEMCALDigit with the given ID
186         
187 if (fDebug>15) Info("GetDigit","Beginning GetDigit");
188
189     if (digitID >= fNDigits) return 0;
190     return (AliEMCALDigit*)(&fDigitsArray[digitID]);
191 }
192
193 TParticle* AliEMCALJetFinderInput::GetTrack(Int_t trackID)
194 {
195 // Returns a pointer to a TParticle specified by the ID
196         
197 if (fDebug>15) Info("GetTrack","Beginning GetTrack with trackID %i",trackID);
198
199     if (trackID >= fNTracks) return 0;
200     return (TParticle*)(&fTracksArray[trackID]);
201 }
202
203 AliEMCALParton* AliEMCALJetFinderInput::GetParton(Int_t partonID)
204 {
205 // Returns a pointer to an AliEMCALParton  specified by the ID
206         
207 if (fDebug>15) Info("GetParton","Beginning GetParton");
208
209     if (partonID >= fNPartons) return 0;
210     return (AliEMCALParton*)(&fPartonsArray[partonID]);
211 }
212
213 TParticle* AliEMCALJetFinderInput::GetParticle(Int_t particleID)
214 {
215 // Returns a pointer to a TParticle specified by the ID
216         
217 if (fDebug>15) Info("GetParticle","Beginning GetParticle");
218
219     if (particleID >= fNParticles) return 0;
220     return (TParticle*)(&fParticlesArray[particleID]);
221 }
222
223
224