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
|
/// \file Visualizer.h
/// \author Johannes de Fine Licht (johannes.definelicht@cern.ch)
#ifndef VECGEOM_UTILITIES_VISUALIZER_H_
#define VECGEOM_UTILITIES_VISUALIZER_H_
#include "VecGeom/base/Config.h"
#ifndef VECGEOM_ROOT
#error "Visualizer currently only available with ROOT enabled."
#endif
#include "VecGeom/base/Global.h"
#include "VecGeom/volumes/PlacedVolume.h"
#include <list>
#include <memory>
#include <utility>
class TApplication;
class TGeoManager;
class TGeoMatrix;
class TGeoShape;
class TGeoVolume;
class TPolyLine3D;
#include "TPolyMarker3D.h"
namespace vecgeom {
inline namespace cxx {
template <typename T>
class AOS3D;
template <typename T>
class SOA3D;
class Transformation3D;
template <typename T>
class Vector3D;
/// \brief Visualize volumes through ROOT.
class Visualizer {
private:
int fVerbosity;
std::list<std::tuple<std::shared_ptr<const TGeoShape>, std::unique_ptr<TGeoMatrix>, std::unique_ptr<TGeoVolume>>>
fVolumes;
std::list<std::unique_ptr<TPolyMarker3D>> fMarkers;
std::list<std::unique_ptr<TPolyLine3D>> fLines;
TApplication *fApp; // ROOT application used for visualization
TGeoManager *fGeoManager; // ROOT geometry manager
public:
Visualizer();
~Visualizer();
TApplication *GetTApp() { return fApp; }
void AddVolume(VPlacedVolume const &volume);
void AddVolume(VPlacedVolume const &volume, Transformation3D const &transformation);
void AddVolume(std::shared_ptr<const TGeoShape> rootVolume);
void AddVolume(std::shared_ptr<const TGeoShape> rootVolume, Transformation3D const &position);
void AddPoint(Vector3D<Precision> const &point, int markercolor = kRed, int markerstyle = 5, int markersize = 1);
void AddPoints(AOS3D<Precision> const &points, int markercolor = kRed, int markerstyle = 5, int markersize = 1);
void AddPoints(SOA3D<Precision> const &points, int markercolor = kRed, int markerstyle = 5, int markersize = 1);
void AddPoints(TPolyMarker3D const &marker);
void AddLine(Vector3D<Precision> const &p0, Vector3D<Precision> const &p1);
void AddLine(TPolyLine3D const &line);
/// Runs a ROOT application, drawing the added volumes and points.
void Show() const;
/// Removes all previously added volumes and points.
void Clear();
/// \param level 0 = no output. 1 = reports when the visualizer is altered.
void SetVerbosity(int level);
private:
template <class ContainerType>
void AddPointsTemplate(ContainerType const &points, int markercolor = kRed, int markerstyle = 5, int markersize = 1);
};
} // namespace cxx
} // End namespace vecgeom
#endif // VECGEOM_MANAGEMENT_VISUALIZER_H_
|
