Support directionality for zones

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

diff --git a/source/libr2c2/trackchain.cpp b/source/libr2c2/trackchain.cpp
index 0c3282ad4845a4893f2c776fef7d2f49d2c48413..f354c6eac7054fb0d9f6c4c59f30578981e33acf 100644 (file)
--- a/source/libr2c2/trackchain.cpp
+++ b/source/libr2c2/trackchain.cpp
@@ -1,9 +1,11 @@
+#include <msp/strings/format.h>
 #include <msp/strings/utils.h>
 #include "layout.h"
 #include "track.h"
 #include "trackchain.h"
 
 using namespace std;
+using namespace Msp;
 
 namespace R2C2 {
 
@@ -135,6 +137,29 @@ bool TrackChain::has_track(Track &t) const
        return tracks.count(&t);
 }
 
+TrackIter TrackChain::iter_for(Track &t, Direction d) const
+{
+       if(!tracks.count(&t))
+               return TrackIter();
+       else if(d==UNSPECIFIED)
+               return TrackIter(&t, 0);
+       else
+               return TrackIter();
+}
+
+TrackIter TrackChain::get_end(unsigned i) const
+{
+       if(i>=2)
+               throw invalid_argument("TrackChain::get_end");
+
+       if(!ends[0])
+               return TrackIter();
+       else if(i==1 && !ends[1])
+               return ends[0].reverse();
+       else
+               return ends[i];
+}
+
 bool TrackChain::is_loop() const
 {
        return !tracks.empty() && !ends[0] && !ends[1];
@@ -151,4 +176,29 @@ void TrackChain::object_removed(Object &obj)
        }
 }
 
+
+void operator<<(LexicalConverter &conv, TrackChain::Direction dir)
+{
+       switch(dir)
+       {
+       case TrackChain::UNSPECIFIED: conv.result("UNSPECIFIED"); return;
+       case TrackChain::UP: conv.result("UP"); return;
+       case TrackChain::DOWN: conv.result("DOWN"); return;
+       default: throw lexical_error(format("conversion of Direction(%d) to string", dir));
+       }
+}
+
+void operator>>(const LexicalConverter &conv, TrackChain::Direction &dir)
+{
+       const string &str = conv.get();
+       if(str=="UNSPECIFIED")
+               dir = TrackChain::UNSPECIFIED;
+       else if(str=="UP")
+               dir = TrackChain::UP;
+       else if(str=="DOWN")
+               dir = TrackChain::DOWN;
+       else
+               throw lexical_error(format("conversion of '%s' to Direction", str));
+}
+
 } // namespace R2C2