diff options
Diffstat (limited to 'intern/cycles/device/device.cpp')
-rw-r--r-- | intern/cycles/device/device.cpp | 198 |
1 files changed, 198 insertions, 0 deletions
diff --git a/intern/cycles/device/device.cpp b/intern/cycles/device/device.cpp new file mode 100644 index 00000000000..4fae9310a7f --- /dev/null +++ b/intern/cycles/device/device.cpp @@ -0,0 +1,198 @@ +/* + * Copyright 2011, Blender Foundation. + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software Foundation, + * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + */ + +#include <stdlib.h> +#include <string.h> + +#include "device.h" +#include "device_intern.h" + +#include "util_cuda.h" +#include "util_debug.h" +#include "util_opengl.h" +#include "util_types.h" +#include "util_vector.h" + +CCL_NAMESPACE_BEGIN + +/* Device Task */ + +DeviceTask::DeviceTask(Type type_) +: type(type_), x(0), y(0), w(0), h(0), rng_state(0), rgba(0), buffer(0), + pass(0), resolution(0), + displace_input(0), displace_offset(0), displace_x(0), displace_w(0) +{ +} + +void DeviceTask::split(ThreadQueue<DeviceTask>& tasks, int num) +{ + if(type == DISPLACE) { + for(int i = 0; i < num; i++) { + int tx = displace_x + (displace_w/num)*i; + int tw = (i == num-1)? displace_w - i*(displace_w/num): displace_w/num; + + DeviceTask task = *this; + + task.displace_x = tx; + task.displace_w = tw; + + tasks.push(task); + } + } + else { + for(int i = 0; i < num; i++) { + int ty = y + (h/num)*i; + int th = (i == num-1)? h - i*(h/num): h/num; + + DeviceTask task = *this; + + task.y = ty; + task.h = th; + + tasks.push(task); + } + } +} + +/* Device */ + +void Device::pixels_alloc(device_memory& mem) +{ + mem_alloc(mem, MEM_READ_WRITE); +} + +void Device::pixels_copy_from(device_memory& mem, int y, int w, int h) +{ + mem_copy_from(mem, sizeof(uchar)*4*y*w, sizeof(uchar)*4*w*h); +} + +void Device::pixels_free(device_memory& mem) +{ + mem_free(mem); +} + +void Device::draw_pixels(device_memory& rgba, int y, int w, int h, int width, int height) +{ + pixels_copy_from(rgba, y, w, h); + + glPixelZoom((float)width/(float)w, (float)height/(float)h); + glRasterPos2f(0, y); + + glDrawPixels(w, h, GL_RGBA, GL_UNSIGNED_BYTE, (void*)rgba.data_pointer); + + glRasterPos2f(0.0f, 0.0f); + glPixelZoom(1.0f, 1.0f); +} + +Device *Device::create(DeviceType type, bool background) +{ + Device *device; + + switch(type) { + case DEVICE_CPU: + device = device_cpu_create(); + break; +#ifdef WITH_CUDA + case DEVICE_CUDA: + if(cuLibraryInit()) + device = device_cuda_create(background); + else + device = NULL; + break; +#endif +#ifdef WITH_MULTI + case DEVICE_MULTI: + device = device_multi_create(background); + break; +#endif +#ifdef WITH_NETWORK + case DEVICE_NETWORK: + device = device_network_create("127.0.0.1"); + break; +#endif +#ifdef WITH_OPENCL + case DEVICE_OPENCL: + device = device_opencl_create(background); + break; +#endif + default: + return NULL; + } + + return device; +} + +DeviceType Device::type_from_string(const char *name) +{ + if(strcmp(name, "cpu") == 0) + return DEVICE_CPU; + else if(strcmp(name, "cuda") == 0) + return DEVICE_CUDA; + else if(strcmp(name, "opencl") == 0) + return DEVICE_OPENCL; + else if(strcmp(name, "network") == 0) + return DEVICE_NETWORK; + else if(strcmp(name, "multi") == 0) + return DEVICE_MULTI; + + return DEVICE_NONE; +} + +string Device::string_from_type(DeviceType type) +{ + if(type == DEVICE_CPU) + return "cpu"; + else if(type == DEVICE_CUDA) + return "cuda"; + else if(type == DEVICE_OPENCL) + return "opencl"; + else if(type == DEVICE_NETWORK) + return "network"; + else if(type == DEVICE_MULTI) + return "multi"; + + return ""; +} + +vector<DeviceType> Device::available_types() +{ + vector<DeviceType> types; + + types.push_back(DEVICE_CPU); + +#ifdef WITH_CUDA + if(cuLibraryInit()) + types.push_back(DEVICE_CUDA); +#endif + +#ifdef WITH_OPENCL + types.push_back(DEVICE_OPENCL); +#endif + +#ifdef WITH_NETWORK + types.push_back(DEVICE_NETWORK); +#endif +#ifdef WITH_MULTI + types.push_back(DEVICE_MULTI); +#endif + + return types; +} + +CCL_NAMESPACE_END + |