Blender V5.0
optix/device_impl.h
Go to the documentation of this file.
1/* SPDX-FileCopyrightText: 2019 NVIDIA Corporation
2 * SPDX-FileCopyrightText: 2019-2022 Blender Foundation
3 *
4 * SPDX-License-Identifier: Apache-2.0 */
5
6#pragma once
7
8#ifdef WITH_OPTIX
9
10# include "device/cuda/device_impl.h"
11# include "device/optix/util.h" // IWYU pragma: keep
12# include "kernel/osl/globals.h"
13
14# include "util/task.h"
15
16CCL_NAMESPACE_BEGIN
17
18class BVHOptiX;
19struct KernelParamsOptiX;
20
21/* List of OptiX program groups. */
22enum {
23 /* Ray generation */
24 PG_RGEN_INTERSECT_CLOSEST,
25 PG_RGEN_INTERSECT_SHADOW,
26 PG_RGEN_INTERSECT_SUBSURFACE,
27 PG_RGEN_INTERSECT_VOLUME_STACK,
28 PG_RGEN_INTERSECT_DEDICATED_LIGHT,
29 PG_RGEN_SHADE_BACKGROUND,
30 PG_RGEN_SHADE_LIGHT,
31 PG_RGEN_SHADE_SURFACE,
32 PG_RGEN_SHADE_SURFACE_RAYTRACE,
33 PG_RGEN_SHADE_SURFACE_MNEE,
34 PG_RGEN_SHADE_VOLUME,
35 PG_RGEN_SHADE_VOLUME_RAY_MARCHING,
36 PG_RGEN_SHADE_SHADOW,
37 PG_RGEN_SHADE_DEDICATED_LIGHT,
38 PG_RGEN_EVAL_DISPLACE,
39 PG_RGEN_EVAL_BACKGROUND,
40 PG_RGEN_EVAL_CURVE_SHADOW_TRANSPARENCY,
41 PG_RGEN_INIT_FROM_CAMERA,
42 PG_RGEN_EVAL_VOLUME_DENSITY,
43
44 /* Miss */
45 PG_MISS,
46
47 /* Hit */
48 PG_HITD, /* Default hit group. */
49 PG_HITS, /* __SHADOW_RECORD_ALL__ hit group. */
50 PG_HITL, /* __BVH_LOCAL__ hit group (only used for triangles). */
51 PG_HITV, /* __VOLUME__ hit group. */
52 PG_HITD_MOTION,
53 PG_HITS_MOTION,
54 PG_HITL_MOTION,
55 PG_HITV_MOTION,
56 PG_HITD_CURVE_LINEAR,
57 PG_HITS_CURVE_LINEAR,
58 PG_HITV_CURVE_LINEAR,
59 PG_HITL_CURVE_LINEAR,
60 PG_HITD_CURVE_LINEAR_MOTION,
61 PG_HITS_CURVE_LINEAR_MOTION,
62 PG_HITV_CURVE_LINEAR_MOTION,
63 PG_HITL_CURVE_LINEAR_MOTION,
64 PG_HITD_CURVE_RIBBON,
65 PG_HITS_CURVE_RIBBON,
66 PG_HITV_CURVE_RIBBON,
67 PG_HITL_CURVE_RIBBON,
68 PG_HITD_POINTCLOUD,
69 PG_HITS_POINTCLOUD,
70 PG_HITV_POINTCLOUD,
71 PG_HITL_POINTCLOUD,
72
73 /* Callable */
74 PG_CALL_SVM_AO,
75 PG_CALL_SVM_BEVEL,
76
77 NUM_PROGRAM_GROUPS
78};
79
80static const int MISS_PROGRAM_GROUP_OFFSET = PG_MISS;
81static const int NUM_MISS_PROGRAM_GROUPS = 1;
82static const int HIT_PROGAM_GROUP_OFFSET = PG_HITD;
83static const int NUM_HIT_PROGRAM_GROUPS = 24;
84static const int CALLABLE_PROGRAM_GROUPS_BASE = PG_CALL_SVM_AO;
85static const int NUM_CALLABLE_PROGRAM_GROUPS = 2;
86
87/* List of OptiX pipelines. */
88enum { PIP_SHADE, PIP_INTERSECT, NUM_PIPELINES };
89
90/* A single shader binding table entry. */
91struct SbtRecord {
92 char header[OPTIX_SBT_RECORD_HEADER_SIZE];
93};
94
95class OptiXDevice : public CUDADevice {
96 public:
97 OptixDeviceContext context = nullptr;
98
99 OptixModule optix_module = nullptr; /* All necessary OptiX kernels are in one module. */
100 OptixModule builtin_modules[4] = {};
101 OptixPipeline pipelines[NUM_PIPELINES] = {};
102 OptixProgramGroup groups[NUM_PROGRAM_GROUPS] = {};
103 OptixPipelineCompileOptions pipeline_options = {};
104
105# ifdef WITH_OSL
106 OSLGlobals osl_globals;
107 vector<OptixModule> osl_modules;
108 vector<OptixProgramGroup> osl_groups;
109 OptixModule osl_camera_module = nullptr;
110 device_vector<uint8_t> osl_colorsystem;
111# endif
112
113 device_vector<SbtRecord> sbt_data;
114 device_only_memory<KernelParamsOptiX> launch_params;
115
116 private:
117 OptixTraversableHandle tlas_handle = 0;
118 vector<unique_ptr<device_only_memory<char>>> delayed_free_bvh_memory;
119 thread_mutex delayed_free_bvh_mutex;
120
121 public:
122 OptiXDevice(const DeviceInfo &info, Stats &stats, Profiler &profiler, bool headless);
123 ~OptiXDevice() override;
124
125 BVHLayoutMask get_bvh_layout_mask(uint /*kernel_features*/) const override;
126
127 string compile_kernel_get_common_cflags(const uint kernel_features);
128
129 void create_optix_module(TaskPool &pool,
130 OptixModuleCompileOptions &module_options,
131 string &ptx_data,
132 OptixModule &module,
133 OptixResult &failure_reason);
134
135 bool load_kernels(const uint kernel_features) override;
136
137 bool load_osl_kernels() override;
138
139 bool build_optix_bvh(BVHOptiX *bvh,
140 OptixBuildOperation operation,
141 const OptixBuildInput &build_input,
142 uint16_t num_motion_steps);
143
144 void build_bvh(BVH *bvh, Progress &progress, bool refit) override;
145
146 void release_bvh(BVH *bvh) override;
147 void free_bvh_memory_delayed();
148
149 void const_copy_to(const char *name, void *host, const size_t size) override;
150
151 void update_launch_params(const size_t offset, void *data, const size_t data_size);
152
153 unique_ptr<DeviceQueue> gpu_queue_create() override;
154
155 OSLGlobals *get_cpu_osl_memory() override;
156};
157
158CCL_NAMESPACE_END
159
160#endif /* WITH_OPTIX */
uint
unsigned int uint
Definition BLI_sys_types.h:64
TaskPool
struct TaskPool TaskPool
Definition BLI_task.h:56
ButType::Progress
@ Progress
Definition UI_interface_c.hh:456
data
BMesh const char void * data
Definition bmesh_iterators_inline.hh:37
size
static DBVT_INLINE btScalar size(const btDbvtVolume &a)
Definition btDbvt.cpp:52
refit
void refit(btStridingMeshInterface *triangles, const btVector3 &aabbMin, const btVector3 &aabbMax)
device_impl.h
CCL_NAMESPACE_END
#define CCL_NAMESPACE_END
Definition device/cuda/compat.h:10
BVH
ccl_device bool BVH_FUNCTION_FULL_NAME BVH(KernelGlobals kg, const ccl_private Ray *ray, ccl_private LocalIntersection *local_isect, const int local_object, ccl_private uint *lcg_state, const int max_hits)
Definition local.h:28
CCL_NAMESPACE_BEGIN
Definition python.cpp:37
blender::ed::asset::shelf::context
int context(const bContext *C, const char *member, bContextDataResult *result)
Definition asset_shelf.cc:694
BVHLayoutMask
int BVHLayoutMask
Definition params.h:50
module
static struct PyModuleDef module
Definition python.cpp:796
name
const char * name
Definition python_compat.hh:32
thread_mutex
std::mutex thread_mutex
Definition thread.h:27
Generated on for Blender by  doxygen 1.16.1