SSOBuffer.h

#pragma once
 
#include <vector>
 
namespace isce3::core::detail {

template<typename T, size_t NMAX = 32>
class SSOBuffer {
public:
    SSOBuffer(size_t n) : size_ {n}
    {
        if (n <= NMAX) {
            ptr_ = stackbuf_;
        } else {
            heapbuf_.resize(n);
            ptr_ = heapbuf_.data();
        }
    }

    size_t size() const { return size_; }

    const T* data() const { return ptr_; }

    T* data() { return ptr_; }

    const T& operator[](size_t i) const { return ptr_[i]; }

    T& operator[](size_t i) { return ptr_[i]; }
 
    // Don't bother implementing copy & assignment, so delete default ones
    SSOBuffer(const SSOBuffer&) = delete;
    SSOBuffer& operator= (const SSOBuffer&) = delete;
 
private:
    T stackbuf_[NMAX];
    std::vector<T> heapbuf_;
    T* ptr_;
    const size_t size_;
};
 
} // namespace isce3::core::detail