Added function load_model

conv3d/loadmodel.py

@@ -0,0 +1,18 @@
+#!/usr/bin/env python3
+
+from .obj import is_obj, OBJParser
+from .ply import is_ply, PLYParser
+
+def load_model(path):
+    parser = None
+
+    if is_obj(path):
+        parser = OBJParser()
+    elif is_ply(path):
+        parser = PLYParser()
+    else:
+        raise Exception("File format not supported")
+
+    parser.parse_file(path)
+
+    return parser

conv3d/model.py

@@ -74,3 +74,4 @@ class Exporter:
     def __init__(self, model):
         self.model = model
 
+

conv3d/obj.py

@@ -3,6 +3,9 @@
 from .model import ModelParser, Exporter, Vertex, TexCoord, Normal, FaceVertex, Face
 from functools import reduce
 
+def is_obj(filename):
+    return filename[-4:] == '.obj'
+
 class OBJParser(ModelParser):
 
     def __init__(self):
@@ -69,7 +72,7 @@ class OBJExporter(Exporter):
                 elif v.texture is not None:
                     sub_arr.append(str(int(v.texture) + 1))
                     if v.normal is not None:
-                        sub_arr.append(str(int(v.normal + 1)))
+                        sub_arr.append(str(int(v.normal) + 1))
                 arr.append('/'.join(sub_arr))
 
             string += ' '.join(arr) + '\n'

conv3d/ply.py

@@ -2,6 +2,9 @@
 
 from .model import ModelParser, Exporter, Vertex, Face, FaceVertex
 
+def is_ply(filename):
+    return filename[-4:] == '.ply'
+
 class PLYParser(ModelParser):
 
     def __init__(self):
@@ -102,4 +105,3 @@ class PLYExporter(Exporter):
 
         return string
 
-