diff options
author | Campbell Barton <ideasman42@gmail.com> | 2009-06-20 18:01:34 +0400 |
---|---|---|
committer | Campbell Barton <ideasman42@gmail.com> | 2009-06-20 18:01:34 +0400 |
commit | ed2887f7209a31ff7bb52147e0baf330477f7548 (patch) | |
tree | 7aa1f5169de340a2acef3e1c8f3049b86277a599 /release | |
parent | f7c0d1b550afbfb7fe2ee2a116f8c66594b034be (diff) |
updated the ply exporter to use the blender 2.5 python api
Note's
* exports the mesh without modifiers or transformations applied
* supports UV's vertex colors
* no support for normals yet (missing from data api)
* registers an operator called EXPORT_OT_ply
* no file selector yet. can only fun from python currently
bpy.ops.EXPORT_OT_ply(filename="/tmp/test.ply")
* removed double lookups on the vertex dict, build face verts to write allong the way.
Diffstat (limited to 'release')
-rw-r--r-- | release/io/export_ply.py | 215 |
1 files changed, 131 insertions, 84 deletions
diff --git a/release/io/export_ply.py b/release/io/export_ply.py index 46d08050302..6471b4b664a 100644 --- a/release/io/export_ply.py +++ b/release/io/export_ply.py @@ -1,16 +1,4 @@ -#!BPY - -""" -Name: 'Stanford PLY (*.ply)...' -Blender: 241 -Group: 'Export' -Tooltip: 'Export active object to Stanford PLY format' -""" - import bpy -import Blender -from Blender import Mesh, Scene, Window, sys, Image, Draw -import BPyMesh __author__ = "Bruce Merry" __version__ = "0.93" @@ -62,85 +50,103 @@ Only one mesh can be exported at a time. def rvec3d(v): return round(v[0], 6), round(v[1], 6), round(v[2], 6) def rvec2d(v): return round(v[0], 6), round(v[1], 6) -def file_callback(filename): +def write(filename, ob, EXPORT_APPLY_MODIFIERS= True, EXPORT_NORMALS= True, EXPORT_UV= True, EXPORT_COLORS= True): if not filename.lower().endswith('.ply'): filename += '.ply' - scn= bpy.data.scenes.active - ob= scn.objects.active if not ob: - Blender.Draw.PupMenu('Error%t|Select 1 active object') + raise Exception("Error, Select 1 active object") return file = open(filename, 'wb') - EXPORT_APPLY_MODIFIERS = Draw.Create(1) - EXPORT_NORMALS = Draw.Create(1) - EXPORT_UV = Draw.Create(1) - EXPORT_COLORS = Draw.Create(1) - #EXPORT_EDGES = Draw.Create(0) - - pup_block = [\ - ('Apply Modifiers', EXPORT_APPLY_MODIFIERS, 'Use transformed mesh data.'),\ - ('Normals', EXPORT_NORMALS, 'Export vertex normal data.'),\ - ('UVs', EXPORT_UV, 'Export texface UV coords.'),\ - ('Colors', EXPORT_COLORS, 'Export vertex Colors.'),\ - #('Edges', EXPORT_EDGES, 'Edges not connected to faces.'),\ - ] - - if not Draw.PupBlock('Export...', pup_block): - return + #EXPORT_EDGES = Draw.Create(0) + """ is_editmode = Blender.Window.EditMode() if is_editmode: Blender.Window.EditMode(0, '', 0) Window.WaitCursor(1) + """ - EXPORT_APPLY_MODIFIERS = EXPORT_APPLY_MODIFIERS.val - EXPORT_NORMALS = EXPORT_NORMALS.val - EXPORT_UV = EXPORT_UV.val - EXPORT_COLORS = EXPORT_COLORS.val - #EXPORT_EDGES = EXPORT_EDGES.val - - mesh = BPyMesh.getMeshFromObject(ob, None, EXPORT_APPLY_MODIFIERS, False, scn) + #mesh = BPyMesh.getMeshFromObject(ob, None, EXPORT_APPLY_MODIFIERS, False, scn) # XXX + mesh = ob.data if not mesh: - Blender.Draw.PupMenu('Error%t|Could not get mesh data from active object') + raise ("Error, could not get mesh data from active object") return - mesh.transform(ob.matrixWorld) + # mesh.transform(ob.matrixWorld) # XXX - faceUV = mesh.faceUV - vertexUV = mesh.vertexUV - vertexColors = mesh.vertexColors + faceUV = len(mesh.uv_layers) > 0 + vertexUV = len(mesh.sticky) > 0 + vertexColors = len(mesh.vcol_layers) > 0 - if (not faceUV) and (not vertexUV): EXPORT_UV = False + if (not faceUV) and (not vertexUV): EXPORT_UV = False if not vertexColors: EXPORT_COLORS = False if not EXPORT_UV: faceUV = vertexUV = False if not EXPORT_COLORS: vertexColors = False + + if faceUV: + active_uv_layer = None + for lay in mesh.uv_layers: + if lay.active: + active_uv_layer= lay.data + break + if not active_uv_layer: + EXPORT_UV = False + faceUV = None + + if vertexColors: + active_col_layer = None + for lay in mesh.vcol_layers: + if lay.active: + active_col_layer= lay.data + if not active_col_layer: + EXPORT_COLORS = False + vertexColors = None # incase color = uvcoord = uvcoord_key = normal = normal_key = None - verts = [] # list of dictionaries + mesh_verts = mesh.verts # save a lookup + ply_verts = [] # list of dictionaries # vdict = {} # (index, normal, uv) -> new index - vdict = [{} for i in xrange(len(mesh.verts))] + vdict = [{} for i in xrange(len(mesh_verts))] + ply_faces = [[] for f in xrange(len(mesh.faces))] vert_count = 0 for i, f in enumerate(mesh.faces): + + smooth = f.smooth + # XXX need face normals + """ if not smooth: normal = tuple(f.no) normal_key = rvec3d(normal) - - if faceUV: uv = f.uv - if vertexColors: col = f.col - for j, v in enumerate(f): + """ + if faceUV: + uv = active_uv_layer[i] + uv = uv.uv1, uv.uv2, uv.uv3, uv.uv4 # XXX - crufty :/ + if vertexColors: + col = active_col_layer[i] + col = col.color1, col.color2, col.color3, col.color4 + + f_verts= list(f.verts) + if not f_verts[3]: f_verts.pop() # XXX face length should be 3/4, not always 4 + + pf= ply_faces[i] + for j, vidx in enumerate(f_verts): + v = mesh_verts[vidx] + """ if smooth: normal= tuple(v.no) normal_key = rvec3d(normal) + """ + normal_key = None # XXX if faceUV: uvcoord= uv[j][0], 1.0-uv[j][1] @@ -149,33 +155,41 @@ def file_callback(filename): uvcoord= v.uvco[0], 1.0-v.uvco[1] uvcoord_key = rvec2d(uvcoord) - if vertexColors: color= col[j].r, col[j].g, col[j].b + if vertexColors: + color= col[j] + color= int(color[0]*255.0), int(color[1]*255.0), int(color[2]*255.0) key = normal_key, uvcoord_key, color - vdict_local = vdict[v.index] + vdict_local = vdict[vidx] + pf_vidx = vdict_local.get(key) # Will be None initially - if (not vdict_local) or (not vdict_local.has_key(key)): - vdict_local[key] = vert_count; - verts.append( (tuple(v.co), normal, uvcoord, color) ) + if pf_vidx == None: # same as vdict_local.has_key(key) + pf_vidx = vdict_local[key] = vert_count; + ply_verts.append((vidx, normal, uvcoord, color)) vert_count += 1 - + + pf.append(pf_vidx) file.write('ply\n') file.write('format ascii 1.0\n') - file.write('comment Created by Blender3D %s - www.blender.org, source file: %s\n' % (Blender.Get('version'), Blender.Get('filename').split('/')[-1].split('\\')[-1] )) + version = "2.5" # Blender.Get('version') + file.write('comment Created by Blender3D %s - www.blender.org, source file: %s\n' % (version, bpy.data.filename.split('/')[-1].split('\\')[-1] )) - file.write('element vertex %d\n' % len(verts)) + file.write('element vertex %d\n' % len(ply_verts)) file.write('property float x\n') file.write('property float y\n') file.write('property float z\n') + + # XXX + """ if EXPORT_NORMALS: file.write('property float nx\n') file.write('property float ny\n') file.write('property float nz\n') - + """ if EXPORT_UV: file.write('property float s\n') file.write('property float t\n') @@ -188,41 +202,74 @@ def file_callback(filename): file.write('property list uchar uint vertex_indices\n') file.write('end_header\n') - for i, v in enumerate(verts): - file.write('%.6f %.6f %.6f ' % v[0]) # co + for i, v in enumerate(ply_verts): + file.write('%.6f %.6f %.6f ' % tuple(mesh_verts[v[0]].co)) # co + """ if EXPORT_NORMALS: file.write('%.6f %.6f %.6f ' % v[1]) # no - - if EXPORT_UV: - file.write('%.6f %.6f ' % v[2]) # uv - if EXPORT_COLORS: - file.write('%u %u %u' % v[3]) # col + """ + if EXPORT_UV: file.write('%.6f %.6f ' % v[2]) # uv + if EXPORT_COLORS: file.write('%u %u %u' % v[3]) # col file.write('\n') - for (i, f) in enumerate(mesh.faces): - file.write('%d ' % len(f)) - smooth = f.smooth - if not smooth: no = rvec3d(f.no) - - if faceUV: uv = f.uv - if vertexColors: col = f.col - for j, v in enumerate(f): - if f.smooth: normal= rvec3d(v.no) - else: normal= no - if faceUV: uvcoord= rvec2d((uv[j][0], 1.0-uv[j][1])) - elif vertexUV: uvcoord= rvec2d((v.uvco[0], 1.0-v.uvco[1])) - if vertexColors: color= col[j].r, col[j].g, col[j].b - - file.write('%d ' % vdict[v.index][normal, uvcoord, color]) - - file.write('\n') + for pf in ply_faces: + if len(pf)==3: file.write('3 %d %d %d\n' % tuple(pf)) + else: file.write('4 %d %d %d %d\n' % tuple(pf)) + file.close() + print("writing", filename, "done") + # XXX + """ if is_editmode: Blender.Window.EditMode(1, '', 0) + """ +""" def main(): Blender.Window.FileSelector(file_callback, 'PLY Export', Blender.sys.makename(ext='.ply')) if __name__=='__main__': main() +""" + +class EXPORT_OT_ply(bpy.types.Operator): + ''' + Operator documentatuon text, will be used for the operator tooltip and python docs. + ''' + __label__ = "Export PLY" + + # List of operator properties, the attributes will be assigned + # to the class instance from the operator settings before calling. + + __props__ = [ + bpy.props.StringProperty(attr="filename", name="File Name", description="File name used for exporting the PLY file", maxlen= 1024, default= "") + ] + + def poll(self, context): + print("Poll") + return context.active_object != None + + def execute(self, context): + # print("Selected: " + context.active_object.name) + + if not self.filename: + raise Exception("filename not set") + + write(self.filename, context.active_object) + + return ('FINISHED',) + + def invoke(self, context, event): + print("Invoke", context, event) + # XXX file selector + self.execute(context) + return ('FINISHED',) + + +bpy.ops.add(EXPORT_OT_ply) + +if __name__ == "__main__": + bpy.ops.EXPORT_OT_ply(filename="/tmp/test.ply") + + |