izzy/UnrealEngineUWP
0
0
Fork 0
mirror of https://github.com/izzy2lost/UnrealEngineUWP.git synced 2026-03-26 18:15:20 -07:00
Code Issues Packages Projects Releases Wiki Activity

U5 - Move TextureCompression into ParallelFor

Browse Source 
[CL 15378859 by Arne Schober in ue5-main branch]
This commit is contained in:
Arne Schober
2021-02-10 15:26:28 -04:00
parent 81656650be
commit df09d6779f
3 changed files with 63 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
Engine/Source/Developer/TextureFormatDXT/Private/TextureFormatDXT.cpp
View File

@@ -13,6 +13,8 @@
#include "TextureCompressorModule.h"
#include "PixelFormat.h"
#include "EngineLogs.h"
#include "Async/ParallelFor.h"
THIRD_PARTY_INCLUDES_START
#include "nvtt/nvtt.h"
THIRD_PARTY_INCLUDES_END
@@ -231,7 +233,7 @@ public:
/**
* Asynchronous NVTT worker.
*/
class FAsyncNVTTWorker : public FNonAbandonableTask
class FAsyncNVTTWorker
{
public:
/**
@@ -249,11 +251,6 @@ public:
bCompressionResults = Compressor->Compress();
}
FORCEINLINE TStatId GetStatId() const
{
RETURN_QUICK_DECLARE_CYCLE_STAT(FAsyncNVTTWorker, STATGROUP_ThreadPoolAsyncTasks);
}
/** Retrieve compression results. */
bool GetCompressionResults() const { return bCompressionResults; }
@@ -263,7 +260,6 @@ private:
/** true if compression was successful. */
bool bCompressionResults;
};
typedef FAsyncTask<FAsyncNVTTWorker> FAsyncNVTTTask;
namespace CompressionSettings
{
@@ -374,22 +370,22 @@ static bool CompressImageUsingNVTT(
// Asynchronously compress each batch.
bool bSuccess = true;
{
TIndirectArray<FAsyncNVTTTask> AsyncTasks;
TArray<FAsyncNVTTWorker> AsyncTasks;
AsyncTasks.Reserve(NumBatches);
for (int32 BatchIndex = 0; BatchIndex < NumBatches; ++BatchIndex)
{
FAsyncNVTTTask* AsyncTask = new FAsyncNVTTTask(&Compressors[BatchIndex]);
AsyncTasks.Add(AsyncTask);
#if WITH_EDITOR
AsyncTask->StartBackgroundTask(GLargeThreadPool);
#else
AsyncTask->StartBackgroundTask();
#endif
AsyncTasks.Emplace(&Compressors[BatchIndex]);
}
ParallelForTemplate(AsyncTasks.Num(), [&AsyncTasks](int32 TaskIndex)
{
AsyncTasks[TaskIndex].DoWork();
}, EParallelForFlags::BackgroundPriority | EParallelForFlags::Unbalanced);
for (int32 BatchIndex = 0; BatchIndex < NumBatches; ++BatchIndex)
{
FAsyncNVTTTask& AsyncTask = AsyncTasks[BatchIndex];
AsyncTask.EnsureCompletion();
bSuccess = bSuccess && AsyncTask.GetTask().GetCompressionResults();
bSuccess = bSuccess && AsyncTasks[BatchIndex].GetCompressionResults();
}
}