CUDAStdArray.hpp

Go to the documentation of this file.

// This code is based on Jet framework.
// Copyright (c) 2018 Doyub Kim
// CubbyFlow is voxel-based fluid simulation engine for computer games.
// Copyright (c) 2020 CubbyFlow Team
// Core Part: Chris Ohk, Junwoo Hwang, Jihong Sin, Seungwoo Yoo
// AI Part: Dongheon Cho, Minseo Kim
// We are making my contributions/submissions to this project solely in our
// personal capacity and are not conveying any rights to any intellectual
// property of any third parties.

#ifndef CUBBYFLOW_CUDA_STD_ARRAY_HPP
#define CUBBYFLOW_CUDA_STD_ARRAY_HPP

#ifdef CUBBYFLOW_USE_CUDA

#include <Core/Matrix/Matrix.hpp>

#include <array>

namespace CubbyFlow
{
template <typename T, size_t N>
class CUDAStdArray
{
 public:
    using ValueType = T;
    using Reference = T&;
    using ConstReference = const T&;
    using Pointer = T*;
    using ConstPointer = const T*;
    using Iterator = Pointer;
    using ConstIterator = ConstPointer;

    CUBBYFLOW_CUDA_HOST_DEVICE CUDAStdArray();

    template <typename... Args>
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAStdArray(ConstReference first, Args... rest);

    CUBBYFLOW_CUDA_HOST CUDAStdArray(const std::array<T, N>& other);

    CUBBYFLOW_CUDA_HOST CUDAStdArray(const Vector<T, N>& other);

    ~CUDAStdArray() = default;

    CUBBYFLOW_CUDA_HOST_DEVICE CUDAStdArray(const CUDAStdArray& other);

    CUBBYFLOW_CUDA_HOST_DEVICE CUDAStdArray(CUDAStdArray&& other) noexcept;

    CUBBYFLOW_CUDA_HOST_DEVICE CUDAStdArray& operator=(
        const CUDAStdArray& other);

    CUBBYFLOW_CUDA_HOST_DEVICE CUDAStdArray& operator=(
        CUDAStdArray&& other) noexcept;

    CUBBYFLOW_CUDA_HOST_DEVICE void Fill(ConstReference val);

    CUBBYFLOW_CUDA_HOST Vector<T, N> ToVector() const;

    CUBBYFLOW_CUDA_HOST_DEVICE Reference operator[](size_t i);

    CUBBYFLOW_CUDA_HOST_DEVICE ConstReference operator[](size_t i) const;

    CUBBYFLOW_CUDA_HOST_DEVICE bool operator==(const CUDAStdArray& other) const;

    CUBBYFLOW_CUDA_HOST_DEVICE bool operator!=(const CUDAStdArray& other) const;

 private:
    template <typename... Args>
    CUBBYFLOW_CUDA_HOST_DEVICE void SetAt(size_t i, ConstReference first,
                                          Args... rest);

    template <typename... Args>
    CUBBYFLOW_CUDA_HOST_DEVICE void SetAt(size_t i, ConstReference first);

    T m_elements[N];
};
}  // namespace CubbyFlow

#include <Core/CUDA/CUDAStdArray-Impl.hpp>

#endif

#endif