Improve algorithms in several editing tools

[r2c2.git] / source / designer / extendtool.h

diff --git a/source/designer/extendtool.h b/source/designer/extendtool.h
index 979c9205ef7038ba907ef60498c74000b5700de1..8e14fcf89a0cf07ea8bdd8b9bb9e064da5a39602 100644 (file)
--- a/source/designer/extendtool.h
+++ b/source/designer/extendtool.h
@@ -3,13 +3,14 @@
 
 #include <set>
 #include <vector>
-#include "libr2c2/track.h"
+#include "libr2c2/object.h"
+#include "libr2c2/trackiter.h"
 #include "tool.h"
 
 class ExtendTool: public Tool
 {
 private:
-       std::set<R2C2::Object *> objects;
+       std::vector<R2C2::TrackIter> unlinked_endpoints;
        std::vector<R2C2::Track *> extend_tracks;
        bool accepted;