45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
DESCRIPTION.
|
||
|
|
Subassembly models setup from SDFormat file.
|
||
|
|
Script selected subassembly models.
|
||
|
|
Support Blender compiled as a Python Module only!
|
||
|
|
"""
|
||
|
|
__version__ = "0.1"
|
||
|
|
|
||
|
|
import logging
|
||
|
|
import sys
|
||
|
|
import xml.etree.ElementTree as ET
|
||
|
|
import os
|
||
|
|
sys.path.append('../')
|
||
|
|
from mathutils import Matrix
|
||
|
|
import bpy
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
logging.basicConfig(level=logging.INFO)
|
||
|
|
|
||
|
|
|
||
|
|
def sdf_mesh_selector(sdf_path):
|
||
|
|
""" Setup scene by SDFormat file description """
|
||
|
|
mytree = ET.parse(sdf_path)
|
||
|
|
myroot = mytree.getroot()
|
||
|
|
|
||
|
|
models_transforms = {}
|
||
|
|
for models in myroot.iter('model'):
|
||
|
|
for models_childs in models:
|
||
|
|
if models_childs.find('pose') is not None:
|
||
|
|
pose = models_childs.find('pose').text
|
||
|
|
logger.debug(models.attrib['name'], pose)
|
||
|
|
pose_vectors = [float(x) * 0.001 for x in pose.split()]
|
||
|
|
models_transforms[models.attrib['name']] = pose_vectors
|
||
|
|
|
||
|
|
# TODO More cleanup and optimise
|
||
|
|
# TODO add models <==> objects compare validation
|
||
|
|
|
||
|
|
for obj in bpy.context.scene.objects:
|
||
|
|
if obj.name in models_transforms:
|
||
|
|
obj.matrix_world = Matrix()
|
||
|
|
else:
|
||
|
|
bpy.data.objects.remove(bpy.data.objects[obj.name], do_unlink=True)
|
||
|
|
logger.info(obj.name, '- is not in assemly objects. Deleted!')
|