153 lines
5.7 KiB
Python
153 lines
5.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (C) 2023 Ilia Kurochkin <brothermechanic@gmail.com>
|
|
#
|
|
# 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 3 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.
|
|
'''
|
|
DESCRIPTION.
|
|
- Build Blender scene from JSON data.
|
|
- Setup hierarchy.
|
|
- Setup materials.
|
|
- Setup LCS points.
|
|
- Apply Bledner scene transforms.
|
|
'''
|
|
__version__ = '0.1'
|
|
|
|
import collections
|
|
import logging
|
|
import random
|
|
import bpy
|
|
from blender.utils.object_transforms import apply_transforms
|
|
from blender.import_cad.import_hierarchy import (fc_placement,
|
|
hierarchy)
|
|
from blender.import_cad.import_materials import (assign_materials,
|
|
assign_black)
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
# COLLECTIONS NAMIG CONVENTION
|
|
part_col_name = 'Parts'
|
|
lcs_col_name = 'LCS'
|
|
hierarchy_col_name = 'Hierarchy'
|
|
lowpoly_col_name = 'Lowpoly'
|
|
# LCS POINT'S SUFFIXES CONVENTION
|
|
inlet = '_in'
|
|
outlet = '_out'
|
|
root = '_root'
|
|
# CG ASSETS SUFFIXES CONVENTION
|
|
hightpoly = '_hp'
|
|
lowpoly = '_lp'
|
|
render = '_render'
|
|
|
|
scene_scale = 0.001
|
|
blackbody_mat_name = 'Robossembler_Black_Body'
|
|
|
|
|
|
def json_to_blend(js_data):
|
|
''' Reads JSON data and creates Blender scene '''
|
|
|
|
part_collection = bpy.data.collections.new(part_col_name)
|
|
bpy.context.scene.collection.children.link(part_collection)
|
|
|
|
lcs_collection = bpy.data.collections.new(lcs_col_name)
|
|
bpy.context.scene.collection.children.link(lcs_collection)
|
|
|
|
hierarchy_collection = bpy.data.collections.new(hierarchy_col_name)
|
|
bpy.context.scene.collection.children.link(hierarchy_collection)
|
|
|
|
fc_file = list(js_data.keys())[0]
|
|
|
|
imported_objects = collections.defaultdict(list)
|
|
|
|
for js_obj in js_data[fc_file]:
|
|
bobj = None
|
|
|
|
if js_data[fc_file][js_obj]['type'] == 'LCS':
|
|
bobj = bpy.data.objects.new(js_obj, None)
|
|
bobj.empty_display_type = 'ARROWS'
|
|
bobj.empty_display_size = round(random.uniform(0.05, 0.15), 3)
|
|
bobj.show_in_front = True
|
|
lcs_collection.objects.link(bobj)
|
|
imported_objects['objs_lcs'].append(bobj.name)
|
|
|
|
elif js_data[fc_file][js_obj]['type'] == 'PART':
|
|
if js_data[fc_file][js_obj].get('mesh'):
|
|
verts = js_data[fc_file][js_obj]['mesh'][0]
|
|
edges = []
|
|
faces = js_data[fc_file][js_obj]['mesh'][1]
|
|
|
|
# create blender object data
|
|
bmesh = bpy.data.meshes.new(name=js_obj)
|
|
bmesh.from_pydata(verts, edges, faces)
|
|
bmesh.update()
|
|
bobj = bpy.data.objects.new(js_obj, bmesh)
|
|
part_collection.objects.link(bobj)
|
|
|
|
if bobj:
|
|
fc_placement(bobj,
|
|
js_data[fc_file][js_obj]['fc_location'],
|
|
js_data[fc_file][js_obj]['fc_rotation'],
|
|
scene_scale)
|
|
if bobj.type == 'MESH':
|
|
bobj.scale = (scene_scale, scene_scale, scene_scale)
|
|
apply_transforms(bobj, scale=True)
|
|
|
|
# construct assembly hierarchy
|
|
hierarchy_objs = hierarchy(bobj,
|
|
js_data[fc_file][js_obj]['hierarchy'],
|
|
scene_scale)
|
|
for hierarchy_obj in hierarchy_objs:
|
|
hierarchy_collection.objects.link(hierarchy_obj)
|
|
imported_objects['objs_hierarchy'].append(hierarchy_obj.name)
|
|
|
|
# one material for the whole object
|
|
if bobj.type == 'MESH':
|
|
if js_data[fc_file][js_obj].get('material'):
|
|
fem_mat = js_data[fc_file][js_obj]['material']
|
|
assign_materials(bobj, fem_mat)
|
|
imported_objects['objs_foreground'].append(bobj.name)
|
|
else:
|
|
assign_black(bobj)
|
|
imported_objects['objs_background'].append(bobj.name)
|
|
|
|
# losted root lcs inlet workaround
|
|
if imported_objects['objs_lcs']:
|
|
root_lcs = None
|
|
for obj_name in imported_objects['objs_lcs']:
|
|
if obj_name.endswith(root):
|
|
root_lcs = bpy.data.objects[obj_name]
|
|
break
|
|
if root_lcs:
|
|
root_inlet_name = '{}{}'.format(root_lcs.name.split(root)[0], inlet)
|
|
if not bpy.data.objects.get(root_inlet_name):
|
|
root_inlet = bpy.data.objects.new(root_inlet_name, None)
|
|
root_inlet.empty_display_type = 'ARROWS'
|
|
root_inlet.empty_display_size = 0.1
|
|
root_inlet.show_in_front = True
|
|
root_inlet.location = root_lcs.location
|
|
root_inlet.rotation_euler = root_lcs.rotation_euler
|
|
root_inlet.parent = root_lcs.parent
|
|
lcs_collection.objects.link(root_inlet)
|
|
imported_objects['objs_lcs'].append(root_inlet.name)
|
|
logger.info('Root Inlet LCS object created!')
|
|
else:
|
|
logger.info('Root Inlet LCS object already exists!')
|
|
else:
|
|
logger.info('Lost Root LCS object!')
|
|
else:
|
|
logger.info('No LCS objects found!')
|
|
|
|
# TODO
|
|
# update do not dork
|
|
logger.info('Generated %s objects without errors',
|
|
len(sum(list(imported_objects.values()), [])))
|
|
return imported_objects
|