112 lines
2.6 KiB
C++
112 lines
2.6 KiB
C++
#pragma once
|
|
|
|
#include <array>
|
|
#include <vector>
|
|
#include <algorithm>
|
|
#include <cstring>
|
|
|
|
// 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() const
|
|
{
|
|
if(!_v.empty())
|
|
return bufpair_t(_v.data(), _v.size());
|
|
else
|
|
return bufpair_t(_stack_buf.data(), _stack_size);
|
|
}
|
|
|
|
std::size_t size() const
|
|
{
|
|
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 {
|