//  Taken from the Boost.Bind library
//
//  bind_stateful_test.cpp
//
//  Copyright (c) 2004 Peter Dimov
//  Copyright (c) 2013 Agustin Berge
//
//  SPDX-License-Identifier: BSL-1.0
// Distributed under the Boost Software License, Version 1.0. (See
// accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
//

#if defined(HPX_MSVC)
#pragma warning(disable : 4786)    // identifier truncated in debug info
#pragma warning(disable : 4710)    // function not inlined
#pragma warning(                                                               \
    disable : 4711)    // function selected for automatic inline expansion
#pragma warning(disable : 4514)    // unreferenced inline removed
#endif

#include <hpx/functional/bind.hpp>

namespace placeholders = hpx::placeholders;

#include <functional>
#include <iostream>

#include <hpx/modules/testing.hpp>

class X
{
private:
    int state_;

public:
    X()
      : state_(0)
    {
    }

// SGI-related compilers have odd compiler-synthesized ctors and dtors
#ifdef __PATHSCALE__
    ~X() {}
#endif

    int state() const
    {
        return state_;
    }

    int operator()()
    {
        return state_ += 17041;
    }

    int operator()(int x1)
    {
        return state_ += x1;
    }

    int operator()(int x1, int x2)
    {
        return state_ += x1 + x2;
    }

    int operator()(int x1, int x2, int x3)
    {
        return state_ += x1 + x2 + x3;
    }

    int operator()(int x1, int x2, int x3, int x4)
    {
        return state_ += x1 + x2 + x3 + x4;
    }

    int operator()(int x1, int x2, int x3, int x4, int x5)
    {
        return state_ += x1 + x2 + x3 + x4 + x5;
    }

    int operator()(int x1, int x2, int x3, int x4, int x5, int x6)
    {
        return state_ += x1 + x2 + x3 + x4 + x5 + x6;
    }

    int operator()(int x1, int x2, int x3, int x4, int x5, int x6, int x7)
    {
        return state_ += x1 + x2 + x3 + x4 + x5 + x6 + x7;
    }

    int operator()(
        int x1, int x2, int x3, int x4, int x5, int x6, int x7, int x8)
    {
        return state_ += x1 + x2 + x3 + x4 + x5 + x6 + x7 + x8;
    }

    int operator()(
        int x1, int x2, int x3, int x4, int x5, int x6, int x7, int x8, int x9)
    {
        return state_ += x1 + x2 + x3 + x4 + x5 + x6 + x7 + x8 + x9;
    }
};

int f0(int& state_)
{
    return state_ += 17041;
}

int f1(int& state_, int x1)
{
    return state_ += x1;
}

int f2(int& state_, int x1, int x2)
{
    return state_ += x1 + x2;
}

int f3(int& state_, int x1, int x2, int x3)
{
    return state_ += x1 + x2 + x3;
}

int f4(int& state_, int x1, int x2, int x3, int x4)
{
    return state_ += x1 + x2 + x3 + x4;
}

int f5(int& state_, int x1, int x2, int x3, int x4, int x5)
{
    return state_ += x1 + x2 + x3 + x4 + x5;
}

int f6(int& state_, int x1, int x2, int x3, int x4, int x5, int x6)
{
    return state_ += x1 + x2 + x3 + x4 + x5 + x6;
}

int f7(int& state_, int x1, int x2, int x3, int x4, int x5, int x6, int x7)
{
    return state_ += x1 + x2 + x3 + x4 + x5 + x6 + x7;
}

int f8(
    int& state_, int x1, int x2, int x3, int x4, int x5, int x6, int x7, int x8)
{
    return state_ += x1 + x2 + x3 + x4 + x5 + x6 + x7 + x8;
}

template <class F>
void test(F f, int a, int b)
{
    HPX_TEST_EQ(f(), a + b);
    HPX_TEST_EQ(f(), a + 2 * b);
    HPX_TEST_EQ(f(), a + 3 * b);
}

