pulsaranimationwidget.cpp 8.55 KB
Newer Older
Oliver Bock's avatar
Oliver Bock committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
/******************************************************************************
 *   Copyright (C) 2008 by Oliver Bock                                        *
 *   oliver.bock[AT]aei.mpg.de                                                *
 *                                                                            *
 *   This file is part of PulsatingScience.                                   *
 *                                                                            *
 *   PulsatingScience is free software: you can redistribute it and/or modify *
 *   it under the terms of the GNU General Public License as published        *
 *   by the Free Software Foundation, version 3 of the License.               *
 *                                                                            *
 *   PulsatingScience is distributed in the hope that it will be useful,      *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of           *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the             *
 *   GNU General Public License for more details.                             *
 *                                                                            *
 *   You should have received a copy of the GNU General Public License        *
 *   along with PulsatingScience. If not, see <http://www.gnu.org/licenses/>. *
 *                                                                            *
 ******************************************************************************/

#include "pulsaranimationwidget.h"

Oliver Bock's avatar
Oliver Bock committed
23 24 25
PulsarAnimationWidget::PulsarAnimationWidget(QWidget *parent) :
	QGLWidget(parent),
	m_frameTimer(),
26 27
	m_pulsarSpinTimer(),
	m_orbitTimer()
Oliver Bock's avatar
Oliver Bock committed
28
{
Oliver Bock's avatar
Oliver Bock committed
29 30
    connect(&m_frameTimer, SIGNAL(timeout()), this, SLOT(updateFrame()));
    connect(&m_pulsarSpinTimer, SIGNAL(timeout()), this, SLOT(updatePulsarSpin()));
31
    connect(&m_orbitTimer, SIGNAL(timeout()), this, SLOT(updateOrbit()));
Oliver Bock's avatar
Oliver Bock committed
32

33 34 35
	m_quadricPulsar = NULL;
	m_quadricPulsarCone1 = NULL;
	m_quadricPulsarCone2 = NULL;
Oliver Bock's avatar
Oliver Bock committed
36 37

	m_framesPerSecond = 25;
38

Oliver Bock's avatar
Oliver Bock committed
39
	setPulsarSpinFrequency(0.5f);
40 41
	m_pulsarRotationAngle = 0.0f;

Oliver Bock's avatar
Oliver Bock committed
42
	setOrbitFrequency(0.2f);
43 44
	setOrbitRadius(4.0f);
	m_orbitRotationAngle = 0.0f;
45 46 47 48 49 50 51 52 53

	m_mouseLastX = 0;
	m_mouseLastY = 0;
	m_mouseAngleH = 90;
	m_mouseAngleV = 30;
	m_cameraZoom = 10.0f;
	m_cameraZoomLBound = 2.0f;

	updateCameraPosition(m_mouseAngleH, m_mouseAngleV, m_cameraZoom);
Oliver Bock's avatar
Oliver Bock committed
54 55 56 57
}

PulsarAnimationWidget::~PulsarAnimationWidget()
{
Oliver Bock's avatar
Oliver Bock committed
58 59 60
	if(m_quadricPulsar) gluDeleteQuadric(m_quadricPulsar);
	if(m_quadricPulsarCone1) gluDeleteQuadric(m_quadricPulsarCone1);
	if(m_quadricPulsarCone2) gluDeleteQuadric(m_quadricPulsarCone2);
Oliver Bock's avatar
Oliver Bock committed
61 62 63 64
}

void PulsarAnimationWidget::initializeGL()
{
Oliver Bock's avatar
Oliver Bock committed
65 66 67 68 69 70 71
	glShadeModel(GL_SMOOTH);
	glClearColor(0.0, 0.0, 0.0, 0.0);
	glClearDepth(1.0f);
	glEnable(GL_DEPTH_TEST);
	glDepthFunc(GL_LEQUAL);
	glHint(GL_PERSPECTIVE_CORRECTION_HINT, GL_NICEST);

72
	GLfloat LightAmbient[] = { 0.3f, 0.3f, 0.3f, 1.0f };
Oliver Bock's avatar
Oliver Bock committed
73 74 75 76 77 78 79 80 81
	GLfloat LightDiffuse[] = { 1.0f, 1.0f, 1.0f, 1.0f };
	GLfloat LightSpecular[] = { 1.0f, 1.0f, 1.0f, 1.0f };
	GLfloat LightPosition[] = { 0.0f, 0.0f, 3.0f, 1.0f };
	GLfloat spot_direction[] = { 0.0, 0.0, -1.0 };

	glLightfv(GL_LIGHT0, GL_AMBIENT, LightAmbient);
	glLightfv(GL_LIGHT0, GL_DIFFUSE, LightDiffuse);
	glLightfv(GL_LIGHT0, GL_SPECULAR, LightSpecular);
	glLightfv(GL_LIGHT0, GL_POSITION, LightPosition);
82
	glLightf(GL_LIGHT0, GL_SPOT_CUTOFF, 50.0);
Oliver Bock's avatar
Oliver Bock committed
83
	glLightfv(GL_LIGHT0, GL_SPOT_DIRECTION, spot_direction);
84
	glLightf(GL_LIGHT0, GL_SPOT_EXPONENT, 10.0);
Oliver Bock's avatar
Oliver Bock committed
85 86 87 88

	glLightModeli(GL_LIGHT_MODEL_LOCAL_VIEWER, GL_TRUE);
	glEnable(GL_LIGHT0);
	glEnable(GL_LIGHTING);
Oliver Bock's avatar
Oliver Bock committed
89

Oliver Bock's avatar
Oliver Bock committed
90 91 92 93 94 95
	m_quadricPulsar = gluNewQuadric();
    m_quadricPulsarCone1 = gluNewQuadric();
    m_quadricPulsarCone2 = gluNewQuadric();
    gluQuadricNormals(m_quadricPulsar, GLU_SMOOTH);
    gluQuadricNormals(m_quadricPulsarCone1, GLU_SMOOTH);
    gluQuadricNormals(m_quadricPulsarCone2, GLU_SMOOTH);
Oliver Bock's avatar
Oliver Bock committed
96 97 98 99
}

