// Copyright 2020-2024 CesiumGS, Inc. and Contributors

#pragma once

#include "Cesium3DTilesSelection/BoundingVolume.h"
#include "CesiumPrimitive.h"
#include "CesiumPrimitiveFeatures.h"
#include "Components/InstancedStaticMeshComponent.h"
#include "Components/StaticMeshComponent.h"
#include "CoreMinimal.h"
#include "Engine/StaticMesh.h"
#include "Materials/MaterialInstanceDynamic.h"
#include "PhysicsEngine/BodySetup.h"
#include "VecMath.h"

#include "CesiumGltfPrimitiveComponent.generated.h"

namespace CesiumGltf {
struct Model;
struct MeshPrimitive;
} // namespace CesiumGltf

/**
 * A component that represents and renders a glTF mesh primitive made
 * from triangles.
 */
UCLASS()
class UCesiumGltfPrimitiveComponent : public UStaticMeshComponent,
                                      public ICesiumPrimitive {
  GENERATED_BODY()

public:
  // Sets default values for this component's properties
  UCesiumGltfPrimitiveComponent();
  virtual ~UCesiumGltfPrimitiveComponent();

  void BeginDestroy() override;

  FBoxSphereBounds CalcBounds(const FTransform& LocalToWorld) const override;

  void
  UpdateTransformFromCesium(const glm::dmat4& CesiumToUnrealTransform) override;

  // from ICesiumPrimitive
  CesiumPrimitiveData& getPrimitiveData() override;
  const CesiumPrimitiveData& getPrimitiveData() const override;
  // from ICesiumLoadedTilePrimitive
  ICesiumLoadedTile& GetLoadedTile() override;
  UStaticMeshComponent& GetMeshComponent() override;
  std::optional<uint32_t> FindTextureCoordinateIndexForGltfAccessor(
      int32_t accessorIndex) const override;

  virtual void OnCreatePhysicsState() override;

private:
  CesiumPrimitiveData _cesiumData;
};

/**
 * A component that represents and renders an instanced glTF mesh
 * primitive made from triangles.
 */
UCLASS()
class UCesiumGltfInstancedComponent : public UInstancedStaticMeshComponent,
                                      public ICesiumPrimitive {
  GENERATED_BODY()

public:
  // Sets default values for this component's properties
  UCesiumGltfInstancedComponent();
  virtual ~UCesiumGltfInstancedComponent();

  void BeginDestroy() override;

  FBoxSphereBounds CalcBounds(const FTransform& LocalToWorld) const override;
  void
  UpdateTransformFromCesium(const glm::dmat4& CesiumToUnrealTransform) override;

  // from ICesiumPrimitive
  CesiumPrimitiveData& getPrimitiveData() override;
  const CesiumPrimitiveData& getPrimitiveData() const override;
  // from ICesiumLoadedTilePrimitive
  ICesiumLoadedTile& GetLoadedTile() override;
  UStaticMeshComponent& GetMeshComponent() override;
  std::optional<uint32_t> FindTextureCoordinateIndexForGltfAccessor(
      int32_t accessorIndex) const override;

  TSharedPtr<FCesiumPrimitiveFeatures> pInstanceFeatures;

private:
  CesiumPrimitiveData _cesiumData;
};