void stateful_function_object_test()
{
    test(hpx::bind(X()), 0, 17041);
    test(hpx::bind(X(), 1), 0, 1);
    test(hpx::bind(X(), 1, 2), 0, 1 + 2);
    test(hpx::bind(X(), 1, 2, 3), 0, 1 + 2 + 3);
    test(hpx::bind(X(), 1, 2, 3, 4), 0, 1 + 2 + 3 + 4);
    test(hpx::bind(X(), 1, 2, 3, 4, 5), 0, 1 + 2 + 3 + 4 + 5);
    test(hpx::bind(X(), 1, 2, 3, 4, 5, 6), 0, 1 + 2 + 3 + 4 + 5 + 6);
    test(hpx::bind(X(), 1, 2, 3, 4, 5, 6, 7), 0, 1 + 2 + 3 + 4 + 5 + 6 + 7);
    test(hpx::bind(X(), 1, 2, 3, 4, 5, 6, 7, 8), 0,
        1 + 2 + 3 + 4 + 5 + 6 + 7 + 8);
    test(hpx::bind(X(), 1, 2, 3, 4, 5, 6, 7, 8, 9), 0,
        1 + 2 + 3 + 4 + 5 + 6 + 7 + 8 + 9);

    X x;

    int n = x.state();

    test(hpx::bind(std::ref(x)), n, 17041);
    n += 3 * 17041;

    test(hpx::bind(std::ref(x), 1), n, 1);
    n += 3 * 1;

    test(hpx::bind(std::ref(x), 1, 2), n, 1 + 2);
    n += 3 * (1 + 2);

    test(hpx::bind(std::ref(x), 1, 2, 3), n, 1 + 2 + 3);
    n += 3 * (1 + 2 + 3);

    test(hpx::bind(std::ref(x), 1, 2, 3, 4), n, 1 + 2 + 3 + 4);
    n += 3 * (1 + 2 + 3 + 4);

    test(hpx::bind(std::ref(x), 1, 2, 3, 4, 5), n, 1 + 2 + 3 + 4 + 5);
    n += 3 * (1 + 2 + 3 + 4 + 5);

    test(hpx::bind(std::ref(x), 1, 2, 3, 4, 5, 6), n, 1 + 2 + 3 + 4 + 5 + 6);
    n += 3 * (1 + 2 + 3 + 4 + 5 + 6);

    test(hpx::bind(std::ref(x), 1, 2, 3, 4, 5, 6, 7), n,
        1 + 2 + 3 + 4 + 5 + 6 + 7);
    n += 3 * (1 + 2 + 3 + 4 + 5 + 6 + 7);

    test(hpx::bind(std::ref(x), 1, 2, 3, 4, 5, 6, 7, 8), n,
        1 + 2 + 3 + 4 + 5 + 6 + 7 + 8);
    n += 3 * (1 + 2 + 3 + 4 + 5 + 6 + 7 + 8);

    test(hpx::bind(std::ref(x), 1, 2, 3, 4, 5, 6, 7, 8, 9), n,
        1 + 2 + 3 + 4 + 5 + 6 + 7 + 8 + 9);
    n += 3 * (1 + 2 + 3 + 4 + 5 + 6 + 7 + 8 + 9);

    HPX_TEST_EQ(x.state(), n);
}

void stateful_function_test()
{
    test(hpx::bind(f0, 0), 0, 17041);
    test(hpx::bind(f1, 0, 1), 0, 1);
    test(hpx::bind(f2, 0, 1, 2), 0, 1 + 2);
    test(hpx::bind(f3, 0, 1, 2, 3), 0, 1 + 2 + 3);
    test(hpx::bind(f4, 0, 1, 2, 3, 4), 0, 1 + 2 + 3 + 4);
    test(hpx::bind(f5, 0, 1, 2, 3, 4, 5), 0, 1 + 2 + 3 + 4 + 5);
    test(hpx::bind(f6, 0, 1, 2, 3, 4, 5, 6), 0, 1 + 2 + 3 + 4 + 5 + 6);
    test(hpx::bind(f7, 0, 1, 2, 3, 4, 5, 6, 7), 0, 1 + 2 + 3 + 4 + 5 + 6 + 7);
    test(hpx::bind(f8, 0, 1, 2, 3, 4, 5, 6, 7, 8), 0,
        1 + 2 + 3 + 4 + 5 + 6 + 7 + 8);
}

int main()
{
    stateful_function_object_test();
    stateful_function_test();
    return hpx::util::report_errors();
}
