| Geant4 Cross Reference |
1 // 1 //
2 // ******************************************* 2 // ********************************************************************
3 // * License and Disclaimer 3 // * License and Disclaimer *
4 // * 4 // * *
5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. 9 // * include a list of copyright holders. *
10 // * 10 // * *
11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. *
17 // * 17 // * *
18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. *
24 // ******************************************* 24 // ********************************************************************
25 25
26 #include "G4VtkPolydataPolyline2DPipeline.hh" 26 #include "G4VtkPolydataPolyline2DPipeline.hh"
27 27
28 #include "G4Polyline.hh" 28 #include "G4Polyline.hh"
29 #include "G4VisAttributes.hh" 29 #include "G4VisAttributes.hh"
30 #include "G4VtkViewer.hh" 30 #include "G4VtkViewer.hh"
31 #include "G4VtkVisContext.hh" 31 #include "G4VtkVisContext.hh"
32 32
33 #include <vtkActor.h> << 33 #include "vtkActor.h"
34 #include <vtkActor2D.h> << 34 #include "vtkActor2D.h"
35 #include <vtkCoordinate.h> << 35 #include "vtkCoordinate.h"
36 #include <vtkLine.h> << 36 #include "vtkLine.h"
37 #include <vtkPolyDataAlgorithm.h> << 37 #include "vtkPolyDataAlgorithm.h"
38 #include <vtkPolyDataMapper.h> << 38 #include "vtkPolyDataMapper.h"
39 #include <vtkPolyDataMapper2D.h> << 39 #include "vtkPolyDataMapper2D.h"
40 #include <vtkProperty.h> << 40 #include "vtkProperty.h"
41 #include <vtkProperty2D.h> << 41 #include "vtkProperty2D.h"
42 42
43 std::size_t G4VtkPolydataPolyline2DPipeline::M 43 std::size_t G4VtkPolydataPolyline2DPipeline::MakeHash(const G4VisAttributes* pVA)
44 { 44 {
45 return std::hash<G4VisAttributes>{}(*pVA); 45 return std::hash<G4VisAttributes>{}(*pVA);
46 } 46 }
47 47
48 G4VtkPolydataPolyline2DPipeline::G4VtkPolydata 48 G4VtkPolydataPolyline2DPipeline::G4VtkPolydataPolyline2DPipeline(
49 G4String nameIn, const G4VtkVisContext& vcIn 49 G4String nameIn, const G4VtkVisContext& vcIn, const G4VisAttributes* pVisAttributes)
50 : G4VtkPolydataPipeline(nameIn, vcIn) 50 : G4VtkPolydataPipeline(nameIn, vcIn)
51 { 51 {
52 // Set pipeline type 52 // Set pipeline type
53 SetTypeName(G4String("G4VtkPolydataPolyline2 53 SetTypeName(G4String("G4VtkPolydataPolyline2DPipeline"));
54 54
55 // Get vis attributes 55 // Get vis attributes
56 G4Color colour = pVisAttributes->GetColour() 56 G4Color colour = pVisAttributes->GetColour();
57 G4double opacity = colour.GetAlpha(); 57 G4double opacity = colour.GetAlpha();
58 G4double lineWidth = pVisAttributes->GetLine 58 G4double lineWidth = pVisAttributes->GetLineWidth();
59 59
60 // make 3d actor invisible and remove from r 60 // make 3d actor invisible and remove from renderer
61 GetActor()->SetVisibility(0); 61 GetActor()->SetVisibility(0);
62 vc.fViewer->renderer->RemoveActor(GetActor() 62 vc.fViewer->renderer->RemoveActor(GetActor());
63 63
64 // Setup 2D mapper 64 // Setup 2D mapper
65 mapper2D = vtkSmartPointer<vtkPolyDataMapper 65 mapper2D = vtkSmartPointer<vtkPolyDataMapper2D>::New();
66 mapper2D->SetInputConnection(GetFinalFilter( 66 mapper2D->SetInputConnection(GetFinalFilter()->GetOutputPort());
67 vtkSmartPointer<vtkCoordinate> coordinate = 67 vtkSmartPointer<vtkCoordinate> coordinate = vtkSmartPointer<vtkCoordinate>::New();
68 coordinate->SetCoordinateSystemToNormalizedV 68 coordinate->SetCoordinateSystemToNormalizedViewport();
69 mapper2D->SetTransformCoordinate(coordinate) 69 mapper2D->SetTransformCoordinate(coordinate);
70 70
71 // Setup 2D actor 71 // Setup 2D actor
72 actor2D = vtkSmartPointer<vtkActor2D>::New() 72 actor2D = vtkSmartPointer<vtkActor2D>::New();
73 GetActor2D()->GetProperty()->SetLineWidth(li 73 GetActor2D()->GetProperty()->SetLineWidth(lineWidth);
74 GetActor2D()->GetProperty()->SetColor(colour 74 GetActor2D()->GetProperty()->SetColor(colour.GetRed(), colour.GetGreen(), colour.GetBlue());
75 GetActor2D()->GetProperty()->SetOpacity(opac 75 GetActor2D()->GetProperty()->SetOpacity(opacity);
76 GetActor2D()->SetMapper(mapper2D); 76 GetActor2D()->SetMapper(mapper2D);
77 GetActor2D()->SetVisibility(1); 77 GetActor2D()->SetVisibility(1);
78 // GetActor2D()->GetPositionCoordinate()->Se 78 // GetActor2D()->GetPositionCoordinate()->SetCoordinateSystemToNormalizedDisplay();
79 79
80 // add to renderer 80 // add to renderer
81 vc.fViewer->renderer->AddActor(GetActor2D()) 81 vc.fViewer->renderer->AddActor(GetActor2D());
82 } 82 }
83 83
84 void G4VtkPolydataPolyline2DPipeline::Clear() 84 void G4VtkPolydataPolyline2DPipeline::Clear()
85 { 85 {
86 renderer->RemoveActor(GetActor2D()); 86 renderer->RemoveActor(GetActor2D());
87 G4VtkPolydataPipeline::Clear(); 87 G4VtkPolydataPipeline::Clear();
88 } 88 }
89 89
90 void G4VtkPolydataPolyline2DPipeline::Modified 90 void G4VtkPolydataPolyline2DPipeline::Modified()
91 { 91 {
92 G4VtkPolydataPipeline::Modified(); 92 G4VtkPolydataPipeline::Modified();
93 GetActor2D()->Modified(); 93 GetActor2D()->Modified();
94 } 94 }
95 95
96 void G4VtkPolydataPolyline2DPipeline::SetPolyd 96 void G4VtkPolydataPolyline2DPipeline::SetPolydata(const G4Polyline& polyline)
97 { 97 {
98 // Data data 98 // Data data
99 const size_t nLines = polyline.size(); 99 const size_t nLines = polyline.size();
100 100
101 for (size_t i = 0; i < nLines; ++i) { 101 for (size_t i = 0; i < nLines; ++i) {
102 auto id = 102 auto id =
103 polydataPoints->InsertNextPoint((polylin 103 polydataPoints->InsertNextPoint((polyline[i].x() + 1) / 2.0, (polyline[i].y() + 1) / 2.0, 0);
104 104
105 if (i < nLines - 1) { 105 if (i < nLines - 1) {
106 vtkSmartPointer<vtkLine> line = vtkSmart 106 vtkSmartPointer<vtkLine> line = vtkSmartPointer<vtkLine>::New();
107 line->GetPointIds()->SetId(0, id); 107 line->GetPointIds()->SetId(0, id);
108 line->GetPointIds()->SetId(1, id + 1); 108 line->GetPointIds()->SetId(1, id + 1);
109 polydataCells->InsertNextCell(line); 109 polydataCells->InsertNextCell(line);
110 } 110 }
111 } 111 }
112 } 112 }
113 113