CoordinateSequence: Add applyAt method

libgeos · Nov 18, 2024 · 32a3bf4 · 32a3bf4
1 parent dc4e7b2
commit 32a3bf4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 9 deletions.
diff --git a/include/geos/geom/CoordinateSequence.h b/include/geos/geom/CoordinateSequence.h
@@ -679,6 +679,16 @@ class GEOS_DLL CoordinateSequence {
         }
     }
 
+    template<typename F>
+    auto applyAt(size_t i, F&& fun) const {
+        switch(getCoordinateType()) {
+            case CoordinateType::XYZ:   return fun(getAt<Coordinate>(i));
+            case CoordinateType::XYM:   return fun(getAt<CoordinateXYM>(i));
+            case CoordinateType::XYZM:  return fun(getAt<CoordinateXYZM>(i));
+            default:                    return fun(getAt<CoordinateXY>(i));
+        }
+    }
+
     template<typename F>
     void forEach(F&& fun) const {
         switch(getCoordinateType()) {

diff --git a/src/geom/SimpleCurve.cpp b/src/geom/SimpleCurve.cpp
@@ -252,16 +252,9 @@ SimpleCurve::getPointN(std::size_t n) const
     assert(getFactory());
     assert(points.get());
 
-    if (hasM() || hasZ()) {
-        CoordinateXYZM c;
-        points->getAt(n, c);
+    return points->applyAt(n, [this](const auto& c) {
         return getFactory()->createPoint(c);
-    }
-    else {
-        CoordinateXY c;
-        points->getAt(n, c);
-        return getFactory()->createPoint(c);
-    }
+    });
 }
 
 std::unique_ptr<Point>