Add a utility class to display a TrackChain as a path

[r2c2.git] / source / 3d / trackchain.cpp

diff --git a/source/3d/trackchain.cpp b/source/3d/trackchain.cpp
new file mode 100644 (file)
index 0000000..3d972d1
--- /dev/null
+++ b/source/3d/trackchain.cpp
@@ -0,0 +1,62 @@
+#include "layout.h"
+#include "path.h"
+#include "trackchain.h"
+#include "track.h"
+
+using namespace Msp;
+
+namespace R2C2 {
+
+TrackChain3D::TrackChain3D(Layout3D &l, TrackChain &r):
+       layout(l),
+       chain(r),
+       layer(0)
+{
+       chain.signal_track_added.connect(sigc::mem_fun(this, &TrackChain3D::track_added));
+       chain.signal_track_removed.connect(sigc::mem_fun(this, &TrackChain3D::track_removed));
+
+       const TrackChain::TrackSet &tracks = chain.get_tracks();
+       for(TrackChain::TrackSet::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
+               track_added(**i);
+}
+
+TrackChain3D::~TrackChain3D()
+{
+       for(PathMap::iterator i=paths.begin(); i!=paths.end(); ++i)
+               delete i->second;
+}
+
+void TrackChain3D::set_color(const GL::Color &c)
+{
+       color = c;
+       for(PathMap::iterator i=paths.begin(); i!=paths.end(); ++i)
+               i->second->set_color(color);
+}
+
+void TrackChain3D::set_layer(float l)
+{
+       layer = l;
+       for(PathMap::iterator i=paths.begin(); i!=paths.end(); ++i)
+               i->second->set_layer(layer);
+}
+
+void TrackChain3D::track_added(Track &track)
+{
+       Track3D &track_3d = layout.get<Track3D>(track);
+       Path3D *path = new Path3D(track_3d);
+       paths.insert(PathMap::value_type(&track, path));
+       path->set_color(color);
+       path->set_layer(layer);
+}
+
+void TrackChain3D::track_removed(Track &track)
+{
+       PathMap::iterator i = paths.find(&track);
+       if(i!=paths.end())
+       {
+               delete i->second;
+               paths.erase(i);
+       }
+}
+
+} // namespace R2C2