GeometryEngine.cpp 8.07 KB
Newer Older
1
#include "..\Header\GeometryEngine.h"
Anakin's avatar
Anakin committed
2
#include "..\Header\MshFile.h"
3
4
#include "..\Header\OglViewerWidget.h"
#include "..\Header\MainWindow.h"
5
#include "..\Header\OutputDevice.h"
Anakin's avatar
Anakin committed
6
#include <QRegExp>
7

Anakin's avatar
Anakin committed
8
9

/////////////////////////////////////////////////////////////////////////
10
// constructor/destructor
Anakin's avatar
Anakin committed
11

12
13
14
GeometryEngine::GeometryEngine(QObject *parent)
	: QObject(parent)
	, m_indexBuf(QOpenGLBuffer::IndexBuffer)
15
	, m_defaultMaterial(FileInterface::getDefaultMaterial())
16
{
Anakin's avatar
Anakin committed
17
	initializeOpenGLFunctions();
18
19
20
21
}

GeometryEngine::~GeometryEngine()
{
Anakin's avatar
Anakin committed
22
	clearData();
23
	delete m_defaultMaterial->texture0;
Anakin's avatar
Anakin committed
24
	delete m_defaultMaterial;
25
26
}

Anakin's avatar
Anakin committed
27
28

/////////////////////////////////////////////////////////////////////////
29
// functions
Anakin's avatar
Anakin committed
30

Anakin's avatar
Anakin committed
31
32
33
34
35
36
37
void GeometryEngine::clearData()
{
	if (m_arrayBuf.isCreated())
		m_arrayBuf.destroy();
	if (m_indexBuf.isCreated())
		m_indexBuf.destroy();

38
39
40
	if (m_materials != Q_NULLPTR)
	{
		for (auto it : *m_materials)
Anakin's avatar
Anakin committed
41
42
43
44
45
46
47
		{
			if (it.texture0 != Q_NULLPTR)
				delete it.texture0;

			if (it.texture1 != Q_NULLPTR)
				delete it.texture1;
		}
48
49
50
		m_materials->clear();
		delete m_materials;
	}
Anakin's avatar
Anakin committed
51
52
53
	m_drawList.clear();
}

54
void GeometryEngine::setupPipeline(QOpenGLShaderProgram *program)
Anakin's avatar
Anakin committed
55
56
57
{
	// Offset for position
	quintptr offset = 0;
58

Anakin's avatar
Anakin committed
59
60
61
62
	// Tell OpenGL programmable pipeline how to locate vertex position data
	int vertexLocation = program->attributeLocation("a_position");
	program->enableAttributeArray(vertexLocation);
	program->setAttributeBuffer(vertexLocation, GL_FLOAT, offset, 3, sizeof(VertexData));
63

Anakin's avatar
Anakin committed
64
65
	// Offset for texture coordinate
	offset += sizeof(QVector3D);
66

Anakin's avatar
Anakin committed
67
68
69
70
	// Tell OpenGL programmable pipeline how to locate vertex texture coordinate data
	int texcoordLocation = program->attributeLocation("a_texcoord");
	program->enableAttributeArray(texcoordLocation);
	program->setAttributeBuffer(texcoordLocation, GL_FLOAT, offset, 2, sizeof(VertexData));
71

72
	//Offset for vertexNormal
Anakin's avatar
Anakin committed
73
74
75
	offset += sizeof(QVector2D);

	// Tell OpenGL programmable pipeline how to locate vertex normal data
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
	int vertNormLocation = program->attributeLocation("a_normal");
	program->enableAttributeArray(vertNormLocation);
	program->setAttributeBuffer(vertNormLocation, GL_FLOAT, offset, 3, sizeof(VertexData));

	//Offset for polygonNormal
	offset += sizeof(QVector3D);

	// Tell OpenGL programmable pipeline how to locate polygon normal data
	int polyNormLocation = program->attributeLocation("a_polyNorm");
	program->enableAttributeArray(polyNormLocation);
	program->setAttributeBuffer(polyNormLocation, GL_FLOAT, offset, 3, sizeof(VertexData));

	//Offset for polygonTangent
	offset += sizeof(QVector3D);

	// Tell OpenGL programmable pipeline how to locate polygon tangent data
	int polyTanLocation = program->attributeLocation("a_polyTan");
	program->enableAttributeArray(polyTanLocation);
	program->setAttributeBuffer(polyTanLocation, GL_FLOAT, offset, 3, sizeof(VertexData));

	//Offset for polygonBitangent
	offset += sizeof(QVector3D);

	// Tell OpenGL programmable pipeline how to locate polygon bitangent data
	int polyBiTanLocation = program->attributeLocation("a_polyBiTan");
	program->enableAttributeArray(polyBiTanLocation);
	program->setAttributeBuffer(polyBiTanLocation, GL_FLOAT, offset, 3, sizeof(VertexData));

}

