From e70610ab1eb83b1e6c549f2646ebf7cd2cc9937d Mon Sep 17 00:00:00 2001 From: Campbell Barton Date: Mon, 3 Jul 2006 03:22:48 +0000 Subject: [PATCH] Update to ply export, give nice messages to the user rather then throwing errors at the user. Export mesh data with modifiers and objects matrix applied. export not only mesh data but text, nurbs etc (thanks to BPYMesh getMeshFromObject) Small updates to ply import, dont set the TEX of a face (no pink faces anymore) --- release/scripts/bpymodules/BPyMesh.py | 9 +- release/scripts/ply_export.py | 174 ++++++++++++++------------ release/scripts/ply_import.py | 29 +++-- 3 files changed, 122 insertions(+), 90 deletions(-) diff --git a/release/scripts/bpymodules/BPyMesh.py b/release/scripts/bpymodules/BPyMesh.py index 02896296feb..a4f745a3d19 100644 --- a/release/scripts/bpymodules/BPyMesh.py +++ b/release/scripts/bpymodules/BPyMesh.py @@ -1,9 +1,8 @@ import Blender -#from BPyMesh_redux import redux # seperated because of its size. -#from BPyMesh_redux import redux # seperated because of its size. -import BPyMesh_redux -reload(BPyMesh_redux) -redux= BPyMesh_redux.redux +from BPyMesh_redux import redux # seperated because of its size. +#import BPyMesh_redux +#reload(BPyMesh_redux) +#redux= BPyMesh_redux.redux # python 2.3 has no reversed() iterator. this will only work on lists and tuples try: diff --git a/release/scripts/ply_export.py b/release/scripts/ply_export.py index 86197692082..e09b9e59133 100644 --- a/release/scripts/ply_export.py +++ b/release/scripts/ply_export.py @@ -1,15 +1,14 @@ #!BPY """ -Name: 'PLY...' -Blender: 237 +Name: 'Stanford PLY (*.ply)...' +Blender: 241 Group: 'Export' -Tooltip: 'Export to Stanford PLY format' +Tooltip: 'Export active object to Stanford PLY format' """ import Blender -import meshtools -import math +import BPyMesh __author__ = "Bruce Merry" __version__ = "0.9" @@ -34,80 +33,101 @@ normals and per-face colours and texture coordinates. # along with this program; if not, write to the Free Software Foundation, # Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. +# Vector rounding se we can use as keys +def rvec3d(v): return round(v.x, 6), round(v.y, 6), round(v.z, 6) +def rvec2d(v): return round(v.x, 6), round(v.y, 6) +def rcol(c): return c.r, c.g, c.b + def file_callback(filename): - if filename.find('.ply', -4) < 0: filename += '.ply' - file = open(filename, "wb") - objects = Blender.Object.GetSelected() - obj = objects[0] - mesh = objects[0].data + if not filename.lower().endswith('.ply'): + filename += '.ply' + + scn= Blender.Scene.GetCurrent() + object= scn.getActiveObject() + if not object: + Blender.Draw.PupMenu('Error%t|Select 1 active object') + return + + file = open(filename, 'wb') + mesh = BPyMesh.getMeshFromObject(object, None, True, False, scn) + if not mesh: + Blender.Draw.PupMenu('Error%t|Could not get mesh data from active object') + return + + mesh.transform(object.matrixWorld) - have_uv = mesh.hasFaceUV() - have_col = meshtools.has_vertex_colors(mesh) - verts = [] # list of dictionaries - vdict = {} # (index, normal, uv) -> new index - for (i, f) in enumerate(mesh.faces): - for (j, v) in enumerate(f.v): - index = v.index - key = index, tuple(v.no) - vdata = {'position': v.co, 'normal': v.no} - if have_uv: - vdata['uv'] = f.uv[j] - key = key + (tuple(f.uv[j]), ) - if have_col: - vdata['col'] = f.col[j] - key = key + ((f.col[j].r, f.col[j].g, f.col[j].b, f.col[j].a), ) - if not vdict.has_key(key): - vdict[key] = len(verts); - verts.append(vdata) - if not i % 100 and meshtools.show_progress: - Blender.Window.DrawProgressBar(float(i) / len(mesh.faces), "Organising vertices") + have_uv = mesh.faceUV + verts = [] # list of dictionaries + vdict = {} # (index, normal, uv) -> new index + for i, f in enumerate(mesh.faces): + f_col= f.col + f_uv= f.uv + for j, v in enumerate(f.v): + index = v.index + + if have_uv: + vdata = v.co, v.no + uv= f_uv[j] + col= f_col[j] + vdata = v.co, v.no, uv, col + key = index, rvec3d(v.no), rcol(col), rvec2d(uv) + else: + vdata = v.co, v.no + key = index, rvec3d(v.no) + + if not vdict.has_key(key): + vdict[key] = len(verts); + verts.append(vdata) + file.write('ply\n') + file.write('format ascii 1.0\n') + file.write('Created by Blender3D %s - www.blender.org, source file: %s\n' % (Blender.Get('version'), Blender.Get('filename').split('/')[-1].split('\\')[-1] )) + + file.write('element vertex %d\n' % len(verts)) + + file.write('property float32 x\n') + file.write('property float32 y\n') + file.write('property float32 z\n') + file.write('property float32 nx\n') + file.write('property float32 ny\n') + file.write('property float32 nz\n') + + if have_uv: + file.write('property float32 s\n') + file.write('property float32 t\n') + file.write('property uint8 red\n') + file.write('property uint8 green\n') + file.write('property uint8 blue\n') + + file.write('element face %d\n' % len(mesh.faces)) + file.write('property list uint8 int32 vertex_indices\n') + file.write('end_header\n') - print >> file, "ply" - print >> file, "format ascii 1.0" - print >> file, "comment created by ply_export.py from Blender" - print >> file, "element vertex %d" % len(verts) - print >> file, "property float32 x" - print >> file, "property float32 y" - print >> file, "property float32 z" - print >> file, "property float32 nx" - print >> file, "property float32 ny" - print >> file, "property float32 nz" - if have_uv: - print >> file, "property float32 s" - print >> file, "property float32 t" - if have_col: - print >> file, "property uint8 red" - print >> file, "property uint8 green" - print >> file, "property uint8 blue" - print >> file, "element face %d" % len(mesh.faces) - print >> file, "property list uint8 int32 vertex_indices" - print >> file, "end_header" - - for (i, v) in enumerate(verts): - print >> file, "%f %f %f %f %f %f" % (tuple(v['position']) + tuple(v['normal'])), - if have_uv: print >> file, "%f %f" % tuple(v['uv']), - if have_col: print >> file, "%u %u %u" % (v['col'].r, v['col'].g, v['col'].b), - print >> file - if not i % 100 and meshtools.show_progress: - Blender.Window.DrawProgressBar(float(i) / len(verts), "Writing vertices") - for (i, f) in enumerate(mesh.faces): - print >> file, "%d" % len(f.v), - for j in range(len(f.v)): - v = f.v[j] - index = v.index - key = index, tuple(v.no) - if have_uv: - key = key + (tuple(f.uv[j]), ) - if have_col: - key = key + ((f.col[j].r, f.col[j].g, f.col[j].b, f.col[j].a), ) - print >> file, "%d" % vdict[key], - print >> file - if not i % 100 and meshtools.show_progress: - Blender.Window.DrawProgressBar(float(i) / len(mesh.faces), "Writing faces") - - Blender.Window.DrawProgressBar(1.0, '') # clear progressbar + for i, v in enumerate(verts): + file.write('%.6f %.6f %.6f ' % tuple(v[0])) # co + file.write('%.6f %.6f %.6f ' % tuple(v[1])) # no + + if have_uv: + file.write('%.6f %.6f %u %u %u' % (v[2].x, v[2].y, v[3].r, v[3].g, v[3].b)) # uv/col + file.write('\n') + + for (i, f) in enumerate(mesh.faces): + file.write('%d ' % len(f)) + + f_col= f.col + f_uv= f.uv + + for j, v in enumerate(f.v): + index = v.index + + if have_uv: + uv= f_uv[j] + col= f_col[j] + key = index, rvec3d(v.no), rcol(col), rvec2d(uv) + else: + key = index, rvec3d(v.no) + + file.write('%d ' % vdict[key]) + file.write('\n') file.close() - message = "Successfully exported " + Blender.sys.basename(filename) - meshtools.print_boxed(message) -Blender.Window.FileSelector(file_callback, "PLY Export") +Blender.Window.FileSelector(file_callback, 'PLY Export', Blender.sys.makename(ext='.ply')) diff --git a/release/scripts/ply_import.py b/release/scripts/ply_import.py index 9df72ca8484..dcdbc11ce5d 100644 --- a/release/scripts/ply_import.py +++ b/release/scripts/ply_import.py @@ -1,8 +1,8 @@ #!BPY """ -Name: 'PLY...' -Blender: 237 +Name: 'Stanford PLY (*.ply)...' +Blender: 241 Group: 'Import' Tip: 'Import a Stanford PLY file' """ @@ -202,6 +202,7 @@ def read(filename): file.close() return (obj_spec, obj); + def add_face(vertices, varr, indices, uvindices, colindices): face = Blender.NMesh.Face([varr[i] for i in indices]) for index in indices: @@ -209,9 +210,11 @@ def add_face(vertices, varr, indices, uvindices, colindices): if uvindices: face.uv.append((vertex[uvindices[0]], 1.0 - vertex[uvindices[1]])) + face.mode &= ~Blender.NMesh.FaceModes.TEX if colindices: if not uvindices: face.uv.append((0, 0)) # Force faceUV face.col.append(Blender.NMesh.Col(vertex[colindices[0]], vertex[colindices[1]], vertex[colindices[2]], 255)) + face.mode &= ~Blender.NMesh.FaceModes.TEX return face def filesel_callback(filename): @@ -273,20 +276,30 @@ def filesel_callback(filename): del obj # Reclaim memory - + + ''' if noindices: normals = 1 else: normals = 0 + ''' + objname = Blender.sys.splitext(Blender.sys.basename(filename))[0] - if not meshtools.overwrite_mesh_name: - objname = meshtools.versioned_name(objname) - Blender.NMesh.PutRaw(mesh, objname, not normals) - Blender.Object.GetSelected()[0].name = objname + scn= Blender.Scene.GetCurrent() + for obj in scn.getChildren(): + obj.sel= 0 + + obj= Blender.Object.New('Mesh', objname) + mesh.name= objname + obj.link(mesh) + scn.link(obj) + obj.sel= 1 + obj.Layers= scn.Layers + Blender.Redraw() Blender.Window.DrawProgressBar(1.0, '') message = 'Successfully imported ' + Blender.sys.basename(filename) + ' ' + str(Blender.sys.time()-t) meshtools.print_boxed(message) -Blender.Window.FileSelector(filesel_callback, 'Import PLY') +Blender.Window.FileSelector(filesel_callback, 'Import PLY', Blender.sys.makename(ext='.ply'))