//  Copyright (c) 2018, Cem Bassoy, cem.bassoy@gmail.com
//  Copyright (c) 2019, Amit Singh, amitsingh19975@gmail.com
//
//  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)
//
//  The authors gratefully acknowledge the support of
//  Google and Fraunhofer IOSB, Ettlingen, Germany
//



#include <boost/test/unit_test.hpp>
#include <boost/numeric/ublas/tensor/extents.hpp>

BOOST_AUTO_TEST_SUITE(test_strides)

using test_types = std::tuple<boost::numeric::ublas::layout::first_order, boost::numeric::ublas::layout::last_order>;

using extents       = boost::numeric::ublas::extents<>;
using first_order   = boost::numeric::ublas::layout::first_order;
using last_order    = boost::numeric::ublas::layout::last_order;


BOOST_AUTO_TEST_CASE_TEMPLATE( test_strides_ctor, value, test_types)
{
  namespace ublas = boost::numeric::ublas;
  constexpr auto layout = value{};

  auto s1   = ublas::to_strides(extents    {1},layout);
  auto s5   = ublas::to_strides(extents    {5},layout);
  auto s11  = ublas::to_strides(extents  {1,1},layout);
  auto s12  = ublas::to_strides(extents  {1,2},layout);
  auto s21  = ublas::to_strides(extents  {2,1},layout);
  auto s23  = ublas::to_strides(extents  {2,3},layout);
  auto s231 = ublas::to_strides(extents{2,3,1},layout);
  auto s123 = ublas::to_strides(extents{1,2,3},layout);
  auto s423 = ublas::to_strides(extents{4,2,3},layout);

  BOOST_CHECK  (!  s1.empty());
  BOOST_CHECK  (!  s5.empty());
  BOOST_CHECK  (! s11.empty());
  BOOST_CHECK  (! s12.empty());
  BOOST_CHECK  (! s21.empty());
  BOOST_CHECK  (! s23.empty());
  BOOST_CHECK  (!s231.empty());
  BOOST_CHECK  (!s123.empty());
  BOOST_CHECK  (!s423.empty());

  BOOST_CHECK_EQUAL (   s1.size(), 1);
  BOOST_CHECK_EQUAL (   s5.size(), 1);
  BOOST_CHECK_EQUAL (  s11.size(), 2);
  BOOST_CHECK_EQUAL (  s12.size(), 2);
  BOOST_CHECK_EQUAL (  s21.size(), 2);
  BOOST_CHECK_EQUAL (  s23.size(), 2);
  BOOST_CHECK_EQUAL ( s231.size(), 3);
  BOOST_CHECK_EQUAL ( s123.size(), 3);
  BOOST_CHECK_EQUAL ( s423.size(), 3);
}

