TreeHeightsFromImage.cpp

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>.
 
#include "TreeHeightsFromImage.h"
 
ATreeHeightsFromImage::ATreeHeightsFromImage()
{
    PrimaryActorTick.bCanEverTick = false;
}
 
void ATreeHeightsFromImage::BeginPlay()
{
    Super::BeginPlay();
 
#if WITH_EDITOR
    if (treeMapImage)
    {
        // Lock the texture to allow access to data
        FTexture2DMipMap& Mip = treeMapImage->GetPlatformData()->Mips[0];
        void* Data = Mip.BulkData.Lock(LOCK_READ_ONLY);
 
        if (Data)
        {
            int32 Width = treeMapImage->GetSizeX();
            int32 Height = treeMapImage->GetSizeY();
 
            if (Width <= 0 || Height <= 0)
                return;
 
            UE_LOG(LogTemp, Warning, TEXT("Image width: %i, Height: %i"), Width, Height);
 
            uint8* Pixels = static_cast<uint8*>(Data);
            for (int32 y = 0; y < Height; y++)
            {
                for (int32 x = 0; x < Width; x++)
                {
                    int32 Index = (y * Width + x) * 4; // 4 bytes per pixel (RGBA)
                    uint8 R = Pixels[Index];
                    uint8 G = Pixels[Index + 1];
                    uint8 B = Pixels[Index + 2];
                    uint8 A = Pixels[Index + 3];
 
                    FColor pixelColor = FColor(B, G, R);
 
                    if (x == 446 && y == 353)
                    {
                        UE_LOG(LogTemp, Warning, TEXT("Coordinates RGB = %i, %i, %i"), B, G, R);
                        FColor ClosestColor = FindClosestColor(pixelColor, colorHeight);
                        UE_LOG(LogTemp, Warning, TEXT("Closest color: R=%d, G=%d, B=%d"), ClosestColor.R, ClosestColor.G, ClosestColor.B);
 
                        for (const auto& Pair : colorHeight)
                        {
                            if (Pair.Value == ClosestColor)
                                UE_LOG(LogTemp, Warning, TEXT("This should be a %im size tree"), Pair.Key);
                        }
 
                    }
 
                    if (x == Width / 2 && y == Height / 2)
                    {
                        UE_LOG(LogTemp, Warning, TEXT("Coordinates middlepoint Width: %i, Height: %i"), x, y);
                        //UE_LOG(LogTemp, Warning, TEXT("ARGB: %i,%i,%i,%i"), A,R,G,B);
                    }
                }
            }
 
            // Unlock the texture
            Mip.BulkData.Unlock();
        }
    }
#endif
}
 
float ATreeHeightsFromImage::EuclideanDistance(const FColor& Color1, const FColor& Color2)
{
    float R1 = Color1.R;
    float G1 = Color1.G;
    float B1 = Color1.B;
    float R2 = Color2.R;
    float G2 = Color2.G;
    float B2 = Color2.B;
    return FMath::Sqrt(FMath::Square(R2 - R1) + FMath::Square(G2 - G1) + FMath::Square(B2 - B1));
}
 
FColor ATreeHeightsFromImage::FindClosestColor(const FColor& TargetColor, const TMap<int32, FColor>& ColorMap)
{
    float MinDistance = FLT_MAX;
    FColor ClosestColor;
 
    for (const auto& Pair : ColorMap)
    {
        float Distance = EuclideanDistance(TargetColor, Pair.Value);
        if (Distance < MinDistance)
        {
            MinDistance = Distance;
            ClosestColor = Pair.Value;
        }
    }
 
    return ClosestColor;
}