model-converter-python/conv3d/obj.py

84 lines
2.5 KiB
Python
Raw Normal View History

2016-11-21 15:59:03 +01:00
#!/usr/bin/env python3
2016-11-21 17:05:52 +01:00
from .model import ModelParser, Exporter, Vertex, TexCoord, Normal, FaceVertex, Face
2016-11-21 15:59:03 +01:00
from functools import reduce
2016-11-22 11:27:42 +01:00
def is_obj(filename):
return filename[-4:] == '.obj'
2016-11-21 15:59:03 +01:00
class OBJParser(ModelParser):
def __init__(self):
super().__init__()
self.materials = []
def parse_line(self, string):
2016-11-22 11:54:59 +01:00
if string == '':
return
split = string.split()
2016-11-21 15:59:03 +01:00
first = split[0]
split = split[1:]
if first == 'usemtl':
self.currentMaterial = split[0]
elif first == 'v':
self.add_vertex(Vertex().from_array(split))
elif first == 'vn':
self.add_normal(Normal().from_array(split))
elif first == 'vt':
self.add_tex_coord(TexCoord().from_array(split))
elif first == 'f':
splits = list(map(lambda x: x.split('/'), split))
for i in range(len(splits)):
for j in range(len(splits[i])):
if splits[i][j] is not '':
splits[i][j] = int(splits[i][j]) - 1
2016-11-21 15:59:03 +01:00
2016-11-21 16:18:49 +01:00
self.add_face(Face().from_array(splits))
2016-11-21 15:59:03 +01:00
class OBJExporter(Exporter):
def __init__(self, model):
super().__init__(model)
def __str__(self):
string = ""
for vertex in self.model.vertices:
string += "v " + ' '.join([str(vertex.x), str(vertex.y), str(vertex.z)]) + "\n"
2016-11-21 15:59:03 +01:00
string += "\n"
2016-11-21 16:18:49 +01:00
if len(self.model.tex_coords) > 0:
for tex_coord in self.model.tex_coords:
string += "vt " + ' '.join([str(tex_coord.x), str(tex_coord.y)]) + "\n"
2016-11-21 15:59:03 +01:00
2016-11-21 16:18:49 +01:00
string += "\n"
2016-11-21 15:59:03 +01:00
2016-11-21 16:18:49 +01:00
if len(self.model.normals) > 0:
for normal in self.model.normals:
string += "vn " + ' '.join([str(normal.x), str(normal.y), str(normal.z)]) + "\n"
2016-11-21 15:59:03 +01:00
2016-11-21 16:18:49 +01:00
string += "\n"
2016-11-21 15:59:03 +01:00
for face in self.model.faces:
string += "f "
arr = []
for v in [face.a, face.b, face.c]:
sub_arr = []
sub_arr.append(str(v.vertex + 1))
2016-11-21 15:59:03 +01:00
if v.normal is None:
if v.texture is not None:
sub_arr.append('')
sub_arr.append(str(v.texture + 1))
2016-11-21 15:59:03 +01:00
elif v.texture is not None:
sub_arr.append(str(v.texture + 1))
2016-11-21 15:59:03 +01:00
if v.normal is not None:
sub_arr.append(str(v.normal + 1))
2016-11-21 15:59:03 +01:00
arr.append('/'.join(sub_arr))
string += ' '.join(arr) + '\n'
return string