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
|
#ifndef VECGEOM_VOLUMES_UNPLACEDBOX_H_
#define VECGEOM_VOLUMES_UNPLACEDBOX_H_
#include "VecGeom/base/Cuda.h"
#include "VecGeom/base/Global.h"
#include "VecGeom/base/AlignedBase.h"
#include "VecGeom/base/Vector3D.h"
#include "VecGeom/volumes/UnplacedVolume.h"
#include "VecGeom/volumes/BoxStruct.h" // the pure box struct
#include "VecGeom/volumes/kernel/BoxImplementation.h"
#include "VecGeom/volumes/UnplacedVolumeImplHelper.h"
namespace vecgeom {
VECGEOM_DEVICE_FORWARD_DECLARE(class UnplacedBox;);
VECGEOM_DEVICE_DECLARE_CONV(class, UnplacedBox);
inline namespace VECGEOM_IMPL_NAMESPACE {
class UnplacedBox : public SIMDUnplacedVolumeImplHelper<BoxImplementation>, public AlignedBase {
private:
BoxStruct<Precision> fBox;
public:
using Kernel = BoxImplementation;
UnplacedBox(Vector3D<Precision> const &dim) : fBox(dim) { ComputeBBox(); }
UnplacedBox(char const *, Vector3D<Precision> const &dim) : fBox(dim) { ComputeBBox(); }
VECCORE_ATT_HOST_DEVICE
UnplacedBox(const Precision dx, const Precision dy, const Precision dz) : fBox(dx, dy, dz)
{
fGlobalConvexity = true;
ComputeBBox();
}
UnplacedBox(char const *, const Precision dx, const Precision dy, const Precision dz) : fBox(dx, dy, dz)
{
fGlobalConvexity = true;
ComputeBBox();
}
VECCORE_ATT_HOST_DEVICE
BoxStruct<Precision> const &GetStruct() const { return fBox; }
VECCORE_ATT_HOST_DEVICE
VECGEOM_FORCE_INLINE
Vector3D<Precision> const &dimensions() const { return fBox.fDimensions; }
VECCORE_ATT_HOST_DEVICE
VECGEOM_FORCE_INLINE
Precision x() const { return dimensions().x(); }
VECCORE_ATT_HOST_DEVICE
VECGEOM_FORCE_INLINE
Precision y() const { return dimensions().y(); }
VECCORE_ATT_HOST_DEVICE
VECGEOM_FORCE_INLINE
Precision z() const { return dimensions().z(); }
void SetX(Precision xx) { fBox.fDimensions[0] = xx; }
void SetY(Precision yy) { fBox.fDimensions[1] = yy; }
void SetZ(Precision zz) { fBox.fDimensions[2] = zz; }
Precision Capacity() const override { return 8.0 * x() * y() * z(); }
// VECCORE_ATT_HOST_DEVICE
Precision SurfaceArea() const override
{
// factor 8 because x(),... are half-lengths
return 8.0 * (x() * y() + y() * z() + x() * z());
}
VECCORE_ATT_HOST_DEVICE
void Extent(Vector3D<Precision> &aMin, Vector3D<Precision> &aMax) const override
{
// Returns the full 3D cartesian extent of the volume
aMin = -fBox.fDimensions;
aMax = fBox.fDimensions;
}
Vector3D<Precision> SamplePointOnSurface() const override;
VECCORE_ATT_HOST_DEVICE
virtual bool Normal(Vector3D<Precision> const &p, Vector3D<Precision> &normal) const override
{
bool valid;
normal = BoxImplementation::NormalKernel(fBox, p, valid);
return valid;
}
VECCORE_ATT_HOST_DEVICE
virtual void Print() const override;
virtual void Print(std::ostream &os) const override;
#ifndef VECCORE_CUDA
virtual SolidMesh *CreateMesh3D(Transformation3D const &trans, size_t nSegments) const override;
#endif
#ifdef VECGEOM_CUDA_INTERFACE
virtual size_t DeviceSizeOf() const override { return DevicePtr<cuda::UnplacedBox>::SizeOf(); }
virtual DevicePtr<cuda::VUnplacedVolume> CopyToGpu() const override;
virtual DevicePtr<cuda::VUnplacedVolume> CopyToGpu(DevicePtr<cuda::VUnplacedVolume> const gpu_ptr) const override;
#endif
#ifndef VECCORE_CUDA
// this is the function called from the VolumeFactory
// this may be specific to the shape
template <TranslationCode trans_code, RotationCode rot_code>
static VPlacedVolume *Create(LogicalVolume const *const logical_volume, Transformation3D const *const transformation,
VPlacedVolume *const placement = NULL);
VPlacedVolume *SpecializedVolume(LogicalVolume const *const volume, Transformation3D const *const transformation,
const TranslationCode trans_code, const RotationCode rot_code,
VPlacedVolume *const placement) const override;
#else
template <TranslationCode trans_code, RotationCode rot_code>
VECCORE_ATT_DEVICE
static VPlacedVolume *Create(LogicalVolume const *const logical_volume, Transformation3D const *const transformation,
const int id, const int copy_no, const int child_id,
VPlacedVolume *const placement = NULL);
VECCORE_ATT_DEVICE VPlacedVolume *SpecializedVolume(LogicalVolume const *const volume,
Transformation3D const *const transformation,
const TranslationCode trans_code, const RotationCode rot_code,
const int id, const int copy_no, const int child_id,
VPlacedVolume *const placement) const override;
#endif
};
} // namespace VECGEOM_IMPL_NAMESPACE
} // namespace vecgeom
#endif
|
