get writeable buffer

fraillt · Jan 9, 2025 · 037c846 · 037c846
1 parent ee992d8
commit 037c846
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/include/bitsery/adapter/buffer.h b/include/bitsery/adapter/buffer.h
@@ -261,6 +261,15 @@ class OutputBufferAdapter
     writeInternalImpl(data, size);
   }
 
+  TValue* writeInternalBuffer(size_t size)
+  {
+    const size_t newOffset = _currOffset + size;
+    maybeResize(newOffset, TResizable{});
+    _currOffset = newOffset;
+    return _beginIt + static_cast<diff_t>(_currOffset);
+  }
+
+
   Buffer* _buffer;
   TIterator _beginIt;
   size_t _currOffset{ 0 };

diff --git a/include/bitsery/details/adapter_common.h b/include/bitsery/details/adapter_common.h
@@ -300,6 +300,15 @@ struct OutputAdapterBaseCRTP
     writeSwappedBuffer(buf, count, ShouldSwap<typename Adapter::TConfig, T>{});
   }
 
+  template<size_t SIZE, typename T>
+  T* writeBuffer(size_t count)
+  {
+    static_assert(std::is_integral<T>(), "");
+    static_assert(sizeof(T) == SIZE, "");
+
+    return static_cast<Adapter*>(this)->writeInternalBuffer(count * sizeof(T));
+  }
+
   template<typename T>
   void writeBits(const T&, size_t)
   {
@@ -375,6 +384,16 @@ struct InputAdapterBaseCRTP
     swapDataBits(buf, count, ShouldSwap<typename Adapter::TConfig, T>{});
   }
 
+  template<size_t SIZE, typename T>
+  T* readBuffer(T* buf, size_t count)
+  {
+    static_assert(std::is_integral<T>(), "");
+    static_assert(sizeof(T) == SIZE, "");
+    static_cast<Adapter*>(this)->readInternalBuffer(
+      reinterpret_cast<typename Adapter::TValue*>(buf), sizeof(T) * count);
+    swapDataBits(buf, count, ShouldSwap<typename Adapter::TConfig, T>{});
+  }
+
   template<typename T>
   void readBits(T&, size_t)
   {