Merge pull request #429 from kylelutz/extents-begin-end

Add begin() and end() methods to extents<N>

Author: kylelutz

include/boost/compute/utility/extents.hpp

@@ -35,6 +35,9 @@ class extents
 public:
     typedef size_t size_type;
     static const size_type static_size = N;
+    typedef boost::array<size_t, N> array_type;
+    typedef typename array_type::iterator iterator;
+    typedef typename array_type::const_iterator const_iterator;
 
     /// Creates an extents object with each component set to zero.
     ///
@@ -99,6 +102,36 @@ class extents
         return m_extents.data();
     }
 
+    iterator begin()
+    {
+        return m_extents.begin();
+    }
+
+    const_iterator begin() const
+    {
+        return m_extents.begin();
+    }
+
+    const_iterator cbegin() const
+    {
+        return m_extents.cbegin();
+    }
+
+    iterator end()
+    {
+        return m_extents.end();
+    }
+
+    const_iterator end() const
+    {
+        return m_extents.end();
+    }
+
+    const_iterator cend() const
+    {
+        return m_extents.cend();
+    }
+
     /// Returns a reference to the extent at \p index.
     size_t& operator[](size_t index)
     {
@@ -124,7 +157,7 @@ class extents
     }
 
 private:
-    boost::array<size_t, N> m_extents;
+    array_type m_extents;
 };
 
 } // end compute namespace

test/test_extents.cpp

@@ -11,6 +11,9 @@
 #define BOOST_TEST_MODULE TestExtents
 #include <boost/test/unit_test.hpp>
 
+#include <algorithm>
+#include <vector>
+
 #include <boost/compute/utility/dim.hpp>
 #include <boost/compute/utility/extents.hpp>
 
@@ -79,4 +82,15 @@ BOOST_AUTO_TEST_CASE(empty_dim)
     BOOST_CHECK(compute::dim<3>() == compute::dim(0, 0, 0));
 }
 
+BOOST_AUTO_TEST_CASE(copy_to_vector)
+{
+    compute::extents<3> exts = compute::dim(4, 5, 6);
+
+    std::vector<size_t> vec(3);
+    std::copy(exts.begin(), exts.end(), vec.begin());
+    BOOST_CHECK_EQUAL(vec[0], 4);
+    BOOST_CHECK_EQUAL(vec[1], 5);
+    BOOST_CHECK_EQUAL(vec[2], 6);
+}
+
 BOOST_AUTO_TEST_SUITE_END()