moved to log_msg

include/c11log/details/stack_buf.h

@@ -0,0 +1,111 @@
+#pragma once
+
+#include <array>
+#include <vector>
+#include <algorithm>
+
+
+// Fast memory storage
+// stores its contents on the stack when possible, in vector<char> otherwise
+// NOTE: User should be remember that returned buffer might be on the stack!!
+
+namespace c11log
+{
+namespace details
+{
+
+template<std::size_t STACK_SIZE=128>
+class stack_buf
+{
+public:
+    stack_buf():_stack_size(0) {}
+    ~stack_buf() {};
+
+    stack_buf(const bufpair_t& buf_to_copy):stack_buf()
+    {
+        append(buf_to_copy);
+    }
+
+    stack_buf(const stack_buf& other)
+    {
+        _stack_size = other._stack_size;
+        if(!other._v.empty())
+            _v = other._v;
+        else if(_stack_size)
+            std::copy(other._stack_buf.begin(), other._stack_buf.begin()+_stack_size, _stack_buf.begin());
+    }
+
+    stack_buf(stack_buf&& other)
+    {
+        _stack_size = other._stack_size;
+        if(!other._v.empty())
+            _v = other._v;
+        else if(_stack_size)
+            std::copy(other._stack_buf.begin(), other._stack_buf.begin()+_stack_size, _stack_buf.begin());
+        other.clear();
+    }
+
+    stack_buf& operator=(const stack_buf& other) = delete;
+    stack_buf& operator=(stack_buf&& other) = delete;
+
+    void append(const char* buf, std::size_t buf_size)
+    {
+        //If we are aleady using _v, forget about the stack
+        if(!_v.empty())
+        {
+            _v.insert(_v.end(), buf, buf+ buf_size);
+        }
+        //Try use the stack
+        else
+        {
+            if(_stack_size+buf_size <= STACK_SIZE)
+            {
+                std::memcpy(&_stack_buf[_stack_size], buf, buf_size);
+                _stack_size+=buf_size;
+            }
+            //Not enough stack space. Copy all to _v
+            else
+            {
+                _v.reserve(_stack_size+buf_size);
+                if(_stack_size)
+                    _v.insert(_v.end(), _stack_buf.begin(), _stack_buf.begin() +_stack_size);
+                _v.insert(_v.end(), buf, buf+buf_size);
+            }
+        }
+    }
+
+    void append(const bufpair_t &buf)
+    {
+        append(buf.first, buf.second);
+    }
+
+    void clear()
+    {
+        _stack_size = 0;
+        _v.clear();
+    }
+
+    bufpair_t get()
+    {
+        if(!_v.empty())
+            return bufpair_t(_v.data(), _v.size());
+        else
+            return bufpair_t(_stack_buf.data(), _stack_size);
+    }
+
+    std::size_t size()
+    {
+        if(!_v.empty())
+            return _v.size();
+        else
+            return _stack_size;
+    }
+
+private:
+    std::vector<char> _v;
+    std::array<char, STACK_SIZE> _stack_buf;
+    std::size_t _stack_size;
+};
+
+}
+} //namespace c11log { namespace details {