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