void GeometryEngine::drawGeometry(QOpenGLShaderProgram *program)
{
	if (!m_arrayBuf.isCreated() || !m_indexBuf.isCreated())
		return;

	// Setup
	// Tell OpenGL which VBOs to use
	m_arrayBuf.bind();
	m_indexBuf.bind();

	// Allways normalize by this
	QMatrix4x4 normMatrix;
	float maxExtent = std::max(std::max(m_boundings.extents[0], m_boundings.extents[1]), m_boundings.extents[2]);
	normMatrix.scale(1 / maxExtent);
	normMatrix.translate(-m_boundings.center[0], -m_boundings.center[1], -m_boundings.center[2]);
Anakin's avatar
Anakin committed
121
	program->setUniformValue("normalizeModel", normMatrix);
122

Anakin's avatar
Anakin committed
123
	// Allways use texture unit 0 and 1
Anakin's avatar
Anakin committed
124
125
	program->setUniformValue("tx0", 0);
	program->setUniformValue("tx1", 1);
Anakin's avatar
Anakin committed
126

127
128
129
	//setup the pipeline
	setupPipeline(program);
	
130
	// Paint
Anakin's avatar
Anakin committed
131
132
133

	for (auto& it : m_drawList)
	{
Anakin's avatar
Anakin committed
134
		bool tmp_transparent(false);
Anakin's avatar
Anakin committed
135
		bool tmp_specular(false);
Anakin's avatar
Anakin committed
136
		bool tmp_normalmap(false);
Anakin's avatar
Anakin committed
137
		bool tmp_glow(false);
Anakin's avatar
Anakin committed
138
		float shininess(0.0);
Anakin's avatar
Anakin committed
139
		QVector3D specularColor;
Anakin's avatar
Anakin committed
140

Anakin's avatar
Anakin committed
141
		// bind the correct texture
142
		if (it.textureIndex < (unsigned)m_materials->size() && m_materials->at(it.textureIndex).texture0 != Q_NULLPTR)
Anakin's avatar
Anakin committed
143
		{
Anakin's avatar
Anakin committed
144
			m_materials->at(it.textureIndex).texture0->bind(0);
Anakin's avatar
Anakin committed
145
			tmp_transparent = m_materials->at(it.textureIndex).transparent;
Anakin's avatar
Anakin committed
146
			tmp_specular = m_materials->at(it.textureIndex).flags[7];
Anakin's avatar
Anakin committed
147
			shininess = m_materials->at(it.textureIndex).shininess;
Anakin's avatar
Anakin committed
148
			specularColor = m_materials->at(it.textureIndex).specularColor.toVector3D();
Anakin's avatar
Anakin committed
149
150
151
152
153
154
155
156
157

			if (m_materials->at(it.textureIndex).rendertype == 27 || m_materials->at(it.textureIndex).rendertype == 28)
			{
				if (m_materials->at(it.textureIndex).texture1 != Q_NULLPTR)
				{
					tmp_normalmap = true;
					m_materials->at(it.textureIndex).texture1->bind(1);
				}
			}
Anakin's avatar
Anakin committed
158
159
160

			if (m_materials->at(it.textureIndex).flags[0] || m_materials->at(it.textureIndex).flags[1] || m_materials->at(it.textureIndex).rendertype == 1)
				tmp_glow = true;
Anakin's avatar
Anakin committed
161
		}
Anakin's avatar
Anakin committed
162
		else
Anakin's avatar
Anakin committed
163
		{
Anakin's avatar
Anakin committed
164
			m_defaultMaterial->texture0->bind(0);
Anakin's avatar
Anakin committed
165
			tmp_transparent = m_defaultMaterial->transparent;
Anakin's avatar
Anakin committed
166
		}
Anakin's avatar
Anakin committed
167

Anakin's avatar
Anakin committed
168
		// Set model matrix
Anakin's avatar
Anakin committed
169
		program->setUniformValue("modelMatrix", it.modelMatrix);
Anakin's avatar
Anakin committed
170

171
		// Set normal matrix
Anakin's avatar
Anakin committed
172
		program->setUniformValue("normalMatrix", (normMatrix * it.modelMatrix).normalMatrix());
Anakin's avatar
Anakin committed
173

Anakin's avatar
Anakin committed
174
		// set some material attributes
Anakin's avatar
Anakin committed
175
176
177
178
179
180
		program->setUniformValue("material.shininess", shininess);
		program->setUniformValue("material.specularColor", specularColor);
		program->setUniformValue("material.isTransparent", tmp_transparent);
		program->setUniformValue("material.hasSpecularmap", tmp_specular);
		program->setUniformValue("material.hasNormalmap", tmp_normalmap);
		program->setUniformValue("material.isGlow", tmp_glow);
Anakin's avatar
Anakin committed
181

Anakin's avatar
Anakin committed
182
		// Draw cube geometry using indices from VBO 1
Anakin's avatar
Anakin committed
183
		glDrawElements(GL_TRIANGLES, it.size, GL_UNSIGNED_INT, (void*)(it.offset * sizeof(GLuint)));
184

Anakin's avatar
Anakin committed
185
	}
186
}
187

