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 "G4VtkPolydataPolygonPipeline.hh" 27 28 #include "G4VisAttributes.hh" 29 #include "G4VtkViewer.hh" 30 #include "G4VtkVisContext.hh" 31 32 #include <vtkActor.h> 33 #include <vtkPolyDataMapper.h> 34 #include <vtkProperty.h> 35 #include <vtkRegularPolygonSource.h> 36 #include <vtkVertexGlyphFilter.h> 37 38 std::size_t G4VtkPolydataPolygonPipeline::MakeHash(const G4VisAttributes* pVA) 39 { 40 return std::hash<G4VisAttributes>{}(*pVA); 41 } 42 43 G4VtkPolydataPolygonPipeline::G4VtkPolydataPolygonPipeline(G4String nameIn, 44 const G4VtkVisContext& vcIn, 45 const G4VisAttributes* pVisAttributes) 46 : G4VtkPolydataPipeline(nameIn, vcIn) 47 { 48 // Set pipeline type 49 SetTypeName(G4String("G4VtkPolydataSpherePipeline")); 50 51 // Get vis attributes 52 G4Color colour = pVisAttributes->GetColour(); 53 G4double opacity = colour.GetAlpha(); 54 G4double lineWidth = pVisAttributes->GetLineWidth(); 55 56 // Draw in world coordinates. 57 vtkSmartPointer<vtkRegularPolygonSource> polygonSource = 58 vtkSmartPointer<vtkRegularPolygonSource>::New(); 59 polygonSource->SetNumberOfSides(4); 60 polygonSource->SetRadius(vc.fSize); 61 62 // vertex glyph filter 63 auto glyphFilter = vtkSmartPointer<vtkVertexGlyphFilter>::New(); 64 glyphFilter->SetInputData(polydata); 65 AddFilter(glyphFilter); 66 67 // Setup actor and mapper 68 mapper->SetInputConnection(glyphFilter->GetOutputPort()); 69 70 GetActor()->GetProperty()->SetLineWidth(lineWidth); 71 GetActor()->GetProperty()->SetColor(colour.GetRed(), colour.GetGreen(), colour.GetBlue()); 72 GetActor()->GetProperty()->SetOpacity(opacity); 73 GetActor()->SetVisibility(1); 74 75 GetActor()->GetProperty()->SetRenderPointsAsSpheres(true); 76 GetActor()->GetProperty()->SetPointSize(vc.fSize * 5); 77 78 vc.fViewer->renderer->AddActor(GetActor()); 79 } 80