Geant4 Cross Reference |
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 "G4VtkPolydataInstanceAppendPipeline.hh" 27 28 #include "G4VtkViewer.hh" 29 #include "G4VtkVisContext.hh" 30 31 #include <vtkActor.h> 32 #include <vtkAppendPolyData.h> 33 #include <vtkDataArray.h> 34 #include <vtkDoubleArray.h> 35 #include <vtkGeneralTransform.h> 36 #include <vtkPointData.h> 37 #include <vtkPolyData.h> 38 #include <vtkPolyDataMapper.h> 39 #include <vtkProperty.h> 40 #include <vtkTransformPolyDataFilter.h> 41 42 std::size_t G4VtkPolydataInstanceAppendPipeline::MakeHash(const G4Polyhedron& polyhedron, 43 const G4VtkVisContext& vc) 44 { 45 // Get view parameters that the user can force through the vis attributes, thereby over-riding the 46 // current view parameter. 47 const G4VisAttributes* pVA = 48 vc.fViewer->GetApplicableVisAttributes(polyhedron.GetVisAttributes()); 49 G4Color colour = pVA->GetColour(); 50 51 // Hash the vis attributes 52 std::size_t hash = std::hash<G4double>{}(colour.GetAlpha()); 53 std::size_t rhash = std::hash<G4double>{}(colour.GetRed()); 54 std::size_t ghash = std::hash<G4double>{}(colour.GetGreen()); 55 std::size_t bhash = std::hash<G4double>{}(colour.GetBlue()); 56 std::size_t phash = std::hash<G4Polyhedron>{}(polyhedron); 57 std::size_t shash = std::hash<G4double>{}(vc.fDrawingStyle); 58 59 std::hash_combine(hash, phash); 60 std::hash_combine(hash, rhash); 61 std::hash_combine(hash, bhash); 62 std::hash_combine(hash, ghash); 63 std::hash_combine(hash, shash); 64 65 return hash; 66 } 67 68 G4VtkPolydataInstanceAppendPipeline::G4VtkPolydataInstanceAppendPipeline(G4String nameIn, 69 const G4VtkVisContext& vcIn) 70 : G4VtkPolydataInstancePipeline(nameIn, vcIn) 71 { 72 // Set pipeline type 73 SetTypeName(G4String("G4VtkPolydataInstanceAppendPipeline")); 74 75 // append filter 76 appendFilter = vtkSmartPointer<vtkAppendPolyData>::New(); 77 AddFilter(appendFilter); 78 79 // set polydata mapper 80 mapper->SetInputConnection(GetFinalFilter()->GetOutputPort()); 81 82 // set actor 83 actor->SetMapper(mapper); 84 actor->SetVisibility(1); 85 86 // colour parameters 87 actor->GetProperty()->SetOpacity(vc.alpha); 88 actor->GetProperty()->SetColor(vc.red, vc.green, vc.blue); 89 90 // shading parameters 91 actor->GetProperty()->SetAmbient(0.2); 92 actor->GetProperty()->SetDiffuse(0.7); 93 actor->GetProperty()->SetSpecular(0.1); 94 actor->GetProperty()->SetSpecularPower(1); 95 96 // add to renderer 97 vc.fViewer->renderer->AddActor(GetActor()); 98 } 99 100 void G4VtkPolydataInstanceAppendPipeline::Print() 101 { 102 G4cout << "G4VtkPolydataInstanceBakePipeline " << GetName() << G4endl; 103 G4VtkPolydataPipeline::Print(); 104 } 105 106 void G4VtkPolydataInstanceAppendPipeline::addInstance(G4double dx, G4double dy, G4double dz, 107 G4double r00, G4double r01, G4double r02, 108 G4double r10, G4double r11, G4double r12, 109 G4double r20, G4double r21, G4double r22, 110 G4double r, G4double g, G4double b, 111 G4double a, const G4String& nameIn) 112 { 113 // Add to base class 114 G4VtkPolydataInstancePipeline::addInstance(dx, dy, dz, r00, r01, r02, r10, r11, r12, r20, r21, 115 r22, r, g, b, a, nameIn); 116 117 // create transform 118 auto transform = vtkSmartPointer<vtkGeneralTransform>::New(); 119 double transformArray[16] = {r00, r01, r02, dx, r10, r11, r12, dy, 120 r20, r21, r22, dz, 0., 0., 0., 1.}; 121 transform->Concatenate(transformArray); 122 transform->Update(); 123 124 // Create transform filter and add to local filter map 125 vtkSmartPointer<vtkTransformPolyDataFilter> tf = 126 vtkSmartPointer<vtkTransformPolyDataFilter>::New(); 127 tf->SetTransform(transform); 128 tf->SetInputConnection(GetFilter(GetNumberOfFilters() - 2)->GetOutputPort()); 129 tf->Update(); 130 transformFilterMap[name] = tf; 131 132 // Add transform filter to append filter 133 GetFinalFilter()->AddInputConnection(tf->GetOutputPort()); 134 } 135 136 void G4VtkPolydataInstanceAppendPipeline::removeInstance(const G4String& nameIn) 137 { 138 // Remove from base class 139 G4VtkPolydataInstancePipeline::removeInstance(nameIn); 140 141 // Remove transform filter from append filter 142 appendFilter->RemoveInputConnection(0, transformFilterMap[name]->GetOutputPort()); 143 144 // Remove from local filter map 145 transformFilterMap.erase(nameIn); 146 } 147