identity.h

#pragma once

#include <funcy/util/chainer.h>

#include <type_traits>
#include <utility>

namespace funcy
{
    template < class Arg >
    struct Identity : Chainer< Identity< Arg > >
    {
        Identity() = default;

        Identity( const Arg& x ) noexcept( std::is_copy_constructible_v< Arg > ) : x_( x )
        {
        }

        Identity( Arg&& x ) noexcept( std::is_move_constructible_v< Arg > ) : x_( std::move( x ) )
        {
        }

        void update( const Arg& x )
        {
            x_ = x;
        }

        const Arg& d0() const noexcept
        {
            return x_;
        }

        template < int >
        const Arg& d1( const Arg& dx ) const noexcept
        {
            return dx;
        }

    private:
        Arg x_;
    };

    template < class Arg >
    auto identity( Arg&& x )
    {
        return Identity( std::forward< Arg >( x ) );
    }
} // namespace funcy