1134723edSLouis Dionne //===----------------------------------------------------------------------===// 2134723edSLouis Dionne // 3134723edSLouis Dionne // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4134723edSLouis Dionne // See https://llvm.org/LICENSE.txt for license information. 5134723edSLouis Dionne // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6134723edSLouis Dionne // 7134723edSLouis Dionne //===----------------------------------------------------------------------===// 8134723edSLouis Dionne 9134723edSLouis Dionne #ifndef _LIBCPP___ALGORITHM_IS_HEAP_UNTIL_H 10134723edSLouis Dionne #define _LIBCPP___ALGORITHM_IS_HEAP_UNTIL_H 11134723edSLouis Dionne 12134723edSLouis Dionne #include <__algorithm/comp.h> 13b07b5bd7SArthur O'Dwyer #include <__algorithm/comp_ref_type.h> 144d81a46fSArthur O'Dwyer #include <__config> 15134723edSLouis Dionne #include <__iterator/iterator_traits.h> 16134723edSLouis Dionne 17134723edSLouis Dionne #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER) 18134723edSLouis Dionne # pragma GCC system_header 19134723edSLouis Dionne #endif 20134723edSLouis Dionne 21134723edSLouis Dionne _LIBCPP_BEGIN_NAMESPACE_STD 22134723edSLouis Dionne 23b07b5bd7SArthur O'Dwyer template <class _Compare, class _RandomAccessIterator> 245146b57bSNikolas Klauser _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _RandomAccessIterator 259783f28cSLouis Dionne __is_heap_until(_RandomAccessIterator __first, _RandomAccessIterator __last, _Compare&& __comp) { 26134723edSLouis Dionne typedef typename iterator_traits<_RandomAccessIterator>::difference_type difference_type; 27134723edSLouis Dionne difference_type __len = __last - __first; 28134723edSLouis Dionne difference_type __p = 0; 29134723edSLouis Dionne difference_type __c = 1; 30134723edSLouis Dionne _RandomAccessIterator __pp = __first; 319783f28cSLouis Dionne while (__c < __len) { 32134723edSLouis Dionne _RandomAccessIterator __cp = __first + __c; 33134723edSLouis Dionne if (__comp(*__pp, *__cp)) 34134723edSLouis Dionne return __cp; 35134723edSLouis Dionne ++__c; 36134723edSLouis Dionne ++__cp; 37134723edSLouis Dionne if (__c == __len) 38134723edSLouis Dionne return __last; 39134723edSLouis Dionne if (__comp(*__pp, *__cp)) 40134723edSLouis Dionne return __cp; 41134723edSLouis Dionne ++__p; 42134723edSLouis Dionne ++__pp; 43134723edSLouis Dionne __c = 2 * __p + 1; 44134723edSLouis Dionne } 45134723edSLouis Dionne return __last; 46134723edSLouis Dionne } 47134723edSLouis Dionne 48b07b5bd7SArthur O'Dwyer template <class _RandomAccessIterator, class _Compare> 49*17e0686aSNikolas Klauser [[__nodiscard__]] inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _RandomAccessIterator 509783f28cSLouis Dionne is_heap_until(_RandomAccessIterator __first, _RandomAccessIterator __last, _Compare __comp) { 51ed2d3644SNikolas Klauser return std::__is_heap_until(__first, __last, static_cast<__comp_ref_type<_Compare> >(__comp)); 52b07b5bd7SArthur O'Dwyer } 53b07b5bd7SArthur O'Dwyer 54134723edSLouis Dionne template <class _RandomAccessIterator> 55*17e0686aSNikolas Klauser [[__nodiscard__]] inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _RandomAccessIterator 569783f28cSLouis Dionne is_heap_until(_RandomAccessIterator __first, _RandomAccessIterator __last) { 5777a00c0dSLouis Dionne return std::__is_heap_until(__first, __last, __less<>()); 58134723edSLouis Dionne } 59134723edSLouis Dionne 60134723edSLouis Dionne _LIBCPP_END_NAMESPACE_STD 61134723edSLouis Dionne 62134723edSLouis Dionne #endif // _LIBCPP___ALGORITHM_IS_HEAP_UNTIL_H 63