CUDAArrayView.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_ARRAY_VIEW_HPP
#define CUBBYFLOW_CUDA_ARRAY_VIEW_HPP
 
#ifdef CUBBYFLOW_USE_CUDA
 
#include <Core/CUDA/CUDAArray.hpp>
 
namespace CubbyFlow
{
template <typename T, size_t N>
class CUDAArrayView final : public CUDAArrayBase<T, N, CUDAArrayView<T, N>>
{
    using Base = CUDAArrayBase<T, N, CUDAArrayView<T, N>>;
    using Base::At;
    using Base::m_size;
    using Base::SetPtrAndSize;
 
 public:
    using Base::data;
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView();
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView(
        T* ptr, const CUDAStdArray<size_t, N>& size);
 
    template <size_t M = N>
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView(
        typename std::enable_if<(M == 1), T>::type* ptr, size_t size);
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView(CUDAArray<T, N>& other);
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView(const CUDAArrayView& other);
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView(CUDAArrayView&& other) noexcept;
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView& operator=(
        const CUDAArrayView& other);
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView& operator=(
        CUDAArrayView&& other) noexcept;
 
    CUBBYFLOW_CUDA_HOST_DEVICE void Set(CUDAArray<T, N>& other);
 
    CUBBYFLOW_CUDA_HOST_DEVICE void Set(const CUDAArrayView& other);
 
    CUBBYFLOW_CUDA_HOST void Fill(const T& val);
};
 
template <typename T, size_t N>
class CUDAArrayView<const T, N> final
    : public CUDAArrayBase<const T, N, CUDAArrayView<const T, N>>
{
    using Base = CUDAArrayBase<const T, N, CUDAArrayView<const T, N>>;
    using Base::m_size;
    using Base::SetPtrAndSize;
 
 public:
    using Base::data;
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView();
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView(
        const T* ptr, const CUDAStdArray<size_t, N>& size);
 
    template <size_t M = N>
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView(
        const typename std::enable_if<(M == 1), T>::type* ptr, size_t size);
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView(const CUDAArray<T, N>& other);
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView(const CUDAArrayView<T, N>& other);
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView(const CUDAArrayView& other);
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView(CUDAArrayView&&) noexcept;
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView& operator=(
        const CUDAArrayView<T, N>& other);
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView& operator=(
        const CUDAArrayView& other);
 
    CUBBYFLOW_CUDA_HOST_DEVICE CUDAArrayView& operator=(
        CUDAArrayView&& other) noexcept;
 
    CUBBYFLOW_CUDA_HOST_DEVICE void Set(const CUDAArray<T, N>& other);
 
    CUBBYFLOW_CUDA_HOST_DEVICE void Set(const CUDAArrayView<T, N>& other);
 
    CUBBYFLOW_CUDA_HOST_DEVICE void Set(const CUDAArrayView& other);
};
 
template <class T>
using CUDAArrayView1 = CUDAArrayView<T, 1>;
 
template <class T>
using CUDAArrayView2 = CUDAArrayView<T, 2>;
 
template <class T>
using CUDAArrayView3 = CUDAArrayView<T, 3>;
 
template <class T>
using CUDAArrayView4 = CUDAArrayView<T, 4>;
 
template <class T>
using ConstCUDAArrayView1 = CUDAArrayView<const T, 1>;
 
template <class T>
using ConstCUDAArrayView2 = CUDAArrayView<const T, 2>;
 
template <class T>
using ConstCUDAArrayView3 = CUDAArrayView<const T, 3>;
 
template <class T>
using ConstCUDAArrayView4 = CUDAArrayView<const T, 4>;
}  // namespace CubbyFlow
 
#include <Core/CUDA/CUDAArrayView-Impl.hpp>
 
#endif
 
#endif