X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fgame%2Fentity.h;h=c4cf83c119b26bce36fce09ac1238be2bebcd1d2;hb=76d859d1429782901799c9886f833cd331b670ce;hp=922f552f4aaac7f642f0cfeda14de22009614da4;hpb=248d62f7240d342982ade65a510be912b867fe49;p=libs%2Fgame.git

diff --git a/source/game/entity.h b/source/game/entity.h
index 922f552..c4cf83c 100644
--- a/source/game/entity.h
+++ b/source/game/entity.h
@@ -2,10 +2,14 @@
 #define MSP_GAME_ENTITY_H_
 
 #include "handle.h"
+#include "owned.h"
 
 namespace Msp::Game {
 
 class Component;
+class Stage;
+class Transform;
+struct TransformValues;
 
 class hierarchy_error: public std::logic_error
 {
@@ -15,26 +19,47 @@ public:
 
 class Entity
 {
+public:
+	enum TransformTag { NO_TRANSFORM };
+
 private:
 	Handle<Entity> parent;
 	std::vector<Handle<Component>> components;
 	std::vector<Handle<Entity>> children;
+	Owned<Transform> transform;
 
 public:
-	Entity(Handle<Entity>);
+	Entity(Handle<Entity>, TransformTag);
+	Entity(Handle<Entity>, const TransformValues &);
 	virtual ~Entity();
 
 	void add_component(Handle<Component>);
 	void remove_component(Handle<Component>);
 
+	template<typename T>
+	Handle<T> get_component();
+
+	Handle<Transform> get_transform() const { return transform; }
+
 	void add_child(Handle<Entity>);
 	void remove_child(Handle<Entity>);
 
 	Handle<Entity> get_parent() const { return parent; }
 	Handle<Entity> get_root();
+	const std::vector<Handle<Entity>> &get_children() const { return children; }
+	Stage &get_stage();
 };
 
 
+template<typename T>
+inline Handle<T> Entity::get_component()
+{
+	for(Handle<Component> c: components)
+		if(Handle<T> tc = dynamic_handle_cast<T>(c))
+			return tc;
+	return nullptr;
+}
+
 inline Handle<Entity> Entity::get_root()
 {
 	Handle<Entity> e = Handle<Entity>::from_object(this);