STEER/AliESDCaloCells.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-2007, 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 //     ESD class to store calorimeter cell data
  20 //     Clone of AliAODCaloCells made by Markus Oldenburg, CERN
  21 //     Author: Gustavo Conesa Balbastre INFN-LNF
  22 //-------------------------------------------------------------------------
  23
  24 #include "AliESDCaloCells.h"
  25
  26 ClassImp(AliESDCaloCells)
  27
  28 //_______________________________________________________________________
  29 AliESDCaloCells::AliESDCaloCells() :
  30   AliVCaloCells(), fNCells(0), fCellNumber(0), fAmplitude(0), fTime(0), fIsSorted(kTRUE), fType(kUndef)
  31 {
  32   // default constructor
  33 }
  34 //_______________________________________________________________________
  35  AliESDCaloCells::AliESDCaloCells(const char* name, const char* title, VCells_t ttype) :
  36    AliVCaloCells(name, title), fNCells(0), fCellNumber(0), fAmplitude(0),  fTime(0), fIsSorted(kTRUE), fType(ttype)
  37 {
  38    // AliVCaloCells constructor
  39  }
  40
  41 //_______________________________________________________________________
  42 AliESDCaloCells::AliESDCaloCells(const AliESDCaloCells& c) :
  43   AliVCaloCells(c), fNCells(c.fNCells),  fCellNumber(0), fAmplitude(0), fTime(0), fIsSorted(c.fIsSorted), fType(c.fType)
  44 {
  45   // copy constructor
  46
  47   fCellNumber = new Short_t[fNCells];
  48   fAmplitude  = new Double32_t[fNCells];
  49
  50   for(Int_t i = 0; i < fNCells; i++){
  51     fCellNumber[i]    = c.fCellNumber[i];
  52     fAmplitude[i]     = c.fAmplitude[i];
  53     fTime[i]          = c.fTime[i];
  54     fAmplitude[i]     = c.fAmplitude[i];
  55
  56   }
  57 }
  58
  59 //_______________________________________________________________________
  60 AliESDCaloCells & AliESDCaloCells::operator =(const AliESDCaloCells& source)
  61 {
  62   // assignment operator
  63
  64   if(&source == this) return *this;
  65
  66   if(fNCells != source.fNCells){
  67     DeleteContainer();
  68     CreateContainer(source.fNCells);
  69   }
  70
  71   fNCells = source.fNCells;
  72   fIsSorted = source.fIsSorted;
  73   fType = source.fType;
  74
  75
  76
  77   for(Int_t i = 0; i < fNCells; i++){
  78     fCellNumber[i]    = source.fCellNumber[i];
  79     fAmplitude[i]     = source.fAmplitude[i];
  80     fTime[i]          = source.fTime[i];
  81   }
  82
  83   return *this;
  84
  85 }
  86
  87 //_______________________________________________________________________
  88 void AliESDCaloCells::Copy(TObject &obj) const {
  89
  90   // this overwrites the virtual TOBject::Copy()
  91   // to allow run time copying without casting
  92   // in AliESDEvent
  93
  94   if(this==&obj)return;
  95   AliESDCaloCells *robj = dynamic_cast<AliESDCaloCells*>(&obj);
  96   if(!robj)return; // not an AliESDCaloCells
  97   *robj = *this;
  98
  99 }
 100
 101 //_______________________________________________________________________
 102 AliVCaloCells* AliESDCaloCells::CopyCaloCells(Bool_t all = kTRUE) const {
 103
 104   // copy the calo cells into a new object. If option all=FALSE, just the object type,
 105   // for mixing
 106
 107   AliVCaloCells *obj = new AliESDCaloCells();
 108
 109   if(all){
 110     obj->SetName (GetName()) ;
 111     obj->SetTitle(GetTitle()) ;
 112     obj->SetType (GetType()) ;
 113
 114     obj->SetNumberOfCells(fNCells);
 115     for (Short_t i = 0; i < fNCells; i++)
 116       obj->SetCell(i,fCellNumber[i],fAmplitude[i],fTime[i]);
 117   }
 118
 119   return obj;
 120 }
 121
 122
 123 //_______________________________________________________________________
 124 AliESDCaloCells::~AliESDCaloCells()
 125 {
 126   // destructor
 127
 128   DeleteContainer();
 129 }
 130
 131 //_______________________________________________________________________
 132 void AliESDCaloCells::CreateContainer(Short_t nCells)
 133 {
 134   // function that creates container to store calorimeter cell data
 135
 136   DeleteContainer();
 137
 138   if (nCells <= 0) {
 139     fNCells = 0;
 140     return;
 141   }
 142
 143   fNCells = nCells;
 144
 145   fCellNumber = new Short_t[fNCells];
 146   fAmplitude  = new Double32_t[fNCells];
 147   fTime       = new Double32_t[fNCells];
 148
 149     // set to zero
 150   for(int i = 0;i<fNCells;++i){
 151     fAmplitude[i] = fCellNumber[i] = 0 ;
 152   }
 153 }
 154
 155 //_______________________________________________________________________
 156 void AliESDCaloCells::DeleteContainer()
 157 {
 158   // deletes allocated memory
 159
 160   if (fCellNumber)
 161   {
 162     delete[] fCellNumber;
 163     fCellNumber = 0;
 164   }
 165
 166   if (fAmplitude)
 167   {
 168     delete[] fAmplitude;
 169     fAmplitude = NULL;
 170   }
 171
 172   if (fTime)
 173   {
 174     delete[] fTime;
 175     fTime = NULL;
 176   }
 177
 178   fNCells = 0;
 179   fIsSorted = kFALSE;
 180 }
 181
 182 //_______________________________________________________________________
 183 void AliESDCaloCells::Sort()
 184 {
 185   // sort the cell array by cell number
 186
 187   Int_t *idxArray = new Int_t[fNCells];
 188   TMath::Sort(fNCells,fCellNumber,idxArray,kFALSE);
 189
 190   Short_t *newIndex        = new Short_t[fNCells];
 191   Double32_t *newAmplitude = new Double32_t[fNCells];
 192   Double32_t *newTime      = new Double32_t[fNCells];
 193   for (Int_t i=0; i < fNCells; i++) {
 194     newIndex[i]     = fCellNumber[idxArray[i]];
 195     newAmplitude[i] = fAmplitude[idxArray[i]];
 196     newTime[i]      = fTime[idxArray[i]];
 197   }
 198   delete [] fCellNumber;
 199   delete [] fAmplitude;
 200   delete [] fTime;
 201
 202   fCellNumber = newIndex;
 203   fAmplitude  = newAmplitude;
 204   fTime       = newTime;
 205
 206   delete [] idxArray;
 207
 208   fIsSorted = kTRUE;
 209 }
 210
 211 //_______________________________________________________________________
 212 Bool_t AliESDCaloCells::SetCell(Short_t pos, Short_t cellNumber, Double32_t amplitude, Double32_t  time)
 213 {
 214   // Sets a cell at the given position
 215
 216   if (pos>=0 && pos < fNCells) {
 217     fCellNumber[pos] = cellNumber;
 218     fAmplitude[pos]  = amplitude;
 219     fTime[pos]       = time;
 220
 221     fIsSorted = kFALSE;
 222     return kTRUE;
 223   } else {
 224     return kFALSE;
 225   }
 226 }