// Copyright 2025, Algoryx Simulation AB.

#pragma once

// AGX Dynamics for Unreal includes.
#include "AMOR/AGX_ShapeContactMergeSplitProperties.h"
#include "AGX_MotionControl.h"
#include "AGX_NativeOwner.h"
#include "AGX_RigidBodyEnums.h"
#include "RigidBodyBarrier.h"

// Unreal Engine includes.
#include "Components/SceneComponent.h"
#include "CoreMinimal.h"

#include "AGX_RigidBodyComponent.generated.h"

class UAGX_ShapeComponent;

struct FAGX_ImportContext;

UCLASS(
	ClassGroup = "AGX", Category = "AGX", Meta = (BlueprintSpawnableComponent),
	Hidecategories = (Cooking, Collision, LOD, Physics, Rendering, Replication))
class AGXUNREAL_API UAGX_RigidBodyComponent : public USceneComponent, public IAGX_NativeOwner
{
	GENERATED_BODY()

public:
	// Sets default values for this component's properties
	UAGX_RigidBodyComponent();

	/**
	 * Set the position of the Rigid Body.
	 *
	 * The semantics depend on if Set Position is called during editing or runtime. During editing
	 * it behaves as-if a drag on the Rigid Body Compnent's transform gizmo, i.e. the Rigid
	 * Body Component is moved to the given location. Durain runtime, when there is a native AGX
	 * Dynamics object, the semantics is as-if AGX Dynamics had moved the Rigid Body, i.e. the
	 * Transform Target is moved so that the Rigid Body Component end up at the given position.
	 *
	 * @param Position The new position of the Rigid Body.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetPosition(FVector Position);

	/**
	 * Get the position of the Rigid Body.
	 *
	 * If there is a native AGX Dynamics object then that position is read and returned. If there
	 * is no native object then the Rigid Body Component's Location is returned. These are usually
	 * in sync but they are not in between AGX Dynamics having updated the native object and the
	 * Rigid Body Component's Tick Component function having updated the Unreal Engine
	 * transformation.
	 *
	 * @return The position of the Rigid Body.
	 */
	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	FVector GetPosition() const;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetRotation(FQuat Rotation);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	FQuat GetRotation() const;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetRotator(FRotator Rotator);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	FRotator GetRotator() const;

	/**
	 * Write the Rigid Body Component's transformation into the native AGX Dynamics object. This is
	 * necessary after manipulating the Scene Component tree to notify AGX Dynamics about the
	 * changes.
	 *
	 * May only be called if there actually is a native for this Rigid Body.
	 */
	UFUNCTION(BlueprintCallable, Category = "Rigid Body")
	bool WriteTransformToNative();

	/**
	 * Read the native AGX Dynamics object's transformation and apply it to the Transform Target.
	 *
	 * This is done automatically on Tick, so there is rarely any need to call this function.
	 *
	 * May only be called if there actually is a native for this Rigid Body.
	 */
	UFUNCTION(BlueprintCallable, Category = "Rigid Body")
	bool ReadTransformFromNative();

	UPROPERTY(EditAnywhere, Category = "AGX Dynamics")
	bool bEnabled = true;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetEnabled(bool InEnabled);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	bool IsEnabled() const;

	UFUNCTION(
		BlueprintCallable, BlueprintPure, Category = "AGX Dynamics",
		Meta = (DeprecatedFunction, DeprecationMessage = "Use IsEnabled instead."))
	bool GetEnabled() const;

	/**
	 * The mass of the Rigid Body [kg].
	 */
	UPROPERTY(
		EditAnywhere, Category = "AGX Dynamics", Meta = (EditCondition = "!bAutoGenerateMass"))
	float Mass;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetMass(float InMass);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	float GetMass() const;

	/**
	 * Whether the mass should be computed automatically.
	 */
	UPROPERTY(EditAnywhere, Category = "AGX Dynamics")
	bool bAutoGenerateMass = true;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetAutoGenerateMass(bool bInAuto);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	bool GetAutoGenerateMass() const;

