memory_output_stream.h

Go to the documentation of this file.

#pragma once
 
#include "roo_io/core/output_stream.h"
 
namespace roo_io {
 
template <typename PtrType>
class MemoryOutputStream : public OutputStream {
 public:
  MemoryOutputStream() : ptr_(nullptr), end_(nullptr), status_(kClosed) {}
 
  MemoryOutputStream(PtrType begin, PtrType end)
      : ptr_(begin), end_(end), status_(kOk) {}
 
  size_t write(const byte* buf, size_t count) override {
    if (status_ != kOk) return 0;
    const size_t available = static_cast<size_t>(end_ - ptr_);
    if (count > available) {
      count = available;
      status_ = kNoSpaceLeftOnDevice;
    }
    memcpy(ptr_, buf, count);
    ptr_ += count;
    return count;
  }
 
  void close() override {
    if (status_ == kOk) {
      status_ = kClosed;
    }
  }
 
  Status status() const override { return status_; }
 
  byte* ptr() const { return ptr_; }
 
 private:
  byte* ptr_;
  const byte* end_;
  Status status_;
};
 
}  // namespace roo_io