GRAPHICS: Add Renderable::collectMaterials()

xoreos · Jan 30, 2014 · 96a20c7 · 96a20c7
1 parent 3abaa6b
commit 96a20c7
Show file tree

Hide file tree

Showing 9 changed files with 76 additions and 1 deletion.
diff --git a/src/graphics/aurora/cube.cpp b/src/graphics/aurora/cube.cpp
@@ -34,6 +34,7 @@
 #include <OgreMesh.h>
 #include <OgreMeshManager.h>
 #include <OgreEntity.h>
+#include <OgreSubEntity.h>
 
 #include "common/ustring.h"
 #include "common/util.h"
@@ -186,6 +187,17 @@ void Cube::setSelectable(bool selectable) {
 	Renderable::setSelectable(selectable);
 }
 
+void Cube::collectMaterials(std::list<Ogre::MaterialPtr> &materials, bool makeDynamic, bool makeTransparent) {
+	Ogre::MaterialPtr material = _entity->getSubEntity(0)->getMaterial();
+	if (makeDynamic)
+		_entity->setMaterial((material = MaterialMan.makeDynamic(material)));
+
+	if (makeTransparent)
+		MaterialMan.setTransparent(material, true);
+
+	materials.push_back(material);
+}
+
 } // End of namespace Aurora
 
 } // End of namespace Graphics
diff --git a/src/graphics/aurora/cube.h b/src/graphics/aurora/cube.h
@@ -53,6 +53,10 @@ class Cube : public Renderable {
 	/** Change whether the cube can be selected (picked) by the user. */
 	void setSelectable(bool selectable);
 
+protected:
+	/** Collect all materials used in the cube, optionally making them dynamic and/or transparent as well. */
+	void collectMaterials(std::list<Ogre::MaterialPtr> &materials, bool makeDynamic = false, bool makeTransparent = false);
+
 private:
 	Ogre::Entity *_entity;
 };

diff --git a/src/graphics/aurora/model.cpp b/src/graphics/aurora/model.cpp
@@ -31,6 +31,7 @@
 #include <OgreSceneManager.h>
 #include <OgreMeshManager.h>
 #include <OgreEntity.h>
+#include <OgreSubEntity.h>
 #include <OgreAnimation.h>
 #include <OgreAnimationState.h>
 
@@ -257,6 +258,16 @@ void Model::showBoundingBox(bool show) {
 	}
 }
 
+void Model::collectMaterials(std::list<Ogre::MaterialPtr> &materials, bool makeDynamic, bool makeTransparent) {
+	for (EntityList::iterator e = _entities.begin(); e != _entities.end(); ++e) {
+		Ogre::MaterialPtr material = (*e)->getSubEntity(0)->getMaterial();
+		if (makeDynamic)
+			(*e)->setMaterial((material = MaterialMan.makeDynamic(material)));
+
+		materials.push_back(material);
+	}
+}
+
 void Model::createNode(NodeEntity *&nodeEntity, State *state, Common::UString name, Ogre::SceneNode *parent) {
 	if (name.empty())
 		name = Common::generateIDRandomString();

diff --git a/src/graphics/aurora/model.h b/src/graphics/aurora/model.h
@@ -127,8 +127,10 @@ class Model : public Renderable {
 	State *_currentState;
 
 
-	void setState(State *state);
+	/** Collect all materials used in the model, optionally making them dynamic and/or transparent as well. */
+	void collectMaterials(std::list<Ogre::MaterialPtr> &materials, bool makeDynamic = false, bool makeTransparent = false);
 
+	void setState(State *state);
 
 	// General loading helpers
 

diff --git a/src/graphics/aurora/quad.cpp b/src/graphics/aurora/quad.cpp
@@ -360,6 +360,16 @@ void Quad::update() {
 	_needUpdate = false;
 }
 
+void Quad::collectMaterials(std::list<Ogre::MaterialPtr> &materials, bool makeDynamic, bool makeTransparent) {
+	if (makeDynamic)
+		_entity->setMaterial((_material = MaterialMan.makeDynamic(_material)));
+
+	if (makeTransparent)
+		MaterialMan.setTransparent(_material, true);
+
+	materials.push_back(_material);
+}
+
 } // End of namespace Aurora
 
 } // End of namespace Graphics
diff --git a/src/graphics/aurora/quad.h b/src/graphics/aurora/quad.h
@@ -76,6 +76,11 @@ class Quad : public Renderable {
 	void setVisible(bool visible);
 
 
+protected:
+	/** Collect all materials used in the quad, optionally making them dynamic and/or transparent as well. */
+	void collectMaterials(std::list<Ogre::MaterialPtr> &materials, bool makeDynamic = false, bool makeTransparent = false);
+
+
 private:
 	Common::UString _textureName;
 

diff --git a/src/graphics/aurora/text.cpp b/src/graphics/aurora/text.cpp
@@ -215,6 +215,26 @@ void Text::setSelectable(bool selectable) {
 	Renderable::setSelectable(selectable);
 }
 
+void Text::collectMaterials(std::list<Ogre::MaterialPtr> &materials, bool makeDynamic, bool makeTransparent) {
+	for (Ogre::Node::ChildNodeIterator c = _rootNode->getChildIterator(); c.hasMoreElements(); c.moveNext()) {
+		Ogre::SceneNode *node = (Ogre::SceneNode *) c.current()->second;
+
+		for (uint i = 0; i < node->numAttachedObjects(); i++) {
+			Ogre::MovableObject *object = node->getAttachedObject(i);
+
+			Ogre::Entity *entity = dynamic_cast<Ogre::Entity *>(object);
+			if (!entity)
+				continue;
+
+			for (uint j = 0; j < entity->getNumSubEntities(); j++) {
+				Ogre::SubEntity *subEntity = entity->getSubEntity(j);
+
+				materials.push_back(subEntity->getMaterial());
+			}
+		}
+	}
+}
+
 } // End of namespace Aurora
 
 } // End of namespace Graphics
diff --git a/src/graphics/aurora/text.h b/src/graphics/aurora/text.h
@@ -63,6 +63,10 @@ class Text : public Renderable {
 	/** Change whether the text object can be selected (picked) by the user. */
 	void setSelectable(bool selectable);
 
+protected:
+	/** Collect all materials used in the text object, optionally making them dynamic and/or transparent as well. */
+	void collectMaterials(std::list<Ogre::MaterialPtr> &materials, bool makeDynamic = false, bool makeTransparent = false);
+
 private:
 	FontHandle _font;
 

diff --git a/src/graphics/renderable.h b/src/graphics/renderable.h
@@ -30,6 +30,10 @@
 #ifndef GRAPHICS_RENDERABLE_H
 #define GRAPHICS_RENDERABLE_H
 
+#include <list>
+
+#include <OgrePrerequisites.h>
+
 #include "common/ustring.h"
 
 namespace Ogre {
@@ -101,6 +105,9 @@ class Renderable {
 
 	void destroyAnimation(const Common::UString &name);
 	void destroyAnimation(Ogre::Animation *anim);
+
+	/** Collect all materials used in the renderable, optionally making them dynamic and/or transparent as well. */
+	virtual void collectMaterials(std::list<Ogre::MaterialPtr> &materials, bool makeDynamic = false, bool makeTransparent = false) = 0;
 };
 
 } // End of namespace Graphics