188
189
190
191
192
193
194
195
196
void GeometryEngine::loadFile(QString filePath)
{
	// cleanup old stuff and recreate buffers
	clearData();
	m_arrayBuf.create();
	m_indexBuf.create();

	//reset view
	emit requestResetView();
197
	OutputDevice::getInstance()->print("loading file..", 0);
198
199
200
201
202
203
204
205

	try
	{
		QVector<Model*>* models;
		QVector<VertexData> vertexData;
		QVector<GLuint> indexData;

		// open file and get the information
206
		MshFile file(filePath);
207
208
209
210
211
212
213
214
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

		models = file.getModels();
		m_materials = file.getMaterials();
		m_boundings = file.getBoundingBox();

		// collect data
		unsigned int indexOffset(0);
		unsigned int vertexOffset(0);
		for (auto& modelIterator : *models)
		{
			for (auto& segmentIterator : modelIterator->segmList)
			{
				// get draw information
				DrawInformation new_info;
				new_info.offset = indexOffset;
				new_info.size = segmentIterator->indices.size();
				new_info.textureIndex = segmentIterator->textureIndex;
				new_info.modelMatrix = modelIterator->m4x4Translation;
				new_info.modelMatrix.rotate(modelIterator->quadRotation);

				// add offset to indices, no need to do it for the first one (maybe it's very big)
				if (vertexOffset != 0)
					for (auto& it : segmentIterator->indices)
						it += vertexOffset;

				// save data
				vertexData += segmentIterator->vertices;
				indexData += segmentIterator->indices;

				if (segmentIterator->textureIndex < (unsigned) m_materials->size() && m_materials->at(segmentIterator->textureIndex).transparent)
					m_drawList.push_back(new_info);
				else
					m_drawList.push_front(new_info);

				// update offset
				indexOffset += new_info.size;
				vertexOffset += segmentIterator->vertices.size();
			}
		}

		// Transfer vertex data to VBO 0
		m_arrayBuf.bind();
		m_arrayBuf.allocate(vertexData.data(), vertexData.size() * sizeof(VertexData));

		// Transfer index data to VBO 1
		m_indexBuf.bind();
		m_indexBuf.allocate(indexData.data(), indexData.size() * sizeof(GLuint));

		emit requestUpdate();
256
		OutputDevice::getInstance()->print("done..", 0);
Anakin's avatar
Anakin committed
257
		OutputDevice::getInstance()->setFileInfo(filePath.right(filePath.size() - filePath.lastIndexOf(QRegExp("/|\\\\")) - 1), m_materials, vertexData.size(), indexData.size() / 3);
258
259
260
261
	}
	catch (std::invalid_argument e)
	{
		clearData();
262
		OutputDevice::getInstance()->print(QString(e.what()), 2);
263
264
265
	}
}