Files
UnrealEngine/Engine/Plugins/MeshPainting/Source/MeshPaintingToolset/Public/BaseMeshPaintComponentAdapter.h
2025-05-18 13:04:45 +08:00

46 lines
2.6 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
#include "IMeshPaintComponentAdapter.h"
#include "TMeshPaintOctree.h"
#include "MeshAdapter.h"
#include "Spatial/MeshAABBTree3.h" // IWYU pragma: keep
#include "UObject/GCObject.h"
typedef UE::Geometry::TIndexMeshArrayAdapter<uint32, double> FIndexMeshArrayAdapterd;
/** Base mesh paint geometry adapter, handles basic sphere intersection using a Octree */
class MESHPAINTINGTOOLSET_API FBaseMeshPaintComponentAdapter : public IMeshPaintComponentAdapter, public FGCObject, public TSharedFromThis<FBaseMeshPaintComponentAdapter>
{
public:
/** Start IMeshPaintGeometryAdapter Overrides */
virtual bool Initialize() override;
virtual const TArray<FVector>& GetMeshVertices() const override;
virtual const TArray<uint32>& GetMeshIndices() const override;
virtual void GetVertexPosition(int32 VertexIndex, FVector& OutVertex) const override;
virtual TArray<uint32> SphereIntersectTriangles(const float ComponentSpaceSquaredBrushRadius, const FVector& ComponentSpaceBrushPosition, const FVector& ComponentSpaceCameraPosition, const bool bOnlyFrontFacing) const override;
virtual void GetInfluencedVertexIndices(const float ComponentSpaceSquaredBrushRadius, const FVector& ComponentSpaceBrushPosition, const FVector& ComponentSpaceCameraPosition, const bool bOnlyFrontFacing, TSet<int32> &InfluencedVertices) const override;
virtual void GetInfluencedVertexData(const float ComponentSpaceSquaredBrushRadius, const FVector& ComponentSpaceBrushPosition, const FVector& ComponentSpaceCameraPosition, const bool bOnlyFrontFacing, TArray<TPair<int32, FVector>>& OutData) const override;
virtual TArray<FVector> SphereIntersectVertices(const float ComponentSpaceSquaredBrushRadius, const FVector& ComponentSpaceBrushPosition, const FVector& ComponentSpaceCameraPosition, const bool bOnlyFrontFacing) const override;
virtual bool RayIntersectAdapter(UE::Geometry::FIndex3i& HitTriangle, FVector& HitPosition, const FVector Start, const FVector End) const override;
/** End IMeshPaintGeometryAdapter Overrides */
virtual FString GetReferencerName() const override
{
return TEXT("FBaseMeshPaintComponentAdapter");
}
virtual bool InitializeVertexData() = 0;
protected:
bool BuildOctree();
protected:
/** Index and Vertex data populated by derived classes in InitializeVertexData */
TArray<FVector> MeshVertices;
TArray<uint32> MeshIndices;
/** Octree used for reducing the cost of sphere intersecting with triangles / vertices */
TUniquePtr<FMeshPaintTriangleOctree> MeshTriOctree;
FIndexMeshArrayAdapterd Adapter;
TUniquePtr<UE::Geometry::TMeshAABBTree3<const FIndexMeshArrayAdapterd>> AABBTree;
};