//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

// bitset(string, pos, n, zero, one); // constexpr since C++23

#include <cuda/std/version>

#ifndef _LIBCUDACXX_HAS_STRING
int main(int, char**)
{
  return 0;
}
#else

#  include <cuda/std/algorithm> // for 'min' and 'max'
#  include <cuda/std/bitset>
#  include <cuda/std/cassert>
#  include <cuda/std/stdexcept> // for 'invalid_argument'
#  include <cuda/std/string>
#  include <cuda/std/type_traits>

#  include "test_macros.h"

template <cuda::std::size_t N>
constexpr void test_string_ctor()
{
#  ifndef TEST_HAS_NO_EXCEPTIONS
  if (!TEST_IS_CONSTANT_EVALUATED)
  {
    try
    {
      cuda::std::string s("xxx1010101010xxxx");
      cuda::std::bitset<N> v(s, s.size() + 1);
      assert(false);
    }
    catch (cuda::std::out_of_range&)
    {}
    try
    {
      cuda::std::string s("xxx1010101010xxxx");
      cuda::std::bitset<N> v(s, s.size() + 1, 10);
      assert(false);
    }
    catch (cuda::std::out_of_range&)
    {}
    try
    {
      cuda::std::string s("xxx1010101010xxxx");
      cuda::std::bitset<N> v(s);
      assert(false);
    }
    catch (cuda::std::invalid_argument&)
    {}
    try
    {
      cuda::std::string s("xxx1010101010xxxx");
      cuda::std::bitset<N> v(s, 2);
      assert(false);
    }
    catch (cuda::std::invalid_argument&)
    {}
    try
    {
      cuda::std::string s("xxx1010101010xxxx");
      cuda::std::bitset<N> v(s, 2, 10);
      assert(false);
    }
    catch (cuda::std::invalid_argument&)
    {}
    try
    {
      cuda::std::string s("xxxbababababaxxxx");
      cuda::std::bitset<N> v(s, 2, 10, 'a', 'b');
      assert(false);
    }
    catch (cuda::std::invalid_argument&)
    {}
  }
#  endif // TEST_HAS_NO_EXCEPTIONS

  static_assert(!cuda::std::is_convertible<cuda::std::string, cuda::std::bitset<N>>::value, "");
  static_assert(cuda::std::is_constructible<cuda::std::bitset<N>, cuda::std::string>::value, "");
  {
    cuda::std::string s("1010101010");
    cuda::std::bitset<N> v(s);
    cuda::std::size_t M = cuda::std::min<cuda::std::size_t>(v.size(), 10);
    for (cuda::std::size_t i = 0; i < M; ++i)
    {
      assert(v[i] == (s[M - 1 - i] == '1'));
    }
    for (cuda::std::size_t i = 10; i < v.size(); ++i)
    {
      assert(v[i] == false);
    }
  }
  {
    cuda::std::string s("xxx1010101010");
    cuda::std::bitset<N> v(s, 3);
    cuda::std::size_t M = cuda::std::min<cuda::std::size_t>(v.size(), 10);
    for (cuda::std::size_t i = 0; i < M; ++i)
    {
      assert(v[i] == (s[3 + M - 1 - i] == '1'));
    }
    for (cuda::std::size_t i = 10; i < v.size(); ++i)
    {
      assert(v[i] == false);
    }
  }
  {
    cuda::std::string s("xxx1010101010xxxx");
    cuda::std::bitset<N> v(s, 3, 10);
    cuda::std::size_t M = cuda::std::min<cuda::std::size_t>(v.size(), 10);
    for (cuda::std::size_t i = 0; i < M; ++i)
    {
      assert(v[i] == (s[3 + M - 1 - i] == '1'));
    }
    for (cuda::std::size_t i = 10; i < v.size(); ++i)
    {
      assert(v[i] == false);
    }
  }
  {
    cuda::std::string s("xxx1a1a1a1a1axxxx");
    cuda::std::bitset<N> v(s, 3, 10, 'a');
    cuda::std::size_t M = cuda::std::min<cuda::std::size_t>(v.size(), 10);
    for (cuda::std::size_t i = 0; i < M; ++i)
    {
      assert(v[i] == (s[3 + M - 1 - i] == '1'));
    }
    for (cuda::std::size_t i = 10; i < v.size(); ++i)
    {
      assert(v[i] == false);
    }
  }
  {
    cuda::std::string s("xxxbababababaxxxx");
    cuda::std::bitset<N> v(s, 3, 10, 'a', 'b');
    cuda::std::size_t M = cuda::std::min<cuda::std::size_t>(v.size(), 10);
    for (cuda::std::size_t i = 0; i < M; ++i)
    {
      assert(v[i] == (s[3 + M - 1 - i] == 'b'));
    }
    for (cuda::std::size_t i = 10; i < v.size(); ++i)
    {
      assert(v[i] == false);
    }
  }
}

struct Nonsense
{
  virtual ~Nonsense() {}
};

constexpr void test_for_non_eager_instantiation()
{
  // Ensure we don't accidentally instantiate `cuda::std::basic_string<Nonsense>`
  // since it may not be well formed and can cause an error in the
  // non-immediate context.
  static_assert(!cuda::std::is_constructible<cuda::std::bitset<3>, Nonsense*>::value, "");
  static_assert(
    !cuda::std::is_constructible<cuda::std::bitset<3>, Nonsense*, cuda::std::size_t, Nonsense&, Nonsense&>::value, "");
}

constexpr bool test()
{
  test_string_ctor<0>();
  test_string_ctor<1>();
  test_string_ctor<31>();
  test_string_ctor<32>();
  test_string_ctor<33>();
  test_string_ctor<63>();
  test_string_ctor<64>();
  test_string_ctor<65>();
  test_string_ctor<1000>();
  test_for_non_eager_instantiation();

  return true;
}

int main(int, char**)
{
  test();
#  if TEST_STD_VER >= 2023
  static_assert(test(), "");
#  endif

  return 0;
}

#endif
