ROOT logo
/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

//-------------------------------------------------------------------------
//     AOD class to store calorimeter cell data
//     Author: Markus Oldenburg, CERN
//-------------------------------------------------------------------------

#ifndef ALIAODCELLS_H
#define ALIAODCELLS_H

#include <AliVCaloCells.h>
#include <TMath.h>

class AliAODCaloCells : public AliVCaloCells
{
 public:
  AliAODCaloCells();
  AliAODCaloCells(const char* name, const char* title, VCells_t ttype=kUndef);
  AliAODCaloCells(const AliAODCaloCells& cells); 
  AliAODCaloCells& operator=(const AliAODCaloCells& cells);
  virtual ~AliAODCaloCells();

  virtual AliVCaloCells* CopyCaloCells(Bool_t all) const;
  virtual void    Copy(TObject &obj) const;
  void            Clear(const Option_t*);
  void            CreateContainer(Short_t nCells);
  void            DeleteContainer();
  void            Sort();
  
  inline Bool_t   GetCell(Short_t pos, Short_t &cellNumber, Double_t &amplitude,  Double_t &time, Int_t &mclabel,      Double_t &efrac) const ;
  Bool_t          SetCell(Short_t pos, Short_t  cellNumber, Double_t  amplitude, Double_t  time, Int_t  mclabel = -1, Double_t  efrac = 0., Bool_t isHG=kFALSE);
  
  Short_t         GetNumberOfCells() const  { return fNCells ; }
  void            SetNumberOfCells(Int_t n) { fNCells = n    ; }
  
  inline Double_t GetCellAmplitude(Short_t cellNumber);
  inline Bool_t   GetCellHighGain(Short_t cellNumber);  //is this cell High Gain
  inline Short_t  GetCellPosition(Short_t cellNumber);
  inline Double_t GetCellTime(Short_t cellNumber);
  
  inline Double_t GetAmplitude(Short_t pos) const;
  inline Bool_t   GetHighGain(Short_t pos) const;
  inline Short_t  GetCellNumber(Short_t pos) const;
  inline Double_t GetTime(Short_t pos) const;
  
  Bool_t          IsEMCAL() const { return (fType == kEMCALCell); }
  Bool_t          IsPHOS()  const { return (fType == kPHOSCell) ; }
  
  Char_t          GetType() const { return fType;}
  void            SetType(Char_t ttype) { fType=ttype; }
  
  // MC & embedding
  inline Int_t    GetCellMCLabel(Short_t cellNumber) ;
  inline Int_t    GetMCLabel(Short_t pos) const ;
  
  inline Double_t GetCellEFraction(Short_t cellNumber) ;
  inline Double_t GetEFraction(Short_t pos) const ;  
  
  inline void     SetEFraction    (Short_t pos,         Double32_t efrac) ;
  inline void     SetCellEFraction(Short_t cellNumber,  Double32_t efrac) ;
  
 protected:
  
  Int_t       fNCells;       // Number of cells
  Bool_t     *fHGLG;         //[fNCells] if sell HG or LG
  Short_t    *fCellNumber;   //[fNCells] array of cell numbers
  Double32_t *fAmplitude;    //[fNCells][0.,0.,16] array with cell amplitudes (= energy!)
  Double32_t *fTime;         //[fNCells][0.,0.,16] array with cell times
  Double32_t *fEFraction;    //[fNCells][0.,0.,16] array with fraction of MC energy and data - for embedding
  Int_t      *fMCLabel;      //[fNCells] array of MC labels
  Bool_t      fIsSorted;     //! true if cell arrays are sorted by index
  Char_t      fType;         // Cell type
  
  ClassDef(AliAODCaloCells, 5);
  
};

Bool_t AliAODCaloCells::GetCell(Short_t pos, Short_t &cellNumber, Double_t &amplitude, 
                                Double_t &time, Int_t & mclabel, Double_t & efrac) const 
{ 
  if (pos>=0 && pos<fNCells) 
  {
    cellNumber = fCellNumber[pos];
    amplitude  = fAmplitude[pos];
    
    if(fTime)      time    = fTime[pos];
    else           time    =-1.;
    if(fMCLabel)   mclabel = fMCLabel[pos];
    else           mclabel =-1 ; 
    if(fEFraction) efrac   = fEFraction[pos];
    else           efrac   = 0 ;
    
    return kTRUE;
    
  } else {
    return kFALSE;
  }
}

