OglViewerWidget.cpp 4.78 KB
Newer Older
1
#include "..\Header\OglViewerWidget.h"
Anakin's avatar
Anakin committed
2
#include "..\Header\MainWindow.h"
3
#include <QMouseEvent>
Anakin's avatar
Anakin committed
4 5
#include <QDropEvent>
#include <QMimeData>
6
#include <math.h>
Anakin's avatar
Anakin committed
7 8
#include <iostream>

Anakin's avatar
Anakin committed
9
#define DEFAULT_Z_DISTANCE -4.0
10

Anakin's avatar
Anakin committed
11 12 13 14

/////////////////////////////////////////////////////////////////////////
// public constructor/destructor

15 16
OglViewerWidget::OglViewerWidget(QWidget *parent) :
	QOpenGLWidget(parent),
Anakin's avatar
Anakin committed
17
	m_dataEngine(0)
18
{
19
	setFocus();
Anakin's avatar
Anakin committed
20
	m_translation.setZ(DEFAULT_Z_DISTANCE);
Anakin's avatar
Anakin committed
21
	setAcceptDrops(true);
Anakin's avatar
Anakin committed
22

23 24 25 26 27 28 29
}

OglViewerWidget::~OglViewerWidget()
{
	// Make sure the context is current when deleting the texture
	// and the buffers.
	makeCurrent();
Anakin's avatar
Anakin committed
30
	delete m_dataEngine;
31 32 33
	doneCurrent();
}

Anakin's avatar
Anakin committed
34 35 36 37

/////////////////////////////////////////////////////////////////////////
// protected functions

38 39 40
void OglViewerWidget::mousePressEvent(QMouseEvent *e)
{
	// Save mouse press position
41 42 43 44 45 46 47
	m_mouse.position = QVector2D(e->localPos());

	// Which button has been pressed?
	if (e->button() == Qt::LeftButton)
		m_mouse.left = true;
	else if (e->button() == Qt::RightButton)
		m_mouse.right = true;
48 49 50 51
}

void OglViewerWidget::mouseReleaseEvent(QMouseEvent *e)
{
52 53 54 55 56
	if (e->button() == Qt::LeftButton)
		m_mouse.left = false;
	else if (e->button() == Qt::RightButton)
		m_mouse.right = false;
}
57

58 59 60 61 62 63
void OglViewerWidget::mouseMoveEvent(QMouseEvent *e)
{
	if (m_mouse.left)
	{
		// get the difference between last press and now
		QVector2D diff = QVector2D(e->localPos()) - m_mouse.position;
64

65 66
		// update the new position
		m_mouse.position = QVector2D(e->localPos());
67

68 69
		// calculate the rotation axis and rotate
		m_rotation = QQuaternion::fromAxisAndAngle(QVector3D(diff.y(), diff.x(), 0.0).normalized(), diff.length() * 0.5) * m_rotation;
70

Anakin's avatar
Anakin committed
71 72 73 74 75 76 77 78 79 80 81 82 83 84
		// request an update
		update();
	}
	else if (m_mouse.right)
	{
		// get the difference between last press and now
		QVector2D diff = QVector2D(e->localPos()) - m_mouse.position;

		// update the new position
		m_mouse.position = QVector2D(e->localPos());

		// calculate the translation
		m_translation += {(float)(diff.x() * 0.01), (float)(diff.y() * -0.01), 0.0};

85 86 87
		// request an update
		update();
	}
88 89
}

Anakin's avatar
Anakin committed
90 91 92 93 94 95
void OglViewerWidget::wheelEvent(QWheelEvent *e)
{
	m_translation += {0.0, 0.0, (float)e->angleDelta().y() / 240};
	update();
}

Anakin's avatar
Anakin committed
96 97 98 99 100 101 102 103 104
void OglViewerWidget::dragEnterEvent(QDragEnterEvent *e)
{
	if (e->mimeData()->hasUrls())
		if(e->mimeData()->urls().size() == 1)
			if(e->mimeData()->urls().first().toLocalFile().endsWith(".msh"))
				e->acceptProposedAction();

}