	/**
	 * Center of mass offset [cm].
	 */
	UPROPERTY(
		EditAnywhere, Category = "AGX Dynamics",
		Meta = (EditCondition = "!bAutoGenerateCenterOfMassOffset"))
	FVector CenterOfMassOffset;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetCenterOfMassOffset(FVector InCoMOffset);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	FVector GetCenterOfMassOffset() const;

	/**
	 * Whether the center of mass offset should be computed automatically.
	 */
	UPROPERTY(EditAnywhere, Category = "AGX Dynamics")
	bool bAutoGenerateCenterOfMassOffset = true;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetAutoGenerateCenterOfMassOffset(bool bInAuto);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	bool GetAutoGenerateCenterOfMassOffset() const;

	/**
	 * Returns the position of the center of mass in world coordinates [cm].
	 * This function is only valid if the Rigid Body has a Native object, which usually is true only
	 * during Play. Returns zero vector if not.
	 */
	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	FVector GetCenterOfMassPosition() const;

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	TArray<UAGX_ShapeComponent*> GetShapes() const;

	/**
	 * Explicitly update mass properties.
	 *
	 * E.g., when the density is changed on a material which this rigid body depends on. This method
	 * uses the current mass property update mask. I.e., if this rigid body has an explicitly
	 * assigned mass, mass will not be updated during this call.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void UpdateMassProperties();

	/**
	 * Calculate the mass of this rigid body using the volume and density of added geometries.
	 *
	 * \note This calculated mass isn't necessary the same as getMassProperties()->getMass() since
	 * the mass in mass properties could be assigned (i.e., explicit).
	 *
	 * @return The total mass of this rigid body, calculated given volume and density of added
	 * geometries.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	double CalculateMass() const;

	/**
	 * The three-component diagonal of the inertia tensor [kgm^2].
	 */
	UPROPERTY(
		EditAnywhere, Category = "AGX Dynamics",
		Meta = (EditCondition = "!bAutoGeneratePrincipalInertia"))
	FVector PrincipalInertia;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetPrincipalInertia(FVector InPrincipalInertia);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	FVector GetPrincipalInertia() const;

	/**
	 * Whether the principal inertia should be computed automatically.
	 */
	UPROPERTY(EditAnywhere, Category = "AGX Dynamics")
	bool bAutoGeneratePrincipalInertia = true;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetAutoGeneratePrincipalInertia(bool bInAuto);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	bool GetAutoGeneratePrincipalInertia() const;

	/**
	 * The velocity of the body [cm/s].
	 */
	UPROPERTY(EditAnywhere, Category = "AGX Dynamics")
	FVector Velocity;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetVelocity(FVector InVelocity);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	FVector GetVelocity() const;

	/**
	 * The angular velocity of the Rigid Body [deg/s].
	 * Following the Unreal Editor rotation widget so that a positive
	 * angular velocity about some axis produces increasing rotation values in the editor widget.
	 *
	 * A positive X angular velocity rotates the Z axis towards the Y axis, i.e., roll right. A
	 * right-handed rotation.
	 *
	 * A positive Y angular velocity rotates the X axis towards the Z axis, i.e., pitch up. A
	 * right-handed rotation.
	 *
	 * A positive Z angular velocity rotates the X axis towards the Y axis, i.e, yaw right. A
	 * left-handed rotation.
	 */
	UPROPERTY(EditAnywhere, Category = "AGX Dynamics")
	FVector AngularVelocity;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetAngularVelocity(FVector InAngularVelocity);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	FVector GetAngularVelocity() const;

	/**
	 * Get the current acceleration of this Rigid Body in world coordinates [cm/s^2].
	 * Will always be zero for KINEMATIC and STATIC bodies.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	FVector GetAcceleration() const;

	/**
	 * Get the current angular acceleration of this Rigid Body in world coordinates [deg/s^2].
	 * Will always be zero for KINEMATIC and STATIC bodies.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	FVector GetAngularAcceleration() const;

	/**
	 * The linear velocity damping of the Rigid Body. Can be used to mimic aerodynamic or
	 * hydrodynamic effects [kg/s].
	 */
	UPROPERTY(EditAnywhere, Category = "AGX Dynamics")
	FVector LinearVelocityDamping;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetLinearVelocityDamping(FVector InLinearVelocityDamping);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	FVector GetLinearVelocityDamping() const;

