=operator implemented (R. Shahoyan)
[u/mrichter/AliRoot.git] / MUON / AliMUON2DMap.cxx
CommitLineData
f6e5d0e9 1/**************************************************************************
2* Copyright(c) 1998-1999, 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#include "AliMUON2DMap.h"
19
20#include "AliLog.h"
f246123b 21#include "AliMUON2DMapIterator.h"
4178b5c7 22#include "AliMUON2DMapIteratorByI.h"
f6e5d0e9 23#include "AliMpExMap.h"
24
3d1463c8 25//-----------------------------------------------------------------------------
5398f946 26/// \class AliMUON2DMap
4178b5c7 27/// Basic implementation of AliMUONVStore container using
9d5f6a64 28/// AliMpExMap internally.
29/// What we store is a "double" map : an AliMpExMap of AliMpExMaps
30///
5398f946 31/// \author Laurent Aphecetche
3d1463c8 32//-----------------------------------------------------------------------------
f6e5d0e9 33
5398f946 34/// \cond CLASSIMP
f6e5d0e9 35ClassImp(AliMUON2DMap)
5398f946 36/// \endcond
f6e5d0e9 37
7332f213 38const Int_t AliMUON2DMap::fgkOptimalSizeForDEManu = 228;
39
4178b5c7 40namespace
41{
42 //___________________________________________________________________________
43 TObject* GetValue(TExMapIter& iter, Int_t& theKey)
44 {
45 /// return the next value corresponding to theKey in iterator iter
46 theKey = -1;
47 Long_t key, value;
48 Bool_t ok = iter.Next(key,value);
49 if (!ok) return 0x0;
50 theKey = (Int_t)(key & 0xFFFF);
51 return reinterpret_cast<TObject*>(value);
52 }
53}
54
f6e5d0e9 55//_____________________________________________________________________________
d91c6144 56AliMUON2DMap::AliMUON2DMap(Bool_t optimizeForDEManu)
4178b5c7 57: AliMUONVStore(),
7332f213 58 fMap(new AliMpExMap(kTRUE)),
d91c6144 59 fOptimizeForDEManu(optimizeForDEManu)
f6e5d0e9 60{
4178b5c7 61 /// Default constructor.
7332f213 62 // hard-coded constant in order not to depend on mapping
63 // if this number ever change, it will not break the code, simply the
64 // automatic resizing will give a warning...
65 if ( fOptimizeForDEManu ) fMap->SetSize(fgkOptimalSizeForDEManu);
f6e5d0e9 66}
67
9d5f6a64 68//_____________________________________________________________________________
69AliMUON2DMap::AliMUON2DMap(const AliMUON2DMap& other)
4178b5c7 70: AliMUONVStore(),
7332f213 71 fMap(new AliMpExMap(*other.fMap)),
72 fOptimizeForDEManu(other.fOptimizeForDEManu)
9d5f6a64 73{
5398f946 74 /// Copy constructor.
9d5f6a64 75}
884a73f1 76
9d5f6a64 77//_____________________________________________________________________________
78AliMUON2DMap&
79AliMUON2DMap::operator=(const AliMUON2DMap& other)
80{
5398f946 81/// Assignment operator
7332f213 82 *fMap = *other.fMap;
83 fOptimizeForDEManu = other.fOptimizeForDEManu;
9d5f6a64 84 return *this;
884a73f1 85}
86
f6e5d0e9 87//_____________________________________________________________________________
88AliMUON2DMap::~AliMUON2DMap()
89{
5398f946 90/// Destructor.
91/// We delete the map, which will delete the objects, as we're owner.
f6e5d0e9 92 delete fMap;
93}
94
9d5f6a64 95//_____________________________________________________________________________
4178b5c7 96AliMUONVStore*
97AliMUON2DMap::Create() const
d91c6144 98{
99 /// Create a void copy of *this.
547c1de0 100 return new AliMUON2DMap(fOptimizeForDEManu);
d91c6144 101}
102
103//_____________________________________________________________________________
4178b5c7 104Bool_t
105AliMUON2DMap::Add(TObject* object)
106{
107 /// Add object, using the decoding of uniqueID into two ints as the key
7332f213 108 if (!object) return kFALSE;
4178b5c7 109 UInt_t uniqueID = object->GetUniqueID();
110 Int_t j = ( uniqueID & 0xFFFF0000 ) >> 16;
111 Int_t i = ( uniqueID & 0xFFFF);
112 return Set(i,j,object,kFALSE);
113}
114
115//_____________________________________________________________________________
f6e5d0e9 116TObject*
4178b5c7 117AliMUON2DMap::FindObject(Int_t i, Int_t j) const
f6e5d0e9 118{
4178b5c7 119 /// Return the value at position (i,j).
4178b5c7 120 AliMpExMap* m = static_cast<AliMpExMap*>(fMap->GetValue(i));
7332f213 121 return m ? m->GetValue(j) : 0x0;
4178b5c7 122}
123
124//_____________________________________________________________________________
125TIterator*
126AliMUON2DMap::CreateIterator() const
127{
128 // Create and return an iterator on this map
129 // Returned iterator must be deleted by user.
7332f213 130 return new AliMUON2DMapIterator(*fMap);
f6e5d0e9 131}
132
133//_____________________________________________________________________________
4178b5c7 134TIterator*
135AliMUON2DMap::CreateIterator(Int_t firstI, Int_t lastI) const
f246123b 136{
137 // Create and return an iterator on this map
138 // Returned iterator must be deleted by user.
7332f213 139 return new AliMUON2DMapIteratorByI(*fMap,firstI,lastI);
f246123b 140}
141
142//_____________________________________________________________________________
4178b5c7 143void
144AliMUON2DMap::Clear(Option_t*)
49f43921 145{
7332f213 146 /// Clear memory
147 fMap->Clear();
4178b5c7 148}
149
150//_____________________________________________________________________________
151Int_t
152AliMUON2DMap::GetSize() const
153{
154 /// Return the number of objects we hold
155 TExMapIter iter(fMap->GetIterator());
156 Int_t i;
157 Int_t theSize(0);
49f43921 158
4178b5c7 159 while ( GetValue(iter,i) )
160 {
161 theSize += GetSize(i);
162 }
163 return theSize;
164}
165
166//_____________________________________________________________________________
167Int_t
168AliMUON2DMap::GetSize(Int_t i) const
169{
170 /// Return the number of objects we hold
171 AliMpExMap* m = static_cast<AliMpExMap*>(fMap->GetValue(i));
7332f213 172 return m ? m->GetSize() : 0;
49f43921 173}
174
175//_____________________________________________________________________________
f6e5d0e9 176Bool_t
177AliMUON2DMap::Set(Int_t i, Int_t j, TObject* object, Bool_t replace)
178{
5398f946 179/// Set the object at position (i,j).
180/// If replace==kTRUE, we don't care if there's an object there already,
181/// otherwise we might refuse to set if the (i,j) location is already
182/// filled (in which case we return kFALSE).
e5cc543e 183
f6e5d0e9 184 TObject* o = fMap->GetValue(i);
185 if ( !o )
186 {
187 AliMpExMap* m = new AliMpExMap(true);
d91c6144 188 if ( fOptimizeForDEManu )
189 {
f0195aa5 190 m->SetSize(451); // same remark as for the SetSize in ctor...
d91c6144 191 }
f6e5d0e9 192 fMap->Add(i,m);
193 o = fMap->GetValue(i);
f6e5d0e9 194 }
4178b5c7 195 AliMpExMap* m = static_cast<AliMpExMap*>(o);
4178b5c7 196
f6e5d0e9 197 o = m->GetValue(j);
4178b5c7 198
199 if ( !o )
f6e5d0e9 200 {
f6e5d0e9 201 m->Add(j,object);
202 }
4178b5c7 203 else
f6e5d0e9 204 {
4178b5c7 205 if ( replace )
206 {
207 delete o;
208 m->Add(j,object);
209 }
210 else
211 {
212 AliError(Form("Object %p is already there for (i,j)=(%d,%d)",o,i,j));
213 return kFALSE;
214 }
f6e5d0e9 215 }
4178b5c7 216
f6e5d0e9 217 return kTRUE;
218}
219