Anakin's avatar
Anakin committed
105 106
void OglViewerWidget::dropEvent(QDropEvent * e)
{
Anakin's avatar
Anakin committed
107
	emit loadFile(e->mimeData()->urls().first().toLocalFile().toStdString().c_str());
Anakin's avatar
Anakin committed
108 109
}

110
void OglViewerWidget::keyPressEvent(QKeyEvent *e)
111
{
112
	if (e->key() == Qt::Key_Space)
Anakin's avatar
Anakin committed
113
	{
Anakin's avatar
Anakin committed
114
		resetView();
Anakin's avatar
Anakin committed
115
	}
Anakin's avatar
Anakin committed
116 117 118 119
	else if (e->key() == Qt::Key_Escape)
	{
		parentWidget()->close();
	}
120 121 122 123 124 125
}

void OglViewerWidget::initializeGL()
{
	initializeOpenGLFunctions();

Anakin's avatar
Anakin committed
126
	glClearColor(0.5000f, 0.8000f, 1.0000f, 0.0000f);
127 128 129 130 131 132 133 134 135

	initShaders();

	// Enable depth buffer
	glEnable(GL_DEPTH_TEST);

	// Enable back face culling
	glEnable(GL_CULL_FACE);

Anakin's avatar
Anakin committed
136
	m_dataEngine = new GeometryEngine;
Anakin's avatar
Anakin committed
137 138
	connect(m_dataEngine, &GeometryEngine::requestResetView, this, &OglViewerWidget::resetView);
	connect(parentWidget(), SIGNAL(loadFile(const char*)), m_dataEngine, SLOT(loadFile(const char*)));
Anakin's avatar
Anakin committed
139
	connect(this, SIGNAL(loadFile(const char*)), m_dataEngine, SLOT(loadFile(const char*)));
140 141 142 143 144 145 146 147 148

}

void OglViewerWidget::resizeGL(int w, int h)
{
	// Calculate aspect ratio
	qreal aspect = qreal(w) / qreal(h ? h : 1);

	// Set near plane to 3.0, far plane to 7.0, field of view 45 degrees
Anakin's avatar
Anakin committed
149
	const qreal zNear = 0.1, zFar = 100.0, fov = 45.0;
150 151

	// Reset projection
152
	m_projection.setToIdentity();
153 154

	// Set perspective projection
155
	m_projection.perspective(fov, aspect, zNear, zFar);
156 157 158 159 160 161 162 163
}

void OglViewerWidget::paintGL()
{
	// Clear color and depth buffer
	glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);

	// Calculate model view transformation
Anakin's avatar
Anakin committed
164
	QMatrix4x4 view;
Anakin's avatar
Anakin committed
165
	view.translate(m_translation);
Anakin's avatar
Anakin committed
166
	view.rotate(m_rotation);
167

Anakin's avatar
Anakin committed
168 169
	// Set view-projection matrix
	m_program.setUniformValue("vp_matrix", m_projection * view);
170 171

	// Draw cube geometry
Anakin's avatar
Anakin committed
172
	m_dataEngine->drawGeometry(&m_program);
173
}
Anakin's avatar
Anakin committed
174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199


/////////////////////////////////////////////////////////////////////////
// private functions

void OglViewerWidget::initShaders()
{
	// Compile vertex shader
	if (!m_program.addShaderFromSourceFile(QOpenGLShader::Vertex, ":/shaders/vshader.glsl"))
		close();

	// Compile fragment shader
	if (!m_program.addShaderFromSourceFile(QOpenGLShader::Fragment, ":/shaders/fshader.glsl"))
		close();

	// Link shader pipeline
	if (!m_program.link())
		close();

	// Bind shader pipeline for use
	if (!m_program.bind())
		close();
}


/////////////////////////////////////////////////////////////////////////
Anakin's avatar
Anakin committed
200
// private slots
Anakin's avatar
Anakin committed
201

Anakin's avatar
Anakin committed
202
void OglViewerWidget::resetView()
Anakin's avatar
Anakin committed
203
{
Anakin's avatar
Anakin committed
204 205 206 207
	m_rotation = QQuaternion();
	m_translation = { 0.0, 0.0, DEFAULT_Z_DISTANCE };
	update();
}