UnrealEngineUWP/Engine/Plugins/Experimental/MeshModelingToolset/Source/MeshModelingTools/Public/AlignObjectsTool.h

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

#pragma once

#include "CoreMinimal.h"
#include "MultiSelectionTool.h"
#include "InteractiveToolBuilder.h"
#include "BaseBehaviors/BehaviorTargetInterfaces.h"
#include "Changes/TransformChange.h"
#include "FrameTypes.h"
#include "BoxTypes.h"
#include "AlignObjectsTool.generated.h"

class UPrimitiveComponent;

/**
 *
 */
UCLASS()
class MESHMODELINGTOOLS_API UAlignObjectsToolBuilder : public UInteractiveToolBuilder
{
	GENERATED_BODY()

public:
	virtual bool CanBuildTool(const FToolBuilderState& SceneState) const override;
	virtual UInteractiveTool* BuildTool(const FToolBuilderState& SceneState) const override;

protected:
	virtual const FToolTargetTypeRequirements& GetTargetRequirements() const override;
};


UENUM()
enum class EAlignObjectsAlignTypes
{
	Pivots,
	BoundingBoxes
};

UENUM()
enum class EAlignObjectsAlignToOptions
{
	FirstSelected,
	LastSelected,
	Combined
};

UENUM()
enum class EAlignObjectsBoxPoint
{
	Center,
	Bottom,
	Top,
	Left,
	Right,
	Front,
	Back,
	Min,
	Max
};



/**
 * Standard properties of the Align Objects Operation
 */
UCLASS()
class MESHMODELINGTOOLS_API UAlignObjectsToolProperties : public UInteractiveToolPropertySet
{
	GENERATED_BODY()
public:

	UPROPERTY(EditAnywhere, Category = Options)
	EAlignObjectsAlignTypes AlignType = EAlignObjectsAlignTypes::BoundingBoxes;

	UPROPERTY(EditAnywhere, Category = Options)
	EAlignObjectsAlignToOptions AlignTo = EAlignObjectsAlignToOptions::Combined;

	UPROPERTY(EditAnywhere, Category = Options, meta = (EditCondition = "AlignType == EAlignObjectsAlignTypes::BoundingBoxes || AlignTo == EAlignObjectsAlignToOptions::Combined"))
	EAlignObjectsBoxPoint BoxPosition = EAlignObjectsBoxPoint::Center;

	UPROPERTY(EditAnywhere, Category = Axes)
	bool bAlignX = false;

	UPROPERTY(EditAnywhere, Category = Axes)
	bool bAlignY = false;

	UPROPERTY(EditAnywhere, Category = Axes)
	bool bAlignZ = true;
};





/**
 * UAlignObjectsTool transforms the input Components so that they are aligned in various ways, depending on the current settings.
 * The object positions move after every change in the parameters. Currently those changes are not transacted.
 * On cancel the original positions are restored, and on accept the positions are updated with a transaction.
 */
UCLASS()
class MESHMODELINGTOOLS_API UAlignObjectsTool : public UMultiSelectionTool, public IClickDragBehaviorTarget
{
	GENERATED_BODY()
	using FTransform3d = UE::Geometry::FTransform3d;
	using FAxisAlignedBox3d = UE::Geometry::FAxisAlignedBox3d;
public:
	UAlignObjectsTool();

	virtual void RegisterActions(FInteractiveToolActionSet& ActionSet) override;

	virtual void SetWorld(UWorld* World, UInteractiveGizmoManager* GizmoManager);

	virtual void Setup() override;
	virtual void Shutdown(EToolShutdownType ShutdownType) override;

	virtual void OnTick(float DeltaTime) override;
	virtual void Render(IToolsContextRenderAPI* RenderAPI) override;

	virtual bool HasCancel() const override { return true; }
	virtual bool HasAccept() const override { return true; }
	virtual bool CanAccept() const override { return true; }

	virtual void OnPropertyModified(UObject* PropertySet, FProperty* Property) override;


	// ICLickDragBehaviorTarget interface
	virtual FInputRayHit CanBeginClickDragSequence(const FInputDeviceRay& PressPos) override;
	virtual void OnClickPress(const FInputDeviceRay& PressPos) override;
	virtual void OnClickDrag(const FInputDeviceRay& DragPos) override;
	virtual void OnClickRelease(const FInputDeviceRay& ReleasePos) override;
	virtual void OnTerminateDragSequence() override;

public:
	UPROPERTY()
	UAlignObjectsToolProperties* AlignProps;


protected:
	UWorld* TargetWorld;
	UInteractiveGizmoManager* GizmoManager;


	struct FAlignInfo
	{
		UPrimitiveComponent* Component;
		FTransform SavedTransform;
		FTransform3d WorldTransform;
		FVector3d WorldPivot;
		FAxisAlignedBox3d WorldBounds;
	};

	TArray<FAlignInfo> ComponentInfo;

	FAxisAlignedBox3d CombinedBounds;
	FAxisAlignedBox3d PivotBounds;
	FVector3d AveragePivot;

	void Precompute();

	bool bAlignDirty = false;
	void UpdateAlignment();
	void UpdateAlignment_Pivots();
	void UpdateAlignment_BoundingBoxes();
};