GeometryEngine.cpp 7.47 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
#include "..\Header\Profiler.h"
Anakin's avatar
Anakin committed
9
10

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

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

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

Anakin's avatar
Anakin committed
28
29

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

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

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

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

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

Anakin's avatar
Anakin committed
60
61
62
63
	// 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));
64

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

Anakin's avatar
Anakin committed
68
69
70
71
	// 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));
72

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

	// Tell OpenGL programmable pipeline how to locate vertex normal data
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
121
122
123
124
125
	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]);
	program->setUniformValue("norm_matrix", normMatrix);

	// Allways use texture unit 0
	program->setUniformValue("texture", 0);
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
		float shininess(0.0);
Anakin's avatar
Anakin committed
137
		QVector3D specularColor;
Anakin's avatar
Anakin committed
138

Anakin's avatar
Anakin committed
139
		// bind the correct texture
140
		if (it.textureIndex < (unsigned)m_materials->size() && m_materials->at(it.textureIndex).texture0 != Q_NULLPTR)
Anakin's avatar
Anakin committed
141
		{
142
			m_materials->at(it.textureIndex).texture0->bind();
Anakin's avatar
Anakin committed
143
			tmp_transparent = m_materials->at(it.textureIndex).transparent;
Anakin's avatar
Anakin committed
144
			tmp_specular = m_materials->at(it.textureIndex).flags[7];
Anakin's avatar
Anakin committed
145
			shininess = m_materials->at(it.textureIndex).shininess;
Anakin's avatar
Anakin committed
146
			specularColor = m_materials->at(it.textureIndex).specularColor.toVector3D();
Anakin's avatar
Anakin committed
147
		}
Anakin's avatar
Anakin committed
148
		else
Anakin's avatar
Anakin committed
149
		{
150
			m_defaultMaterial->texture0->bind();
Anakin's avatar
Anakin committed
151
			tmp_transparent = m_defaultMaterial->transparent;
Anakin's avatar
Anakin committed
152
		}
Anakin's avatar
Anakin committed
153
154
155
		// Set model matrix
		program->setUniformValue("m_matrix", it.modelMatrix);

156
157
158
		// Set normal matrix
		program->setUniformValue("n_matrix", (normMatrix * it.modelMatrix).normalMatrix());

Anakin's avatar
Anakin committed
159
		// set some more values
Anakin's avatar
Anakin committed
160
		program->setUniformValue("b_transparent", tmp_transparent);
Anakin's avatar
Anakin committed
161
		program->setUniformValue("b_specular", tmp_specular);
Anakin's avatar
Anakin committed
162

Anakin's avatar
Anakin committed
163
164
165
166
		// set some material attributes
		program->setUniformValue("materialShininess", shininess);
		program->setUniformValue("materialSpecularColor", specularColor);

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

Anakin's avatar
Anakin committed
170
	}
171
}
172

173
174
void GeometryEngine::loadFile(QString filePath)
{
Anakin's avatar
Anakin committed
175
	TIC("Start");
176
177
178
179
180
181
182
	// cleanup old stuff and recreate buffers
	clearData();
	m_arrayBuf.create();
	m_indexBuf.create();

	//reset view
	emit requestResetView();
183
	OutputDevice::getInstance()->print("loading file..", 0);
184
185
186
187
188
189
190
191

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

		// open file and get the information
192
		MshFile file(filePath);
193
194
195
196
197
198
199
200
201
202
203
204
205
206
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

		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();
242
		OutputDevice::getInstance()->print("done..", 0);
Anakin's avatar
Anakin committed
243
		OutputDevice::getInstance()->setFileInfo(filePath.right(filePath.size() - filePath.lastIndexOf(QRegExp("/|\\\\")) - 1), m_materials, vertexData.size(), indexData.size() / 3);
244
245
246
247
	}
	catch (std::invalid_argument e)
	{
		clearData();
248
		OutputDevice::getInstance()->print(QString(e.what()), 2);
249
	}
Anakin's avatar
Anakin committed
250
251

	TOC("End");
252
253
}