191 lines
8.3 KiB
C
191 lines
8.3 KiB
C
|
// Copyright 2020 Phyronnaz
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include "CoreMinimal.h"
|
||
|
|
#include "VoxelAssets/VoxelDataAssetData.h"
|
||
|
|
#include "VoxelImporters/VoxelMeshImporter.h"
|
||
|
|
#include "VoxelTools/Tools/VoxelToolWithAlignment.h"
|
||
|
|
#include "VoxelMeshTool.generated.h"
|
||
|
|
|
||
|
|
class UTextureRenderTarget2D;
|
||
|
|
|
||
|
|
UCLASS()
|
||
|
|
class VOXEL_API UVoxelMeshTool : public UVoxelToolWithAlignment
|
||
|
|
{
|
||
|
|
GENERATED_BODY()
|
||
|
|
|
||
|
|
public:
|
||
|
|
UPROPERTY(Category = "Tool Preview Settings", EditAnywhere, BlueprintReadWrite, meta = (HideInPanel))
|
||
|
|
TObjectPtr<UMaterialInterface> ToolMaterial = nullptr;
|
||
|
|
|
||
|
|
public:
|
||
|
|
UPROPERTY(Category = "Tool Settings", EditAnywhere, BlueprintReadWrite)
|
||
|
|
TObjectPtr<UStaticMesh> Mesh = nullptr;
|
||
|
|
|
||
|
|
// Relative to the radius
|
||
|
|
UPROPERTY(Category = "Tool Settings", EditAnywhere, BlueprintReadWrite, meta = (UIMin=0, UIMax=1))
|
||
|
|
float Stride = 0.f;
|
||
|
|
|
||
|
|
// Do a smooth import by converting the voxel densities & the mesh to true distance fields, and doing a smooth union/subtraction on these
|
||
|
|
// NOTE: Disabled if bProgressiveStamp = true
|
||
|
|
// NOTE: Will disable bImportColorsFromMesh/bImportUVsFromMesh
|
||
|
|
UPROPERTY(Category = "Tool Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "!bProgressiveStamp"))
|
||
|
|
bool bSmoothImport = false;
|
||
|
|
|
||
|
|
// Relative to radius
|
||
|
|
UPROPERTY(Category = "Tool Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bSmoothImport && !bProgressiveStamp", UIMin = 0, UIMax = 1))
|
||
|
|
float Smoothness = 0.5f;
|
||
|
|
|
||
|
|
// Will slowly grow/shrink the surface towards the mesh
|
||
|
|
// NOTE: Will disable SmoothImport
|
||
|
|
// NOTE: Will disable bImportColorsFromMesh/bImportUVsFromMesh
|
||
|
|
UPROPERTY(Category = "Tool Settings", EditAnywhere, BlueprintReadWrite)
|
||
|
|
bool bProgressiveStamp = false;
|
||
|
|
|
||
|
|
// Speed of the progressive stamp
|
||
|
|
// Make sure your mesh is intersecting the voxel world!
|
||
|
|
UPROPERTY(Category = "Tool Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bProgressiveStamp", UIMin = 0, UIMax = 1))
|
||
|
|
float Speed = 0.1f;
|
||
|
|
|
||
|
|
UPROPERTY(Category = "Tool Settings", EditAnywhere, BlueprintReadWrite, AdvancedDisplay)
|
||
|
|
FVoxelMeshImporterSettingsBase MeshImporterSettings;
|
||
|
|
|
||
|
|
public:
|
||
|
|
UPROPERTY(Category = "Sculpt Settings", EditAnywhere, BlueprintReadWrite)
|
||
|
|
bool bSculpt = true;
|
||
|
|
|
||
|
|
public:
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite)
|
||
|
|
bool bPaint = true;
|
||
|
|
|
||
|
|
// Use to restrict editing on some channels
|
||
|
|
UPROPERTY(Category = "Paint Settings", AdvancedDisplay, EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint", Bitmask, BitmaskEnum = EVoxelMaterialMask_BP))
|
||
|
|
int32 PaintMask = EVoxelMaterialMask::All;
|
||
|
|
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint", ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
bool bPaintColors = true;
|
||
|
|
|
||
|
|
// Import the colors directly from the mesh by sampling ColorsMaterial at the mesh UVs
|
||
|
|
// NOTE: Will be disabled if bSmoothImport = true or bProgressiveStamp = true
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint && !bSmoothImport && !bProgressiveStamp && bPaintColors", ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
bool bImportColorsFromMesh = true;
|
||
|
|
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint && !bSmoothImport && !bProgressiveStamp && bPaintColors && bImportColorsFromMesh", ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
TObjectPtr<UMaterialInterface> ColorsMaterial = nullptr;
|
||
|
|
|
||
|
|
// Used if bImportColorsFromMesh = false
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint && bPaintColors", ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
FColor ColorToPaint = FColor::White;
|
||
|
|
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint", ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
bool bPaintUVs = true;
|
||
|
|
|
||
|
|
// Import the uvs directly from the mesh by sampling UVsMaterial at the mesh UVs
|
||
|
|
// NOTE: Will be disabled if bSmoothImport = true or bProgressiveStamp = true
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint && !bSmoothImport && !bProgressiveStamp && bPaintUVs", ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
bool bImportUVsFromMesh = true;
|
||
|
|
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint && !bSmoothImport && !bProgressiveStamp && bPaintUVs", ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
TObjectPtr<UMaterialInterface> UVsMaterial = nullptr;
|
||
|
|
|
||
|
|
// Used if bImportUVsFromMesh = false
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint && bPaintUVs", ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
FVector2D UV0ToPaint = FVector2D::ZeroVector;
|
||
|
|
|
||
|
|
// Used if bImportUVsFromMesh = false
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint && bPaintUVs", ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
FVector2D UV1ToPaint = FVector2D::ZeroVector;
|
||
|
|
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint", ShowForMaterialConfigs = "SingleIndex, MultiIndex"))
|
||
|
|
bool bPaintIndex = false;
|
||
|
|
|
||
|
|
UPROPERTY(Category = "Paint Settings", EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint && bPaintIndex", ShowForMaterialConfigs = "SingleIndex, MultiIndex"))
|
||
|
|
uint8 IndexToPaint = 0;
|
||
|
|
|
||
|
|
// For debug
|
||
|
|
UPROPERTY(Category = "Paint Settings", AdvancedDisplay, VisibleAnywhere, BlueprintReadOnly, Transient, meta = (ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
TObjectPtr<UTextureRenderTarget2D> UVsRenderTarget = nullptr;
|
||
|
|
|
||
|
|
// For debug
|
||
|
|
UPROPERTY(Category = "Paint Settings", AdvancedDisplay, VisibleAnywhere, BlueprintReadOnly, Transient, meta = (ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
TObjectPtr<UTextureRenderTarget2D> ColorsRenderTarget = nullptr;
|
||
|
|
|
||
|
|
UPROPERTY(Category = "Paint Settings", AdvancedDisplay, EditAnywhere, BlueprintReadWrite, meta = (EditCondition = "bPaint && !bSmoothImport && !bProgressiveStamp"), meta = (ShowForMaterialConfigs = "RGB, SingleIndex"))
|
||
|
|
int32 RenderTargetSize = 4096;
|
||
|
|
|
||
|
|
public:
|
||
|
|
// Relative to the size of the mesh
|
||
|
|
UPROPERTY(Category = "Transform", EditAnywhere, BlueprintReadWrite, meta = (UIMin = -1, UIMax = 1))
|
||
|
|
FVector PositionOffset = FVector::ZeroVector;
|
||
|
|
|
||
|
|
// If false the mesh scale will be set to match the radius
|
||
|
|
// If true the mesh scale will ignore the radius, and only use the scale below
|
||
|
|
UPROPERTY(Category = "Transform", EditAnywhere, BlueprintReadWrite)
|
||
|
|
bool bAbsoluteScale = false;
|
||
|
|
|
||
|
|
UPROPERTY(Category = "Transform", EditAnywhere, BlueprintReadWrite)
|
||
|
|
FVector Scale = FVector::OneVector;
|
||
|
|
|
||
|
|
UPROPERTY(Category = "Transform", EditAnywhere, BlueprintReadWrite)
|
||
|
|
bool bAlignToNormal = true;
|
||
|
|
|
||
|
|
UPROPERTY(Category = "Transform", EditAnywhere, BlueprintReadWrite)
|
||
|
|
bool bAlignToMovement = true;
|
||
|
|
|
||
|
|
// Applied after position and scale offset
|
||
|
|
UPROPERTY(Category = "Transform", EditAnywhere, BlueprintReadWrite)
|
||
|
|
FRotator RotationOffset = FRotator::ZeroRotator;
|
||
|
|
|
||
|
|
public:
|
||
|
|
UVoxelMeshTool();
|
||
|
|
|
||
|
|
//~ Begin UVoxelToolBase Interface
|
||
|
|
virtual void GetToolConfig(FVoxelToolBaseConfig& OutConfig) const override;
|
||
|
|
virtual void UpdateRender(UMaterialInstanceDynamic* OverlayMaterialInstance, UMaterialInstanceDynamic* MeshMaterialInstance) override;
|
||
|
|
virtual FVoxelIntBoxWithValidity DoEdit() override;
|
||
|
|
//~ End UVoxelToolBase Interface
|
||
|
|
|
||
|
|
private:
|
||
|
|
struct FMeshData
|
||
|
|
{
|
||
|
|
TWeakObjectPtr<UStaticMesh> StaticMesh;
|
||
|
|
FVoxelMeshImporterInputData Data;
|
||
|
|
FBox Bounds;
|
||
|
|
};
|
||
|
|
TUniquePtr<const FMeshData> CachedMeshData;
|
||
|
|
|
||
|
|
const FMeshData* GetMeshData();
|
||
|
|
void GetTransform(
|
||
|
|
const FMeshData& MeshData,
|
||
|
|
FVector& OutMeshScale,
|
||
|
|
FTransform& OutTransformNoTranslation,
|
||
|
|
FTransform& OutTransformWithTranslation) const;
|
||
|
|
|
||
|
|
private:
|
||
|
|
struct FAssetData
|
||
|
|
{
|
||
|
|
FTransform Transform;
|
||
|
|
FVoxelDataAssetData Data;
|
||
|
|
FIntVector PositionOffset = FIntVector::ZeroValue;
|
||
|
|
};
|
||
|
|
struct FDistanceFieldData
|
||
|
|
{
|
||
|
|
FTransform Transform;
|
||
|
|
float BoxExtension = 0;
|
||
|
|
FVoxelMeshImporterSettingsBase ImporterSettings;
|
||
|
|
|
||
|
|
TArray<float> Data;
|
||
|
|
TArray<FVector3f> SurfacePositions;
|
||
|
|
FIntVector Size;
|
||
|
|
FIntVector PositionOffset = FIntVector::ZeroValue;
|
||
|
|
};
|
||
|
|
TUniquePtr<const FAssetData> AssetData;
|
||
|
|
TUniquePtr<const FDistanceFieldData> DistanceFieldData;
|
||
|
|
|
||
|
|
UPROPERTY(Transient)
|
||
|
|
FVoxelMeshImporterRenderTargetCache RenderTargetCache;
|
||
|
|
|
||
|
|
UPROPERTY(Transient)
|
||
|
|
FVoxelMeshImporterSettings AssetData_ImporterSettings;
|
||
|
|
};
|