SWBF2-Classic-Msh-Viewer/QtMeshViewer/Source/GeometryEngine.cpp

219 lines
6.2 KiB
C++
Raw Normal View History

#include "..\Header\GeometryEngine.h"
#include "..\Header\MshFile.h"
#include "..\Header\OglViewerWidget.h"
#include "..\Header\MainWindow.h"
#include <cmath>
2017-01-07 14:59:16 +00:00
#include <QRegExp>
2016-12-29 13:06:25 +00:00
/////////////////////////////////////////////////////////////////////////
// public constructor/destructor
GeometryEngine::GeometryEngine(QObject *parent)
: QObject(parent)
, m_indexBuf(QOpenGLBuffer::IndexBuffer)
{
2016-12-29 12:37:15 +00:00
initializeOpenGLFunctions();
m_defaultMaterial = FileInterface::getDefaultMaterial();
}
GeometryEngine::~GeometryEngine()
{
clearData();
delete m_defaultMaterial->texture;
delete m_defaultMaterial;
}
2016-12-29 13:06:25 +00:00
/////////////////////////////////////////////////////////////////////////
// private functions
2017-01-07 14:59:16 +00:00
void GeometryEngine::clearData()
{
if (m_arrayBuf.isCreated())
m_arrayBuf.destroy();
if (m_indexBuf.isCreated())
m_indexBuf.destroy();
if (m_materials != Q_NULLPTR)
{
for (auto it : *m_materials)
delete it.texture;
m_materials->clear();
delete m_materials;
}
2017-01-07 14:59:16 +00:00
m_drawList.clear();
}
/////////////////////////////////////////////////////////////////////////
// public slots
void GeometryEngine::loadFile(QString filePath)
{
// cleanup old stuff and recreate buffers
clearData();
m_arrayBuf.create();
m_indexBuf.create();
//reset view
emit requestResetView();
emit sendMessage("loading file..", 0);
try
{
QVector<Model*>* models;
QVector<VertexData> vertexData;
QVector<GLuint> indexData;
// open file and get the information
MshFile file(filePath, this);
models = file.getModels();
m_materials = file.getMaterials();
m_boundings = file.getBoundingBox();
2016-12-31 12:04:03 +00:00
// collect data
unsigned int indexOffset(0);
unsigned int vertexOffset(0);
2016-12-31 12:04:03 +00:00
for (auto& modelIterator : *models)
{
for (auto& segmentIterator : modelIterator->segmList)
{
// get draw information
DrawInformation new_info;
new_info.offset = indexOffset;
2016-12-31 12:04:03 +00:00
new_info.size = segmentIterator->indices.size();
new_info.textureIndex = segmentIterator->textureIndex;
new_info.modelMatrix = modelIterator->m4x4Translation;
2017-01-02 15:43:38 +00:00
new_info.modelMatrix.rotate(modelIterator->quadRotation);
2016-12-31 12:04:03 +00:00
// add offset to indices, no need to do it for the first one (maybe it's very big)
2017-01-07 14:59:16 +00:00
if (vertexOffset != 0)
for (auto& it : segmentIterator->indices)
it += vertexOffset;
2016-12-31 12:04:03 +00:00
// save data
vertexData += segmentIterator->vertices;
indexData += segmentIterator->indices;
2017-01-15 14:51:12 +00:00
if (segmentIterator->textureIndex < m_materials->size() && m_materials->at(segmentIterator->textureIndex).transparent)
m_drawList.push_back(new_info);
else
m_drawList.push_front(new_info);
2016-12-31 12:04:03 +00:00
// update offset
indexOffset += new_info.size;
vertexOffset += segmentIterator->vertices.size();
2016-12-31 12:04:03 +00:00
}
}
// 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();
emit sendMessage("done..", 0);
emit sendFileInfo(filePath.right(filePath.size() - filePath.lastIndexOf(QRegExp("/|\\\\")) - 1), m_materials, vertexData.size(), indexData.size() / 3);
}
catch (std::invalid_argument e)
{
2017-01-05 15:18:01 +00:00
clearData();
emit sendMessage(QString(e.what()), 2);
}
}
void GeometryEngine::drawGeometry(QOpenGLShaderProgram *program, bool wireframe)
2016-12-29 12:37:15 +00:00
{
if (!m_arrayBuf.isCreated() || !m_indexBuf.isCreated())
return;
// Setup
2016-12-29 12:37:15 +00:00
// Tell OpenGL which VBOs to use
2016-12-29 13:06:25 +00:00
m_arrayBuf.bind();
m_indexBuf.bind();
2016-12-29 12:37:15 +00:00
// 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
2016-12-29 12:37:15 +00:00
program->setUniformValue("texture", 0);
2016-12-29 12:37:15 +00:00
// Offset for position
quintptr offset = 0;
2016-12-29 12:37:15 +00:00
// 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));
2016-12-29 12:37:15 +00:00
// Offset for texture coordinate
offset += sizeof(QVector3D);
2016-12-29 12:37:15 +00:00
// 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));
2017-01-08 13:41:53 +00:00
//Offset for normal
offset += sizeof(QVector2D);
// Tell OpenGL programmable pipeline how to locate vertex normal data
int normLocation = program->attributeLocation("a_normal");
program->enableAttributeArray(normLocation);
program->setAttributeBuffer(normLocation, GL_FLOAT, offset, 3, sizeof(VertexData));
// Paint
for (auto& it : m_drawList)
{
2017-01-15 14:51:12 +00:00
bool tmp_transparent(false);
float shininess;
QVector3D specularColor;
2017-01-15 14:51:12 +00:00
// bind the correct texture
if (it.textureIndex < m_materials->size() && m_materials->at(it.textureIndex).texture != Q_NULLPTR)
2017-01-15 14:51:12 +00:00
{
m_materials->at(it.textureIndex).texture->bind();
2017-01-15 14:51:12 +00:00
tmp_transparent = m_materials->at(it.textureIndex).transparent;
shininess = m_materials->at(it.textureIndex).shininess;
specularColor = m_materials->at(it.textureIndex).specularColor.toVector3D();
2017-01-15 14:51:12 +00:00
}
2016-12-31 12:04:03 +00:00
else
2017-01-15 14:51:12 +00:00
{
m_defaultMaterial->texture->bind();
tmp_transparent = m_defaultMaterial->transparent;
2017-01-15 14:51:12 +00:00
}
// Set model matrix
program->setUniformValue("m_matrix", it.modelMatrix);
// Set normal matrix
program->setUniformValue("n_matrix", (normMatrix * it.modelMatrix).normalMatrix());
2017-01-15 14:51:12 +00:00
// decide if this is transparent
program->setUniformValue("b_transparent", tmp_transparent);
// set some material attributes
program->setUniformValue("materialShininess", shininess);
program->setUniformValue("materialSpecularColor", specularColor);
// Draw cube geometry using indices from VBO 1
2017-01-16 15:41:52 +00:00
if(wireframe)
glPolygonMode(GL_FRONT_AND_BACK, GL_LINE);
glDrawElements(GL_TRIANGLES, it.size, GL_UNSIGNED_INT, (void*)(it.offset * sizeof(GLuint)));
glPolygonMode(GL_FRONT_AND_BACK, GL_FILL);
}
}