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
|
// Copyright 2009 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
#pragma once
#include <thread>
#include "MultiDeviceFrameBuffer.h"
// ospray
#include "common/Future.h"
#include "fb/FrameBuffer.h"
namespace ospray {
namespace api {
struct MultiDeviceRenderTask : public Future
{
MultiDeviceRenderTask(MultiDeviceFrameBuffer *fb, std::function<float()> fcn);
~MultiDeviceRenderTask() override;
bool isFinished(OSPSyncEvent event = OSP_TASK_FINISHED) override;
void wait(OSPSyncEvent event) override;
void cancel() override;
float getProgress() override;
float getTaskDuration() override;
private:
Ref<MultiDeviceFrameBuffer> fb;
std::atomic<float> taskDuration{0.f};
std::atomic<bool> finished;
std::thread thread;
};
// Inlined definitions //////////////////////////////////////////////////////
inline MultiDeviceRenderTask::MultiDeviceRenderTask(
MultiDeviceFrameBuffer *fb, std::function<float()> fcn)
: fb(fb), finished(false)
{
thread = std::thread([this, fcn]() {
taskDuration = fcn();
finished = true;
});
}
inline MultiDeviceRenderTask::~MultiDeviceRenderTask()
{
if (thread.joinable()) {
thread.join();
}
}
inline bool MultiDeviceRenderTask::isFinished(OSPSyncEvent)
{
return finished == true;
}
inline void MultiDeviceRenderTask::wait(OSPSyncEvent)
{
if (thread.joinable()) {
thread.join();
}
}
inline void MultiDeviceRenderTask::cancel()
{
for (size_t i = 0; i < fb->objects.size(); ++i) {
FrameBuffer *fbi = (FrameBuffer *)fb->objects[i];
fbi->cancelFrame();
}
}
inline float MultiDeviceRenderTask::getProgress()
{
// The frame is done when all the subdevices are done, so the total progress
// works out to be the average
float progress = 0.0;
for (size_t i = 0; i < fb->objects.size(); ++i) {
FrameBuffer *fbi = (FrameBuffer *)fb->objects[i];
progress += fbi->getCurrentProgress();
}
return progress / fb->objects.size();
}
inline float MultiDeviceRenderTask::getTaskDuration()
{
return taskDuration.load();
}
} // namespace api
} // namespace ospray
|
