diff options
Diffstat (limited to 'intern/cycles/blender/blender_python.cpp')
-rw-r--r-- | intern/cycles/blender/blender_python.cpp | 1063 |
1 files changed, 0 insertions, 1063 deletions
diff --git a/intern/cycles/blender/blender_python.cpp b/intern/cycles/blender/blender_python.cpp deleted file mode 100644 index d681517c9e1..00000000000 --- a/intern/cycles/blender/blender_python.cpp +++ /dev/null @@ -1,1063 +0,0 @@ -/* - * Copyright 2011-2013 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. - */ - -#include <Python.h> - -#include "blender/CCL_api.h" - -#include "blender/blender_device.h" -#include "blender/blender_session.h" -#include "blender/blender_sync.h" -#include "blender/blender_util.h" - -#include "render/denoising.h" -#include "render/merge.h" - -#include "util/util_debug.h" -#include "util/util_foreach.h" -#include "util/util_logging.h" -#include "util/util_md5.h" -#include "util/util_opengl.h" -#include "util/util_openimagedenoise.h" -#include "util/util_path.h" -#include "util/util_string.h" -#include "util/util_task.h" -#include "util/util_tbb.h" -#include "util/util_types.h" - -#ifdef WITH_OSL -# include "render/osl.h" - -# include <OSL/oslconfig.h> -# include <OSL/oslquery.h> -#endif - -CCL_NAMESPACE_BEGIN - -namespace { - -/* Flag describing whether debug flags were synchronized from scene. */ -bool debug_flags_set = false; - -void *pylong_as_voidptr_typesafe(PyObject *object) -{ - if (object == Py_None) - return NULL; - return PyLong_AsVoidPtr(object); -} - -PyObject *pyunicode_from_string(const char *str) -{ - /* Ignore errors if device API returns invalid UTF-8 strings. */ - return PyUnicode_DecodeUTF8(str, strlen(str), "ignore"); -} - -/* Synchronize debug flags from a given Blender scene. - * Return truth when device list needs invalidation. - */ -static void debug_flags_sync_from_scene(BL::Scene b_scene) -{ - DebugFlagsRef flags = DebugFlags(); - PointerRNA cscene = RNA_pointer_get(&b_scene.ptr, "cycles"); - /* Synchronize shared flags. */ - flags.viewport_static_bvh = get_enum(cscene, "debug_bvh_type"); - /* Synchronize CPU flags. */ - flags.cpu.avx2 = get_boolean(cscene, "debug_use_cpu_avx2"); - flags.cpu.avx = get_boolean(cscene, "debug_use_cpu_avx"); - flags.cpu.sse41 = get_boolean(cscene, "debug_use_cpu_sse41"); - flags.cpu.sse3 = get_boolean(cscene, "debug_use_cpu_sse3"); - flags.cpu.sse2 = get_boolean(cscene, "debug_use_cpu_sse2"); - flags.cpu.bvh_layout = (BVHLayout)get_enum(cscene, "debug_bvh_layout"); - /* Synchronize CUDA flags. */ - flags.cuda.adaptive_compile = get_boolean(cscene, "debug_use_cuda_adaptive_compile"); - /* Synchronize OptiX flags. */ - flags.optix.use_debug = get_boolean(cscene, "debug_use_optix_debug"); -} - -/* Reset debug flags to default values. - * Return truth when device list needs invalidation. - */ -static void debug_flags_reset() -{ - DebugFlagsRef flags = DebugFlags(); - flags.reset(); -} - -} /* namespace */ - -void python_thread_state_save(void **python_thread_state) -{ - *python_thread_state = (void *)PyEval_SaveThread(); -} - -void python_thread_state_restore(void **python_thread_state) -{ - PyEval_RestoreThread((PyThreadState *)*python_thread_state); - *python_thread_state = NULL; -} - -static const char *PyC_UnicodeAsByte(PyObject *py_str, PyObject **coerce) -{ - const char *result = PyUnicode_AsUTF8(py_str); - if (result) { - /* 99% of the time this is enough but we better support non unicode - * chars since blender doesn't limit this. - */ - return result; - } - else { - PyErr_Clear(); - if (PyBytes_Check(py_str)) { - return PyBytes_AS_STRING(py_str); - } - else if ((*coerce = PyUnicode_EncodeFSDefault(py_str))) { - return PyBytes_AS_STRING(*coerce); - } - else { - /* Clear the error, so Cycles can be at least used without - * GPU and OSL support, - */ - PyErr_Clear(); - return ""; - } - } -} - -static PyObject *init_func(PyObject * /*self*/, PyObject *args) -{ - PyObject *path, *user_path, *temp_path; - int headless; - - if (!PyArg_ParseTuple(args, "OOOi", &path, &user_path, &temp_path, &headless)) { - return nullptr; - } - - PyObject *path_coerce = nullptr, *user_path_coerce = nullptr, *temp_path_coerce = nullptr; - path_init(PyC_UnicodeAsByte(path, &path_coerce), - PyC_UnicodeAsByte(user_path, &user_path_coerce), - PyC_UnicodeAsByte(temp_path, &temp_path_coerce)); - Py_XDECREF(path_coerce); - Py_XDECREF(user_path_coerce); - Py_XDECREF(temp_path_coerce); - - BlenderSession::headless = headless; - - DebugFlags().running_inside_blender = true; - - VLOG(2) << "Debug flags initialized to:\n" << DebugFlags(); - - Py_RETURN_NONE; -} - -static PyObject *exit_func(PyObject * /*self*/, PyObject * /*args*/) -{ - ShaderManager::free_memory(); - TaskScheduler::free_memory(); - Device::free_memory(); - Py_RETURN_NONE; -} - -static PyObject *create_func(PyObject * /*self*/, PyObject *args) -{ - PyObject *pyengine, *pypreferences, *pydata, *pyscreen, *pyregion, *pyv3d, *pyrv3d; - int preview_osl; - - if (!PyArg_ParseTuple(args, - "OOOOOOOi", - &pyengine, - &pypreferences, - &pydata, - &pyscreen, - &pyregion, - &pyv3d, - &pyrv3d, - &preview_osl)) { - return NULL; - } - - /* RNA */ - ID *bScreen = (ID *)PyLong_AsVoidPtr(pyscreen); - - PointerRNA engineptr; - RNA_pointer_create(NULL, &RNA_RenderEngine, (void *)PyLong_AsVoidPtr(pyengine), &engineptr); - BL::RenderEngine engine(engineptr); - - PointerRNA preferencesptr; - RNA_pointer_create( - NULL, &RNA_Preferences, (void *)PyLong_AsVoidPtr(pypreferences), &preferencesptr); - BL::Preferences preferences(preferencesptr); - - PointerRNA dataptr; - RNA_main_pointer_create((Main *)PyLong_AsVoidPtr(pydata), &dataptr); - BL::BlendData data(dataptr); - - PointerRNA regionptr; - RNA_pointer_create(bScreen, &RNA_Region, pylong_as_voidptr_typesafe(pyregion), ®ionptr); - BL::Region region(regionptr); - - PointerRNA v3dptr; - RNA_pointer_create(bScreen, &RNA_SpaceView3D, pylong_as_voidptr_typesafe(pyv3d), &v3dptr); - BL::SpaceView3D v3d(v3dptr); - - PointerRNA rv3dptr; - RNA_pointer_create(bScreen, &RNA_RegionView3D, pylong_as_voidptr_typesafe(pyrv3d), &rv3dptr); - BL::RegionView3D rv3d(rv3dptr); - - /* create session */ - BlenderSession *session; - - if (rv3d) { - /* interactive viewport session */ - int width = region.width(); - int height = region.height(); - - session = new BlenderSession(engine, preferences, data, v3d, rv3d, width, height); - } - else { - /* offline session or preview render */ - session = new BlenderSession(engine, preferences, data, preview_osl); - } - - return PyLong_FromVoidPtr(session); -} - -static PyObject *free_func(PyObject * /*self*/, PyObject *value) -{ - delete (BlenderSession *)PyLong_AsVoidPtr(value); - - Py_RETURN_NONE; -} - -static PyObject *render_func(PyObject * /*self*/, PyObject *args) -{ - PyObject *pysession, *pydepsgraph; - - if (!PyArg_ParseTuple(args, "OO", &pysession, &pydepsgraph)) - return NULL; - - BlenderSession *session = (BlenderSession *)PyLong_AsVoidPtr(pysession); - - PointerRNA depsgraphptr; - RNA_pointer_create(NULL, &RNA_Depsgraph, (ID *)PyLong_AsVoidPtr(pydepsgraph), &depsgraphptr); - BL::Depsgraph b_depsgraph(depsgraphptr); - - /* Allow Blender to execute other Python scripts. */ - python_thread_state_save(&session->python_thread_state); - - session->render(b_depsgraph); - - python_thread_state_restore(&session->python_thread_state); - - Py_RETURN_NONE; -} - -static PyObject *render_frame_finish_func(PyObject * /*self*/, PyObject *args) -{ - PyObject *pysession; - - if (!PyArg_ParseTuple(args, "O", &pysession)) { - return nullptr; - } - - BlenderSession *session = (BlenderSession *)PyLong_AsVoidPtr(pysession); - - /* Allow Blender to execute other Python scripts. */ - python_thread_state_save(&session->python_thread_state); - - session->render_frame_finish(); - - python_thread_state_restore(&session->python_thread_state); - - Py_RETURN_NONE; -} - -static PyObject *draw_func(PyObject * /*self*/, PyObject *args) -{ - PyObject *py_session, *py_graph, *py_screen, *py_space_image; - - if (!PyArg_ParseTuple(args, "OOOO", &py_session, &py_graph, &py_screen, &py_space_image)) { - return nullptr; - } - - BlenderSession *session = (BlenderSession *)PyLong_AsVoidPtr(py_session); - - ID *b_screen = (ID *)PyLong_AsVoidPtr(py_screen); - - PointerRNA b_space_image_ptr; - RNA_pointer_create(b_screen, - &RNA_SpaceImageEditor, - pylong_as_voidptr_typesafe(py_space_image), - &b_space_image_ptr); - BL::SpaceImageEditor b_space_image(b_space_image_ptr); - - session->draw(b_space_image); - - Py_RETURN_NONE; -} - -/* pixel_array and result passed as pointers */ -static PyObject *bake_func(PyObject * /*self*/, PyObject *args) -{ - PyObject *pysession, *pydepsgraph, *pyobject; - const char *pass_type; - int pass_filter, width, height; - - if (!PyArg_ParseTuple(args, - "OOOsiii", - &pysession, - &pydepsgraph, - &pyobject, - &pass_type, - &pass_filter, - &width, - &height)) - return NULL; - - BlenderSession *session = (BlenderSession *)PyLong_AsVoidPtr(pysession); - - PointerRNA depsgraphptr; - RNA_pointer_create(NULL, &RNA_Depsgraph, PyLong_AsVoidPtr(pydepsgraph), &depsgraphptr); - BL::Depsgraph b_depsgraph(depsgraphptr); - - PointerRNA objectptr; - RNA_id_pointer_create((ID *)PyLong_AsVoidPtr(pyobject), &objectptr); - BL::Object b_object(objectptr); - - python_thread_state_save(&session->python_thread_state); - - session->bake(b_depsgraph, b_object, pass_type, pass_filter, width, height); - - python_thread_state_restore(&session->python_thread_state); - - Py_RETURN_NONE; -} - -static PyObject *view_draw_func(PyObject * /*self*/, PyObject *args) -{ - PyObject *pysession, *pygraph, *pyv3d, *pyrv3d; - - if (!PyArg_ParseTuple(args, "OOOO", &pysession, &pygraph, &pyv3d, &pyrv3d)) - return NULL; - - BlenderSession *session = (BlenderSession *)PyLong_AsVoidPtr(pysession); - - if (PyLong_AsVoidPtr(pyrv3d)) { - /* 3d view drawing */ - int viewport[4]; - glGetIntegerv(GL_VIEWPORT, viewport); - - session->view_draw(viewport[2], viewport[3]); - } - - Py_RETURN_NONE; -} - -static PyObject *reset_func(PyObject * /*self*/, PyObject *args) -{ - PyObject *pysession, *pydata, *pydepsgraph; - - if (!PyArg_ParseTuple(args, "OOO", &pysession, &pydata, &pydepsgraph)) - return NULL; - - BlenderSession *session = (BlenderSession *)PyLong_AsVoidPtr(pysession); - - PointerRNA dataptr; - RNA_main_pointer_create((Main *)PyLong_AsVoidPtr(pydata), &dataptr); - BL::BlendData b_data(dataptr); - - PointerRNA depsgraphptr; - RNA_pointer_create(NULL, &RNA_Depsgraph, PyLong_AsVoidPtr(pydepsgraph), &depsgraphptr); - BL::Depsgraph b_depsgraph(depsgraphptr); - - python_thread_state_save(&session->python_thread_state); - - session->reset_session(b_data, b_depsgraph); - - python_thread_state_restore(&session->python_thread_state); - - Py_RETURN_NONE; -} - -static PyObject *sync_func(PyObject * /*self*/, PyObject *args) -{ - PyObject *pysession, *pydepsgraph; - - if (!PyArg_ParseTuple(args, "OO", &pysession, &pydepsgraph)) - return NULL; - - BlenderSession *session = (BlenderSession *)PyLong_AsVoidPtr(pysession); - - PointerRNA depsgraphptr; - RNA_pointer_create(NULL, &RNA_Depsgraph, PyLong_AsVoidPtr(pydepsgraph), &depsgraphptr); - BL::Depsgraph b_depsgraph(depsgraphptr); - - python_thread_state_save(&session->python_thread_state); - - session->synchronize(b_depsgraph); - - python_thread_state_restore(&session->python_thread_state); - - Py_RETURN_NONE; -} - -static PyObject *available_devices_func(PyObject * /*self*/, PyObject *args) -{ - const char *type_name; - if (!PyArg_ParseTuple(args, "s", &type_name)) { - return NULL; - } - - DeviceType type = Device::type_from_string(type_name); - /* "NONE" is defined by the add-on, see: `CyclesPreferences.get_device_types`. */ - if ((type == DEVICE_NONE) && (strcmp(type_name, "NONE") != 0)) { - PyErr_Format(PyExc_ValueError, "Device \"%s\" not known.", type_name); - return NULL; - } - - uint mask = (type == DEVICE_NONE) ? DEVICE_MASK_ALL : DEVICE_MASK(type); - mask |= DEVICE_MASK_CPU; - - vector<DeviceInfo> devices = Device::available_devices(mask); - PyObject *ret = PyTuple_New(devices.size()); - - for (size_t i = 0; i < devices.size(); i++) { - DeviceInfo &device = devices[i]; - string type_name = Device::string_from_type(device.type); - PyObject *device_tuple = PyTuple_New(4); - PyTuple_SET_ITEM(device_tuple, 0, pyunicode_from_string(device.description.c_str())); - PyTuple_SET_ITEM(device_tuple, 1, pyunicode_from_string(type_name.c_str())); - PyTuple_SET_ITEM(device_tuple, 2, pyunicode_from_string(device.id.c_str())); - PyTuple_SET_ITEM(device_tuple, 3, PyBool_FromLong(device.has_peer_memory)); - PyTuple_SET_ITEM(ret, i, device_tuple); - } - - return ret; -} - -#ifdef WITH_OSL - -static PyObject *osl_update_node_func(PyObject * /*self*/, PyObject *args) -{ - PyObject *pydata, *pynodegroup, *pynode; - const char *filepath = NULL; - - if (!PyArg_ParseTuple(args, "OOOs", &pydata, &pynodegroup, &pynode, &filepath)) - return NULL; - - /* RNA */ - PointerRNA dataptr; - RNA_main_pointer_create((Main *)PyLong_AsVoidPtr(pydata), &dataptr); - BL::BlendData b_data(dataptr); - - PointerRNA nodeptr; - RNA_pointer_create((ID *)PyLong_AsVoidPtr(pynodegroup), - &RNA_ShaderNodeScript, - (void *)PyLong_AsVoidPtr(pynode), - &nodeptr); - BL::ShaderNodeScript b_node(nodeptr); - - /* update bytecode hash */ - string bytecode = b_node.bytecode(); - - if (!bytecode.empty()) { - MD5Hash md5; - md5.append((const uint8_t *)bytecode.c_str(), bytecode.size()); - b_node.bytecode_hash(md5.get_hex().c_str()); - } - else - b_node.bytecode_hash(""); - - /* query from file path */ - OSL::OSLQuery query; - - if (!OSLShaderManager::osl_query(query, filepath)) - Py_RETURN_FALSE; - - /* add new sockets from parameters */ - set<void *> used_sockets; - - for (int i = 0; i < query.nparams(); i++) { - const OSL::OSLQuery::Parameter *param = query.getparam(i); - - /* skip unsupported types */ - if (param->varlenarray || param->isstruct || param->type.arraylen > 1) - continue; - - /* Read metadata. */ - bool is_bool_param = false; - ustring param_label = param->name; - - for (const OSL::OSLQuery::Parameter &metadata : param->metadata) { - if (metadata.type == TypeDesc::STRING) { - if (metadata.name == "widget") { - /* Boolean socket. */ - if (metadata.sdefault[0] == "boolean" || metadata.sdefault[0] == "checkBox") { - is_bool_param = true; - } - } - else if (metadata.name == "label") { - /* Socket label. */ - param_label = metadata.sdefault[0]; - } - } - } - /* determine socket type */ - string socket_type; - BL::NodeSocket::type_enum data_type = BL::NodeSocket::type_VALUE; - float4 default_float4 = make_float4(0.0f, 0.0f, 0.0f, 1.0f); - float default_float = 0.0f; - int default_int = 0; - string default_string = ""; - bool default_boolean = false; - - if (param->isclosure) { - socket_type = "NodeSocketShader"; - data_type = BL::NodeSocket::type_SHADER; - } - else if (param->type.vecsemantics == TypeDesc::COLOR) { - socket_type = "NodeSocketColor"; - data_type = BL::NodeSocket::type_RGBA; - - if (param->validdefault) { - default_float4[0] = param->fdefault[0]; - default_float4[1] = param->fdefault[1]; - default_float4[2] = param->fdefault[2]; - } - } - else if (param->type.vecsemantics == TypeDesc::POINT || - param->type.vecsemantics == TypeDesc::VECTOR || - param->type.vecsemantics == TypeDesc::NORMAL) { - socket_type = "NodeSocketVector"; - data_type = BL::NodeSocket::type_VECTOR; - - if (param->validdefault) { - default_float4[0] = param->fdefault[0]; - default_float4[1] = param->fdefault[1]; - default_float4[2] = param->fdefault[2]; - } - } - else if (param->type.aggregate == TypeDesc::SCALAR) { - if (param->type.basetype == TypeDesc::INT) { - if (is_bool_param) { - socket_type = "NodeSocketBool"; - data_type = BL::NodeSocket::type_BOOLEAN; - if (param->validdefault) { - default_boolean = (bool)param->idefault[0]; - } - } - else { - socket_type = "NodeSocketInt"; - data_type = BL::NodeSocket::type_INT; - if (param->validdefault) - default_int = param->idefault[0]; - } - } - else if (param->type.basetype == TypeDesc::FLOAT) { - socket_type = "NodeSocketFloat"; - data_type = BL::NodeSocket::type_VALUE; - if (param->validdefault) - default_float = param->fdefault[0]; - } - else if (param->type.basetype == TypeDesc::STRING) { - socket_type = "NodeSocketString"; - data_type = BL::NodeSocket::type_STRING; - if (param->validdefault) - default_string = param->sdefault[0].string(); - } - else - continue; - } - else - continue; - - /* Update existing socket. */ - bool found_existing = false; - if (param->isoutput) { - for (BL::NodeSocket &b_sock : b_node.outputs) { - if (b_sock.identifier() == param->name) { - if (b_sock.bl_idname() != socket_type) { - /* Remove if type no longer matches. */ - b_node.outputs.remove(b_data, b_sock); - } - else { - /* Reuse and update label. */ - if (b_sock.name() != param_label) { - b_sock.name(param_label.string()); - } - used_sockets.insert(b_sock.ptr.data); - found_existing = true; - } - break; - } - } - } - else { - for (BL::NodeSocket &b_sock : b_node.inputs) { - if (b_sock.identifier() == param->name) { - if (b_sock.bl_idname() != socket_type) { - /* Remove if type no longer matches. */ - b_node.inputs.remove(b_data, b_sock); - } - else { - /* Reuse and update label. */ - if (b_sock.name() != param_label) { - b_sock.name(param_label.string()); - } - used_sockets.insert(b_sock.ptr.data); - found_existing = true; - } - break; - } - } - } - - if (!found_existing) { - /* Create new socket. */ - BL::NodeSocket b_sock = (param->isoutput) ? b_node.outputs.create(b_data, - socket_type.c_str(), - param_label.c_str(), - param->name.c_str()) : - b_node.inputs.create(b_data, - socket_type.c_str(), - param_label.c_str(), - param->name.c_str()); - - /* set default value */ - if (data_type == BL::NodeSocket::type_VALUE) { - set_float(b_sock.ptr, "default_value", default_float); - } - else if (data_type == BL::NodeSocket::type_INT) { - set_int(b_sock.ptr, "default_value", default_int); - } - else if (data_type == BL::NodeSocket::type_RGBA) { - set_float4(b_sock.ptr, "default_value", default_float4); - } - else if (data_type == BL::NodeSocket::type_VECTOR) { - set_float3(b_sock.ptr, "default_value", float4_to_float3(default_float4)); - } - else if (data_type == BL::NodeSocket::type_STRING) { - set_string(b_sock.ptr, "default_value", default_string); - } - else if (data_type == BL::NodeSocket::type_BOOLEAN) { - set_boolean(b_sock.ptr, "default_value", default_boolean); - } - - used_sockets.insert(b_sock.ptr.data); - } - } - - /* remove unused parameters */ - bool removed; - - do { - removed = false; - - for (BL::NodeSocket &b_input : b_node.inputs) { - if (used_sockets.find(b_input.ptr.data) == used_sockets.end()) { - b_node.inputs.remove(b_data, b_input); - removed = true; - break; - } - } - - for (BL::NodeSocket &b_output : b_node.outputs) { - if (used_sockets.find(b_output.ptr.data) == used_sockets.end()) { - b_node.outputs.remove(b_data, b_output); - removed = true; - break; - } - } - } while (removed); - - Py_RETURN_TRUE; -} - -static PyObject *osl_compile_func(PyObject * /*self*/, PyObject *args) -{ - const char *inputfile = NULL, *outputfile = NULL; - - if (!PyArg_ParseTuple(args, "ss", &inputfile, &outputfile)) - return NULL; - - /* return */ - if (!OSLShaderManager::osl_compile(inputfile, outputfile)) - Py_RETURN_FALSE; - - Py_RETURN_TRUE; -} -#endif - -static PyObject *system_info_func(PyObject * /*self*/, PyObject * /*value*/) -{ - string system_info = Device::device_capabilities(); - return pyunicode_from_string(system_info.c_str()); -} - -static bool image_parse_filepaths(PyObject *pyfilepaths, vector<string> &filepaths) -{ - if (PyUnicode_Check(pyfilepaths)) { - const char *filepath = PyUnicode_AsUTF8(pyfilepaths); - filepaths.push_back(filepath); - return true; - } - - PyObject *sequence = PySequence_Fast(pyfilepaths, - "File paths must be a string or sequence of strings"); - if (sequence == NULL) { - return false; - } - - for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(sequence); i++) { - PyObject *item = PySequence_Fast_GET_ITEM(sequence, i); - const char *filepath = PyUnicode_AsUTF8(item); - if (filepath == NULL) { - PyErr_SetString(PyExc_ValueError, "File paths must be a string or sequence of strings."); - Py_DECREF(sequence); - return false; - } - filepaths.push_back(filepath); - } - Py_DECREF(sequence); - - return true; -} - -static PyObject *denoise_func(PyObject * /*self*/, PyObject *args, PyObject *keywords) -{ -#if 1 - (void)args; - (void)keywords; -#else - static const char *keyword_list[] = { - "preferences", "scene", "view_layer", "input", "output", "tile_size", "samples", NULL}; - PyObject *pypreferences, *pyscene, *pyviewlayer; - PyObject *pyinput, *pyoutput = NULL; - int tile_size = 0, samples = 0; - - if (!PyArg_ParseTupleAndKeywords(args, - keywords, - "OOOO|Oii", - (char **)keyword_list, - &pypreferences, - &pyscene, - &pyviewlayer, - &pyinput, - &pyoutput, - &tile_size, - &samples)) { - return NULL; - } - - /* Get device specification from preferences and scene. */ - PointerRNA preferencesptr; - RNA_pointer_create( - NULL, &RNA_Preferences, (void *)PyLong_AsVoidPtr(pypreferences), &preferencesptr); - BL::Preferences b_preferences(preferencesptr); - - PointerRNA sceneptr; - RNA_id_pointer_create((ID *)PyLong_AsVoidPtr(pyscene), &sceneptr); - BL::Scene b_scene(sceneptr); - - DeviceInfo device = blender_device_info(b_preferences, b_scene, true); - - /* Get denoising parameters from view layer. */ - PointerRNA viewlayerptr; - RNA_pointer_create((ID *)PyLong_AsVoidPtr(pyscene), - &RNA_ViewLayer, - PyLong_AsVoidPtr(pyviewlayer), - &viewlayerptr); - PointerRNA cviewlayer = RNA_pointer_get(&viewlayerptr, "cycles"); - - DenoiseParams params; - params.radius = get_int(cviewlayer, "denoising_radius"); - params.strength = get_float(cviewlayer, "denoising_strength"); - params.feature_strength = get_float(cviewlayer, "denoising_feature_strength"); - params.relative_pca = get_boolean(cviewlayer, "denoising_relative_pca"); - params.neighbor_frames = get_int(cviewlayer, "denoising_neighbor_frames"); - - /* Parse file paths list. */ - vector<string> input, output; - - if (!image_parse_filepaths(pyinput, input)) { - return NULL; - } - - if (pyoutput) { - if (!image_parse_filepaths(pyoutput, output)) { - return NULL; - } - } - else { - output = input; - } - - if (input.empty()) { - PyErr_SetString(PyExc_ValueError, "No input file paths specified."); - return NULL; - } - if (input.size() != output.size()) { - PyErr_SetString(PyExc_ValueError, "Number of input and output file paths does not match."); - return NULL; - } - - /* Create denoiser. */ - DenoiserPipeline denoiser(device); - denoiser.params = params; - denoiser.input = input; - denoiser.output = output; - - if (tile_size > 0) { - denoiser.tile_size = make_int2(tile_size, tile_size); - } - if (samples > 0) { - denoiser.samples_override = samples; - } - - /* Run denoiser. */ - if (!denoiser.run()) { - PyErr_SetString(PyExc_ValueError, denoiser.error.c_str()); - return NULL; - } -#endif - - Py_RETURN_NONE; -} - -static PyObject *merge_func(PyObject * /*self*/, PyObject *args, PyObject *keywords) -{ - static const char *keyword_list[] = {"input", "output", NULL}; - PyObject *pyinput, *pyoutput = NULL; - - if (!PyArg_ParseTupleAndKeywords( - args, keywords, "OO", (char **)keyword_list, &pyinput, &pyoutput)) { - return NULL; - } - - /* Parse input list. */ - vector<string> input; - if (!image_parse_filepaths(pyinput, input)) { - return NULL; - } - - /* Parse output string. */ - if (!PyUnicode_Check(pyoutput)) { - PyErr_SetString(PyExc_ValueError, "Output must be a string."); - return NULL; - } - string output = PyUnicode_AsUTF8(pyoutput); - - /* Merge. */ - ImageMerger merger; - merger.input = input; - merger.output = output; - - if (!merger.run()) { - PyErr_SetString(PyExc_ValueError, merger.error.c_str()); - return NULL; - } - - Py_RETURN_NONE; -} - -static PyObject *debug_flags_update_func(PyObject * /*self*/, PyObject *args) -{ - PyObject *pyscene; - if (!PyArg_ParseTuple(args, "O", &pyscene)) { - return NULL; - } - - PointerRNA sceneptr; - RNA_id_pointer_create((ID *)PyLong_AsVoidPtr(pyscene), &sceneptr); - BL::Scene b_scene(sceneptr); - - debug_flags_sync_from_scene(b_scene); - - VLOG(2) << "Debug flags set to:\n" << DebugFlags(); - - debug_flags_set = true; - - Py_RETURN_NONE; -} - -static PyObject *debug_flags_reset_func(PyObject * /*self*/, PyObject * /*args*/) -{ - debug_flags_reset(); - if (debug_flags_set) { - VLOG(2) << "Debug flags reset to:\n" << DebugFlags(); - debug_flags_set = false; - } - Py_RETURN_NONE; -} - -static PyObject *enable_print_stats_func(PyObject * /*self*/, PyObject * /*args*/) -{ - BlenderSession::print_render_stats = true; - Py_RETURN_NONE; -} - -static PyObject *get_device_types_func(PyObject * /*self*/, PyObject * /*args*/) -{ - vector<DeviceType> device_types = Device::available_types(); - bool has_cuda = false, has_optix = false, has_hip = false; - foreach (DeviceType device_type, device_types) { - has_cuda |= (device_type == DEVICE_CUDA); - has_optix |= (device_type == DEVICE_OPTIX); - has_hip |= (device_type == DEVICE_HIP); - } - PyObject *list = PyTuple_New(3); - PyTuple_SET_ITEM(list, 0, PyBool_FromLong(has_cuda)); - PyTuple_SET_ITEM(list, 1, PyBool_FromLong(has_optix)); - PyTuple_SET_ITEM(list, 2, PyBool_FromLong(has_hip)); - return list; -} - -static PyObject *set_device_override_func(PyObject * /*self*/, PyObject *arg) -{ - PyObject *override_string = PyObject_Str(arg); - string override = PyUnicode_AsUTF8(override_string); - Py_DECREF(override_string); - - bool include_cpu = false; - const string cpu_suffix = "+CPU"; - if (string_endswith(override, cpu_suffix)) { - include_cpu = true; - override = override.substr(0, override.length() - cpu_suffix.length()); - } - - if (override == "CPU") { - BlenderSession::device_override = DEVICE_MASK_CPU; - } - else if (override == "CUDA") { - BlenderSession::device_override = DEVICE_MASK_CUDA; - } - else if (override == "OPTIX") { - BlenderSession::device_override = DEVICE_MASK_OPTIX; - } - else if (override == "HIP") { - BlenderSession::device_override = DEVICE_MASK_HIP; - } - else { - printf("\nError: %s is not a valid Cycles device.\n", override.c_str()); - Py_RETURN_FALSE; - } - - if (include_cpu) { - BlenderSession::device_override = (DeviceTypeMask)(BlenderSession::device_override | - DEVICE_MASK_CPU); - } - - Py_RETURN_TRUE; -} - -static PyMethodDef methods[] = { - {"init", init_func, METH_VARARGS, ""}, - {"exit", exit_func, METH_VARARGS, ""}, - {"create", create_func, METH_VARARGS, ""}, - {"free", free_func, METH_O, ""}, - {"render", render_func, METH_VARARGS, ""}, - {"render_frame_finish", render_frame_finish_func, METH_VARARGS, ""}, - {"draw", draw_func, METH_VARARGS, ""}, - {"bake", bake_func, METH_VARARGS, ""}, - {"view_draw", view_draw_func, METH_VARARGS, ""}, - {"sync", sync_func, METH_VARARGS, ""}, - {"reset", reset_func, METH_VARARGS, ""}, -#ifdef WITH_OSL - {"osl_update_node", osl_update_node_func, METH_VARARGS, ""}, - {"osl_compile", osl_compile_func, METH_VARARGS, ""}, -#endif - {"available_devices", available_devices_func, METH_VARARGS, ""}, - {"system_info", system_info_func, METH_NOARGS, ""}, - - /* Standalone denoising */ - {"denoise", (PyCFunction)denoise_func, METH_VARARGS | METH_KEYWORDS, ""}, - {"merge", (PyCFunction)merge_func, METH_VARARGS | METH_KEYWORDS, ""}, - - /* Debugging routines */ - {"debug_flags_update", debug_flags_update_func, METH_VARARGS, ""}, - {"debug_flags_reset", debug_flags_reset_func, METH_NOARGS, ""}, - - /* Statistics. */ - {"enable_print_stats", enable_print_stats_func, METH_NOARGS, ""}, - - /* Compute Device selection */ - {"get_device_types", get_device_types_func, METH_VARARGS, ""}, - {"set_device_override", set_device_override_func, METH_O, ""}, - - {NULL, NULL, 0, NULL}, -}; - -static struct PyModuleDef module = { - PyModuleDef_HEAD_INIT, - "_cycles", - "Blender cycles render integration", - -1, - methods, - NULL, - NULL, - NULL, - NULL, -}; - -CCL_NAMESPACE_END - -void *CCL_python_module_init() -{ - PyObject *mod = PyModule_Create(&ccl::module); - -#ifdef WITH_OSL - /* TODO(sergey): This gives us library we've been linking against. - * In theory with dynamic OSL library it might not be - * accurate, but there's nothing in OSL API which we - * might use to get version in runtime. - */ - int curversion = OSL_LIBRARY_VERSION_CODE; - PyModule_AddObject(mod, "with_osl", Py_True); - Py_INCREF(Py_True); - PyModule_AddObject( - mod, - "osl_version", - Py_BuildValue("(iii)", curversion / 10000, (curversion / 100) % 100, curversion % 100)); - PyModule_AddObject( - mod, - "osl_version_string", - PyUnicode_FromFormat( - "%2d, %2d, %2d", curversion / 10000, (curversion / 100) % 100, curversion % 100)); -#else - PyModule_AddObject(mod, "with_osl", Py_False); - Py_INCREF(Py_False); - PyModule_AddStringConstant(mod, "osl_version", "unknown"); - PyModule_AddStringConstant(mod, "osl_version_string", "unknown"); -#endif - -#ifdef WITH_EMBREE - PyModule_AddObject(mod, "with_embree", Py_True); - Py_INCREF(Py_True); -#else /* WITH_EMBREE */ - PyModule_AddObject(mod, "with_embree", Py_False); - Py_INCREF(Py_False); -#endif /* WITH_EMBREE */ - - if (ccl::openimagedenoise_supported()) { - PyModule_AddObject(mod, "with_openimagedenoise", Py_True); - Py_INCREF(Py_True); - } - else { - PyModule_AddObject(mod, "with_openimagedenoise", Py_False); - Py_INCREF(Py_False); - } - - return (void *)mod; -} |