MUON/AliMUON2DMap.cxx

   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"
  21 #include "AliMUON2DMapIterator.h"
  22 #include "AliMUON2DMapIteratorByI.h"
  23 #include "AliMpExMap.h"
  24 #include "AliMpManuList.h"
  25 #include "AliMpDEManager.h"
  26 #include "AliMpConstants.h"
  27
  28 //-----------------------------------------------------------------------------
  29 /// \class AliMUON2DMap
  30 /// Basic implementation of AliMUONVStore container using
  31 /// AliMpExMap internally.
  32 /// What we store is a "double" map : an AliMpExMap of AliMpExMaps
  33 ///
  34 /// \author Laurent Aphecetche
  35 //-----------------------------------------------------------------------------
  36
  37 /// \cond CLASSIMP
  38 ClassImp(AliMUON2DMap)
  39 /// \endcond
  40
  41 namespace
  42 {
  43   //___________________________________________________________________________
  44   TObject* GetValue(TExMapIter& iter, Int_t& theKey)
  45   {
  46     /// return the next value corresponding to theKey in iterator iter
  47     theKey = -1;
  48     Long_t key, value;
  49     Bool_t ok = iter.Next(key,value);
  50     if (!ok) return 0x0;
  51     theKey = (Int_t)(key & 0xFFFF);
  52     return reinterpret_cast<TObject*>(value);
  53   }
  54 }
  55
  56 //_____________________________________________________________________________
  57 AliMUON2DMap::AliMUON2DMap(Bool_t optimizeForDEManu)
  58 : AliMUONVStore(),
  59   fMap(0x0),
  60   fOptimizeForDEManu(optimizeForDEManu)
  61 {
  62   /// Default constructor.
  63     Clear();
  64 }
  65
  66 //_____________________________________________________________________________
  67 AliMUON2DMap::AliMUON2DMap(const AliMUON2DMap& other)
  68 : AliMUONVStore(),
  69 fMap(0x0),
  70 fOptimizeForDEManu(kFALSE)
  71 {
  72  /// Copy constructor.
  73
  74  other.CopyTo(*this);
  75 }
  76
  77 //_____________________________________________________________________________
  78 AliMUON2DMap&
  79 AliMUON2DMap::operator=(const AliMUON2DMap& other)
  80 {
  81 /// Assignment operator
  82
  83   other.CopyTo(*this);
  84   return *this;
  85 }
  86
  87 //_____________________________________________________________________________
  88 AliMUON2DMap::~AliMUON2DMap()
  89 {
  90 /// Destructor.
  91 /// We delete the map, which will delete the objects, as we're owner.
  92
  93   delete fMap;
  94 }
  95
  96 //_____________________________________________________________________________
  97 AliMUONVStore*
  98 AliMUON2DMap::Create() const
  99 {
 100   /// Create a void copy of *this.
 101   return new AliMUON2DMap(fOptimizeForDEManu);
 102 }
 103
 104 //_____________________________________________________________________________
 105 void
 106 AliMUON2DMap::CopyTo(AliMUON2DMap& dest) const
 107 {
 108   /// Copy this into dest.
 109
 110   delete dest.fMap;
 111   dest.fMap = new AliMpExMap(*fMap);
 112   dest.fOptimizeForDEManu = fOptimizeForDEManu;
 113 }
 114
 115 //_____________________________________________________________________________
 116 Bool_t
 117 AliMUON2DMap::Add(TObject* object)
 118 {
 119   /// Add object, using the decoding of uniqueID into two ints as the key
 120   UInt_t uniqueID = object->GetUniqueID();
 121   Int_t j = ( uniqueID & 0xFFFF0000 ) >> 16;
 122   Int_t i = ( uniqueID & 0xFFFF);
 123   return Set(i,j,object,kFALSE);
 124 }
 125
 126 //_____________________________________________________________________________
 127 TObject*
 128 AliMUON2DMap::FindObject(Int_t i, Int_t j) const
 129 {
 130   /// Return the value at position (i,j).
 131
 132   AliMpExMap* m = static_cast<AliMpExMap*>(fMap->GetValue(i));
 133   if (m) return m->GetValue(j);
 134   return 0x0;
 135 }
 136
 137 //_____________________________________________________________________________
 138 TIterator*
 139 AliMUON2DMap::CreateIterator() const
 140 {
 141   // Create and return an iterator on this map
 142   // Returned iterator must be deleted by user.
 143   if ( fMap )
 144   {
 145     return new AliMUON2DMapIterator(*fMap);
 146   }
 147   return 0x0;
 148 }
 149
 150 //_____________________________________________________________________________
 151 TIterator*
 152 AliMUON2DMap::CreateIterator(Int_t firstI, Int_t lastI) const
 153 {
 154   // Create and return an iterator on this map
 155   // Returned iterator must be deleted by user.
 156   if ( fMap )
 157   {
 158     return new AliMUON2DMapIteratorByI(*fMap,firstI,lastI);
 159   }
 160   return 0x0;
 161 }
 162
 163 //_____________________________________________________________________________
 164 void
 165 AliMUON2DMap::Clear(Option_t*)
 166 {
 167   /// Reset
 168   delete fMap;
 169
 170   fMap = new AliMpExMap(kTRUE);
 171
 172   if ( fOptimizeForDEManu )
 173   {
 174     Int_t nDEs(0);
 175     for ( Int_t i = 0; i < AliMpConstants::NofChambers(); ++i )
 176     {
 177       nDEs += AliMpDEManager::GetNofDEInChamber(i);
 178     }
 179     fMap->SetSize(nDEs);
 180   }
 181 }
 182
 183 //_____________________________________________________________________________
 184 Int_t
 185 AliMUON2DMap::GetSize() const
 186 {
 187   /// Return the number of objects we hold
 188   TExMapIter iter(fMap->GetIterator());
 189   Int_t i;
 190   Int_t theSize(0);
 191
 192   while ( GetValue(iter,i) )
 193   {
 194     theSize += GetSize(i);
 195   }
 196   return theSize;
 197 }
 198
 199 //_____________________________________________________________________________
 200 Int_t
 201 AliMUON2DMap::GetSize(Int_t i) const
 202 {
 203   /// Return the number of objects we hold
 204   AliMpExMap* m = static_cast<AliMpExMap*>(fMap->GetValue(i));
 205   if (m)
 206   {
 207     return m->GetSize();
 208   }
 209   return 0;
 210 }
 211
 212 //_____________________________________________________________________________
 213 Bool_t
 214 AliMUON2DMap::Set(Int_t i, Int_t j, TObject* object, Bool_t replace)
 215 {
 216 /// Set the object at position (i,j).
 217 /// If replace==kTRUE, we don't care if there's an object there already,
 218 /// otherwise we might refuse to set if the (i,j) location is already
 219 /// filled (in which case we return kFALSE).
 220
 221   TObject* o = fMap->GetValue(i);
 222   if ( !o )
 223   {
 224     AliMpExMap* m = new AliMpExMap(true);
 225     if ( fOptimizeForDEManu )
 226     {
 227       Int_t n(AliMpManuList::NumberOfManus(i));
 228       if (!n)
 229       {
 230         AliError(Form("This does not look right : i = %d is supposed to "
 231                       "be a DetElemId with n = %d manus!",i,n));
 232       }
 233       else
 234       {
 235         m->SetSize(n);
 236       }
 237     }
 238     fMap->Add(i,m);
 239     o = fMap->GetValue(i);
 240   }
 241   AliMpExMap* m = static_cast<AliMpExMap*>(o);
 242 //  AliMpExMap* m = dynamic_cast<AliMpExMap*>(o);
 243 //  if (!m) AliFatal(Form("fMap[%d] not of the expected type",i));
 244
 245   o = m->GetValue(j);
 246
 247   if ( !o )
 248   {
 249     m->Add(j,object);
 250   }
 251   else
 252   {
 253     if ( replace )
 254     {
 255       delete o;
 256       m->Add(j,object);
 257     }
 258     else
 259     {
 260       AliError(Form("Object %p is already there for (i,j)=(%d,%d)",o,i,j));
 261       return kFALSE;
 262     }
 263   }
 264
 265   return kTRUE;
 266 }
 267
 268
 269
 270
 271