unrealUntil.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
import unreal
from MPath import MPath
import unrealLib as ulib
import importlib
importlib.reload(ulib)
ANode = ulib.ANode
UNode = ulib.UNode

e_util = unreal.EditorUtilityLibrary()
a_util = unreal.EditorAssetLibrary()
str_util = unreal.StringLibrary()
sys_util = unreal.SystemLibrary()
l_util = unreal.EditorLevelLibrary()


def show_message(message, level='log'):
    '''
    显示消息的方式 默认两种方式 第一种 在outlog 中打印log 其中logo分三个 级别  对应 log warning error
    第二种方式弹窗的方式 进行显示， 分两种 第一种 只弹确认窗口，第二种弹出 确认与否 窗口 其中 log 方式为 确认窗口，以外为确认与否窗口
    :param message:  需要显示的内容
    :param level: 级别  log warning error 三种
    :param window: 是否开启窗口告知
    :return:
    '''
    dialog = unreal.EditorDialog()
    dialog.show_message(level,message,unreal.AppMsgType.OK)

def progress_dialog(func,iter_obj,*args,task_name=None,force_quit=True):
    """
    带进度提示的 批量处理
    :param func:  需要执行的 功能函数
    :param iter_obj:   迭代对象
    :param args:   功能函数是否包含需要提供的其他参数
    :param task_name:  任务名称 默认会显示在 进度窗口 标题栏
    :param force_quit:  是否可以强制退出
    :return:
    """
    total_task = len(iter_obj)
    task_name = task_name if task_name else func.__name__
    with unreal.ScopedSlowTask(total_task,task_name) as slow_task:
        slow_task.make_dialog(force_quit)
        for index,obj in enumerate(iter_obj):
            if force_quit and slow_task.should_cancel():
                break
            result = func(obj,*args)
            slow_task.enter_progress_frame(int(float(index)/float(total_task)),result)



def ls(outliner=True, sl=False, node_type=''):
    """
    方法 同 maya 如果需要 得到 原始 clarisse 的节点 把 package 设置成 false 即可
    :param outliner: 默认对大纲中的对象进行选择操作
    :param node_type:
    :param sl:
    :return:
    """
    if sl:
        # 获取当前所选择
        if node_type:
            # 找到指定类型
            all_sel_obj = selected(outliner=outliner)
            target_type_obs = [obj for obj in all_sel_obj if obj.type == node_type]
            return target_type_obs
        else:
            return selected(outliner=outliner)

    else:
        # 查找场景所有 对象
        if outliner:
            all_nodes = [UNode(nd) for nd in l_util.get_all_level_actors()]
        else:
            # 获取当前 文件夹下的所有 asset
            all_nodes = [ANode(nd) for nd in a_util.list_assets(sys_util.get_project_content_directory())]
        if node_type:
            # 返回场景中所有指定类型的对象
            res_nds = [nd for nd in all_nodes if nd.type == node_type]
            return res_nds
        else:
            # 返回场景中所有对象
            return all_nodes


def selected(outliner=True, cl=False):
    """
    获取当前所选择的节点
    :param outliner: 是否是大纲中进行选择
    :param cl: clear 清空当前选择
    :param package:
    :return:
    """
    if outliner:
        if cl:
            l_util.select_nothing()
            return []
        else:
            return [UNode(nd) for nd in l_util.get_selected_level_actors()]
    else:
        return [ANode(nd) for nd in e_util.get_selected_assets()]


def listDir(source_dir, recursive=True):
    '''
    查找当前路径下的 子资产 路径可以是系统路径也可以是UE路径
    :param source_dir:
    :param recursive:
    :return:
    '''
    isGamePath = isSysPath(source_dir)

    return [ANode(nd) for nd in a_util.list_assets(source_dir, recursive=recursive)]


def isSysPath(path):
    return False if path.startswith('/Game') else True


