Buffer.h

#pragma once

#include "forward.h"

#include <array>
#include <gdal_priv.h>

#include "detail/GDALDataTypeUtil.h"

namespace isce3 { namespace io { namespace gdal {

class Buffer {
public:

    constexpr
    Buffer(const void * data,
           GDALDataType datatype,
           const std::array<int, 2> & shape);

    constexpr
    Buffer(void * data,
           GDALDataType datatype,
           const std::array<int, 2> & shape,
           GDALAccess access = GA_Update);

    constexpr
    Buffer(const void * data,
           GDALDataType datatype,
           const std::array<int, 2> & shape,
           const std::array<std::size_t, 2> & strides);

    constexpr
    Buffer(void * data,
           GDALDataType datatype,
           const std::array<int, 2> & shape,
           const std::array<std::size_t, 2> & strides,
           GDALAccess access = GA_Update);

    constexpr
    void * data() { return _data; }

    constexpr
    const void * data() const { return _data; }

    constexpr
    GDALDataType datatype() const { return _datatype; }

    constexpr
    std::size_t itemsize() const { return detail::getSize(_datatype); }

    constexpr
    const std::array<int, 2> & shape() const { return _shape; }

    constexpr
    int length() const { return _shape[0]; }

    constexpr
    int width() const { return _shape[1]; }

    constexpr
    const std::array<std::size_t, 2> & strides() const { return _strides; }

    constexpr
    std::size_t rowstride() const { return _strides[0]; }

    constexpr
    std::size_t colstride() const { return _strides[1]; }

    constexpr
    GDALAccess access() const { return _access; }

    template<typename T>
    constexpr
    TypedBuffer<T> cast() const;

private:
    void * _data;
    GDALDataType _datatype;
    std::array<int, 2> _shape;
    std::array<std::size_t, 2> _strides;
    GDALAccess _access;
};

template<typename T>
class TypedBuffer {
public:

    constexpr
    TypedBuffer(const T * data,
                const std::array<int, 2> & shape);

    constexpr
    TypedBuffer(T * data,
                const std::array<int, 2> & shape,
                GDALAccess access = GA_Update);

    constexpr
    TypedBuffer(const T * data,
                const std::array<int, 2> & shape,
                const std::array<std::size_t, 2> & strides);

    constexpr
    TypedBuffer(T * data,
                const std::array<int, 2> & shape,
                const std::array<std::size_t, 2> & strides,
                GDALAccess access = GA_Update);

    constexpr
    T * data() { return _data; }

    constexpr
    const T * data() const { return _data; }

    constexpr
    GDALDataType datatype() const { return detail::Type2GDALDataType<T>::datatype; }

    constexpr
    std::size_t itemsize() const { return sizeof(T); }

    constexpr
    const std::array<int, 2> & shape() const { return _shape; }

    constexpr
    int length() const { return _shape[0]; }

    constexpr
    int width() const { return _shape[1]; }

    constexpr
    const std::array<std::size_t, 2> & strides() const { return _strides; }

    constexpr
    std::size_t rowstride() const { return _strides[0]; }

    constexpr
    std::size_t colstride() const { return _strides[1]; }

    constexpr
    GDALAccess access() const { return _access; }

private:
    T * _data;
    std::array<int, 2> _shape;
    std::array<std::size_t, 2> _strides;
    GDALAccess _access;
};

}}}

#define ISCE_IO_GDAL_BUFFER_ICC
#include "Buffer.icc"
#undef ISCE_IO_GDAL_BUFFER_ICC