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
|
#include "../benchmark/ArgParser.h"
#include "ShapeTester.h"
#include "VecGeom/management/RootGeoManager.h"
#include "VecGeom/volumes/LogicalVolume.h"
#include "VecGeom/management/GeoManager.h"
#include "TGeoManager.h"
#include "TGeoVolume.h"
using namespace vecgeom;
// benchmarking any available shape (logical volume) found in a ROOT file
// usage: shape_testFromROOTFile detector.root logicalvolumename [option]
// logicalvolumename should not contain trailing pointer information
template <typename ImplT>
int runTester(ImplT const *shape, int npoints, bool debug, bool stat);
int main(int argc, char *argv[])
{
OPTION_INT(npoints, 10000);
OPTION_BOOL(debug, false);
OPTION_BOOL(stat, false);
if (argc < 3) {
std::cerr << "***** Error: need to give root geometry file and logical volume name\n";
std::cerr << "Ex: ./shape_testFromROOTFile ExN03.root World\n";
exit(1);
}
TGeoManager::Import(argv[1]);
std::string testvolume(argv[2]);
int errCode = 0;
int found = 0;
TGeoVolume *foundvolume = NULL;
// now try to find shape with logical volume name given on the command line
TObjArray *vlist = gGeoManager->GetListOfVolumes();
for (auto i = 0; i < vlist->GetEntries(); ++i) {
TGeoVolume *vol = reinterpret_cast<TGeoVolume *>(vlist->At(i));
std::string fullname(vol->GetName());
std::size_t founds = fullname.compare(testvolume);
if (founds == 0) {
found++;
foundvolume = vol;
std::cerr << "found matching volume " << fullname << " of type " << vol->GetShape()->ClassName() << "\n";
}
}
std::cerr << "volume found " << found << " times \n";
foundvolume->GetShape()->InspectShape();
std::cerr << "volume capacity " << foundvolume->GetShape()->Capacity() << "\n";
// now get the shape and benchmark it
if (!foundvolume) {
std::cerr << " NO SUCH VOLUME [" << testvolume << "] FOUND ... EXITING \n";
errCode = 2048; // errCode: 1000 0000 0000
return errCode;
}
LogicalVolume *converted = RootGeoManager::Instance().Convert(foundvolume);
VPlacedVolume *shape = converted->Place();
shape->Print();
return runTester<VPlacedVolume>(shape, npoints, debug, stat);
}
template <typename ImplT>
int runTester(ImplT const *shape, int npoints, bool debug, bool stat)
{
ShapeTester<ImplT> tester;
tester.setDebug(debug);
tester.setStat(stat);
tester.SetTestBoundaryErrors(false);
tester.SetMaxPoints(npoints);
int errcode = tester.Run(shape);
std::cout << "Final Error count for Shape *** " << shape->GetName() << "*** = " << errcode << "\n";
std::cout << "=========================================================\n";
if (shape) delete shape;
return errcode;
}
|
