Pool.h

Go to the documentation of this file.

 
#pragma once
 
#include "Object.h"
 
namespace ConfigDB
{
class PoolData
{
public:
    PoolData(size_t itemSize) : itemSize(itemSize)
    {
        assert(itemSize > 0 && itemSize <= 255);
    }
 
    PoolData(const PoolData& other)
    {
        *this = other;
    }
 
    PoolData(PoolData&& other);
 
    PoolData& operator=(const PoolData& other);
 
    size_t getCount() const
    {
        return count;
    }
 
    size_t getCapacity() const
    {
        return count + space;
    }
 
    size_t getItemSize() const
    {
        return itemSize;
    }
 
    size_t getItemSize(size_t count) const
    {
        return count * itemSize;
    }
 
    void* operator[](unsigned index)
    {
        return getItemPtr(index);
    }
 
    const void* operator[](unsigned index) const
    {
        return getItemPtr(index);
    }
 
    void clear()
    {
        free(buffer);
        buffer = nullptr;
        count = space = 0;
    }
 
    bool ensureCapacity(size_t capacity);
 
    size_t usage() const
    {
        return getCapacity() * itemSize;
    }
 
protected:
    void* allocate(size_t itemCount);
    void deallocate(size_t itemCount);
 
    void* getItemPtr(unsigned index)
    {
        assert(index < count);
        return static_cast<uint8_t*>(buffer) + getItemSize(index);
    }
 
    const void* getItemPtr(unsigned index) const
    {
        assert(index < count);
        return static_cast<const uint8_t*>(buffer) + getItemSize(index);
    }
 
    void* buffer{};
    uint16_t count{0};
    uint8_t space{0};
    uint8_t itemSize;
};
 
struct CountedString {
public:
    const char* value{};
    uint16_t length{};
 
    CountedString() = default;
 
    CountedString(const char* value, uint16_t length) : value(value), length(length)
    {
    }
 
    explicit CountedString(const String& s) : value(s.c_str()), length(s.length())
    {
    }
 
    bool operator==(const CountedString& other) const
    {
        return length == other.length && memcmp(value, other.value, length) == 0;
    }
 
    explicit operator bool() const
    {
        return value && length;
    }
 
    explicit operator String() const
    {
        return String(value, length);
    }
 
    uint16_t getStorageSize() const
    {
        return 1 + (length > 0x80) + length;
    }
};
 
class StringPool : public PoolData
{
public:
    StringPool() : PoolData(1)
    {
    }
 
    ~StringPool()
    {
        clear();
    }
 
    StringId find(const CountedString& string) const;
 
    StringId add(const CountedString& string);
 
    StringId findOrAdd(const CountedString& string)
    {
        return find(string) ?: add(string);
    }
 
    CountedString operator[](StringId ref) const
    {
        unsigned offset = ref - 1;
        return offset < count ? getString(offset) : CountedString{};
    }
 
    const char* getBuffer() const
    {
        return static_cast<const char*>(buffer);
    }
 
private:
    CountedString getString(unsigned offset) const;
};
 
class ArrayData : public PoolData
{
public:
    using PoolData::PoolData;
 
    void* insert(unsigned index, const void* data = nullptr, size_t itemCount = 1);
 
    void* add(const void* data = nullptr, size_t itemCount = 1)
    {
        return insert(getCount(), data, itemCount);
    }
 
    bool remove(unsigned index);
 
    void dispose()
    {
        clear();
        itemSize = 0;
    }
};
 
class ArrayPool : public PoolData
{
public:
    ArrayPool() : PoolData(sizeof(ArrayData))
    {
    }
 
    ArrayPool(const ArrayPool& other);
 
    ~ArrayPool()
    {
        clear();
    }
 
    ArrayId add(const ObjectInfo& object)
    {
        return add(object.structSize);
    }
 
    ArrayId add(const PropertyInfo& prop)
    {
        return add(prop.getSize());
    }
 
    ArrayData& operator[](ArrayId id)
    {
        return *static_cast<ArrayData*>(getItemPtr(id - 1));
    }
 
    const ArrayData& operator[](ArrayId id) const
    {
        return *static_cast<const ArrayData*>(getItemPtr(id - 1));
    }
 
    void clear();
 
    size_t usage() const
    {
        size_t n = PoolData::usage();
        auto data = static_cast<const ArrayData*>(buffer);
        for(unsigned i = 0; i < count; ++i) {
            n += data->usage();
            ++data;
        }
        return n;
    }
 
private:
    ArrayId add(size_t itemSize);
};
 
} // namespace ConfigDB