Skip to content

Commit

Permalink
Clean and update
Browse files Browse the repository at this point in the history
  • Loading branch information
@xavier150
xavier150 committed Jan 23, 2024
1 parent 9a59559 commit 2bf29f2
Show file tree
Hide file tree
Showing 7 changed files with 178 additions and 121 deletions.
11 changes: 7 additions & 4 deletions blender-for-unrealengine/bfu_import_module/__init__.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
from . import bps
from . import import_module_utils
from . import import_module_unreal_utils
from . import import_module_post_treatment
from . import asset_import
from . import sequencer_import

Expand All @@ -30,15 +31,17 @@
importlib.reload(import_module_utils)
if "import_module_unreal_utils" in locals():
importlib.reload(import_module_unreal_utils)
if "import_module_post_treatment" in locals():
importlib.reload(import_module_post_treatment)
if "asset_import" in locals():
importlib.reload(asset_import)
if "sequencer_import" in locals():
importlib.reload(sequencer_import)

def run_asset_import(assets_data):
def run_asset_import(assets_data, show_finished_popup=True):
if asset_import.ready_for_asset_import():
return asset_import.ImportAllAssets(assets_data)
return asset_import.ImportAllAssets(assets_data, show_finished_popup)

def run_sequencer_import(sequence_data):
def run_sequencer_import(sequence_data, show_finished_popup=True):
if sequencer_import.ready_for_sequence_import():
return sequencer_import.CreateSequencer(sequence_data)
return sequencer_import.CreateSequencer(sequence_data, show_finished_popup)
162 changes: 53 additions & 109 deletions blender-for-unrealengine/bfu_import_module/asset_import.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,9 @@
from . import bps
from . import import_module_utils
from . import import_module_unreal_utils
from . import import_module_post_treatment

try: # TO DO: Found a better way to check that.
try:
import unreal
except ImportError:
import unreal_engine as unreal
Expand Down Expand Up @@ -61,23 +62,28 @@ def GetAdditionalData():
return import_module_utils.JsonLoadFile(asset_data["additional_tracks_path"])
return None

additional_data = GetAdditionalData()
asset_additional_data = GetAdditionalData()



if asset_data["asset_type"] == "Animation" or asset_data["asset_type"] == "SkeletalMesh":
origin_skeletal_mesh = None
origin_skeleton = None

find_asset = unreal.find_asset(asset_data["animation_skeleton_path"])
if isinstance(find_asset, unreal.Skeleton):
OriginSkeleton = find_asset
origin_skeleton = find_asset
elif isinstance(find_asset, unreal.SkeletalMesh):
OriginSkeleton = find_asset.skeleton
origin_skeletal_mesh = find_asset
origin_skeleton = find_asset.skeleton
else:
OriginSkeleton = None
if OriginSkeleton:
origin_skeleton = None
if origin_skeleton:
pass
#print("Setting skeleton asset: " + OriginSkeleton.get_full_name())
else:
message = "Could not find skeleton at the path: " + asset_data["animation_skeleton_path"]
message = "WARNING: Could not find skeleton." + "\n"
message += '"animation_skeleton_path": ' + asset_data["animation_skeleton_path"]
import_module_unreal_utils.show_warning_message("Skeleton not found.", message)

# docs.unrealengine.com/4.26/en-US/PythonAPI/class/AssetImportTask.html
Expand Down Expand Up @@ -139,25 +145,8 @@ def GetMeshImportData():
GetAlembicImportData().conversion_settings.set_editor_property("rotation", rotation)

# Vertex color
vertex_override_color = None
vertex_color_import_option = None
if additional_data:

vertex_color_import_option = unreal.VertexColorImportOption.REPLACE # Default
if "vertex_color_import_option" in additional_data:
if additional_data["vertex_color_import_option"] == "IGNORE":
vertex_color_import_option = unreal.VertexColorImportOption.IGNORE
elif additional_data["vertex_color_import_option"] == "OVERRIDE":
vertex_color_import_option = unreal.VertexColorImportOption.OVERRIDE
elif additional_data["vertex_color_import_option"] == "REPLACE":
vertex_color_import_option = unreal.VertexColorImportOption.REPLACE

if "vertex_override_color" in additional_data:
vertex_override_color = unreal.LinearColor(
additional_data["vertex_override_color"][0],
additional_data["vertex_override_color"][1],
additional_data["vertex_override_color"][2]
)
vertex_override_color = import_module_unreal_utils.get_vertex_override_color(asset_additional_data)
vertex_color_import_option = import_module_unreal_utils.get_vertex_color_import_option(asset_additional_data)

# #################################[Change]

