import openvdb as vdb
import numpy

filename = '/work/stanford/Dragon.obj'

print 'reading: %s' % filename

points = []
triangles = []
quads = []

f = open(filename, 'r')
lines = f.readlines()

for line in lines:
    t = line.split()
    if t[0] == 'v':         # vertex
        points.append( [ float(t[1]), float(t[2]), float(t[3]) ] )
    elif t[0] == 'f':
        if len(t) == 4:     # triangle
            triangles.append( [ int(t[1])-1, int(t[2])-1, int(t[3])-1 ] )
        elif len(t) == 5:   # quad
            quads.append( [ int(t[1])-1, int(t[2])-1, int(t[3])-1, int(t[4])-1 ] )


np_points = numpy.ndarray((len(points), 3), dtype=float)
np_triangles = numpy.ndarray((len(triangles), 3), dtype=int)
np_quads = numpy.ndarray((len(quads), 4), dtype=int)

for i in range(0, len(points)):
    np_points[i] = points[i]

for i in range(0, len(triangles)):
    np_triangles[i] = triangles[i]

for i in range(0, len(quads)):
    np_quads[i] = quads[i]

xform = vdb.createLinearTransform(voxelSize=0.5)
grid = vdb.FloatGrid.createLevelSetFromPolygons(np_points, np_triangles, np_quads, transform=xform)
grid.name = 'surface'

print 'name:', grid.name, ' class:', grid.gridClass, ' background value:', grid.background, ' voxel size:', grid.transform.voxelSize()[0], ' active voxels:', grid.activeVoxelCount()

vdb.write('/tmp/foo.vdb', grid)