def convertSysPathToGamePath(sysPath):
    '''
    转换系统路径到 UE Path  转换过程中需要考虑 资产名称的后缀名格式，以及 uasset 和 umap 得问题
    :param sysPath:
    :return:
    '''
    sysPath = MPath(sysPath)
    if sysPath.exists:
        # 在系统文件目录状态下需要存在
        # 有三种情况
        # 如果是 folder
        if sysPath.is_folder:
            return MPath(sysPath.replace(sys_util.get_project_content_directory(), '/Game/'))
        # 如果是 uasset or umap
        else:
            ue_path = MPath(str(sysPath).replace(str(sys_util.get_project_content_directory()), '/Game/'))
            ue_path = ue_path.parent.child(ue_path.base_name + '.' + ue_path.base_name)
            if a_util.does_asset_exist(ue_path):
                # ue 中可以被识别的文件
                return ue_path
            else:
                # ue 中不能被识别的文件
                unreal.log_error('current uasset unrecognized in unreal. ')
                return None
    else:
        unreal.log_error('current uasset is not exist on disk. ')
        return None


def convertGamePathToSysPath(gamePath):
    '''
    转换UnrealPath 到系统路径
    :param GamePath:
    :return:
    '''
    if a_util.does_asset_exist(gamePath):
        # 首先是需要在系統文件夾下存在 并且能被識別
        asset_data = unreal.AssetData(gamePath)
        print(gamePath)
        if asset_data.get_asset().get_class().get_name() == 'World':
            file_path = MPath(sys_util.get_project_content_directory() + gamePath[6:]).parent.child(
                MPath(gamePath).base_name + '.umap')
            if not file_path.exists:
                # 检查下文件是否在硬盘上是否存在
                unreal.log_warning('{} file is not exists.'.format(file_path))
            return file_path
        else:
            file_path = MPath(sys_util.get_project_content_directory() + gamePath[6:]).parent.child(
                MPath(gamePath).base_name + '.uasset')
            if not file_path.exists:
                # 检查下文件是否在硬盘上是否存在
                unreal.log_warning('{} file is not exists.'.format(file_path))
            return file_path
    elif a_util.does_directory_exist(gamePath):
        # 传入进来的路径为 文件夹路径
        folder_path = MPath(sys_util.get_project_content_directory() + gamePath[6:])
        if not folder_path.exists:
            # 检查下当前文件夹是否在硬盘上存在
            unreal.log_warning('{} folder is not exists.'.format(folder_path))
        return folder_path
    else:
        unreal.log_error('current {} path invaild.'.format(gamePath))
        return None


def removeUnuseAsset():
    """
    移除未使用的资产
    :return:
    """
    pass


def createNode(node_type, node_name=None, parent=None):
    """
    创建 node
    :param node_type:
    :param node_name:
    :param parent: <clarisseNode>
    :param package:
    :return:
    """
    pass


def listMethod(nd, r_name='', show_help=False):
    for i in dir(nd):
        if r_name:
            if r_name.lower() in i.lower():
                print(i)
                if show_help:
                    help(getattr(nd, i))
            continue
        print(i)
        if show_help:
            help(getattr(nd, i))


def debug_function():
    """
    用于调试代码的函数
    :return:
    """
    print("debug_function exec!!!")
    # path = sys_util.get_project_content_directory() # 返回文件夹结构的系统路径
    nd = selected(outliner=False)[0]
    # print(nd.node_asset)
    # listMethod(unreal.FilePath)
    # listMethod(nd.node)
    unreal.log(nd.node.get_name())

    # unreal.log(dp.convert_relative_path_to_full(nd.node_path))

    # unreal.log(a_util.list_assets(node_path, recursive=False))
    # if nd:
    #     unreal.log(nd.node_path)
    #     nd_data = unreal.AssetData(nd.node_path)
    #     unreal.log(nd_data)

    # cbd = unreal.ContentBrowserDataSubsystem()
    # ds = cbd.get_active_data_sources()

    # print(a_util.does_directory_exist('/Game/VirtualProduction/testLevel'))
    # listMethod(unreal, 'main')
    # print(type(nd))

    # l_util.load_level("/Game/")
    # bp_node = a_util.load_blueprint_class(nd.node_path)
    # a_util.save_loaded_assets()
    # l_util.spawn_actor_from_class(bp_node,unreal.Vector(),unreal.Rotator())
    # nd ='/Game/VitrualProduction/testLevel.testLevel'
    # l_util.load_level(nd)
    # nd = l_util.get_selected_level_actors()[0]
    # nd = l_util.get_actor_reference(l_util.get_selected_level_actors()[0])

    # unreal.log(nd.get_class().get_name())
    # listMethod(nd.get_class())

    pass