Don't crash if a train has no router

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

diff --git a/source/designer/extendtool.h b/source/designer/extendtool.h
index 8a7ebebe6fadd75de56549756c0e61bba6acb2b0..866d258a68e6e4f72b6ac5b1c99e187e2b909b82 100644 (file)
--- a/source/designer/extendtool.h
+++ b/source/designer/extendtool.h
@@ -1,6 +1,7 @@
 #ifndef EXTENDTOOL_H_
 #define EXTENDTOOL_H_
 
+#include <map>
 #include <set>
 #include <vector>
 #include "libr2c2/object.h"
@@ -10,11 +11,13 @@
 class ExtendTool: public Tool
 {
 private:
+       std::map<float, const R2C2::TrackType *> types_by_length;
+       unsigned max_preference;
        std::vector<R2C2::TrackIter> unlinked_endpoints;
        std::vector<R2C2::Track *> extend_tracks;
 
 public:
-       ExtendTool(Designer &, Msp::Input::Mouse &, const std::set<R2C2::Object *> &);
+       ExtendTool(Designer &, Msp::Input::Keyboard &, Msp::Input::Mouse &, const std::set<R2C2::Object *> &);
        virtual ~ExtendTool();
 
        void connect();