Added constructor to buffer

man-group · Mar 11, 2024 · af3504d · af3504d
1 parent 878d5d8
commit af3504d
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/include/sparrow/buffer.hpp b/include/sparrow/buffer.hpp
@@ -180,6 +180,7 @@ namespace sparrow
 
         buffer() = default;
         explicit buffer(size_type size);
+        buffer(size_type size, value_type);
         buffer(pointer data, size_type size);
 
         ~buffer();
@@ -463,6 +464,13 @@ namespace sparrow
     {
     }
 
+    template <class T>
+    buffer<T>::buffer(size_type size, value_type value)
+        : base_type{allocate(size), size}
+    {
+        std::fill_n(base_type::data(), size, value);
+    }
+
     template <class T>
     buffer<T>::buffer(pointer data, size_type size)
         : base_type{data, size}
@@ -527,7 +535,7 @@ namespace sparrow
         resize(n);
         if (old_size < n)
         {
-            std::fill(base_type::data() + old_size, base_type::data() + n, value);
+            std::fill_n(base_type::data() + old_size, n, value);
         }
     }
 

diff --git a/test/test_buffer.cpp b/test/test_buffer.cpp
@@ -60,6 +60,15 @@ namespace sparrow
             CHECK_EQ(b2.data(), mem);
             CHECK_EQ(b2.size(), expected_size);
             CHECK_EQ(b2.data()[2], uint8_t(2));
+
+            const uint8_t expected_value = 3;
+            buffer_test_type b3(expected_size, expected_value);
+            CHECK_NE(b3.data(), nullptr);
+            CHECK_EQ(b3.size(), expected_size);
+            for (std::size_t i = 0; i < expected_size; ++i)
+            {
+                CHECK_EQ(b3[i], expected_value);
+            }
         }
 
         TEST_CASE("copy semantic")