~ | Vertex normal obj handling

This commit is contained in:
TheRedShip
2025-02-24 20:19:06 +01:00
committed by tomoron
parent 30ca774491
commit 62d0863ff4
11 changed files with 108 additions and 50 deletions

View File

@ -58,6 +58,15 @@ glm::vec2 ObjParser::getUV(std::stringstream &line)
return(res);
}
glm::vec3 ObjParser::getNormals(std::stringstream &line)
{
glm::vec3 res;
if(!(line >> res.x) || !(line >> res.y) || (!(line >> res.z)) && !line.eof())
throw std::runtime_error("syntax error in obj file while parsing normal vertex");
return(res);
}
long int ObjParser::checkVertexIndex(int index, size_t size)
{
if((size_t)index > size || index == 0 || (index < 0 && (size_t)(-index) > size))
@ -97,10 +106,11 @@ int ObjParser::pointInTriangle(glm::vec3 pts[3], std::vector<glm::vec3> vertices
return(0);
}
bool ObjParser::addTriangleFromPolygon(std::vector<glm::vec3> &vertices, std::vector<glm::vec2> &textureVertices, int inv)
bool ObjParser::addTriangleFromPolygon(std::vector<glm::vec3> &vertices, std::vector<glm::vec2> &textureVertices, std::vector<glm::vec3> &normalVertices, int inv)
{
glm::vec3 v1, v2, v3;
glm::vec2 vt1, vt2, vt3;
glm::vec3 vn1, vn2, vn3;
float dot;
@ -110,11 +120,13 @@ bool ObjParser::addTriangleFromPolygon(std::vector<glm::vec3> &vertices, std::ve
{
v1 = vertices.back();
vt1 = textureVertices.back();
vn1 = normalVertices.back();
}
else
{
v1 = vertices[i - 1];
vt1 = textureVertices[i - 1];
vn1 = normalVertices[i - 1];
}
v2 = vertices[i];
v3 = vertices[(i + 1) % vertices.size()];
@ -122,6 +134,9 @@ bool ObjParser::addTriangleFromPolygon(std::vector<glm::vec3> &vertices, std::ve
vt2 = textureVertices[i];
vt3 = textureVertices[(i + 1) % textureVertices.size()];
vn2 = normalVertices[i];
vn3 = normalVertices[(i + 1) % normalVertices.size()];
if (inv)
dot = glm::cross(v2 - v1, v2 - v3).z;
else
@ -133,14 +148,19 @@ bool ObjParser::addTriangleFromPolygon(std::vector<glm::vec3> &vertices, std::ve
continue;
vertices.erase(vertices.begin() + i);
textureVertices.erase(textureVertices.begin() + i);
normalVertices.erase(normalVertices.begin() + i);
std::vector<glm::vec2> texture;
texture.push_back(vt1);
texture.push_back(vt2);
texture.push_back(vt3);
addTriangle(v1, v2, v3, texture);
std::vector<glm::vec3> normals;
normals.push_back(vn1);
normals.push_back(vn2);
normals.push_back(vn3);
addTriangle(v1, v2, v3, texture, normals);
return(1);
}
return(0);
@ -159,7 +179,7 @@ std::vector<std::string> ObjParser::objSplit(std::string str, std::string delim)
return(res);
}
void ObjParser::getFaceVertices(std::vector<glm::vec3> &faceVertices, std::vector<glm::vec2> &textureVertices, std::stringstream &line)
void ObjParser::getFaceVertices(std::vector<glm::vec3> &faceVertices, std::vector<glm::vec2> &textureVertices, std::vector<glm::vec3> &normalVertices, std::stringstream &line)
{
std::string el;
std::vector<std::string> sp;
@ -174,6 +194,9 @@ void ObjParser::getFaceVertices(std::vector<glm::vec3> &faceVertices, std::vecto
if (sp.size() > 1 && sp[1].length())
textureVertices.push_back(_textureVertices[checkVertexIndex(std::stoi(sp[1]), _textureVertices.size())]);
if (sp.size() > 2 && sp[2].length())
normalVertices.push_back(_normalVertices[checkVertexIndex(std::stoi(sp[2]), _normalVertices.size())]);
faceVertices.push_back(_vertices[checkVertexIndex(std::stoi(sp[0]), _vertices.size())]);
}
}
@ -182,29 +205,34 @@ void ObjParser::addFace(std::stringstream &line)
{
std::vector<glm::vec3> faceVertices;
std::vector<glm::vec2> textureVertices;
std::vector<glm::vec3> normalVertices;
getFaceVertices(faceVertices, textureVertices, line);
getFaceVertices(faceVertices, textureVertices, normalVertices, line);
if(!line.eof())
throw std::runtime_error("OBJ : an error occured while paring face");
if(faceVertices.size() < 3)
throw std::runtime_error("OBJ : face does not have enough vertices");
while(faceVertices.size() > 3)
if (!addTriangleFromPolygon(faceVertices, textureVertices, 0))
if(!addTriangleFromPolygon(faceVertices, textureVertices, 1))
if (!addTriangleFromPolygon(faceVertices, textureVertices, normalVertices, 0))
if(!addTriangleFromPolygon(faceVertices, textureVertices, normalVertices, 1))
return ;
if(!line.eof())
throw std::runtime_error("OBJ: an error occured while parsing face");
addTriangle(faceVertices[0], faceVertices[1], faceVertices[2], textureVertices);
addTriangle(faceVertices[0], faceVertices[1], faceVertices[2], textureVertices, normalVertices);
}
void ObjParser::addTriangle(glm::vec3 v1, glm::vec3 v2, glm::vec3 v3, std::vector<glm::vec2> texture_vertices)
void ObjParser::addTriangle(glm::vec3 v1, glm::vec3 v2, glm::vec3 v3, std::vector<glm::vec2> texture_vertices, std::vector<glm::vec3> normal_vertices)
{
glm::vec2 vt1 = glm::vec2(0.);
glm::vec2 vt2 = glm::vec2(0.);
glm::vec2 vt3 = glm::vec2(0.);
glm::vec3 vn1 = glm::vec3(0.);
glm::vec3 vn2 = glm::vec3(0.);
glm::vec3 vn3 = glm::vec3(0.);
if (texture_vertices.size() == 3)
{
vt1 = texture_vertices[0];
@ -212,7 +240,14 @@ void ObjParser::addTriangle(glm::vec3 v1, glm::vec3 v2, glm::vec3 v3, std::vecto
vt3 = texture_vertices[2];
}
_triangles.push_back(Triangle(v1, v2, v3, vt1, vt2, vt3, _mat));
if (normal_vertices.size() == 3)
{
vn1 = glm::normalize(normal_vertices[0]);
vn2 = glm::normalize(normal_vertices[1]);
vn3 = glm::normalize(normal_vertices[2]);
}
_triangles.push_back(Triangle(v1, v2, v3, vt1, vt2, vt3, vn1, vn2, vn3, _mat));
}
void ObjParser::parseMtl(std::stringstream &input_line, Scene &scene)
@ -286,7 +321,9 @@ void ObjParser::parseMtl(std::stringstream &input_line, Scene &scene)
float x, y, z;
if(!(lineStream >> x >> y >> z))
throw std::runtime_error("OBJ: syntax error while getting material emission");
mat->emission = (x + y + z) / 3;
if (mat->color == glm::vec3(0.0f))
mat->color = glm::vec3(x, y, z);
mat->emission = (x + y + z);
}
else if(identifier == "Ni")
{
@ -384,6 +421,8 @@ void ObjParser::parse(Scene &scene, glm::vec3 offset, float scale, glm::mat4 tra
_vertices.push_back(getVertex(lineStream));
else if (identifier == "vt")
_textureVertices.push_back(getUV(lineStream));
else if (identifier == "vn")
_normalVertices.push_back(getNormals(lineStream));
else if (identifier == "f")
addFace(lineStream);
else if (identifier == "mtllib")

View File

@ -115,17 +115,7 @@ void Scene::addObject(Object *obj)
}
else if (obj->getType() == Object::Type::TRIANGLE)
{
GPUTriangle gpu_triangle;
auto triangle = static_cast<Triangle *>(obj);
gpu_triangle.position = triangle->getPosition();
gpu_triangle.mat_index = triangle->getMaterialIndex();
gpu_triangle.vertex1 = triangle->getVertex2();
gpu_triangle.vertex2 = triangle->getVertex3();
gpu_triangle.normal = triangle->getNormal();
_gpu_triangles.push_back(gpu_triangle);
gpu_obj.vertex1 = triangle->getVertex2();
gpu_obj.vertex2 = triangle->getVertex3();
@ -188,12 +178,15 @@ void Scene::addBvh(std::vector<Triangle> &triangles, glm::vec3 offset, float sc
gpu_triangle.vertex1 = triangles[i].getVertex2();
gpu_triangle.vertex2 = triangles[i].getVertex3();
gpu_triangle.normal = triangles[i].getNormal();
gpu_triangle.texture_vertex1 = triangles[i].getTextureVertex1();
gpu_triangle.texture_vertex2 = triangles[i].getTextureVertex2();
gpu_triangle.texture_vertex3 = triangles[i].getTextureVertex3();
gpu_triangle.normal_vertex1 = triangles[i].getNormalVertex1();
gpu_triangle.normal_vertex2 = triangles[i].getNormalVertex2();
gpu_triangle.normal_vertex3 = triangles[i].getNormalVertex3();
_gpu_triangles.push_back(gpu_triangle);
}