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 // --
  25 // --
  26
  27
  28
  29 #include "AliEMCALJetFinderInput.h"
  30 #include "AliEMCALDigit.h"
  31 #include "AliEMCALGeometry.h"
  32 #include "AliEMCALParton.h"
  33 class TClonesArray;
  34
  35
  36 ClassImp(AliEMCALJetFinderInput)
  37
  38 AliEMCALJetFinderInput::AliEMCALJetFinderInput():
  39 fDigitsArray(0),
  40 fTracksArray(0),
  41 fPartonsArray(0),
  42 fParticlesArray(0)
  43 {
  44 // Default constructor
  45 // Defines all TClonesArrays
  46 // Creates full array of Digits - all other arrays will have objects created as necessary
  47
  48   //if (fDebug>0) Info("AliEMCALJetFinderInput","Beginning Constructor");
  49   fDebug=0;
  50   fInitialised = kFALSE;
  51   fNDigits = 19152;     // This is the number of digits
  52   fNMaxDigits = fNDigits;
  53   fNMaxTracks = 3000;
  54   fNMaxParticles = 2000;
  55   fNMaxPartons = 4;
  56   fNTracks        = 0;
  57   fNPartons       = 0;
  58   fNParticles     = 0;
  59
  60 }
  61
  62 void AliEMCALJetFinderInput::InitArrays()
  63 {
  64 if (fDebug>0) Info("AliEMCALJetFinderInput","Beginning InitArrays");
  65 fNDigits = 19152;     // This is the number of digits
  66 fNMaxDigits = fNDigits;
  67 fNMaxTracks = 3000;
  68 fNMaxParticles = 2000;
  69 fNMaxPartons = 4;
  70
  71 fDigitsArray  = new TClonesArray ("AliEMCALDigit",fNDigits);     // This is the digits array for the EMCAL
  72 fTracksArray  = new TClonesArray("TParticle",fNMaxTracks);
  73 fPartonsArray = new TClonesArray("AliEMCALParton",fNMaxPartons);
  74 fParticlesArray       = new TClonesArray ("TParticle",fNMaxParticles);
  75
  76 for (Int_t i = 0; i < 19152; i++)
  77 {
  78         AliEMCALDigit tempdigit(0,0,i+1,0,1.0,-1);
  79         tempdigit.SetAmp(0);
  80 //       AliEMCALDigit(0,0,i+1,0,1.0,-1)
  81         new((*fDigitsArray)[i]) AliEMCALDigit(tempdigit);       // Default digit amplitude is zero
  82 }
  83 fNTracks        = 0;
  84 fNPartons       = 0;
  85 fNParticles     = 0;
  86 fInitialised = kTRUE;
  87 }
  88
  89
  90 AliEMCALJetFinderInput::~AliEMCALJetFinderInput()
  91 {
  92 // Destructor -
  93
  94 if (fDebug>0) Info("~AliEMCALJetFinderInput","Beginning Destructor");
  95 if (fInitialised)
  96  {
  97        fDigitsArray->Delete();
  98        fTracksArray->Delete();
  99        fPartonsArray->Delete();
 100        fParticlesArray->Delete();
 101        delete fDigitsArray;
 102        delete fTracksArray;
 103        delete fPartonsArray;
 104        delete fParticlesArray;
 105 }
 106
 107 }
 108
 109 void AliEMCALJetFinderInput::Reset(AliEMCALJetFinderResetType_t resettype)
 110 {
 111 // Clears the contents of all the arrays and returns to the state we were in
 112 // after the constructor
 113 if (fDebug>1) Info("Reset","Beginning Reset");
 114
 115   if (!fInitialised) InitArrays();
 116
 117   if (  resettype == kResetData   ||
 118         resettype == kResetDigits ||
 119         resettype == kResetAll   ){
 120                 if (fInitialised)
 121                 {
 122                         fDigitsArray->Delete();
 123                         fTracksArray->Delete();
 124                         fPartonsArray->Delete();
 125                         fParticlesArray->Delete();
 126                 }
 127                 for (Int_t i = 0; i < 19152; i++)
 128                 {
 129                         AliEMCALDigit tempdigit(0,0,i+1,0,1.0,-1);
 130                         tempdigit.SetAmp(0);
 131                         new((*fDigitsArray)[i]) AliEMCALDigit(tempdigit);       // Default digit amplitude is zero
 132                 }
 133   }
 134   if (  resettype == kResetData   ||
 135         resettype == kResetTracks ||
 136         resettype == kResetAll   ){
 137           fNTracks = 0;
 138   }
 139  if (  resettype == kResetData    ||
 140        resettype == kResetPartons ||
 141        resettype == kResetAll   ){
 142          fNPartons=0;
 143  }
 144  if (  resettype == kResetData      ||
 145        resettype == kResetParticles ||
 146        resettype == kResetAll   ){
 147          fNParticles = 0;
 148  }
 149
 150 }
 151 void AliEMCALJetFinderInput::AddEnergyToDigit(Int_t digitID,Int_t denergy)
 152 {
 153 // Adds energy to the digit specified - Note these are tower IDs and
 154 // so start at 1!
 155
 156 if (fDebug>15) Info("AddEnergyToDigit","Beginning AddEnergyToDigit");
 157  if (!fInitialised) InitArrays();
 158
 159  Int_t amp = 0;
 160  AliEMCALDigit *mydigit = (AliEMCALDigit*)(*fDigitsArray)[digitID];
 161  amp = mydigit->GetAmp();
 162  mydigit->SetAmp(amp+denergy);
 163 }
 164
 165 void AliEMCALJetFinderInput::AddTrack(TParticle track)
 166 {
 167 // Adds a TParticle to the track array
 168
 169 if (fDebug>5) Info("AddTrack","Beginning AddTrack");
 170
 171         if (!fInitialised) InitArrays();
 172         if (fNTracks < fNMaxTracks){
 173                 new((*fTracksArray)[fNTracks]) TParticle(track);
 174                 fNTracks++;
 175                 if (fDebug>5) Info("AddTracks","Added Tracks %i",fNTracks);
 176         }else
 177         {
 178                 Error("AddTracks","Cannot AddTrack - maximum exceeded");
 179         }
 180
 181 }
 182
 183 void AliEMCALJetFinderInput::AddTrack(TMCParticle *track)
 184 {
 185 // Adds a TParticle to the particle array
 186
 187 if (fDebug>5) Info("AddTrack","Beginning AddTrack");
 188
 189  if (!fInitialised) InitArrays();
 190  if (fNTracks < fNMaxTracks){
 191    new((*fTracksArray)[fNTracks])
 192      TParticle(track->GetKF(), track->GetKS(), track->GetParent(), 0,
 193                track->GetFirstChild(), track->GetLastChild(),
 194                track->GetPx(), track->GetPy(), track->GetPz(),
 195                track->GetEnergy(),
 196                track->GetVx(), track->GetVy(), track->GetVz(),
 197                track->GetTime());
 198    fNTracks++;
 199    if (fDebug>5) Info("AddTrack","Added Track %i",fNTracks);
 200  } else {
 201    Error("AddTrack","Cannot AddTrack - maximum exceeded");
 202  }
 203 }
 204
 205
 206 void AliEMCALJetFinderInput::AddParton(AliEMCALParton *parton)
 207 {
 208 // Adds an AliEMCALParton to the parton array
 209
 210 if (fDebug>5) Info("AddParton","Beginning AddParton");
 211
 212         if (!fInitialised) InitArrays();
 213         if (fNPartons < fNMaxPartons){
 214                 new((*fPartonsArray)[fNPartons]) AliEMCALParton(*parton);
 215                 fNPartons++;
 216                 if (fDebug>5) Info("AddParton","Added Parton %i",fNPartons);
 217         }else
 218         {
 219                 Error("AddParton","Cannot AddParton - maximum exceeded");
 220         }
 221 }
 222
 223 void AliEMCALJetFinderInput::AddParticle(TParticle *particle)
 224 {
 225 // Adds a TParticle to the particle array
 226
 227 if (fDebug>5) Info("AddParticle","Beginning AddParticle");
 228
 229         if (!fInitialised) InitArrays();
 230         if (fNParticles < fNMaxParticles){
 231                 new((*fParticlesArray)[fNParticles]) TParticle(*particle);
 232                 fNParticles++;
 233                 if (fDebug>5) Info("AddParticle","Added Particle %i",fNParticles);
 234         }else
 235         {
 236                 Error("AddParticle","Cannot AddParticle - maximum exceeded");
 237         }
 238
 239 }
 240
 241 void AliEMCALJetFinderInput::AddParticle(TMCParticle *particle)
 242 {
 243 // Adds a TParticle to the particle array
 244
 245 if (fDebug>5) Info("AddParticle","Beginning AddParticle");
 246
 247  if (!fInitialised) InitArrays();
 248  if (fNParticles < fNMaxParticles){
 249    new((*fParticlesArray)[fNParticles])
 250      TParticle(particle->GetKF(), particle->GetKS(), particle->GetParent(), 0,
 251                particle->GetFirstChild(), particle->GetLastChild(),
 252                particle->GetPx(), particle->GetPy(), particle->GetPz(),
 253                particle->GetEnergy(),
 254                particle->GetVx(), particle->GetVy(), particle->GetVz(),
 255                particle->GetTime());
 256    fNParticles++;
 257    if (fDebug>5) Info("AddParticle","Added Particle %i",fNParticles);
 258  } else {
 259    Error("AddParticle","Cannot AddParticle - maximum exceeded");
 260  }
 261 }
 262
 263
 264 AliEMCALDigit* AliEMCALJetFinderInput::GetDigit(Int_t digitID)
 265 {
 266 // Returns a pointer to an AliEMCALDigit with the given ID
 267
 268 if (fDebug>15) Info("GetDigit","Beginning GetDigit");
 269
 270     if (digitID >= fNDigits) return 0;
 271     return (AliEMCALDigit*)((*fDigitsArray)[digitID]);
 272 }
 273
 274 TParticle* AliEMCALJetFinderInput::GetTrack(Int_t trackID)
 275 {
 276 // Returns a pointer to a TParticle specified by the ID
 277
 278 if (fDebug>15) Info("GetTrack","Beginning GetTrack with trackID %i",trackID);
 279
 280     if (trackID >= fNTracks) return 0;
 281     return (TParticle*)((*fTracksArray)[trackID]);
 282 }
 283
 284 AliEMCALParton* AliEMCALJetFinderInput::GetParton(Int_t partonID)
 285 {
 286 // Returns a pointer to an AliEMCALParton  specified by the ID
 287
 288 if (fDebug>15) Info("GetParton","Beginning GetParton");
 289
 290     if (partonID >= fNPartons) return 0;
 291     return (AliEMCALParton*)((*fPartonsArray)[partonID]);
 292 }
 293
 294 TParticle* AliEMCALJetFinderInput::GetParticle(Int_t particleID)
 295 {
 296 // Returns a pointer to a TParticle specified by the ID
 297
 298 if (fDebug>15) Info("GetParticle","Beginning GetParticle");
 299
 300     if (particleID >= fNParticles) return 0;
 301     return (TParticle*)((*fParticlesArray)[particleID]);
 302 }
 303
 304
 305