Array3D.hpp

// INFINITY_API_PUBLIC
 
#pragma once
 
#include <Infinity/Types/Containers/Array.hpp>
#include <Infinity/Types/Math/Vector2.hpp>
#include <Infinity/Types/Math/Vector3.hpp>
#include <Infinity/Types/Math/Vector4.hpp>
 
#include <vector>
#include <cassert>
 
namespace Infinity::IO::Data
{
    template<typename T>
    class Array3DSerializer;
}
 
namespace Infinity::Types::Containers
{
 
    template<typename T>
    class INFINITY_API_TEMPLATE Array3D : public Array<T>
    {
        public:
            Array3D() : Array<T>(), _width(0), _height(0), _depth(0) { }
            
            Array3D(size_t width, size_t height, size_t depth) : Array<T>(width * height * depth), _width(width), _height(height), _depth(depth) { }
            
            Array3D(size_t width, size_t height, size_t depth, Array<T> data) : Array<T>(data), _width(width), _height(height), _depth(depth) { }
            
            Array3D(size_t width, size_t height, size_t depth, T* data) : Array<T>(data, width * height * depth), _width(width), _height(height), _depth(depth) { }
            
            Array3D(size_t width, size_t height, size_t depth, const std::vector<T>& data) : Array<T>(data), _width(width), _height(height), _depth(depth) { }
 
            virtual ~Array3D() { }
 
            std::unique_ptr<Core::Base> clone() const override
            {
                return std::make_unique<Array3D<T>>(*this);
            }
 
            const Infinity::Types::TypeID& typeId() const override { return Infinity::Types::getTypeID<Array3D<T>>(); }
 
            inline size_t width() const noexcept
            {
                return _width;
            }
 
            inline size_t height() const noexcept
            {
                return _height;
            }
 
            inline size_t depth() const noexcept
            {
                return _depth;
            }
 
            using Array<T>::at;
 
            T& at(size_t x, size_t y, size_t z)
            {
                return at((z * _width * _height) + (y * _width) + x);
            }
 
            const T& at(size_t x, size_t y, size_t z) const
            {
                return at((z * _width * _height) + (y * _width) + x);
            }
 
            virtual size_t dimensions() const noexcept override
            {
                return 3;
            }
 
            void resize(size_t width, size_t height, size_t depth)
            {
                size_t newSize = width * height * depth;
                
                if (newSize == Array<T>::size())
                {
                    _width = width;
                    _height = height;
                    _depth = depth;
                    return;
                }
                
                Array<T>::resize(newSize);
                
                _width = width;
                _height = height;
                _depth = depth;
            }
 
        protected:
            using Array<T>::resize;
 
        private:
            size_t _width;   
            size_t _height;  
            size_t _depth;   
 
            friend class Infinity::IO::Data::Array3DSerializer<T>;
    };
 
    using Texture3D = Array3D<Math::Vector4ub>;
}
 
// Type registration for common scalar specializations
INFINITY_TYPE(Infinity::Types::Containers::Array3D<int8_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<int16_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<int32_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<int64_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<uint8_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<uint16_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<uint32_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<uint64_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<float>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<double>)
 
// Type registration for Vector2 specializations
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector2>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector3>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector4>)
 
// Type registration for Vector2d (double) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector2d>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector3d>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector4d>)
 
// Type registration for Vector2b (int8_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector2b>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector3b>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector4b>)
 
// Type registration for Vector2s (int16_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector2s>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector3s>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector4s>)
 
// Type registration for Vector2i (int32_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector2i>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector3i>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector4i>)
 
// Type registration for Vector2l (int64_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector2l>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector3l>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector4l>)
 
// Type registration for Vector2ub (uint8_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector2ub>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector3ub>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector4ub>)
 
// Type registration for Vector2us (uint16_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector2us>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector3us>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector4us>)
 
// Type registration for Vector2ui (uint32_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector2ui>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector3ui>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector4ui>)
 
// Type registration for Vector2ul (uint64_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector2ul>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector3ul>)
INFINITY_TYPE(Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector4ul>)
 
// Type registration for common 3D texture array types (arrays of 3D textures)
INFINITY_TYPE(Infinity::Types::Containers::Array<Infinity::Types::Containers::Array3D<uint8_t>>)
INFINITY_TYPE(Infinity::Types::Containers::Array<Infinity::Types::Containers::Array3D<int8_t>>)
INFINITY_TYPE(Infinity::Types::Containers::Array<Infinity::Types::Containers::Array3D<float>>)
INFINITY_TYPE(Infinity::Types::Containers::Array<Infinity::Types::Containers::Array3D<Infinity::Types::Math::Vector4ub>>)