model-converter-python/d3/model/formats/obj.py

156 lines
5.0 KiB
Python
Raw Normal View History

2016-12-02 15:52:01 +01:00
from ..basemodel import ModelParser, Exporter, Vertex, TexCoord, Normal, FaceVertex, Face
from ..mesh import Material, MeshPart
2016-11-21 15:59:03 +01:00
from functools import reduce
2016-11-29 12:05:03 +01:00
import os.path
import PIL.Image
2016-11-29 17:24:39 +01:00
import sys
2016-11-29 12:05:03 +01:00
2016-11-21 15:59:03 +01:00
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):
2016-11-30 15:37:58 +01:00
def __init__(self, up_conversion = None):
super().__init__(up_conversion)
2016-11-29 12:13:43 +01:00
self.current_material = None
2016-11-29 17:24:39 +01:00
self.mtl = None
2016-11-21 15:59:03 +01:00
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:]
2016-11-29 17:24:39 +01:00
if first == 'usemtl' and self.mtl is not None:
self.current_material = self.mtl[split[0]]
2016-11-29 12:05:03 +01:00
elif first == 'mtllib':
path = os.path.join(os.path.dirname(self.path), split[0])
2016-11-29 17:24:39 +01:00
if os.path.isfile(path):
self.mtl = MTLParser(self)
self.mtl.parse_file(path)
else:
print('Warning : ' + path + 'not found ', file=sys.stderr)
2016-11-21 15:59:03 +01:00
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-12-02 09:26:07 +01:00
# if Face3
if len(split) == 3:
face = Face().from_array(splits)
face.material = self.current_material
self.add_face(face)
elif len(split) == 4:
2016-12-02 09:53:07 +01:00
face = Face().from_array(splits[:3])
2016-12-02 09:26:07 +01:00
face.material = self.current_material
self.add_face(face)
face = Face().from_array([splits[0], splits[2], splits[3]])
face.material = self.current_material
self.add_face(face)
else:
print('Face with more than 4 vertices are not supported', file=sys.stderr)
2016-11-21 15:59:03 +01:00
2016-11-29 12:05:03 +01:00
class MTLParser:
def __init__(self, parent):
self.parent = parent
self.materials = []
self.current_mtl = None
def parse_line(self, string):
if string == '':
return
split = string.split()
first = split[0]
split = split[1:]
if first == 'newmtl':
self.current_mtl = Material(split[0])
self.materials.append(self.current_mtl)
elif first == 'Ka':
self.current_mtl.Ka = Vertex().from_array(split)
elif first == 'Kd':
self.current_mtl.Kd = Vertex().from_array(split)
elif first == 'Ks':
self.current_mtl.Ks = Vertex().from_array(split)
elif first == 'map_Kd':
self.current_mtl.map_Kd = PIL.Image.open(os.path.join(os.path.dirname(self.parent.path), split[0]))
2016-11-29 12:05:03 +01:00
def parse_file(self, path):
with open(path) as f:
for line in f.readlines():
line = line.rstrip()
self.parse_line(line)
def __getitem__(self, key):
for material in self.materials:
if material.name == key:
return material
2016-11-29 12:05:03 +01:00
2016-11-21 15:59:03 +01:00
class OBJExporter(Exporter):
def __init__(self, model):
super().__init__(model)
def __str__(self):
2016-11-30 11:47:21 +01:00
current_material = ''
2016-11-21 15:59:03 +01:00
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
2016-11-30 11:47:21 +01:00
faces = sum(map(lambda x: x.faces, self.model.parts), [])
for face in faces:
2016-11-30 14:57:58 +01:00
if face.material is not None and face.material.name != current_material:
2016-11-30 11:47:21 +01:00
current_material = face.material.name
string += "usemtl " + current_material + "\n"
2016-11-21 15:59:03 +01:00
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:
2016-11-30 11:47:21 +01:00
if v.tex_coord is not None:
2016-11-21 15:59:03 +01:00
sub_arr.append('')
2016-11-30 11:47:21 +01:00
sub_arr.append(str(v.tex_coord + 1))
elif v.tex_coord is not None:
sub_arr.append(str(v.tex_coord + 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