X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fdesigner%2Fselection.h;h=89f7100e3e6e854945219bf074ad7f4ff3388e4f;hb=dfeb2d2896fe18e55b1f6c4a8028b7cfb878dede;hp=9d49c76d3b924e467f9a60d6efddaade34e341bb;hpb=283542ae424b94786652fbf0c67b883b63a8e7a0;p=r2c2.git

diff --git a/source/designer/selection.h b/source/designer/selection.h
index 9d49c76..89f7100 100644
--- a/source/designer/selection.h
+++ b/source/designer/selection.h
@@ -1,16 +1,9 @@
-/* $Id$
-
-This file is part of the MSP Märklin suite
-Copyright © 2006-2008 Mikkosoft Productions, Mikko Rasa
-Distributed under the GPL
-*/
-
 #ifndef SELECTION_H_
 #define SELECTION_H_
 
 #include <set>
 #include <sigc++/sigc++.h>
-#include "3d/track.h"
+#include "libr2c2/object.h"
 
 class Selection
 {
@@ -18,26 +11,47 @@ public:
 	sigc::signal<void> signal_changed;
 
 private:
-	std::set<Marklin::Track *> tracks;
+	std::set<R2C2::Object *> objects;
 
 public:
-	const std::set<Marklin::Track *> &get_tracks() const { return tracks; }
-	Marklin::Track *get_track() const;
-	unsigned size() const { return tracks.size(); }
+	const std::set<R2C2::Object *> &get_objects() const { return objects; }
+	R2C2::Object *get_object() const;
+
+	template<typename T>
+	std::set<T *> get_objects() const
+	{
+		std::set<T *> result;
+		for(std::set<R2C2::Object *>::const_iterator i=objects.begin(); i!=objects.end(); ++i)
+			if(T *to = dynamic_cast<T *>(*i))
+				result.insert(to);
+		return result;
+	}
+
+	template<typename T>
+	T *get_object() const
+	{
+		for(std::set<R2C2::Object *>::const_iterator i=objects.begin(); i!=objects.end(); ++i)
+			if(T *to = dynamic_cast<T *>(*i))
+				return to;
+		return 0;
+	}
+
+	unsigned size() const { return objects.size(); }
+	bool empty() const { return objects.empty(); }
 
 	void clear();
 
 	template<typename Iter>
 	void replace(Iter begin, Iter end)
 	{
-		tracks.clear();
-		tracks.insert(begin, end);
+		objects.clear();
+		objects.insert(begin, end);
 		signal_changed.emit();
 	}
 
-	void add_track(Marklin::Track *);
-	void remove_track(Marklin::Track *);
-	void toggle_track(Marklin::Track *);
+	void add_object(R2C2::Object *);
+	void remove_object(R2C2::Object *);
+	void toggle_object(R2C2::Object *);
 
 	void select_more();
 	void select_linked();