123 lines
4.5 KiB
C++
123 lines
4.5 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "Factories/MaterialImportHelpers.h"
|
|
#include "AssetRegistryModule.h"
|
|
#include "AssetData.h"
|
|
#include "ARFilter.h"
|
|
#include "Materials/MaterialInterface.h"
|
|
#include "Misc/Paths.h"
|
|
|
|
UMaterialInterface* UMaterialImportHelpers::FindExistingMaterialFromSearchLocation(const FString& MaterialFullName, const FString& BasePackagePath, EMaterialSearchLocation SearchLocation, FText& OutError)
|
|
{
|
|
if (SearchLocation == EMaterialSearchLocation::DoNotSearch)
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
// Search in memory
|
|
constexpr bool bExactClass = false;
|
|
UMaterialInterface* FoundMaterial = FindObject<UMaterialInterface>(nullptr, *MaterialFullName, bExactClass);
|
|
|
|
if (FoundMaterial == nullptr)
|
|
{
|
|
FString SearchPath = FPaths::GetPath(BasePackagePath);
|
|
|
|
// Search in asset's local folder
|
|
FoundMaterial = FindExistingMaterial(SearchPath, MaterialFullName, false, OutError);
|
|
|
|
// Search recursively in asset's folder
|
|
if (FoundMaterial == nullptr &&
|
|
(SearchLocation != EMaterialSearchLocation::Local))
|
|
{
|
|
FoundMaterial = FindExistingMaterial(SearchPath, MaterialFullName, true, OutError);
|
|
}
|
|
|
|
if (FoundMaterial == nullptr &&
|
|
(SearchLocation == EMaterialSearchLocation::UnderParent ||
|
|
SearchLocation == EMaterialSearchLocation::UnderRoot ||
|
|
SearchLocation == EMaterialSearchLocation::AllAssets))
|
|
{
|
|
// Search recursively in parent's folder
|
|
SearchPath = FPaths::GetPath(SearchPath);
|
|
if (!SearchPath.IsEmpty())
|
|
{
|
|
FoundMaterial = FindExistingMaterial(SearchPath, MaterialFullName, true, OutError);
|
|
}
|
|
}
|
|
if (FoundMaterial == nullptr &&
|
|
(SearchLocation == EMaterialSearchLocation::UnderRoot ||
|
|
SearchLocation == EMaterialSearchLocation::AllAssets))
|
|
{
|
|
// Search recursively in root folder of asset
|
|
FString OutPackageRoot, OutPackagePath, OutPackageName;
|
|
FPackageName::SplitLongPackageName(SearchPath, OutPackageRoot, OutPackagePath, OutPackageName);
|
|
if (!SearchPath.IsEmpty())
|
|
{
|
|
FoundMaterial = FindExistingMaterial(OutPackageRoot, MaterialFullName, true, OutError);
|
|
}
|
|
}
|
|
if (FoundMaterial == nullptr &&
|
|
SearchLocation == EMaterialSearchLocation::AllAssets)
|
|
{
|
|
// Search everywhere
|
|
FoundMaterial = FindExistingMaterial(TEXT("/"), MaterialFullName, true, OutError);
|
|
}
|
|
}
|
|
|
|
return FoundMaterial;
|
|
}
|
|
|
|
UMaterialInterface* UMaterialImportHelpers::FindExistingMaterial(const FString& BasePath, const FString& MaterialFullName, const bool bRecursivePaths, FText& OutError)
|
|
{
|
|
UMaterialInterface* Material = nullptr;
|
|
|
|
FAssetRegistryModule& AssetRegistryModule = FModuleManager::LoadModuleChecked<FAssetRegistryModule>("AssetRegistry");
|
|
IAssetRegistry& AssetRegistry = AssetRegistryModule.Get();
|
|
|
|
// Finish/update any scans
|
|
TArray<FString> ScanPaths;
|
|
if (BasePath.IsEmpty() || BasePath == TEXT("/"))
|
|
{
|
|
FPackageName::QueryRootContentPaths(ScanPaths);
|
|
}
|
|
else
|
|
{
|
|
ScanPaths.Add(BasePath);
|
|
}
|
|
const bool bForceRescan = false;
|
|
AssetRegistry.ScanPathsSynchronous(ScanPaths, bForceRescan);
|
|
|
|
FARFilter Filter;
|
|
Filter.bRecursiveClasses = true;
|
|
Filter.bRecursivePaths = bRecursivePaths;
|
|
Filter.ClassNames.Add(UMaterialInterface::StaticClass()->GetFName());
|
|
Filter.PackagePaths.Add(FName(*BasePath));
|
|
|
|
TArray<FAssetData> AssetData;
|
|
AssetRegistry.GetAssets(Filter, AssetData);
|
|
|
|
TArray<UMaterialInterface*> FoundMaterials;
|
|
for (const FAssetData& Data: AssetData)
|
|
{
|
|
if (Data.AssetName == FName(*MaterialFullName))
|
|
{
|
|
Material = Cast<UMaterialInterface>(Data.GetAsset());
|
|
if (Material != nullptr)
|
|
{
|
|
FoundMaterials.Add(Material);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (FoundMaterials.Num() > 1)
|
|
{
|
|
check(Material != nullptr);
|
|
OutError =
|
|
FText::Format(NSLOCTEXT("MaterialImportHelpers", "MultipleMaterialsFound", "Found {0} materials matching name '{1}'. Using '{2}'."),
|
|
FText::FromString(FString::FromInt(FoundMaterials.Num())),
|
|
FText::FromString(MaterialFullName),
|
|
FText::FromString(Material->GetOutermost()->GetName()));
|
|
}
|
|
return Material;
|
|
}
|