diff --git a/rcg_pipeline/rcg_pipeline/render_asset.py b/rcg_pipeline/rcg_pipeline/render_asset.py index 2d5480b..d6954dc 100644 --- a/rcg_pipeline/rcg_pipeline/render_asset.py +++ b/rcg_pipeline/rcg_pipeline/render_asset.py @@ -71,7 +71,7 @@ def assembly_builder(item, libs_data, libs_data_dir, collection=None, parent=Non if item['type'] == 'LOCATOR': item_obj = bpy.data.objects.new(item['name'], None) item_obj.empty_display_type = 'CUBE' - item_obj.empty_display_size = 0.01 + item_obj.empty_display_size = 0.0005 collection.objects.link(item_obj) elif item['type'] == 'LCS': @@ -117,7 +117,7 @@ def assembly_builder(item, libs_data, libs_data_dir, collection=None, parent=Non if local_obj: local_obj[0].name = item['base_name'] item_obj.empty_display_type = 'PLAIN_AXES' - item_obj.empty_display_size = 0.01 + item_obj.empty_display_size = 0.0005 else: logger.error('Unknown object type %s of %s', item['type'], item['name']) @@ -190,6 +190,8 @@ def assembly_rebuilder(): def build_render_assets(project_dir): ''' ''' + assets_data = [] + libs_data_path = os.path.join(project_dir, 'libs.json') if not os.path.exists(libs_data_path): raise Exception('No libs database found! Check %s directory' % project_dir) @@ -236,5 +238,19 @@ def build_render_assets(project_dir): bpy.ops.wm.save_as_mainfile(filepath=blend_path) logger.info('Render asset %s generated!', tree_item['name']) + assets_data.append( + { + 'type': 'RENDER', + 'name': tree_item['name'], + 'path': os.path.relpath(blend_path, project_dir), + 'thumbnail': '' + } + ) logger.info('%s Render Assets was generated!', len(tree_item_list)) + # write db file + assets_data_path = os.path.join(project_dir, 'assets.json') + with open(assets_data_path, 'w', encoding='utf-8') as assets_data_file: + json.dump(assets_data, assets_data_file, ensure_ascii=False, indent=4) + logger.info('Database saved successfully to %s!', assets_data_path) + return render_assets_dir