Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer 3 // * License and Disclaimer * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. 9 // * include a list of copyright holders. * 10 // * 10 // * * 11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. * 17 // * 17 // * * 18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************* 24 // ******************************************************************** 25 // 25 // 26 // G4LogicalVolumeStore implementation << 27 // 26 // 28 // 10.07.95, P.Kent, G.Cosmo << 27 // $Id$ >> 28 // >> 29 // G4LogicalVolumeStore >> 30 // >> 31 // Implementation for singleton container >> 32 // >> 33 // History: >> 34 // 10.07.95 P.Kent Initial version 29 // ------------------------------------------- 35 // -------------------------------------------------------------------- 30 36 31 #include "G4Types.hh" 37 #include "G4Types.hh" 32 #include "G4LogicalVolumeStore.hh" 38 #include "G4LogicalVolumeStore.hh" 33 #include "G4GeometryManager.hh" 39 #include "G4GeometryManager.hh" 34 40 35 #include "G4AutoLock.hh" << 36 << 37 namespace << 38 { << 39 G4Mutex mapMutex = G4MUTEX_INITIALIZER; << 40 } << 41 << 42 // ******************************************* 41 // *************************************************************************** 43 // Static class variables 42 // Static class variables 44 // ******************************************* 43 // *************************************************************************** 45 // 44 // 46 G4LogicalVolumeStore* G4LogicalVolumeStore::fg << 45 G4LogicalVolumeStore* G4LogicalVolumeStore::fgInstance = 0; 47 G4ThreadLocal G4VStoreNotifier* G4LogicalVolum << 46 G4VStoreNotifier* G4LogicalVolumeStore::fgNotifier = 0; 48 G4ThreadLocal G4bool G4LogicalVolumeStore::loc << 47 G4bool G4LogicalVolumeStore::locked = false; 49 48 50 // ******************************************* 49 // *************************************************************************** 51 // Protected constructor: Construct underlying 50 // Protected constructor: Construct underlying container with 52 // initial size of 100 entries 51 // initial size of 100 entries 53 // ******************************************* 52 // *************************************************************************** 54 // 53 // 55 G4LogicalVolumeStore::G4LogicalVolumeStore() 54 G4LogicalVolumeStore::G4LogicalVolumeStore() 56 << 55 : std::vector<G4LogicalVolume*>() 57 { 56 { 58 reserve(100); 57 reserve(100); 59 } 58 } 60 59 61 // ******************************************* 60 // *************************************************************************** 62 // Destructor 61 // Destructor 63 // ******************************************* 62 // *************************************************************************** 64 // 63 // 65 G4LogicalVolumeStore::~G4LogicalVolumeStore() 64 G4LogicalVolumeStore::~G4LogicalVolumeStore() 66 { 65 { 67 Clean(); // Delete all volumes in the store << 66 Clean(); 68 G4LogicalVolume::Clean(); // Delete allocat << 69 } 67 } 70 68 71 // ******************************************* 69 // *************************************************************************** 72 // Delete all elements from the store 70 // Delete all elements from the store 73 // ******************************************* 71 // *************************************************************************** 74 // 72 // 75 void G4LogicalVolumeStore::Clean() 73 void G4LogicalVolumeStore::Clean() 76 { 74 { 77 // Do nothing if geometry is closed 75 // Do nothing if geometry is closed 78 // 76 // 79 if (G4GeometryManager::GetInstance()->IsGeom 77 if (G4GeometryManager::GetInstance()->IsGeometryClosed()) 80 { 78 { 81 G4cout << "WARNING - Attempt to delete the 79 G4cout << "WARNING - Attempt to delete the logical volume store" 82 << " while geometry closed !" << G4 80 << " while geometry closed !" << G4endl; 83 return; 81 return; 84 } 82 } 85 83 86 // Locks store for deletion of volumes. De-r 84 // Locks store for deletion of volumes. De-registration will be 87 // performed at this stage. G4LogicalVolumes 85 // performed at this stage. G4LogicalVolumes will not de-register themselves. 88 // 86 // 89 locked = true; 87 locked = true; 90 88 >> 89 size_t i=0; 91 G4LogicalVolumeStore* store = GetInstance(); 90 G4LogicalVolumeStore* store = GetInstance(); 92 91 93 for(auto pos=store->cbegin(); pos!=store->ce << 92 #ifdef G4GEOMETRY_VOXELDEBUG >> 93 G4cout << "Deleting Logical Volumes ... "; >> 94 #endif >> 95 >> 96 for(iterator pos=store->begin(); pos!=store->end(); pos++) 94 { 97 { 95 if (fgNotifier != nullptr) { fgNotifier->N << 98 if (fgNotifier) { fgNotifier->NotifyDeRegistration(); } 96 if (*pos != nullptr) { (*pos)->Lock(); del << 99 if (*pos) { (*pos)->Lock(); delete *pos; } >> 100 i++; 97 } 101 } 98 102 99 store->bmap.clear(); store->mvalid = false; << 103 #ifdef G4GEOMETRY_VOXELDEBUG >> 104 if (store->size() < i-1) >> 105 { G4cout << "No volumes deleted. Already deleted by user ?" << G4endl; } >> 106 else >> 107 { G4cout << i-1 << " volumes deleted !" << G4endl; } >> 108 #endif >> 109 100 locked = false; 110 locked = false; 101 store->clear(); 111 store->clear(); 102 } 112 } 103 113 104 // ******************************************* 114 // *************************************************************************** 105 // Associate user notifier to the store 115 // Associate user notifier to the store 106 // ******************************************* 116 // *************************************************************************** 107 // 117 // 108 void G4LogicalVolumeStore::SetNotifier(G4VStor 118 void G4LogicalVolumeStore::SetNotifier(G4VStoreNotifier* pNotifier) 109 { 119 { 110 GetInstance(); 120 GetInstance(); 111 fgNotifier = pNotifier; 121 fgNotifier = pNotifier; 112 } 122 } 113 123 114 // ******************************************* 124 // *************************************************************************** 115 // Bring contents of internal map up to date a << 116 // ******************************************* << 117 // << 118 void G4LogicalVolumeStore::UpdateMap() << 119 { << 120 G4AutoLock l(&mapMutex); // to avoid thread << 121 if (mvalid) return; << 122 bmap.clear(); << 123 for(auto pos=GetInstance()->cbegin(); pos!=G << 124 { << 125 const G4String& vol_name = (*pos)->GetName << 126 auto it = bmap.find(vol_name); << 127 if (it != bmap.cend()) << 128 { << 129 it->second.push_back(*pos); << 130 } << 131 else << 132 { << 133 std::vector<G4LogicalVolume*> vol_vec { << 134 bmap.insert(std::make_pair(vol_name, vol << 135 } << 136 } << 137 mvalid = true; << 138 l.unlock(); << 139 } << 140 << 141 // ******************************************* << 142 // Add volume to container 125 // Add volume to container 143 // ******************************************* 126 // *************************************************************************** 144 // 127 // 145 void G4LogicalVolumeStore::Register(G4LogicalV 128 void G4LogicalVolumeStore::Register(G4LogicalVolume* pVolume) 146 { 129 { 147 G4LogicalVolumeStore* store = GetInstance(); << 130 GetInstance()->push_back(pVolume); 148 store->push_back(pVolume); << 131 if (fgNotifier) { fgNotifier->NotifyRegistration(); } 149 const G4String& vol_name = pVolume->GetName( << 150 auto it = store->bmap.find(vol_name); << 151 if (it != store->bmap.cend()) << 152 { << 153 it->second.push_back(pVolume); << 154 } << 155 else << 156 { << 157 std::vector<G4LogicalVolume*> vol_vec { pV << 158 store->bmap.insert(std::make_pair(vol_name << 159 } << 160 if (fgNotifier != nullptr) { fgNotifier->Not << 161 store->mvalid = true; << 162 } 132 } 163 133 164 // ******************************************* 134 // *************************************************************************** 165 // Remove volume from container 135 // Remove volume from container 166 // ******************************************* 136 // *************************************************************************** 167 // 137 // 168 void G4LogicalVolumeStore::DeRegister(G4Logica 138 void G4LogicalVolumeStore::DeRegister(G4LogicalVolume* pVolume) 169 { 139 { 170 G4LogicalVolumeStore* store = GetInstance(); << 171 if (!locked) // Do not de-register if loc 140 if (!locked) // Do not de-register if locked ! 172 { 141 { 173 if (fgNotifier != nullptr) { fgNotifier->N << 142 if (fgNotifier) { fgNotifier->NotifyDeRegistration(); } 174 for (auto i=store->cbegin(); i!=store->cen << 143 for (iterator i=GetInstance()->begin(); i!=GetInstance()->end(); i++) 175 { 144 { 176 if (**i==*pVolume) 145 if (**i==*pVolume) 177 { 146 { 178 store->erase(i); << 147 GetInstance()->erase(i); 179 break; 148 break; 180 } 149 } 181 } 150 } 182 const G4String& vol_name = pVolume->GetNam << 183 auto it = store->bmap.find(vol_name); << 184 if (it != store->bmap.cend()) << 185 { << 186 if (it->second.size() > 1) << 187 { << 188 for (auto i=it->second.cbegin(); i!=it << 189 { << 190 if (**i==*pVolume) << 191 { << 192 it->second.erase(i); << 193 break; << 194 } << 195 } << 196 } << 197 else << 198 { << 199 store->bmap.erase(it); << 200 } << 201 } << 202 } 151 } 203 } 152 } 204 153 205 // ******************************************* 154 // *************************************************************************** 206 // Retrieve the first or last volume pointer i << 155 // Retrieve the first volume pointer in the container having that name 207 // ******************************************* 156 // *************************************************************************** 208 // 157 // 209 G4LogicalVolume* 158 G4LogicalVolume* 210 G4LogicalVolumeStore::GetVolume(const G4String << 159 G4LogicalVolumeStore::GetVolume(const G4String& name, G4bool verbose) const 211 G4bool reverse << 212 { 160 { 213 G4LogicalVolumeStore* store = GetInstance(); << 161 for (iterator i=GetInstance()->begin(); i!=GetInstance()->end(); i++) 214 if (!store->mvalid) { store->UpdateMap(); } << 215 auto pos = store->bmap.find(name); << 216 if(pos != store->bmap.cend()) << 217 { 162 { 218 if ((verbose) && (pos->second.size()>1)) << 163 if ((*i)->GetName() == name) { return *i; } 219 { << 220 std::ostringstream message; << 221 message << "There exists more than ONE l << 222 << name << "!" << G4endl << 223 << "Returning the first found."; << 224 G4Exception("G4LogicalVolumeStore::GetVo << 225 "GeomMgt1001", JustWarning, << 226 } << 227 if(reverseSearch) << 228 { << 229 return pos->second[pos->second.size()-1] << 230 } << 231 else << 232 { << 233 return pos->second[0]; << 234 } << 235 } 164 } 236 if (verbose) 165 if (verbose) 237 { 166 { 238 std::ostringstream message; 167 std::ostringstream message; 239 message << "Volume NOT found in store !" 168 message << "Volume NOT found in store !" << G4endl 240 << " Volume " << name << " 169 << " Volume " << name << " NOT found in store !" << G4endl 241 << " Returning NULL pointe 170 << " Returning NULL pointer."; 242 G4Exception("G4LogicalVolumeStore::GetVol 171 G4Exception("G4LogicalVolumeStore::GetVolume()", 243 "GeomMgt1001", JustWarning, m 172 "GeomMgt1001", JustWarning, message); 244 } 173 } 245 return nullptr; << 174 return 0; 246 } 175 } 247 176 248 // ******************************************* 177 // *************************************************************************** 249 // Return ptr to Store, setting if necessary 178 // Return ptr to Store, setting if necessary 250 // ******************************************* 179 // *************************************************************************** 251 // 180 // 252 G4LogicalVolumeStore* G4LogicalVolumeStore::Ge 181 G4LogicalVolumeStore* G4LogicalVolumeStore::GetInstance() 253 { 182 { 254 static G4LogicalVolumeStore worldStore; 183 static G4LogicalVolumeStore worldStore; 255 if (fgInstance == nullptr) << 184 if (!fgInstance) 256 { 185 { 257 fgInstance = &worldStore; 186 fgInstance = &worldStore; 258 } 187 } 259 return fgInstance; 188 return fgInstance; 260 } 189 } 261 190