#[cfg(target_arch = "spirv")]
use crate::integer::Integer;
use crate::{
integer::{SignedInteger, UnsignedInteger},
scalar::Scalar,
vector::Vector,
};
#[cfg(target_arch = "spirv")]
use core::arch::asm;
mod atomics;
mod barrier;
mod demote_to_helper_invocation_ext;
mod derivative;
mod primitive;
mod ray_tracing;
mod subgroup;
pub use atomics::*;
pub use barrier::*;
pub use demote_to_helper_invocation_ext::*;
pub use derivative::*;
pub use primitive::*;
pub use ray_tracing::*;
pub use subgroup::*;
#[spirv_std_macros::gpu_only]
#[doc(alias = "OpAny")]
#[inline]
pub fn any<V: Vector<bool, N>, const N: usize>(vector: V) -> bool {
let mut result = false;
unsafe {
asm! {
"%bool = OpTypeBool",
"%vector = OpLoad _ {vector}",
"%result = OpAny %bool %vector",
"OpStore {result} %result",
vector = in(reg) &vector,
result = in(reg) &mut result
}
}
result
}
#[spirv_std_macros::gpu_only]
#[doc(alias = "OpAll")]
#[inline]
pub fn all<V: Vector<bool, N>, const N: usize>(vector: V) -> bool {
let mut result = false;
unsafe {
asm! {
"%bool = OpTypeBool",
"%vector = OpLoad _ {vector}",
"%result = OpAll %bool %vector",
"OpStore {result} %result",
vector = in(reg) &vector,
result = in(reg) &mut result
}
}
result
}
#[spirv_std_macros::gpu_only]
#[doc(alias = "OpVectorExtractDynamic")]
#[inline]
pub unsafe fn vector_extract_dynamic<T: Scalar, const N: usize>(
vector: impl Vector<T, N>,
index: usize,
) -> T {
let mut result = T::default();
asm! {
"%vector = OpLoad _ {vector}",
"%element = OpVectorExtractDynamic _ %vector {index}",
"OpStore {element} %element",
vector = in(reg) &vector,
index = in(reg) index,
element = in(reg) &mut result
}
result
}
#[spirv_std_macros::gpu_only]
#[doc(alias = "OpVectorInsertDynamic")]
#[inline]
pub unsafe fn vector_insert_dynamic<T: Scalar, V: Vector<T, N>, const N: usize>(
vector: V,
index: usize,
element: T,
) -> V {
let mut result = V::default();
asm! {
"%vector = OpLoad _ {vector}",
"%element = OpLoad _ {element}",
"%new_vector = OpVectorInsertDynamic _ %vector %element {index}",
"OpStore {result} %new_vector",
vector = in(reg) &vector,
index = in(reg) index,
element = in(reg) &element,
result = in(reg) &mut result,
}
result
}
#[spirv_std_macros::gpu_only]
#[doc(alias = "OpKill", alias = "discard")]
#[allow(clippy::empty_loop)]
pub fn kill() -> ! {
unsafe { asm!("OpKill", options(noreturn)) }
}
#[cfg(all(
target_feature = "Int64",
target_feature = "ShaderClockKHR",
target_feature = "ext:SPV_KHR_shader_clock"
))]
#[spirv_std_macros::gpu_only]
#[doc(alias = "OpReadClockKHR")]
pub unsafe fn read_clock_khr<const SCOPE: u32>() -> u64 {
let mut result: u64;
asm! {
"%uint = OpTypeInt 32 0",
"%scope = OpConstant %uint {scope}",
"{result} = OpReadClockKHR typeof*{result} %scope",
result = out(reg) result,
scope = const SCOPE,
};
result
}
#[cfg(all(
target_feature = "ShaderClockKHR",
target_feature = "ext:SPV_KHR_shader_clock"
))]
#[spirv_std_macros::gpu_only]
#[doc(alias = "OpReadClockKHR")]
pub unsafe fn read_clock_uvec2_khr<V: Vector<u32, 2>, const SCOPE: u32>() -> V {
let mut result = V::default();
asm! {
"%uint = OpTypeInt 32 0",
"%scope = OpConstant %uint {scope}",
"%result = OpReadClockKHR typeof*{result} %scope",
"OpStore {result} %result",
result = in(reg) &mut result,
scope = const SCOPE,
};
result
}
#[cfg(target_arch = "spirv")]
unsafe fn call_glsl_op_with_ints<T: Integer, const OP: u32>(a: T, b: T) -> T {
let mut result = T::default();
asm!(
"%glsl = OpExtInstImport \"GLSL.std.450\"",
"%a = OpLoad _ {a}",
"%b = OpLoad _ {b}",
"%result = OpExtInst typeof*{result} %glsl {op} %a %b",
"OpStore {result} %result",
a = in(reg) &a,
b = in(reg) &b,
result = in(reg) &mut result,
op = const OP
);
result
}
#[spirv_std_macros::gpu_only]
pub fn unsigned_min<T: UnsignedInteger>(a: T, b: T) -> T {
unsafe { call_glsl_op_with_ints::<_, 38>(a, b) }
}
#[spirv_std_macros::gpu_only]
pub fn unsigned_max<T: UnsignedInteger>(a: T, b: T) -> T {
unsafe { call_glsl_op_with_ints::<_, 41>(a, b) }
}
#[spirv_std_macros::gpu_only]
pub fn signed_min<T: SignedInteger>(a: T, b: T) -> T {
unsafe { call_glsl_op_with_ints::<_, 39>(a, b) }
}
#[spirv_std_macros::gpu_only]
pub fn signed_max<T: SignedInteger>(a: T, b: T) -> T {
unsafe { call_glsl_op_with_ints::<_, 42>(a, b) }
}
pub trait IndexUnchecked<T> {
unsafe fn index_unchecked(&self, index: usize) -> &T;
unsafe fn index_unchecked_mut(&mut self, index: usize) -> &mut T;
}
impl<T> IndexUnchecked<T> for [T] {
#[cfg(target_arch = "spirv")]
unsafe fn index_unchecked(&self, index: usize) -> &T {
let mut result_slot = core::mem::MaybeUninit::uninit();
asm! {
"%slice_ptr = OpLoad _ {slice_ptr_ptr}",
"%data_ptr = OpCompositeExtract _ %slice_ptr 0",
"%result = OpAccessChain _ %data_ptr {index}",
"OpStore {result_slot} %result",
slice_ptr_ptr = in(reg) &self,
index = in(reg) index,
result_slot = in(reg) result_slot.as_mut_ptr(),
}
result_slot.assume_init()
}
#[cfg(not(target_arch = "spirv"))]
unsafe fn index_unchecked(&self, index: usize) -> &T {
self.get_unchecked(index)
}
#[cfg(target_arch = "spirv")]
unsafe fn index_unchecked_mut(&mut self, index: usize) -> &mut T {
let mut result_slot = core::mem::MaybeUninit::uninit();
asm! {
"%slice_ptr = OpLoad _ {slice_ptr_ptr}",
"%data_ptr = OpCompositeExtract _ %slice_ptr 0",
"%result = OpAccessChain _ %data_ptr {index}",
"OpStore {result_slot} %result",
slice_ptr_ptr = in(reg) &self,
index = in(reg) index,
result_slot = in(reg) result_slot.as_mut_ptr(),
}
result_slot.assume_init()
}
#[cfg(not(target_arch = "spirv"))]
unsafe fn index_unchecked_mut(&mut self, index: usize) -> &mut T {
self.get_unchecked_mut(index)
}
}
impl<T, const N: usize> IndexUnchecked<T> for [T; N] {
#[cfg(target_arch = "spirv")]
unsafe fn index_unchecked(&self, index: usize) -> &T {
let mut result_slot = core::mem::MaybeUninit::uninit();
asm! {
"%result = OpAccessChain _ {array_ptr} {index}",
"OpStore {result_slot} %result",
array_ptr = in(reg) self,
index = in(reg) index,
result_slot = in(reg) result_slot.as_mut_ptr(),
}
result_slot.assume_init()
}
#[cfg(not(target_arch = "spirv"))]
unsafe fn index_unchecked(&self, index: usize) -> &T {
self.get_unchecked(index)
}
#[cfg(target_arch = "spirv")]
unsafe fn index_unchecked_mut(&mut self, index: usize) -> &mut T {
let mut result_slot = core::mem::MaybeUninit::uninit();
asm! {
"%result = OpAccessChain _ {array_ptr} {index}",
"OpStore {result_slot} %result",
array_ptr = in(reg) self,
index = in(reg) index,
result_slot = in(reg) result_slot.as_mut_ptr(),
}
result_slot.assume_init()
}
#[cfg(not(target_arch = "spirv"))]
unsafe fn index_unchecked_mut(&mut self, index: usize) -> &mut T {
self.get_unchecked_mut(index)
}
}