72 lines
1.7 KiB
C++
72 lines
1.7 KiB
C++
// Copyright (c) 2012-2021 Wojciech Figat. All rights reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "../GPUTask.h"
|
|
#include "../GPUTasksContext.h"
|
|
#include "Engine/Graphics/GPUResourceProperty.h"
|
|
|
|
/// <summary>
|
|
/// GPU resource copy task.
|
|
/// </summary>
|
|
class GPUCopyResourceTask : public GPUTask
|
|
{
|
|
private:
|
|
|
|
GPUResourceReference _srcResource;
|
|
GPUResourceReference _dstResource;
|
|
|
|
public:
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="GPUCopyResourceTask"/> class.
|
|
/// </summary>
|
|
/// <param name="src">The source resource.</param>
|
|
/// <param name="dst">The destination resource.</param>
|
|
GPUCopyResourceTask(GPUResource* src, GPUResource* dst)
|
|
: GPUTask(Type::CopyResource)
|
|
, _srcResource(src)
|
|
, _dstResource(dst)
|
|
{
|
|
_srcResource.OnUnload.Bind<GPUCopyResourceTask, &GPUCopyResourceTask::OnResourceUnload>(this);
|
|
_dstResource.OnUnload.Bind<GPUCopyResourceTask, &GPUCopyResourceTask::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->CopyResource(_dstResource, _srcResource);
|
|
|
|
return Result::Ok;
|
|
}
|
|
|
|
void OnEnd() override
|
|
{
|
|
_srcResource.Unlink();
|
|
_dstResource.Unlink();
|
|
|
|
// Base
|
|
GPUTask::OnEnd();
|
|
}
|
|
};
|