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