//============================================================================
//  Copyright (c) Kitware, Inc.
//  All rights reserved.
//  See LICENSE.txt for details.
//
//  This software is distributed WITHOUT ANY WARRANTY; without even
//  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
//  PURPOSE.  See the above copyright notice for more information.
//============================================================================

#ifndef vtk_m_internal_ListTagDetail_h
#define vtk_m_internal_ListTagDetail_h

#if !defined(vtk_m_ListTag_h) && !defined(VTKM_TEST_HEADER_BUILD)
#error ListTagDetail.h must be included from ListTag.h
#endif

#include <vtkm/Types.h>
#include <vtkm/internal/brigand.hpp>

namespace vtkm
{
namespace detail
{

//-----------------------------------------------------------------------------

/// Base class that all ListTag classes inherit from. Helps identify lists
/// in macros like VTKM_IS_LIST_TAG.
///
struct ListRoot
{
};

template <class... T>
using ListBase = brigand::list<T...>;

/// list value that is used to represent a list actually matches all values
struct UniversalTag
{
  //We never want this tag constructed, and by deleting the constructor
  //we get an error when trying to use this class with ForEach.
  UniversalTag() = delete;
};

//-----------------------------------------------------------------------------
template <typename... ListTags>
struct ListJoin
{
  using type = brigand::append<ListTags...>;
};

template <typename ListTag>
struct ListJoin<vtkm::detail::ListBase<vtkm::detail::UniversalTag>, ListTag>
{
  using type = vtkm::detail::ListBase<vtkm::detail::UniversalTag>;
};

template <typename ListTag>
struct ListJoin<ListTag, vtkm::detail::ListBase<vtkm::detail::UniversalTag>>
{
  using type = vtkm::detail::ListBase<vtkm::detail::UniversalTag>;
};

//-----------------------------------------------------------------------------
template <typename Type, typename List>
struct ListContainsImpl;

//-----------------------------------------------------------------------------
template <typename Type>
struct ListContainsImpl<Type, brigand::empty_sequence>
{
  static constexpr bool value = false;
};

//-----------------------------------------------------------------------------
template <typename Type>
struct ListContainsImpl<Type, brigand::list<vtkm::detail::UniversalTag>>
{
  static constexpr bool value = true;
};

//-----------------------------------------------------------------------------
template <typename Type, typename T1>
struct ListContainsImpl<Type, brigand::list<T1>>
{
  static constexpr bool value = std::is_same<Type, T1>::value;
};

//-----------------------------------------------------------------------------
template <typename Type, typename T1, typename T2>
struct ListContainsImpl<Type, brigand::list<T1, T2>>
{
  static constexpr bool value = std::is_same<Type, T1>::value || std::is_same<Type, T2>::value;
};

//-----------------------------------------------------------------------------
template <typename Type, typename T1, typename T2, typename T3>
struct ListContainsImpl<Type, brigand::list<T1, T2, T3>>
{
  static constexpr bool value =
    std::is_same<Type, T1>::value || std::is_same<Type, T2>::value || std::is_same<Type, T3>::value;
};

//-----------------------------------------------------------------------------
template <typename Type, typename T1, typename T2, typename T3, typename T4>
struct ListContainsImpl<Type, brigand::list<T1, T2, T3, T4>>
{
  static constexpr bool value = std::is_same<Type, T1>::value || std::is_same<Type, T2>::value ||
    std::is_same<Type, T3>::value || std::is_same<Type, T4>::value;
};

//-----------------------------------------------------------------------------
template <typename Type, typename List>
struct ListContainsImpl
{
  using find_result = brigand::find<List, std::is_same<brigand::_1, Type>>;
  using size = brigand::size<find_result>;
  static constexpr bool value = (size::value != 0);
};

//-----------------------------------------------------------------------------
template <typename BrigandList>
struct ListSizeImpl;

template <typename... Ts>
struct ListSizeImpl<brigand::list<Ts...>>
{
  static constexpr vtkm::IdComponent value = vtkm::IdComponent{ sizeof...(Ts) };
};

//-----------------------------------------------------------------------------
template <typename Type, typename RemainingList, vtkm::IdComponent NumBefore>
struct ListIndexOfImpl;

template <typename Type, vtkm::IdComponent NumBefore>
struct ListIndexOfImpl<Type, brigand::list<>, NumBefore>
{
  // Could not find index.
  static constexpr vtkm::IdComponent value = -1;
};

template <typename Type, typename T1, typename... RemainingList, vtkm::IdComponent NumBefore>
struct ListIndexOfImpl<Type, brigand::list<T1, RemainingList...>, NumBefore>
{
  static constexpr vtkm::IdComponent value =
    ListIndexOfImpl<Type, brigand::list<RemainingList...>, NumBefore + 1>::value;
};

template <typename Type, typename... RemainingList, vtkm::IdComponent NumBefore>
struct ListIndexOfImpl<Type, brigand::list<Type, RemainingList...>, NumBefore>
{
  static constexpr vtkm::IdComponent value = NumBefore;
};

//-----------------------------------------------------------------------------
template <class T, class U, class ListTag>
struct intersect_tags
{
  using has_u = ListContainsImpl<U, ListTag>;
  using type = typename std::conditional<has_u::value, brigand::push_back<T, U>, T>::type;
};

//-----------------------------------------------------------------------------
template <typename ListTag1, typename ListTag2>
struct ListIntersect
{
  using type =
    brigand::fold<ListTag1,
                  brigand::list<>,
                  intersect_tags<brigand::_state, brigand::_element, brigand::pin<ListTag2>>>;
};

template <typename ListTag>
struct ListIntersect<vtkm::detail::ListBase<vtkm::detail::UniversalTag>, ListTag>
{
  using type = ListTag;
};

template <typename ListTag>
struct ListIntersect<ListTag, vtkm::detail::ListBase<vtkm::detail::UniversalTag>>
{
  using type = ListTag;
};

template <typename SameListTag>
struct ListIntersect<SameListTag, SameListTag>
{
  using type = SameListTag;
};

template <typename Functor, typename... Args>
VTKM_CONT void ListForEachImpl(Functor&&, brigand::list<>, Args&&...)
{
}

template <typename Functor, typename T1, typename... Args>
VTKM_CONT void ListForEachImpl(Functor&& f, brigand::list<T1>, Args&&... args)
{
  f(T1{}, std::forward<Args>(args)...);
}

template <typename Functor, typename T1, typename T2, typename... Args>
VTKM_CONT void ListForEachImpl(Functor&& f, brigand::list<T1, T2>, Args&&... args)
{
  f(T1{}, std::forward<Args>(args)...);
  f(T2{}, std::forward<Args>(args)...);
}

template <typename Functor, typename T1, typename T2, typename T3, typename... Args>
VTKM_CONT void ListForEachImpl(Functor&& f, brigand::list<T1, T2, T3>, Args&&... args)
{
  f(T1{}, std::forward<Args>(args)...);
  f(T2{}, std::forward<Args>(args)...);
  f(T3{}, std::forward<Args>(args)...);
}

template <typename Functor,
          typename T1,
          typename T2,
          typename T3,
          typename T4,
          typename... ArgTypes,
          typename... Args>
VTKM_CONT void ListForEachImpl(Functor&& f,
                               brigand::list<T1, T2, T3, T4, ArgTypes...>&&,
                               Args&&... args)
{
  f(T1{}, std::forward<Args>(args)...);
  f(T2{}, std::forward<Args>(args)...);
  f(T3{}, std::forward<Args>(args)...);
  f(T4{}, std::forward<Args>(args)...);
  ListForEachImpl(
    std::forward<Functor>(f), brigand::list<ArgTypes...>{}, std::forward<Args>(args)...);
}

template <typename R1, typename R2>
struct ListCrossProductImpl
{
  // This is a lazy Cartesian product generator.
  // This version was settled on as being the best default
  // version as all compilers including Intel handle this
  // implementation without issue for very large cross products
  using type = brigand::reverse_fold<
    brigand::list<R1, R2>,
    brigand::list<brigand::list<>>,
    brigand::lazy::join<brigand::lazy::transform<
      brigand::_2,
      brigand::defer<brigand::lazy::join<brigand::lazy::transform<
        brigand::parent<brigand::_1>,
        brigand::defer<brigand::bind<
          brigand::list,
          brigand::lazy::push_front<brigand::_1, brigand::parent<brigand::_1>>>>>>>>>>;
};

//-----------------------------------------------------------------------------
template <typename List, typename Type>
struct ListAppendUniqueImpl
{
  using type = typename std::conditional<ListContainsImpl<Type, List>::value,
                                         List,
                                         typename ListJoin<List, ListBase<Type>>::type>::type;
};

} // namespace detail

//-----------------------------------------------------------------------------
/// A basic tag for a list of typenames. This struct can be subclassed
/// and still behave like a list tag.
template <typename... ArgTypes>
struct ListTagBase : detail::ListRoot
{
  using list = detail::ListBase<ArgTypes...>;
};
}

#endif //vtk_m_internal_ListTagDetail_h