	/**
	 * The angular velocity damping of the Rigid Body. Can be used to mimic aerodynamic or
	 * hydrodynamic effects [kg/s].
	 */
	UPROPERTY(EditAnywhere, Category = "AGX Dynamics")
	FVector AngularVelocityDamping;

	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SetAngularVelocityDamping(FVector InAngularVelocityDamping);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	FVector GetAngularVelocityDamping() const;

	UPROPERTY(EditAnywhere, Category = "AGX Dynamics")
	TEnumAsByte<enum EAGX_MotionControl> MotionControl;

	UFUNCTION(
		BlueprintCallable, Category = "AGX Dynamics", Meta = (InMotionControl = "MC_DYNAMICS"))
	void SetMotionControl(TEnumAsByte<enum EAGX_MotionControl> InMotionControl);

	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	TEnumAsByte<enum EAGX_MotionControl> GetMotionControl() const;

	/**
	 * Decides to where transformation updates from AGX Dynamics should be written during Play.
	 */
	UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "AGX Dynamics")
	TEnumAsByte<enum EAGX_TransformTarget> TransformTarget;

	UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "AGX AMOR")
	FAGX_ShapeContactMergeSplitProperties MergeSplitProperties;

	/**
	 * Creates Merge Split Properties for this Rigid Body. This only needs to be called once per
	 * Play instance.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX AMOR")
	void CreateMergeSplitProperties();

	/**
	 * Returns true if this Rigid Body has been automatically merged and is currently merged.
	 * Returns false otherwise.
	 */
	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX AMOR")
	bool IsAutomaticallyMerged();

	/**
	 * Explicitly split this Rigid Body in case it has been automatically merged.
	 * Returns true if this Rigid Body has Merge Split properties, has been automatically merged and
	 * was successfully split. Returns false otherwise.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX AMOR")
	bool Split();

	/**
	 * Add an external force, given in the world coordinate frame, that will be affecting this body
	 * in the next solve. The force will be applied at the center of mass.
	 *
	 * This member function may only be called when a native AGX Dynamics representation has already
	 * been created, i.e., HasNative() returns true;
	 *
	 * @param Force The force to add, given in the world coordinate frame.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void AddForceAtCenterOfMass(FVector Force);

	/**
	 * Add an external force, given in the world coordinate frame, applied at a point specified in
	 * the world coordinate frame, that will be affecting this body in the next solve. If the
	 * location is different from the center of mass then a torque will be calculated and added as
	 * well.
	 *
	 * This member function may only be called when a native AGX Dynamics representation has already
	 * been created, i.e., HasNative() returns true;
	 *
	 * @param Force The force to add, given in the world coordinate frame.
	 * @param Location The location in the world coordinate frame where the force should be applied.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void AddForceAtWorldLocation(FVector Force, FVector Location);

	/**
	 * Add an external force, given in the world coordinate frame, applied at a point specified in
	 * the local coordinate frame, that will be affecting this body in the next solve. If the
	 * location is different from the center of mass then a torque will be calculated and added as
	 * well.
	 *
	 * This member function may only be called when a native AGX Dynamics representation has already
	 * been created, i.e., HasNative() returns true;
	 *
	 * @param Force The force to add, given in world coordinate frame.
	 * @param Location The location in the local coordinate frame where the force should be applied.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void AddForceAtLocalLocation(FVector Force, FVector Location);

	/**
	 * Add an external force, given in the local coordinate frame, applied at a point specified in
	 * the local coordinate frame, that will be affecting this body in the next solve. If the
	 * location is different from the center of mass then a torque will be calculated and added as
	 * well.
	 *
	 * This member function may only be called when a native AGX Dynamics representation has already
	 * been created, i.e., HasNative() returns true;
	 *
	 * @param LocalForce The force to add, given in local coordinate frame.
	 * @param Location The location in the local coordinate frame where the force should be applied.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void AddLocalForceAtLocalLocation(FVector LocalForce, FVector Location);

	/**
	 * Get the external forces accumulated so far by the AddForce member functions, to be applied
	 * to this body in the next solve.
	 * @return The external forces for the next solve accumulated so far.
	 */
	UFUNCTION(BlueprintCallable, BlueprintPure, Category = "AGX Dynamics")
	FVector GetForce() const;

	/**
	 * Adds an external torque, given in center of mass coordinate frame, that will be affecting
	 * this body in the next solve [Nm].
	 *
	 * @param Torque The torque to add, given in center of mass coordinate frame.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void AddTorqueLocal(FVector Torque);

	/**
	 * Add an external torque, given in the world coordinate frame, that will be affecting this body
	 * in the next solve [Nm].
	 *
	 * @param Torque The torque to add, given in world coordinate frame.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void AddTorqueWorld(FVector Torque);

	/**
	 * Get the external torques accumulated so far by the AddTorque member functions, to be applied
	 * to this body in the next solve.
	 *
	 * @return The external torques for the next solve accumulated so far.
	 */
	FVector GetTorque() const;

	/// @see MoveTo_Rotator.
	void MoveTo(FVector Position, FRotator Rotation, double Duration);

	// @see MoveTo_Quat.
	void MoveTo(FVector Position, FQuat Rotation, double Duration);

	/**
	 * Sets a constant velocity and angular velocity such that the Rigid Body reaches the desired
	 * world Position and Rotation in the given Duration. Duration determines the time that the
	 * Rigid Body has to get to the final position [s].
	 * The velocity and angular velocity is left unchanged even after reaching the given position
	 * and rotation, i.e. the Rigid Body will continue to move in the same direction until the
	 * velocity is explicitly updated.
	 *
	 * If using this function with MotionControl set to anything other than Kinematics, the Rigid
	 * Body may not reach the desired Position and Rotation since external forces may act on it
	 * during the movement.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Rigid Body")
	void MoveTo_Rotator(FVector Position, FRotator Rotation, double Duration);

	/**
	 * Sets a constant velocity and angular velocity such that the Rigid Body reaches the desired
	 * world Position and Rotation in the given Duration. Duration determines the time that the
	 * Rigid Body has to get to the final position [s].
	 * The velocity and angular velocity is left unchanged even after reaching the given position
	 * and rotation, i.e. the Rigid Body will continue to move in the same direction until the
	 * velocity is explicitly updated.
	 *
	 * If using this function with MotionControl set to anything other than Kinematics, the Rigid
	 * Body may not reach the desired Position and Rotation since external forces may act on it
	 * during the movement.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Rigid Body")
	void MoveTo_Quat(FVector Position, FQuat Rotation, double Duration);

	/**
	 * Same as MoveTo, but using local coordinates.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void MoveToLocal(FVector PositionLocal, FRotator RotationLocal, float Duration);

	/**
	 * Find all Shape Components attached, including transitively, to this Rigid Body Component and
	 * add them to the body on the AGX Dynamics side as well.
	 *
	 * Should be called after attaching new Shape Components to a Rigid Body Component.
	 *
	 * @note Will only add shapes, currently does not support removing shapes.
	 */
	UFUNCTION(BlueprintCallable, Category = "AGX Dynamics")
	void SynchronizeShapes();

	/// Get the native AGX Dynamics representation of this rigid body. Create it if necessary.
	FRigidBodyBarrier* GetOrCreateNative();

	/// Return the native AGX Dynamics representation of this rigid body. May return nullptr.
	FRigidBodyBarrier* GetNative();

	const FRigidBodyBarrier* GetNative() const;

	/**
	 * Get access to the underlying native AGX Dynamics object that this Rigid Body Component
	 * represents. If there is no underlying AGX Dynamics object then the returned Barrier object
	 * will be invalid and Has Native will return false. Do not call any other function on a Barrier
	 * object for which Has Native returns false.
	 *
	 * @return Barrier for the native AGX Dynamics object this Rigid Body Component represents.
	 */
	UFUNCTION(
		BlueprintCallable, BlueprintPure, Category = "AGX Rigid Body",
		Meta = (DisplayName = "Get Native"))
	FRigidBodyBarrier& GetNativeByRef();

	// ~Begin IAGX_NativeOwner interface.
	virtual bool HasNative() const override;
	virtual uint64 GetNativeAddress() const override;
	virtual void SetNativeAddress(uint64 NativeAddress) override;
	// ~End IAGX_NativeOwner interface.

