mirror of
https://codeberg.org/comaps/comaps
synced 2025-12-19 13:03:36 +00:00
committed by
Konstantin Pastbin
parent
c9cbb64f12
commit
76ffc99abd
65
libs/coding/byte_stream.hpp
Normal file
65
libs/coding/byte_stream.hpp
Normal file
@@ -0,0 +1,65 @@
|
||||
#pragma once
|
||||
|
||||
#include "base/base.hpp"
|
||||
|
||||
#include <cstddef>
|
||||
#include <cstdint>
|
||||
#include <cstring>
|
||||
|
||||
class ArrayByteSource
|
||||
{
|
||||
public:
|
||||
explicit ArrayByteSource(const void * p) : m_p(static_cast<uint8_t const *>(p)) {}
|
||||
|
||||
uint8_t ReadByte() { return *m_p++; }
|
||||
|
||||
void Read(void * ptr, size_t size)
|
||||
{
|
||||
memcpy(ptr, m_p, size);
|
||||
m_p += size;
|
||||
}
|
||||
|
||||
void const * Ptr() const { return m_p; }
|
||||
uint8_t const * PtrUint8() const { return m_p; }
|
||||
|
||||
void Advance(size_t size)
|
||||
{
|
||||
m_p += size;
|
||||
}
|
||||
|
||||
private:
|
||||
uint8_t const * m_p;
|
||||
};
|
||||
|
||||
template <class StorageT> class PushBackByteSink
|
||||
{
|
||||
public:
|
||||
explicit PushBackByteSink(StorageT & storage)
|
||||
: m_Storage(storage)
|
||||
{
|
||||
}
|
||||
|
||||
void Write(void const * p, size_t size)
|
||||
{
|
||||
// assume input buffer as buffer of bytes
|
||||
uint8_t const * pp = static_cast<uint8_t const *>(p);
|
||||
m_Storage.insert(m_Storage.end(), pp, pp + size);
|
||||
}
|
||||
|
||||
size_t Pos() const
|
||||
{
|
||||
return m_Storage.size();
|
||||
}
|
||||
private:
|
||||
StorageT & m_Storage;
|
||||
};
|
||||
|
||||
class CountingSink
|
||||
{
|
||||
public:
|
||||
CountingSink() : m_Count(0) {}
|
||||
inline void Write(void const *, size_t size) { m_Count += size; }
|
||||
inline size_t GetCount() const { return m_Count; }
|
||||
private:
|
||||
size_t m_Count;
|
||||
};
|
||||
Reference in New Issue
Block a user