forked from sPHENIX-Collaboration/acts
/
GeometryView3D.hpp
184 lines (167 loc) · 8.11 KB
/
GeometryView3D.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
// This file is part of the Acts project.
//
// Copyright (C) 2020 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
#pragma once
#include "Acts/Geometry/AbstractVolume.hpp"
#include "Acts/Geometry/GeometryContext.hpp"
#include "Acts/Utilities/Definitions.hpp"
#include "Acts/Visualization/IVisualization3D.hpp"
#include "Acts/Visualization/ViewConfig.hpp"
namespace Acts {
class Layer;
class Surface;
class SurfaceArray;
class TrackingVolume;
struct Polyhedron;
static ViewConfig s_viewSensitive = ViewConfig({0, 180, 240});
static ViewConfig s_viewPassive = ViewConfig({240, 280, 0});
static ViewConfig s_viewVolume = ViewConfig({220, 220, 0});
static ViewConfig s_viewGrid = ViewConfig({220, 0, 0});
static ViewConfig s_viewLine = ViewConfig({0, 0, 220});
struct GeometryView3D {
/// Helper method to draw Polyhedron objects
///
/// @param helper [in, out] The visualization helper
/// @param Polyhedron The surface to be drawn
/// @param ViewConfig The drawing configuration
static void drawPolyhedron(IVisualization3D& helper,
const Polyhedron& polyhedron,
const ViewConfig& ViewConfig = s_viewVolume);
/// Helper method to draw Surface objects
///
/// @param helper [in, out] The visualization helper
/// @param surface The surface to be drawn
/// @param gctx The geometry context for which it is drawn
/// @param transform An option additional transform
/// @param ViewConfig The drawing configuration
static void drawSurface(
IVisualization3D& helper, const Surface& surface,
const GeometryContext& gctx,
const Transform3D& transform = Transform3D::Identity(),
const ViewConfig& ViewConfig = s_viewSensitive);
/// Helper method to draw SurfaceArray objects
///
/// @param helper [in, out] The visualization helper
/// @param surfaceArray The surface to be drawn
/// @param gctx The geometry context for which it is drawn
/// @param transform An option additional transform
/// @param sensitiveConfig The drawing configuration for sensitive surfaces
/// @param passiveConfig The drawing configuration for passive surfaces
/// @param gridCongig The drawing configuraiton for grid
static void drawSurfaceArray(
IVisualization3D& helper, const SurfaceArray& surfaceArray,
const GeometryContext& gctx,
const Transform3D& transform = Transform3D::Identity(),
const ViewConfig& sensitiveConfig = s_viewSensitive,
const ViewConfig& passiveConfig = s_viewPassive,
const ViewConfig& gridConfig = s_viewGrid);
/// Helper method to draw AbstractVolume objects
///
/// @param helper [in, out] The visualization helper
/// @param volume The volume to be drawn
/// @param gctx The geometry context for which it is drawn
/// @param transform An option additional transform
/// @param viewConfig The drawing configuration for boundary surfaces
static void drawVolume(IVisualization3D& helper, const AbstractVolume& volume,
const GeometryContext& gctx,
const Transform3D& transform = Transform3D::Identity(),
const ViewConfig& viewConfig = s_viewVolume);
/// Helper method to draw AbstractVolume objects
///
/// @param helper [in, out] The visualization helper
/// @param volume The tracking volume to be drawn
/// @param gctx The geometry context for which it is drawn
/// @param layerConfig The drawing configuration for passive surfaces
/// @param sensitiveConfig The drawing configuration for sensitive surfaces
/// @param gridConfig The drawing configuraiton for grid display
static void drawLayer(IVisualization3D& helper, const Layer& layer,
const GeometryContext& gctx,
const ViewConfig& layerConfig = s_viewPassive,
const ViewConfig& sensitiveConfig = s_viewSensitive,
const ViewConfig& gridConfig = s_viewGrid);
/// Helper method to draw AbstractVolume objects
///
/// @param helper [in, out] The visualization helper
/// @param volume The tracking volume to be drawn
/// @param gctx The geometry context for which it is drawn
/// @param containerView The drawing configuration for a container volume
/// @param volumeView The drawing configuration for the navigation level
/// volume
/// @param layerView The drawing configuration for passive surfaces
/// @param sensitiveView The drawing configuration for sensitive surfaces
/// @param gridView The drawing configuraiton for grid display
/// @param writeIt The prescription to write it or not
/// @param tag The (optional) additional output tag
static void drawTrackingVolume(
IVisualization3D& helper, const TrackingVolume& tVolume,
const GeometryContext& gctx,
const ViewConfig& containerView = s_viewVolume,
const ViewConfig& volumeView = s_viewVolume,
const ViewConfig& layerView = s_viewPassive,
const ViewConfig& sensitiveView = s_viewSensitive,
const ViewConfig& gridView = s_viewGrid, bool writeIt = true,
const std::string& tag = "");
/// Helper method to draw lines - base for all lines
///
/// @param helper [in, out] The visualization helper
/// @param start The start point
/// @param end The end point
/// @param arrows [ -1 | 0 | 1 | 2 ] = [ start | none | end | both ]
/// @param arrowLength wrt halflength
/// @param arrowWidth wrt thickness
/// @param ViewConfig The drawing configuration for this segement
static void drawSegmentBase(IVisualization3D& helper, const Vector3D& start,
const Vector3D& end, int arrows = 0,
double arrowLength = 0., double arrowWidth = 0.,
const ViewConfig& viewConfig = s_viewLine);
/// Convenience function : line
///
/// @param helper [in, out] The visualization helper
/// @param start The start point
/// @param end The end point
/// @param ViewConfig The drawing configuration for this segement
static void drawSegment(IVisualization3D& helper, const Vector3D& start,
const Vector3D& end,
const ViewConfig& viewConfig = s_viewLine);
/// Convenience function : arrow pointing back
///
/// @param helper [in, out] The visualization helper
/// @param start The start point
/// @param end The end point
/// @param arrowLength wrt thickness
/// @param arrowWidth wrt thickness
/// @param ViewConfig The drawing configuration for this segement
static void drawArrowBackward(IVisualization3D& helper, const Vector3D& start,
const Vector3D& end, double arrowLength,
double arrowWidth,
const ViewConfig& viewConfig = s_viewLine);
/// Convenience function : arrow pointing forwad
///
/// @param helper [in, out] The visualization helper
/// @param start The start point
/// @param end The end point
/// @param arrowLength wrt thickness
/// @param arrowWidth wrt thickness
/// @param ViewConfig The drawing configuration for this segement
static void drawArrowForward(IVisualization3D& helper, const Vector3D& start,
const Vector3D& end, double arrowLength,
double arrowWidth,
const ViewConfig& viewConfig = s_viewLine);
/// Convenience function : arrow pointing both directions
///
/// @param helper [in, out] The visualization helper
/// @param start The start point
/// @param end The end point
/// @param arrowLength wrt thickness
/// @param arrowWidth wrt thickness
/// @param ViewConfig The drawing configuration for this segement
static void drawArrowsBoth(IVisualization3D& helper, const Vector3D& start,
const Vector3D& end, double arrowLength,
double arrowWidth,
const ViewConfig& viewConfig = s_viewLine);
};
} // namespace Acts