RayTracer/objToPov.py at master · wallisc/RayTracer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
# Converts a .obj file to a povray file using smooth_triangles
# Christopher Wallis
# Example usage: 'python objToPov.py bunny.obj'

import sys, re

DEFAULT_CAMERA = """
camera {
    location <1, 6, 12>
    up <0, 1, 0>
    right <1.33, 0, 0>
    look_at <0, 0, 0>
}
"""

DEFAULT_LIGHTS = """
light_source { <30, 10, 30> color rgb <1.0, 1.0, 1.0> }
"""

DEFAULT_MATERIAL = """
   pigment { image_map "blitz.bmp" }
   finish { ambient 0.2 diffuse 0.6 specular 0.4 roughness 0.05}
"""

def vecToStr(vec):
   return "<" + ", ".join(["%.6f" % i for i in vec]) + ">"

def convertToPOV(fileName):
   objCheck = re.compile(r".*\.obj")

   if objCheck.match(fileName) == None:
      print fileName + " is not a valid .obj file\n"
      return

   file = open(fileName, 'r')
   outputName = fileName[:len(fileName) - 4] + ".pov"
   outFile = open(outputName, 'w')

   outFile.write(DEFAULT_CAMERA)
   outFile.write(DEFAULT_LIGHTS)

   normList = []
   vertList = []
   texList = []
   normalCheck = re.compile(r"vn\s.*")
   vertexCheck = re.compile(r"v\s.*")
   textureCheck = re.compile(r"vt\s.*")
   faceCheck = re.compile(r"f\s.*")

   line = file.readline()
   while line != '':
      if normalCheck.match(line):
         normals = re.split("\s", line)
         # Skip the vn tag (i.e. start at index 1)
         n1 = float(normals[1])
         n2 = float(normals[2])
         n3 = float(normals[3])
         normList.append((n1, n2, n3))

      elif vertexCheck.match(line):
         vertex = re.split("\s", line)
         # Skip the v tag (i.e. start at index 1)
         v1 = float(vertex[1])
         v2 = float(vertex[2])
         v3 = float(vertex[3])
         vertList.append((v1, v2, v3))

      elif textureCheck.match(line):
         vertex = re.split("\s", line)
         # Skip the vt tag (i.e. start at index 1)
         v1 = float(vertex[1])
         v2 = float(vertex[2])
         texList.append((v1, v2))

      elif faceCheck.match(line):
         break

      line = file.readline()


   # If the list of normals, uvs, and vertices have been fully constructed
   faceCount = 0
   while line != '':
      if faceCheck.match(line):
         faceCount += 1
         outFile.write("smooth_triangle {\n")
         face = re.split("\s", line)
         uvIdxList = []
         for idx in range(1, 4):
            point = re.split("/", face[idx])
            # Note: OBJ files are 1-indexed
            vertIdx = int(point[0]) - 1

            # If vertex textures were specified
            if len(point) == 3:
               uvIdxList.append(int(point[1]) - 1)

            normIdx = int(point[len(point) - 1]) - 1
            outFile.write("   " + vecToStr(vertList[vertIdx]) + ", "
                                + vecToStr(normList[normIdx]))

            if idx < 3: outFile.write(",\n")
            else: outFile.write("\n")

         if len(uvIdxList) == 3:
            outFile.write("   uv {")
            for idx in range(0, 3):
               outFile.write(vecToStr(texList[uvIdxList[idx]]))
               if idx < 2: outFile.write(", ")
               else: outFile.write("}\n")


         outFile.write(DEFAULT_MATERIAL)
         outFile.write("}\n\n")

      line = file.readline()

   print "Succesfully wrote out " + outputName
   print "Number of triangles: " + repr(faceCount)

def main():
   args = sys.argv
   for i in range(1, len(args)):
      convertToPOV(args[i])

main()