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