blob: bb35880d37e49b032188c6c3eac3576e904fbc0f (
plain)
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
|
/* SPDX-License-Identifier: Apache-2.0
* Copyright 2011-2022 Blender Foundation */
#include "scene/tables.h"
#include "device/device.h"
#include "scene/scene.h"
#include "scene/stats.h"
#include "util/log.h"
#include "util/time.h"
CCL_NAMESPACE_BEGIN
/* Lookup Tables */
LookupTables::LookupTables()
{
need_update_ = true;
}
LookupTables::~LookupTables()
{
assert(lookup_tables.size() == 0);
}
void LookupTables::device_update(Device *, DeviceScene *dscene, Scene *scene)
{
if (!need_update())
return;
scoped_callback_timer timer([scene](double time) {
if (scene->update_stats) {
scene->update_stats->tables.times.add_entry({"device_update", time});
}
});
VLOG(1) << "Total " << lookup_tables.size() << " lookup tables.";
if (lookup_tables.size() > 0)
dscene->lookup_table.copy_to_device();
need_update_ = false;
}
void LookupTables::device_free(Device *, DeviceScene *dscene)
{
dscene->lookup_table.free();
}
bool LookupTables::need_update() const
{
return need_update_;
}
static size_t round_up_to_multiple(size_t size, size_t chunk)
{
return ((size + chunk - 1) / chunk) * chunk;
}
size_t LookupTables::add_table(DeviceScene *dscene, vector<float> &data)
{
assert(data.size() > 0);
need_update_ = true;
Table new_table;
new_table.offset = 0;
new_table.size = round_up_to_multiple(data.size(), TABLE_CHUNK_SIZE);
/* find space to put lookup table */
list<Table>::iterator table;
for (table = lookup_tables.begin(); table != lookup_tables.end(); table++) {
if (new_table.offset + new_table.size <= table->offset) {
lookup_tables.insert(table, new_table);
break;
}
else
new_table.offset = table->offset + table->size;
}
if (table == lookup_tables.end()) {
/* add at the end */
lookup_tables.push_back(new_table);
dscene->lookup_table.resize(new_table.offset + new_table.size);
}
/* copy table data and return offset */
float *dtable = dscene->lookup_table.data();
memcpy(dtable + new_table.offset, &data[0], sizeof(float) * data.size());
return new_table.offset;
}
void LookupTables::remove_table(size_t *offset)
{
if (*offset == TABLE_OFFSET_INVALID) {
/* The table isn't even allocated, so just return here. */
return;
}
need_update_ = true;
list<Table>::iterator table;
for (table = lookup_tables.begin(); table != lookup_tables.end(); table++) {
if (table->offset == *offset) {
lookup_tables.erase(table);
*offset = TABLE_OFFSET_INVALID;
return;
}
}
assert(table != lookup_tables.end());
}
CCL_NAMESPACE_END
|