Add some shortcut functions for getting endpoints

[r2c2.git] / source / libmarklin / trackiter.cpp

diff --git a/source/libmarklin/trackiter.cpp b/source/libmarklin/trackiter.cpp
index 5f633997cc26e3dcdf0ac0354946325928f77377..b98022862dd903c25779c507f5f2dfc540199e5b 100644 (file)
--- a/source/libmarklin/trackiter.cpp
+++ b/source/libmarklin/trackiter.cpp
@@ -29,6 +29,14 @@ TrackIter::TrackIter(Track *t, unsigned e):
                throw InvalidParameterValue("Endpoint index not valid for track");
 }
 
+const TrackType::Endpoint &TrackIter::endpoint() const
+{
+       if(!_track)
+               throw InvalidState("TrackIter is null");
+
+       return _track->get_type().get_endpoint(_entry);
+}
+
 int TrackIter::get_exit(unsigned path) const
 {
        const vector<TrackType::Endpoint> &eps = _track->get_type().get_endpoints();