Start obj conversion tool

obj2array.py

@@ -0,0 +1,136 @@
+# Python tool to convert a 3D model from the text obj format to C arrays to be
+# used with small3dlib.
+#
+# Usage:
+#
+# TODO
+#
+# by drummfish
+# released under CC0 1.0.
+
+import sys
+
+fileName = sys.argv[1]
+
+VERTEX_SCALE = 512
+U_SCALE = 512
+V_SCALE = 512
+NAME = "model"
+COMPACT = True
+INDEXED_UVS = False
+
+objFile = open(fileName)
+
+vertices = []
+uvs = []
+triangles = []
+triangleUVs = []
+
+# parse the file:
+
+for line in objFile:
+  if line[:2] == "v ":
+    coords = line[2:].split()
+    vertex = [float(coords[i]) for i in range(3)]
+    vertices.append(vertex)
+  elif line[:3] == "vt ":
+    coords = line[3:].split()
+    vertex = [float(coords[i]) for i in range(2)]
+    uvs.append(vertex)
+  elif line[:2] == "f ":
+    indices = line[2:].split()
+
+    if len(indices) != 3:
+      raise(Exception("The model is not triangulated!"))
+
+    t = []
+    u = []
+ 
+    for i in indices:
+      components = i.split("/")
+      t.append(int(components[0]) - 1)
+      u.append(int(components[0]) - 1)
+
+    triangles.append(t)
+    triangleUVs.append(t)
+
+# print the result:
+
+def arrayString(name, array, components, scales, align, short):
+  result = name + " = [\n"
+
+  if COMPACT:
+    lineLen = 0
+    first = True
+    n = 0
+
+    for v in array:
+      for c in v:
+        item = ""
+
+        if first:
+          first = False
+        else:
+          result += ","
+          lineLen += 1
+
+          if lineLen >= 80:
+            result += "\n"
+            lineLen = 0
+
+        num = c * scales[n % len(scales)]
+
+        if short:
+          item += str(num)
+        else:
+          item += ("" if num >= 0 else "-") + "0x%x" % abs(num)
+
+        if lineLen + len(item) > 80:
+          result += "\n"
+          lineLen = 0
+       
+        result += item
+        lineLen += len(item)
+        n += 1
+
+    result += "]\n"
+
+  else: # non-compact
+    n = 0
+    endIndex = len(array) - 1
+
+    for v in array:
+      line = "  " + ", ".join([str(int(v[c] * scales[c % len(scales)])).rjust(align) for c in range(components)])
+
+      if n < endIndex:
+        line += ","
+
+      line = line.ljust((components + 2) * (align + 1)) + "// " + str(n * components) + "\n"
+      result += line
+      n += 1
+
+    result += "]; // " + name + "\n"
+
+  return result
+
+result = ""
+
+print(arrayString(NAME + "Vertices",vertices,3,[VERTEX_SCALE],5,False))
+print(arrayString(NAME + "TriangleIndices",triangles,3,[1],5,True))
+
+if INDEXED_UVS:
+  print(arrayString(NAME + "UVs",vertices,2,[U_SCALE,V_SCALE],5,False))
+  print(arrayString(NAME + "UVIndices",triangleUVs,3,[1],5,True))
+else:
+  uvs2 = []
+  for item in triangleUVs:
+    uvs2.append([
+      uvs[item[0]][0],
+      uvs[item[0]][1],
+      uvs[item[1]][0],
+      uvs[item[1]][1],
+      uvs[item[2]][0],
+      uvs[item[2]][1]])
+
+  print(arrayString(NAME + "TriangleUVs",uvs2,6,[U_SCALE,V_SCALE],5,False))
+