| 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 "G4VtkPolydataPolyline2DPipeline.hh"
27
28 #include "G4Polyline.hh"
29 #include "G4VisAttributes.hh"
30 #include "G4VtkViewer.hh"
31 #include "G4VtkVisContext.hh"
32
33 #include <vtkActor.h>
34 #include <vtkActor2D.h>
35 #include <vtkCoordinate.h>
36 #include <vtkLine.h>
37 #include <vtkPolyDataAlgorithm.h>
38 #include <vtkPolyDataMapper.h>
39 #include <vtkPolyDataMapper2D.h>
40 #include <vtkProperty.h>
41 #include <vtkProperty2D.h>
42
43 std::size_t G4VtkPolydataPolyline2DPipeline::MakeHash(const G4VisAttributes* pVA)
44 {
45 return std::hash<G4VisAttributes>{}(*pVA);
46 }
47
48 G4VtkPolydataPolyline2DPipeline::G4VtkPolydataPolyline2DPipeline(
49 G4String nameIn, const G4VtkVisContext& vcIn, const G4VisAttributes* pVisAttributes)
50 : G4VtkPolydataPipeline(nameIn, vcIn)
51 {
52 // Set pipeline type
53 SetTypeName(G4String("G4VtkPolydataPolyline2DPipeline"));
54
55 // Get vis attributes
56 G4Color colour = pVisAttributes->GetColour();
57 G4double opacity = colour.GetAlpha();
58 G4double lineWidth = pVisAttributes->GetLineWidth();
59
60 // make 3d actor invisible and remove from renderer
61 GetActor()->SetVisibility(0);
62 vc.fViewer->renderer->RemoveActor(GetActor());
63
64 // Setup 2D mapper
65 mapper2D = vtkSmartPointer<vtkPolyDataMapper2D>::New();
66 mapper2D->SetInputConnection(GetFinalFilter()->GetOutputPort());
67 vtkSmartPointer<vtkCoordinate> coordinate = vtkSmartPointer<vtkCoordinate>::New();
68 coordinate->SetCoordinateSystemToNormalizedViewport();
69 mapper2D->SetTransformCoordinate(coordinate);
70
71 // Setup 2D actor
72 actor2D = vtkSmartPointer<vtkActor2D>::New();
73 GetActor2D()->GetProperty()->SetLineWidth(lineWidth);
74 GetActor2D()->GetProperty()->SetColor(colour.GetRed(), colour.GetGreen(), colour.GetBlue());
75 GetActor2D()->GetProperty()->SetOpacity(opacity);
76 GetActor2D()->SetMapper(mapper2D);
77 GetActor2D()->SetVisibility(1);
78 // GetActor2D()->GetPositionCoordinate()->SetCoordinateSystemToNormalizedDisplay();
79
80 // add to renderer
81 vc.fViewer->renderer->AddActor(GetActor2D());
82 }
83
84 void G4VtkPolydataPolyline2DPipeline::Clear()
85 {
86 renderer->RemoveActor(GetActor2D());
87 G4VtkPolydataPipeline::Clear();
88 }
89
90 void G4VtkPolydataPolyline2DPipeline::Modified()
91 {
92 G4VtkPolydataPipeline::Modified();
93 GetActor2D()->Modified();
94 }
95
96 void G4VtkPolydataPolyline2DPipeline::SetPolydata(const G4Polyline& polyline)
97 {
98 // Data data
99 const size_t nLines = polyline.size();
100
101 for (size_t i = 0; i < nLines; ++i) {
102 auto id =
103 polydataPoints->InsertNextPoint((polyline[i].x() + 1) / 2.0, (polyline[i].y() + 1) / 2.0, 0);
104
105 if (i < nLines - 1) {
106 vtkSmartPointer<vtkLine> line = vtkSmartPointer<vtkLine>::New();
107 line->GetPointIds()->SetId(0, id);
108 line->GetPointIds()->SetId(1, id + 1);
109 polydataCells->InsertNextCell(line);
110 }
111 }
112 }
113