X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibmarklin%2Fblockiter.cpp;h=245f9634a5d0b18fab9c9c24985fe3cd34aac67c;hb=975ea87cc7be179618b06291cb2506a2523cad1f;hp=96159577cb6c488aa062ba33298f0499075cddfd;hpb=2707158feba5f9740d51618692914db05410f9a6;p=r2c2.git

diff --git a/source/libmarklin/blockiter.cpp b/source/libmarklin/blockiter.cpp
index 9615957..245f963 100644
--- a/source/libmarklin/blockiter.cpp
+++ b/source/libmarklin/blockiter.cpp
@@ -34,10 +34,18 @@ TrackIter BlockIter::track_iter() const
 	if(!_block)
 		return TrackIter();
 
-	const Block::Endpoint &ep = _block->get_endpoints()[_entry];
+	const Block::Endpoint &ep = _block->get_endpoint(_entry);
 	return TrackIter(ep.track, ep.track_ep);
 }
 
+const Block::Endpoint &BlockIter::endpoint() const
+{
+	if(!_block)
+		throw InvalidState("BlockIter is null");
+
+	return _block->get_endpoint(_entry);
+}
+
 int BlockIter::get_exit(const Route *route) const
 {
 	const vector<Block::Endpoint> &eps = _block->get_endpoints();