void PulsarAnimationWidget::resizeGL(int w, int h)
{
Oliver Bock's avatar
Oliver Bock committed
100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117
	glViewport(0, 0, w, h);

	glMatrixMode(GL_PROJECTION);
	glLoadIdentity();

	gluPerspective(45.0f, (GLfloat)w / (GLfloat)h, 0.1f, 100.0f);

	glMatrixMode(GL_MODELVIEW);
	glLoadIdentity();
}

void PulsarAnimationWidget::paintGL()
{
	glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);

	glMatrixMode(GL_MODELVIEW);
	glLoadIdentity();

118 119
	gluLookAt(m_cameraPosX, m_cameraPosY, m_cameraPosZ,
			  0.0, 0.0, 0.0,
120
			  0.0, 1.0, 0.0);
Oliver Bock's avatar
Oliver Bock committed
121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136

	// TODO: should be located elsewhere
	static GLfloat no_mat[] = { 0.0, 0.0, 0.0, 1.0 };
	static GLfloat mat_ambient[] = { 0.7, 0.7, 0.7, 1.0 };
	static GLfloat mat_ambient_color[] = { 0.8, 0.8, 0.2, 1.0 };
	static GLfloat mat_diffuse[] = { 0.5, 0.5, 0.5, 1.0 };
	static GLfloat mat_specular[] = { 1.0, 1.0, 1.0, 1.0 };
	static GLfloat no_shininess[] = { 0.0 };
	static GLfloat low_shininess[] = { 5.0 };
	static GLfloat high_shininess[] = { 100.0 };
	static GLfloat mat_emission[] = { 0.3, 0.2, 0.2, 0.0 };
	glMaterialfv(GL_FRONT, GL_AMBIENT, no_mat);
	glMaterialfv(GL_FRONT, GL_DIFFUSE, mat_diffuse);
	glMaterialfv(GL_FRONT, GL_SPECULAR, mat_specular);
	glMaterialfv(GL_FRONT, GL_SHININESS, low_shininess);
	glMaterialfv(GL_FRONT, GL_EMISSION, no_mat);
Oliver Bock's avatar
Oliver Bock committed
137

138 139
	glRotatef(m_orbitRotationAngle, 0.0f, 1.0f, 0.0f);

140
	glPushMatrix();
141
	glTranslatef(0.0f, 0.0f, m_orbitRadiusNeg);
142 143 144 145 146
	gluSphere(m_quadricPulsar, 1.0f, 32, 32);
	glPopMatrix();

	glTranslatef(0.0f, 0.0f, m_orbitRadius);

Oliver Bock's avatar
Oliver Bock committed
147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162
	glPushMatrix();
	glRotatef(m_pulsarRotationAngle, 0.0f, 1.0f, 0.0f);
	gluSphere(m_quadricPulsar, 1.0f, 32, 32);
	glPopMatrix();

	glPushMatrix();
	glRotatef(m_pulsarRotationAngle-90, 0.0f, 1.0f, 0.0f);
	glTranslatef(0.0f, 0.0f, -4.0f);
	gluCylinder(m_quadricPulsarCone1, 0.5f, 0.0f, 3.0f, 32, 32);
	glPopMatrix();

	glPushMatrix();
	glRotatef(m_pulsarRotationAngle+90, 0.0f, 1.0f, 0.0f);
	glTranslatef(0.0f, 0.0f, -4.0f);
	gluCylinder(m_quadricPulsarCone2, 0.5f, 0.0f, 3.0f, 32, 32);
	glPopMatrix();
Oliver Bock's avatar
Oliver Bock committed
163 164
}

Oliver Bock's avatar
Oliver Bock committed
165 166 167 168
void PulsarAnimationWidget::runAnimation()
{
	m_frameTimer.start(1000.0 / m_framesPerSecond);
	m_pulsarSpinTimer.start(m_pulsarSpinFrequency);
169
	m_orbitTimer.start(m_orbitFrequency);
Oliver Bock's avatar
Oliver Bock committed
170
}
Oliver Bock's avatar
Oliver Bock committed
171

