Welcome to mirror list, hosted at ThFree Co, Russian Federation.

git.blender.org/blender.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
path: root/release/scripts/op/io_mesh_ply/import_ply.py
diff options
context:
space:
mode:
Diffstat (limited to 'release/scripts/op/io_mesh_ply/import_ply.py')
-rw-r--r--release/scripts/op/io_mesh_ply/import_ply.py339
1 files changed, 0 insertions, 339 deletions
diff --git a/release/scripts/op/io_mesh_ply/import_ply.py b/release/scripts/op/io_mesh_ply/import_ply.py
deleted file mode 100644
index 6de90b1e86e..00000000000
--- a/release/scripts/op/io_mesh_ply/import_ply.py
+++ /dev/null
@@ -1,339 +0,0 @@
-# ##### BEGIN GPL LICENSE BLOCK #####
-#
-# 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
-#
-# ##### END GPL LICENSE BLOCK #####
-
-# <pep8 compliant>
-
-import re
-import struct
-
-
-class element_spec(object):
- __slots__ = ("name",
- "count",
- "properties",
- )
-
- def __init__(self, name, count):
- self.name = name
- self.count = count
- self.properties = []
-
- def load(self, format, stream):
- if format == 'ascii':
- stream = re.split('\s+', stream.readline())
- return [x.load(format, stream) for x in self.properties]
-
- 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",
- )
-
- 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):
- if format == 'ascii':
- if num_type == 's':
- ans = []
- for i in range(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 = [mapper(x) for x in stream[:count]]
- stream[:count] = []
- return ans
- else:
- if num_type == 's':
- ans = []
- for i in range(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)
-
- def load(self, format, stream):
- if self.list_type is not 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",
- )
- 'A list of element_specs'
- def __init__(self):
- self.specs = []
-
- def load(self, format, stream):
- return dict([(i.name, [i.load(format, stream) for j in range(i.count)]) for i in self.specs])
-
- '''
- # Longhand for above LC
- answer = {}
- for i in self.specs:
- answer[i.name] = []
- for j in range(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(filepath):
- 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',
- 'ushort': 'H',
- 'int': 'i',
- 'int32': 'i',
- 'uint': 'I',
- 'uint32': 'I',
- 'float': 'f',
- 'float32': 'f',
- 'float64': 'd',
- 'double': 'd',
- 'string': 's'}
- obj_spec = object_spec()
-
- try:
- file = open(filepath, '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(filepath, 'rb')
-
- # skip the header...
- while not file.readline().startswith('end_header'):
- pass
-
- obj = obj_spec.load(format_specs[format], file)
-
- except IOError:
- try:
- file.close()
- except:
- pass
-
- return None
- try:
- file.close()
- except:
- pass
-
- return obj_spec, obj
-
-
-import bpy
-
-
-def load_ply(filepath):
- import time
- from io_utils import load_image, unpack_list, unpack_face_list
-
- t = time.time()
- obj_spec, obj = read(filepath)
- if obj is None:
- print('Invalid file')
- return
-
- uvindices = colindices = None
- # noindices = None # Ignore normals
-
- 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
- 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
-
- # 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)
-
- 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 range(len_ind - 2):
- add_face(verts, (ind[0], ind[j + 1], ind[j + 2]), uvindices, colindices)
-
- ply_name = bpy.path.display_name_from_filepath(filepath)
-
- mesh = bpy.data.meshes.new(name=ply_name)
-
- mesh.vertices.add(len(obj['vertex']))
-
- mesh.vertices.foreach_set("co", [a for v in obj['vertex'] for a in (v[vindices_x], v[vindices_y], v[vindices_z])])
-
- if mesh_faces:
- mesh.faces.add(len(mesh_faces))
- mesh.faces.foreach_set("vertices_raw", unpack_face_list(mesh_faces))
-
- if uvindices or colindices:
- if uvindices:
- uvlay = mesh.uv_textures.new()
- if colindices:
- vcol_lay = mesh.vertex_colors.new()
-
- if uvindices:
- for i, f in enumerate(uvlay.data):
- ply_uv = mesh_uvs[i]
- for j, uv in enumerate(f.uv):
- uv[:] = ply_uv[j]
-
- if colindices:
- faces = obj['face']
- for i, f in enumerate(vcol_lay.data):
- # XXX, colors dont come in right, needs further investigation.
- ply_col = mesh_colors[i]
- if len(faces[i]) == 4:
- f_col = f.color1, f.color2, f.color3, f.color4
- else:
- f_col = f.color1, f.color2, f.color3
-
- for j, col in enumerate(f_col):
- col.r, col.g, col.b = ply_col[j]
-
- mesh.update()
-
- scn = bpy.context.scene
- #scn.objects.selected = [] # XXX25
-
- obj = bpy.data.objects.new(ply_name, mesh)
- scn.objects.link(obj)
- scn.objects.active = obj
- obj.select = True
-
- print('\nSuccessfully imported %r in %.3f sec' % (filepath, time.time() - t))
-
-
-def load(operator, context, filepath=""):
- load_ply(filepath)
- return {'FINISHED'}
generated by cgit v1.2.3 (git 2.25.1) at 2024-07-18 01:05:17 +0300