feat: encode uint8_t with binarystream

feelfreelinux · Feb 12, 2024 · 288613e · 288613e
1 parent 2986005
commit 288613e
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/main/io/BinaryStream.cpp b/main/io/BinaryStream.cpp
@@ -34,6 +34,12 @@ BinaryStream& BinaryStream::operator>>(char& value) {
   return *this;
 }
 
+BinaryStream& BinaryStream::operator>>(uint8_t& value) {
+  ensureReadable();
+  istr->read((char*)&value, 1);
+  return *this;
+}
+
 BinaryStream& BinaryStream::operator>>(std::byte& value) {
   ensureReadable();
   istr->read((char*)&value, 1);
@@ -109,6 +115,13 @@ BinaryStream& BinaryStream::operator<<(std::byte value) {
   return *this;
 }
 
+BinaryStream& BinaryStream::operator<<(uint8_t value) {
+  ensureWritable();
+  ostr->write((const char*)&value, sizeof(value));
+
+  return *this;
+}
+
 BinaryStream& BinaryStream::operator<<(int16_t value) {
   ensureWritable();
   if (flipBytes)

diff --git a/main/io/include/BinaryStream.h b/main/io/include/BinaryStream.h
@@ -59,6 +59,7 @@ class BinaryStream {
   // Read operations
   BinaryStream& operator>>(char& value);
   BinaryStream& operator>>(std::byte& value);
+  BinaryStream& operator>>(uint8_t& value);
   BinaryStream& operator>>(int16_t& value);
   BinaryStream& operator>>(uint16_t& value);
   BinaryStream& operator>>(int32_t& value);
@@ -69,6 +70,7 @@ class BinaryStream {
   // Write operations
   BinaryStream& operator<<(char value);
   BinaryStream& operator<<(std::byte value);
+  BinaryStream& operator<<(uint8_t value);
   BinaryStream& operator<<(int16_t value);
   BinaryStream& operator<<(uint16_t value);
   BinaryStream& operator<<(int32_t value);