// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include "CoreMinimal.h"

#include "MetaHumanFaceAnimationSolver.generated.h"


UENUM()
enum class EDepthMapInfluenceValue : uint8
{
	None = 0,
	Low,
	High,
};

UENUM()
enum class ETeethMode : uint8
{
	TrackingPoints = 0,
	Estimated,
};

/** MetaHuman Face Animation Solver
*
*   Holds configuration info used by the solver.
*
*/
UCLASS()
class METAHUMANFACEANIMATIONSOLVER_API UMetaHumanFaceAnimationSolver : public UObject
{
	GENERATED_BODY()

public:

	// Delegate called when something changes in the face animation solver that others should know about
	DECLARE_MULTICAST_DELEGATE(FOnInternalsChanged)

#if WITH_EDITOR

	//~Begin UObject interface
	virtual void PostEditChangeProperty(struct FPropertyChangedEvent& InPropertyChangedEvent) override;
	virtual void PostTransacted(const FTransactionObjectEvent& InTransactionEvent) override;
	//~End UObject interface

#endif

	UPROPERTY(EditAnywhere, Category = "Parameters", meta = (InlineEditConditionToggle))
	bool bOverrideDeviceConfig = false;

	UPROPERTY(EditAnywhere, Category = "Parameters", meta = (EditCondition = "bOverrideDeviceConfig"))
	TObjectPtr<class UMetaHumanConfig> DeviceConfig;

	UPROPERTY(EditAnywhere, Category = "Parameters", meta = (InlineEditConditionToggle))
	bool bOverrideDepthMapInfluence = false;

	/* The amount by which the depth-map is used to influence the solve result */
	UPROPERTY(EditAnywhere, Category = "Parameters", meta = (EditCondition = "bOverrideDepthMapInfluence"))
	EDepthMapInfluenceValue DepthMapInfluence = EDepthMapInfluenceValue::High;

	UPROPERTY(EditAnywhere, Category = "Parameters", meta = (InlineEditConditionToggle))
	bool bOverrideEyeSolveSmoothness = false;

	/* The amount of smoothing to be applied to the eye gaze control results */
	UPROPERTY(EditAnywhere, Category = "Parameters", meta = (ClampMin = "0.0", ClampMax = "1.0", UIMin = "0.0", UIMax = "1.0", EditCondition = "bOverrideEyeSolveSmoothness"))
	float EyeSolveSmoothness = 0.1;

	UPROPERTY(EditAnywhere, Category = "Parameters", meta = (InlineEditConditionToggle))
	bool bOverrideTeethMode = false;

	/* Whether teeth tracking points are used or teeth position is estimated */
	UPROPERTY(EditAnywhere, Category = "Parameters", meta = (EditCondition = "bOverrideTeethMode"))
	ETeethMode TeethMode = ETeethMode::TrackingPoints;

	bool CanProcess() const;
	bool SettingsOverridden() const;
	bool GetConfigDisplayName(class UCaptureData* InCaptureData, FString& OutName) const;

	FString GetSolverTemplateData(class UCaptureData* InCaptureData = nullptr) const;
	FString GetSolverConfigData(class UCaptureData* InCaptureData = nullptr) const;
	FString GetSolverDefinitionsData(class UCaptureData* InCaptureData = nullptr) const;
	FString GetSolverHierarchicalDefinitionsData(class UCaptureData* InCaptureData = nullptr) const;
	FString GetSolverPCAFromDNAData(class UCaptureData* InCaptureData = nullptr) const;

	FOnInternalsChanged& OnInternalsChanged();

private:

	class UMetaHumanConfig* GetEffectiveConfig(class UCaptureData* InCaptureData) const;

	FString JsonObjectAsString(TSharedPtr<class FJsonObject> InJsonObject) const;

	FOnInternalsChanged OnInternalsChangedDelegate;

	void NotifyInternalsChanged();
};