651ae4e21a053ade0ee5b22a1089b4af0bf55cca
[blender-staging.git] / intern / cycles / render / tables.cpp
1 /*
2  * Copyright 2011-2013 Blender Foundation
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #include "device/device.h"
18 #include "render/scene.h"
19 #include "render/tables.h"
20
21 #include "util/util_debug.h"
22 #include "util/util_logging.h"
23
24 CCL_NAMESPACE_BEGIN
25
26 /* Lookup Tables */
27
28 LookupTables::LookupTables()
29 {
30         need_update = true;
31 }
32
33 LookupTables::~LookupTables()
34 {
35         assert(lookup_tables.size() == 0);
36 }
37
38 void LookupTables::device_update(Device *, DeviceScene *dscene)
39 {
40         if(!need_update)
41                 return;
42
43         VLOG(1) << "Total " << lookup_tables.size() << " lookup tables.";
44
45         if(lookup_tables.size() > 0)
46                 dscene->lookup_table.copy_to_device();
47
48         need_update = false;
49 }
50
51 void LookupTables::device_free(Device *, DeviceScene *dscene)
52 {
53         dscene->lookup_table.free();
54 }
55
56 static size_t round_up_to_multiple(size_t size, size_t chunk)
57 {
58         return ((size + chunk - 1)/chunk) * chunk;
59 }
60
61 size_t LookupTables::add_table(DeviceScene *dscene, vector<float>& data)
62 {
63         assert(data.size() > 0);
64
65         need_update = true;
66
67         Table new_table;
68         new_table.offset = 0;
69         new_table.size = round_up_to_multiple(data.size(), TABLE_CHUNK_SIZE);
70
71         /* find space to put lookup table */
72         list<Table>::iterator table;
73
74         for(table = lookup_tables.begin(); table != lookup_tables.end(); table++) {
75                 if(new_table.offset + new_table.size <= table->offset) {
76                         lookup_tables.insert(table, new_table);
77                         break;
78                 }
79                 else
80                         new_table.offset = table->offset + table->size;
81         }
82
83         if(table == lookup_tables.end()) {
84                 /* add at the end */
85                 lookup_tables.push_back(new_table);
86                 dscene->lookup_table.resize(new_table.offset + new_table.size);
87         }
88
89         /* copy table data and return offset */
90         float *dtable = dscene->lookup_table.data();
91         memcpy(dtable + new_table.offset, &data[0], sizeof(float) * data.size());
92
93         return new_table.offset;
94 }
95
96 void LookupTables::remove_table(size_t *offset)
97 {
98         if(*offset == TABLE_OFFSET_INVALID) {
99                 /* The table isn't even allocated, so just return here. */
100                 return;
101         }
102
103         need_update = true;
104
105         list<Table>::iterator table;
106
107         for(table = lookup_tables.begin(); table != lookup_tables.end(); table++) {
108                 if(table->offset == *offset) {
109                         lookup_tables.erase(table);
110                         *offset = TABLE_OFFSET_INVALID;
111                         return;
112                 }
113         }
114
115         assert(table != lookup_tables.end());
116 }
117
118 CCL_NAMESPACE_END
119