Skip to content

Commit

Permalink
AK: Add helper type for serializing structures into buffer
Browse files Browse the repository at this point in the history
  • Loading branch information
ccapitalK authored and alimpfard committed Jul 18, 2021
1 parent 04226f0 commit 2174152
Showing 1 changed file with 47 additions and 0 deletions.
47 changes: 47 additions & 0 deletions AK/BinaryBufferWriter.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
/*
* Copyright (c) 2021, Sahan Fernando <[email protected]>
*
* SPDX-License-Identifier: BSD-2-Clause
*/

#pragma once

#include <AK/Span.h>
#include <AK/StdLibExtraDetails.h>

namespace AK {

class BinaryBufferWriter {
public:
BinaryBufferWriter(Bytes target)
: m_target(target)
{
}

template<typename T>
requires(AK::Detail::IsTriviallyConstructible<T>) T& append_structure()
{
VERIFY((reinterpret_cast<FlatPtr>(m_target.data()) + m_offset) % alignof(T) == 0);
VERIFY(m_offset + sizeof(T) <= m_target.size());
T* allocated = new (m_target.data() + m_offset) T;
m_offset += sizeof(T);
return *allocated;
}

void skip_bytes(size_t num_bytes)
{
VERIFY(m_offset + num_bytes <= m_target.size());
m_offset += num_bytes;
}

[[nodiscard]] size_t current_offset() const
{
return m_offset;
}

private:
Bytes m_target;
size_t m_offset { 0 };
};

}

0 comments on commit 2174152

Please sign in to comment.