#include <fit/fix.hpp>
#include <fit/static.hpp>
#include <fit/reveal.hpp>
#include "test.hpp"

#include <memory>

struct factorial_t
{
    template<class Self, class T>
    FIT_FIX_CONSTEXPR T operator()(Self s, T x) const
    {
        return x == 0 ? 1 : x * s(x-1);
    }
};

struct factorial_move_t
{
    std::unique_ptr<int> i;
    factorial_move_t() : i(new int(1))
    {}
    template<class Self, class T>
    FIT_FIX_CONSTEXPR T operator()(const Self& s, T x) const
    {
        return x == 0 ? *i : x * s(x-1);
    }
};

static constexpr fit::fix_adaptor<factorial_t> factorial = {};
static constexpr fit::static_<fit::fix_adaptor<factorial_move_t> > factorial_move = {};

FIT_TEST_CASE()
{
    const int r1 = factorial(5);
    const int r2 = fit::reveal(factorial)(5);
    FIT_TEST_CHECK(r1 == 5*4*3*2*1);
    FIT_TEST_CHECK(r2 == 5*4*3*2*1);
#if FIT_FIX_HAS_CONSTEXPR
    FIT_STATIC_TEST_CHECK(r1 == 5*4*3*2*1);
#endif
#if FIT_HAS_GENERIC_LAMBDA
    int r3 = fit::fix([](auto s, auto x) -> decltype(x) { return x == 0 ? 1 : x * s(x-1); })(5);
    FIT_TEST_CHECK(r3 == 5*4*3*2*1);
#endif
}

FIT_TEST_CASE()
{
    const int r = factorial_move(5);
    FIT_TEST_CHECK(r == 5*4*3*2*1);
    FIT_TEST_CHECK(fit::fix(factorial_move_t())(5) == 5*4*3*2*1);
}