flow: rw/ro sentinel support

Author: skypjack

src/entt/graph/flow.hpp

@@ -287,7 +287,7 @@ class basic_flow {
      * @param last An iterator past the last element of the range of elements.
      * @return This flow builder.
      */
-    basic_flow &ro(stl::input_iterator auto first, stl::input_iterator auto last) {
+    basic_flow &ro(stl::input_iterator auto first, stl::sentinel_for<decltype(first)> auto last) {
         static_assert(stl::is_convertible_v<decltype(*first), id_type>, "Unexpected value type");
 
         for(; first != last; ++first) {
@@ -313,7 +313,7 @@ class basic_flow {
      * @param last An iterator past the last element of the range of elements.
      * @return This flow builder.
      */
-    basic_flow &rw(stl::input_iterator auto first, stl::input_iterator auto last) {
+    basic_flow &rw(stl::input_iterator auto first, stl::sentinel_for<decltype(first)> auto last) {
         static_assert(stl::is_convertible_v<decltype(*first), id_type>, "Unexpected value type");
 
         for(; first != last; ++first) {

test/entt/graph/flow.cpp

@@ -194,6 +194,20 @@ TEST(Flow, RangeRW) {
     ASSERT_FALSE(graph.contains(1u, 0u));
 }
 
+TEST(Flow, RangeWithSentinel) {
+    const std::array<entt::id_type, 3u> res{0u, 1u};
+    entt::flow flow{};
+
+    flow.bind(0u).rw(std::counted_iterator{res.begin(), 1}, std::default_sentinel);
+    flow.bind(1u).ro(std::counted_iterator{res.begin(), 2}, std::default_sentinel);
+
+    const auto graph = flow.graph();
+
+    ASSERT_EQ(graph.size(), 2u);
+    ASSERT_TRUE(graph.contains(0u, 1u));
+    ASSERT_FALSE(graph.contains(1u, 0u));
+}
+
 TEST(Flow, Graph) {
     using namespace entt::literals;