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 // << 27 /// \file DicomPhantomParameterisationColour.c 26 /// \file DicomPhantomParameterisationColour.cc 28 /// \brief Implementation of the DicomPhantomP 27 /// \brief Implementation of the DicomPhantomParameterisationColour class 29 28 30 #include "DicomPhantomParameterisationColour.h 29 #include "DicomPhantomParameterisationColour.hh" 31 30 32 #include "DicomHandler.hh" << 31 #include "globals.hh" 33 << 32 #include "G4VisAttributes.hh" 34 #include "G4LogicalVolume.hh" << 35 #include "G4Material.hh" 33 #include "G4Material.hh" 36 #include "G4VPhysicalVolume.hh" 34 #include "G4VPhysicalVolume.hh" 37 #include "G4VVisManager.hh" << 35 #include "G4LogicalVolume.hh" 38 #include "G4VisAttributes.hh" << 39 #include "globals.hh" << 40 << 41 //....oooOO0OOooo........oooOO0OOooo........oo << 42 << 43 G4String DicomPhantomParameterisationColour::f << 44 DicomHandler::GetDicomDataPath() + "/ColourM << 45 36 46 //....oooOO0OOooo........oooOO0OOooo........oo 37 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 47 DicomPhantomParameterisationColour::DicomPhant << 38 DicomPhantomParameterisationColour::DicomPhantomParameterisationColour() 48 : G4PhantomParameterisation() << 49 { 39 { 50 ReadColourData(colourFile); << 40 ReadColourData(); 51 SetSkipEqualMaterials(false); << 52 } 41 } 53 42 54 //....oooOO0OOooo........oooOO0OOooo........oo 43 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 55 DicomPhantomParameterisationColour::~DicomPhan << 44 DicomPhantomParameterisationColour::~DicomPhantomParameterisationColour() >> 45 { >> 46 } 56 47 57 //....oooOO0OOooo........oooOO0OOooo........oo 48 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 58 void DicomPhantomParameterisationColour::ReadC << 49 void DicomPhantomParameterisationColour::ReadColourData() 59 { 50 { 60 //----- Add a G4VisAttributes for materials 51 //----- Add a G4VisAttributes for materials not defined in file; 61 G4VisAttributes* blankAtt = new G4VisAttribu 52 G4VisAttributes* blankAtt = new G4VisAttributes; 62 blankAtt->SetVisibility(FALSE); << 53 blankAtt->SetVisibility( FALSE ); 63 fColours["Default"] = blankAtt; 54 fColours["Default"] = blankAtt; 64 55 65 //----- Read file 56 //----- Read file 66 std::ifstream fin(colourFile.c_str()); << 57 std::ifstream fin("ColourMap.dat"); 67 G4int nMate; 58 G4int nMate; 68 G4String mateName; 59 G4String mateName; 69 G4double cred, cgreen, cblue, copacity; 60 G4double cred, cgreen, cblue, copacity; 70 fin >> nMate; 61 fin >> nMate; 71 for (G4int ii = 0; ii < nMate; ii++) { << 62 for( G4int ii = 0; ii < nMate; ii++ ){ 72 fin >> mateName; << 63 fin >> mateName >> cred >> cgreen >> cblue >> copacity; 73 if (fin.eof()) break; << 64 G4Colour colour( cred, cgreen, cblue, copacity ); 74 fin >> cred >> cgreen >> cblue >> copacity << 65 G4VisAttributes* visAtt = new G4VisAttributes( colour ); 75 G4Colour colour(cred, cgreen, cblue, copac << 76 G4VisAttributes* visAtt = new G4VisAttribu << 77 visAtt->SetVisibility(true); << 78 fColours[mateName] = visAtt; 66 fColours[mateName] = visAtt; 79 fColours2[ii] = new G4VisAttributes(*visAt << 80 } 67 } 81 } << 82 68 >> 69 } >> 70 83 //....oooOO0OOooo........oooOO0OOooo........oo 71 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 84 G4Material* DicomPhantomParameterisationColour << 72 G4Material* DicomPhantomParameterisationColour:: 85 << 73 ComputeMaterial(const G4int copyNo, G4VPhysicalVolume * physVol, const G4VTouchable *) 86 << 74 { 87 { << 75 G4Material* mate = G4PhantomParameterisation::ComputeMaterial( copyNo, physVol, 0 ); 88 G4Material* mate = G4PhantomParameterisation << 76 if( physVol ) { 89 << 90 if (G4VVisManager::GetConcreteInstance() && << 91 G4String mateName = mate->GetName(); 77 G4String mateName = mate->GetName(); 92 std::string::size_type iuu = mateName.find 78 std::string::size_type iuu = mateName.find("__"); 93 if (iuu != std::string::npos) mateName = m << 79 if( iuu != std::string::npos ) { 94 << 80 mateName = mateName.substr( 0, iuu ); 95 if (0 < fColours.count(mateName)) << 96 physVol->GetLogicalVolume()->SetVisAttri << 97 else { << 98 bool found = false; << 99 for (const auto& itr : fColours) { << 100 G4String mat_color = itr.first; << 101 auto len = mat_color.length(); << 102 if (mateName.find(mat_color) == 0 && m << 103 physVol->GetLogicalVolume()->SetVisA << 104 found = true; << 105 } << 106 if (found) break; << 107 } << 108 if (!found) { << 109 G4int matIndex = G4int(GetMaterialInde << 110 static uintmax_t n = 0; << 111 if (n++ < 100) << 112 G4cout << "Unknown material name " < << 113 if (fColours2.find(matIndex) != fColou << 114 physVol->GetLogicalVolume()->SetVisA << 115 else << 116 physVol->GetLogicalVolume()->SetVisA << 117 } << 118 } 81 } 119 physVol->GetLogicalVolume()->SetMaterial(m << 82 std::map<G4String,G4VisAttributes*>::const_iterator ite = fColours.find(mateName); 120 } << 83 if( ite != fColours.end() ){ >> 84 physVol->GetLogicalVolume()->SetVisAttributes( (*ite).second ); >> 85 } else { >> 86 physVol->GetLogicalVolume()->SetVisAttributes( (*(fColours.begin()) ).second ); // set it as unseen >> 87 } >> 88 } 121 89 122 return mate; 90 return mate; 123 } 91 } >> 92 124 93