BOOST_AUTO_TEST_CASE( test_strides_ctor_access_first_order)
{
  namespace ublas = boost::numeric::ublas;
  constexpr auto layout = first_order{};

  auto s1   = ublas::to_strides(extents    {1},layout);
  auto s5   = ublas::to_strides(extents    {5},layout);
  auto s11  = ublas::to_strides(extents  {1,1},layout);
  auto s12  = ublas::to_strides(extents  {1,2},layout);
  auto s21  = ublas::to_strides(extents  {2,1},layout);
  auto s23  = ublas::to_strides(extents  {2,3},layout);
  auto s231 = ublas::to_strides(extents{2,3,1},layout);
  auto s123 = ublas::to_strides(extents{1,2,3},layout);
  auto s423 = ublas::to_strides(extents{4,2,3},layout);

  BOOST_REQUIRE_EQUAL ( s11 .size(),2);
  BOOST_REQUIRE_EQUAL ( s12 .size(),2);
  BOOST_REQUIRE_EQUAL ( s21 .size(),2);
  BOOST_REQUIRE_EQUAL ( s23 .size(),2);
  BOOST_REQUIRE_EQUAL ( s231.size(),3);
  BOOST_REQUIRE_EQUAL ( s123.size(),3);
  BOOST_REQUIRE_EQUAL ( s423.size(),3);


  BOOST_CHECK_EQUAL ( s11[0], 1);
  BOOST_CHECK_EQUAL ( s11[1], 1);

  BOOST_CHECK_EQUAL ( s12[0], 1);
  BOOST_CHECK_EQUAL ( s12[1], 1);

  BOOST_CHECK_EQUAL ( s21[0], 1);
  BOOST_CHECK_EQUAL ( s21[1], 1);


  BOOST_CHECK_EQUAL ( s23[0], 1);
  BOOST_CHECK_EQUAL ( s23[1], 2);

  BOOST_CHECK_EQUAL ( s231[0], 1);
  BOOST_CHECK_EQUAL ( s231[1], 2);
  BOOST_CHECK_EQUAL ( s231[2], 6);

  BOOST_CHECK_EQUAL ( s123[0], 1);
  BOOST_CHECK_EQUAL ( s123[1], 1);
  BOOST_CHECK_EQUAL ( s123[2], 2);

  BOOST_CHECK_EQUAL ( s423[0], 1);
  BOOST_CHECK_EQUAL ( s423[1], 4);
  BOOST_CHECK_EQUAL ( s423[2], 8);
}

BOOST_AUTO_TEST_CASE( test_strides_ctor_access_last_order)
{
  namespace ublas = boost::numeric::ublas;
  constexpr auto layout = last_order{};

  auto s1   = ublas::to_strides(extents    {1},layout);
  auto s5   = ublas::to_strides(extents    {5},layout);
  auto s11  = ublas::to_strides(extents  {1,1},layout);
  auto s12  = ublas::to_strides(extents  {1,2},layout);
  auto s21  = ublas::to_strides(extents  {2,1},layout);
  auto s23  = ublas::to_strides(extents  {2,3},layout);
  auto s231 = ublas::to_strides(extents{2,3,1},layout);
  auto s123 = ublas::to_strides(extents{1,2,3},layout);
  auto s423 = ublas::to_strides(extents{4,2,3},layout);

  BOOST_REQUIRE_EQUAL ( s11 .size(),2);
  BOOST_REQUIRE_EQUAL ( s12 .size(),2);
  BOOST_REQUIRE_EQUAL ( s21 .size(),2);
  BOOST_REQUIRE_EQUAL ( s23 .size(),2);
  BOOST_REQUIRE_EQUAL ( s231.size(),3);
  BOOST_REQUIRE_EQUAL ( s123.size(),3);
  BOOST_REQUIRE_EQUAL ( s423.size(),3);

  BOOST_CHECK_EQUAL  ( s11[0], 1);
  BOOST_CHECK_EQUAL  ( s11[1], 1);

  BOOST_CHECK_EQUAL   ( s12[0], 1);
  BOOST_CHECK_EQUAL   ( s12[1], 1);

  BOOST_CHECK_EQUAL   ( s21[0], 1);
  BOOST_CHECK_EQUAL   ( s21[1], 1);

  BOOST_CHECK_EQUAL   ( s23[0], 3);
  BOOST_CHECK_EQUAL   ( s23[1], 1);

  BOOST_CHECK_EQUAL   ( s231[0], 3);
  BOOST_CHECK_EQUAL   ( s231[1], 1);
  BOOST_CHECK_EQUAL   ( s231[2], 1);

  BOOST_CHECK_EQUAL   ( s123[0], 6);
  BOOST_CHECK_EQUAL   ( s123[1], 3);
  BOOST_CHECK_EQUAL   ( s123[2], 1);

  BOOST_CHECK_EQUAL   ( s423[0], 6);
  BOOST_CHECK_EQUAL   ( s423[1], 3);
  BOOST_CHECK_EQUAL   ( s423[2], 1);
}

BOOST_AUTO_TEST_SUITE_END()