// Copyright 2009-2021 Intel Corporation // SPDX-License-Identifier: Apache-2.0 #pragma once #include "rtcore.h" namespace embree { namespace instance_id_stack { static_assert(RTC_MAX_INSTANCE_LEVEL_COUNT > 0, "RTC_MAX_INSTANCE_LEVEL_COUNT must be greater than 0."); /******************************************************************************* * Instance ID stack manipulation. * This is used from the instance intersector. ******************************************************************************/ /* * Push an instance to the stack. */ RTC_FORCEINLINE bool push(RTCIntersectContext* context, unsigned instanceId) { #if RTC_MAX_INSTANCE_LEVEL_COUNT > 1 const bool spaceAvailable = context->instStackSize < RTC_MAX_INSTANCE_LEVEL_COUNT; /* We assert here because instances are silently dropped when the stack is full. This might be quite hard to find in production. */ assert(spaceAvailable); if (likely(spaceAvailable)) context->instID[context->instStackSize++] = instanceId; return spaceAvailable; #else const bool spaceAvailable = (context->instID[0] == RTC_INVALID_GEOMETRY_ID); assert(spaceAvailable); if (likely(spaceAvailable)) context->instID[0] = instanceId; return spaceAvailable; #endif } /* * Pop the last instance pushed to the stack. * Do not call on an empty stack. */ RTC_FORCEINLINE void pop(RTCIntersectContext* context) { assert(context); #if RTC_MAX_INSTANCE_LEVEL_COUNT > 1 assert(context->instStackSize > 0); context->instID[--context->instStackSize] = RTC_INVALID_GEOMETRY_ID; #else assert(context->instID[0] != RTC_INVALID_GEOMETRY_ID); context->instID[0] = RTC_INVALID_GEOMETRY_ID; #endif } /* * Optimized instance id stack copy. * The copy() functions will either copy full * stacks or copy only until the last valid element has been copied, depending * on RTC_MAX_INSTANCE_LEVEL_COUNT. */ RTC_FORCEINLINE void copy_UU(const unsigned* src, unsigned* tgt) { #if (RTC_MAX_INSTANCE_LEVEL_COUNT == 1) tgt[0] = src[0]; #else for (unsigned l = 0; l < RTC_MAX_INSTANCE_LEVEL_COUNT; ++l) { tgt[l] = src[l]; if (RTC_MAX_INSTANCE_LEVEL_COUNT > 4) if (src[l] == RTC_INVALID_GEOMETRY_ID) break; } #endif } template <int K> RTC_FORCEINLINE void copy_UV(const unsigned* src, vuint<K>* tgt) { #if (RTC_MAX_INSTANCE_LEVEL_COUNT == 1) tgt[0] = src[0]; #else for (unsigned l = 0; l < RTC_MAX_INSTANCE_LEVEL_COUNT; ++l) { tgt[l] = src[l]; if (RTC_MAX_INSTANCE_LEVEL_COUNT > 4) if (src[l] == RTC_INVALID_GEOMETRY_ID) break; } #endif } template <int K> RTC_FORCEINLINE void copy_UV(const unsigned* src, vuint<K>* tgt, size_t j) { #if (RTC_MAX_INSTANCE_LEVEL_COUNT == 1) tgt[0][j] = src[0]; #else for (unsigned l = 0; l < RTC_MAX_INSTANCE_LEVEL_COUNT; ++l) { tgt[l][j] = src[l]; if (RTC_MAX_INSTANCE_LEVEL_COUNT > 4) if (src[l] == RTC_INVALID_GEOMETRY_ID) break; } #endif } template <int K> RTC_FORCEINLINE void copy_UV(const unsigned* src, vuint<K>* tgt, const vbool<K>& mask) { #if (RTC_MAX_INSTANCE_LEVEL_COUNT == 1) vuint<K>::store(mask, tgt, src[0]); #else for (unsigned l = 0; l < RTC_MAX_INSTANCE_LEVEL_COUNT; ++l) { vuint<K>::store(mask, tgt + l, src[l]); if (RTC_MAX_INSTANCE_LEVEL_COUNT > 4) if (src[l] == RTC_INVALID_GEOMETRY_ID) break; } #endif } template <int K> RTC_FORCEINLINE void copy_VU(const vuint<K>* src, unsigned* tgt, size_t i) { #if (RTC_MAX_INSTANCE_LEVEL_COUNT == 1) tgt[0] = src[0][i]; #else for (unsigned l = 0; l < RTC_MAX_INSTANCE_LEVEL_COUNT; ++l) { tgt[l] = src[l][i]; if (RTC_MAX_INSTANCE_LEVEL_COUNT > 4) if (src[l][i] == RTC_INVALID_GEOMETRY_ID) break; } #endif } template <int K> RTC_FORCEINLINE void copy_VV(const vuint<K>* src, vuint<K>* tgt, size_t i, size_t j) { #if (RTC_MAX_INSTANCE_LEVEL_COUNT == 1) tgt[0][j] = src[0][i]; #else for (unsigned l = 0; l < RTC_MAX_INSTANCE_LEVEL_COUNT; ++l) { tgt[l][j] = src[l][i]; if (RTC_MAX_INSTANCE_LEVEL_COUNT > 4) if (src[l][i] == RTC_INVALID_GEOMETRY_ID) break; } #endif } template <int K> RTC_FORCEINLINE void copy_VV(const vuint<K>* src, vuint<K>* tgt, const vbool<K>& mask) { #if (RTC_MAX_INSTANCE_LEVEL_COUNT == 1) vuint<K>::store(mask, tgt, src[0]); #else vbool<K> done = !mask; for (unsigned l = 0; l < RTC_MAX_INSTANCE_LEVEL_COUNT; ++l) { vuint<K>::store(mask, tgt + l, src[l]); if (RTC_MAX_INSTANCE_LEVEL_COUNT > 4) { done |= src[l] == RTC_INVALID_GEOMETRY_ID; if (all(done)) break; } } #endif } } // namespace instance_id_stack } // namespace embree