def load(op, ctx, filepath=None, **kwargs):
c = Collada(filepath, ignore=[DaeBrokenRefError])
impclass = get_import(c)
imp = impclass(ctx, c, os.path.dirname(filepath), **kwargs)
tf = kwargs['transformation']
print('Begin Load')
with prevented_updates(ctx):
if tf in ('MUL', 'APPLY'):
for i, obj in enumerate(c.scene.objects('geometry')):
b_geoms = imp.geometry(obj)
if tf == 'MUL':
tf_mat = Matrix(obj.matrix)
for b_obj in b_geoms:
b_obj.matrix_world = tf_mat
elif tf == 'PARENT':
_dfs(c.scene, imp.node)
for i, obj in enumerate(c.scene.objects('light')):
imp.light(obj, i)
for obj in c.scene.objects('camera'):
imp.camera(obj)
print('Finish Load')
return {'FINISHED'}
def read_collada_surfaces(session,
path,
name=None,
color=(200, 200, 200, 255),
**kw):
'''Open a collada file.'''
from collada import Collada
c = Collada(path)
if name is None:
from os.path import basename
name = basename(path)
from chimerax.geometry import Place
splist = surfaces_from_nodes(c.scene.nodes, color, Place(), {}, session)
if len(splist) > 1:
from chimerax.core.models import Model
s = Model(name, session)
s.add(splist)
elif len(splist) == 1:
s = splist[0]
s.name = name
else:
from chimerax.core.errors import UserError
raise UserError('Collada file has no TriangleSets: %s' % name)
set_instance_positions_and_colors(s.all_drawings())
ai = c.assetInfo
if ai:
s.collada_unit_name = ai.unitname
s.collada_contributors = ai.contributors
return [s], ('Opened collada file %s' % name)
def display_collada(dae_file):
"""
Display the DAE mesh. Requires pycollada.
"""
print("Displaying %s" % dae_file)
from collada import Collada, DaeUnsupportedError, DaeBrokenRefError
mesh = Collada(dae_file, ignore=[DaeUnsupportedError, DaeBrokenRefError])
for geometry in mesh.scene.objects('geometry'):
for prim in geometry.primitives():
# use primitive-specific ways to get triangles
prim_type = type(prim).__name__
if prim_type == 'BoundTriangleSet':
triangles = prim
elif prim_type == 'BoundPolylist':
triangles = prim.triangleset()
else:
# Unsupported mesh type
triangles = None
if triangles is not None:
x = triangles.vertex[:, 0]
y = triangles.vertex[:, 1]
z = triangles.vertex[:, 2]
mlab.triangular_mesh(x,
y,
z,
triangles.vertex_index,
color=(0, 0, 1))
def mesh_to_collada(mesh):
'''
Supports per-vertex color, but nothing else.
'''
import numpy as np
try:
from collada import Collada, scene
except ImportError:
raise ImportError(
"lace.serialization.dae.mesh_to_collade requires package pycollada."
)
def create_material(dae):
from collada import material, scene
effect = material.Effect("effect0", [],
"phong",
diffuse=(1, 1, 1),
specular=(0, 0, 0),
double_sided=True)
mat = material.Material("material0", "mymaterial", effect)
dae.effects.append(effect)
dae.materials.append(mat)
return scene.MaterialNode("materialref", mat, inputs=[])
def geometry_from_mesh(dae, mesh):
from collada import source, geometry
srcs = []
# v
srcs.append(source.FloatSource("verts-array", mesh.v, ('X', 'Y', 'Z')))
input_list = source.InputList()
input_list.addInput(0, 'VERTEX', "#verts-array")
# vc
if mesh.vc is not None:
input_list.addInput(len(srcs), 'COLOR', "#color-array")
srcs.append(
source.FloatSource("color-array", mesh.vc[mesh.f.ravel()],
('X', 'Y', 'Z')))
# f
geom = geometry.Geometry(str(mesh), "geometry0", "mymesh", srcs)
indices = np.dstack([mesh.f for _ in srcs]).ravel()
triset = geom.createTriangleSet(indices, input_list, "materialref")
geom.primitives.append(triset)
# e
if mesh.e is not None:
indices = np.dstack([mesh.e for _ in srcs]).ravel()
lineset = geom.createLineSet(indices, input_list, "materialref")
geom.primitives.append(lineset)
dae.geometries.append(geom)
return geom
dae = Collada()
geom = geometry_from_mesh(dae, mesh)
node = scene.Node(
"node0", children=[scene.GeometryNode(geom, [create_material(dae)])])
myscene = scene.Scene("myscene", [node])
dae.scenes.append(myscene)
dae.scene = myscene
return dae
def main():
"""
Generate .dae file.
"""
mesh = Collada()
effect = material.Effect('effect0', [],
'phong',
diffuse=(1, 0, 0),
specular=(0, 1, 0))
mat = material.Material('material0', 'mymaterial', effect)
mesh.effects.append(effect)
mesh.materials.append(mat)
vert_floats = numpy.array([
-50, 50, 50, 50, 50, 50, -50, -50, 50, 50, -50, 50, -50, 50, -50, 50,
50, -50, -50, -50, -50, 50, -50, -50
])
normal_floats = numpy.array([
0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0,
0, -1, 0, 0, -1, 0, 0, -1, 0, 0, -1, 0, -1, 0, 0, -1, 0, 0, -1, 0, 0,
-1, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0, -1, 0, 0, -1, 0, 0,
-1, 0, 0, -1
])
componests = ('X', 'Y', 'Z')
vert_src = source.FloatSource('cubeverts-array', vert_floats, componests)
normal_src = source.FloatSource('cubenormals-array', normal_floats,
componests)
input_list = source.InputList()
input_list.addInput(0, 'VERTEX', '#cubeverts-array')
input_list.addInput(1, 'NORMAL', '#cubenormals-array')
indices = numpy.array([
0, 0, 2, 1, 3, 2, 0, 0, 3, 2, 1, 3, 0, 4, 1, 5, 5, 6, 0, 4, 5, 6, 4, 7,
6, 8, 7, 9, 3, 10, 6, 8, 3, 10, 2, 11, 0, 12, 4, 13, 6, 14, 0, 12, 6,
14, 2, 15, 3, 16, 7, 17, 5, 18, 3, 16, 5, 18, 1, 19, 5, 20, 7, 21, 6,
22, 5, 20, 6, 22, 4, 23
])
geom = geometry.Geometry(mesh, 'geometry0', 'mycube',
[vert_src, normal_src])
triset = geom.createTriangleSet(indices, input_list, 'materialref')
geom.primitives.append(triset)
mesh.geometries.append(geom)
matnode = scene.MaterialNode('materialref', mat, inputs=[])
geomnode = scene.GeometryNode(geom, [matnode])
node = scene.Node('node0', children=[geomnode])
myscene = scene.Scene('myscene', [node])
mesh.scenes.append(myscene)
mesh.scene = myscene
mesh.write('collada_sample.dae')
def load(file):
mesh = Collada(file)
index_count = 0
all_indices = []
all_vertices = []
for geometry in mesh.geometries:
for primitive in geometry.primitives:
indices = []
for tri in primitive.indices:
indices.extend(tri[:, 0] + index_count)
index_count += len(primitive.vertex)
vertices = primitive.vertex
all_vertices.extend(vertices)
all_indices.extend(indices)
return np.array(all_vertices).reshape(len(all_vertices),
3), np.array(all_indices)
def _import_dae(self, file):
"""\
Import a collada object.
"""
solid = Collada(file)
self._triangles = []
for geometry in solid.geometries:
for triangle_set in geometry.primitives:
for triangle in triangle_set:
v = triangle.vertices
try:
points = [Point(v[0][0], v[0][1], v[0][2]), \
Point(v[1][0], v[1][1], v[1][2]), \
Point(v[2][0], v[2][1], v[2][2])]
self._triangles.append(Triangle(*points))
except IndexError:
pass
def build_scene(env, outfilename):
#architecture is :
#-env.name
#--exterior
#----ingredients
#--compartments
#----surface
#------ingredients
#----interior
#------ingredients
#create the document and a node for rootenv
collada_xml = Collada()
collada_xml.assetInfo.unitname = "centimeter"
collada_xml.assetInfo.unitmeter = 0.01
collada_xml.assetInfo.upaxis = "Y_UP"
root_env = scene.Node(env.name)
myscene = scene.Scene(env.name + "_Scene", [root_env])
collada_xml.scenes.append(myscene)
collada_xml.scene = myscene
#
bbsurface = None
r = env.exteriorRecipe
if r: collada_xml, root_env = buildRecipe(r, r.name, collada_xml, root_env)
for o in env.compartments:
rs = o.surfaceRecipe
if rs:
#p,s,bb=up (767.0) #used for lipids
#pp,ss,bbsurface = up (700.0)
collada_xml, root_env = buildRecipe(rs,
str(o.name) + "_surface",
collada_xml,
root_env,
mask=bbsurface)
ri = o.innerRecipe
if ri:
collada_xml, root_env = buildRecipe(ri,
str(o.name) + "_interior",
collada_xml,
root_env,
mask=bbsurface)
collada_xml, root_env = buildCompartmentsGeom(env.compartments[0],
collada_xml, root_env)
collada_xml.write(outfilename)
return collada_xml
def load(op, ctx, filepath=None, **kwargs):
c = Collada(filepath, ignore=[DaeBrokenRefError])
impclass = get_import(c)
imp = impclass(ctx, c, os.path.dirname(filepath), **kwargs)
tf = kwargs['transformation']
allvertices = np.vstack([
i.vertex for j in c.scene.objects('geometry') for i in j.primitives()
if isinstance(i, BoundTriangleSet)
])
scale = np.max(allvertices.max(axis=0) - allvertices.min(axis=0)) / 2
mean = (allvertices.min(axis=0) + allvertices.max(axis=0)) * 0.5
for i, obj in enumerate(c.scene.objects('geometry')):
b_geoms = imp.geometry(obj, mean, scale)
return {'FINISHED'}
def read_collada_surfaces(path, session, color=(178, 178, 178, 255)):
from os.path import basename
from ..models import Model
s = Model(basename(path))
from collada import Collada
c = Collada(path)
from ..geometry.place import Place
splist = surface_pieces_from_nodes(c.scene.nodes, s, color, Place(), {})
set_positions_and_colors(splist)
ai = c.assetInfo
if ai:
s.collada_unit_name = ai.unitname
s.collada_contributors = ai.contributors
return s
def __init__(self, collada_file_path):
model = Collada(collada_file_path)
self.vao = []
self.inverse_transform_matrices = [
value for _, value in model.controllers[0].joint_matrices.items()
]
self.joints_order = {
"Armature_" + joint_name: index
for index, joint_name in enumerate(
np.squeeze(model.controllers[0].weight_joints.data))
}
self.joint_count = len(self.joints_order)
for node in model.scenes[0].nodes:
if node.id == 'Armature':
self.root_joint = Joint(
node.children[0].id,
self.inverse_transform_matrices[self.joints_order.get(
node.children[0].id)])
self.root_joint.children.extend(
self.__load_armature(node.children[0]))
del self.inverse_transform_matrices
if node.id == "Cube":
self.__load_mesh_data(node.children[0])
self.render_static_matrices = [
np.identity(4) for _ in range(len(self.joints_order))
]
self.render_animation_matrices = [
i for i in range(len(self.joints_order))
]
anims = et.parse(collada_file_path).getroot().find(
"{http://www.collada.org/2005/11/COLLADASchema}library_animations"
).findall("{http://www.collada.org/2005/11/COLLADASchema}animation")
self.__load_keyframes(anims)
self.prev_iter = 0
def __init__(self, collada_file_path):
model = Collada(collada_file_path)
self.vao = []
self.inverse_transform_matrices = [
value for _, value in model.controllers[0].joint_matrices.items()
]
self.joints_order = {
"Armature_" + joint_name: index
for index, joint_name in enumerate(
np.squeeze(model.controllers[0].weight_joints.data))
}
self.joint_count = len(self.joints_order)
for node in model.scenes[0].nodes:
if node.id == 'Armature':
self.root_joint = Joint(
node.children[0].id,
self.inverse_transform_matrices[self.joints_order.get(
node.children[0].id)])
self.root_joint.children.extend(
self.__load_armature(node.children[0]))
del self.inverse_transform_matrices
if node.id == "Cube":
self.__load_mesh_data(node.children[0])
self.render_static_matrices = [
np.identity(4) for _ in range(len(self.joints_order))
]
self.render_animation_matrices = [
i for i in range(len(self.joints_order))
]
self.__load_keyframes(model.animations)
self.doing_animation = False
self.frame_start_time = None
self.animation_keyframe_pointer = 0
def export_dae(vertices, indices, fname):
mesh = Collada()
effect = material.Effect("effect0", [], "phong", diffuse=(1, 0, 0), specular=(0, 1, 0))
mat = material.Material("material0", "mymaterial", effect)
mesh.effects.append(effect)
mesh.materials.append(mat)
vert_src = source.FloatSource('verts-array', numpy.array(vertices), ('X', 'Y', 'Z'))
inlist = source.InputList()
inlist.addInput(0, 'VERTEX', '#verts-array')
indices = numpy.array(indices)
geom = geometry.Geometry(mesh, 'geometry0', 'linac', [vert_src])
triset = geom.createTriangleSet(indices, inlist, "materialref")
geom.primitives.append(triset)
mesh.geometries.append(geom)
matnode = scene.MaterialNode("materialref", mat, inputs=[])
geomnode = scene.GeometryNode(geom, [matnode])
node = scene.Node("node0", children=[geomnode])
myscene = scene.Scene("myscene", [node])
mesh.scenes.append(myscene)
mesh.scene = myscene
mesh.write(fname)
def _save_dae(self, name):
"""\
Save the model as a collada file.
"""
mesh = Collada()
effect = material.Effect("effect0", [], "phong", diffuse=(0.9,0.9,0.9), \
specular=(0.1,0.1,0.1))
mat = material.Material("material0", "mymaterial", effect)
mesh.effects.append(effect)
mesh.materials.append(mat)
vert_floats = []
norm_floats = []
for vertex in self.vertices:
vert_floats.append(vertex.x)
vert_floats.append(vertex.y)
vert_floats.append(vertex.z)
for normal in self.normals:
norm_floats.append(normal.x)
norm_floats.append(normal.y)
norm_floats.append(normal.z)
indices = array(self.collada_indices)
vert_src = source.FloatSource("vert-array", array(vert_floats), \
('X', 'Y', 'Z'))
norm_src = source.FloatSource("norm-array", array(norm_floats), \
('X', 'Y', 'Z'))
geom = geometry.Geometry(mesh, "geometry0", "solid", [vert_src, norm_src])
input_list = source.InputList()
input_list.addInput(0, 'VERTEX', "#vert-array")
input_list.addInput(1, 'NORMAL', "#norm-array")
triset = geom.createTriangleSet(indices, input_list, "materialref")
geom.primitives.append(triset)
mesh.geometries.append(geom)
matnode = scene.MaterialNode("materialref", mat, inputs=[])
geomnode = scene.GeometryNode(geom, [matnode])
node = scene.Node("node0", children=[geomnode])
myscene = scene.Scene("myscene", [node])
mesh.scenes.append(myscene)
mesh.scene = myscene
mesh.write(name+'.dae')
def collada_from_scene(scene, name="triagain"):
collada = Collada()
geometry_nodes = ([
geometry_node_from_mesh(
collada=collada,
mesh=child.mesh,
name=f"mesh_geometry_{i}",
) for i, child in enumerate(child for child in scene.children
if isinstance(child, InternalMesh))
] + [
geometry_node_from_polyline(
collada,
child.polyline,
name=f"polyline_geometry_{i}",
color=normalize_color(child.color),
) for i, child in enumerate(child for child in scene.children
if isinstance(child, InternalLine))
] + [
geometry_node_from_points(
collada,
points=np.array([point.point for point in points]),
radius=scene.point_radius,
color=normalize_color(color),
name=f"point_geometry_{i}",
) for i, (color, points) in enumerate(
groupby(
lambda point: point.color,
[
child for child in scene.children
if isinstance(child, InternalPoint)
],
).items())
])
scene = Scene(name, [Node("root", children=geometry_nodes)])
collada.scenes.append(scene)
collada.scene = scene
return collada
def import_Collada(filename):
'''
Returns a list of objects represented as a list of triangles.
A triangle is list of vertices.
A vertex is a list of points.
'''
from collada import Collada
from StringIO import StringIO
import sys, inspect #This allows it to except bad Collada files by ignore all errors
clsmembers = inspect.getmembers(sys.modules['collada.common'],
inspect.isclass)
ignore_types = [cls[1] for cls in clsmembers]
mesh = Collada(filename, ignore=ignore_types)
geoms = mesh.geometries #This is the list of geometries
geom_tris = []
for geom in geoms:
for primitive in geom.primitives:
triangles = list(primitive)
#Appends each triangles set with each triangle being just a list of vertice
geom_tris.append([tri.vertices for tri in triangles])
return geom_tris
def generate_collada(coords, relative_texture_path):
"""generate the pycollada mesh out of the coordinates array
Arguments:
coords {[[[]]]} -- 2d array of 3d coordinates
relative_texture_path {str} -- relative path to the texture, relative to the generated collada file
Returns:
Collada -- final collada mesh
"""
# create the mesh
mesh = Collada()
# create source arrays
vert_src = source.FloatSource('verts-array', generate_vertex_array(coords),
('X', 'Y', 'Z'))
normal_src = source.FloatSource('normals-array',
generate_normal_array(coords),
('X', 'Y', 'Z'))
uv_src = source.FloatSource('uv-array', generate_uv_array(coords),
('S', 'T'))
# create geometry and add the sources
geom = geometry.Geometry(mesh, 'geometry', 'terrain',
[vert_src, normal_src, uv_src])
# define inputs to triangle set
input_list = source.InputList()
input_list.addInput(0, 'VERTEX', '#verts-array')
input_list.addInput(1, 'NORMAL', '#normals-array')
input_list.addInput(2, 'TEXCOORD', '#uv-array', set='0')
# create index array
indices = generate_index_array(coords)
# repeat each of the entries for vertex, normal, uv
indices = np.repeat(indices, 3)
# create triangle set, add it to list of geometries in the mesh
triset = geom.createTriangleSet(indices, input_list, 'material')
geom.primitives.append(triset)
mesh.geometries.append(geom)
# add a material
image = material.CImage('material-image', relative_texture_path)
surface = material.Surface('material-image-surface', image)
sampler2d = material.Sampler2D('material-image-sampler', surface)
material_map = material.Map(sampler2d, 'UVSET0')
effect = material.Effect('material-effect', [surface, sampler2d],
'lambert',
emission=(0.0, 0.0, 0.0, 1),
ambient=(0.0, 0.0, 0.0, 1),
diffuse=material_map,
transparent=material_map,
transparency=0.0,
double_sided=True)
mat = material.Material('materialID', 'material', effect)
mesh.effects.append(effect)
mesh.materials.append(mat)
mesh.images.append(image)
# instantiate geometry into a scene node
matnode = scene.MaterialNode('material', mat, inputs=[])
geomnode = scene.GeometryNode(geom, [matnode])
node = scene.Node('model', children=[geomnode])
# create scene
myscene = scene.Scene('scene', [node])
mesh.scenes.append(myscene)
mesh.scene = myscene
return mesh
def __init__(self, objects, filepath, directory, kwargs):
self._is_zae = kwargs["export_as"] == "zae"
self._export_textures = kwargs["export_textures"]
self._add_blender_extensions = kwargs["add_blender_extensions"]
self._filepath = filepath
self._dir = directory
self._ext_files_map = {}
self._ext_files_revmap = {}
if self._is_zae:
self._zip = zipfile.ZipFile(self._filepath, "w")
class ZipAttr:
compress_type = zipfile.ZIP_DEFLATED
file_attr = 0o100644 << 16
date_time = time.gmtime()[:6]
scene_name = "scene.dae"
@classmethod
def new_item(celf, filename):
item = zipfile.ZipInfo()
item.compress_type = celf.compress_type
item.external_attr = celf.file_attr
item.date_time = celf.date_time
item.filename = filename
return item
#end new_item
#end ZipAttr
self._zipattr = ZipAttr
# First item in archive is uncompressed and named “mimetype”, and
# contents is MIME type for archive. This way it ends up at a fixed
# offset (filename at 30 bytes from start, contents at 38 bytes) for
# easy detection by format-sniffing tools. This convention is used
# by ODF and other formats similarly based on Zip archives.
mimetype = zipfile.ZipInfo()
mimetype.filename = "mimetype"
mimetype.compress_type = zipfile.ZIP_STORED
mimetype.external_attr = ZipAttr.file_attr
mimetype.date_time = (2020, 8, 23, 1, 33, 52)
# about when I started getting .zae export working
self._zip.writestr(mimetype, b"model/vnd.collada+xml+zip")
# extrapolating from the fact that the official type for .dae files
# is “model/vnd.collada+xml”
else:
self._zip = None
self._zipattr = None
#end if
self._up_axis = kwargs["up_axis"]
if self._up_axis == "Z_UP":
self._orient = Matrix.Identity(4)
elif self._up_axis == "X_UP":
self._orient = Matrix.Rotation(-120 * DEG, 4, Vector(1, -1, 1))
else: # "Y_UP" or unspecified
self._orient = Matrix.Rotation(-90 * DEG, 4, "X")
#end if
obj_children = {}
for obj in objects:
parent = obj.parent
if parent == None:
parentname = None
else:
parentname = parent.name
#end if
if parentname not in obj_children:
obj_children[parentname] = set()
#end if
obj_children[parentname].add(obj.name)
#end for
self._obj_children = obj_children
self._selected_only = kwargs["use_selection"]
self._geometries = {}
self._materials = {}
self._collada = Collada()
self._collada.xmlnode.getroot().set("version",
kwargs["collada_version"])
self._collada.assetInfo.unitmeter = 1
self._collada.assetInfo.unitname = "metre"
self._collada.assetInfo.upaxis = self._up_axis
self._collada.assetInfo.save()
root_technique = self.blender_technique(
True, self._collada.xmlnode.getroot())
if root_technique != None:
prefixes = E.id_prefixes()
for k in sorted(DATABLOCK.__members__.keys()):
v = DATABLOCK[k]
if not v.internal_only:
prefix = E.prefix(name=k, value=v.nameid(""))
prefixes.append(prefix)
#end if
#end for
root_technique.append(prefixes)
#end if
self._scene = Scene(DATABLOCK.SCENE.nameid("main"), [])
self._collada.scenes.append(self._scene)
self._collada.scene = self._scene
self._id_seq = 0
def getColladaMesh(filename, node, link):
"""Read collada file."""
if not colladaIsAvailable:
sys.stderr.write('Collada module not found, please install it with:\n')
sys.stderr.write(' python -m pip install pycollada\n')
sys.stderr.write('Skipping "%s"\n' % filename)
return
print('Parsing Mesh: ' + filename)
colladaMesh = Collada(filename)
index = -1
if hasattr(node, 'material') and node.material:
for geometry in list(colladaMesh.scene.objects('geometry')):
for data in list(geometry.primitives()):
visual = Visual()
index += 1
visual.position = node.position
visual.rotation = node.rotation
visual.material.diffuse.red = node.material.diffuse.red
visual.material.diffuse.green = node.material.diffuse.green
visual.material.diffuse.blue = node.material.diffuse.blue
visual.material.diffuse.alpha = node.material.diffuse.alpha
visual.material.texture = node.material.texture
name = '%s_%d' % (os.path.splitext(
os.path.basename(filename))[0], index)
if type(data.original) is lineset.LineSet:
visual.geometry.lineset = True
if name in Geometry.reference:
visual.geometry = Geometry.reference[name]
else:
Geometry.reference[name] = visual.geometry
visual.geometry.name = name
visual.geometry.scale = node.geometry.scale
for val in data.vertex:
visual.geometry.trimesh.coord.append(numpy.array(val))
for val in data.vertex_index:
visual.geometry.trimesh.coordIndex.append(val)
if data.texcoordset: # non-empty
for val in data.texcoordset[0]:
visual.geometry.trimesh.texCoord.append(val)
if data.texcoord_indexset: # non-empty
for val in data.texcoord_indexset[0]:
visual.geometry.trimesh.texCoordIndex.append(val)
if hasattr(
data, '_normal'
) and data._normal is not None and data._normal.size > 0:
for val in data._normal:
visual.geometry.trimesh.normal.append(
numpy.array(val))
if hasattr(
data, '_normal_index'
) and data._normal_index is not None and data._normal_index.size > 0:
for val in data._normal_index:
visual.geometry.trimesh.normalIndex.append(val)
if data.material and data.material.effect:
if data.material.effect.emission and isinstance(
data.material.effect.emission, tuple):
visual.material.emission = colorVector2Instance(
data.material.effect.emission)
if data.material.effect.ambient and isinstance(
data.material.effect.ambient, tuple):
visual.material.ambient = colorVector2Instance(
data.material.effect.ambient)
if data.material.effect.specular and isinstance(
data.material.effect.specular, tuple):
visual.material.specular = colorVector2Instance(
data.material.effect.specular)
if data.material.effect.shininess:
visual.material.shininess = data.material.effect.shininess
if data.material.effect.index_of_refraction:
visual.material.index_of_refraction = data.material.effect.index_of_refraction
if data.material.effect.diffuse:
if numpy.size(data.material.effect.diffuse) > 1\
and all([isinstance(x, numbers.Number) for x in data.material.effect.diffuse]):
# diffuse is defined by values
visual.material.diffuse = colorVector2Instance(
data.material.effect.diffuse)
else:
# diffuse is defined by *.tif files
visual.material.texture = 'textures/' + \
data.material.effect.diffuse.sampler.surface.image.path.split('/')[-1]
txt = os.path.splitext(visual.material.texture)[1]
if txt == '.tiff' or txt == '.tif':
for dirname, dirnames, filenames in os.walk(
'.'):
for file in filenames:
if file == str(
visual.material.texture.split(
'/')[-1]):
try:
tifImage = Image.open(
os.path.join(
dirname, file))
img = './' + robotName + '_textures'
tifImage.save(
os.path.splitext(
os.path.join(
img, file))[0] +
'.png')
visual.material.texture = (
robotName + '_textures/' +
os.path.splitext(file)[0] +
'.png')
print('translated image ' +
visual.material.texture)
except IOError:
visual.material.texture = ""
print('failed to open ' +
os.path.join(
dirname, file))
else:
visual.material.diffuse = colorVector2Instance(
[1.0, 1.0, 1.0, 1.0])
link.visual.append(visual)
else:
for geometry in list(colladaMesh.scene.objects('geometry')):
for data in list(geometry.primitives()):
collision = Collision()
collision.position = node.position
collision.rotation = node.rotation
collision.geometry.scale = node.geometry.scale
for value in data.vertex:
collision.geometry.trimesh.coord.append(numpy.array(value))
for value in data.vertex_index:
collision.geometry.trimesh.coordIndex.append(value)
link.collision.append(collision)
def read_collada_file(model_path: Union[Path, str]) -> RawModelData:
"""Reads the model and returns it's data in the desirable format."""
if isinstance(model_path, Path):
model_path = str(model_path)
collada_obj = Collada(model_path)
# assumes that there is a single controller
if len(collada_obj.controllers) != 1:
raise RuntimeError(
f'there should be exactly 1 controller, got {len(collada_obj.controllers)}'
)
controller: Skin = collada_obj.controllers[0]
triangles: TriangleSet = controller.geometry.primitives[0]
# get vertices
vertices = triangles.vertex
n_vertices = vertices.shape[0]
# apply the bind_shape_matrix to all of the vertices
t_bind_shape = Transformation(controller.bind_shape_matrix)
vertices = t_bind_shape.apply(vertices)
# get faces
faces = triangles.vertex_index
n_faces = triangles.ntriangles
_check_all_vertices_in_faces(n_vertices, faces)
# get texture
vertex_texture_coords = _get_vertex_texture_coords(
n_vertices, faces, triangles.texcoord_indexset, triangles.texcoordset)
# get normals
vertex_normals = _get_vertex_normals(n_vertices, faces,
triangles.normal_index,
triangles.normal)
# get joint data
root_joint, joint_list, n_bound_joints, n_total_joints = _get_joints_data(
collada_obj.scene.nodes, controller)
# get weight matrix
weight_matrix = _get_weight_matrix(n_vertices, n_bound_joints,
controller.weights.data.flatten(),
controller.weight_index,
controller.joint_index)
# read animation data
animations = collada_obj.animations
n_keyframes = 0
if len(animations) > 0:
n_keyframes = _read_animations(animations, root_joint)
return RawModelData(
n_vertices=n_vertices,
n_faces=n_faces,
n_bound_joints=n_bound_joints,
n_total_joints=n_total_joints,
vertices=vertices,
faces=faces,
vertex_normals=vertex_normals,
vertex_texture_coords=vertex_texture_coords,
weight_matrix=weight_matrix,
joint_list=joint_list,
root_joint=root_joint,
n_keyframes=n_keyframes,
)
def export(file):
# Load scene from file
collada_file = Collada(file)
scene = collada_file.scene
# Create folder to store resources
folder_path = resources_path / Path(file.name).stem
if not folder_path.exists():
os.mkdir(folder_path)
# Export textures
mat_map = {}
model_folder_path = Path(file.name).parent
for material in collada_file.materials:
if hasattr(material.effect.diffuse, "sampler"):
mat_path = model_folder_path / Path(
material.effect.diffuse.sampler.surface.image.path)
shutil.copy(mat_path, folder_path / mat_path.name)
mat_map[material.effect.id] = Path(
file.name).stem + "/" + mat_path.name
# Process skinned meshes
skin_data = {}
node_list = []
bind_mats = []
for controller in collada_file.controllers:
joints = list(controller.sourcebyid[controller.joint_source])
joint_list = []
weight_list = []
for i in range(len(controller.joint_index)):
joint_indices = controller.joint_index[i]
vertex_joints = []
for joint_index in joint_indices:
joint_node = joints[joint_index]
if joint_node not in node_list:
node_list.append(joint_node)
vertex_joints.append(node_list.index(joint_node))
joint_list.append(vertex_joints)
weight_indices = controller.weight_index[i]
vertex_weights = []
for weight_index in weight_indices:
vertex_weights.append(
list(controller.weights)[weight_index][0].item())
weight_list.append(vertex_weights)
skin_data[controller.geometry.id] = {
"joints": joint_list,
"weights": weight_list
}
# Export meshes
for geometry in collada_file.geometries:
# Create new mesh
new_mesh = {}
vertex_list = []
texcoords_list = []
normals_list = []
joints_list = []
weights_list = []
for primitive in geometry.primitives:
# Set mesh's texture
if primitive.material in mat_map:
new_mesh["texture"] = mat_map[primitive.material]
for tri in list(primitive):
# Go over vertices, texture coordinates, and normals
for vertex in tri.vertices:
for coord in vertex:
vertex_list.append(coord.item())
for texcoord in tri.texcoords:
for coord in texcoord:
texcoords_list.append(coord[0].item())
texcoords_list.append(coord[1].item())
for normal in tri.normals:
for coord in normal:
normals_list.append(coord.item())
# Go over weights
if geometry.id in skin_data:
for index in tri.indices:
joints_list.append(
skin_data[geometry.id]["joints"][index])
weights_list.append(
skin_data[geometry.id]["weights"][index])
new_mesh["joints"] = joints_list
new_mesh["weights"] = weights_list
new_mesh["vertices"] = vertex_list
new_mesh["texcoords"] = texcoords_list
new_mesh["normals"] = normals_list
# Add metadata
new_mesh["metadata"] = {"version": "0.2", "type": "mesh"}
# Export JSON files
with open(folder_path / f"{geometry.name}.json", "w") as mesh_file:
json.dump(new_mesh, mesh_file)
# Export sub-tree
joint_node_dict = {}
root_node = ColladaImporter.process_node(scene, np.identity(4),
joint_node_dict, node_list,
Path(file.name).stem)
node_dict = GraphNode.scene_graph_to_dict(root_node)
with open(folder_path / f"{Path(file.name).stem}.json",
"w") as scene_file:
json.dump(node_dict, scene_file)
def getColladaMesh(filename, node, link):
"""Read collada file.
Args:
filename (str): path to the Collada (.dae) file.
node (DOM): XML node.
link (Link): link instance.
"""
colladaMesh = Collada(filename)
if hasattr(node, 'material') and node.material:
for geometry in list(colladaMesh.scene.objects('geometry')):
for data in list(geometry.primitives()):
visual = Visual()
visual.position = node.position
visual.rotation = node.rotation
visual.material.diffuse.red = node.material.diffuse.red
visual.material.diffuse.green = node.material.diffuse.green
visual.material.diffuse.blue = node.material.diffuse.blue
visual.material.diffuse.alpha = node.material.diffuse.alpha
visual.material.texture = node.material.texture
visual.geometry.scale = node.geometry.scale
for val in data.vertex:
visual.geometry.trimesh.coord.append(numpy.array(val))
for val in data.vertex_index:
visual.geometry.trimesh.coordIndex.append(val)
if data.texcoordset: # non-empty
for val in data.texcoordset[0]:
visual.geometry.trimesh.texCoord.append(val)
if data.texcoord_indexset: # non-empty
for val in data.texcoord_indexset[0]:
visual.geometry.trimesh.texCoordIndex.append(val)
if hasattr(
data, '_normal'
) and data._normal is not None and data._normal.size > 0:
for val in data._normal:
visual.geometry.trimesh.normal.append(numpy.array(val))
if hasattr(
data, '_normal_index'
) and data._normal_index is not None and data._normal_index.size > 0:
for val in data._normal_index:
visual.geometry.trimesh.normalIndex.append(val)
if data.material and data.material.effect:
if data.material.effect.emission:
visual.material.emission = colorVector2Instance(
data.material.effect.emission)
if data.material.effect.ambient:
visual.material.ambient = colorVector2Instance(
data.material.effect.ambient)
if data.material.effect.specular:
visual.material.specular = colorVector2Instance(
data.material.effect.specular)
if data.material.effect.shininess:
visual.material.shininess = data.material.effect.shininess
if data.material.effect.index_of_refraction:
visual.material.index_of_refraction = data.material.effect.index_of_refraction
if data.material.effect.diffuse:
if numpy.size(data.material.effect.diffuse) > 1\
and all([isinstance(x, numbers.Number) for x in data.material.effect.diffuse]):
# diffuse is defined by values
visual.material.diffuse = colorVector2Instance(
data.material.effect.diffuse)
else:
# diffuse is defined by *.tif files
visual.material.texture = 'textures/' + data.material.effect.diffuse.sampler.surface.image.path.split(
'/')[-1]
txt = os.path.splitext(visual.material.texture)[1]
if txt == '.tiff' or txt == '.tif':
for dirname, dirnames, filenames in os.walk(
'.'):
for file in filenames:
if file == str(
visual.material.texture.split(
'/')[-1]):
try:
tifImage = Image.open(
os.path.join(
dirname, file))
img = './' + robotName + '_textures'
tifImage.save(
os.path.splitext(
os.path.join(
img, file))[0] +
'.png')
visual.material.texture = robotName + '_textures/' + os.path.splitext(
file)[0] + '.png'
print('translated image ' +
visual.material.texture)
except IOError:
visual.material.texture = ""
print('failed to open ' +
os.path.join(
dirname, file))
link.visual.append(visual)
else:
for geometry in list(colladaMesh.scene.objects('geometry')):
for data in list(geometry.primitives()):
collision = Collision()
collision.position = node.position
collision.rotation = node.rotation
collision.geometry.scale = node.geometry.scale
for value in data.vertex:
collision.geometry.trimesh.coord.append(numpy.array(value))
for value in data.vertex_index:
collision.geometry.trimesh.coordIndex.append(value)
link.collision.append(collision)
def instancesToCollada(self,
parent_object,
collada_xml=None,
instance_node=True,
**kw):
try:
from upy.transformation import decompose_matrix
from collada import Collada
from collada import material
from collada import source
from collada import geometry
from collada import scene
except:
return
inst_parent = parent_object #self.getCurrentSelection()[0]
ch = self.getChilds(inst_parent)
transpose = True
if "transpose" in kw:
transpose = kw["transpose"]
#instance master
if "mesh" in kw and kw["mesh"] is not None:
inst_master = kw["mesh"]
f, v, vn = self.DecomposeMesh(kw["mesh"],
edit=False,
copy=False,
tri=True,
transform=False)
else:
inst_master = self.getMasterInstance(ch[0])
print "master is ", inst_master
#grabb v,f,n of inst_master
f, v, vn = self.DecomposeMesh(inst_master,
edit=False,
copy=False,
tri=True,
transform=False)
#special case when come from x-z swap
# v=[[vv[2],vv[1],vv[0]] for vv in v] # go back to regular
#-90degree rotation onY
mry90 = self.rotation_matrix(-math.pi / 2.0, [0.0, 1.0, 0.0]) #?
v = self.ApplyMatrix(v, mry90) #same for the normal?
vn = self.ApplyMatrix(vn, mry90) #same for the normal?
iname = self.getName(inst_master)
pname = self.getName(inst_parent)
if collada_xml is None:
collada_xml = Collada()
collada_xml.assetInfo.unitname = "centimeter"
collada_xml.assetInfo.unitmeter = 0.01
mat = self.getMaterialObject(inst_master)
if len(mat):
mat = mat[0]
props = self.getMaterialProperty(mat, color=1) #,specular_color=1)
print "colors is ", props
effect = material.Effect(
"effect" + iname, [],
"phong",
diffuse=[props["color"][0], props["color"][1], props["color"][2], 1.0])
# specular = props["specular_color"])
mat = material.Material("material" + iname, iname + "material", effect)
collada_xml.effects.append(effect)
collada_xml.materials.append(mat)
matnode = scene.MaterialNode(iname + "material" + "ref", mat, inputs=[])
#the geom
#invert Z ? for C4D?
vertzyx = numpy.array(v) #* numpy.array([1,1,-1])
z, y, x = vertzyx.transpose()
vertxyz = numpy.vstack([x, y, z]).transpose() #* numpy.array([1,1,-1])
vert_src = source.FloatSource(iname + "_verts-array", vertxyz.flatten(),
('X', 'Y', 'Z'))
norzyx = numpy.array(vn)
nz, ny, nx = norzyx.transpose()
norxyz = numpy.vstack([nx, ny, nz]).transpose() * numpy.array([1, 1, -1])
normal_src = source.FloatSource(iname + "_normals-array", norxyz.flatten(),
('X', 'Y', 'Z'))
geom = geometry.Geometry(collada_xml, "geometry" + iname, iname,
[vert_src, normal_src]) # normal_src])
input_list = source.InputList()
input_list.addInput(0, 'VERTEX', "#" + iname + "_verts-array")
input_list.addInput(0, 'NORMAL', "#" + iname + "_normals-array")
#invert all the face
fi = numpy.array(f, int) #[:,::-1]
triset = geom.createTriangleSet(fi.flatten(), input_list,
iname + "materialref")
geom.primitives.append(triset)
collada_xml.geometries.append(geom)
#the noe
#instance here ?
#creae the instance maser node :
if instance_node:
master_geomnode = scene.GeometryNode(geom, [matnode]) #doesn work ?
master_node = scene.Node("node_" + iname, children=[
master_geomnode,
]) #,transforms=[tr,rz,ry,rx,s])
g = []
for c in ch:
#collada.scene.NodeNode
if instance_node:
geomnode = scene.NodeNode(master_node)
else:
geomnode = scene.GeometryNode(geom, [matnode])
matrix = self.ToMat(
self.getTransformation(c)) #.transpose()#.flatten()
if transpose:
matrix = numpy.array(matrix).transpose()
scale, shear, euler, translate, perspective = decompose_matrix(matrix)
scale = self.getScale(c)
p = translate #matrix[3,:3]/100.0#unit problem
tr = scene.TranslateTransform(p[0], p[1], p[2])
rx = scene.RotateTransform(1, 0, 0, numpy.degrees(euler[0]))
ry = scene.RotateTransform(0, 1, 0, numpy.degrees(euler[1]))
rz = scene.RotateTransform(0, 0, 1, numpy.degrees(euler[2]))
# rx=scene.RotateTransform(-1,0,0,numpy.degrees(euler[0]))
# ry=scene.RotateTransform(0,-1,0,numpy.degrees(euler[1]))
# rz=scene.RotateTransform(0,0,1,numpy.degrees(euler[2]))
s = scene.ScaleTransform(scale[0], scale[1], scale[2])
#n = scene.NodeNode(master_node,transforms=[tr,rz,ry,rx,s])
# gnode = scene.Node(self.getName(c)+"_inst", children=[geomnode,])
n = scene.Node(
self.getName(c),
children=[
geomnode,
],
transforms=[tr, rz, ry, rx, s]
) #scene.MatrixTransform(matrix)[scene.MatrixTransform(numpy.array(matrix).reshape(16,))]
# n = scene.Node(self.getName(c), children=[geomnode,],
# transforms=[scene.MatrixTransform(numpy.array(matrix).reshape(16,))]) #scene.MatrixTransform(matrix)[scene.MatrixTransform(numpy.array(matrix).reshape(16,))]
g.append(n)
node = scene.Node(
pname, children=g) #,transforms=[scene.RotateTransform(0,1,0,90.0)])
if "parent_node" in kw:
kw["parent_node"].children.append(node)
node = kw["parent_node"]
if not len(collada_xml.scenes):
myscene = scene.Scene("myscene", [node])
collada_xml.scenes.append(myscene)
collada_xml.scene = myscene
else:
if "parent_node" not in kw:
collada_xml.scene.nodes.append(node)
if instance_node:
collada_xml.nodes.append(master_node)
return collada_xml
def __init__(self, filename, scale=1.0):
Group.__init__(self)
self._dae = Collada(filename)
self._load_mesh(self._dae, scale=scale)
ndata = isocontour.newDatasetRegFloat3D(newgrid3D, origin, stepsize)
# print "pfff"
isoc = isocontour.getContour3d(ndata, 0, 0, isovalue,
isocontour.NO_COLOR_VARIABLE)
vert = np.zeros((isoc.nvert, 3)).astype('f')
norm = np.zeros((isoc.nvert, 3)).astype('f')
col = np.zeros((isoc.nvert)).astype('f')
tri = np.zeros((isoc.ntri, 3)).astype('i')
isocontour.getContour3dData(isoc, vert, norm, col, tri, 0)
#print vert
if maskGrid.crystal:
vert = maskGrid.crystal.toCartesian(vert)
return vert, norm, tri
collada_xml = Collada()
collada_xml.assetInfo.unitname = "centimeter"
collada_xml.assetInfo.unitmeter = 0.01
collada_xml.assetInfo.upaxis = "Y_UP"
root_env = scene.Node(env.name)
myscene = scene.Scene(env.name + "_Scene", [root_env])
collada_xml.scenes.append(myscene)
collada_xml.scene = myscene
name = "myMolecule"
matnode = oneMaterial(name, collada_xml)
master_node = buildMeshGeom(name, v, f, collada_xml, matnode)
collada_xml.nodes.append(master_node)
collada_xml.write("test.dae")
def PyFactoryCreateFile(self, uclass: Class, parent: Object, name: str,
filename: str) -> Object:
# load the collada file
dae = Collada(filename)
ue.log_warning(dae)
self.do_import = False
self.open_collada_wizard()
if not self.do_import:
return None
# create a new UStaticMesh with the specified name and parent
static_mesh = StaticMesh(name, parent)
# prepare a new model with the specified build settings
source_model = StaticMeshSourceModel(
BuildSettings=MeshBuildSettings(bRecomputeNormals=False,
bRecomputeTangents=True,
bUseMikkTSpace=True,
bBuildAdjacencyBuffer=True,
bRemoveDegenerates=True))
# extract vertices, uvs and normals from the da file (numpy.ravel will flatten the arrays to simple array of floats)
triset = dae.geometries[0].primitives[0]
self.vertices = numpy.ravel(triset.vertex[triset.vertex_index])
# take the first uv channel (there could be multiple channels, like the one for lightmapping)
self.uvs = numpy.ravel(
triset.texcoordset[0][triset.texcoord_indexset[0]])
self.normals = numpy.ravel(triset.normal[triset.normal_index])
# fix mesh data
self.FixMeshData()
# create a new mesh, FRawMesh is an ptopmized wrapper exposed by the python plugin. read: no reflection involved
mesh = FRawMesh()
# assign vertices
mesh.set_vertex_positions(self.vertices)
# uvs are required
mesh.set_wedge_tex_coords(self.uvs)
# normals are optionals
mesh.set_wedge_tangent_z(self.normals)
# assign indices (not optimized, just return the list of triangles * 3...)
mesh.set_wedge_indices(numpy.arange(0, len(triset) * 3))
# assign the FRawMesh to the LOD0 (the model we created before)
mesh.save_to_static_mesh_source_model(source_model)
# assign LOD0 to the SataticMesh and build it
static_mesh.SourceModels = [source_model]
static_mesh.static_mesh_build()
static_mesh.static_mesh_create_body_setup()
static_mesh.StaticMaterials = [
StaticMaterial(
MaterialInterface=self.ImportOptions.DefaultMaterial,
MaterialSlotName='Main')
]
return static_mesh