File: VisualizePolyhedron.cpp

package info (click to toggle)
vecgeom 1.2.8%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 24,016 kB
  • sloc: cpp: 88,803; ansic: 6,888; python: 1,035; sh: 582; sql: 538; makefile: 23
file content (54 lines) | stat: -rw-r--r-- 1,735 bytes parent folder | download
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
#include "VecGeomTest/Visualizer.h"
#include "VecGeom/volumes/Polyhedron.h"
#include "VecGeom/volumes/utilities/VolumeUtilities.h"
#include "TPolyMarker3D.h"

using namespace vecgeom;

int main()
{
  constexpr int nSamples = 100000;

  Precision phiStart = kPi / 4., deltaPhi = kPi / 3. + kPi / 4.;
  constexpr int sides   = 8;
  constexpr int nPlanes = 6;

  Precision rmin[] = {0., 0., 3., 3., 0., 0.};
  Precision rmax[] = {3.5, 3.5, 3.5, 3.5, 3.5, 3.5};
  Precision z[]    = {-3., -2.5, -2.5, 2.5, 2.5, 3.};

  // Precision rmin[] = {0.,0.,0.,0.};
  // Precision rmax[] = {15.,15.,15.,10.};
  // Precision z[]    = {0.,20.,30.,40.};

  // Precision rmin[] = {0.,0.,0.,0.,0.,0.};
  // Precision rmax[] = {10.,20.,20.,10.,10.,5.};
  // Precision z[]    = {0.,20.,20.,40.,40.,50.};

  // Precision z[]    = {0.,20.,20.,40.,45.,50.,50.,60.};
  // Precision rmin[] = {0.,0.,0.,0.,0.,0.,0.,0.};
  // Precision rmax[] = {10.,20.,20.,10.,10.,5.,5.,20.};

  // Precision z[]    = {-2, -1, 1, 2};
  // Precision rmin[] = {0, 1, 0.5, 0};
  // Precision rmax[] = {1, 2, 2, 1};

  SimplePolyhedron polyhedron("Visualizer Polyhedron", phiStart, deltaPhi, sides, nPlanes, z, rmin, rmax);
  TPolyMarker3D pm(nSamples);
  pm.SetMarkerColor(kRed);
  pm.SetMarkerStyle(6);

  for (int i = 0; i < nSamples; ++i) {
    Vector3D<Precision> sample;
    sample = polyhedron.GetUnplacedVolume()
                 ->SamplePointOnSurface(); // volumeUtilities::SamplePoint(Vector3D<Precision>(20, 20, 20));
    assert(polyhedron.Inside(sample) == vecgeom::EInside::kSurface);
    pm.SetNextPoint(sample[0], sample[1], sample[2]);
  }

  Visualizer visualizer;
  visualizer.AddVolume(polyhedron);
  visualizer.AddPoints(pm);
  visualizer.Show();
  return 0;
}