Qt+opengl加载各种类型的3D模型，.glb .obj 等；Qt+opengl加载各种类型的3D模型，.glb .obj

#include "model.h" #include <QDir> void Model::loadModel(string path) { Assimp::Importer import; const aiScene *scene = import.ReadFile(path, aiProcess_Triangulate | aiProcess_FlipUVs); if(!scene || scene->mFlags & AI_SCENE_FLAGS_INCOMPLETE || !scene->mRootNode) { qDebug() << "ERROR::ASSIMP::" << import.GetErrorString(); return; } directory = path.substr(0, path.find_last_of('/')); modelDirectory = directory + "/model"; QString modelStr = QString::fromStdString(modelDirectory); QDir dir; if (!dir.exists(modelStr)) dir.mkdir(modelStr); for (int i = 0; i < scene->mNumTextures; i++) { aiTexture* texture = scene->mTextures[i]; char fileName[100]; sprintf(fileName, "%s/%s.%s", modelDirectory.c_str(), texture->mFilename.C_Str(), texture->achFormatHint); QFile file(fileName); if (file.exists()) { m_mapPath.insert(i, QString(fileName)); continue; } if (!file.open(QIODevice::WriteOnly)) break; unsigned char* buffer = new unsigned char[texture->mWidth * 4]; memset(buffer, 0, texture->mWidth * 4); for (int x = 0; x < texture->mWidth; ++x) { //拷贝RGBA数据到缓冲区 int index = x * 4; buffer[index] = texture->pcData[x].b; // Blue buffer[index + 1] = texture->pcData[x].g; // Green buffer[index + 2] = texture->pcData[x].r; // Red buffer[index + 3] = texture->pcData[x].a; // Alpha } file.write((char*)buffer, texture->mWidth * 4); file.close(); m_mapPath.insert(i, QString(fileName)); delete[]buffer; } processNode(scene->mRootNode, scene); } void Model::processNode(aiNode *node, const aiScene *scene) { // process all the node's meshes (if any) for(unsigned int i = 0; i < node->mNumMeshes; i++) { aiMesh *mesh = scene->mMeshes[node->mMeshes[i]]; meshes.push_back(processMesh(mesh, scene)); } // then do the same for each of its children for(unsigned int i = 0; i < node->mNumChildren; i++) { processNode(node->mChildren[i], scene); } } Mesh Model::processMesh(aiMesh *mesh, const aiScene *scene) { vector<Vertex> vertices; vector<unsigned int> indices; vector<Texture> textures; for(unsigned int i = 0; i < mesh->mNumVertices; i++) { if(m_maxX<mesh->mVertices[i].x) m_maxX=mesh->mVertices[i].x; if(m_maxY<mesh->mVertices[i].y) m_maxY=mesh->mVertices[i].y; if(m_minX>mesh->mVertices[i].x) m_minX=mesh->mVertices[i].x; if(m_minY>mesh->mVertices[i].y) m_minY=mesh->mVertices[i].y; Vertex vertex; // 处理顶点位置、法线和纹理坐标 QVector3D vector; vector.setX(mesh->mVertices[i].x); vector.setY(mesh->mVertices[i].y); vector.setZ(mesh->mVertices[i].z); vertex.Position = vector; vector.setX(mesh->mNormals[i].x); vector.setY(mesh->mNormals[i].y); vector.setZ(mesh->mNormals[i].z); vertex.Normal = vector; if(mesh->mTextureCoords[0]) // 有纹理坐标？ { QVector2D vec; vec.setX(mesh->mTextureCoords[0][i].x); vec.setY(mesh->mTextureCoords[0][i].y); vertex.TexCoords = vec; } else vertex.TexCoords = QVector2D(0.0f, 0.0f); vertices.push_back(vertex); } // 处理索引 for(unsigned int i = 0; i < mesh->mNumFaces; i++) { aiFace face = mesh->mFaces[i]; for(unsigned int j = 0; j < face.mNumIndices; j++) indices.push_back(face.mIndices[j]); } // 处理材质 if(mesh->mMaterialIndex >= 0) { aiMaterial *material = scene->mMaterials[mesh->mMaterialIndex]; vector<Texture> diffuseMaps = loadMaterialTextures(material, aiTextureType_DIFFUSE, "texture_diffuse"); textures.insert(textures.end(), diffuseMaps.begin(), diffuseMaps.end()); vector<Texture> specularMaps = loadMaterialTextures(material, aiTextureType_SPECULAR, "texture_specular"); textures.insert(textures.end(), specularMaps.begin(), specularMaps.end()); } return Mesh(m_glFuns,vertices, indices, textures); } vector<Texture> Model::loadMaterialTextures(aiMaterial *mat, aiTextureType type, string typeName) { vector<Texture> textures; for(unsigned int i = 0; i < mat->GetTextureCount(type); i++) { aiString str; mat->GetTexture(type, i, &str); bool skip = false; for(unsigned int j = 0; j < textures_loaded.size(); j++) { if(std::strcmp(textures_loaded[j].path.data(), str.C_Str()) == 0) { textures.push_back(textures_loaded[j]); skip = true; break; } } if(!skip){ Texture texture; //texture.id = TextureFromFile(str.C_Str(), directory); texture.texture = TextureFromFile(str.C_Str(), directory); texture.type = typeName; texture.path = str.C_Str(); textures.push_back(texture); textures_loaded.push_back(texture); } } return textures; } QOpenGLTexture *Model::TextureFromFile(const char *path, const string &directory) { QOpenGLTexture * texture = nullptr; string filename = string(path); if (filename.find('*') != std::string::npos) { QString qs = QString::fromStdString(filename); QStringList strList = qs.split('*'); QString fileName = m_mapPath.value(strList.at(1).toInt()); QImage image(fileName); texture = new QOpenGLTexture(image); texture->setMinificationFilter(QOpenGLTexture::Nearest); texture->setMinificationFilter(QOpenGLTexture::Linear); texture->setWrapMode(QOpenGLTexture::Repeat); //if (texture == NULL) // qDebug() << "texture is NULL"; //else // qDebug() << filename.c_str() << "loaded"; } else { filename = directory + '/' + filename; qDebug() << "filename = " << QString::fromStdString(filename); QImage image(filename.c_str()); texture = new QOpenGLTexture(image.mirrored()); //if (texture == NULL) // qDebug() << "texture is NULL"; //else // qDebug() << filename.c_str() << "loaded"; } return texture; }