forked from bartvdbraak/blender
78 lines
2.4 KiB
C
78 lines
2.4 KiB
C
/*
|
|
* Copyright 2011-2016 Blender Foundation
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* 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.
|
|
*/
|
|
|
|
#ifndef __KERNEL_SPLIT_DATA_H__
|
|
#define __KERNEL_SPLIT_DATA_H__
|
|
|
|
#include "kernel/split/kernel_split_data_types.h"
|
|
|
|
#include "kernel/kernel_globals.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
ccl_device_inline uint64_t split_data_buffer_size(KernelGlobals *kg, size_t num_elements)
|
|
{
|
|
(void)kg; /* Unused on CPU. */
|
|
|
|
uint64_t size = 0;
|
|
#define SPLIT_DATA_ENTRY(type, name, num) +align_up(num_elements *num * sizeof(type), 16)
|
|
size = size SPLIT_DATA_ENTRIES;
|
|
#undef SPLIT_DATA_ENTRY
|
|
|
|
uint64_t closure_size = sizeof(ShaderClosure) * (kernel_data.integrator.max_closures - 1);
|
|
|
|
#ifdef __BRANCHED_PATH__
|
|
size += align_up(num_elements * (sizeof(ShaderData) + closure_size), 16);
|
|
#endif
|
|
|
|
size += align_up(num_elements * (sizeof(ShaderData) + closure_size), 16);
|
|
|
|
return size;
|
|
}
|
|
|
|
ccl_device_inline void split_data_init(KernelGlobals *kg,
|
|
ccl_global SplitData *split_data,
|
|
size_t num_elements,
|
|
ccl_global void *data,
|
|
ccl_global char *ray_state)
|
|
{
|
|
(void)kg; /* Unused on CPU. */
|
|
|
|
ccl_global char *p = (ccl_global char *)data;
|
|
|
|
#define SPLIT_DATA_ENTRY(type, name, num) \
|
|
split_data->name = (type *)p; \
|
|
p += align_up(num_elements * num * sizeof(type), 16);
|
|
SPLIT_DATA_ENTRIES;
|
|
#undef SPLIT_DATA_ENTRY
|
|
|
|
uint64_t closure_size = sizeof(ShaderClosure) * (kernel_data.integrator.max_closures - 1);
|
|
|
|
#ifdef __BRANCHED_PATH__
|
|
split_data->_branched_state_sd = (ShaderData *)p;
|
|
p += align_up(num_elements * (sizeof(ShaderData) + closure_size), 16);
|
|
#endif
|
|
|
|
split_data->_sd = (ShaderData *)p;
|
|
p += align_up(num_elements * (sizeof(ShaderData) + closure_size), 16);
|
|
|
|
split_data->ray_state = ray_state;
|
|
}
|
|
|
|
CCL_NAMESPACE_END
|
|
|
|
#endif /* __KERNEL_SPLIT_DATA_H__ */
|