Add some shortcut functions for getting endpoints

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

diff --git a/source/libmarklin/trackiter.h b/source/libmarklin/trackiter.h
index e7560611277ea06c7a45a9fa68fa93327ea3e8f6..e2f1d85dfde9e682cc0d3a7b2f555bdd0d0313d9 100644 (file)
--- a/source/libmarklin/trackiter.h
+++ b/source/libmarklin/trackiter.h
@@ -10,6 +10,7 @@ Distributed under the GPL
 
 #include <set>
 #include <msp/core/refptr.h>
+#include "tracktype.h"
 
 namespace Marklin {
 
@@ -30,6 +31,7 @@ public:
 
        Track *track() const { return _track; }
        unsigned entry() const { return _entry; }
+       const TrackType::Endpoint &endpoint() const;
 
 private:
        int get_exit(unsigned) const;