87 lines
2.5 KiB
C++
87 lines
2.5 KiB
C++
// Copyright (c) 2012-2024 Wojciech Figat. All rights reserved.
|
|
|
|
#include "CollectAssetsStep.h"
|
|
#include "Engine/Content/Content.h"
|
|
#include "Engine/Content/Asset.h"
|
|
#include "Engine/Content/AssetReference.h"
|
|
#include "Engine/Content/Assets/Texture.h"
|
|
#include "Engine/Core/Log.h"
|
|
#include "Engine/Content/Assets/CubeTexture.h"
|
|
#include "Engine/Content/Assets/Shader.h"
|
|
#include "Engine/Content/Cache/AssetsCache.h"
|
|
|
|
bool CollectAssetsStep::Process(CookingData& data, Asset* asset)
|
|
{
|
|
// Skip virtual/temporary assets
|
|
if (asset->IsVirtual())
|
|
return false;
|
|
|
|
// Keep reference to the asset
|
|
AssetReference<Asset> ref(asset);
|
|
|
|
// Asset should have loaded data
|
|
if (asset->WaitForLoaded())
|
|
return false;
|
|
|
|
// Gather asset references
|
|
_references.Clear();
|
|
asset->Locker.Lock();
|
|
asset->GetReferences(_references);
|
|
asset->Locker.Unlock();
|
|
_assetsQueue.Add(_references);
|
|
|
|
return false;
|
|
}
|
|
|
|
bool CollectAssetsStep::Perform(CookingData& data)
|
|
{
|
|
LOG(Info, "Searching for assets to include in a build. Using {0} root assets.", data.RootAssets.Count());
|
|
data.StepProgress(TEXT("Collecting assets"), 0);
|
|
|
|
// Initialize assets queue
|
|
_assetsQueue.Clear();
|
|
_assetsQueue.EnsureCapacity(1024);
|
|
for (auto i = data.RootAssets.Begin(); i.IsNotEnd(); ++i)
|
|
_assetsQueue.Add(i->Item);
|
|
|
|
// Iterate through the assets graph
|
|
AssetInfo assetInfo;
|
|
while (_assetsQueue.HasItems())
|
|
{
|
|
BUILD_STEP_CANCEL_CHECK;
|
|
|
|
const auto assetId = _assetsQueue.Dequeue();
|
|
|
|
// Skip already processed or invalid assets
|
|
if (!assetId.IsValid()
|
|
|| !Content::GetRegistry()->FindAsset(assetId, assetInfo)
|
|
|| data.Assets.Contains(assetId))
|
|
continue;
|
|
|
|
// Skip some assets (with no refs and not required to load)
|
|
if (assetInfo.TypeName == Texture::TypeName ||
|
|
assetInfo.TypeName == CubeTexture::TypeName ||
|
|
assetInfo.TypeName == Shader::TypeName)
|
|
{
|
|
LOG_STR(Info, assetInfo.Path);
|
|
data.Assets.Add(assetId);
|
|
continue;
|
|
}
|
|
|
|
// Load asset
|
|
const auto asset = Content::LoadAsync<Asset>(assetId);
|
|
if (asset == nullptr)
|
|
continue;
|
|
|
|
// Process that asset
|
|
LOG_STR(Info, asset->GetPath());
|
|
data.Assets.Add(assetId);
|
|
Process(data, asset);
|
|
}
|
|
|
|
data.Stats.TotalAssets = data.Assets.Count();
|
|
LOG(Info, "Found {0} assets to deploy!", data.Assets.Count());
|
|
|
|
return false;
|
|
}
|