Code refactor: move more memory allocation logic into device API.
[blender-staging.git] / intern / cycles / render / tables.cpp
index be0d4afbe2c728ccb00185338b5aa4a93d01a958..5cda977b7f1b53e0f537428057016bbd38bcc992 100644 (file)
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
- * limitations under the License
+ * limitations under the License.
  */
 
-#include "device.h"
-#include "scene.h"
-#include "tables.h"
+#include "device/device.h"
+#include "render/scene.h"
+#include "render/tables.h"
 
-#include "util_debug.h"
+#include "util/util_debug.h"
+#include "util/util_logging.h"
 
 CCL_NAMESPACE_BEGIN
 
@@ -34,20 +35,22 @@ LookupTables::~LookupTables()
        assert(lookup_tables.size() == 0);
 }
 
-void LookupTables::device_update(Device *device, DeviceScene *dscene)
+void LookupTables::device_update(Device *, DeviceScene *dscene)
 {
        if(!need_update)
                return;
 
-       device->tex_alloc("__lookup_table", dscene->lookup_table);
+       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 *device, DeviceScene *dscene)
+void LookupTables::device_free(Device *, DeviceScene *dscene)
 {
-       device->tex_free(dscene->lookup_table);
-       dscene->lookup_table.clear();
+       dscene->lookup_table.free();
 }
 
 static size_t round_up_to_multiple(size_t size, size_t chunk)
@@ -84,19 +87,27 @@ size_t LookupTables::add_table(DeviceScene *dscene, vector<float>& data)
        }
 
        /* copy table data and return offset */
-       dscene->lookup_table.copy_at(&data[0], new_table.offset, data.size());
+       float *dtable = dscene->lookup_table.get_data();
+       memcpy(dtable + new_table.offset, &data[0], sizeof(float) * data.size());
+
        return new_table.offset;
 }
 
-void LookupTables::remove_table(size_t 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) {
+               if(table->offset == *offset) {
                        lookup_tables.erase(table);
+                       *offset = TABLE_OFFSET_INVALID;
                        return;
                }
        }