72 lines
2.0 KiB
C++
72 lines
2.0 KiB
C++
// Copyright (c) 2012-2022 Wojciech Figat. All rights reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "../GPUTask.h"
|
|
#include "../GPUTasksContext.h"
|
|
#include "Engine/Graphics/GPUResourceProperty.h"
|
|
|
|
/// <summary>
|
|
/// GPU subresource copy task.
|
|
/// </summary>
|
|
class GPUCopySubresourceTask : public GPUTask
|
|
{
|
|
private:
|
|
GPUResourceReference _srcResource;
|
|
GPUResourceReference _dstResource;
|
|
uint32 _srcSubresource, _dstSubresource;
|
|
|
|
public:
|
|
/// <summary>
|
|
/// Init
|
|
/// </summary>
|
|
/// <param name="src">The source resource.</param>
|
|
/// <param name="dst">The destination resource.</param>
|
|
/// <param name="srcSubresource">The source subresource index.</param>
|
|
/// <param name="dstSubresource">The destination subresource index.</param>
|
|
GPUCopySubresourceTask(GPUResource* src, GPUResource* dst, uint32 srcSubresource, uint32 dstSubresource)
|
|
: GPUTask(Type::CopyResource)
|
|
, _srcResource(src)
|
|
, _dstResource(dst)
|
|
, _srcSubresource(srcSubresource)
|
|
, _dstSubresource(dstSubresource)
|
|
{
|
|
_srcResource.OnUnload.Bind<GPUCopySubresourceTask, &GPUCopySubresourceTask::OnResourceUnload>(this);
|
|
_dstResource.OnUnload.Bind<GPUCopySubresourceTask, &GPUCopySubresourceTask::OnResourceUnload>(this);
|
|
}
|
|
|
|
private:
|
|
void OnResourceUnload(GPUResourceReference* ref)
|
|
{
|
|
Cancel();
|
|
}
|
|
|
|
public:
|
|
// [GPUTask]
|
|
bool HasReference(Object* resource) const override
|
|
{
|
|
return _srcResource == resource || _dstResource == resource;
|
|
}
|
|
|
|
protected:
|
|
// [GPUTask]
|
|
Result run(GPUTasksContext* context) override
|
|
{
|
|
if (_srcResource.IsMissing() || _dstResource.IsMissing())
|
|
return Result::MissingResources;
|
|
|
|
context->GPU->CopySubresource(_dstResource, _dstSubresource, _srcResource, _srcSubresource);
|
|
|
|
return Result::Ok;
|
|
}
|
|
|
|
void OnEnd() override
|
|
{
|
|
_srcResource.Unlink();
|
|
_dstResource.Unlink();
|
|
|
|
// Base
|
|
GPUTask::OnEnd();
|
|
}
|
|
};
|