#if WITH_EDITOR
	// ~Begin UObject interface.
	virtual void PostInitProperties() override;
	virtual void PostEditChangeChainProperty(FPropertyChangedChainEvent& Event) override;
#endif
	// ~End UObject interface.

	void CopyFrom(const FRigidBodyBarrier& Barrier, FAGX_ImportContext* Context);

	static TArray<UAGX_RigidBodyComponent*> GetFromActor(const AActor* Actor);
	static UAGX_RigidBodyComponent* GetFirstFromActor(const AActor* Actor);

	//~ Begin UActorComponent Interface
	virtual void BeginPlay() override;
	virtual void EndPlay(const EEndPlayReason::Type Reason) override;
	virtual void TickComponent(
		float DeltaTime, ELevelTick TickType,
		FActorComponentTickFunction* ThisTickFunction) override;
	virtual TStructOnScope<FActorComponentInstanceData> GetComponentInstanceData() const override;
	//~ End UActorComponent Interface

	// ~Begin USceneComponent interface.
#if WITH_EDITOR
	virtual void PostEditComponentMove(bool bFinished) override;
	virtual void OnChildDetached(USceneComponent* Child) override;
	virtual void OnChildAttached(USceneComponent* Child) override;
#endif
	// ~End USceneComponent interface.

