Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Maya Preset for alembic loaders #16

Merged
merged 4 commits into from
Jul 9, 2024
Merged

Maya Preset for alembic loaders #16

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
985f697
maya preset for alembic importer
moonyuet Jul 9, 2024
dbaaa99
ruff cosmetic fix
moonyuet Jul 9, 2024
2411189
ruff cosmetic fix
moonyuet Jul 9, 2024
47056ed
ruff cosmetic fix
moonyuet Jul 9, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
49 changes: 36 additions & 13 deletions client/ayon_unreal/plugins/load/load_skeletalmesh_abc.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
get_representation_path,
AYON_CONTAINER_ID
)
from ayon_core.lib import EnumDef
from ayon_unreal.api import plugin
from ayon_unreal.api.pipeline import (
AYON_ASSET_DIR,
Expand All @@ -26,8 +27,23 @@ class SkeletalMeshAlembicLoader(plugin.Loader):

root = AYON_ASSET_DIR

@classmethod
def get_options(cls, contexts):
return [
EnumDef(
"abc_conversion_preset",
label="Alembic Conversion Preset",
items={
"custom": "custom",
"maya": "maya"
},
default="maya"
)
]


@staticmethod
def get_task(filename, asset_dir, asset_name, replace, default_conversion):
def get_task(filename, asset_dir, asset_name, replace, loaded_options):
task = unreal.AssetImportTask()
options = unreal.AbcImportSettings()
conversion_settings = unreal.AbcConversionSettings(
Expand All @@ -46,12 +62,18 @@ def get_task(filename, asset_dir, asset_name, replace, default_conversion):
options.set_editor_property(
'import_type', unreal.AlembicImportType.SKELETAL)

if not default_conversion:
conversion_settings = unreal.AbcConversionSettings(
preset=unreal.AbcConversionPreset.CUSTOM,
flip_u=False, flip_v=False,
rotation=[0.0, 0.0, 0.0],
scale=[1.0, 1.0, 1.0])
if not loaded_options.get("default_conversion"):
conversion_settings = None
abc_conversion_preset = loaded_options.get("abc_conversion_preset")
if abc_conversion_preset == "maya":
conversion_settings = unreal.AbcConversionSettings(
preset= unreal.AbcConversionPreset.MAYA)
else:
conversion_settings = unreal.AbcConversionSettings(
preset=unreal.AbcConversionPreset.CUSTOM,
flip_u=False, flip_v=False,
rotation=[0.0, 0.0, 0.0],
scale=[1.0, 1.0, 1.0])
options.conversion_settings = conversion_settings

task.options = options
Expand All @@ -60,12 +82,12 @@ def get_task(filename, asset_dir, asset_name, replace, default_conversion):

def import_and_containerize(
self, filepath, asset_dir, asset_name, container_name,
default_conversion=False
loaded_options
):
unreal.EditorAssetLibrary.make_directory(asset_dir)

task = self.get_task(
filepath, asset_dir, asset_name, False, default_conversion)
filepath, asset_dir, asset_name, False, loaded_options)

unreal.AssetToolsHelpers.get_asset_tools().import_asset_tasks([task])

Expand Down Expand Up @@ -125,9 +147,10 @@ def load(self, context, name, namespace, options):
else:
name_version = f"{name}_v{version:03d}"

default_conversion = False
if options.get("default_conversion"):
default_conversion = options.get("default_conversion")
loaded_options = {
"default_conversion": options.get("default_conversion", False),
"abc_conversion_preset": options.get("abc_conversion_preset", "maya")
}

tools = unreal.AssetToolsHelpers().get_asset_tools()
asset_dir, container_name = tools.create_unique_asset_name(
Expand All @@ -139,7 +162,7 @@ def load(self, context, name, namespace, options):
path = self.filepath_from_context(context)

self.import_and_containerize(path, asset_dir, asset_name,
container_name, default_conversion)
container_name, loaded_options)

product_type = context["product"]["productType"]
self.imprint(
Expand Down
53 changes: 37 additions & 16 deletions client/ayon_unreal/plugins/load/load_staticmesh_abc.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
create_container,
imprint,
)
from ayon_core.lib import EnumDef
import unreal # noqa


Expand All @@ -26,8 +27,22 @@ class StaticMeshAlembicLoader(plugin.Loader):

root = AYON_ASSET_DIR

@classmethod
def get_options(cls, contexts):
return [
EnumDef(
"abc_conversion_preset",
label="Alembic Conversion Preset",
items={
"custom": "custom",
"maya": "maya"
},
default="maya"
)
]

@staticmethod
def get_task(filename, asset_dir, asset_name, replace, default_conversion):
def get_task(filename, asset_dir, asset_name, replace, loaded_options):
task = unreal.AssetImportTask()
options = unreal.AbcImportSettings()
sm_settings = unreal.AbcStaticMeshSettings()
Expand All @@ -46,12 +61,18 @@ def get_task(filename, asset_dir, asset_name, replace, default_conversion):

sm_settings.set_editor_property('merge_meshes', True)

if not default_conversion:
conversion_settings = unreal.AbcConversionSettings(
preset=unreal.AbcConversionPreset.CUSTOM,
flip_u=False, flip_v=False,
rotation=[0.0, 0.0, 0.0],
scale=[1.0, 1.0, 1.0])
if not loaded_options.get("default_conversion"):
conversion_settings = None
abc_conversion_preset = loaded_options.get("abc_conversion_preset")
if abc_conversion_preset == "maya":
conversion_settings = unreal.AbcConversionSettings(
preset= unreal.AbcConversionPreset.MAYA)
else:
conversion_settings = unreal.AbcConversionSettings(
preset=unreal.AbcConversionPreset.CUSTOM,
flip_u=False, flip_v=False,
rotation=[0.0, 0.0, 0.0],
scale=[1.0, 1.0, 1.0])
options.conversion_settings = conversion_settings

options.static_mesh_settings = sm_settings
Expand All @@ -61,12 +82,12 @@ def get_task(filename, asset_dir, asset_name, replace, default_conversion):

def import_and_containerize(
self, filepath, asset_dir, asset_name, container_name,
default_conversion=False
loaded_options
):
unreal.EditorAssetLibrary.make_directory(asset_dir)

task = self.get_task(
filepath, asset_dir, asset_name, False, default_conversion)
filepath, asset_dir, asset_name, False, loaded_options)

