Differential D12367 Diff 41308 intern/cycles/kernel/svm/svm_ao.h

Changeset View

Standalone View

intern/cycles/kernel/svm/svm_ao.h

	/*			/*
	* Copyright 2011-2018 Blender Foundation			* Copyright 2011-2018 Blender Foundation
	*			*
	* Licensed under the Apache License, Version 2.0 (the "License");			* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.			* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at			* You may obtain a copy of the License at
	*			*
	* http://www.apache.org/licenses/LICENSE-2.0			* http://www.apache.org/licenses/LICENSE-2.0
	*			*
	* Unless required by applicable law or agreed to in writing, software			* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,			* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.			* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and			* See the License for the specific language governing permissions and
	* limitations under the License.			* limitations under the License.
	*/			*/

				#include "kernel/bvh/bvh.h"

	CCL_NAMESPACE_BEGIN			CCL_NAMESPACE_BEGIN

	#ifdef __SHADER_RAYTRACE__			#ifdef __SHADER_RAYTRACE__

	ccl_device_noinline float svm_ao(KernelGlobals *kg,			ccl_device float svm_ao(INTEGRATOR_STATE_CONST_ARGS,
	ShaderData *sd,			ShaderData *sd,
	float3 N,			float3 N,
	ccl_addr_space PathState *state,
	float max_dist,			float max_dist,
	int num_samples,			int num_samples,
	int flags)			int flags)
	{			{
	if (flags & NODE_AO_GLOBAL_RADIUS) {			if (flags & NODE_AO_GLOBAL_RADIUS) {
	max_dist = kernel_data.background.ao_distance;			max_dist = kernel_data.integrator.ao_bounces_distance;
	}			}

	/* Early out if no sampling needed. */			/* Early out if no sampling needed. */
	if (max_dist <= 0.0f \|\| num_samples < 1 \|\| sd->object == OBJECT_NONE) {			if (max_dist <= 0.0f \|\| num_samples < 1 \|\| sd->object == OBJECT_NONE) {
	return 1.0f;			return 1.0f;
	}			}

	/* Can't raytrace from shaders like displacement, before BVH exists. */			/* Can't raytrace from shaders like displacement, before BVH exists. */
	if (kernel_data.bvh.bvh_layout == BVH_LAYOUT_NONE) {			if (kernel_data.bvh.bvh_layout == BVH_LAYOUT_NONE) {
	return 1.0f;			return 1.0f;
	}			}

	if (flags & NODE_AO_INSIDE) {			if (flags & NODE_AO_INSIDE) {
	N = -N;			N = -N;
	}			}

	float3 T, B;			float3 T, B;
	make_orthonormals(N, &T, &B);			make_orthonormals(N, &T, &B);

				/* TODO: support ray-tracing in shadow shader evaluation? */
				RNGState rng_state;
				path_state_rng_load(INTEGRATOR_STATE_PASS, &rng_state);

	int unoccluded = 0;			int unoccluded = 0;
	for (int sample = 0; sample < num_samples; sample++) {			for (int sample = 0; sample < num_samples; sample++) {
	float disk_u, disk_v;			float disk_u, disk_v;
	path_branched_rng_2D(			path_branched_rng_2D(kg, &rng_state, sample, num_samples, PRNG_BEVEL_U, &disk_u, &disk_v);
	kg, state->rng_hash, state, sample, num_samples, PRNG_BEVEL_U, &disk_u, &disk_v);

	float2 d = concentric_sample_disk(disk_u, disk_v);			float2 d = concentric_sample_disk(disk_u, disk_v);
	float3 D = make_float3(d.x, d.y, safe_sqrtf(1.0f - dot(d, d)));			float3 D = make_float3(d.x, d.y, safe_sqrtf(1.0f - dot(d, d)));

	/* Create ray. */			/* Create ray. */
	Ray ray;			Ray ray;
	ray.P = ray_offset(sd->P, N);			ray.P = ray_offset(sd->P, N);
	ray.D = D.x * T + D.y * B + D.z * N;			ray.D = D.x * T + D.y * B + D.z * N;
	ray.t = max_dist;			ray.t = max_dist;
	ray.time = sd->time;			ray.time = sd->time;
	ray.dP = sd->dP;			ray.dP = differential_zero_compact();
	ray.dD = differential3_zero();			ray.dD = differential_zero_compact();

	if (flags & NODE_AO_ONLY_LOCAL) {			if (flags & NODE_AO_ONLY_LOCAL) {
	if (!scene_intersect_local(kg, &ray, NULL, sd->object, NULL, 0)) {			if (!scene_intersect_local(kg, &ray, NULL, sd->object, NULL, 0)) {
	unoccluded++;			unoccluded++;
	}			}
	}			}
	else {			else {
	Intersection isect;			Intersection isect;
	if (!scene_intersect(kg, &ray, PATH_RAY_SHADOW_OPAQUE, &isect)) {			if (!scene_intersect(kg, &ray, PATH_RAY_SHADOW_OPAQUE, &isect)) {
	unoccluded++;			unoccluded++;
	}			}
	}			}
	}			}

	return ((float)unoccluded) / num_samples;			return ((float)unoccluded) / num_samples;
	}			}

	ccl_device void svm_node_ao(			# if defined(__KERNEL_OPTIX__)
	KernelGlobals kg, ShaderData sd, ccl_addr_space PathState state, float stack, uint4 node)			ccl_device_inline
				# else
				ccl_device_noinline
				# endif
				void
				svm_node_ao(INTEGRATOR_STATE_CONST_ARGS, ShaderData sd, float stack, uint4 node)
				{
				# if defined(__KERNEL_OPTIX__)
				optixDirectCall<void>(0, INTEGRATOR_STATE_PASS, sd, stack, node);
				}

				extern "C" __device__ void __direct_callable__svm_node_ao(INTEGRATOR_STATE_CONST_ARGS,
				ShaderData *sd,
				float *stack,
				uint4 node)
	{			{
				# endif
	uint flags, dist_offset, normal_offset, out_ao_offset;			uint flags, dist_offset, normal_offset, out_ao_offset;
	svm_unpack_node_uchar4(node.y, &flags, &dist_offset, &normal_offset, &out_ao_offset);			svm_unpack_node_uchar4(node.y, &flags, &dist_offset, &normal_offset, &out_ao_offset);

	uint color_offset, out_color_offset, samples;			uint color_offset, out_color_offset, samples;
	svm_unpack_node_uchar3(node.z, &color_offset, &out_color_offset, &samples);			svm_unpack_node_uchar3(node.z, &color_offset, &out_color_offset, &samples);

	float dist = stack_load_float_default(stack, dist_offset, node.w);			float dist = stack_load_float_default(stack, dist_offset, node.w);
	float3 normal = stack_valid(normal_offset) ? stack_load_float3(stack, normal_offset) : sd->N;			float3 normal = stack_valid(normal_offset) ? stack_load_float3(stack, normal_offset) : sd->N;
	float ao = svm_ao(kg, sd, normal, state, dist, samples, flags);			float ao = svm_ao(INTEGRATOR_STATE_PASS, sd, normal, dist, samples, flags);

	if (stack_valid(out_ao_offset)) {			if (stack_valid(out_ao_offset)) {
	stack_store_float(stack, out_ao_offset, ao);			stack_store_float(stack, out_ao_offset, ao);
	}			}

	if (stack_valid(out_color_offset)) {			if (stack_valid(out_color_offset)) {
	float3 color = stack_load_float3(stack, color_offset);			float3 color = stack_load_float3(stack, color_offset);
	stack_store_float3(stack, out_color_offset, ao * color);			stack_store_float3(stack, out_color_offset, ao * color);
	}			}
	}			}

	#endif /* __SHADER_RAYTRACE__ */			#endif /* __SHADER_RAYTRACE__ */

	CCL_NAMESPACE_END			CCL_NAMESPACE_END