ROOT logo
#ifndef ALIESDCALOCELLS_H
#define ALIESDCALOCELLS_H
/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
/* $Id$ */
/* $Log $ */

//-------------------------------------------------------------------------
//     ESD class to store calorimeter cell data
//     Clone of AliAODCaloCells made by Markus Oldenburg, CERN
//     Author: Gustavo Conesa Balbastre INFN-LNF
//
//-------------------------------------------------------------------------


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

class AliESDCaloCells : public AliVCaloCells
{
 public:

  AliESDCaloCells();
  AliESDCaloCells(const char* name, const char* title, VCells_t ttype=kUndef);
  AliESDCaloCells(const AliESDCaloCells & cells);
  AliESDCaloCells & operator=(const AliESDCaloCells& source);
  virtual ~AliESDCaloCells();
  
  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();
  
  Bool_t          IsEMCAL()  const { return (fType == kEMCALCell); }
  Bool_t          IsPHOS()   const { return (fType == kPHOSCell) ; }
  Char_t          GetType()  const { return  fType               ; }
  void            SetType(Char_t t){ fType = t                   ; }
  
  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 Double_t GetTime(Short_t pos) const;
  inline Short_t  GetCellNumber(Short_t pos) const;

  // 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(AliESDCaloCells, 5);
};


Bool_t AliESDCaloCells::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];
    time       = fTime[pos];
    
    if(fMCLabel)   mclabel = fMCLabel[pos];
    else           mclabel =-1 ; 
    if(fEFraction) efrac   = fEFraction[pos];
    else           efrac   = 0 ;
    
    return kTRUE;
    
  } else 
  {
    return kFALSE;
  }
}

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

  Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
  if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber ) {
    return fAmplitude[pos];
  } else {
    return 0.;
  }
}

Bool_t AliESDCaloCells::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 ESD, for data HG flag stored in MCLabel 
        return !(fMCLabel[pos]==-2) ;
      else
	return kTRUE ;
    }
  } else {
    return kFALSE;
  }
}

Double_t AliESDCaloCells::GetCellTime(Short_t cellNumber)
{ 
  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 AliESDCaloCells::GetAmplitude(Short_t pos) const 
{ 
  if (pos>=0 && pos<fNCells) {
    return fAmplitude[pos];
  } else {
    return 0.;
  }
}
Bool_t AliESDCaloCells::GetHighGain(Short_t pos) const 
{ 
  if (pos>=0 && pos<fNCells) {
    if(fHGLG)
      return fHGLG[pos];
    else{
      if(fMCLabel)//old version of ESD, for data HG flag stored in MCLabel 
        return !(fMCLabel[pos]==-2) ;
      else
	return kTRUE ;
    }
  } else {
    return kFALSE;
  }
}

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

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

Short_t AliESDCaloCells::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 AliESDCaloCells::GetMCLabel(Short_t pos) const 
{ 
  if (pos>=0 && pos<fNCells && fMCLabel) {
    return fMCLabel[pos];
  } else {
    return -1 ;
  }
}

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

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

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

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