Rewrite goal initialization for TrainRouteMetric

[r2c2.git] / source / libr2c2 / trackchain.h

diff --git a/source/libr2c2/trackchain.h b/source/libr2c2/trackchain.h
index 729a89cd9c6910882fe641721f86751e93df21e0..74d194884e48eac531991f3f552bd95302abd612 100644 (file)
--- a/source/libr2c2/trackchain.h
+++ b/source/libr2c2/trackchain.h
@@ -4,6 +4,7 @@
 #include <set>
 #include <stdexcept>
 #include <sigc++/trackable.h>
+#include <msp/datafile/statement.h>
 #include "trackiter.h"
 
 namespace R2C2 {
@@ -22,6 +23,13 @@ public:
 class TrackChain: public sigc::trackable
 {
 public:
+       enum Direction
+       {
+               UNSPECIFIED,
+               UP,
+               DOWN
+       };
+
        typedef std::set<Track *> TrackSet;
 
 protected:
@@ -62,11 +70,22 @@ private:
 public:
        const TrackSet &get_tracks() const { return tracks; }
        bool has_track(Track &) const;
+       virtual TrackIter iter_for(Track &, Direction) const;
+       TrackIter get_end(unsigned) const;
+       virtual TrackIter get_end(Direction) const { return get_end(0); }
+       bool is_loop() const;
 
 private:
        void object_removed(Object &);
+
+public:
+       virtual Msp::DataFile::Statement save_reference() const = 0;
 };
 
+
+void operator<<(Msp::LexicalConverter &, TrackChain::Direction);
+void operator>>(const Msp::LexicalConverter &, TrackChain::Direction &);
+
 } // namespace R2C2
 
 #endif