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 << 26 // G4LogicalVolumeStore implementation for singleton container 27 // 27 // 28 // 10.07.95, P.Kent, G.Cosmo << 28 // 10.07.95 - P.Kent, Initial version 29 // ------------------------------------------- 29 // -------------------------------------------------------------------- 30 30 31 #include "G4Types.hh" 31 #include "G4Types.hh" 32 #include "G4LogicalVolumeStore.hh" 32 #include "G4LogicalVolumeStore.hh" 33 #include "G4GeometryManager.hh" 33 #include "G4GeometryManager.hh" 34 34 35 #include "G4AutoLock.hh" << 36 << 37 namespace << 38 { << 39 G4Mutex mapMutex = G4MUTEX_INITIALIZER; << 40 } << 41 << 42 // ******************************************* 35 // *************************************************************************** 43 // Static class variables 36 // Static class variables 44 // ******************************************* 37 // *************************************************************************** 45 // 38 // 46 G4LogicalVolumeStore* G4LogicalVolumeStore::fg 39 G4LogicalVolumeStore* G4LogicalVolumeStore::fgInstance = nullptr; 47 G4ThreadLocal G4VStoreNotifier* G4LogicalVolum 40 G4ThreadLocal G4VStoreNotifier* G4LogicalVolumeStore::fgNotifier = nullptr; 48 G4ThreadLocal G4bool G4LogicalVolumeStore::loc 41 G4ThreadLocal G4bool G4LogicalVolumeStore::locked = false; 49 42 50 // ******************************************* 43 // *************************************************************************** 51 // Protected constructor: Construct underlying 44 // Protected constructor: Construct underlying container with 52 // initial size of 100 entries 45 // initial size of 100 entries 53 // ******************************************* 46 // *************************************************************************** 54 // 47 // 55 G4LogicalVolumeStore::G4LogicalVolumeStore() 48 G4LogicalVolumeStore::G4LogicalVolumeStore() 56 << 49 : std::vector<G4LogicalVolume*>() 57 { 50 { 58 reserve(100); 51 reserve(100); 59 } 52 } 60 53 61 // ******************************************* 54 // *************************************************************************** 62 // Destructor 55 // Destructor 63 // ******************************************* 56 // *************************************************************************** 64 // 57 // 65 G4LogicalVolumeStore::~G4LogicalVolumeStore() 58 G4LogicalVolumeStore::~G4LogicalVolumeStore() 66 { 59 { 67 Clean(); // Delete all volumes in the store 60 Clean(); // Delete all volumes in the store 68 G4LogicalVolume::Clean(); // Delete allocat 61 G4LogicalVolume::Clean(); // Delete allocated sub-instance data 69 } 62 } 70 63 71 // ******************************************* 64 // *************************************************************************** 72 // Delete all elements from the store 65 // Delete all elements from the store 73 // ******************************************* 66 // *************************************************************************** 74 // 67 // 75 void G4LogicalVolumeStore::Clean() 68 void G4LogicalVolumeStore::Clean() 76 { 69 { 77 // Do nothing if geometry is closed 70 // Do nothing if geometry is closed 78 // 71 // 79 if (G4GeometryManager::GetInstance()->IsGeom << 72 if (G4GeometryManager::IsGeometryClosed()) 80 { 73 { 81 G4cout << "WARNING - Attempt to delete the 74 G4cout << "WARNING - Attempt to delete the logical volume store" 82 << " while geometry closed !" << G4 75 << " while geometry closed !" << G4endl; 83 return; 76 return; 84 } 77 } 85 78 86 // Locks store for deletion of volumes. De-r 79 // Locks store for deletion of volumes. De-registration will be 87 // performed at this stage. G4LogicalVolumes 80 // performed at this stage. G4LogicalVolumes will not de-register themselves. 88 // 81 // 89 locked = true; 82 locked = true; 90 83 >> 84 size_t i = 0; 91 G4LogicalVolumeStore* store = GetInstance(); 85 G4LogicalVolumeStore* store = GetInstance(); 92 86 >> 87 #ifdef G4GEOMETRY_VOXELDEBUG >> 88 G4cout << "Deleting Logical Volumes ... "; >> 89 #endif >> 90 93 for(auto pos=store->cbegin(); pos!=store->ce 91 for(auto pos=store->cbegin(); pos!=store->cend(); ++pos) 94 { 92 { 95 if (fgNotifier != nullptr) { fgNotifier->N 93 if (fgNotifier != nullptr) { fgNotifier->NotifyDeRegistration(); } 96 if (*pos != nullptr) { (*pos)->Lock(); del 94 if (*pos != nullptr) { (*pos)->Lock(); delete *pos; } >> 95 ++i; 97 } 96 } 98 97 99 store->bmap.clear(); store->mvalid = false; << 98 #ifdef G4GEOMETRY_VOXELDEBUG >> 99 if (store->size() < i-1) >> 100 { G4cout << "No volumes deleted. Already deleted by user ?" << G4endl; } >> 101 else >> 102 { G4cout << i-1 << " volumes deleted !" << G4endl; } >> 103 #endif >> 104 100 locked = false; 105 locked = false; 101 store->clear(); 106 store->clear(); 102 } 107 } 103 108 104 // ******************************************* 109 // *************************************************************************** 105 // Associate user notifier to the store 110 // Associate user notifier to the store 106 // ******************************************* 111 // *************************************************************************** 107 // 112 // 108 void G4LogicalVolumeStore::SetNotifier(G4VStor 113 void G4LogicalVolumeStore::SetNotifier(G4VStoreNotifier* pNotifier) 109 { 114 { 110 GetInstance(); 115 GetInstance(); 111 fgNotifier = pNotifier; 116 fgNotifier = pNotifier; 112 } 117 } 113 118 114 // ******************************************* 119 // *************************************************************************** 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 120 // Add volume to container 143 // ******************************************* 121 // *************************************************************************** 144 // 122 // 145 void G4LogicalVolumeStore::Register(G4LogicalV 123 void G4LogicalVolumeStore::Register(G4LogicalVolume* pVolume) 146 { 124 { 147 G4LogicalVolumeStore* store = GetInstance(); << 125 GetInstance()->push_back(pVolume); 148 store->push_back(pVolume); << 126 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 } 127 } 163 128 164 // ******************************************* 129 // *************************************************************************** 165 // Remove volume from container 130 // Remove volume from container 166 // ******************************************* 131 // *************************************************************************** 167 // 132 // 168 void G4LogicalVolumeStore::DeRegister(G4Logica 133 void G4LogicalVolumeStore::DeRegister(G4LogicalVolume* pVolume) 169 { 134 { 170 G4LogicalVolumeStore* store = GetInstance(); << 171 if (!locked) // Do not de-register if loc 135 if (!locked) // Do not de-register if locked ! 172 { 136 { 173 if (fgNotifier != nullptr) { fgNotifier->N 137 if (fgNotifier != nullptr) { fgNotifier->NotifyDeRegistration(); } 174 for (auto i=store->cbegin(); i!=store->cen << 138 for (auto i=GetInstance()->cbegin(); i!=GetInstance()->cend(); ++i) 175 { 139 { 176 if (**i==*pVolume) 140 if (**i==*pVolume) 177 { 141 { 178 store->erase(i); << 142 GetInstance()->erase(i); 179 break; 143 break; 180 } 144 } 181 } 145 } 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 } 146 } 203 } 147 } 204 148 205 // ******************************************* 149 // *************************************************************************** 206 // Retrieve the first or last volume pointer i << 150 // Retrieve the first volume pointer in the container having that name 207 // ******************************************* 151 // *************************************************************************** 208 // 152 // 209 G4LogicalVolume* 153 G4LogicalVolume* 210 G4LogicalVolumeStore::GetVolume(const G4String << 154 G4LogicalVolumeStore::GetVolume(const G4String& name, G4bool verbose) const 211 G4bool reverse << 212 { 155 { 213 G4LogicalVolumeStore* store = GetInstance(); << 156 for (auto i=GetInstance()->cbegin(); i!=GetInstance()->cend(); ++i) 214 if (!store->mvalid) { store->UpdateMap(); } << 215 auto pos = store->bmap.find(name); << 216 if(pos != store->bmap.cend()) << 217 { 157 { 218 if ((verbose) && (pos->second.size()>1)) << 158 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 } 159 } 236 if (verbose) 160 if (verbose) 237 { 161 { 238 std::ostringstream message; 162 std::ostringstream message; 239 message << "Volume NOT found in store !" 163 message << "Volume NOT found in store !" << G4endl 240 << " Volume " << name << " 164 << " Volume " << name << " NOT found in store !" << G4endl 241 << " Returning NULL pointe 165 << " Returning NULL pointer."; 242 G4Exception("G4LogicalVolumeStore::GetVol 166 G4Exception("G4LogicalVolumeStore::GetVolume()", 243 "GeomMgt1001", JustWarning, m 167 "GeomMgt1001", JustWarning, message); 244 } 168 } 245 return nullptr; << 169 return 0; 246 } 170 } 247 171 248 // ******************************************* 172 // *************************************************************************** 249 // Return ptr to Store, setting if necessary 173 // Return ptr to Store, setting if necessary 250 // ******************************************* 174 // *************************************************************************** 251 // 175 // 252 G4LogicalVolumeStore* G4LogicalVolumeStore::Ge 176 G4LogicalVolumeStore* G4LogicalVolumeStore::GetInstance() 253 { 177 { 254 static G4LogicalVolumeStore worldStore; 178 static G4LogicalVolumeStore worldStore; 255 if (fgInstance == nullptr) 179 if (fgInstance == nullptr) 256 { 180 { 257 fgInstance = &worldStore; 181 fgInstance = &worldStore; 258 } 182 } 259 return fgInstance; 183 return fgInstance; 260 } 184 } 261 185