sandey/blender
1
0
Fork 0
forked from bartvdbraak/blender
Code Pull Requests Activity

pep8 cleanup.

Browse Source
This commit is contained in:
Campbell Barton 2011-01-13 23:31:04 +00:00
parent a1850ed876
commit de5d321e71
1 changed files with 301 additions and 327 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

628
release/scripts/op/io_mesh_ply/import_ply.py
View File

@ -1,354 +1,328 @@
#!BPY # ##### BEGIN GPL LICENSE BLOCK #####
"""
Name: 'Stanford PLY (*.ply)...'
Blender: 248
Group: 'Import'
Tip: 'Import a Stanford PLY file'
"""
__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 # This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License # modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2 # as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version. # of the License, or (at your option) any later version.
# #
# This program is distributed in the hope that it will be useful, # This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of # but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details. # GNU General Public License for more details.
# #
# You should have received a copy of the GNU General Public License # You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation, # along with this program; if not, write to the Free Software Foundation,
# Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
# ##### END GPL LICENSE BLOCK #####
# <pep8 compliant>
# 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.
# Portions of this code are taken from mod_meshtools.py in Blender
# 2.32.
import Blender import Blender
try: try:
import re, struct import re
import struct
except: except:
struct= None struct = None
class element_spec(object): class element_spec(object):
__slots__ = 'name', 'count', 'properties' __slots__ = 'name', 'count', 'properties'
def __init__(self, name, count):
self.name = name
self.count = count
self.properties = []
def load(self, format, stream): def __init__(self, name, count):
if format == 'ascii': self.name = name
stream = re.split('\s+', stream.readline()) self.count = count
return map(lambda x: x.load(format, stream), self.properties) self.properties = []
def load(self, format, stream):
if format == 'ascii':
stream = re.split('\s+', stream.readline())
return map(lambda x: x.load(format, stream), self.properties)
def index(self, name):
for i, p in enumerate(self.properties):
if p.name == name:
return i
return -1
def index(self, name):
for i, p in enumerate(self.properties):
if p.name == name: return i
return -1
class property_spec(object): class property_spec(object):
__slots__ = 'name', 'list_type', 'numeric_type' __slots__ = 'name', 'list_type', 'numeric_type'
def __init__(self, name, list_type, numeric_type):
self.name = name
self.list_type = list_type
self.numeric_type = numeric_type
def read_format(self, format, count, num_type, stream): def __init__(self, name, list_type, numeric_type):
if format == 'ascii': self.name = name
if (num_type == 's'): self.list_type = list_type
ans = [] self.numeric_type = numeric_type
for i in xrange(count):
s = stream[i] def read_format(self, format, count, num_type, stream):
if len(s) < 2 or s[0] != '"' or s[-1] != '"': if format == 'ascii':
print 'Invalid string', s if (num_type == 's'):
print 'Note: ply_import.py does not handle whitespace in strings' ans = []
return None for i in xrange(count):
ans.append(s[1:-1]) s = stream[i]
stream[:count] = [] if len(s) < 2 or s[0] != '"' or s[-1] != '"':
return ans print 'Invalid string', s
if (num_type == 'f' or num_type == 'd'): print 'Note: ply_import.py does not handle whitespace in strings'
mapper = float return None
else: ans.append(s[1:-1])
mapper = int stream[:count] = []
ans = map(lambda x: mapper(x), stream[:count]) return ans
stream[:count] = [] if (num_type == 'f' or num_type == 'd'):
return ans mapper = float
else: else:
if (num_type == 's'): mapper = int
ans = [] ans = map(lambda x: mapper(x), stream[:count])
for i in xrange(count): stream[:count] = []
fmt = format + 'i' return ans
data = stream.read(struct.calcsize(fmt)) else:
length = struct.unpack(fmt, data)[0] if (num_type == 's'):
fmt = '%s%is' % (format, length) ans = []
data = stream.read(struct.calcsize(fmt)) for i in xrange(count):
s = struct.unpack(fmt, data)[0] fmt = format + 'i'
ans.append(s[:-1]) # strip the NULL data = stream.read(struct.calcsize(fmt))
return ans length = struct.unpack(fmt, data)[0]
else: fmt = '%s%is' % (format, length)
fmt = '%s%i%s' % (format, count, num_type) data = stream.read(struct.calcsize(fmt))
data = stream.read(struct.calcsize(fmt)); s = struct.unpack(fmt, data)[0]
return struct.unpack(fmt, data) 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)
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]
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): class object_spec(object):
__slots__ = 'specs' __slots__ = 'specs'
'A list of element_specs' 'A list of element_specs'
def __init__(self): def __init__(self):
self.specs = [] self.specs = []
def load(self, format, stream): def load(self, format, stream):
return dict([(i.name,[i.load(format, stream) for j in xrange(i.count) ]) for i in self.specs]) return dict([(i.name, [i.load(format, stream) for j in xrange(i.count)]) for i in self.specs])
''' '''
# Longhand for above LC # Longhand for above LC
answer = {} answer = {}
for i in self.specs: for i in self.specs:
answer[i.name] = [] answer[i.name] = []
for j in xrange(i.count): for j in xrange(i.count):
if not j % 100 and meshtools.show_progress: if not j % 100 and meshtools.show_progress:
Blender.Window.DrawProgressBar(float(j) / i.count, 'Loading ' + i.name) Blender.Window.DrawProgressBar(float(j) / i.count, 'Loading ' + i.name)
answer[i.name].append(i.load(format, stream)) answer[i.name].append(i.load(format, stream))
return answer return answer
''' '''
def read(filename): def read(filename):
format = '' format = ''
version = '1.0' version = '1.0'
format_specs = {'binary_little_endian': '<', format_specs = {'binary_little_endian': '<',
'binary_big_endian': '>', 'binary_big_endian': '>',
'ascii': 'ascii'} 'ascii': 'ascii'}
type_specs = {'char': 'b', type_specs = {'char': 'b',
'uchar': 'B', 'uchar': 'B',
'int8': 'b', 'int8': 'b',
'uint8': 'B', 'uint8': 'B',
'int16': 'h', 'int16': 'h',
'uint16': 'H', 'uint16': 'H',
'ushort': 'H', 'ushort': 'H',
'int': 'i', 'int': 'i',
'int32': 'i', 'int32': 'i',
'uint': 'I', 'uint': 'I',
'uint32': 'I', 'uint32': 'I',
'float': 'f', 'float': 'f',
'float32': 'f', 'float32': 'f',
'float64': 'd', 'float64': 'd',
'double': 'd', 'double': 'd',
'string': 's'} 'string': 's'}
obj_spec = object_spec() obj_spec = object_spec()
try:
file = open(filename, 'rU') # Only for parsing the header, not binary data
signature = file.readline()
if not signature.startswith('ply'):
print 'Signature line was invalid'
return None
while 1:
tokens = re.split(r'[ \n]+', file.readline())
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
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
obj = obj_spec.load(format_specs[format], file)
except IOError, (errno, strerror):
try:
file.close()
except:
pass
return None
try:
file.close()
except:
pass
return obj_spec, obj
try:
file = open(filename, 'rU') # Only for parsing the header, not binary data
signature = file.readline()
if not signature.startswith('ply'):
print 'Signature line was invalid'
return None
while 1:
tokens = re.split(r'[ \n]+', file.readline())
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
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
obj = obj_spec.load(format_specs[format], file)
except IOError, (errno, strerror):
try: file.close()
except: pass
return None
try: file.close()
except: pass
return (obj_spec, obj);
def load_ply(filename): def load_ply(filename):
t = Blender.sys.time() t = Blender.sys.time()
obj_spec, obj = read(filename) obj_spec, obj = read(filename)
if obj == None: if obj == None:
print 'Invalid file' print 'Invalid file'
return return
uvindices = colindices = None uvindices = colindices = None
# noindices = None # Ignore normals # noindices = None # Ignore normals
for el in obj_spec.specs: for el in obj_spec.specs:
if el.name == 'vertex': if el.name == 'vertex':
vindices = vindices_x, vindices_y, vindices_z = (el.index('x'), el.index('y'), el.index('z')) 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')) # noindices = (el.index('nx'), el.index('ny'), el.index('nz'))
# if -1 in noindices: noindices = None # if -1 in noindices: noindices = None
uvindices = (el.index('s'), el.index('t')) uvindices = (el.index('s'), el.index('t'))
if -1 in uvindices: uvindices = None if -1 in uvindices:
colindices = (el.index('red'), el.index('green'), el.index('blue')) uvindices = None
if -1 in colindices: colindices = None colindices = (el.index('red'), el.index('green'), el.index('blue'))
elif el.name == 'face': if -1 in colindices:
findex = el.index('vertex_indices') colindices = None
elif el.name == 'face':
mesh_faces = [] findex = el.index('vertex_indices')
mesh_uvs = []
mesh_colors = [] mesh_faces = []
mesh_uvs = []
def add_face(vertices, indices, uvindices, colindices): mesh_colors = []
mesh_faces.append(indices)
if uvindices: mesh_uvs.append([ (vertices[index][uvindices[0]], 1.0 - vertices[index][uvindices[1]]) for index in indices]) def add_face(vertices, indices, uvindices, colindices):
if colindices: mesh_colors.append([ (vertices[index][colindices[0]], vertices[index][colindices[1]], vertices[index][colindices[2]]) for index in indices]) mesh_faces.append(indices)
if uvindices:
mesh_uvs.append([(vertices[index][uvindices[0]], 1.0 - vertices[index][uvindices[1]]) for index in indices])
if uvindices or colindices: if colindices:
# If we have Cols or UVs then we need to check the face order. mesh_colors.append([(vertices[index][colindices[0]], vertices[index][colindices[1]], vertices[index][colindices[2]]) for index in indices])
add_face_simple = add_face
if uvindices or colindices:
# EVIL EEKADOODLE - face order annoyance. # If we have Cols or UVs then we need to check the face order.
def add_face(vertices, indices, uvindices, colindices): add_face_simple = add_face
if len(indices)==4:
if indices[2]==0 or indices[3]==0: # EVIL EEKADOODLE - face order annoyance.
indices= indices[2], indices[3], indices[0], indices[1] def add_face(vertices, indices, uvindices, colindices):
elif len(indices)==3: if len(indices) == 4:
if indices[2]==0: if indices[2] == 0 or indices[3] == 0:
indices= indices[1], indices[2], indices[0] indices = indices[2], indices[3], indices[0], indices[1]
elif len(indices) == 3:
add_face_simple(vertices, indices, uvindices, colindices) if indices[2] == 0:
indices = indices[1], indices[2], indices[0]
verts = obj['vertex']
add_face_simple(vertices, indices, uvindices, colindices)
if 'face' in obj:
for f in obj['face']: verts = obj['vertex']
ind = f[findex]
len_ind = len(ind) if 'face' in obj:
if len_ind <= 4: for f in obj['face']:
add_face(verts, ind, uvindices, colindices) ind = f[findex]
else: len_ind = len(ind)
# Fan fill the face if len_ind <= 4:
for j in xrange(len_ind - 2): add_face(verts, ind, uvindices, colindices)
add_face(verts, (ind[0], ind[j + 1], ind[j + 2]), uvindices, colindices) else:
# Fan fill the face
mesh = Blender.Mesh.New() for j in xrange(len_ind - 2):
add_face(verts, (ind[0], ind[j + 1], ind[j + 2]), uvindices, colindices)
mesh.verts.extend([(v[vindices_x], v[vindices_y], v[vindices_z]) for v in obj['vertex']])
mesh = Blender.Mesh.New()
if mesh_faces:
mesh.faces.extend(mesh_faces, smooth=True, ignoreDups=True) mesh.verts.extend([(v[vindices_x], v[vindices_y], v[vindices_z]) for v in obj['vertex']])
if uvindices or colindices: if mesh_faces:
if uvindices: mesh.faceUV = True mesh.faces.extend(mesh_faces, smooth=True, ignoreDups=True)
if colindices: mesh.vertexColors = True
if uvindices or colindices:
for i, f in enumerate(mesh.faces): if uvindices:
if uvindices: mesh.faceUV = True
ply_uv = mesh_uvs[i] if colindices:
for j, uv in enumerate(f.uv): mesh.vertexColors = True
uv[:] = ply_uv[j]
for i, f in enumerate(mesh.faces):
if colindices: if uvindices:
ply_col = mesh_colors[i] ply_uv = mesh_uvs[i]
for j, col in enumerate(f.col): for j, uv in enumerate(f.uv):
col.r, col.g, col.b = ply_col[j] uv[:] = ply_uv[j]
mesh.calcNormals() if colindices:
ply_col = mesh_colors[i]
for j, col in enumerate(f.col):
objname = Blender.sys.splitext(Blender.sys.basename(filename))[0] col.r, col.g, col.b = ply_col[j]
scn= Blender.Scene.GetCurrent()
scn.objects.selected = [] mesh.calcNormals()
mesh.name= objname objname = Blender.sys.splitext(Blender.sys.basename(filename))[0]
scn.objects.active = scn.objects.new(mesh) scn = Blender.Scene.GetCurrent()
scn.objects.selected = []
Blender.Redraw()
Blender.Window.DrawProgressBar(1.0, '') mesh.name = objname
print '\nSuccessfully imported "%s" in %.3f sec' % (filename, Blender.sys.time()-t) scn.objects.active = scn.objects.new(mesh)
Blender.Redraw()
Blender.Window.DrawProgressBar(1.0, '')
print '\nSuccessfully imported "%s" in %.3f sec' % (filename, Blender.sys.time() - t)
def main(): def main():
if not struct: if not struct:
msg = 'This importer requires a full python install' msg = 'This importer requires a full python install'
if Blender.mode == 'background': print msg if Blender.mode == 'background':
else: Blender.Draw.PupMenu(msg) print msg
return else:
Blender.Draw.PupMenu(msg)
Blender.Window.FileSelector(load_ply, 'Import PLY', '*.ply') return
if __name__=='__main__': Blender.Window.FileSelector(load_ply, 'Import PLY', '*.ply')
main()
''' if __name__ == '__main__':
import bpy main()
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)
'''