dl_array2.h

浏览该文件的文档.

#pragma once
 
 
namespace dl
{
template<typename T, size_t W, size_t H>
class Array2
{
public:
    T* operator[](std::size_t j)
    {
        assert(j < H);
        return &_data[j * W];
    }

    const T* operator[](std::size_t j) const
    {
        assert(j < H);
        return &_data[j * W];
    }
 
    const T& Get(size_t j, size_t i) const
    {
        assert(j < H);
        assert(i < W);
        return _data[j * W + i];
    }
    T& GetRef(size_t j, size_t i)
    {
        assert(j < H);
        assert(i < W);
        return _data[j * W + i];
    }
 
    bool CheckRange(size_t j, size_t i) const
    {
        return j < H && i < W;
    }
    bool CheckRange(const SizeT& xy) const
    {
        return CheckRange(xy.y, xy.x);
    }

    void Fill(const T& v)
    {
        for (auto& iter : _data)
            iter = v;
    }
 
    std::byte* GetBuffer()
    {
        return (std::byte*)_data.data();
    }
 
    T* GetData()
    {
        return _data.data();
    }
 
    unsigned GetWidth()
    {
        return (unsigned)W;
    }
 
    unsigned GetHeight()
    {
        return (unsigned)H;
    }
 
    Size GetSize()
    {
        return { (unsigned)W, (unsigned)H };
    }
 
    size_t GetSizeByte()
    {
        return _data.size() * sizeof(T);
    }
 
    auto begin()
    {
        return _data.begin();
    }
    auto end()
    {
        return _data.end();
    }
private:
    std::array<T, W * H> _data;
};
 
 
}