UE4/5资产管理插件 UAssetBrowser

// Copyright 2017-2021 marynate. All Rights Reserved. #include "ExtAssetData.h" #include "ExtContentBrowser.h" #include "ExtContentBrowserSingleton.h" #include "ExtPackageUtils.h" #include "ExtContentBrowserUtils.h" #include "SExtContentBrowser.h" #include "SContentBrowserPathPicker.h" #include "HAL/FileManager.h" #include "Misc/ConfigCacheIni.h" #include "Misc/PackageName.h" #include "Misc/ScopedSlowTask.h" #include "AssetRegistry/AssetRegistryState.h" #include "AssetRegistry/AssetDataTagMap.h" #include "Engine/AssetManager.h" #include "Engine/Blueprint.h" #include "Engine/UserDefinedStruct.h" #include "Engine/UserDefinedEnum.h" #include "AssetRegistry/ARFilter.h" #include "ObjectTools.h" #include "Blueprint/BlueprintSupport.h" #include "Framework/Application/SlateApplication.h" #include "ContentBrowserModule.h" #include "UObject/UObjectIterator.h" #include "HAL/RunnableThread.h" #include "PackageTools.h" #include "AssetToolsModule.h" #include "FileHelpers.h" #include "UObject/CoreRedirects.h" #include "Misc/RedirectCollector.h" #include "Serialization/ArchiveReplaceObjectRef.h" #include "Engine/MapBuildDataRegistry.h" #include "Engine/PrimaryAssetLabel.h" #include "Async/Async.h" #include "Interfaces/IMainFrameModule.h" #include "CollectionManagerTypes.h" #include "CollectionManagerModule.h" #include "ICollectionManager.h" #include "DesktopPlatformModule.h" #include "FileUtilities/ZipArchiveWriter.h" #include "HAL/PlatformFilemanager.h" #include "Misc/FileHelper.h" #include "CollectionManagerModule.h" #include "InstancedReferenceSubobjectHelper.h" #include "Kismet2/BlueprintEditorUtils.h" #include "UObject/MetaData.h" #include "IContentBrowserDataModule.h" // Serialization #include "AssetRegistry/AssetDataTagMapSerializationDetails.h" #include "Serialization/LargeMemoryWriter.h" #include "Serialization/ArchiveProxy.h" #include "UObject/NameBatchSerialization.h" #if ECB_WIP_IMPORT_FOLDER_COLOR_OVERRIDE #include "AssetViewUtils.h" #endif // Drag and Drop #include "Input/DragAndDrop.h" #include "LevelEditorViewport.h" #define LOCTEXT_NAMESPACE "ExtContentBrowser" const FGuid FExtAssetRegistryVersion::GUID(0xDC449DF5, 0x817F4F19, 0x899A33C9, 0x0AB230B4); FCustomVersionRegistration GRegisterExtAssetRegistryCustomVersion(FExtAssetRegistryVersion::GUID, FExtAssetRegistryVersion::LatestVersion, TEXT("ExtAssetRegistryVer")); FString FExtAssetSupport::AssetPackageExtension = TEXT(".uasset"); FString FExtAssetSupport::MapPackageExtension = TEXT(".umap"); FString FExtAssetSupport::TextAssetPackageExtension = TEXT(".utxt"); FString FExtAssetSupport::TextMapPackageExtension = TEXT(".utxtmap"); namespace FPathsUtil { FString WithEndSlash(const FString& InPath) { FString PathWithEndSlash(InPath); FPaths::NormalizeDirectoryName(PathWithEndSlash); PathWithEndSlash.Append(TEXT("/")); return PathWithEndSlash; } bool IsSubOrSamePath(const FString& InSubPath, const FString& InRootPath) { FString SubPath(InSubPath); FPaths::NormalizeDirectoryName(SubPath); FString RootPath(InRootPath); FPaths::NormalizeDirectoryName(RootPath); if (SubPath.Equals(RootPath)) { return true; } RootPath.Append(TEXT("/")); if (SubPath.StartsWith(RootPath)) { return true; } return false; } bool IsFileInDir(const FString& InFile, const FString& InRootPath) { FString RootPath = WithEndSlash(InRootPath); return (InFile.StartsWith(RootPath)); } // return: num of paths been merged int32 SortAndMergeDirs(const TArray<FString>& InPathsToCombine, TArray<FString>& OutCombinedPaths) { TArray<FString> PathsToCombine = InPathsToCombine; PathsToCombine.Sort(); const int32 NumPaths = PathsToCombine.Num(); TArray<int32> IndicesToRemove; for (int32 IndexA = 0; IndexA < NumPaths - 1; ++IndexA) { FString& A = PathsToCombine[IndexA]; for (int32 IndexB = NumPaths - 1; IndexB > IndexA; --IndexB) { if (IndicesToRemove.Contains(IndexB)) { continue; } FString& B = PathsToCombine[IndexB]; //if (B.StartsWith(A, ESearchCase::IgnoreCase)) // todo: Mac -> Casesensitive if (FPathsUtil::IsSubOrSamePath(B, A)) { IndicesToRemove.Add(IndexB); } } } if (IndicesToRemove.Num() > 0) { for (int32 Index = 0; Index < IndicesToRemove.Num(); ++Index) { PathsToCombine.RemoveAt(IndicesToRemove[Index]); } } OutCombinedPaths = PathsToCombine; return IndicesToRemove.Num(); } } namespace FExtAssetCoreUtil { bool FExtAssetFeedbackContext::ReceivedUserCancel() { if (!bTaskWasCancelledCache) { bTaskWasCancelledCache = FExtFeedbackContextEditor::ReceivedUserCancel(); } return bTaskWasCancelledCache; } /////////////////////////////////////////// FExtAssetWorkReporter::FExtAssetWorkReporter(const TSharedPtr<IExtAssetProgressReporter>& InReporter, const FText& InDescription, float InAmountOfWork, float InIncrementOfWork, bool bInterruptible) : Reporter(InReporter) , DefaultIncrementOfWork(InIncrementOfWork) { if (Reporter.IsValid()) { Reporter->BeginWork(InDescription, InAmountOfWork, bInterruptible); } } FExtAssetWorkReporter::~FExtAssetWorkReporter() { if (Reporter.IsValid()) { Reporter->EndWork(); } } void FExtAssetWorkReporter::ReportNextStep(const FText& InMessage, float InIncrementOfWork) { if (Reporter.IsValid()) { Reporter->ReportProgress(InIncrementOfWork, InMessage); } } void FExtAssetWorkReporter::ReportNextStep(const FText& InMessage) { ReportNextStep(InMessage, DefaultIncrementOfWork); } bool FExtAssetWorkReporter::IsWorkCancelled() const { if (Reporter.IsValid()) { return Reporter->IsWorkCancelled(); } return false; } /////////////////////////////////////////// void FExtAssetProgressUIReporter::BeginWork(const FText& InTitle, float InAmountOfWork, bool bInterruptible/* = true*/) { ProgressTasks.Emplace(new FScopedSlowTask(InAmountOfWork, InTitle, true, FeedbackContext.IsValid() ? *FeedbackContext.Get() : *GWarn)); ProgressTasks.Last()->MakeDialog(bInterruptible); } void FExtAssetProgressUIReporter::EndWork() { if (ProgressTasks.Num() > 0) { ProgressTasks.Pop(); } } void FExtAssetProgressUIReporter::ReportProgress(float Progress, const FText& InMessage) { if (ProgressTasks.Num() > 0) { TSharedPtr<FScopedSlowTask>& ProgressTask = ProgressTasks.Last(); ProgressTask->EnterProgressFrame(Progress, InMessage); } } bool FExtAssetProgressUIReporter::IsWorkCancelled() { if (!bIsCancelled && ProgressTasks.Num() > 0) { const TSharedPtr<FScopedSlowTask>& ProgressTask = ProgressTasks.Last(); bIsCancelled |= ProgressTask->ShouldCancel(); } return bIsCancelled; } FFeedbackContext* FExtAssetProgressUIReporter::GetFeedbackContext() const { return FeedbackContext.IsValid() ? FeedbackContext.Get() : GWarn; } /////////////////////////////////////////// void FExtAssetProgressTextReporter::BeginWork(const FText& InTitle, float InAmountOfWork, bool bInterruptible/* = true*/) { ECB_LOG(Display, TEXT("Start: %s ..."), *InTitle.ToString()); ++TaskDepth; } void FExtAssetProgressTextReporter::EndWork() { if (TaskDepth > 0) { --TaskDepth; } } void FExtAssetProgressTextReporter::ReportProgress(float Progress, const FText& InMessage) { if (TaskDepth > 0) { ECB_LOG(Display, TEXT("Doing %s ..."), *InMessage.ToString()); } } bool FExtAssetProgressTextReporter::IsWorkCancelled() { return false; } FFeedbackContext* FExtAssetProgressTextReporter::GetFeedbackContext() const { return FeedbackContext.Get(); } } //////////////////////////////////////////////////// // FExtConfigUtil // struct FExtConfigUtil { static FString GetConfigDirByContentRoot(const FString& InAssetContentRoot) { static FString EmptyString(""); FString ConfigDir = InAssetContentRoot / TEXT("../Saved/Config/") / ANSI_TO_TCHAR(FPlatformProperties::PlatformName()); FPaths::CollapseRelativeDirectories(ConfigDir); if (IFileManager::Get