feat: add pad helper to write N zeroed bytes

Author: craftablescience

include/BufferStream.h

@@ -274,6 +274,14 @@ class BufferStream {
 		return this->write(obj);
 	}
 
+	template<BufferStreamPODType T = std::byte>
+	BufferStream& pad(std::uint64_t n) {
+		for (std::uint64_t i = 0; i < n * sizeof(T); i++) {
+			this->write<std::byte>({});
+		}
+		return *this;
+	}
+
 	template<BufferStreamPODType T, std::uint64_t N>
 	BufferStream& read(T(&obj)[N]) {
 		if (this->useExceptions && this->bufferPos + sizeof(T) * N > this->bufferLen) {

include/FileStream.h

@@ -235,6 +235,14 @@ class FileStream {
 		return this->write(obj);
 	}
 
+	template<BufferStreamPODType T = std::byte>
+	FileStream& pad(std::uint64_t n) {
+		for (std::uint64_t i = 0; i < n * sizeof(T); i++) {
+			this->write<std::byte>({});
+		}
+		return *this;
+	}
+
 	template<BufferStreamPODType T, std::uint64_t N>
 	FileStream& read(T(&obj)[N]) {
 		return this->read(&obj, N);

test/BufferStream.cpp

@@ -154,6 +154,22 @@ TEST(BufferStream, skip) {
 	EXPECT_EQ(stream.tell(), 3);
 }
 
+TEST(BufferStream, pad) {
+	std::vector<unsigned char> buffer;
+	BufferStream stream{buffer};
+
+	EXPECT_EQ(stream.tell(), 0);
+
+	stream.pad(1);
+	EXPECT_EQ(stream.tell(), 1);
+
+	stream.pad(2);
+	EXPECT_EQ(stream.tell(), 3);
+
+	stream.pad<std::int16_t>(1);
+	EXPECT_EQ(stream.tell(), 3 + sizeof(std::int16_t));
+}
+
 TEST(BufferStream, read_int_ref) {
 	{
 		int x = 10;