Double_t AliAODCaloCells::GetCellAmplitude(Short_t cellNumber)
{ 
  if (!fIsSorted) {
    Sort();
    fIsSorted=kTRUE;
  }

  Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
  if (pos>=0 && fCellNumber[pos] == cellNumber) {
    return fAmplitude[pos];
  } else {
    return 0.;
  }
}
Bool_t AliAODCaloCells::GetCellHighGain(Short_t cellNumber)
{ 
  if (!fIsSorted) {
    Sort();
    fIsSorted=kTRUE;
  }

  Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
  if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber ) {
    if(fHGLG)
      return fHGLG[pos];
    else{
      if(fMCLabel) //old version of AOD, 
         return !(fMCLabel[pos]==-2) ;
      else
	 return kTRUE ;
    }
  } else {
    return kFALSE;
  }
}

Double_t AliAODCaloCells::GetCellTime(Short_t cellNumber)
{ 
  if(!fTime) return -1;
  
  if (!fIsSorted) {
    Sort();
    fIsSorted=kTRUE;
  }
  
  Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
  if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber) {
    return fTime[pos];
  } else {
    return -1.;
  }
}

Double_t AliAODCaloCells::GetAmplitude(Short_t pos) const 
{ 
  if (pos>=0 && pos<fNCells) {
    return fAmplitude[pos];
  } else {
    return 0.;
  }
}
Bool_t AliAODCaloCells::GetHighGain(Short_t pos) const 
{ 
  if (pos>=0 && pos<fNCells) {
    if(fHGLG)
      return fHGLG[pos];
    else{
      if(fMCLabel)    //Old version of AOD store this flag in MCLabel
        return !(fMCLabel[pos]==-2) ;
      else
	return kTRUE ;
    }
  } else {
    return kFALSE;
  }
}


Double_t AliAODCaloCells::GetTime(Short_t pos) const 
{ 
  if (pos>=0 && pos<fNCells && fTime) {
    return fTime[pos];
  } else {
    return -1.;
  }
}

Short_t AliAODCaloCells::GetCellNumber(Short_t pos) const 
{ 
  if (pos>=0 && pos<fNCells) {
    return fCellNumber[pos];
  } else {
    return fNCells;
  }
}

Short_t AliAODCaloCells::GetCellPosition(Short_t cellNumber)
{ 
  if (!fIsSorted) {
    Sort();
    fIsSorted=kTRUE;
  }

   Int_t nabove, nbelow, middle;
   Short_t pos = -1;

   nabove = fNCells + 1;
   nbelow = 0;
   while (nabove - nbelow > 1) {
      middle = (nabove + nbelow) / 2;
      if (cellNumber == fCellNumber[middle-1]) {
	  pos =   middle - 1;
	  break;
      }
      if (cellNumber  < fCellNumber[middle-1]) nabove = middle;
      else                                     nbelow = middle;
   }

  return pos;
}

Int_t AliAODCaloCells::GetMCLabel(Short_t pos) const 
{ 
  if (pos>=0 && pos<fNCells && fMCLabel) {
    return fMCLabel[pos];
  } else {
    return 0;
  }
}

Double_t AliAODCaloCells::GetEFraction(Short_t pos) const 
{ 
  if (pos>=0 && pos<fNCells && fEFraction) {
    return fEFraction[pos];
  } else {
    return 0.;
  }
}

Int_t AliAODCaloCells::GetCellMCLabel(Short_t cellNumber)
{ 
  if(!fMCLabel) return -1;
  
  if (!fIsSorted) {
    Sort();
    fIsSorted=kTRUE;
  }
  
  Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
  if (pos>=0 && fCellNumber[pos] == cellNumber) {
    return fMCLabel[pos];
  } else {
    return 0;
  }
}

