| 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 #ifndef G4VTKPOLYDATAPIPELINE_HH
27 #define G4VTKPOLYDATAPIPELINE_HH
28
29 #include "G4Point3D.hh"
30 #include "G4VVtkPipeline.hh"
31
32 #include <vtkSmartPointer.h>
33
34 #include <vector>
35
36 class G4String;
37 class G4VtkVisContext;
38 class G4Polyhedron;
39 class G4Polyline;
40
41 class vtkPoints;
42 class vtkCellArray;
43 class vtkPolyData;
44 class vtkPolyDataAlgorithm;
45 class vtkPolyDataMapper;
46 class vtkActor;
47
48 class G4VtkPolydataPipeline : public G4VVtkPipeline
49 {
50 public:
51 G4VtkPolydataPipeline(G4String name, const G4VtkVisContext& vc);
52 ~G4VtkPolydataPipeline() override = default;
53
54 void AddFilter(vtkSmartPointer<vtkPolyDataAlgorithm> f) { filters.push_back(f); }
55 vtkSmartPointer<vtkPolyDataAlgorithm> GetFilter(G4int iFilter) { return filters[iFilter]; }
56 G4int GetNumberOfFilters() { return (G4int)filters.size(); }
57 vtkSmartPointer<vtkPolyDataAlgorithm> GetFinalFilter() { return filters[filters.size() - 1]; }
58
59 void Enable() override;
60 void Disable() override;
61
62 void Print() override;
63 void Modified() override;
64 void Clear() override;
65
66 virtual vtkSmartPointer<vtkActor> GetActor() { return actor; }
67
68 virtual void SetPolydata(const G4Polyhedron& polyhedron);
69 virtual void SetPolydata(const G4Polyline& polyline);
70 virtual void SetPolydata(vtkPolyData *polydata);
71 virtual void SetPolydataData(const G4Point3D& p);
72 virtual void SetPolydataData(double x, double y, double z);
73
74 virtual void SetActorTransform(G4double dx, G4double dy, G4double dz, G4double r00,
75 G4double r01, G4double r02, G4double r10, G4double r11,
76 G4double r12, G4double r20, G4double r21, G4double r22);
77 virtual void SetActorColour(G4double r, G4double g, G4double b, G4double a);
78
79 virtual vtkSmartPointer<vtkPolyData> GetPolydata() { return polydata; }
80
81 virtual G4double* GetBounds() { return actor->GetBounds(); }
82
83 static std::size_t MakeHash(const G4Polyhedron& p, const G4VtkVisContext& vc);
84
85 protected:
86 vtkSmartPointer<vtkPoints> polydataPoints;
87 vtkSmartPointer<vtkCellArray> polydataCells;
88 vtkSmartPointer<vtkPolyData> polydata;
89
90 std::vector<vtkSmartPointer<vtkPolyDataAlgorithm>>
91 filters; // derived types can store filters in this vector
92 vtkSmartPointer<vtkPolyDataMapper> mapper; // polydagta mapper
93 vtkSmartPointer<vtkActor> actor; // all pipelines require an actor
94 };
95
96 #endif // G4VTKPOLYDATAPIPELINE_HH
97