// Boost.Range library
//
//  Copyright Neil Groves 2009. Use, modification and
//  distribution is subject to 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)
//
//
// For more information, see http://www.boost.org/libs/range/
//
#include <boost/range/adaptor/transformed.hpp>

#include <boost/test/test_tools.hpp>
#include <boost/test/unit_test.hpp>

#include <boost/assign.hpp>
#include <boost/bind/bind.hpp>
#include <boost/range/algorithm_ext.hpp>

#include <algorithm>
#include <list>
#include <set>
#include <vector>

namespace boost
{
    namespace
    {
        struct double_x
        {
            typedef int result_type;
            int operator()(int x) const { return x * 2; }
        };

        struct halve_x
        {
            typedef int result_type;
            int operator()(int x) const { return x / 2; }
        };

        struct lambda_init
        {
        };

        struct lambda
        {
            typedef int result_type;

            lambda(const lambda_init& init) {}
            lambda(const lambda& rhs) {}

            int operator()(int x) const { return x + 1; }

        private:
            lambda() {}
            lambda& operator=(const lambda& rhs) { return *this; }
        };

        template< class Container, class TransformFn >
        void transformed_test_impl_core( Container& c, TransformFn fn )
        {
            using namespace boost::adaptors;

            std::vector< int > test_result1;
            boost::push_back(test_result1, c | transformed(fn));

            std::vector< int > test_result2;
            boost::push_back(test_result2, adaptors::transform(c, fn));

            std::vector< int > reference;
            std::transform(c.begin(), c.end(), std::back_inserter(reference), fn);

            BOOST_CHECK_EQUAL_COLLECTIONS( reference.begin(), reference.end(),
                                           test_result1.begin(), test_result1.end() );

            BOOST_CHECK_EQUAL_COLLECTIONS( reference.begin(), reference.end(),
                                           test_result2.begin(), test_result2.end() );
        }

        template< class Rng >
        void check_copy_assign(Rng r)
        {
            Rng r2 = r;
            r2 = r;
        }

        template< class Container, class TransformFn >
        void transformed_range_copy_assign(Container& c, TransformFn fn)
        {
            using namespace boost::adaptors;
            check_copy_assign(c | transformed(fn));
            check_copy_assign(adaptors::transform(c, fn));
        }

        template< class Container, class TransformFn, class TransformFnInit >
        void transformed_test_fn_impl()
        {
            using namespace boost::assign;

            Container c;
            TransformFnInit init;
            TransformFn fn( init );

            // Test empty
            transformed_test_impl_core(c, fn);

            // Test one element
            c += 1;
            transformed_test_impl_core(c, fn);

            // Test many elements
            c += 1,1,1,2,2,2,2,2,3,4,5,6,7,8,9;
            transformed_test_impl_core(c, fn);

            // test the range and iterator are copy assignable
            transformed_range_copy_assign(c, fn);
        }

        template< class Container >
        void transformed_test_impl()
        {
            transformed_test_fn_impl< Container, double_x, double_x >();
            transformed_test_fn_impl< Container, halve_x, halve_x >();
            transformed_test_fn_impl< Container, lambda, lambda_init >();
        }

        void transformed_test()
        {
            transformed_test_impl< std::vector< int > >();
            transformed_test_impl< std::list< int > >();
            transformed_test_impl< std::set< int > >();
            transformed_test_impl< std::multiset< int > >();
        }

        struct foo_bind
        {
            int foo() const { return 7; }
        };

        void transformed_bind()
        {
            using namespace boost::adaptors;
            using namespace boost::placeholders;

            std::vector<foo_bind> input(5);
            std::vector<int> output;
            boost::range::push_back(
                    output,
                    input | transformed(boost::bind(&foo_bind::foo, _1)));

            BOOST_CHECK_EQUAL(output.size(), input.size());

            std::vector<int> reference_output(5, 7);
            BOOST_CHECK_EQUAL_COLLECTIONS(
                        output.begin(), output.end(),
                        reference_output.begin(), reference_output.end());
        }
    }
}

boost::unit_test::test_suite*
init_unit_test_suite(int argc, char* argv[])
{
    boost::unit_test::test_suite* test
        = BOOST_TEST_SUITE( "RangeTestSuite.adaptor.transformed" );

    test->add(BOOST_TEST_CASE(&boost::transformed_test));
    test->add(BOOST_TEST_CASE(&boost::transformed_bind));

    return test;
}
