xref: /freebsd/contrib/llvm-project/libcxx/include/__algorithm/set_union.h (revision 848ee2a3a8b47c9905fc51fefcf60eb371edbb98)
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 <__config>
13 #include <__algorithm/comp.h>
14 #include <__algorithm/comp_ref_type.h>
15 #include <__algorithm/copy.h>
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_PUSH_MACROS
23 #include <__undef_macros>
24 
25 _LIBCPP_BEGIN_NAMESPACE_STD
26 
27 template <class _Compare, class _InputIterator1, class _InputIterator2, class _OutputIterator>
28 _LIBCPP_CONSTEXPR_AFTER_CXX17 _OutputIterator
29 __set_union(_InputIterator1 __first1, _InputIterator1 __last1,
30             _InputIterator2 __first2, _InputIterator2 __last2, _OutputIterator __result, _Compare __comp)
31 {
32     for (; __first1 != __last1; ++__result)
33     {
34         if (__first2 == __last2)
35             return _VSTD::copy(__first1, __last1, __result);
36         if (__comp(*__first2, *__first1))
37         {
38             *__result = *__first2;
39             ++__first2;
40         }
41         else
42         {
43             if (!__comp(*__first1, *__first2))
44                 ++__first2;
45             *__result = *__first1;
46             ++__first1;
47         }
48     }
49     return _VSTD::copy(__first2, __last2, __result);
50 }
51 
52 template <class _InputIterator1, class _InputIterator2, class _OutputIterator, class _Compare>
53 inline _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_AFTER_CXX17
54 _OutputIterator
55 set_union(_InputIterator1 __first1, _InputIterator1 __last1,
56           _InputIterator2 __first2, _InputIterator2 __last2, _OutputIterator __result, _Compare __comp)
57 {
58     typedef typename __comp_ref_type<_Compare>::type _Comp_ref;
59     return _VSTD::__set_union<_Comp_ref>(__first1, __last1, __first2, __last2, __result, __comp);
60 }
61 
62 template <class _InputIterator1, class _InputIterator2, class _OutputIterator>
63 inline _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_AFTER_CXX17
64 _OutputIterator
65 set_union(_InputIterator1 __first1, _InputIterator1 __last1,
66           _InputIterator2 __first2, _InputIterator2 __last2, _OutputIterator __result)
67 {
68     return _VSTD::set_union(__first1, __last1, __first2, __last2, __result,
69                           __less<typename iterator_traits<_InputIterator1>::value_type,
70                                  typename iterator_traits<_InputIterator2>::value_type>());
71 }
72 
73 _LIBCPP_END_NAMESPACE_STD
74 
75 _LIBCPP_POP_MACROS
76 
77 #endif // _LIBCPP___ALGORITHM_SET_UNION_H
78