76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
from model import ModelParser, Exporter, Vertex, TexCoord, Normal, FaceVertex, Face
|
||
|
|
from functools import reduce
|
||
|
|
|
||
|
|
class OBJParser(ModelParser):
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__()
|
||
|
|
self.materials = []
|
||
|
|
|
||
|
|
def parse_line(self, string):
|
||
|
|
split = string.split(' ')
|
||
|
|
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] = str(int(splits[i][j]) - 1)
|
||
|
|
|
||
|
|
self.addFace(Face().from_array(splits))
|
||
|
|
|
||
|
|
class OBJExporter(Exporter):
|
||
|
|
def __init__(self, model):
|
||
|
|
super().__init__(model)
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
string = ""
|
||
|
|
|
||
|
|
for vertex in self.model.vertices:
|
||
|
|
string += "n " + ' '.join([vertex.x, vertex.y, vertex.z]) + "\n"
|
||
|
|
|
||
|
|
string += "\n"
|
||
|
|
|
||
|
|
for tex_coord in self.model.tex_coords:
|
||
|
|
string += "vt " + ' '.join([tex_coord.x, tex_coord.y]) + "\n"
|
||
|
|
|
||
|
|
string += "\n"
|
||
|
|
|
||
|
|
for normal in self.model.normals:
|
||
|
|
string += "vn " + ' '.join([normal.x, normal.y, normal.z]) + "\n"
|
||
|
|
|
||
|
|
string += "\n"
|
||
|
|
|
||
|
|
for face in self.model.faces:
|
||
|
|
string += "f "
|
||
|
|
arr = []
|
||
|
|
for v in [face.a, face.b, face.c]:
|
||
|
|
sub_arr = []
|
||
|
|
sub_arr.append(v.vertex)
|
||
|
|
if v.normal is None:
|
||
|
|
if v.texture is not None:
|
||
|
|
sub_arr.append('')
|
||
|
|
sub_arr.append(v.texture)
|
||
|
|
elif v.texture is not None:
|
||
|
|
sub_arr.append(v.texture)
|
||
|
|
if v.normal is not None:
|
||
|
|
sub_arr.append(v.normal)
|
||
|
|
arr.append('/'.join(sub_arr))
|
||
|
|
|
||
|
|
string += ' '.join(arr) + '\n'
|
||
|
|
return string
|
||
|
|
|