9d04778abc68e2eb68636663ea37070f7caccdfa
[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 *device, DeviceScene *dscene)
39 {
40         if(!need_update)
41                 return;
42
43         VLOG(1) << "Total " << lookup_tables.size() << " lookup tables.";
44
45         device->tex_free(dscene->lookup_table);
46
47         if(lookup_tables.size() > 0)
48                 device->tex_alloc(dscene->lookup_table);
49
50         need_update = false;
51 }
52
53 void LookupTables::device_free(Device *device, DeviceScene *dscene)
54 {
55         device->tex_free(dscene->lookup_table);
56         dscene->lookup_table.clear();
57 }
58
59 static size_t round_up_to_multiple(size_t size, size_t chunk)
60 {
61         return ((size + chunk - 1)/chunk) * chunk;
62 }
63
64 size_t LookupTables::add_table(DeviceScene *dscene, vector<float>& data)
65 {
66         assert(data.size() > 0);
67
68         need_update = true;
69
70         Table new_table;
71         new_table.offset = 0;
72         new_table.size = round_up_to_multiple(data.size(), TABLE_CHUNK_SIZE);
73
74         /* find space to put lookup table */
75         list<Table>::iterator table;
76
77         for(table = lookup_tables.begin(); table != lookup_tables.end(); table++) {
78                 if(new_table.offset + new_table.size <= table->offset) {
79                         lookup_tables.insert(table, new_table);
80                         break;
81                 }
82                 else
83                         new_table.offset = table->offset + table->size;
84         }
85
86         if(table == lookup_tables.end()) {
87                 /* add at the end */
88                 lookup_tables.push_back(new_table);
89                 dscene->lookup_table.resize(new_table.offset + new_table.size);
90         }
91
92         /* copy table data and return offset */
93         float *dtable = dscene->lookup_table.get_data();
94         memcpy(dtable + new_table.offset, &data[0], sizeof(float) * data.size());
95
96         return new_table.offset;
97 }
98
99 void LookupTables::remove_table(size_t *offset)
100 {
101         if(*offset == TABLE_OFFSET_INVALID) {
102                 /* The table isn't even allocated, so just return here. */
103                 return;
104         }
105
106         need_update = true;
107
108         list<Table>::iterator table;
109
110         for(table = lookup_tables.begin(); table != lookup_tables.end(); table++) {
111                 if(table->offset == *offset) {
112                         lookup_tables.erase(table);
113                         *offset = TABLE_OFFSET_INVALID;
114                         return;
115                 }
116         }
117
118         assert(table != lookup_tables.end());
119 }
120
121 CCL_NAMESPACE_END
122