94 lines
2.3 KiB
C++
94 lines
2.3 KiB
C++
#include "Subsystems/RRDADataAssetSubsystem.h"
|
|
|
|
#include "RefinedRDApi.h"
|
|
#include "DataAssets/RRDADirtDataAsset.h"
|
|
#include "Engine/AssetManager.h"
|
|
|
|
URRDADataAssetSubsystem::URRDADataAssetSubsystem() {}
|
|
|
|
URRDADataAssetSubsystem* URRDADataAssetSubsystem::Get(UObject* Context)
|
|
{
|
|
if (IsValid(Context))
|
|
{
|
|
return Context->GetWorld()->GetSubsystem<URRDADataAssetSubsystem>();
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
URRDADataAssetSubsystem* URRDADataAssetSubsystem::GetChecked(UObject* Context)
|
|
{
|
|
URRDADataAssetSubsystem* Subsystem = Get(Context);
|
|
fgcheck(Subsystem);
|
|
return Subsystem;
|
|
}
|
|
|
|
void URRDADataAssetSubsystem::Initialize(FSubsystemCollectionBase& Collection)
|
|
{
|
|
Super::Initialize(Collection);
|
|
|
|
StartScanForDataAssets();
|
|
}
|
|
|
|
void URRDADataAssetSubsystem::Deinitialize()
|
|
{
|
|
Super::Deinitialize();
|
|
}
|
|
|
|
void URRDADataAssetSubsystem::StartScanForDataAssets()
|
|
{
|
|
UAssetManager& Manager = UAssetManager::Get();
|
|
Manager.LoadPrimaryAssetsWithType(FPrimaryAssetType("RRDADataAsset"));
|
|
|
|
mDisabledDataAssets.Empty();
|
|
mAllDataAssets.Empty();
|
|
|
|
ReCacheBoilerDataAssets();
|
|
ReCacheCoolerDataAssets();
|
|
ReCacheDirtDataAssets();
|
|
ReCacheHeaterDataAssets();
|
|
ReCacheTurbineDataAssets();
|
|
}
|
|
|
|
void URRDADataAssetSubsystem::ReCacheDirtDataAssets()
|
|
{
|
|
mDirtAssets.Empty();
|
|
TSet<URRDADirtDataAsset*> DirtDataAssets;
|
|
if(FindAllDataAssetsOfClass(DirtDataAssets))
|
|
{
|
|
for (URRDADirtDataAsset* DirtDataAsset : DirtDataAssets)
|
|
{
|
|
mDirtAssets.Add(DirtDataAsset->mItem, DirtDataAsset);
|
|
}
|
|
}
|
|
UE_LOG(LogRRDApi, Log, TEXT("ReCacheDirtDataAssets: %d"), mDirtAssets.Num());
|
|
}
|
|
|
|
void URRDADataAssetSubsystem::ReCacheTurbineDataAssets()
|
|
{
|
|
UE_LOG(LogRRDApi, Log, TEXT("ReCacheTurbineDataAssets"));
|
|
}
|
|
|
|
void URRDADataAssetSubsystem::ReCacheBoilerDataAssets() {
|
|
UE_LOG(LogRRDApi, Log, TEXT("ReCacheBoilerDataAssets"));
|
|
}
|
|
|
|
void URRDADataAssetSubsystem::ReCacheCoolerDataAssets() {
|
|
UE_LOG(LogRRDApi, Log, TEXT("ReCacheCoolerDataAssets"));
|
|
}
|
|
|
|
void URRDADataAssetSubsystem::ReCacheHeaterDataAssets() {
|
|
UE_LOG(LogRRDApi, Log, TEXT("ReCacheHeaterDataAssets"));
|
|
}
|
|
|
|
int32 URRDADataAssetSubsystem::GetAllDirtItems(TArray<TSubclassOf<UFGItemDescriptor>>& Items) const
|
|
{
|
|
return mDirtAssets.GetKeys(Items);
|
|
}
|
|
|
|
URRDADirtDataAsset* URRDADataAssetSubsystem::GetDataForDirtItem(TSubclassOf<UFGItemDescriptor> Item) const
|
|
{
|
|
if(!Item) return nullptr;
|
|
return *mDirtAssets.Find(Item);
|
|
}
|
|
|