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
|
/// @file UnplacedGenericPolycone.h
/// @author Raman Sehgal (raman.sehgal@cern.ch)
#ifndef VECGEOM_VOLUMES_UNPLACEDGENERICPOLYCONE_H_
#define VECGEOM_VOLUMES_UNPLACEDGENERICPOLYCONE_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/GenericPolyconeStruct.h"
#include "VecGeom/volumes/kernel/GenericPolyconeImplementation.h"
#include "VecGeom/volumes/UnplacedVolumeImplHelper.h"
#include "VecGeom/volumes/ReducedPolycone.h"
namespace vecgeom {
VECGEOM_DEVICE_FORWARD_DECLARE(class UnplacedGenericPolycone;);
VECGEOM_DEVICE_DECLARE_CONV(class, UnplacedGenericPolycone);
inline namespace VECGEOM_IMPL_NAMESPACE {
class UnplacedGenericPolycone : public LoopUnplacedVolumeImplHelper<GenericPolyconeImplementation>, public AlignedBase {
private:
GenericPolyconeStruct<Precision> fGenericPolycone;
// Original Polycone Parameters
Precision fSPhi;
Precision fDPhi;
int fNumRZ;
Vector<Precision> fR;
Vector<Precision> fZ;
// Used for Extent
Vector3D<Precision> fAMin;
Vector3D<Precision> fAMax;
public:
/*
* All the required GenericPolycone Constructor
*
*/
VECCORE_ATT_HOST_DEVICE
UnplacedGenericPolycone();
VECCORE_ATT_HOST_DEVICE
UnplacedGenericPolycone(Precision phiStart, // initial phi starting angle
Precision phiTotal, // total phi angle
int numRZ, // number corners in r,z space (must be an even number)
Precision const *r, // r coordinate of these corners
Precision const *z); // z coordinate of these corners
VECCORE_ATT_HOST_DEVICE
GenericPolyconeStruct<Precision> const &GetStruct() const { return fGenericPolycone; }
/*
* Required Getters and Setters
*
*/
VECCORE_ATT_HOST_DEVICE
VECGEOM_FORCE_INLINE
Precision GetSPhi() const { return fSPhi; }
VECCORE_ATT_HOST_DEVICE
VECGEOM_FORCE_INLINE
Precision GetDPhi() const { return fDPhi; }
VECCORE_ATT_HOST_DEVICE
VECGEOM_FORCE_INLINE
int GetNumRz() const { return fNumRZ; }
VECCORE_ATT_HOST_DEVICE
VECGEOM_FORCE_INLINE
Vector<Precision> GetR() const { return fR; }
VECCORE_ATT_HOST_DEVICE
VECGEOM_FORCE_INLINE
Vector<Precision> GetZ() const { return fZ; }
VECCORE_ATT_HOST_DEVICE
void Extent(Vector3D<Precision> &, Vector3D<Precision> &) const override;
VECCORE_ATT_HOST_DEVICE
bool Normal(Vector3D<Precision> const &point, Vector3D<Precision> &norm) const override;
Precision Capacity() const override { return fGenericPolycone.fCubicVolume; }
// Using the Generic implementation from UnplacedVolume
Precision SurfaceArea() const override { return EstimateSurfaceArea(); }
Vector3D<Precision> SamplePointOnSurface() const override;
std::string GetEntityType() const { return "GenericPolycone"; }
public:
virtual int MemorySize() const final { return sizeof(*this); }
VECCORE_ATT_HOST_DEVICE
virtual void Print() const override;
virtual void Print(std::ostream &os) const override;
#ifdef VECGEOM_CUDA_INTERFACE
virtual size_t DeviceSizeOf() const override { return DevicePtr<cuda::UnplacedGenericPolycone>::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
};
using GenericUnplacedGenericPolycone = UnplacedGenericPolycone;
} // namespace VECGEOM_IMPL_NAMESPACE
} // namespace vecgeom
#endif
|