Oliver Bock's avatar
Oliver Bock committed
172 173 174 175
void PulsarAnimationWidget::pauseAnimation()
{
	m_frameTimer.stop();
	m_pulsarSpinTimer.stop();
176
	m_orbitTimer.stop();
Oliver Bock's avatar
Oliver Bock committed
177 178 179 180 181 182
}

void PulsarAnimationWidget::stopAnimation()
{
	m_frameTimer.stop();
	m_pulsarSpinTimer.stop();
183
	m_orbitTimer.stop();
Oliver Bock's avatar
Oliver Bock committed
184 185

	m_pulsarRotationAngle = 0.0;
186
	m_orbitRotationAngle = 0.0;
Oliver Bock's avatar
Oliver Bock committed
187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204
	updateGL();
}

void PulsarAnimationWidget::updateFrame()
{
	updateGL();
}

void PulsarAnimationWidget::updatePulsarSpin()
{
	if(m_pulsarRotationAngle <= 360.0) {
		m_pulsarRotationAngle++;
	}
	else {
		m_pulsarRotationAngle = 0.0;
	}
}

205 206 207 208 209 210 211 212 213 214
void PulsarAnimationWidget::updateOrbit()
{
	if(m_orbitRotationAngle <= 360.0) {
		m_orbitRotationAngle++;
	}
	else {
		m_orbitRotationAngle = 0.0;
	}
}

215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257
void PulsarAnimationWidget::mouseMoveEvent(QMouseEvent *event)
{
	Qt::MouseButtons buttons = event->buttons();
	if((buttons & Qt::LeftButton) == Qt::LeftButton) {
		if(m_mouseLastX != 0) {
			m_mouseAngleH -= (m_mouseLastX - event->x()) / 2;
			m_mouseAngleH = m_mouseAngleH < 360? m_mouseAngleH : 0;

		}
		if(m_mouseLastY != 0) {
			m_mouseAngleV -= (m_mouseLastY - event->y()) / 2;
			m_mouseAngleV = m_mouseAngleV < 90? m_mouseAngleV : 90;
			m_mouseAngleV = m_mouseAngleV > -90? m_mouseAngleV : -90;
		}

		m_mouseLastX = event->x();
		m_mouseLastY = event->y();
	}
	else if((buttons & Qt::RightButton) == Qt::RightButton) {
		if(m_mouseLastY != 0) {
			m_cameraZoom -= (m_mouseLastY - event->y()) / 2;
			m_cameraZoom = m_cameraZoom >= m_cameraZoomLBound? m_cameraZoom : m_cameraZoomLBound;
			updateCameraPosition(m_mouseAngleH, m_mouseAngleV, m_cameraZoom);
		}

		m_mouseLastY = event->y();
	}

	updateCameraPosition(m_mouseAngleH, m_mouseAngleV, m_cameraZoom);
}

void PulsarAnimationWidget::mouseReleaseEvent(QMouseEvent *event)
{
	m_mouseLastX = 0;
	m_mouseLastY = 0;
}

void PulsarAnimationWidget::updateCameraPosition(const int angleH, const int angleV, const float zoom)
{
	static const float deg2rad = PI/180.0f;

	m_cameraPosX = sin(angleH * deg2rad) * zoom;
	m_cameraPosY = sin(angleV * deg2rad) * zoom;
Oliver Bock's avatar
Oliver Bock committed
258
	m_cameraPosZ = cos(angleH * deg2rad) * cos(fabs(angleV * deg2rad)) * zoom;
259 260 261 262

	updateGL();
}

Oliver Bock's avatar
Oliver Bock committed
263 264 265 266 267 268 269
void PulsarAnimationWidget::setFramePerSecond(const unsigned int fps)
{
	m_framesPerSecond = fps;
}

void PulsarAnimationWidget::setPulsarSpinFrequency(const float frequency)
{
Oliver Bock's avatar
Oliver Bock committed
270
	m_pulsarSpinFrequency = (unsigned int) (1000 / (frequency * 360.0f) + 0.5f);
Oliver Bock's avatar
Oliver Bock committed
271
	m_pulsarSpinTimer.setInterval(m_pulsarSpinFrequency);
Oliver Bock's avatar
Oliver Bock committed
272
}
273 274 275

void PulsarAnimationWidget::setOrbitRadius(const float radius)
{
276 277
	m_orbitRadius = radius;
	m_orbitRadiusNeg = -1.0 * radius;
278 279 280 281
}

void PulsarAnimationWidget::setOrbitFrequency(const float frequency)
{
Oliver Bock's avatar
Oliver Bock committed
282
	m_orbitFrequency = (unsigned int) (1000 / (frequency * 360.0f) + 0.5f);
283 284
	m_orbitTimer.setInterval(m_orbitFrequency);
}