Rather odd attempt to screenshot the view

This commit is contained in:
Menotdan
2023-05-09 16:14:37 -04:00
parent ec60c90963
commit c73ab05d70
3 changed files with 85 additions and 1 deletions

View File

@@ -212,7 +212,7 @@ namespace FlaxEditor.GUI.Dialogs
private void OnEyedropColor()
{
throw new NotImplementedException();
Editor.Log("Color: " + Screenshot.GetPixelAt(7, 6).ToString());
}
private void OnRGBAChanged()

View File

@@ -3,6 +3,7 @@
#include "Screenshot.h"
#include "Engine/Core/Log.h"
#include "Engine/Core/Math/Math.h"
#include "Engine/Core/Math/Color32.h"
#include "Engine/Graphics/RenderTask.h"
#include "Engine/Platform/FileSystem.h"
#include "Engine/Graphics/Textures/TextureData.h"
@@ -129,6 +130,66 @@ void CaptureScreenshot::OnFail()
ThreadPoolTask::OnFail();
}
/// <summary>
/// Capture screenshot helper
/// </summary>
/// <seealso cref="ThreadPoolTask" />
class GetPixelData : public ThreadPoolTask
{
friend Screenshot;
private:
TextureData _data;
Color32 _color;
public:
int32 x;
int32 y;
public:
/// <summary>
/// Gets the texture data container.
/// </summary>
/// <returns>Texture data</returns>
FORCE_INLINE TextureData& GetData()
{
return _data;
}
FORCE_INLINE Color32 GetColor()
{
return _color;
}
protected:
// [ThreadPoolTask]
bool Run() override;
void OnFail() override;
};
bool GetPixelData::Run()
{
LOG(Warning, "REAL");
TextureMipData *mipData = _data.GetData(0, 0);
Array<Color32> pixels;
mipData->GetPixels(pixels, _data.Width, _data.Height, _data.Format);
_color = pixels[(y * _data.Width) + x];
LOG(Warning, "really real");
LOG(Warning, "Color: {0} {1} {2}", _color.R, _color.B, _color.G);
return false;
}
void GetPixelData::OnFail()
{
LOG(Warning, "Cannot get pixel data.");
// Base
ThreadPoolTask::OnFail();
}
void Screenshot::Capture(GPUTexture* target, const StringView& path)
{
// Validate
@@ -242,3 +303,18 @@ void Screenshot::Capture(const StringView& path)
Capture(mainTask, path);
}
Color32 Screenshot::GetPixelAt(int32 x, int32 y) {
GPUSwapChain* swapChain = Engine::MainWindow->GetSwapChain();
auto getPixelTask = New<GetPixelData>();
getPixelTask->x = x;
getPixelTask->y = y;
Task* downloadTask = swapChain->DownloadDataAsync(getPixelTask->GetData());
downloadTask->ContinueWith(getPixelTask);
downloadTask->Start();
//downloadTask->Wait(750);
return getPixelTask->GetColor();
}

View File

@@ -39,4 +39,12 @@ API_CLASS(Static) class FLAXENGINE_API Screenshot
/// </summary>
/// <param name="path">The custom file location. Use null or empty to use default one.</param>
API_FUNCTION() static void Capture(const StringView& path = StringView::Empty);
/// <summary>
/// Get the pixel at specified coordinates.
/// </summary>
/// <param name="x">The x coordinate to read.</param>
/// <param name="y">The y coordinate to read.</param>
/// <returns>The color </returns>
API_FUNCTION() static Color32 GetPixelAt(int32 x, int32 y);
};