test handle dups (#10)

* test handle dups

* bind more functions

* update headers

Co-authored-by: TANG ZHIXIONG <dvora4tzx@gmail.com>

Author: district10

CMakeLists.txt

@@ -22,7 +22,7 @@ elseif(CMAKE_BUILD_TYPE STREQUAL "Release")
   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O3")
 endif()
 
-include_directories(BEFORE ${PROJECT_SOURCE_DIR}/headers/include)
+include_directories(${PROJECT_SOURCE_DIR}/headers/include)
 
 set(CMAKE_CXX_STANDARD 17)
 set(PYBIND11_CPP_STANDARD -std=c++17)

headers

@@ -284,6 +284,12 @@ struct PolylineRuler
         }
         return *ranges_;
     }
+    double range(int seg_idx) const { return ranges()[seg_idx]; }
+    double range(int seg_idx, double t) const
+    {
+        auto &ranges = this->ranges();
+        return ranges[seg_idx] * (1.0 - t) + ranges[seg_idx + 1] * t;
+    }
 
     double length() const { return ranges()[N_ - 1]; }
 
@@ -386,6 +392,15 @@ struct PolylineRuler
                            is_wgs84_);
     }
 
+    Eigen::Vector3d at(double range) const { return extended_along(range); }
+    Eigen::Vector3d at(int seg_idx) const { return polyline_.row(seg_idx); }
+    Eigen::Vector3d at(int seg_idx, double t) const
+    {
+        return interpolate(polyline_.row(seg_idx),     //
+                           polyline_.row(seg_idx + 1), //
+                           t, is_wgs84_);
+    }
+
     std::pair<Eigen::Vector3d, Eigen::Vector3d>
     arrow(double range, bool smooth_joint = true) const
     {

src/polyline_ruler.hpp

@@ -87,6 +87,11 @@ CUBAO_INLINE void bind_polyline_ruler(py::module &m)
             "polyline"_a, py::kw_only(), "is_wgs84"_a = false)
         .def("ranges", py::overload_cast<>(&PolylineRuler::ranges, py::const_),
              rvp::reference_internal)
+        .def("range", py::overload_cast<int>(&PolylineRuler::range, py::const_),
+             "segment_index"_a)
+        .def("range",
+             py::overload_cast<int, double>(&PolylineRuler::range, py::const_),
+             py::kw_only(), "segment_index"_a, "t"_a)
         //
         .def("length", &PolylineRuler::length)
         //
@@ -102,6 +107,13 @@ CUBAO_INLINE void bind_polyline_ruler(py::module &m)
              py::overload_cast<double, bool>(&PolylineRuler::dir, py::const_),
              "range"_a, py::kw_only(), "smooth_joint"_a = true)
         .def("extended_along", &PolylineRuler::extended_along, "range"_a)
+        .def("at", py::overload_cast<double>(&PolylineRuler::at, py::const_),
+             "range"_a)
+        .def("at", py::overload_cast<int>(&PolylineRuler::at, py::const_),
+             "segment_index"_a)
+        .def("at",
+             py::overload_cast<int, double>(&PolylineRuler::at, py::const_),
+             "segment_index"_a, py::kw_only(), "t"_a)
         .def("arrow", &PolylineRuler::arrow, "range"_a, //
              py::kw_only(), "smooth_joint"_a = true)
         .def("arrows",

src/pybind11_polyline_ruler.hpp

@@ -303,3 +303,20 @@ def test_polyline_ruler():
     assert len(ranges) == 12 and ranges[-1] == 110
     ranges, xyzs, dirs = ruler.arrows(10.0 - 1e-9)
     assert len(ranges) == 13 and ranges[-1] == 110
+
+
+def test_polyline_ruler_duplicates():
+    ruler = PolylineRuler([[0, 0, 0], [10, 0, 0], [10, 0, 0], [100, 0, 0]])
+    assert np.all(ruler.ranges() == [0, 10, 10, 100])
+    assert np.all(ruler.dirs() == [[1, 0, 0], [1, 0, 0], [1, 0, 0]])
+
+
+def test_polyline_ruler_at():
+    ruler = PolylineRuler([[0, 0, 0], [10, 0, 0], [10, 0, 0], [100, 0, 0]])
+    assert np.all(ruler.ranges() == [0, 10, 10, 100])
+    xyz = ruler.at(2.0)
+    assert np.all(xyz == [2, 0, 0])
+    xyz = ruler.at(2)
+    assert np.all(xyz == [10, 0, 0])
+    xyz = ruler.at(2, t=0.5)
+    assert np.all(xyz == [55, 0, 0])