#if WITH_EDITOR
	void OnComponentView();
	bool TransformRootComponentAllowed() const;
#endif

	/*
	 * The import Guid of this Component. Only used by the AGX Dynamics for Unreal import system.
	 * Should never be assigned manually.
	 */
	UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "AGX Dynamics Import Guid")
	FGuid ImportGuid;

	/*
	 * The import name of this Component. Only used by the AGX Dynamics for Unreal import system.
	 * Should never be assigned manually.
	 */
	UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "AGX Dynamics Import Name")
	FString ImportName;

private: // Deprecated functions.
	UFUNCTION(
		BlueprintCallable, Category = "AGX Dynamics",
		Meta =
			(DeprecatedFunction,
			 DeprecationMessage = "Use CalculateMass instead of CalculateMass_BP"))
	float CalculateMass_BP() const;

private:
#if WITH_EDITOR
	// Fill in a bunch of callbacks in PropertyDispatcher so we don't have to manually check each
	// and every UPROPERTY in PostEditChangeProperty and PostEditChangeChainProperty.
	void InitPropertyDispatcher();
#endif

	// Create the native AGX Dynamics object.
	void InitializeNative();

	// Set native's MotionControl and ensure Unreal has corresponding mobility.
	void InitializeMotionControl();

	/**
	 * Should be called whenever properties (excluding transform and shapes) need to be pushed
	 * onto the native in runtime.
	 */
	void WritePropertiesToNative();

	/// A variant of WriteTransformToNative that only writes if we have a Native to write to.
	void TryWriteTransformToNative();

#if WITH_EDITOR
	virtual bool CanEditChange(const FProperty* InProperty) const override;
	void DisableTransformRootCompIfMultiple();
#endif

private:
	// The AGX Dynamics object only exists while simulating. Initialized in
	// BeginPlay and released in EndPlay.
	FRigidBodyBarrier NativeBarrier;
};
