117 lines
3.6 KiB
Python
117 lines
3.6 KiB
Python
|
import os
|
||
|
|
import sys
|
||
|
|
from ctypes import *
|
||
|
|
|
||
|
|
from DHI.core.util import MayaUtil
|
||
|
|
from maya import cmds
|
||
|
|
|
||
|
|
|
||
|
|
def resolve_python_to_use(version):
|
||
|
|
pydna_version_mapping = {
|
||
|
|
'2022': 'python3',
|
||
|
|
'2023': 'python397',
|
||
|
|
'2024': 'python3108',
|
||
|
|
'2025': 'python311',
|
||
|
|
}
|
||
|
|
return pydna_version_mapping[version] or 'python3'
|
||
|
|
|
||
|
|
|
||
|
|
def _resolve_running_platform():
|
||
|
|
import platform
|
||
|
|
platform = platform.system()
|
||
|
|
|
||
|
|
if platform not in ["Windows", "Linux"]:
|
||
|
|
platform = "Windows"
|
||
|
|
|
||
|
|
print(f"Running on {platform} ")
|
||
|
|
return platform
|
||
|
|
|
||
|
|
|
||
|
|
def _load_pydna(current_folder, loaded_platform, maya_version):
|
||
|
|
"""
|
||
|
|
Initialize required libraries based on Maya version
|
||
|
|
Returns
|
||
|
|
"""
|
||
|
|
python_version = resolve_python_to_use(maya_version)
|
||
|
|
pydna_folder = os.path.join(current_folder, "plugins", loaded_platform, "pydna", python_version)
|
||
|
|
|
||
|
|
if pydna_folder not in sys.path:
|
||
|
|
print(f"Adding PYDNA to path {pydna_folder}")
|
||
|
|
sys.path.append(pydna_folder)
|
||
|
|
|
||
|
|
if loaded_platform == "Linux":
|
||
|
|
print("Setting LD_LIBRARY_PATH to %s" % pydna_folder)
|
||
|
|
try:
|
||
|
|
os.environ["LD_LIBRARY_PATH"] = pydna_folder
|
||
|
|
cdll.LoadLibrary(os.path.join(pydna_folder, "libdna.so.7.1.0"))
|
||
|
|
except Exception as exc:
|
||
|
|
print("Failed to load library:", exc)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
|
||
|
|
def _load_dna_calib(current_folder, loaded_platform, maya_version):
|
||
|
|
pydnac_folder = os.path.join(current_folder, "plugins", loaded_platform, maya_version)
|
||
|
|
|
||
|
|
if pydnac_folder not in sys.path:
|
||
|
|
print(f"Adding PYDNACalib to path {pydnac_folder}")
|
||
|
|
sys.path.append(pydnac_folder)
|
||
|
|
|
||
|
|
if loaded_platform == "Linux":
|
||
|
|
print("Setting LD_LIBRARY_PATH to %s" % pydnac_folder)
|
||
|
|
try:
|
||
|
|
os.environ["LD_LIBRARY_PATH"] = pydnac_folder
|
||
|
|
cdll.LoadLibrary(os.path.join(pydnac_folder, "libdnacalib.so.6"))
|
||
|
|
except Exception as exc:
|
||
|
|
print("Failed to load library:", exc)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
|
||
|
|
def _load_rig_logic(current_folder, platform, maya_version):
|
||
|
|
try:
|
||
|
|
rl_plugin_name = "libembeddedRL4" if platform == "Linux" else "embeddedRL4"
|
||
|
|
|
||
|
|
MayaUtil.load_plugin(
|
||
|
|
rl_plugin_name,
|
||
|
|
os.path.join(current_folder, "plugins", platform, maya_version),
|
||
|
|
platform,
|
||
|
|
)
|
||
|
|
except Exception as ex:
|
||
|
|
print("Failed to load rig logic plugin:", ex)
|
||
|
|
|
||
|
|
|
||
|
|
def _load_rbf(current_folder, platform, maya_version):
|
||
|
|
try:
|
||
|
|
MayaUtil.load_plugin(
|
||
|
|
"MayaUE4RBFPlugin" + maya_version,
|
||
|
|
os.path.join(current_folder, "plugins", platform, maya_version),
|
||
|
|
platform,
|
||
|
|
".mll",
|
||
|
|
)
|
||
|
|
MayaUtil.load_plugin(
|
||
|
|
"MayaUERBFPlugin",
|
||
|
|
os.path.join(current_folder, "plugins", platform, maya_version),
|
||
|
|
platform,
|
||
|
|
".mll",
|
||
|
|
)
|
||
|
|
except Exception as ex:
|
||
|
|
print("Failed to load plugins:", ex)
|
||
|
|
|
||
|
|
|
||
|
|
def load():
|
||
|
|
platform = _resolve_running_platform()
|
||
|
|
maya_version = cmds.about(version=True)
|
||
|
|
print(f"Running Maya {maya_version} on {platform}")
|
||
|
|
|
||
|
|
current_folder = os.path.dirname(os.path.abspath(__file__))
|
||
|
|
print(f"Current folder: {current_folder}")
|
||
|
|
|
||
|
|
if current_folder not in sys.path:
|
||
|
|
print(f"Adding current folder: {current_folder} to sys.path...")
|
||
|
|
sys.path.append(current_folder)
|
||
|
|
|
||
|
|
print("Loading plugins...")
|
||
|
|
_load_pydna(current_folder, platform, maya_version)
|
||
|
|
_load_dna_calib(current_folder, platform, maya_version)
|
||
|
|
_load_rig_logic(current_folder, platform, maya_version)
|
||
|
|
_load_rbf(current_folder, platform, maya_version)
|