Expand All @@ -181,12 +170,12 @@ def GetMeshImportData():

else:
if asset_data["asset_type"] == "Animation" or asset_data["asset_type"] == "SkeletalMesh":
if OriginSkeleton:
task.get_editor_property('options').set_editor_property('Skeleton', OriginSkeleton)
if origin_skeleton:
task.get_editor_property('options').set_editor_property('Skeleton', origin_skeleton)
else:
if asset_data["asset_type"] == "Animation":
fail_reason = 'Skeleton ' + asset_data["animation_skeleton_path"] + ' Not found for ' + asset_data["asset_name"] + ' asset.'
return fail_reason
return fail_reason, None, None
else:
print("Skeleton is not set, a new skeleton asset will be created...")

Expand Down Expand Up @@ -250,9 +239,9 @@ def GetMeshImportData():
# ###############[ pre import ]################

# Check is the file alredy exit
if additional_data:
if "preview_import_path" in additional_data:
task_asset_full_path = task.destination_path+"/"+additional_data["preview_import_path"]+"."+additional_data["preview_import_path"]
if asset_additional_data:
if "preview_import_path" in asset_additional_data:
task_asset_full_path = task.destination_path+"/"+asset_additional_data["preview_import_path"]+"."+asset_additional_data["preview_import_path"]
find_asset = unreal.find_asset(task_asset_full_path)
if find_asset:

Expand Down Expand Up @@ -284,18 +273,18 @@ def GetMeshImportData():
unreal.AssetToolsHelpers.get_asset_tools().import_asset_tasks([task])

if len(task.imported_object_paths) > 0:
asset = unreal.find_asset(task.imported_object_paths[0])
asset_path = task.imported_object_paths[0]
asset = unreal.find_asset(asset_path)
else:
asset = None

if asset is None:
fail_reason = 'Error zero imported object for: ' + asset_data["asset_name"]
return fail_reason
return fail_reason, None, None

if asset_data["asset_type"] == "Animation":
# For animation remove the extra mesh
p = task.imported_object_paths[0]
if type(unreal.find_asset(p)) is not unreal.AnimSequence:
if type(asset) is not unreal.AnimSequence:
animAssetName = p.split('.')[0]+'_anim.'+p.split('.')[1]+'_anim'
animAssetNameDesiredPath = p.split('.')[0]+'.'+p.split('.')[1]
animAsset = unreal.find_asset(animAssetName)
Expand All @@ -305,7 +294,7 @@ def GetMeshImportData():
asset = animAsset
else:
fail_reason = 'animAsset ' + asset_data["asset_name"] + ' not found for after inport: ' + animAssetName
return fail_reason
return fail_reason, None, None

# ###############[ Post treatment ]################
asset_import_data = asset.get_editor_property('asset_import_data')
Expand Down Expand Up @@ -341,7 +330,7 @@ def GetMeshImportData():

if asset_data["asset_type"] == "SkeletalMesh":
asset_import_data.set_editor_property('normal_import_method', unreal.FBXNormalImportMethod.FBXNIM_IMPORT_NORMALS_AND_TANGENTS)
if OriginSkeleton is None:
if origin_skeleton is None:
#Unreal create a new skeleton when no skeleton was selected, so addon rename it.
p = task.imported_object_paths[0]
old_skeleton_name = p.split('.')[0]+'_Skeleton.'+p.split('.')[1]+'_Skeleton'
Expand All @@ -355,7 +344,7 @@ def GetMeshImportData():
# Socket
if asset_data["asset_type"] == "SkeletalMesh":
# Import the SkeletalMesh socket(s)
sockets_to_add = additional_data["Sockets"]
sockets_to_add = asset_additional_data["Sockets"]
for socket in sockets_to_add:
old_socket = asset.find_socket(socket["SocketName"])
if old_socket:
Expand All @@ -378,56 +367,15 @@ def GetMeshImportData():
# skeleton.add_socket(new_socket)

# Lod
if asset_data["asset_type"] == "StaticMesh" or asset_data["asset_type"] == "SkeletalMesh":
if asset_data["asset_type"] == "StaticMesh":
unreal.EditorStaticMeshLibrary.remove_lods(asset) # Import the StaticMesh lod(s)

if asset_data["asset_type"] == "SkeletalMesh" or asset_data["asset_type"] == "StaticMesh":

def ImportStaticLod(lod_name, lod_number):
if "LevelOfDetail" in additional_data:
if lod_name in additional_data["LevelOfDetail"]:
lodTask = unreal.AssetImportTask()
lodTask.filename = additional_data["LevelOfDetail"][lod_name]
destination_path = os.path.normpath(asset_data["full_import_path"]).replace('\\', '/')
lodTask.destination_path = destination_path
lodTask.automated = True
lodTask.replace_existing = True

