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
|
#include "VecGeom/volumes/LogicalVolume.h"
#include "VecGeom/volumes/Box.h"
#include "VecGeom/volumes/Polyhedron.h"
#include "VecGeomTest/Benchmarker.h"
#include "VecGeom/management/GeoManager.h"
#include "ArgParser.h"
#include <fstream>
using namespace vecgeom;
UnplacedPolyhedron *NoInnerRadii()
{
constexpr int nPlanes = 5;
Precision zPlanes[nPlanes] = {-4, -2, 0, 2, 4};
Precision rInner[nPlanes] = {0, 0, 0, 0, 0};
Precision rOuter[nPlanes] = {2, 3, 2, 3, 2};
return new UnplacedPolyhedron(5, nPlanes, zPlanes, rInner, rOuter);
}
UnplacedPolyhedron *WithInnerRadii()
{
constexpr int nPlanes = 5;
Precision zPlanes[nPlanes] = {-4, -1, 0, 1, 4};
Precision rInner[nPlanes] = {1, 0.75, 0.5, 0.75, 1};
Precision rOuter[nPlanes] = {1.5, 1.5, 1.5, 1.5, 1.5};
return new UnplacedPolyhedron(5, nPlanes, zPlanes, rInner, rOuter);
}
UnplacedPolyhedron *WithPhiSectionConvex()
{
constexpr int nPlanes = 5;
Precision zPlanes[nPlanes] = {-4, -1, 0, 1, 4};
Precision rInner[nPlanes] = {1, 0.75, 0.5, 0.75, 1};
Precision rOuter[nPlanes] = {1.5, 1.5, 1.5, 1.5, 1.5};
return new UnplacedPolyhedron(15 * kDegToRad, 45 * kDegToRad, 5, nPlanes, zPlanes, rInner, rOuter);
}
UnplacedPolyhedron *WithPhiSectionNonConvex()
{
constexpr int nPlanes = 5;
Precision zPlanes[nPlanes] = {-4, -1, 0, 1, 4};
Precision rInner[nPlanes] = {1, 0.75, 0.5, 0.75, 1};
Precision rOuter[nPlanes] = {1.5, 1.5, 1.5, 1.5, 1.5};
return new UnplacedPolyhedron(15 * kDegToRad, 340 * kDegToRad, 5, nPlanes, zPlanes, rInner, rOuter);
}
UnplacedPolyhedron *ManySegments()
{
constexpr int nPlanes = 17;
Precision zPlanes[nPlanes] = {-8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5, 6, 7, 8};
Precision rInner[nPlanes] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1};
Precision rOuter[nPlanes] = {2, 2, 3, 3, 2, 2, 3, 3, 2, 2, 3, 3, 2, 2, 3, 3, 2};
return new UnplacedPolyhedron(6, nPlanes, zPlanes, rInner, rOuter);
}
UnplacedPolyhedron *SameZsection()
{
constexpr int nPlanes = 5;
Precision zPlanes[nPlanes] = {-2, -1, 1, 1, 2};
Precision rInner[nPlanes] = {0, 1, 0.5, 1, 0};
Precision rOuter[nPlanes] = {1, 2, 2, 2.5, 1};
return new UnplacedPolyhedron(15 * kDegToRad, 340 * kDegToRad, 5, nPlanes, zPlanes, rInner, rOuter);
};
int main(int argc, char *argv[])
{
OPTION_INT(npoints, 10000);
OPTION_INT(nrep, 4);
// Polyhedron type:
// 0=NoInnerRadii
// 1=WithInnerRadii
// 2=WithPhiSectionConvex
// 3=WithPhiSectionNonConvex
// 4=ManySegments
// 5=SameZsection
OPTION_INT(type, 3);
UnplacedBox worldUnplaced = UnplacedBox(5, 5, 10);
auto RunBenchmark = [&worldUnplaced](UnplacedPolyhedron const *shape, char const *label, int npoints,
int nrep) -> int {
LogicalVolume logical("pgon", shape);
LogicalVolume worldLogical(&worldUnplaced);
Transformation3D transformation(0, 0, 0);
worldLogical.PlaceDaughter("pgonplaced", &logical, &transformation);
GeoManager::Instance().SetWorldAndClose(worldLogical.Place());
Benchmarker benchmarker(GeoManager::Instance().GetWorld());
benchmarker.SetVerbosity(2);
benchmarker.SetPoolMultiplier(1);
benchmarker.SetRepetitions(nrep);
benchmarker.SetPointCount(npoints);
auto errcode = benchmarker.RunBenchmark();
std::list<BenchmarkResult> results = benchmarker.PopResults();
std::ofstream outStream;
outStream.open(label, std::fstream::app);
BenchmarkResult::WriteCsvHeader(outStream);
for (auto i = results.begin(), iEnd = results.end(); i != iEnd; ++i) {
i->WriteToCsv(outStream);
}
outStream.close();
return errcode;
};
std::cout << "________________________________________________________________________________\n";
switch (type) {
case 0:
std::cout << "Testing NoInnerRadii with npoints = " << npoints << " nrep = " << nrep << std::endl;
std::cout << "________________________________________________________________________________\n";
return RunBenchmark(NoInnerRadii(), "polyhedron_no-inner-radii.csv", npoints, nrep);
case 1:
std::cout << "Testing WithInnerRadii with npoints = " << npoints << " nrep = " << nrep << std::endl;
std::cout << "________________________________________________________________________________\n";
return RunBenchmark(WithInnerRadii(), "polyhedron_with-inner-radii.csv", npoints, nrep);
case 2:
std::cout << "Testing WithPhiSectionConvex with npoints = " << npoints << " nrep = " << nrep << std::endl;
std::cout << "________________________________________________________________________________\n";
return RunBenchmark(WithPhiSectionConvex(), "polyhedron_phi-section-conv.csv", npoints, nrep);
case 3:
std::cout << "Testing WithPhiSectionNonConvex with npoints = " << npoints << " nrep = " << nrep << std::endl;
std::cout << "________________________________________________________________________________\n";
return RunBenchmark(WithPhiSectionNonConvex(), "polyhedron_phi-section-non-conv.csv", npoints, nrep);
case 4:
std::cout << "Testing ManySegments with npoints = " << npoints << " nrep = " << nrep << std::endl;
std::cout << "________________________________________________________________________________\n";
return RunBenchmark(ManySegments(), "polyhedron_many-segments.csv", npoints, nrep);
case 5:
std::cout << "Testing SameZsection with npoints = " << npoints << " nrep = " << nrep << std::endl;
std::cout << "________________________________________________________________________________\n";
return RunBenchmark(SameZsection(), "polyhedron_sameZsection.csv", npoints, nrep);
default:
std::cout << "Unknown polyhedron type." << std::endl;
}
return 1;
}
|
