strong_int.h

Go to the documentation of this file.

#pragma once
 
#include "di/format/prelude.h"
#include "di/meta/algorithm.h"
#include "di/meta/core.h"
#include "di/meta/language.h"
 
namespace di::util {
namespace detail {
    struct NoopMixin {};
 
    template<typename T>
    struct MixinHelper : meta::TypeConstant<NoopMixin> {};
 
    template<typename T>
    requires(requires { typename T::Mixin; })
    struct MixinHelper<T> : meta::TypeConstant<typename T::Mixin> {};
 
    template<typename T>
    using Mixin = meta::Type<MixinHelper<T>>;
 
    template<typename T>
    concept Tag = requires { typename T::Type; };
 
    template<typename T>
    struct SSizeTypeHelper : meta::TypeConstant<meta::MakeSigned<meta::Type<T>>> {};
 
    template<typename T>
    requires(requires { typename T::SSizeType; })
    struct SSizeTypeHelper<T> : meta::TypeConstant<typename T::SSizeType> {};
 
    template<Tag T>
    using SSizeType = meta::Type<SSizeTypeHelper<T>>;
 
    template<typename T>
    constexpr inline bool format_as_pointer = false;
 
    template<typename T>
    requires(T::format_as_pointer)
    constexpr inline bool format_as_pointer<T> = true;
}
 
template<detail::Tag Tag>
class StrongInt : public detail::Mixin<Tag> {
public:
    using IsAtom = void;
 
    using Type = Tag::Type;
    using SSizeType = detail::SSizeType<Tag>;
 
    StrongInt() = default;
 
    constexpr explicit StrongInt(Type value) : m_value(value) {}
 
    constexpr auto raw_value() const -> Type { return m_value; }
 
    constexpr auto operator+=(Type x) -> StrongInt& {
        m_value += x;
        return *this;
    }
    constexpr auto operator+=(SSizeType x) -> StrongInt&
    requires(!concepts::SameAs<SSizeType, Type>)
    {
        m_value += x;
        return *this;
    }
 
    constexpr auto operator-=(Type x) -> StrongInt& {
        m_value -= x;
        return *this;
    }
    constexpr auto operator-=(SSizeType x) -> StrongInt&
    requires(!concepts::SameAs<SSizeType, Type>)
    {
        m_value -= x;
        return *this;
    }
 
    constexpr auto operator++() -> StrongInt& {
        ++m_value;
        return *this;
    }
    constexpr auto operator++(int) -> StrongInt {
        auto copy = *this;
        ++m_value;
        return copy;
    }
 
    constexpr auto operator--() -> StrongInt& {
        --m_value;
        return *this;
    }
    constexpr auto operator--(int) -> StrongInt {
        auto copy = *this;
        --m_value;
        return copy;
    }
 
private:
    constexpr friend auto operator+(StrongInt a, Type b) -> StrongInt { return StrongInt(a.raw_value() + b); }
    constexpr friend auto operator+(StrongInt a, SSizeType b) -> StrongInt
    requires(!concepts::SameAs<SSizeType, Type>)
    {
        return StrongInt(a.raw_value() + b);
    }
    constexpr friend auto operator+(Type a, StrongInt b) -> StrongInt { return StrongInt(a + b.raw_value()); }
    constexpr friend auto operator+(SSizeType a, StrongInt b) -> StrongInt
    requires(!concepts::SameAs<SSizeType, Type>)
    {
        return StrongInt(a + b.raw_value());
    }
 
    constexpr friend auto operator-(StrongInt a, Type b) -> StrongInt { return StrongInt(a.raw_value() - b); }
    constexpr friend auto operator-(StrongInt a, SSizeType b) -> StrongInt
    requires(!concepts::SameAs<SSizeType, Type>)
    {
        return StrongInt(a.raw_value() - b);
    }
    constexpr friend auto operator-(Type a, StrongInt b) -> StrongInt { return StrongInt(a - b.raw_value()); }
    constexpr friend auto operator-(SSizeType a, StrongInt b) -> StrongInt
    requires(!concepts::SameAs<SSizeType, Type>)
    {
        return StrongInt(a - b.raw_value());
    }
 
    constexpr friend auto operator-(StrongInt a, StrongInt b) -> SSizeType { return a.raw_value() - b.raw_value(); }
 
    constexpr friend auto operator==(StrongInt a, StrongInt b) -> bool { return a.raw_value() == b.raw_value(); }
    constexpr friend auto operator<=>(StrongInt a, StrongInt b) -> di::strong_ordering {
        return a.raw_value() <=> b.raw_value();
    }
 
    template<concepts::Encoding Enc>
    constexpr friend auto tag_invoke(types::Tag<format::formatter_in_place>, InPlaceType<StrongInt>,
                                     FormatParseContext<Enc>& parse_context, bool debug) {
        using Base = meta::Conditional<detail::format_as_pointer<Tag>, void*, Type>;
        return format::formatter<Base, Enc>(parse_context, debug) % [](concepts::CopyConstructible auto formatter) {
            return [=](concepts::FormatContext auto& context, StrongInt self) {
                if constexpr (concepts::Pointer<Base>) {
                    auto* pointer = reinterpret_cast<void*>(self.raw_value());
                    return formatter(context, pointer);
                } else {
                    return formatter(context, self.raw_value());
                }
            };
        };
    }
 
    Type m_value { 0 };
};
}
 
namespace di {
using util::StrongInt;
}