wehub-resource-sync
106 lines
3.3 KiB
C#
106 lines
3.3 KiB
C#
using T3.Core.DataTypes.ShaderGraph;
|
|
using T3.Core.Utils;
|
|
|
|
namespace Lib.field.generate.sdf;
|
|
|
|
[Guid("b85198a8-6475-45e1-b2d2-83ec8f59d6ed")]
|
|
internal sealed class CappedTorusSDF : Instance<CappedTorusSDF>
|
|
, ITransformable
|
|
, IGraphNodeOp
|
|
{
|
|
[Output(Guid = "170b545f-c7c0-47d9-8b19-d8d8e8a3d2fa")]
|
|
public readonly Slot<ShaderGraphNode> Result = new();
|
|
|
|
// ITransformable interface implementation (Gizmo support)
|
|
IInputSlot ITransformable.TranslationInput => Center;
|
|
IInputSlot ITransformable.RotationInput => null;
|
|
IInputSlot ITransformable.ScaleInput => null;
|
|
public Action<Instance, EvaluationContext> TransformCallback { get; set; }
|
|
|
|
public CappedTorusSDF()
|
|
{
|
|
ShaderNode = new ShaderGraphNode(this);
|
|
Result.Value = ShaderNode;
|
|
Result.UpdateAction += Update;
|
|
}
|
|
|
|
private void Update(EvaluationContext context)
|
|
{
|
|
TransformCallback?.Invoke(this, context); // Needed for Gizmo support
|
|
|
|
var axis = Axis.GetEnumValue<AxisTypes>(context);
|
|
|
|
var templateChanged = axis != _axis;
|
|
if (templateChanged)
|
|
{
|
|
_axis = axis;
|
|
ShaderNode.FlagCodeChanged();
|
|
}
|
|
|
|
ShaderNode.Update(context);
|
|
}
|
|
|
|
public ShaderGraphNode ShaderNode { get; }
|
|
|
|
void IGraphNodeOp.AddDefinitions(CodeAssembleContext c)
|
|
{
|
|
c.Globals["fCappedTorus"] = """
|
|
float fCappedTorus(float3 p, float size, float ra, float rb)
|
|
{
|
|
float an = 2.5 * (0.5 + 0.5 * (size * 1.1 + 3));
|
|
float2 sc = float2(sin(an),cos(an));
|
|
p.x = abs(p.x);
|
|
float k = (sc.y*p.x>sc.x*p.y) ? dot(p.xy,sc) : length(p.xy);
|
|
return sqrt(dot(p,p) + ra*ra - 2.0*ra*k ) - rb;
|
|
}
|
|
""";
|
|
}
|
|
|
|
public void GetPreShaderCode(CodeAssembleContext c, int inputIndex)
|
|
{
|
|
var a = _axisCodes0[(int)_axis];
|
|
|
|
|
|
|
|
var n = ShaderNode;
|
|
c.AppendCall($"f{c}.w = fCappedTorus(p{c}.{a} - {n}Center.{a}, {n}Fill, {n}Radius, {n}Thickness);");
|
|
c.AppendCall($"f{c}.xyz = p.w < 0.5 ? p{c}.xyz : 1;"); // save local space
|
|
}
|
|
|
|
|
|
private readonly string[] _axisCodes0 =
|
|
[
|
|
"yzx",
|
|
"xzy",
|
|
"xyz",
|
|
];
|
|
|
|
private AxisTypes _axis;
|
|
|
|
private enum AxisTypes
|
|
{
|
|
X,
|
|
Y,
|
|
Z,
|
|
}
|
|
|
|
|
|
[GraphParam]
|
|
[Input(Guid = "f5a24e80-741a-49c5-9a0b-d7074e80940a")]
|
|
public readonly InputSlot<Vector3> Center = new();
|
|
|
|
[GraphParam]
|
|
[Input(Guid = "d37914dd-7bbd-47fa-ad90-9921621afb9e")]
|
|
public readonly InputSlot<float> Fill = new();
|
|
|
|
[GraphParam]
|
|
[Input(Guid = "00cec9de-3be4-403f-8677-dcc2ef473708")]
|
|
public readonly InputSlot<float> Radius = new();
|
|
|
|
[GraphParam]
|
|
[Input(Guid = "e1dc5cdc-24d2-409f-9ea6-6dd14fdb70d3")]
|
|
public readonly InputSlot<float> Thickness = new();
|
|
|
|
[Input(Guid = "01DEC852-5B80-4705-8648-C6FD68199E83", MappedType = typeof(AxisTypes))]
|
|
public readonly InputSlot<int> Axis = new();
|
|
} |