61 lines
1.6 KiB
C++
61 lines
1.6 KiB
C++
|
// Copyright Epic Games, Inc. All Rights Reserved.
|
|||
|
|
|
|||
|
|
#include "Factory/QuestAssetFactory.h"
|
|||
|
|
#include "ProjectFish/DataAsset/QuestAsset.h"
|
|||
|
|
#include "AssetRegistry/AssetRegistryModule.h"
|
|||
|
|
#include "AssetRegistry/AssetData.h"
|
|||
|
|
|
|||
|
|
UQuestAssetFactory::UQuestAssetFactory()
|
|||
|
|
{
|
|||
|
|
SupportedClass = UQuestAsset::StaticClass();
|
|||
|
|
bCreateNew = true;
|
|||
|
|
bEditAfterNew = true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
UObject* UQuestAssetFactory::FactoryCreateNew(UClass* Class, UObject* InParent, FName Name, EObjectFlags Flags,
|
|||
|
|
UObject* Context, FFeedbackContext* Warn)
|
|||
|
|
{
|
|||
|
|
UQuestAsset* NewQuestAsset = NewObject<UQuestAsset>(InParent, Class, Name, Flags | RF_Transactional);
|
|||
|
|
|
|||
|
|
// 自动生成唯一的 QuestID
|
|||
|
|
if (NewQuestAsset)
|
|||
|
|
{
|
|||
|
|
NewQuestAsset->QuestID = GenerateUniqueQuestID();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return NewQuestAsset;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int32 UQuestAssetFactory::GenerateUniqueQuestID() const
|
|||
|
|
{
|
|||
|
|
int32 MaxQuestID = 0;
|
|||
|
|
|
|||
|
|
// 获取资产注册表模块
|
|||
|
|
FAssetRegistryModule& AssetRegistryModule = FModuleManager::LoadModuleChecked<FAssetRegistryModule>("AssetRegistry");
|
|||
|
|
IAssetRegistry& AssetRegistry = AssetRegistryModule.Get();
|
|||
|
|
|
|||
|
|
// 查找所有 QuestAsset 资产
|
|||
|
|
TArray<FAssetData> AssetDataList;
|
|||
|
|
AssetRegistry.GetAssetsByClass(UQuestAsset::StaticClass()->GetClassPathName(), AssetDataList, true);
|
|||
|
|
|
|||
|
|
// 遍历所有已存在的任务资产，找到最大的 QuestID
|
|||
|
|
for (const FAssetData& AssetData : AssetDataList)
|
|||
|
|
{
|
|||
|
|
if (UQuestAsset* QuestAsset = Cast<UQuestAsset>(AssetData.GetAsset()))
|
|||
|
|
{
|
|||
|
|
if (QuestAsset->QuestID > MaxQuestID)
|
|||
|
|
{
|
|||
|
|
MaxQuestID = QuestAsset->QuestID;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 返回最大ID + 1，如果没有现有任务则从1开始
|
|||
|
|
return MaxQuestID + 1;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
bool UQuestAssetFactory::ShouldShowInNewMenu() const
|
|||
|
|
{
|
|||
|
|
return true;
|
|||
|
|
}
|