[cfe-commits] [libcxx] r104719 - in /libcxx/trunk: include/numeric test/numerics/numeric.ops/numeric.iota/iota.pass.cpp

[Howard Hinnant]

Author: hhinnant
Date: Wed May 26 13:53:44 2010
New Revision: 104719

URL: http://llvm.org/viewvc/llvm-project?rev=104719&view=rev
Log:
[numeric.iota]

Added:
    libcxx/trunk/test/numerics/numeric.ops/numeric.iota/iota.pass.cpp
Modified:
    libcxx/trunk/include/numeric

Modified: libcxx/trunk/include/numeric
URL: http://llvm.org/viewvc/llvm-project/libcxx/trunk/include/numeric?rev=104719&r1=104718&r2=104719&view=diff
==============================================================================
--- libcxx/trunk/include/numeric (original)
+++ libcxx/trunk/include/numeric Wed May 26 13:53:44 2010
@@ -50,6 +50,9 @@
     OutputIterator
     adjacent_difference(InputIterator first, InputIterator last, OutputIterator result, BinaryOperation binary_op);
 
+template <class ForwardIterator, class T>
+    void iota(ForwardIterator first, ForwardIterator last, T value);
+
 }  // std
 
 */
@@ -178,6 +181,15 @@
     return __result;
 }
 
+template <class _ForwardIterator, class _Tp>
+inline _LIBCPP_INLINE_VISIBILITY
+void
+iota(_ForwardIterator __first, _ForwardIterator __last, _Tp __value)
+{
+    for (; __first != __last; ++__first, ++__value)
+        *__first = __value;
+}
+
 _LIBCPP_END_NAMESPACE_STD
 
 #endif  // _LIBCPP_NUMERIC

Added: libcxx/trunk/test/numerics/numeric.ops/numeric.iota/iota.pass.cpp
URL: http://llvm.org/viewvc/llvm-project/libcxx/trunk/test/numerics/numeric.ops/numeric.iota/iota.pass.cpp?rev=104719&view=auto
==============================================================================
--- libcxx/trunk/test/numerics/numeric.ops/numeric.iota/iota.pass.cpp (added)
+++ libcxx/trunk/test/numerics/numeric.ops/numeric.iota/iota.pass.cpp Wed May 26 13:53:44 2010
@@ -0,0 +1,38 @@
+//===----------------------------------------------------------------------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+// <numeric>
+
+// template <class ForwardIterator, class T>
+//     void iota(ForwardIterator first, ForwardIterator last, T value);
+
+#include <numeric>
+#include <cassert>
+
+#include "../iterators.h"
+
+template <class InIter>
+void
+test()
+{
+    int ia[] = {1, 2, 3, 4, 5};
+    int ir[] = {5, 6, 7, 8, 9};
+    const unsigned s = sizeof(ia) / sizeof(ia[0]);
+    std::iota(InIter(ia), InIter(ia+s), 5);
+    for (unsigned i = 0; i < s; ++i)
+        assert(ia[i] == ir[i]);
+}
+
+int main()
+{
+    test<forward_iterator<int*> >();
+    test<bidirectional_iterator<int*> >();
+    test<random_access_iterator<int*> >();
+    test<int*>();
+}