Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/examples/extended/medical/DICOM/src/DicomPhantomParameterisationColour.cc

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

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