Added classes for UPC analysis

[u/mrichter/AliRoot.git] / ITS / UPGRADE / AliITSUSensMap.h

diff --git a/ITS/UPGRADE/AliITSUSensMap.h b/ITS/UPGRADE/AliITSUSensMap.h
index f9bdcb1770a2359cf5e06fc765b2a0486970a492..965f277bc8223d3404f2283e926bf0844b0c2f8a 100644 (file)
--- a/ITS/UPGRADE/AliITSUSensMap.h
+++ b/ITS/UPGRADE/AliITSUSensMap.h
@@ -17,7 +17,6 @@
 // ***********************************************************************
 #include <TClonesArray.h>
 #include <TBtree.h>
-
 #define _ROWWISE_SORT_
 
 class AliITSUSensMap: public TObject 
@@ -27,21 +26,22 @@ class AliITSUSensMap: public TObject
   enum {kDisableBit=BIT(14)};
   //
   AliITSUSensMap();
-  AliITSUSensMap(const char* className, UInt_t dim0,UInt_t dim1);
+  AliITSUSensMap(const char* className, UInt_t dimCol,UInt_t dimRow,UInt_t dimCycle=1);
   virtual ~AliITSUSensMap();
   AliITSUSensMap(const AliITSUSensMap &source);
   AliITSUSensMap& operator=(const AliITSUSensMap &source);
   void Clear(Option_t* option = "");
-  void DeleteItem(UInt_t i,UInt_t j);
+  void DeleteItem(UInt_t col,UInt_t row, Int_t cycle);
   void DeleteItem(TObject* obj);
   //
-  void  SetDimensions(UInt_t dim0,UInt_t dim1)   {fDim0 = dim0; fDim1 = dim1;}
-  void  GetMaxIndex(UInt_t &ni,UInt_t &nj) const {ni=fDim0;nj=fDim1;}
-  Int_t GetMaxIndex()                      const {return fDim0*fDim1;}
+  void  SetDimensions(UInt_t dimCol,UInt_t dimRow,UInt_t dimCycle=1);
+  void  GetMaxIndex(UInt_t &col,UInt_t &row,UInt_t &cycle) const {col=fDimCol; row=fDimRow; cycle=fDimCycle;}
+  Int_t GetMaxIndex()                      const {return fDimCol*fDimRow*(fDimCycle*2+1);}
   Int_t GetEntries()                       const {return fBTree->GetEntries();}
   Int_t GetEntriesUnsorted()               const {return fItems->GetEntriesFast();}
-  void  GetMapIndex(UInt_t index,UInt_t &i,UInt_t &j) const {return GetCell(index, i,j);}
-  TObject* GetItem(UInt_t i,UInt_t j)            {SetUniqueID(GetIndex(i,j)); return fBTree->FindObject(this);}
+  void  GetMapIndex(UInt_t index,UInt_t &col,UInt_t &row,Int_t &cycle) const {return GetCell(index,fDimCol,fDimRow,fDimCycle,col,row,cycle);}
+  void  GetCell(UInt_t index,UInt_t &col,UInt_t &row,Int_t &cycle)     const {return GetCell(index,fDimCol,fDimRow,fDimCycle,col,row,cycle);}
+  TObject* GetItem(UInt_t col,UInt_t row,Int_t cycle)  {SetUniqueID(GetIndex(col,row,cycle)); return fBTree->FindObject(this);}
   TObject* GetItem(UInt_t index)                 {SetUniqueID(index);         return fBTree->FindObject(this);}
   TObject* GetItem(const TObject* obj)           {return fBTree->FindObject(obj);}
   TObject* At(Int_t i)                     const {return fBTree->At(i);}             //!!! Access in sorted order !!!
@@ -49,8 +49,7 @@ class AliITSUSensMap: public TObject
   TObject* RegisterItem(TObject* obj)            {fBTree->Add(obj); return obj;}
   TObject* GetFree()                             {return (*fItems)[fItems->GetEntriesFast()];}
   //
-  void   GetCell(UInt_t index,UInt_t &i,UInt_t &j) const;
-  UInt_t GetIndex(UInt_t i,UInt_t j)       const;
+  UInt_t   GetIndex(UInt_t col,UInt_t row,Int_t cycle=0)  const;
   //
   TClonesArray* GetItems()                 const {return fItems;}
   TBtree*       GetItemsBTree()            const {return fBTree;}
@@ -62,25 +61,46 @@ class AliITSUSensMap: public TObject
   static Bool_t IsDisabled(TObject* obj)         {return obj ? obj->TestBit(kDisableBit) : kFALSE;}
   static void   Disable(TObject* obj)            {if (obj) obj->SetBit(kDisableBit);}
   static void   Enable(TObject* obj)             {if (obj) obj->ResetBit(kDisableBit);}
+  static void   GetCell(UInt_t index,UInt_t dcol,UInt_t drow,UInt_t dcycle,UInt_t &col,UInt_t &row,Int_t &cycle);
   //
  protected:
   //
-  UInt_t fDim0;              // 1st dimension of the matrix
-  UInt_t fDim1;              // 2nd dimention of the matrix
+  UInt_t fDimCol;              // 1st dimension of the matrix, col index may span from 0 to fDimCol
+  UInt_t fDimRow;              // 2nd dimention of the matrix, row index may span from 0 to fDimRow
+  UInt_t fDimCycle;            // readout cycle range, may span from -fDimCycle to fDimCycle
   TClonesArray*    fItems;   // pListItems array
   TBtree*          fBTree;   // tree for ordered access
   //
   ClassDef(AliITSUSensMap,1) // list of sensor signals (should be sortable objects)
 };     
 
-inline UInt_t AliITSUSensMap::GetIndex(UInt_t i,UInt_t j) const  
+//______________________________________________________________________
+inline UInt_t AliITSUSensMap::GetIndex(UInt_t col,UInt_t row, Int_t cycle) const  
 {
   // linearized ID of digit
+  UInt_t cyclePos = cycle+fDimCycle; // cycle may span from -fDimCycle to fDimCycle
 #ifdef _ROWWISE_SORT_
-  return j*fDim0+i; // sorted in row, then in column
+  return fDimCol*(cyclePos*fDimRow+row)+col; // sorted in row, then in column
 #else
-  return j+i*fDim1; // sorted in column, then in row
+  return fDimRow*(cyclePos*fDimCol+col)+row; // sorted in column, then in row
 #endif
 }
 
+//______________________________________________________________________
+inline void AliITSUSensMap::GetCell(UInt_t index,UInt_t dcol,UInt_t drow,UInt_t dcycle,UInt_t &col,UInt_t &row,Int_t &cycle) 
+{
+  // returns the i,j index numbers from the linearized index computed with GetIndex
+  UInt_t dcr = dcol*drow;
+  cycle = int(index/dcr) - dcycle;
+  index %= dcr;
+#ifdef _ROWWISE_SORT_
+  col = index%dcol;   // sorted in row, then in column
+  row = index/dcol;
+#else
+  col = index/drow;   // sorted in column, then in row
+  row = index%drow;
+#endif  
+}
+
+
 #endif