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

#pragma once 

#include "ChaosClothAsset/AddWeightMapNode.h"
#include "Dataflow/DataflowFunctionProperty.h"
#include "GeometryCollection/ManagedArrayCollection.h"
#include "ChaosClothAsset/ConnectableValue.h"
#include "Dataflow/DataflowCollectionAddScalarVertexPropertyNode.h"

#include "WeightMapNode.generated.h"

#if UE_ENABLE_INCLUDE_ORDER_DEPRECATED_IN_5_5
namespace Dataflow = UE::Dataflow;
#else
namespace UE_DEPRECATED(5.5, "Use UE::Dataflow instead.") Dataflow {}
#endif

/** Which mesh to update with the node's weight map */
UENUM()
enum class EChaosClothAssetWeightMapMeshTarget : uint8
{
	Simulation,
	Render
};



/** Painted weight map attributes node. */
PRAGMA_DISABLE_DEPRECATION_WARNINGS  // For Name implicit operators.
USTRUCT(Meta = (DataflowCloth))
struct FChaosClothAssetWeightMapNode : public FDataflowVertexAttributeEditableNode
{
PRAGMA_ENABLE_DEPRECATION_WARNINGS
	GENERATED_USTRUCT_BODY()
	DATAFLOW_NODE_DEFINE_INTERNAL(FChaosClothAssetWeightMapNode, "WeightMap", "Cloth", "Cloth Weight Map")
	DATAFLOW_NODE_RENDER_TYPE("SurfaceRender", FName("FClothCollection"), "Collection")

public:
	static constexpr float ReplaceChangedPassthroughValue = UE_BIG_NUMBER;

	UPROPERTY(Meta = (DataflowInput, DataflowOutput, DataflowPassthrough = "Collection"))
	FManagedArrayCollection Collection;

	/** The collection used to transfer weight map from. */
	UPROPERTY(Meta = (DataflowInput))
	FManagedArrayCollection TransferCollection;

	/** The name to be set as a weight map attribute. */
	UPROPERTY(EditAnywhere, Category = "Weight Map")
	FChaosClothAssetConnectableOStringValue OutputName;

	/** The name to populate this map from and override based on Map Override Type. Output Name will be used if Input Name is empty. */
	UPROPERTY(EditAnywhere, Category = "Weight Map")
	FChaosClothAssetConnectableIStringValue InputName = { TEXT("") };

	UPROPERTY(EditAnywhere, Category = "Weight Map")
	EChaosClothAssetWeightMapMeshTarget MeshTarget = EChaosClothAssetWeightMapMeshTarget::Simulation;

	/** How to apply this node's weight values onto existing maps. Changing this value will change the output map.
	 *  To change how the node's stored weights are calculated, change the equivalent value on the Weight Map Paint Tool context.*/
	UPROPERTY(EditAnywhere, Category = "Weight Map")
	EChaosClothAssetWeightMapOverrideType MapOverrideType = EChaosClothAssetWeightMapOverrideType::ReplaceChanged;

	/**
	 * The type of transfer used to transfer the weight map when a TransferCollection is connected and MeshTarget is Simulation.
	 * Render weight maps always do a 3D transfer.
	 * This property is disabled when no TransferCollection input has been connected.
	 */
	UPROPERTY(EditAnywhere, Category = "Weight Map")
	EChaosClothAssetWeightMapTransferType TransferType = EChaosClothAssetWeightMapTransferType::Use2DSimMesh;

	/** Transfer the weight map from the connected Transfer Collection containing a weight map with Input Name (or Output Name if Input Name is empty). */
	UPROPERTY(EditAnywhere, Category = "Weight Map", Meta = (ButtonImage = "Icons.Convert"))
	FDataflowFunctionProperty Transfer;

	FChaosClothAssetWeightMapNode(const UE::Dataflow::FNodeParameters& InParam, FGuid InGuid = FGuid::NewGuid());

private:
	friend class UClothEditorWeightMapPaintTool;

	class FWeightMapNodeChange;
	static TUniquePtr<class FToolCommandChange> CHAOSCLOTHASSETDATAFLOWNODES_API MakeWeightMapNodeChange(const FChaosClothAssetWeightMapNode& Node);

	const TArray<float>& GetVertexWeights() const { return VertexWeights; }
	TArray<float>& GetVertexWeights() { return VertexWeights; }

	// These methods are exported for UClothEditorWeightMapPaintTool which lives in a different module.
	FName CHAOSCLOTHASSETDATAFLOWNODES_API GetInputName(UE::Dataflow::FContext& Context) const;

	void CHAOSCLOTHASSETDATAFLOWNODES_API SetVertexWeights(const TConstArrayView<float> InputMap, const TArray<float>& FinalValues);

	// Input and FinalOutputMap can be the same array, but should not otherwise be interleaved.
	void CHAOSCLOTHASSETDATAFLOWNODES_API CalculateFinalVertexWeightValues(const TConstArrayView<float> InputMap, TArrayView<float> FinalOutputMap) const;

	//~ Begin FDataflowNode interface
	virtual void Evaluate(UE::Dataflow::FContext& Context, const FDataflowOutput* Out) const override;
	virtual void Serialize(FArchive& Ar) override;
	virtual FDataflowOutput* RedirectSerializedOutput(const FName& MissingOutputName) override;
	//~ End FDataflowNode interface

	//~ Begin - FDataflowVertexAttributeEditableNode API
	virtual void GetSupportedViewModes(UE::Dataflow::FContext& Context, TArray<FName>& OutViewModeNames) const override;
	virtual void GetVertexAttributeValues(UE::Dataflow::FContext& Context, TArray<float>& OutValues) const override;
	virtual void SetVertexAttributeValues(UE::Dataflow::FContext& Context, const TArray<float>& InValues, const TArray<int32>& InWeightIndices) override;
	virtual void GetExtraVertexMapping(UE::Dataflow::FContext& Context, FName SelectedViewMode, TArray<int32>& OutMappingToWeight, TArray<TArray<int32>>& OutMappingFromWeight) const override;
	virtual const TArray<float>& GetStoredAttributeValues() const override { return VertexWeights; }
	virtual void SwapStoredAttributeValuesWith(TArray<float>& OtherValues) override;
	//~ End - FDataflowVertexAttributeEditableNode API

	void OnTransfer(UE::Dataflow::FContext& Context);

	UPROPERTY()
	TArray<float> VertexWeights;

	UE_DEPRECATED(5.5, "Use OutputName instead.")
	UPROPERTY()
	FString Name;  // TODO: Discard in future v2. but keep for backward compatibility here as some weight maps have been created with it
};