commit 407c29208009a6b767a8424c28a0909eb5871996
author Wenzel Jakob <wenzel.jakob@epfl.ch> Tue Jun 21 15:43:01 2016 +0200
committer GitHub <noreply@github.com> Tue Jun 21 15:43:01 2016 +0200
tree d9011bb03ae828716e60e7e532c60b7ddd87945f
parent 9edfd20d40405eba19d4bb4651690a25f3a00cd3
parent daed1abc9c6fb3c43b6c633391316249fb263dad

Merge pull request #247 from aldanor/iterators

Use prefix increment in make_iterator

include/pybind11/common.h

diff --git a/include/pybind11/common.h b/include/pybind11/common.h
index 560be43..daafa7d 100644
--- a/include/pybind11/common.h
+++ b/include/pybind11/common.h
@@ -116,7 +116,7 @@
 extern "C" {
     struct _Py_atomic_address { void *value; };
     PyAPI_DATA(_Py_atomic_address) _PyThreadState_Current;
-};
+}
 #endif
 
 #define PYBIND11_TRY_NEXT_OVERLOAD ((PyObject *) 1) // special failure return code

include/pybind11/pybind11.h

diff --git a/include/pybind11/pybind11.h b/include/pybind11/pybind11.h
index 1438e34..d472531 100644
--- a/include/pybind11/pybind11.h
+++ b/include/pybind11/pybind11.h
@@ -1028,7 +1028,10 @@
     (void) wr.release();
 }
 
-template <typename Iterator> struct iterator_state { Iterator it, end; };
+template <typename Iterator> struct iterator_state {
+    Iterator it, end;
+    bool first;
+};
 
 NAMESPACE_END(detail)
 
@@ -1044,13 +1047,17 @@
         class_<state>(handle(), "")
             .def("__iter__", [](state &s) -> state& { return s; })
             .def("__next__", [](state &s) -> ValueType {
+                if (!s.first)
+                    ++s.it;
+                else
+                    s.first = false;
                 if (s.it == s.end)
                     throw stop_iteration();
-                return *s.it++;
+                return *s.it;
             }, return_value_policy::reference_internal, std::forward<Extra>(extra)...);
     }
 
-    return (iterator) cast(state { first, last });
+    return (iterator) cast(state { first, last, true });
 }
 
 template <typename Type, typename... Extra> iterator make_iterator(Type &value, Extra&&... extra) {