UE5 蓝图节点：从图片读取灰度值并输出到数组

UE5 蓝图节点：从图片读取灰度值并输出到数组

本文提供一个 C++ 编写的 UE5 Actor 蓝图节点，用于读取输入图片的灰度值并将其存储到一个数组中。该节点可以方便地在蓝图中使用，并提供完整的 C++ 代码示例。

MyActor.h：

#pragma once

#include "CoreMinimal.h"
#include "GameFramework/Actor.h"
#include "MyActor.generated.h"

UCLASS()
class MYPROJECT_API AMyActor : public AActor
{
    GENERATED_BODY()

public:
    // Sets default values for this actor's properties
    AMyActor();

    // Called every frame
    virtual void Tick(float DeltaTime) override;

    // Input image to read from
    UPROPERTY(EditAnywhere, Category = 'MyActor')
    UTexture2D* InputImage;

    // Output array to store grayscale values
    UPROPERTY(BlueprintReadOnly, Category = 'MyActor')
    TArray<float> GrayscaleValues;

protected:
    // Called when the game starts or when spawned
    virtual void BeginPlay() override;

private:
    // Helper function to convert color to grayscale
    float ConvertToGrayscale(FColor Color) const;
};

MyActor.cpp：

#include "MyActor.h"
#include "Engine/Texture2D.h"
#include "Engine/TextureRenderTarget2D.h"
#include "Engine/TextureRenderTarget.h"

// Sets default values
AMyActor::AMyActor()
{
    // Set this actor to call Tick() every frame.  You can turn this off to improve performance if you don't need it.
    PrimaryActorTick.bCanEverTick = true;

    // Initialize input image and output array
    InputImage = nullptr;
    GrayscaleValues.Empty();
}

// Called when the game starts or when spawned
void AMyActor::BeginPlay()
{
    Super::BeginPlay();

    // Check if input image is valid
    if (InputImage)
    {
        // Get the texture size
        const int32 Width = InputImage->GetSizeX();
        const int32 Height = InputImage->GetSizeY();

        // Lock the texture data for reading
        FColor* TextureData = static_cast<FColor*>(InputImage->PlatformData->Mips[0].BulkData.LockReadOnly());

        // Iterate over each pixel and convert to grayscale
        for (int32 Y = 0; Y < Height; ++Y)
        {
            for (int32 X = 0; X < Width; ++X)
            {
                // Get the color value at this pixel
                const FColor Color = TextureData[Y * Width + X];

                // Convert to grayscale and add to the output array
                const float GrayscaleValue = ConvertToGrayscale(Color);
                GrayscaleValues.Add(GrayscaleValue);
            }
        }

        // Unlock the texture data
        InputImage->PlatformData->Mips[0].BulkData.Unlock();
    }
}

// Called every frame
void AMyActor::Tick(float DeltaTime)
{
    Super::Tick(DeltaTime);

}

// Helper function to convert color to grayscale
float AMyActor::ConvertToGrayscale(FColor Color) const
{
    const float Red = static_cast<float>(Color.R);
    const float Green = static_cast<float>(Color.G);
    const float Blue = static_cast<float>(Color.B);
    const float GrayscaleValue = 0.2989f * Red + 0.5870f * Green + 0.1140f * Blue;
    return GrayscaleValue;
}

在UE5的蓝图中，您可以将此Actor作为一个组件来使用，并且可以通过蓝图实现访问InputImage和GrayscaleValues属性。

使用说明：

1. 将 MyActor.h 和 MyActor.cpp 文件添加到您的 UE5 项目中。
2. 在您的蓝图中添加一个 MyActor 组件。
3. 在 MyActor 组件的属性面板中，设置 InputImage 属性，选择您要读取的图片。
4. 在您的蓝图中，可以通过 Get GrayscaleValues 节点访问 GrayscaleValues 属性，获取读取到的灰度值数组。

示例蓝图：

注意：

• 该节点仅适用于单通道图像。
• 该节点使用的是简单的灰度转换公式，您可以根据需要调整公式。
• 在实际使用中，您可以根据需求添加其他功能，例如处理多通道图像、调整灰度值范围等。