Array2D.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 Array2DSerializer;
}
 
namespace Infinity::Types::Containers
{
 
    template<typename T>
    class INFINITY_API_TEMPLATE Array2D : public Array<T>
    {
        public:
            Array2D() : Array<T>(), _width(0), _height(0) { }
            
            Array2D(size_t width, size_t height) : Array<T>(width * height), _width(width), _height(height) { }
            
            Array2D(size_t width, size_t height, Array<T> data) : Array<T>(data), _width(width), _height(height) { }
            
            Array2D(size_t width, size_t height, T* data) : Array<T>(data, width * height), _width(width), _height(height) { }
            
            Array2D(size_t colStride, const std::vector<T>& data) : Array<T>(data), _width(data.size() / colStride), _height(colStride) { }
 
            virtual ~Array2D() { }
 
            std::unique_ptr<Core::Base> clone() const override
            {
                return std::make_unique<Array2D<T>>(*this);
            }
 
            const Infinity::Types::TypeID& typeId() const override { return Infinity::Types::getTypeID<Array2D<T>>(); }
 
            inline size_t width() const noexcept
            {
                return _width;
            }
 
            inline size_t height() const noexcept
            {
                return _height;
            }
 
            using Array<T>::at;
 
            T& at(size_t x, size_t y)
            {
                return at(y * _width + x);
            }
 
            const T& at(size_t x, size_t y) const
            {
                return at(y * _width + x);
            }
 
            virtual size_t dimensions() const noexcept override
            {
                return 2;
            }
 
            void resize(size_t width, size_t height)
            {
                size_t newSize = width * height;
                
                if (newSize == Array<T>::size())
                {
                    _width = width;
                    _height = height;
                    return;
                }
                
                Array<T>::resize(newSize);
                
                _width = width;
                _height = height;
            }
        
        protected:
            using Array<T>::resize;
 
        private:
            size_t _width;   
            size_t _height;  
 
            friend class Infinity::IO::Data::Array2DSerializer<T>;
    };
 
    using GrayscaleTexture = Array2D<uint8_t>;
    
    using GreyscaleTexture = Array2D<uint8_t>;
    
    using Splatmap = Array2D<uint8_t>;
    
    using RTexture = Array2D<uint8_t>;
    
    using RGTexture = Array2D<Math::Vector2ub>;
    
    using RGBTexture = Array2D<Math::Vector3ub>;
    
    using RGBATexture = Array2D<Math::Vector4ub>;
    
    using Texture = Array2D<Math::Vector4ub>;
    
    using Heightmap = Array2D<float>;
    
    using NoiseTexture = Array2D<float>;
    
    using Mask = Array2D<float>;
}
 
// Type registration for common scalar specializations
INFINITY_TYPE(Infinity::Types::Containers::Array2D<int8_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<int16_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<int32_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<int64_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<uint8_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<uint16_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<uint32_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<uint64_t>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<float>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<double>)
 
// Type registration for Vector2 specializations
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector2>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector3>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector4>)
 
// Type registration for Vector2d (double) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector2d>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector3d>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector4d>)
 
// Type registration for Vector2b (int8_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector2b>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector3b>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector4b>)
 
// Type registration for Vector2s (int16_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector2s>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector3s>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector4s>)
 
// Type registration for Vector2i (int32_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector2i>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector3i>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector4i>)
 
// Type registration for Vector2l (int64_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector2l>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector3l>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector4l>)
 
// Type registration for Vector2ub (uint8_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector2ub>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector3ub>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector4ub>)
 
// Type registration for Vector2us (uint16_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector2us>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector3us>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector4us>)
 
// Type registration for Vector2ui (uint32_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector2ui>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector3ui>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector4ui>)
 
// Type registration for Vector2ul (uint64_t) specializations
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector2ul>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector3ul>)
INFINITY_TYPE(Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector4ul>)
 
// Type registration for common 2D texture array types (arrays of 2D textures)
INFINITY_TYPE(Infinity::Types::Containers::Array<Infinity::Types::Containers::Array2D<uint8_t>>)
INFINITY_TYPE(Infinity::Types::Containers::Array<Infinity::Types::Containers::Array2D<int8_t>>)
INFINITY_TYPE(Infinity::Types::Containers::Array<Infinity::Types::Containers::Array2D<float>>)
INFINITY_TYPE(Infinity::Types::Containers::Array<Infinity::Types::Containers::Array2D<Infinity::Types::Math::Vector4ub>>)