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)
    int32 GetColorFromPixel(FVector point);
 
    const TMap<int32, 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)}
    };
 
    const TMap<int32, FColor> colorHeightRovaniemi = {
        {0,  FColor(48, 18, 59)},
        {1,  FColor(61, 52, 137)},
        {2,  FColor(68, 85, 196)},
        {3,  FColor(71, 115, 235)},
        {4,  FColor(67, 144, 254)},
        {5,  FColor(51, 173, 247)},
        {6,  FColor(31, 201, 221)},
        {7,  FColor(24, 223, 191)},
        {8,  FColor(42, 239, 161)},
        {9,  FColor(81, 250, 123)},
        {10, FColor(126, 255, 85)},
        {11, FColor(164, 252, 60)},
        {12, FColor(194, 242, 52)},
        {13, FColor(222, 224, 55)},
        {14, FColor(242, 201, 58)},
        {15, FColor(253, 175, 53)},
        {16, FColor(254, 143, 41)},
        {17, FColor(246, 109, 25)},
        {18, FColor(233, 77, 13)},
        {19, FColor(214, 53, 6)},
        {20, FColor(189, 32, 2)},
        {21, FColor(157, 16, 1)},
        {22, 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;
 
};