TreeHeightsFromImageComponent.h

Go to the documentation of this file.

// Copyright (c) 2024 FrostBit Software Lab at the Lapland University of Applied Sciences
//
// This work is licensed under the terms of the MIT license.
// For a copy, see <https://opensource.org/licenses/MIT>.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Components/ActorComponent.h"
#include "Math/Color.h"
#include "Engine/Texture2D.h"
#include "TreeHeightsFromImageComponent.generated.h"
 
UCLASS(ClassGroup = (Custom), meta = (BlueprintSpawnableComponent))
class AGRARSENSE_API UTreeHeightsFromImageComponent : public UActorComponent
{
    GENERATED_BODY()
 
public:
 
    UTreeHeightsFromImageComponent();
 
protected:
 
    virtual void BeginPlay() override;
 
    void CacheTexture();
 
    float EuclideanDistance(const FColor& Color1, const FColor& Color2);
 
    FColor FindClosestColor(const FColor& TargetColor, const TMap<int32, FColor>& ColorMap);
 
    UFUNCTION(BlueprintCallable)
    int GetColorFromPixel(FVector point);
 
    const TMap<int, FColor> colorHeight = {
        {0, FColor(48, 18, 59)},
        {1, FColor(59, 48, 129)},
        {2, FColor(67, 77, 183)},
        {3, FColor(70, 104, 222)},
        {4, FColor(70, 130, 247)},
        {5, FColor(61, 156, 252)},
        {6, FColor(44, 182, 239)},
        {7, FColor(30, 205, 216)},
        {8, FColor(26, 223, 190)},
        {9, FColor(42, 237, 163)},
        {10, FColor(73, 247, 130)},
        {11, FColor(111, 253, 97)},
        {12, FColor(147, 253, 71)},
        {13, FColor(177, 248, 56)},
        {14, FColor(203, 236, 53)},
        {15, FColor(226, 219, 56)},
        {16, FColor(243, 199, 57)},
        {17, FColor(252, 176, 53)},
        {18, FColor(253, 148, 43)},
        {19, FColor(249, 118, 30)},
        {20, FColor(239, 89, 17)},
        {21, FColor(224, 65, 9)},
        {22, FColor(205, 45, 5)},
        {23, FColor(181, 28, 2)},
        {24, FColor(153, 14, 1)},
        {25, FColor(122, 4, 3)}
    };
 
    UPROPERTY()
    int32 Width;
    UPROPERTY()
    int32 Height;
 
public:
 
    UPROPERTY(EditAnywhere, Category = "Generate")
    bool GenerateTrigger;
 
#if WITH_EDITOR
    virtual void PostEditChangeProperty(FPropertyChangedEvent& PropertyChangedEvent) override;
#endif
 
    UPROPERTY(EditAnywhere)
    UTexture2D* treeMapImage;
 
    UPROPERTY()
    TArray<uint8> Pixels;
 
};