Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/visualization/Vtk/src/G4VtkPolydataInstanceBakePipeline.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 #include "G4VtkPolydataInstanceBakePipeline.hh"
 27 
 28 #include "G4VtkViewer.hh"
 29 #include "G4VtkVisContext.hh"
 30 
 31 #include <vtkActor.h>
 32 #include <vtkDataArray.h>
 33 #include <vtkDoubleArray.h>
 34 #include <vtkPointData.h>
 35 #include <vtkPolyData.h>
 36 #include <vtkPolyDataMapper.h>
 37 #include <vtkProperty.h>
 38 
 39 std::size_t G4VtkPolydataInstanceBakePipeline::MakeHash(const G4Polyhedron& polyhedron,
 40                                                         const G4VtkVisContext& vc)
 41 {
 42   // Get view parameters that the user can force through the vis attributes, thereby over-riding the
 43   // current view parameter.
 44   const G4VisAttributes* pVA =
 45     vc.fViewer->GetApplicableVisAttributes(polyhedron.GetVisAttributes());
 46   G4Colour colour = pVA->GetColour();
 47 
 48   // Hash the vis attributes
 49   std::size_t hash = std::hash<G4double>{}(colour.GetAlpha());
 50   std::size_t shash = std::hash<G4double>{}(vc.fDrawingStyle);
 51 
 52   std::hash_combine(hash, shash);
 53 
 54   return hash;
 55 }
 56 
 57 G4VtkPolydataInstanceBakePipeline::G4VtkPolydataInstanceBakePipeline(G4String nameIn,
 58                                                                      const G4VtkVisContext& vcIn)
 59   : G4VtkPolydataInstancePipeline(nameIn, vcIn)
 60 {
 61   iVert = 0;
 62   iFace = 0;
 63   polydataPointData = vtkSmartPointer<vtkDoubleArray>::New();
 64   polydataPointData->SetNumberOfComponents(3);
 65   polydataPointData->SetName("Colors");
 66   polydata->GetPointData()->SetScalars(polydataPointData);
 67 
 68   mapper->SetColorModeToDirectScalars();
 69 
 70   actor->GetProperty()->SetOpacity(vc.alpha);
 71 }
 72 
 73 void G4VtkPolydataInstanceBakePipeline::SetPolydata(const G4Polyhedron& polyhedron)
 74 {
 75   polyhedronPrototype = &polyhedron;
 76 }
 77 
 78 void G4VtkPolydataInstanceBakePipeline::Print()
 79 {
 80   G4cout << "G4VtkPolydataInstanceBakePipeline " << GetName() << G4endl;
 81   G4VtkPolydataPipeline::Print();
 82 }
 83 
 84 void G4VtkPolydataInstanceBakePipeline::addInstance(G4double dx, G4double dy, G4double dz,
 85                                                     G4double r00, G4double r01, G4double r02,
 86                                                     G4double r10, G4double r11, G4double r12,
 87                                                     G4double r20, G4double r21, G4double r22,
 88                                                     G4double r, G4double g, G4double b, G4double a,
 89                                                     const G4String& nameIn)
 90 {
 91   G4VtkPolydataInstancePipeline::addInstance(dx, dy, dz, r00, r01, r02, r10, r11, r12, r20, r21,
 92                                              r22, r, g, b, a, nameIn);
 93 
 94   vtkIdType vStart;
 95   vtkIdType vEnd;
 96   vtkIdType fStart;
 97   vtkIdType fEnd;
 98 
 99   vStart = iVert;
100   fStart = iFace;
101 
102   G4bool notLastFace;
103   do {
104     G4Point3D vertex[4];
105     G4int edgeFlag[4];
106     G4Normal3D normals[4];
107     G4int nEdges;
108     notLastFace = polyhedronPrototype->GetNextFacet(nEdges, vertex, edgeFlag, normals);
109 
110     vtkSmartPointer<vtkIdList> poly = vtkSmartPointer<vtkIdList>::New();
111     // loop over vertices
112     for (int i = 0; i < nEdges; i++) {
113       // note : G4Transform3D does not have a matrix element constructor
114       G4Point3D bakedVertex =
115         G4Point3D(vertex[i].x() * r00 + vertex[i].y() * r01 + vertex[i].z() * r02 + dx,
116                   vertex[i].x() * r10 + vertex[i].y() * r11 + vertex[i].z() * r12 + dy,
117                   vertex[i].x() * r20 + vertex[i].y() * r21 + vertex[i].z() * r22 + dz);
118       polydataPoints->InsertNextPoint(bakedVertex.x(), bakedVertex.y(), bakedVertex.z());
119       poly->InsertNextId(iVert);
120       iVert++;
121       double cols[] = {r, g, b, 0.5};
122       polydataPointData->InsertNextTuple(cols);
123     }
124 
125     polydataCells->InsertNextCell(poly);
126     iFace++;
127 
128   } while (notLastFace);
129 
130   vEnd = iVert;
131   fEnd = iFace;
132 
133   // Add vertex and face to maps to allow for removal later
134   instanceVertexMap[name] = std::pair<vtkIdType, vtkIdType>(vStart, vEnd);
135   instanceFaceMap[name] = std::pair<vtkIdType, vtkIdType>(fStart, fEnd);
136 }
137 
138 void G4VtkPolydataInstanceBakePipeline::removeInstance(const G4String& /*name*/) {}
139