1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

use crate::partitioning::{QbvhNode, QbvhProxy};
use crate::utils::{Array1, DefaultStorage};

#[cfg(all(feature = "std", feature = "cuda"))]
use crate::utils::CudaArray1;
#[cfg(feature = "cuda")]
use crate::utils::{CudaArrayPointer1, CudaStorage, CudaStoragePtr};

/// Trait describing all the types needed for storing a Qbvh’s data.
pub trait QbvhStorage<LeafData> {
    /// Type of the array containing the Qbvh nodes.
    type Nodes: Array1<QbvhNode>;
    /// Type of an array containing u32.
    type ArrayU32: Array1<u32>;
    /// Type of the array containing the Qbvh leaves.
    type ArrayProxies: Array1<QbvhProxy<LeafData>>;
}

#[cfg(feature = "std")]
impl<LeafData> QbvhStorage<LeafData> for DefaultStorage {
    type Nodes = Vec<QbvhNode>;
    type ArrayU32 = Vec<u32>;
    type ArrayProxies = Vec<QbvhProxy<LeafData>>;
}

#[cfg(all(feature = "std", feature = "cuda"))]
impl<LeafData: cust_core::DeviceCopy> QbvhStorage<LeafData> for CudaStorage {
    type Nodes = CudaArray1<QbvhNode>;
    type ArrayU32 = CudaArray1<u32>;
    type ArrayProxies = CudaArray1<QbvhProxy<LeafData>>;
}

#[cfg(feature = "cuda")]
impl<LeafData: cust_core::DeviceCopy> QbvhStorage<LeafData> for CudaStoragePtr {
    type Nodes = CudaArrayPointer1<QbvhNode>;
    type ArrayU32 = CudaArrayPointer1<u32>;
    type ArrayProxies = CudaArrayPointer1<QbvhProxy<LeafData>>;
}