[Bf-blender-cvs] SVN commit: /data/svn/bf-blender [21033] branches/blender2.5/blender/ release/io/export_ply.py: updated the ply exporter to use the blender 2. 5 python api
Campbell Barton
ideasman42 at gmail.com
Sat Jun 20 16:01:34 CEST 2009
Revision: 21033
http://projects.blender.org/plugins/scmsvn/viewcvs.php?view=rev&root=bf-blender&revision=21033
Author: campbellbarton
Date: 2009-06-20 16:01:34 +0200 (Sat, 20 Jun 2009)
Log Message:
-----------
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.
Modified Paths:
--------------
branches/blender2.5/blender/release/io/export_ply.py
Modified: branches/blender2.5/blender/release/io/export_ply.py
===================================================================
--- branches/blender2.5/blender/release/io/export_ply.py 2009-06-20 13:55:47 UTC (rev 21032)
+++ branches/blender2.5/blender/release/io/export_ply.py 2009-06-20 14:01:34 UTC (rev 21033)
@@ -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 @@
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
-
+ """
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) # XXX
+ mesh = ob.data
- mesh = BPyMesh.getMeshFromObject(ob, None, EXPORT_APPLY_MODIFIERS, False, scn)
-
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 @@
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 @@
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")
+
+
More information about the Bf-blender-cvs
mailing list