from ..basemodel import ModelParser, Exporter, Vertex, TexCoord, Normal, FaceVertex, Face from ..mesh import Material, MeshPart from functools import reduce import os.path import sys def is_obj(filename): return filename[-4:] == '.obj' class OBJParser(ModelParser): def __init__(self, up_conversion = None): super().__init__(up_conversion) self.current_material = None self.mtl = None def parse_line(self, string): if string == '': return split = string.split() first = split[0] split = split[1:] if first == 'usemtl' and self.mtl is not None: self.current_material = self.mtl[split[0]] elif first == 'mtllib': path = os.path.join(os.path.dirname(self.path), split[0]) if os.path.isfile(path): self.mtl = MTLParser(self) self.mtl.parse_file(path) else: print('Warning : ' + path + 'not found ', file=sys.stderr) 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 # if Face3 if len(split) == 3: face = Face().from_array(splits) face.material = self.current_material self.add_face(face) elif len(split) == 4: face = Face().from_array(splits[:3]) 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) 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': try: import PIL.Image self.current_mtl.map_Kd = PIL.Image.open(os.path.join(os.path.dirname(self.parent.path), split[0])) except: pass 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 class OBJExporter(Exporter): def __init__(self, model): super().__init__(model) def __str__(self): current_material = '' string = "" for vertex in self.model.vertices: string += "v " + ' '.join([str(vertex.x), str(vertex.y), str(vertex.z)]) + "\n" string += "\n" 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" string += "\n" 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" string += "\n" faces = sum(map(lambda x: x.faces, self.model.parts), []) for face in faces: if face.material is not None and face.material.name != current_material: current_material = face.material.name string += "usemtl " + current_material + "\n" string += "f " arr = [] for v in [face.a, face.b, face.c]: sub_arr = [] sub_arr.append(str(v.vertex + 1)) if v.normal is None: if v.tex_coord is not None: sub_arr.append('') sub_arr.append(str(v.tex_coord + 1)) elif v.tex_coord is not None: sub_arr.append(str(v.tex_coord + 1)) if v.normal is not None: sub_arr.append(str(v.normal + 1)) arr.append('/'.join(sub_arr)) string += ' '.join(arr) + '\n' return string