Double_t AliAODCaloCells::GetCellEFraction(Short_t cellNumber)
{ 
  if(!fEFraction) return 0;

  if (!fIsSorted) {
    Sort();
    fIsSorted=kTRUE;
  }
  
  Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
  if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber) {
    return fEFraction[pos];
  } else {
    return -1.;
  }
}

void AliAODCaloCells::SetEFraction(Short_t pos,  Double32_t efrac)
{
  // Sets the fraction of energy from MC with respect to data at the given position
  
  
  if (pos>=0 && pos < fNCells) 
  {
    if(!fEFraction) fEFraction = new Double32_t[fNCells];
    fEFraction[pos]  = efrac;
  } 
}

void AliAODCaloCells::SetCellEFraction(Short_t cellNumber, Double32_t efrac)
{ 
  if (!fIsSorted) {
    Sort();
    fIsSorted=kTRUE;
  }
  
  Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
  if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber) 
  {
    if(!fEFraction) fEFraction = new Double32_t[fNCells];
    fEFraction[pos] = efrac;
  } 
}


#endif
 AliAODCaloCells.h:1
 AliAODCaloCells.h:2
 AliAODCaloCells.h:3
 AliAODCaloCells.h:4
 AliAODCaloCells.h:5
 AliAODCaloCells.h:6
 AliAODCaloCells.h:7
 AliAODCaloCells.h:8
 AliAODCaloCells.h:9
 AliAODCaloCells.h:10
 AliAODCaloCells.h:11
 AliAODCaloCells.h:12
 AliAODCaloCells.h:13
 AliAODCaloCells.h:14
 AliAODCaloCells.h:15
 AliAODCaloCells.h:16
 AliAODCaloCells.h:17
 AliAODCaloCells.h:18
 AliAODCaloCells.h:19
 AliAODCaloCells.h:20
 AliAODCaloCells.h:21
 AliAODCaloCells.h:22
 AliAODCaloCells.h:23
 AliAODCaloCells.h:24
 AliAODCaloCells.h:25
 AliAODCaloCells.h:26
 AliAODCaloCells.h:27
 AliAODCaloCells.h:28
 AliAODCaloCells.h:29
 AliAODCaloCells.h:30
 AliAODCaloCells.h:31
 AliAODCaloCells.h:32
 AliAODCaloCells.h:33
 AliAODCaloCells.h:34
 AliAODCaloCells.h:35
 AliAODCaloCells.h:36
 AliAODCaloCells.h:37
 AliAODCaloCells.h:38
 AliAODCaloCells.h:39
 AliAODCaloCells.h:40
 AliAODCaloCells.h:41
 AliAODCaloCells.h:42
 AliAODCaloCells.h:43
 AliAODCaloCells.h:44
 AliAODCaloCells.h:45
 AliAODCaloCells.h:46
 AliAODCaloCells.h:47
 AliAODCaloCells.h:48
 AliAODCaloCells.h:49
 AliAODCaloCells.h:50
 AliAODCaloCells.h:51
 AliAODCaloCells.h:52
 AliAODCaloCells.h:53
 AliAODCaloCells.h:54
 AliAODCaloCells.h:55
 AliAODCaloCells.h:56
 AliAODCaloCells.h:57
 AliAODCaloCells.h:58
 AliAODCaloCells.h:59
 AliAODCaloCells.h:60
 AliAODCaloCells.h:61
 AliAODCaloCells.h:62
 AliAODCaloCells.h:63
 AliAODCaloCells.h:64
 AliAODCaloCells.h:65
 AliAODCaloCells.h:66
 AliAODCaloCells.h:67
 AliAODCaloCells.h:68
 AliAODCaloCells.h:69
 AliAODCaloCells.h:70
 AliAODCaloCells.h:71
 AliAODCaloCells.h:72
 AliAODCaloCells.h:73
 AliAODCaloCells.h:74
 AliAODCaloCells.h:75
 AliAODCaloCells.h:76
 AliAODCaloCells.h:77
 AliAODCaloCells.h:78
 AliAODCaloCells.h:79
 AliAODCaloCells.h:80
 AliAODCaloCells.h:81
 AliAODCaloCells.h:82
 AliAODCaloCells.h:83
 AliAODCaloCells.h:84
 AliAODCaloCells.h:85
 AliAODCaloCells.h:86
 AliAODCaloCells.h:87
 AliAODCaloCells.h:88
 AliAODCaloCells.h:89
 AliAODCaloCells.h:90
 AliAODCaloCells.h:91
 AliAODCaloCells.h:92
 AliAODCaloCells.h:93
 AliAODCaloCells.h:94
 AliAODCaloCells.h:95
 AliAODCaloCells.h:96
 AliAODCaloCells.h:97
 AliAODCaloCells.h:98
 AliAODCaloCells.h:99
 AliAODCaloCells.h:100
 AliAODCaloCells.h:101
 AliAODCaloCells.h:102
 AliAODCaloCells.h:103
 AliAODCaloCells.h:104
 AliAODCaloCells.h:105
 AliAODCaloCells.h:106
 AliAODCaloCells.h:107
 AliAODCaloCells.h:108
 AliAODCaloCells.h:109
 AliAODCaloCells.h:110
 AliAODCaloCells.h:111
 AliAODCaloCells.h:112
 AliAODCaloCells.h:113
 AliAODCaloCells.h:114
 AliAODCaloCells.h:115
 AliAODCaloCells.h:116
 AliAODCaloCells.h:117
 AliAODCaloCells.h:118
 AliAODCaloCells.h:119
 AliAODCaloCells.h:120
 AliAODCaloCells.h:121
 AliAODCaloCells.h:122
 AliAODCaloCells.h:123
 AliAODCaloCells.h:124
 AliAODCaloCells.h:125
 AliAODCaloCells.h:126
 AliAODCaloCells.h:127
 AliAODCaloCells.h:128
 AliAODCaloCells.h:129
 AliAODCaloCells.h:130
 AliAODCaloCells.h:131
 AliAODCaloCells.h:132
 AliAODCaloCells.h:133
 AliAODCaloCells.h:134
 AliAODCaloCells.h:135
 AliAODCaloCells.h:136
 AliAODCaloCells.h:137
 AliAODCaloCells.h:138
 AliAODCaloCells.h:139
 AliAODCaloCells.h:140
 AliAODCaloCells.h:141
 AliAODCaloCells.h:142
 AliAODCaloCells.h:143
 AliAODCaloCells.h:144
 AliAODCaloCells.h:145
 AliAODCaloCells.h:146
 AliAODCaloCells.h:147
 AliAODCaloCells.h:148
 AliAODCaloCells.h:149
 AliAODCaloCells.h:150
 AliAODCaloCells.h:151
 AliAODCaloCells.h:152
 AliAODCaloCells.h:153
 AliAODCaloCells.h:154
 AliAODCaloCells.h:155
 AliAODCaloCells.h:156
 AliAODCaloCells.h:157
 AliAODCaloCells.h:158
 AliAODCaloCells.h:159
 AliAODCaloCells.h:160
 AliAODCaloCells.h:161
 AliAODCaloCells.h:162
 AliAODCaloCells.h:163
 AliAODCaloCells.h:164
 AliAODCaloCells.h:165
 AliAODCaloCells.h:166
 AliAODCaloCells.h:167
 AliAODCaloCells.h:168
 AliAODCaloCells.h:169
 AliAODCaloCells.h:170
 AliAODCaloCells.h:171
 AliAODCaloCells.h:172
 AliAODCaloCells.h:173
 AliAODCaloCells.h:174
 AliAODCaloCells.h:175
 AliAODCaloCells.h:176
 AliAODCaloCells.h:177
 AliAODCaloCells.h:178
 AliAODCaloCells.h:179
 AliAODCaloCells.h:180
 AliAODCaloCells.h:181
 AliAODCaloCells.h:182
 AliAODCaloCells.h:183
 AliAODCaloCells.h:184
 AliAODCaloCells.h:185
 AliAODCaloCells.h:186
 AliAODCaloCells.h:187
 AliAODCaloCells.h:188
 AliAODCaloCells.h:189
 AliAODCaloCells.h:190
 AliAODCaloCells.h:191
 AliAODCaloCells.h:192
 AliAODCaloCells.h:193
 AliAODCaloCells.h:194
 AliAODCaloCells.h:195
 AliAODCaloCells.h:196
 AliAODCaloCells.h:197
 AliAODCaloCells.h:198
 AliAODCaloCells.h:199
 AliAODCaloCells.h:200
 AliAODCaloCells.h:201
 AliAODCaloCells.h:202
 AliAODCaloCells.h:203
 AliAODCaloCells.h:204
 AliAODCaloCells.h:205
 AliAODCaloCells.h:206
 AliAODCaloCells.h:207
 AliAODCaloCells.h:208
 AliAODCaloCells.h:209
 AliAODCaloCells.h:210
 AliAODCaloCells.h:211
 AliAODCaloCells.h:212
 AliAODCaloCells.h:213
 AliAODCaloCells.h:214
 AliAODCaloCells.h:215
 AliAODCaloCells.h:216
 AliAODCaloCells.h:217
 AliAODCaloCells.h:218
 AliAODCaloCells.h:219
 AliAODCaloCells.h:220
 AliAODCaloCells.h:221
 AliAODCaloCells.h:222
 AliAODCaloCells.h:223
 AliAODCaloCells.h:224
 AliAODCaloCells.h:225
 AliAODCaloCells.h:226
 AliAODCaloCells.h:227
 AliAODCaloCells.h:228
 AliAODCaloCells.h:229
 AliAODCaloCells.h:230
 AliAODCaloCells.h:231
 AliAODCaloCells.h:232
 AliAODCaloCells.h:233
 AliAODCaloCells.h:234
 AliAODCaloCells.h:235
 AliAODCaloCells.h:236
 AliAODCaloCells.h:237
 AliAODCaloCells.h:238
 AliAODCaloCells.h:239
 AliAODCaloCells.h:240
 AliAODCaloCells.h:241
 AliAODCaloCells.h:242
 AliAODCaloCells.h:243
 AliAODCaloCells.h:244
 AliAODCaloCells.h:245
 AliAODCaloCells.h:246
 AliAODCaloCells.h:247
 AliAODCaloCells.h:248
 AliAODCaloCells.h:249
 AliAODCaloCells.h:250
 AliAODCaloCells.h:251
 AliAODCaloCells.h:252
 AliAODCaloCells.h:253
 AliAODCaloCells.h:254
 AliAODCaloCells.h:255
 AliAODCaloCells.h:256
 AliAODCaloCells.h:257
 AliAODCaloCells.h:258
 AliAODCaloCells.h:259
 AliAODCaloCells.h:260
 AliAODCaloCells.h:261
 AliAODCaloCells.h:262
 AliAODCaloCells.h:263
 AliAODCaloCells.h:264
 AliAODCaloCells.h:265
 AliAODCaloCells.h:266
 AliAODCaloCells.h:267
 AliAODCaloCells.h:268
 AliAODCaloCells.h:269
 AliAODCaloCells.h:270
 AliAODCaloCells.h:271
 AliAODCaloCells.h:272
 AliAODCaloCells.h:273
 AliAODCaloCells.h:274
 AliAODCaloCells.h:275
 AliAODCaloCells.h:276
 AliAODCaloCells.h:277
 AliAODCaloCells.h:278
 AliAODCaloCells.h:279
 AliAODCaloCells.h:280
 AliAODCaloCells.h:281
 AliAODCaloCells.h:282
 AliAODCaloCells.h:283
 AliAODCaloCells.h:284
 AliAODCaloCells.h:285
 AliAODCaloCells.h:286
 AliAODCaloCells.h:287
 AliAODCaloCells.h:288
 AliAODCaloCells.h:289
 AliAODCaloCells.h:290
 AliAODCaloCells.h:291
 AliAODCaloCells.h:292
 AliAODCaloCells.h:293
 AliAODCaloCells.h:294
 AliAODCaloCells.h:295
 AliAODCaloCells.h:296
 AliAODCaloCells.h:297
 AliAODCaloCells.h:298
 AliAODCaloCells.h:299
 AliAODCaloCells.h:300
 AliAODCaloCells.h:301
 AliAODCaloCells.h:302
 AliAODCaloCells.h:303
 AliAODCaloCells.h:304