blender/release/scripts/ply_import.py

352 lines
9.6 KiB
Python
Raw Normal View History

#!BPY
"""
Name: 'Stanford PLY (*.ply)...'
Blender: 248
Group: 'Import'
Tip: 'Import a Stanford PLY file'
"""
2006-01-26 15:39:38 +00:00
__author__ = 'Bruce Merry'
__version__ = '0.93'
__bpydoc__ = """\
This script imports Stanford PLY files into Blender. It supports per-vertex
normals, and per-face colours and texture coordinates.
Usage:
Run this script from "File->Import" and select the desired PLY file.
"""
# Copyright (C) 2004, 2005: Bruce Merry, bmerry@cs.uct.ac.za
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
# 20th Oct 2008, 0.93 - Updated by Campbell Barton AKA ideasman42, use Mesh rather then NMesh, dont import normals, vcolors work again.
# Updated by Campbell Barton AKA Ideasman42, 10% faster code.
2006-01-26 15:39:38 +00:00
# Portions of this code are taken from mod_meshtools.py in Blender
# 2.32.
import Blender
try:
import re, struct
except:
struct= None
class element_spec(object):
__slots__ = 'name', 'count', 'properties'
2006-01-26 15:39:38 +00:00
def __init__(self, name, count):
self.name = name
self.count = count
self.properties = []
2006-01-26 15:39:38 +00:00
def load(self, format, stream):
if format == 'ascii':
stream = re.split('\s+', stream.readline())
return map(lambda x: x.load(format, stream), self.properties)
2006-01-26 15:39:38 +00:00
def index(self, name):
for i, p in enumerate(self.properties):
if p.name == name: return i
return -1
class property_spec(object):
__slots__ = 'name', 'list_type', 'numeric_type'
2006-01-26 15:39:38 +00:00
def __init__(self, name, list_type, numeric_type):
self.name = name
self.list_type = list_type
self.numeric_type = numeric_type
2006-01-26 15:39:38 +00:00
def read_format(self, format, count, num_type, stream):
if format == 'ascii':
if (num_type == 's'):
ans = []
for i in xrange(count):
s = stream[i]
if len(s) < 2 or s[0] != '"' or s[-1] != '"':
print 'Invalid string', s
print 'Note: ply_import.py does not handle whitespace in strings'
return None
ans.append(s[1:-1])
stream[:count] = []
return ans
if (num_type == 'f' or num_type == 'd'):
mapper = float
else:
mapper = int
ans = map(lambda x: mapper(x), stream[:count])
stream[:count] = []
return ans
else:
if (num_type == 's'):
ans = []
for i in xrange(count):
fmt = format + 'i'
data = stream.read(struct.calcsize(fmt))
length = struct.unpack(fmt, data)[0]
fmt = '%s%is' % (format, length)
data = stream.read(struct.calcsize(fmt))
s = struct.unpack(fmt, data)[0]
ans.append(s[:-1]) # strip the NULL
return ans
else:
fmt = '%s%i%s' % (format, count, num_type)
data = stream.read(struct.calcsize(fmt));
return struct.unpack(fmt, data)
2006-01-26 15:39:38 +00:00
def load(self, format, stream):
if (self.list_type != None):
count = int(self.read_format(format, 1, self.list_type, stream)[0])
return self.read_format(format, count, self.numeric_type, stream)
else:
return self.read_format(format, 1, self.numeric_type, stream)[0]
class object_spec(object):
__slots__ = 'specs'
2006-01-26 15:39:38 +00:00
'A list of element_specs'
def __init__(self):
self.specs = []
2006-01-26 15:39:38 +00:00
def load(self, format, stream):
return dict([(i.name,[i.load(format, stream) for j in xrange(i.count) ]) for i in self.specs])
'''
# Longhand for above LC
2006-01-26 15:39:38 +00:00
answer = {}
for i in self.specs:
answer[i.name] = []
for j in xrange(i.count):
if not j % 100 and meshtools.show_progress:
Blender.Window.DrawProgressBar(float(j) / i.count, 'Loading ' + i.name)
answer[i.name].append(i.load(format, stream))
return answer
'''
def read(filename):
2006-01-26 15:39:38 +00:00
format = ''
version = '1.0'
format_specs = {'binary_little_endian': '<',
'binary_big_endian': '>',
'ascii': 'ascii'}
type_specs = {'char': 'b',
'uchar': 'B',
'int8': 'b',
'uint8': 'B',
'int16': 'h',
'uint16': 'H',
'int': 'i',
'int32': 'i',
'uint': 'I',
'uint32': 'I',
'float': 'f',
'float32': 'f',
'float64': 'd',
'string': 's'}
obj_spec = object_spec()
2006-01-26 15:39:38 +00:00
try:
file = open(filename, 'rU') # Only for parsing the header, not binary data
2006-01-26 15:39:38 +00:00
signature = file.readline()
if not signature.startswith('ply'):
2006-01-26 15:39:38 +00:00
print 'Signature line was invalid'
return None
2006-01-26 15:39:38 +00:00
while 1:
tokens = re.split(r'[ \n]+', file.readline())
2006-01-26 15:39:38 +00:00
if (len(tokens) == 0):
continue
if (tokens[0] == 'end_header'):
break
elif (tokens[0] == 'comment' or tokens[0] == 'obj_info'):
continue
elif (tokens[0] == 'format'):
if (len(tokens) < 3):
print 'Invalid format line'
return None
if (tokens[1] not in format_specs): # .keys()): # keys is implicit
2006-01-26 15:39:38 +00:00
print 'Unknown format', tokens[1]
return None
if (tokens[2] != version):
print 'Unknown version', tokens[2]
return None
format = tokens[1]
elif (tokens[0] == 'element'):
if (len(tokens) < 3):
print 'Invalid element line'
return None
obj_spec.specs.append(element_spec(tokens[1], int(tokens[2])))
elif (tokens[0] == 'property'):
if (not len(obj_spec.specs)):
print 'Property without element'
return None
if (tokens[1] == 'list'):
obj_spec.specs[-1].properties.append(property_spec(tokens[4], type_specs[tokens[2]], type_specs[tokens[3]]))
else:
obj_spec.specs[-1].properties.append(property_spec(tokens[2], None, type_specs[tokens[1]]))
if format != 'ascii':
file.close() # was ascii, now binary
file = open(filename, 'rb')
# skip the header...
while not file.readline().startswith('end_header'):
pass
2006-01-26 15:39:38 +00:00
obj = obj_spec.load(format_specs[format], file)
2006-01-26 15:39:38 +00:00
except IOError, (errno, strerror):
try: file.close()
except: pass
2006-01-26 15:39:38 +00:00
return None
try: file.close()
except: pass
2006-01-26 15:39:38 +00:00
return (obj_spec, obj);
def load_ply(filename):
2006-01-26 15:39:38 +00:00
t = Blender.sys.time()
obj_spec, obj = read(filename)
2006-01-26 15:39:38 +00:00
if obj == None:
print 'Invalid file'
return
uvindices = colindices = None
# noindices = None # Ignore normals
2006-01-26 15:39:38 +00:00
for el in obj_spec.specs:
if el.name == 'vertex':
vindices = vindices_x, vindices_y, vindices_z = (el.index('x'), el.index('y'), el.index('z'))
# noindices = (el.index('nx'), el.index('ny'), el.index('nz'))
# if -1 in noindices: noindices = None
uvindices = (el.index('s'), el.index('t'))
if -1 in uvindices: uvindices = None
colindices = (el.index('red'), el.index('green'), el.index('blue'))
if -1 in colindices: colindices = None
2006-01-26 15:39:38 +00:00
elif el.name == 'face':
findex = el.index('vertex_indices')
mesh_faces = []
mesh_uvs = []
mesh_colors = []
def add_face(vertices, indices, uvindices, colindices):
mesh_faces.append(indices)
if uvindices: mesh_uvs.append([ (vertices[index][uvindices[0]], 1.0 - vertices[index][uvindices[1]]) for index in indices])
if colindices: mesh_colors.append([ (vertices[index][colindices[0]], vertices[index][colindices[1]], vertices[index][colindices[2]]) for index in indices])
if uvindices or colindices:
# If we have Cols or UVs then we need to check the face order.
add_face_simple = add_face
2006-01-26 15:39:38 +00:00
# EVIL EEKADOODLE - face order annoyance.
def add_face(vertices, indices, uvindices, colindices):
if len(indices)==4:
if indices[2]==0 or indices[3]==0:
indices= indices[2], indices[3], indices[0], indices[1]
elif len(indices)==3:
if indices[2]==0:
indices= indices[1], indices[2], indices[0]
add_face_simple(vertices, indices, uvindices, colindices)
2006-01-26 15:39:38 +00:00
verts = obj['vertex']
if 'face' in obj:
for f in obj['face']:
ind = f[findex]
len_ind = len(ind)
if len_ind <= 4:
add_face(verts, ind, uvindices, colindices)
else:
# Fan fill the face
for j in xrange(len_ind - 2):
add_face(verts, (ind[0], ind[j + 1], ind[j + 2]), uvindices, colindices)
mesh = Blender.Mesh.New()
mesh.verts.extend([(v[vindices_x], v[vindices_y], v[vindices_z]) for v in obj['vertex']])
if mesh_faces:
mesh.faces.extend(mesh_faces, smooth=True, ignoreDups=True)
if uvindices or colindices:
if uvindices: mesh.faceUV = True
if colindices: mesh.vertexColors = True
for i, f in enumerate(mesh.faces):
if uvindices:
ply_uv = mesh_uvs[i]
for j, uv in enumerate(f.uv):
uv[:] = ply_uv[j]
if colindices:
ply_col = mesh_colors[i]
for j, col in enumerate(f.col):
col.r, col.g, col.b = ply_col[j]
2006-01-26 15:39:38 +00:00
mesh.calcNormals()
2006-01-26 15:39:38 +00:00
objname = Blender.sys.splitext(Blender.sys.basename(filename))[0]
scn= Blender.Scene.GetCurrent()
scn.objects.selected = []
mesh.name= objname
scn.objects.active = scn.objects.new(mesh)
2006-01-26 15:39:38 +00:00
Blender.Redraw()
Blender.Window.DrawProgressBar(1.0, '')
print '\nSuccessfully imported "%s" in %.3f sec' % (filename, Blender.sys.time()-t)
def main():
if not struct:
msg = 'This importer requires a full python install'
if Blender.mode == 'background': print msg
else: Blender.Draw.PupMenu(msg)
return
Blender.Window.FileSelector(load_ply, 'Import PLY', '*.ply')
if __name__=='__main__':
main()
'''
import bpy
import os
files = os.popen('find /fe/ply -iname "*.ply"').readlines()
files.sort()
tot = len(files)
for i, f in enumerate(files):
if i < 26 or i > 1000000:
continue
#if i != 12686:
# continue
f = f.strip()
print f, i, tot
sce = bpy.data.scenes.new(f.split('/')[-1])
bpy.data.scenes.active = sce
# Window.
load_ply(f)
'''