unreal.AssetToolsHelpers.get_asset_tools().import_asset_tasks([task])

Expand Down Expand Up @@ -116,7 +137,7 @@ def load(self, context, name, namespace, options):
"""
# Create directory for asset and Ayon container
folder_path = context["folder"]["path"]
folder_name = context["folder"]["path"]
folder_name = context["folder"]["name"]

suffix = "_CON"
asset_name = f"{folder_name}_{name}" if folder_name else f"{name}"
Expand All @@ -126,10 +147,10 @@ def load(self, context, name, namespace, options):
name_version = f"{name}_hero"
else:
name_version = f"{name}_v{version:03d}"

default_conversion = False
if options.get("default_conversion"):
default_conversion = options.get("default_conversion")
loaded_options = {
"default_conversion": options.get("default_conversion", False),
"abc_conversion_preset": options.get("abc_conversion_preset", "maya")
}

tools = unreal.AssetToolsHelpers().get_asset_tools()
asset_dir, container_name = tools.create_unique_asset_name(
Expand All @@ -141,7 +162,7 @@ def load(self, context, name, namespace, options):
path = self.filepath_from_context(context)

self.import_and_containerize(path, asset_dir, asset_name,
container_name, default_conversion)
container_name, loaded_options)

product_type = context["product"]["productType"]
self.imprint(
Expand All @@ -156,7 +177,7 @@ def load(self, context, name, namespace, options):
asset_content = unreal.EditorAssetLibrary.list_assets(
asset_dir, recursive=True, include_folder=False
)

unreal.log(asset_content)
for a in asset_content:
unreal.EditorAssetLibrary.save_asset(a)

Expand Down