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
|
/// \file BVHSafetyEstimator.h
/// \author Guilherme Amadio
#ifndef VECGEOM_NAVIGATION_BVHSAFETYESTIMATOR_H_
#define VECGEOM_NAVIGATION_BVHSAFETYESTIMATOR_H_
#include "VecGeom/management/BVHManager.h"
#include "VecGeom/navigation/VSafetyEstimator.h"
namespace vecgeom {
inline namespace VECGEOM_IMPL_NAMESPACE {
/**
* @brief Safety estimator class using the bounding volume hierarchy of each
* logical volume for acceleration.
*/
class BVHSafetyEstimator : public VSafetyEstimatorHelper<BVHSafetyEstimator> {
private:
/** Constructor. Private since this is a singleton class accessed only via the @c Instance() static method. */
VECCORE_ATT_DEVICE
BVHSafetyEstimator() : VSafetyEstimatorHelper<BVHSafetyEstimator>() {}
public:
static constexpr const char *gClassNameString = "BVHSafetyEstimator";
#ifndef VECCORE_CUDA
/** Return instance of this singleton class. */
static VSafetyEstimator *Instance()
{
static BVHSafetyEstimator instance;
return &instance;
}
#else
// If used on device, this needs to be implemented in a .cu file rather than in this header
// This hack is used also by NewSimpleNavigator, implemented in LogicalVolume.cpp
// This is now implemented in BVHManager.cu
VECCORE_ATT_DEVICE
static VSafetyEstimator *Instance();
#endif
/**
* Compute safety of a point given in the local coordinates of the placed volume @p pvol.
* @param[in] localpoint Point in the local coordinates of the placed volume.
* @param[in] pvol Placed volume.
*/
VECCORE_ATT_HOST_DEVICE
Precision ComputeSafetyForLocalPoint(Vector3D<Precision> const &localpoint, VPlacedVolume const *pvol) const final
{
Precision safety = pvol->SafetyToOut(localpoint);
if (safety > 0.0 && pvol->GetDaughters().size() > 0)
safety = BVHManager::GetBVH(pvol->GetLogicalVolume())->ComputeSafety(localpoint, safety);
return safety;
}
/**
* Compute safety of a point given in the local coordinates of the logical volume @p lvol against
* all its child volumes. Uses the bounding volume hierarchy associated with the logical volume
* for acceleration.
* @param[in] localpoint Point in the local coordinates of the placed volume.
* @param[in] lvol Logical volume.
*/
VECCORE_ATT_HOST_DEVICE
Precision ComputeSafetyToDaughtersForLocalPoint(Vector3D<Precision> const &localpoint,
LogicalVolume const *lvol) const final
{
return BVHManager::GetBVH(lvol)->ComputeSafety(localpoint, kInfLength);
}
/**
* Compute safety of a point given in the local coordinates of the placed volume @p pvol.
* @param[in] localpoint Points in SIMD layout in the local coordinates of the placed volume.
* @param[in] pvol Placed volume.
* @param[in] m Mask of active SIMD lanes.
*/
VECCORE_ATT_HOST_DEVICE
Real_v ComputeSafetyForLocalPoint(Vector3D<Real_v> const &localpoint, VPlacedVolume const *pvol, Bool_v m) const final
{
using vecCore::Get;
using vecCore::Set;
using vecCore::VectorSize;
Real_v safeties(kInfLength);
for (size_t i = 0; i < VectorSize<Real_v>(); ++i) {
if (Get(m, i) == true) {
Vector3D<Precision> point(Get(localpoint[0], i), Get(localpoint[1], i), Get(localpoint[2], i));
Set(safeties, i, ComputeSafetyForLocalPoint(point, pvol));
}
}
return safeties;
}
/**
* Vector interface to compute safety for a set of points given in the local coordinates of the placed volume @p pvol.
* @param[in] localpoints Points in the local coordinates of the placed volume.
* @param[in] pvol Placed volume.
* @param[out] safeties Output safeties.
*/
void ComputeSafetyForLocalPoints(SOA3D<Precision> const &localpoints, VPlacedVolume const *pvol,
Precision *safeties) const final
{
for (size_t i = 0; i < localpoints.size(); ++i)
safeties[i] = ComputeSafetyForLocalPoint({localpoints.x(i), localpoints.y(i), localpoints.z(i)}, pvol);
}
};
} // namespace VECGEOM_IMPL_NAMESPACE
} // namespace vecgeom
#endif
|