# Set vertex color import settings to replicate base StaticMesh's behaviour
if asset_data["asset_type"] == "Alembic":
lodTask.set_editor_property('options', unreal.AbcImportSettings())
else:
lodTask.set_editor_property('options', unreal.FbxImportUI())

lodTask.get_editor_property('options').static_mesh_import_data.set_editor_property('vertex_color_import_option', vertex_color_import_option)
lodTask.get_editor_property('options').static_mesh_import_data.set_editor_property('vertex_override_color', vertex_override_color.to_rgbe())

unreal.AssetToolsHelpers.get_asset_tools().import_asset_tasks([lodTask])
if len(lodTask.imported_object_paths) > 0:
lodAsset = unreal.find_asset(lodTask.imported_object_paths[0])
slot_replaced = unreal.EditorStaticMeshLibrary.set_lod_from_static_mesh(asset, lod_number, lodAsset, 0, True)
unreal.EditorAssetLibrary.delete_asset(lodTask.imported_object_paths[0])

def ImportSkeletalLod(lod_name, lod_number):
if "LevelOfDetail" in additional_data:
if lod_name in additional_data["LevelOfDetail"]:
# Unreal python no longer support Skeletal mesh LODS import.
pass

if asset_data["asset_type"] == "StaticMesh":
ImportStaticLod("lod_1", 1)
ImportStaticLod("lod_2", 2)
ImportStaticLod("lod_3", 3)
ImportStaticLod("lod_4", 4)
ImportStaticLod("lod_5", 5)

elif asset_data["asset_type"] == "SkeletalMesh":
ImportSkeletalLod("lod_1", 1)
ImportSkeletalLod("lod_2", 2)
ImportSkeletalLod("lod_3", 3)
ImportSkeletalLod("lod_4", 4)
ImportSkeletalLod("lod_5", 5)
if asset_data["asset_type"] == "StaticMesh":
import_module_post_treatment.set_static_mesh_lods(asset, asset_data, asset_additional_data)

if asset_data["asset_type"] == "SkeletalMesh":
import_module_post_treatment.set_skeletal_mesh_lods(asset, asset_data, asset_additional_data)

# Preview mesh
if asset_data["asset_type"] == "Animation":
import_module_post_treatment.set_sequence_preview_skeletal_mesh(asset, origin_skeletal_mesh)

# Vertex color
if vertex_override_color:
Expand All @@ -443,12 +391,7 @@ def ImportSkeletalLod(lod_name, lod_number):



def ImportAllAssets(assets_data):




bfu_unreal_import_location = assets_data['bfu_unreal_import_location']
def ImportAllAssets(assets_data, show_finished_popup=True):
ImportedList = []
ImportFailList = []

Expand Down Expand Up @@ -523,22 +466,23 @@ def PrepareImportAsset(asset_data):
unreal.EditorAssetLibrary.sync_browser_to_objects(PathList)
print('=========================')

if len(ImportFailList) > 0:
message = 'Some asset(s) could not be imported.' + "\n"
else:
message = 'All assets imported with success!' + "\n"

message += "Import finished in " + counter.get_str_time() + "\n"
message += "\n"
for import_row in import_log:
message += import_row + "\n"
if show_finished_popup:
if len(ImportFailList) > 0:
message = 'Some asset(s) could not be imported.' + "\n"
else:
message = 'All assets imported with success!' + "\n"

if len(ImportFailList) > 0:
message += "Import finished in " + counter.get_str_time() + "\n"
message += "\n"
for error in ImportFailList:
message += error + "\n"
for import_row in import_log:
message += import_row + "\n"

if len(ImportFailList) > 0:
message += "\n"
for error in ImportFailList:
message += error + "\n"

title = "Import finished!"
import_module_unreal_utils.show_simple_message(title, message)
title = "Import finished!"
import_module_unreal_utils.show_simple_message(title, message)

return True
4 changes: 2 additions & 2 deletions blender-for-unrealengine/bfu_import_module/asset_import_script.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -60,9 +60,9 @@ def RunImportScriptWithJsonData():

import_module_path = import_assets_data["info"]["import_modiule_path"] # Module to run
imported_module, module_name = load_module(import_module_path)

imported_module.run_asset_import(import_assets_data, False)
unload_module(module_name)
imported_module.run_asset_import(import_assets_data)


if __name__ == "__main__":
RunImportScriptWithJsonData()
Loading

0 comments on commit 2bf29f2

Please sign in to comment.