Fix for ITS dictionaries
[u/mrichter/AliRoot.git] / ITS / UPGRADE / AliITSUSensMap.h
1 #ifndef ALIITSUSENSMAP_H
2 #define ALIITSUSENSMAP_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice     */
5
6 //***********************************************************************
7 //
8 // It consist of a TClonesArray of objects
9 // and B-tree for fast sorted access
10 // This array can be accessed via 2 indexed
11 // it is used at digitization level by 
12 // all the ITS subdetectors
13 //
14 // The items should be added to the map like this:
15 // map->RegisterItem( new(map->GetFree()) ItemConstructor(...) );
16 //
17 // ***********************************************************************
18 #include <TClonesArray.h>
19 #include <TBtree.h>
20 #define _ROWWISE_SORT_
21
22 class AliITSUSensMap: public TObject 
23 {
24
25  public:
26   enum {kDisableBit=BIT(14)};
27   //
28   AliITSUSensMap();
29   AliITSUSensMap(const char* className, UInt_t dimCol,UInt_t dimRow,UInt_t dimCycle=1);
30   virtual ~AliITSUSensMap();
31   AliITSUSensMap(const AliITSUSensMap &source);
32   AliITSUSensMap& operator=(const AliITSUSensMap &source);
33   void Clear(Option_t* option = "");
34   void DeleteItem(UInt_t col,UInt_t row, Int_t cycle);
35   void DeleteItem(TObject* obj);
36   //
37   void  SetDimensions(UInt_t dimCol,UInt_t dimRow,UInt_t dimCycle=1);
38   void  GetMaxIndex(UInt_t &col,UInt_t &row,UInt_t &cycle) const {col=fDimCol; row=fDimRow; cycle=fDimCycle;}
39   Int_t GetMaxIndex()                      const {return fDimCol*fDimRow*(fDimCycle*2+1);}
40   Int_t GetEntries()                       const {return fBTree->GetEntries();}
41   Int_t GetEntriesUnsorted()               const {return fItems->GetEntriesFast();}
42   void  GetMapIndex(UInt_t index,UInt_t &col,UInt_t &row,Int_t &cycle) const {return GetCell(index,fDimCol,fDimRow,fDimCycle,col,row,cycle);}
43   void  GetCell(UInt_t index,UInt_t &col,UInt_t &row,Int_t &cycle)     const {return GetCell(index,fDimCol,fDimRow,fDimCycle,col,row,cycle);}
44   TObject* GetItem(UInt_t col,UInt_t row,Int_t cycle)  {SetUniqueID(GetIndex(col,row,cycle)); return fBTree->FindObject(this);}
45   TObject* GetItem(UInt_t index)                 {SetUniqueID(index);         return fBTree->FindObject(this);}
46   TObject* GetItem(const TObject* obj)           {return fBTree->FindObject(obj);}
47   TObject* At(Int_t i)                     const {return fBTree->At(i);}             //!!! Access in sorted order !!!
48   TObject* AtUnsorted(Int_t i)             const {return fItems->At(i);}             //!!! Access in unsorted order !!!
49   TObject* RegisterItem(TObject* obj)            {fBTree->Add(obj); return obj;}
50   TObject* GetFree()                             {return (*fItems)[fItems->GetEntriesFast()];}
51   //
52   UInt_t   GetIndex(UInt_t col,UInt_t row,Int_t cycle=0)  const;
53   //
54   TClonesArray* GetItems()                 const {return fItems;}
55   TBtree*       GetItemsBTree()            const {return fBTree;}
56   //
57   Bool_t        IsSortable()                const {return kTRUE;}
58   Bool_t        IsEqual(const TObject* obj) const {return GetUniqueID()==obj->GetUniqueID();}
59   Int_t         Compare(const TObject* obj) const {return (GetUniqueID()<obj->GetUniqueID()) ? -1 : ((GetUniqueID()>obj->GetUniqueID()) ? 1 : 0 );}
60   //
61   static Bool_t IsDisabled(TObject* obj)         {return obj ? obj->TestBit(kDisableBit) : kFALSE;}
62   static void   Disable(TObject* obj)            {if (obj) obj->SetBit(kDisableBit);}
63   static void   Enable(TObject* obj)             {if (obj) obj->ResetBit(kDisableBit);}
64   static void   GetCell(UInt_t index,UInt_t dcol,UInt_t drow,UInt_t dcycle,UInt_t &col,UInt_t &row,Int_t &cycle);
65   //
66  protected:
67   //
68   UInt_t fDimCol;              // 1st dimension of the matrix, col index may span from 0 to fDimCol
69   UInt_t fDimRow;              // 2nd dimention of the matrix, row index may span from 0 to fDimRow
70   UInt_t fDimCycle;            // readout cycle range, may span from -fDimCycle to fDimCycle
71   TClonesArray*    fItems;   // pListItems array
72   TBtree*          fBTree;   // tree for ordered access
73   //
74   ClassDef(AliITSUSensMap,1) // list of sensor signals (should be sortable objects)
75 };      
76
77 //______________________________________________________________________
78 inline UInt_t AliITSUSensMap::GetIndex(UInt_t col,UInt_t row, Int_t cycle) const  
79 {
80   // linearized ID of digit
81   UInt_t cyclePos = cycle+fDimCycle; // cycle may span from -fDimCycle to fDimCycle
82 #ifdef _ROWWISE_SORT_
83   return fDimCol*(cyclePos*fDimRow+row)+col; // sorted in row, then in column
84 #else
85   return fDimRow*(cyclePos*fDimCol+col)+row; // sorted in column, then in row
86 #endif
87 }
88
89 //______________________________________________________________________
90 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) 
91 {
92   // returns the i,j index numbers from the linearized index computed with GetIndex
93   UInt_t dcr = dcol*drow;
94   cycle = int(index/dcr) - dcycle;
95   index %= dcr;
96 #ifdef _ROWWISE_SORT_
97   col = index%dcol;   // sorted in row, then in column
98   row = index/dcol;
99 #else
100   col = index/drow;   // sorted in column, then in row
101   row = index%drow;
102 #endif  
103 }
104
105
106 #endif