111 lines
3.0 KiB
C++
111 lines
3.0 KiB
C++
// Copyright (c) 2012-2022 Wojciech Figat. All rights reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "../GPUTask.h"
|
|
#include "../GPUTasksContext.h"
|
|
#include "Engine/Graphics/Textures/GPUTexture.h"
|
|
#include "Engine/Graphics/GPUResourceProperty.h"
|
|
#include "Engine/Core/Types/DataContainer.h"
|
|
|
|
/// <summary>
|
|
/// GPU texture mip upload task.
|
|
/// </summary>
|
|
class GPUUploadTextureMipTask : public GPUTask
|
|
{
|
|
protected:
|
|
|
|
GPUTextureReference _texture;
|
|
int32 _mipIndex;
|
|
BytesContainer _data;
|
|
|
|
public:
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="GPUUploadTextureMipTask"/> class.
|
|
/// </summary>
|
|
/// <param name="texture">The target texture.</param>
|
|
/// <param name="mipIndex">The target texture mip data.</param>
|
|
/// <param name="data">The data to upload.</param>
|
|
/// <param name="copyData">True if copy data to the temporary buffer, otherwise the input data will be used directly. Then ensure it is valid during the copy operation period (for the next few frames).</param>
|
|
GPUUploadTextureMipTask(GPUTexture* texture, int32 mipIndex, Span<byte> data, bool copyData)
|
|
: GPUTask(Type::UploadTexture)
|
|
, _texture(texture)
|
|
, _mipIndex(mipIndex)
|
|
{
|
|
_texture.OnUnload.Bind<GPUUploadTextureMipTask, &GPUUploadTextureMipTask::OnResourceUnload>(this);
|
|
|
|
if (copyData)
|
|
_data.Copy(data);
|
|
else
|
|
_data.Link(data);
|
|
}
|
|
|
|
private:
|
|
|
|
void OnResourceUnload(GPUTextureReference* ref)
|
|
{
|
|
Cancel();
|
|
}
|
|
|
|
public:
|
|
|
|
// [GPUTask]
|
|
bool HasReference(Object* resource) const override
|
|
{
|
|
return _texture == resource;
|
|
}
|
|
|
|
protected:
|
|
|
|
// [GPUTask]
|
|
Result run(GPUTasksContext* context) override
|
|
{
|
|
const auto texture = _texture.Get();
|
|
if (texture == nullptr)
|
|
return Result::MissingResources;
|
|
ASSERT(texture->IsAllocated());
|
|
|
|
// Cache data
|
|
const int32 arraySize = texture->ArraySize();
|
|
uint32 rowPitch, slicePitch;
|
|
texture->ComputePitch(_mipIndex, rowPitch, slicePitch);
|
|
ASSERT((uint32)_data.Length() >= slicePitch * arraySize);
|
|
|
|
// Update all array slices
|
|
const byte* dataSource = _data.Get();
|
|
for (int32 arrayIndex = 0; arrayIndex < arraySize; arrayIndex++)
|
|
{
|
|
context->GPU->UpdateTexture(texture, arrayIndex, _mipIndex, dataSource, rowPitch, slicePitch);
|
|
dataSource += slicePitch;
|
|
}
|
|
|
|
return Result::Ok;
|
|
}
|
|
|
|
void OnSync() override
|
|
{
|
|
auto texture = _texture.Get();
|
|
if (texture)
|
|
{
|
|
// Check if the new mips has been just uploaded
|
|
if (_mipIndex == texture->HighestResidentMipIndex() - 1)
|
|
{
|
|
// Mark as mip loaded
|
|
texture->SetResidentMipLevels(texture->ResidentMipLevels() + 1);
|
|
}
|
|
}
|
|
|
|
// Base
|
|
GPUTask::OnSync();
|
|
}
|
|
|
|
void OnEnd() override
|
|
{
|
|
_texture.Unlink();
|
|
|
|
// Base
|
|
GPUTask::OnEnd();
|
|
}
|
|
};
|