openmw/components/esm/esmcommon.hpp

#ifndef OPENMW_ESM_COMMON_H
#define OPENMW_ESM_COMMON_H

#include <cassert>
#include <cstdint>
#include <cstring>
#include <filesystem>
#include <limits>
#include <string>
#include <string_view>
#include <type_traits>
#include <vector>

namespace ESM
{
    enum RecordFlag
    {
        // This flag exists, but is not used to determine if a record has been deleted while loading
        FLAG_Deleted = 0x00000020,
        FLAG_Persistent = 0x00000400,
        FLAG_Ignored = 0x00001000,
        FLAG_Blocked = 0x00002000
    };

    using StringSizeType = std::uint32_t;

    template <std::size_t capacity>
    struct FixedString
    {
        static_assert(capacity > 0);

        static constexpr std::size_t sCapacity = capacity;

        char mData[capacity];

        FixedString() = default;

        template <std::size_t size>
        constexpr FixedString(const char (&value)[size]) noexcept
            : mData()
        {
            if constexpr (capacity == sizeof(std::uint32_t))
            {
                static_assert(capacity == size || capacity + 1 == size);
                if constexpr (capacity + 1 == size)
                    assert(value[capacity] == '\0');
                for (std::size_t i = 0; i < capacity; ++i)
                    mData[i] = value[i];
            }
            else
            {
                const std::size_t length = std::min(capacity, size);
                for (std::size_t i = 0; i < length; ++i)
                    mData[i] = value[i];
                mData[std::min(capacity - 1, length)] = '\0';
            }
        }

        constexpr explicit FixedString(std::uint32_t value) noexcept
            : mData()
        {
            static_assert(capacity == sizeof(std::uint32_t));
            for (std::size_t i = 0; i < capacity; ++i)
                mData[i] = static_cast<char>((value >> (i * std::numeric_limits<std::uint8_t>::digits))
                    & std::numeric_limits<std::uint8_t>::max());
        }

        template <class T>
        constexpr explicit FixedString(T value) noexcept
            : FixedString(static_cast<std::uint32_t>(value))
        {
        }

        std::string_view toStringView() const noexcept { return std::string_view(mData, strnlen(mData, capacity)); }

        std::string toString() const { return std::string(toStringView()); }

        std::uint32_t toInt() const noexcept
        {
            static_assert(capacity == sizeof(std::uint32_t));
            std::uint32_t value;
            std::memcpy(&value, mData, capacity);
            return value;
        }

        void clear() noexcept { std::memset(mData, 0, capacity); }

        void assign(std::string_view value) noexcept
        {
            if (value.empty())
            {
                clear();
                return;
            }

            if (value.size() < capacity)
            {
                if constexpr (capacity == sizeof(std::uint32_t))
                    std::memset(mData, 0, capacity);
                std::memcpy(mData, value.data(), value.size());
                if constexpr (capacity != sizeof(std::uint32_t))
                    mData[value.size()] = '\0';
                return;
            }

            std::memcpy(mData, value.data(), capacity);

            if constexpr (capacity != sizeof(std::uint32_t))
                mData[capacity - 1] = '\0';
        }

        FixedString& operator=(std::uint32_t value) noexcept
        {
            static_assert(capacity == sizeof(value));
            std::memcpy(&mData, &value, capacity);
            return *this;
        }
    };

    template <std::size_t capacity>
    inline bool operator==(const FixedString<capacity>& lhs, std::string_view rhs) noexcept
    {
        for (std::size_t i = 0, n = std::min(rhs.size(), capacity); i < n; ++i)
        {
            if (lhs.mData[i] != rhs[i])
                return false;
            if (lhs.mData[i] == '\0')
                return true;
        }
        return rhs.size() <= capacity || rhs[capacity] == '\0';
    }

    template <std::size_t capacity, class T, typename = std::enable_if_t<std::is_same_v<T, char>>>
    inline bool operator==(const FixedString<capacity>& lhs, const T* const& rhs) noexcept
    {
        return lhs == std::string_view(rhs, capacity);
    }

    template <std::size_t capacity, std::size_t rhsSize>
    inline bool operator==(const FixedString<capacity>& lhs, const char (&rhs)[rhsSize]) noexcept
    {
        return strnlen(rhs, rhsSize) == strnlen(lhs.mData, capacity) && std::strncmp(lhs.mData, rhs, capacity) == 0;
    }

    inline bool operator==(const FixedString<4>& lhs, std::uint32_t rhs) noexcept
    {
        return lhs.toInt() == rhs;
    }

    inline bool operator==(const FixedString<4>& lhs, const FixedString<4>& rhs) noexcept
    {
        return lhs.toInt() == rhs.toInt();
    }

    template <class T, typename = std::enable_if_t<std::is_same_v<T, char>>>
    inline bool operator==(const FixedString<4>& lhs, const T* const& rhs) noexcept
    {
        return lhs == std::string_view(rhs, 5);
    }

    template <std::size_t capacity, class Rhs>
    inline bool operator!=(const FixedString<capacity>& lhs, const Rhs& rhs) noexcept
    {
        return !(lhs == rhs);
    }

    using NAME = FixedString<4>;
    using NAME32 = FixedString<32>;
    using NAME64 = FixedString<64>;

    static_assert(std::is_standard_layout_v<NAME> && std::is_trivial_v<NAME>);
    static_assert(std::is_standard_layout_v<NAME32> && std::is_trivial_v<NAME32>);
    static_assert(std::is_standard_layout_v<NAME64> && std::is_trivial_v<NAME64>);

    static_assert(sizeof(NAME) == 4);
    static_assert(sizeof(NAME32) == 32);
    static_assert(sizeof(NAME64) == 64);

    /* This struct defines a file 'context' which can be saved and later
       restored by an ESMReader instance. It will save the position within
       a file, and when restored will let you read from that position as
       if you never left it.
     */
    struct ESM_Context
    {
        std::filesystem::path filename;
        std::streamsize leftRec;
        std::uint32_t leftSub;
        std::streamsize leftFile;
        NAME recName, subName;
        // When working with multiple esX files, we will generate lists of all files that
        //  actually contribute to a specific cell. Therefore, we need to store the index
        //  of the file belonging to this contest. See CellStore::(list/load)refs for details.
        int index;
        std::vector<int> parentFileIndices;

        // True if subName has been read but not used.
        bool subCached;

        // File position. Only used for stored contexts, not regularly
        // updated within the reader itself.
        size_t filePos;
    };

}

#endif