// Copyright 2020 Phyronnaz

#pragma once

#include "CoreMinimal.h"
#include "PhysicsEngine/BodySetup.h" // Can't forward decl anything with uobjects generated constructors...
#include "Components/PrimitiveComponent.h"
#include "VoxelWorldRootComponent.generated.h"

UCLASS(editinlinenew)
class VOXEL_API UVoxelWorldRootComponent : public UPrimitiveComponent
{
	GENERATED_BODY()

public:
	UVoxelWorldRootComponent();
	~UVoxelWorldRootComponent();
	
	ECollisionTraceFlag CollisionTraceFlag = {};

	//~ Begin UPrimitiveComponent Interface
	virtual UBodySetup* GetBodySetup() override final;
	virtual FPrimitiveSceneProxy* CreateSceneProxy() override final;
	virtual FBoxSphereBounds CalcBounds(const FTransform& LocalToWorld) const override;
	//~ End UPrimitiveComponent Interface

	// Only need to tick when created
	void TickWorldRoot();

public:
#if WITH_PHYSX && PHYSICS_INTERFACE_PHYSX
	void UpdateConvexCollision(uint64 Id, const FBox& Bounds, TArray<FKConvexElem>&& ConvexElements, TArray<physx::PxConvexMesh*>&& ConvexMeshes);
	void SetCookedTriMeshes(const TArray<physx::PxTriangleMesh*>& TriMeshes);
#endif

private:
	UPROPERTY(Transient)
	TObjectPtr<UBodySetup> BodySetup;

	FBoxSphereBounds LocalBounds;

	// For debug draw
	FCriticalSection BodySetupLock;
	
#if WITH_PHYSX && PHYSICS_INTERFACE_PHYSX
	struct FConvexElements
	{
		const FBox Bounds;
		const TArray<FKConvexElem> ConvexElements;
		const TArray<physx::PxConvexMesh*> ConvexMeshes;

		FConvexElements(const FBox& InBounds, TArray<FKConvexElem>&& InConvexElements, TArray<physx::PxConvexMesh*>&& InConvexMeshes);
		~FConvexElements();
	};
	TMap<uint64, TUniquePtr<FConvexElements>> Elements;

	bool bRebuildQueued = false;
	
	void RebuildConvexCollision();
#endif

	friend class FVoxelRenderSimpleCollisionSceneProxy;
};