Modified RenderMesh so that it can renders multiple meshes, each with…

… a plain color. A new object has been created to keep the old functionality (one mesh, each point its own color).
cguebert · Dec 9, 2015 · bd763ba · bd763ba
1 parent 527513a
commit bd763ba
Showing 1 changed file with 76 additions and 4 deletions.
diff --git a/modules/Render/RenderMesh.cpp b/modules/Render/RenderMesh.cpp
@@ -23,6 +23,78 @@ class RenderMesh : public Renderer
 	PANDA_CLASS(RenderMesh, Renderer)
 
 	RenderMesh(PandaDocument* parent)
+		: Renderer(parent)
+		, meshes(initData("mesh", "Mesh to render"))
+		, color(initData("color", "Color of the points"))
+		, shader(initData("shader", "Shaders used during the rendering"))
+	{
+		addInput(meshes);
+		addInput(color);
+		addInput(shader);
+
+		color.getAccessor().push_back(Color::black());
+
+		shader.setWidgetData("Vertex;Fragment");
+		auto shaderAcc = shader.getAccessor();
+		shaderAcc->setSourceFromFile(QOpenGLShader::Vertex, "shaders/PT_uniColor_noTex.v.glsl");
+		shaderAcc->setSourceFromFile(QOpenGLShader::Fragment, "shaders/PT_uniColor_noTex.f.glsl");
+	}
+
+	void render()
+	{
+		const QVector<Mesh>& inMeshes = meshes.getValue();
+		QVector<Color> listColor = color.getValue();
+
+		int nbMeshes = inMeshes.size();
+		int nbColor = listColor.size();
+
+		if(nbMeshes && nbColor)
+		{
+			if (nbColor < nbMeshes)
+				nbColor = 1;
+
+			if (!shader.getValue().apply(shaderProgram))
+				return;
+
+			int colorLocation = shaderProgram.uniformLocation("color");
+			shaderProgram.setUniformValue("MVP", getMVPMatrix());
+			shaderProgram.enableAttributeArray("vertex");
+
+			for (int i = 0; i < nbMeshes; ++i)
+			{
+				const auto& mesh = inMeshes[i];
+				int nbTri = mesh.nbTriangles();
+
+				shaderProgram.setAttributeArray("vertex", mesh.getPoints().front().data(), 2);
+
+				auto color = listColor[i % nbColor];
+				shaderProgram.setUniformValue(colorLocation, color.r, color.g, color.b, color.a);
+
+				glDrawElements(GL_TRIANGLES, nbTri * 3, GL_UNSIGNED_INT, mesh.getTriangles().data());
+			}
+
+			shaderProgram.disableAttributeArray("vertex");
+		}
+	}
+
+protected:
+	Data< QVector<Mesh> > meshes;
+	Data< QVector<Color> > color;
+	Data< Shader > shader;
+
+	QOpenGLShaderProgram shaderProgram;
+};
+
+int RenderMeshClass = RegisterObject<RenderMesh>("Render/Filled/Plain mesh").setDescription("Draw a mesh");
+
+//****************************************************************************//
+
+class RenderMeshColoredPoints : public Renderer
+{
+public:
+	PANDA_CLASS(RenderMeshColoredPoints, Renderer)
+
+		RenderMeshColoredPoints(PandaDocument* parent)
 		: Renderer(parent)
 		, mesh(initData("mesh", "Mesh to render"))
 		, color(initData("color", "Color of the points"))
@@ -49,12 +121,12 @@ class RenderMesh : public Renderer
 		int nbTri = inMesh.nbTriangles();
 		int nbColor = listColor.size();
 
-		if(nbTri && nbColor)
+		if (nbTri && nbColor)
 		{
-			if(nbColor < nbPts)
+			if (nbColor < nbPts)
 				listColor.fill(listColor[0], nbPts);
 
-			if(!shader.getValue().apply(shaderProgram))
+			if (!shader.getValue().apply(shaderProgram))
 				return;
 
 			shaderProgram.setUniformValue("MVP", getMVPMatrix());
@@ -80,7 +152,7 @@ class RenderMesh : public Renderer
 	QOpenGLShaderProgram shaderProgram;
 };
 
-int RenderMeshClass = RegisterObject<RenderMesh>("Render/Filled/Mesh").setDescription("Draw a mesh");
+int RenderMeshColoredPointsClass = RegisterObject<RenderMeshColoredPoints>("Render/Filled/Mesh").setDescription("Draw a mesh, each point having its own color");
 
 //****************************************************************************//