Geant4 Cross Reference |
1 // 2 // ******************************************************************** 3 // * License and Disclaimer * 4 // * * 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. * 10 // * * 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitation of liability. * 17 // * * 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************************************** 25 // 26 // G4ScoringProbe 27 // -------------------------------------------------------------------- 28 29 #include "G4ScoringProbe.hh" 30 31 #include "G4Box.hh" 32 #include "G4LogicalVolume.hh" 33 #include "G4LogicalVolumeStore.hh" 34 #include "G4VPhysicalVolume.hh" 35 #include "G4PVPlacement.hh" 36 #include "G4Region.hh" 37 #include "G4RegionStore.hh" 38 #include "G4Threading.hh" 39 #include "G4Material.hh" 40 #include "G4NistManager.hh" 41 42 #include "G4MultiFunctionalDetector.hh" 43 #include "G4VPrimitiveScorer.hh" 44 45 #include "G4ScoringManager.hh" 46 #include "G4StatDouble.hh" 47 48 #include "G4SystemOfUnits.hh" 49 #include "G4VisAttributes.hh" 50 51 #include "G4AutoLock.hh" 52 53 namespace 54 { 55 G4Mutex logvolmutex = G4MUTEX_INITIALIZER; 56 } 57 58 G4ScoringProbe::G4ScoringProbe(const G4String& lvName, G4double half_size, 59 G4bool checkOverlap) 60 : G4VScoringMesh(lvName) 61 , chkOverlap(checkOverlap) 62 , layeredMaterialName("none") 63 , layeredMaterial(nullptr) 64 { 65 fShape = MeshShape::probe; 66 logVolName = lvName; 67 probeSize = half_size; 68 G4double hs[] = { half_size, half_size, half_size }; 69 SetSize(hs); 70 G4int nBin[] = { 1, 1, 1 }; 71 SetNumberOfSegments(nBin); 72 regName = lvName + "_region"; 73 if(G4Threading::IsMasterThread()) 74 { 75 new G4Region(regName); 76 } 77 } 78 79 void G4ScoringProbe::List() const 80 { 81 G4cout << "G4ScoringProbe : " << logVolName << G4endl; 82 std::size_t np = posVec.size(); 83 for(std::size_t i = 0; i < np; ++i) 84 { 85 G4cout << " >> probe #" << i << " at " << posVec[i] << G4endl; 86 } 87 G4VScoringMesh::List(); 88 } 89 90 void G4ScoringProbe::SetupGeometry(G4VPhysicalVolume* worldPhys) 91 { 92 if(G4Threading::IsMasterThread()) 93 { 94 auto worldLog = worldPhys->GetLogicalVolume(); 95 auto region = G4RegionStore::GetInstance()->GetRegion(regName); 96 assert(region != nullptr); 97 region->AddRootLogicalVolume(worldLog); 98 region->SetWorld(worldPhys); 99 100 auto boxSolid = 101 new G4Box(logVolName + "_solid", probeSize, probeSize, probeSize); 102 fMeshElementLogical = 103 new G4LogicalVolume(boxSolid, layeredMaterial, logVolName + "_log"); 104 105 std::size_t np = posVec.size(); 106 for(std::size_t i = 0; i < np; ++i) 107 { 108 new G4PVPlacement(nullptr, posVec[i], fMeshElementLogical, logVolName + "_phy", 109 worldLog, false, (G4int)i, chkOverlap); 110 } 111 112 auto wisatt = new G4VisAttributes(G4Colour(.5, .5, .5)); 113 wisatt->SetVisibility(false); 114 worldLog->SetVisAttributes(wisatt); 115 auto visatt = new G4VisAttributes(G4Colour(.5, .5, .5)); 116 visatt->SetVisibility(true); 117 fMeshElementLogical->SetVisAttributes(visatt); 118 } 119 else 120 { 121 G4AutoLock l(&logvolmutex); 122 fMeshElementLogical = 123 G4LogicalVolumeStore::GetInstance()->GetVolume(logVolName, false); 124 assert(fMeshElementLogical != nullptr); 125 l.unlock(); 126 } 127 128 fMeshElementLogical->SetSensitiveDetector(fMFD); 129 } 130 131 G4bool G4ScoringProbe::SetMaterial(G4String val) 132 { 133 if(val == "none") 134 { 135 layeredMaterialName = val; 136 layeredMassFlg = false; 137 layeredMaterial = nullptr; 138 } 139 else 140 { 141 G4AutoLock l(&logvolmutex); 142 auto mat = G4NistManager::Instance()->FindOrBuildMaterial(val); 143 if(mat == nullptr) 144 { 145 return false; 146 } 147 layeredMaterialName = val; 148 layeredMassFlg = true; 149 layeredMaterial = mat; 150 if(G4Threading::IsMasterThread()) 151 { 152 auto region = G4RegionStore::GetInstance()->GetRegion(regName); 153 assert(region != nullptr); 154 region->UpdateMaterialList(); 155 } 156 l.unlock(); 157 } 158 return true; 159 } 160