1 files changed, 16 insertions, 5 deletions
diff --git a/src/gui/ParticleRenderer.cc b/src/gui/ParticleRenderer.cc
index 3c8fe86..05f1b63 100644
--- a/src/gui/ParticleRenderer.cc
+++ b/src/gui/ParticleRenderer.cc
@@ -9,9 +9,11 @@
 #include "ParticleRenderer.h"
 #include "util.h"
 
+using namespace vhc::util;
+
 namespace vhc {
 
-ParticleRenderer::ParticleRenderer() {
+ParticleRenderer::ParticleRenderer(): _drawSpheres(false) {
 
 }
 
@@ -22,11 +24,20 @@ ParticleRenderer::~ParticleRenderer() {
 void ParticleRenderer::render(const Particle& particle) const {
 	glPushMatrix();
 	glTranslated(particle.getPosition().getX(), particle.getPosition().getY(), particle.getPosition().getZ());
-	glBegin(GL_POINTS);
-	//glVertex3d(0, 0, 0);
-	util::sphere(0.01);
-	glEnd();
+
+	if (_drawSpheres) {
+		glBegin(GL_POINTS);
+		glVertex3d(0, 0, 0);
+		glEnd();
+	}
+	else {
+		util::sphere(0.01);
+	}
 	glPopMatrix();
 }
 
+void ParticleRenderer::drawSpheres(bool value) {
+	_drawSpheres = value;
+}
+
 }