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