CaveInteraction/Evaluation.cxx at main · GallAlex/CaveInteraction · GitHub

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
#include "Evaluation.h"
#include <vtkSphereSource.h>
#include <vtkConeSource.h>
#include <vtkPolyDataMapper.h>
#include <vtkActor.h>
#include <vtkRenderer.h>
#include <vtkOpenXRRenderer.h>
#include <vtkOpenXRRenderWindow.h>
#include <vtkOpenXRRenderWindowInteractor.h>
#include <vtkProperty.h>
#include <cstdlib>
#include <ctime>
#include "CaveInteractorStyle.h"

Evaluation::Evaluation() {}
Evaluation::~Evaluation() {}

void Evaluation::CreateScene()
{
    srand(static_cast<unsigned int>(time(nullptr)));

    vtkNew<vtkOpenXRRenderWindow> xrRenderWindow;
    xrRenderWindow->Initialize();

    vtkNew<vtkOpenXRRenderer> xrRenderer;
    xrRenderWindow->AddRenderer(xrRenderer);

    vtkNew<vtkOpenXRRenderWindowInteractor> xrInteractor;
    vtkNew<CaveInteractorStyle> style;
    xrInteractor->SetInteractorStyle(style);
    style->SetRendererAndWindow(xrRenderer, xrRenderWindow);
    xrInteractor->SetRenderWindow(xrRenderWindow);

    for (int i = 0; i < 20; i++)
    {
        vtkNew<vtkSphereSource> sphereSource;
        sphereSource->SetRadius(3.0); // Radius

        vtkNew<vtkPolyDataMapper> mapper;
        mapper->SetInputConnection(sphereSource->GetOutputPort());

        vtkNew<vtkActor> actor;
        actor->SetMapper(mapper);

        // Set random position for the spheres
        double x = static_cast<double>(rand() % 40 - 20);
        double y = static_cast<double>(rand() % 40 - 20);
        double z = static_cast<double>(rand() % 40 - 20);
        actor->SetPosition(x, y, z);

        xrRenderer->AddActor(actor);
    }

    // Add one occluded sphere with different colour
    vtkNew<vtkSphereSource> yellowSphere;
    yellowSphere->SetRadius(3.0);

    vtkNew<vtkPolyDataMapper> shMapper;
    shMapper->SetInputConnection(yellowSphere->GetOutputPort());

    vtkNew<vtkActor> yellowActor;
    yellowActor->SetMapper(shMapper);

    vtkNew<vtkProperty> shProperty;
    shProperty->SetColor(1.0, 0.65, 0.0);
    yellowActor->SetProperty(shProperty);

    yellowActor->SetPosition(0, 0, -15);
    xrRenderer->AddActor(yellowActor);

    AddOccludingSphere(xrRenderer, 0, 0, -10);
    AddOccludingSphere(xrRenderer, -5, 0, -12);

    xrRenderer->SetBackground(0.1, 0.2, 0.3);

    xrInteractor->Start();
}

void Evaluation::AddOccludingSphere(vtkRenderer* renderer, double x, double y, double z)
{
    vtkNew<vtkSphereSource> sphereSource;
    sphereSource->SetRadius(3.0);

    vtkNew<vtkPolyDataMapper> mapper;
    mapper->SetInputConnection(sphereSource->GetOutputPort());

    vtkNew<vtkActor> actor;
    actor->SetMapper(mapper);
    actor->SetPosition(x, y, z);

    renderer->AddActor(actor);
}