xref: /freebsd/contrib/llvm-project/libcxx/include/__algorithm/set_union.h (revision 5956d97f4b3204318ceb6aa9c77bd0bc6ea87a41)
1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #ifndef _LIBCPP___ALGORITHM_SET_UNION_H
10 #define _LIBCPP___ALGORITHM_SET_UNION_H
11 
12 #include <__algorithm/comp.h>
13 #include <__algorithm/comp_ref_type.h>
14 #include <__algorithm/copy.h>
15 #include <__config>
16 #include <__iterator/iterator_traits.h>
17 
18 #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
19 #pragma GCC system_header
20 #endif
21 
22 _LIBCPP_BEGIN_NAMESPACE_STD
23 
24 template <class _Compare, class _InputIterator1, class _InputIterator2, class _OutputIterator>
25 _LIBCPP_CONSTEXPR_AFTER_CXX17 _OutputIterator
26 __set_union(_InputIterator1 __first1, _InputIterator1 __last1,
27             _InputIterator2 __first2, _InputIterator2 __last2, _OutputIterator __result, _Compare __comp)
28 {
29     for (; __first1 != __last1; ++__result)
30     {
31         if (__first2 == __last2)
32             return _VSTD::copy(__first1, __last1, __result);
33         if (__comp(*__first2, *__first1))
34         {
35             *__result = *__first2;
36             ++__first2;
37         }
38         else
39         {
40             if (!__comp(*__first1, *__first2))
41                 ++__first2;
42             *__result = *__first1;
43             ++__first1;
44         }
45     }
46     return _VSTD::copy(__first2, __last2, __result);
47 }
48 
49 template <class _InputIterator1, class _InputIterator2, class _OutputIterator, class _Compare>
50 inline _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_AFTER_CXX17
51 _OutputIterator
52 set_union(_InputIterator1 __first1, _InputIterator1 __last1,
53           _InputIterator2 __first2, _InputIterator2 __last2, _OutputIterator __result, _Compare __comp)
54 {
55     typedef typename __comp_ref_type<_Compare>::type _Comp_ref;
56     return _VSTD::__set_union<_Comp_ref>(__first1, __last1, __first2, __last2, __result, __comp);
57 }
58 
59 template <class _InputIterator1, class _InputIterator2, class _OutputIterator>
60 inline _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_AFTER_CXX17
61 _OutputIterator
62 set_union(_InputIterator1 __first1, _InputIterator1 __last1,
63           _InputIterator2 __first2, _InputIterator2 __last2, _OutputIterator __result)
64 {
65     return _VSTD::set_union(__first1, __last1, __first2, __last2, __result,
66                           __less<typename iterator_traits<_InputIterator1>::value_type,
67                                  typename iterator_traits<_InputIterator2>::value_type>());
68 }
69 
70 _LIBCPP_END_NAMESPACE_STD
71 
72 #endif // _LIBCPP___ALGORITHM_SET_UNION_H
73