ProxyRefOpMixin.hpp

Go to the documentation of this file.

// Copyright 2022 Bernhard Manfred Gruber
// SPDX-License-Identifier: MPL-2.0
 
#pragma once
 
#include "macros.hpp"
 
namespace llama
{
    LLAMA_EXPORT
    template<typename Derived, typename ValueType>
    struct ProxyRefOpMixin
    {
    private:
        LLAMA_FN_HOST_ACC_INLINE constexpr auto derived() -> Derived&
        {
            return static_cast<Derived&>(*this);
        }
 
        // in principle, load() could be const, but we use it only from non-const functions
        LLAMA_FN_HOST_ACC_INLINE constexpr auto load() -> ValueType
        {
            return static_cast<ValueType>(derived());
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr void store(ValueType t)
        {
            derived() = std::move(t);
        }
 
    public:
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator+=(const ValueType& rhs) -> Derived&
        {
            ValueType lhs = load();
            lhs += rhs;
            store(lhs);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator-=(const ValueType& rhs) -> Derived&
        {
            ValueType lhs = load();
            lhs -= rhs;
            store(lhs);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator*=(const ValueType& rhs) -> Derived&
        {
            ValueType lhs = load();
            lhs *= rhs;
            store(lhs);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator/=(const ValueType& rhs) -> Derived&
        {
            ValueType lhs = load();
            lhs /= rhs;
            store(lhs);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator%=(const ValueType& rhs) -> Derived&
        {
            ValueType lhs = load();
            lhs %= rhs;
            store(lhs);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator<<=(const ValueType& rhs) -> Derived&
        {
            ValueType lhs = load();
            lhs <<= rhs;
            store(lhs);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator>>=(const ValueType& rhs) -> Derived&
        {
            ValueType lhs = load();
            lhs >>= rhs;
            store(lhs);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator&=(const ValueType& rhs) -> Derived&
        {
            ValueType lhs = load();
            lhs &= rhs;
            store(lhs);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator|=(const ValueType& rhs) -> Derived&
        {
            ValueType lhs = load();
            lhs |= rhs;
            store(lhs);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator^=(const ValueType& rhs) -> Derived&
        {
            ValueType lhs = load();
            lhs ^= rhs;
            store(lhs);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator++() -> Derived&
        {
            ValueType v = load();
            ++v;
            store(v);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator++(int) -> ValueType
        {
            ValueType v = load();
            ValueType old = v++;
            store(v);
            return old;
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator--() -> Derived&
        {
            ValueType v = load();
            --v;
            store(v);
            return derived();
        }
 
        LLAMA_FN_HOST_ACC_INLINE constexpr auto operator--(int) -> ValueType
        {
            ValueType v = load();
            ValueType old = v--;
            store(v);
            return old;
        }
 
        LLAMA_FN_HOST_ACC_INLINE friend constexpr void swap(Derived a, Derived b) noexcept
        {
            const auto va = static_cast<ValueType>(a);
            const auto vb = static_cast<ValueType>(b);
            a = vb;
            b = va;
        }
    };
} // namespace llama