Automatized the thumbnail generation, and adding easily all lights

Update export to find export path without the operator

.gitignore

@@ -7,24 +7,9 @@
 __pycache__/
 **/__pycache__/
 
-
-# Unit Test #
-*.zip
-tests/blender/**
-tests/blender/*.dmg
-!tests/blender/.keep
-
 ### Blender ###
 *.blend1
 
-### Unreal ###
-tests/unreal_sample/DerivedDataCache
-tests/unreal_sample/Intermediate
-tests/unreal_sample/Saved
-
-# Secret file
-**/token.txt
-
 # Virtual Environment
 **/venv/**
 venv/

Fange_Pipeline/__init__.py

@@ -1,9 +1,17 @@
 import bpy
 
-from .ui.export import GRAOU_PT_panel
-from .operators.outline import ConfigBlendScene
+# UI and Interface
+from .ui.asset import GraouPanel_asset
+from .ui.export import GRAOU_PT_export
+from .ui.icon import GraouPanel_thumbnail
+from .ui.setup import GRAOU_PT_setup
+# All operators
 from .operators.exports import ExportForFange
+from .operators.outline import ConfigBlendScene
 from .operators.misc import MakeBasicCollision
+from .operators.lighting import ConfigLighting
+from .operators.thumbnails import ConfigRendering
+# Preferences and properties
 from .preference import GRAOU_AddonPreference
 from .properties.main import FangeProperties
 
@@ -20,10 +28,10 @@ bl_info = {
 }
 
 modules_class = [
-    # Main Property
-    ExportForFange, MakeBasicCollision, ConfigBlendScene,
-    # UI
-    GRAOU_PT_panel,
+    # Main operators property
+    ExportForFange, MakeBasicCollision, ConfigBlendScene, ConfigLighting, ConfigRendering,
+    # UI, the order are the way to select how show each panel
+    GRAOU_PT_setup, GraouPanel_asset, GRAOU_PT_export, GraouPanel_thumbnail,
     # Preference
     GRAOU_AddonPreference, FangeProperties
 ]

Fange_Pipeline/operators/exports.py

@@ -1,21 +1,21 @@
 import bpy
 
-from ..properties.models import FangeProject
-from pathlib import Path
+from ..utils import get_export_path, get_asset_name
 
 
 class ExportForFange(bpy.types.Operator):
-    """Export a building for fange"""
+    """Export a building for fange.
+Requiert your blend file are write on your hard-drive."""
     bl_idname = 'graou.building_export'
     bl_label = 'Easily export a building asset'
 
     def __init__(self):
         self.coll_layout = bpy.data.collections.get('Placeholder')
-        self.path = FangeProject()
-        self.asset = self.get_asset_name()
-        self.category = self.get_asset_type()
+        self.asset = get_asset_name()
+        self.category = get_export_path()
 
         self.instance_type_dict = {}
+        self.temp_copy = {}
 
     @classmethod
     def poll(cls, context):
@@ -26,7 +26,6 @@ class ExportForFange(bpy.types.Operator):
 
     def execute(self, context):
         bpy.ops.object.select_all(action='DESELECT')
-        print(f'[Pipeline] Start to export the building props.')
 
         # Make a check if the file are saved on the disk
         if not bpy.data.is_saved:
@@ -37,111 +36,88 @@ class ExportForFange(bpy.types.Operator):
             self.report({'ERROR'}, 'Can\'t find the asset category.')
             return {'CANCELLED'}
 
-        if len(self.coll_layout.collection_children):
-            for coll in self.coll_layout.children:
-                print(f'[Pipeline] Update "{coll.name}" mesh')
-                child = bpy.data.collections.get(coll.name)
+        if not len(self.coll_layout.children):
+            self.report({'WARNING'}, 'No children collection, nothing can be exported.')
+            return {'CANCELLED'}
 
-                for ob in child.all_objects:
-                    print(f'\tLook "{ob.name}", his type are "{type(ob.data)}"')
-                    # TODO Support another type object, not only SM/SK
-                    if ob.type == 'MESH' or 'EMPTY':
-                        print(f'[Pipeline] Check ob {ob.name} and is type {type(ob)}')
-                        ob.select_set(True)
+        for coll in self.coll_layout.children:
+            print(f'[Pipeline] Update "{coll.name}" mesh')
+            child = bpy.data.collections.get(coll.name)
 
-                        if ob.type == 'EMPTY':
-                            if ob.instance_type != 'NONE':
-                                self.instance_type_dict[ob.name] = ob.instance_type
-                                ob.instance_type = 'NONE'
+            for ob in child.all_objects:
+                print(f'\tLook "{ob.name}", his type are "{type(ob.data)}"')
+                # TODO Support another type object, not only SM/SK
+                if ob.type == 'MESH' or 'EMPTY' or 'CURVE':
+                    print(f'[Pipeline] Check ob {ob.name} and is type {type(ob)}')
+                    ob.select_set(True)
 
-                abs_export = self.category.joinpath(self.asset, "Meshes")
-                if not abs_export.exists():
-                    abs_export.mkdir(parents=True)
+                    if ob.type == 'EMPTY':
+                        if ob.instance_type != 'NONE':
+                            self.instance_type_dict[ob.name] = ob.instance_type
+                            ob.instance_type = 'NONE'
 
-                if coll.name != 'Socket':
-                    asset_name = f"SM_{coll.name}.fbx"
-                else:
-                    asset_name = f"{coll.name}.fbx"
+                    if ob.type == 'CURVE':
+                        # Make a new objet with the Geo
+                        bpy.ops.object.convert(target='CURVE')
 
-                # TODO Use the preset system
-                bpy.ops.export_scene.fbx(filepath=abs_export.joinpath(asset_name).as_posix(),
-                                         use_selection=True,
-                                         use_visible=False,
-                                         use_active_collection=False,
-                                         global_scale=1.0,
-                                         apply_unit_scale=True,
-                                         apply_scale_options='FBX_SCALE_NONE',
-                                         use_space_transform=True,
-                                         bake_space_transform=True,
-                                         object_types={'MESH', 'EMPTY'},
-                                         use_mesh_modifiers=True,
-                                         use_mesh_modifiers_render=True,
-                                         mesh_smooth_type='OFF',
-                                         colors_type='SRGB',
-                                         prioritize_active_color=False,
-                                         use_subsurf=False,
-                                         use_mesh_edges=False,
-                                         use_tspace=False,
-                                         use_triangles=False,
-                                         use_custom_props=False,
-                                         add_leaf_bones=True,
-                                         primary_bone_axis='Y',
-                                         secondary_bone_axis='X',
-                                         use_armature_deform_only=False,
-                                         armature_nodetype='NULL',
-                                         bake_anim=False,
-                                         bake_anim_use_all_bones=True,
-                                         bake_anim_use_nla_strips=True,
-                                         bake_anim_use_all_actions=True,
-                                         bake_anim_force_startend_keying=True,
-                                         bake_anim_step=1.0,
-                                         bake_anim_simplify_factor=1.0,
-                                         path_mode='AUTO',
-                                         embed_textures=False,
-                                         batch_mode='OFF',
-                                         use_batch_own_dir=True,
-                                         axis_forward='X',
-                                         axis_up='Y'
-                                         )
+            abs_export = self.category.joinpath(self.asset, "Meshes")
+            if not abs_export.exists():
+                abs_export.mkdir(parents=True)
 
-                print(f'[Pipeline] Export here "{abs_export}"')
-                bpy.ops.object.select_all(action='DESELECT')
+            if coll.name != 'Socket':
+                asset_name = f"SM_{coll.name}.fbx"
+            else:
+                asset_name = f"{coll.name}.fbx"
 
-                self.set_instance_type()
+            # TODO Use the preset system
+            bpy.ops.export_scene.fbx(filepath=abs_export.joinpath(asset_name).as_posix(),
+                                     use_selection=True,
+                                     use_visible=False,
+                                     use_active_collection=False,
+                                     global_scale=1.0,
+                                     apply_unit_scale=True,
+                                     apply_scale_options='FBX_SCALE_NONE',
+                                     use_space_transform=True,
+                                     bake_space_transform=True,
+                                     object_types={'MESH', 'EMPTY'},
+                                     use_mesh_modifiers=True,
+                                     use_mesh_modifiers_render=True,
+                                     mesh_smooth_type='OFF',
+                                     colors_type='SRGB',
+                                     prioritize_active_color=False,
+                                     use_subsurf=False,
+                                     use_mesh_edges=False,
+                                     use_tspace=False,
+                                     use_triangles=False,
+                                     use_custom_props=False,
+                                     add_leaf_bones=True,
+                                     primary_bone_axis='Y',
+                                     secondary_bone_axis='X',
+                                     use_armature_deform_only=False,
+                                     armature_nodetype='NULL',
+                                     bake_anim=False,
+                                     bake_anim_use_all_bones=True,
+                                     bake_anim_use_nla_strips=True,
+                                     bake_anim_use_all_actions=True,
+                                     bake_anim_force_startend_keying=True,
+                                     bake_anim_step=1.0,
+                                     bake_anim_simplify_factor=1.0,
+                                     path_mode='AUTO',
+                                     embed_textures=False,
+                                     batch_mode='OFF',
+                                     use_batch_own_dir=True,
+                                     axis_forward='X',
+                                     axis_up='Y'
+                                     )
 
-        # for coll in coll_layout.children:
-        #    print(f'[Pipeline] Check {coll}. Item type {type(coll)}')
+            print(f'[Pipeline] Export here "{abs_export}"')
+            bpy.ops.object.select_all(action='DESELECT')
+
+            self.set_instance_type()
 
         self.report({'INFO'}, 'Placeholder exported')
         return {'FINISHED'}
 
-    def get_asset_type(self) -> Path:
-        """Look the file path, to understand if the asset are a Character, Buildings or Props"""
-        abs_blend_path = Path(bpy.data.filepath)
-        print(f'[Pipeline] Get blend file path "{abs_blend_path}"')
-        print(f'[Pipeline] Convert to List "{abs_blend_path.parts}"')
-
-        if 'Buildings' in abs_blend_path.parts:
-            print(f'[Pipeline] Export here "{self.path.buildings}"')
-            return self.path.buildings
-
-        elif 'Pros' in abs_blend_path.parts:
-            print(f'[Pipeline] Export here "{self.path.props}"')
-            return self.path.props
-
-        elif 'Characters' in abs_blend_path.parts:
-            print(f'[Pipeline] Export here "{self.path.characters}"')
-            return self.path.characters
-
-        else:
-            print(f'[Pipeline] Can\'t find asset category')
-            return Path()
-
-    @staticmethod
-    def get_asset_name():
-        abs_blend_path = Path(bpy.data.filepath)
-        return abs_blend_path.stem
-
     def set_instance_type(self):
         for key, value in self.instance_type_dict.items():
             ob = bpy.data.objects.get(key)

Fange_Pipeline/operators/lighting.py

@@ -0,0 +1,64 @@
+import bpy
+
+from pathlib import Path
+
+
+class ConfigLighting(bpy.types.Operator):
+    """Add or conform a lighting build"""
+    bl_idname = 'graou.lighting'
+    bl_label = 'Config or update a lighting'
+
+    def execute(self, context):
+        # Look if a light object are in the scene to clear
+        self.clear_light_on_scene()
+
+        r = self.get_blend_resources_file()
+
+        world = bpy.data.worlds.get('WorldIconRendering')
+        if not world:
+            with bpy.data.libraries.load(r, link=True) as (data_from, data_to):
+                data_to.worlds = data_from.worlds
+
+            world = bpy.data.worlds.get('WorldIconRendering')
+
+        context.scene.world = world
+
+        if not bpy.data.collections.get('Lighting'):
+            print(f'[Pipeline] Need to import the lighting collection')
+            with bpy.data.libraries.load(r, link=False) as (data_from, data_to):
+                for collection in data_from.collections:
+                    print(f'[Pipeline] Collection can be import "{collection}"')
+                    if 'Lighting' in collection:
+                        print(f'[Pipeline] Append {collection}.')
+                        data_to.collections.append(collection)
+
+            print('[Pipeline] Link lighting collection with the scene collection')
+            bpy.data.collections['Scene'].children.link(bpy.data.collections.get('Lighting'))
+
+        return {'FINISHED'}
+
+    @staticmethod
+    def get_blend_resources_file():
+        """Get the blend file shared with the addon"""
+        addon_folder = Path(bpy.utils.user_resource('SCRIPTS'))
+        addon_name = __name__.split(".")[0]
+
+        return addon_folder.joinpath("addons", addon_name, "resources", "Lighting.blend").as_posix()
+
+    def clear_light_on_scene(self):
+        bpy.ops.object.select_all(action='DESELECT')
+
+        for ob in bpy.data.objects:
+            if ob.type == 'LIGHT' and not self.find_collection(ob):
+                ob.select_set(True)
+                bpy.ops.object.delete()
+
+            else:
+                ob.select_set(False)
+
+    @staticmethod
+    def find_collection(ob, name='Lighting'):
+        """Check if an object are attached with the Lighting collection"""
+        for collection in ob.users_collection:
+            if collection.name == name:
+                return True

Fange_Pipeline/operators/misc.py

@@ -11,4 +11,5 @@ class MakeBasicCollision(bpy.types.Operator):
         return bpy.context.object
 
     def execute(self, context):
+        # TODO
         return {'FINISHED'}

Fange_Pipeline/operators/outline.py

@@ -86,5 +86,7 @@ class SetCollectionSocket(bpy.types.Operator):
 
     def __init__(self):
         self.socket = bpy.data.collections.get('Socket')
+
     def execute(self, context):
+        # TODO
         return {'FINISHED'}

Fange_Pipeline/operators/thumbnails.py

@@ -0,0 +1,44 @@
+import bpy
+
+from ..utils import get_export_path, get_asset_name
+from pathlib import Path
+
+
+class ConfigRendering(bpy.types.Operator):
+    """Setup camera and rendering config"""
+    bl_idname = 'graou_config.rendering_thumbnail'
+    bl_label = 'Setup the blend file to be ready'
+
+    def __init__(self):
+        self.scene = bpy.data.scenes['Scene']
+
+        self.path_export = get_export_path()
+        self.asset_name = get_asset_name()
+        self.filename_export = Path(self.path_export, self.asset_name, f'TX_Icon{self.asset_name}.tga')
+
+    def execute(self, context):
+        self.set_camera_used()
+        self.set_rendering_panel()
+        self.set_output_file()
+
+        bpy.ops.render.render(write_still=True, use_viewport=True)
+
+        return {'FINISHED'}
+
+    def set_rendering_panel(self):
+        self.scene.render.engine = 'BLENDER_EEVEE'
+        self.scene.eevee.use_gtao = True
+        self.scene.eevee.use_ssr = True
+        self.scene.render.use_high_quality_normals = True
+        self.scene.render.film_transparent = True
+
+    def set_output_file(self):
+        self.scene.render.resolution_x = self.scene.render.resolution_y = 512
+        self.scene.render.image_settings.file_format = 'TARGA'
+        self.scene.render.image_settings.color_mode = 'RGBA'
+        self.scene.render.filepath = self.filename_export.as_posix()
+
+    def set_camera_used(self):
+        """Find the best camera position"""
+        # TODO
+        pass

Fange_Pipeline/ui/asset.py

@@ -0,0 +1,12 @@
+from .main import GraouPanel
+
+
+class GraouPanel_asset(GraouPanel):
+    bl_idname = 'GRAOU_PT_asset'
+    bl_label = 'Generate Assets'
+
+    def draw(self, context):
+        layout = self.layout
+
+        layout.label(text='Asset:')
+        layout.operator('graou.make_collision', text='Generate collision', icon='MOD_PHYSICS')

Fange_Pipeline/ui/export.py

@@ -1,44 +1,13 @@
-import bpy
-import os
-
-# from pathlib import Path
-#
-# preview_collection = {}
-# icon_sauropod_path = Path(os.path.dirname(os.path.abspath(__file__)), "icons")
-#
-# pcoll = bpy.utils.previews.new()
-#
-# for entry in os.scandir(icon_sauropod_path):
-#     if entry.name.endswith(".png"):
-#         name = os.path.splitext(entry.name)[0]
-#         print(f'[Pipeline] Add icon "{name}"')
-#         pcoll.load(name.upper(), entry.path, "IMAGE")
+from .main import GraouPanel
 
 
-class GRAOU_PT_panel(bpy.types.Panel):
+class GRAOU_PT_export(GraouPanel):
     bl_idname = 'GRAOU_PT_MAIN'
-    bl_space_type = 'VIEW_3D'
-    bl_region_type = 'UI'
-    bl_label = 'Pipeline'
-    bl_category = 'Graou Studio'
+    bl_label = 'Export'
 
     def draw(self, context):
         layout = self.layout
 
-        # layout.label(text='Graou Pipeline', icon_value=pcoll["GRAOU"].icon_id)
-
-        col = layout.column(align=True)
-        col.label(text='Main Config:')
-        col.operator('graou.build_scene', text='Init Scene', icon='OUTLINER')
-        col.prop(context.scene.graou_props, 'socket_collection', text='Use socket', toggle=True)
-
-        layout.separator()
-
-        layout.label(text='Asset:')
-        layout.operator('graou.make_collision', text='Generate collision', icon='MOD_PHYSICS')
-
-        layout.separator()
-
         layout.label(text='Export scene:')
         box = layout.box()
         box.label(text='Sanity Check')

Fange_Pipeline/ui/icon.py

@@ -0,0 +1,13 @@
+from .main import GraouPanel
+
+
+class GraouPanel_thumbnail(GraouPanel):
+    bl_idname = 'GRAOU_PT_thumbnail'
+    bl_label = 'Thumbnail'
+
+    def draw(self, context):
+        layout = self.layout
+
+        layout.label(text='Thumbnail:')
+        layout.operator('graou_config.rendering_thumbnail', text='Generate Thumbnail', icon='FILE_IMAGE')
+        # TODO Add the thumbnail view

Fange_Pipeline/ui/main.py

@@ -0,0 +1,25 @@
+import bpy
+# import os
+
+# from pathlib import Path
+#
+# preview_collection = {}
+# icon_sauropod_path = Path(os.path.dirname(os.path.abspath(__file__)), "icons")
+#
+# pcoll = bpy.utils.previews.new()
+#
+# for entry in os.scandir(icon_sauropod_path):
+#     if entry.name.endswith(".png"):
+#         name = os.path.splitext(entry.name)[0]
+#         print(f'[Pipeline] Add icon "{name}"')
+#         pcoll.load(name.upper(), entry.path, "IMAGE")
+
+# layout.label(text='Graou Pipeline', icon_value=pcoll["GRAOU"].icon_id)
+
+
+class GraouPanel(bpy.types.Panel):
+    bl_idname = 'GRAOU_PT_MAIN'
+    bl_space_type = 'VIEW_3D'
+    bl_region_type = 'UI'
+    bl_label = 'Export'
+    bl_category = 'Graou Studio'

Fange_Pipeline/ui/setup.py

@@ -0,0 +1,15 @@
+from .main import GraouPanel
+
+
+class GRAOU_PT_setup(GraouPanel):
+    bl_idname = 'GRAOU_PT_setup'
+    bl_label = 'Setup Pipeline'
+
+    def draw(self, context):
+        layout = self.layout
+
+        col = layout.column(align=True)
+        col.label(text='Main Config:')
+        col.operator('graou.build_scene', text='Init Scene', icon='OUTLINER')
+        col.prop(context.scene.graou_props, 'socket_collection', text='Use socket', toggle=True)
+        col.operator('graou.lighting', text='Set basic lighting', icon='OUTLINER_OB_LIGHT')

Fange_Pipeline/utils.py

@@ -1,5 +1,6 @@
 import bpy
 
+from .properties.models import FangeProject
 from pathlib import Path
 
 
@@ -21,8 +22,25 @@ def get_export_preset():
         print(f'[Pipeline] Preset folder not found')
 
 
-def apply_mesh_and_join():
-    """
-    This function make a new mesh with all selected asset.
-    """
-    pass
+def get_export_path():
+    """Look the file path, to understand if the asset are a Character, Buildings or Props"""
+    abs_blend_path = Path(bpy.data.filepath)
+    game_path = FangeProject()
+
+    if 'Buildings' in abs_blend_path.parts:
+        return game_path.buildings
+
+    elif 'Props' in abs_blend_path.parts:
+        return game_path.props
+
+    elif 'Characters' in abs_blend_path.parts:
+        return game_path.characters
+
+    else:
+        return Path()
+
+
+def get_asset_name() -> str:
+    """From the .blend file, return the blend name."""
+    abs_blend_path = Path(bpy.data.filepath)
+    return abs_blend_path.stem