Add Ragdoll Editor utilities for easier ragdoll setup and editing
This commit is contained in:
Wojtek Figat
@@ -1,6 +1,14 @@
|
||||
// Copyright (c) 2012-2021 Wojciech Figat. All rights reserved.
|
||||
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using FlaxEditor.GUI;
|
||||
using FlaxEditor.GUI.ContextMenu;
|
||||
using FlaxEditor.SceneGraph;
|
||||
using FlaxEditor.SceneGraph.Actors;
|
||||
using FlaxEngine;
|
||||
using FlaxEngine.GUI;
|
||||
|
||||
namespace FlaxEditor.CustomEditors.Dedicated
|
||||
{
|
||||
@@ -16,13 +24,295 @@ namespace FlaxEditor.CustomEditors.Dedicated
|
||||
{
|
||||
base.Initialize(layout);
|
||||
|
||||
// Add info box
|
||||
if (IsSingleObject && Values[0] is Ragdoll ragdoll)
|
||||
var ragdoll = Values.Count == 1 ? Values[0] as Ragdoll : null;
|
||||
if (ragdoll == null)
|
||||
return;
|
||||
|
||||
var editorGroup = layout.Group("Ragdoll Editor");
|
||||
editorGroup.Panel.Open(false);
|
||||
|
||||
// Info
|
||||
var text = $"Total mass: {Utils.RoundTo1DecimalPlace(ragdoll.TotalMass)}kg";
|
||||
var label = editorGroup.Label(text);
|
||||
label.Label.AutoHeight = true;
|
||||
|
||||
if (ragdoll.Parent is AnimatedModel animatedModel && animatedModel.SkinnedModel)
|
||||
{
|
||||
var text = $"Total mass: {Utils.RoundTo1DecimalPlace(ragdoll.TotalMass)}kg";
|
||||
var label = layout.Label(text);
|
||||
label.Label.AutoHeight = true;
|
||||
// Builder
|
||||
var grid = editorGroup.CustomContainer<UniformGridPanel>();
|
||||
var gridControl = grid.CustomControl;
|
||||
gridControl.ClipChildren = false;
|
||||
gridControl.Height = Button.DefaultHeight;
|
||||
gridControl.SlotsHorizontally = 3;
|
||||
gridControl.SlotsVertically = 1;
|
||||
grid.Button("Rebuild").Button.ButtonClicked += OnRebuild;
|
||||
grid.Button("Rebuild/Add bone").Button.ButtonClicked += OnRebuildBone;
|
||||
grid.Button("Remove bone").Button.ButtonClicked += OnRemoveBone;
|
||||
}
|
||||
|
||||
if (Presenter.Owner is Windows.PropertiesWindow || Presenter.Owner is Windows.Assets.PrefabWindow)
|
||||
{
|
||||
// Selection
|
||||
var grid = editorGroup.CustomContainer<UniformGridPanel>();
|
||||
var gridControl = grid.CustomControl;
|
||||
gridControl.ClipChildren = false;
|
||||
gridControl.Height = Button.DefaultHeight;
|
||||
gridControl.SlotsHorizontally = 3;
|
||||
gridControl.SlotsVertically = 1;
|
||||
grid.Button("Select all joints").Button.Clicked += OnSelectAllJoints;
|
||||
grid.Button("Select all colliders").Button.Clicked += OnSelectAllColliders;
|
||||
grid.Button("Select all bodies").Button.Clicked += OnSelectAllBodies;
|
||||
}
|
||||
}
|
||||
|
||||
class RebuildContextMenu : ContextMenuBase
|
||||
{
|
||||
private Action<AnimatedModelNode.RebuildOptions> _rebuild;
|
||||
private AnimatedModelNode.RebuildOptions _options = new AnimatedModelNode.RebuildOptions();
|
||||
|
||||
public RebuildContextMenu(Action<AnimatedModelNode.RebuildOptions> rebuild)
|
||||
{
|
||||
_rebuild = rebuild;
|
||||
|
||||
const float width = 280.0f;
|
||||
const float height = 220.0f;
|
||||
Size = new Vector2(width, height);
|
||||
|
||||
// Title
|
||||
var title = new Label(2, 2, width - 4, 23.0f)
|
||||
{
|
||||
Font = new FontReference(FlaxEngine.GUI.Style.Current.FontLarge),
|
||||
Text = "Ragdoll Options",
|
||||
Parent = this
|
||||
};
|
||||
|
||||
// Buttons
|
||||
var rebuildButton = new Button(2.0f, title.Bottom + 2.0f, width - 4.0f, 20.0f)
|
||||
{
|
||||
Text = "Rebuild",
|
||||
Parent = this
|
||||
};
|
||||
rebuildButton.Clicked += OnRebuild;
|
||||
|
||||
// Actual panel
|
||||
var panel1 = new Panel(ScrollBars.Vertical)
|
||||
{
|
||||
Bounds = new Rectangle(0, rebuildButton.Bottom + 2.0f, width, height - rebuildButton.Bottom - 2.0f),
|
||||
Parent = this
|
||||
};
|
||||
var editor = new CustomEditorPresenter(null);
|
||||
editor.Panel.AnchorPreset = AnchorPresets.HorizontalStretchTop;
|
||||
editor.Panel.IsScrollable = true;
|
||||
editor.Panel.Parent = panel1;
|
||||
|
||||
editor.Select(_options);
|
||||
}
|
||||
|
||||
private void OnRebuild()
|
||||
{
|
||||
Hide();
|
||||
_rebuild(_options);
|
||||
}
|
||||
|
||||
protected override void OnShow()
|
||||
{
|
||||
Focus();
|
||||
|
||||
base.OnShow();
|
||||
}
|
||||
|
||||
public override void Hide()
|
||||
{
|
||||
if (!Visible)
|
||||
return;
|
||||
|
||||
Focus(null);
|
||||
|
||||
base.Hide();
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
public override bool OnKeyDown(KeyboardKeys key)
|
||||
{
|
||||
if (key == KeyboardKeys.Escape)
|
||||
{
|
||||
Hide();
|
||||
return true;
|
||||
}
|
||||
|
||||
return base.OnKeyDown(key);
|
||||
}
|
||||
}
|
||||
|
||||
private void OnRebuild(Button button)
|
||||
{
|
||||
var cm = new RebuildContextMenu(Rebuild);
|
||||
cm.Show(button.Parent, button.BottomLeft);
|
||||
}
|
||||
|
||||
private void Rebuild(AnimatedModelNode.RebuildOptions options)
|
||||
{
|
||||
var ragdoll = (Ragdoll)Values[0];
|
||||
var animatedModel = (AnimatedModel)ragdoll.Parent;
|
||||
|
||||
// Remove existing bodies
|
||||
var bodies = ragdoll.Children.Where(x => x is RigidBody && x.IsActive);
|
||||
var actions = new List<IUndoAction>();
|
||||
foreach (var body in bodies)
|
||||
{
|
||||
var action = new Actions.DeleteActorsAction(new List<SceneGraphNode> { SceneGraphFactory.FindNode(body.ID) });
|
||||
action.Do();
|
||||
actions.Add(action);
|
||||
}
|
||||
Presenter.Undo?.AddAction(new MultiUndoAction(actions));
|
||||
|
||||
// Build ragdoll
|
||||
SceneGraph.Actors.AnimatedModelNode.BuildRagdoll(animatedModel, options, ragdoll);
|
||||
}
|
||||
|
||||
private void OnRebuildBone(Button button)
|
||||
{
|
||||
PickBone(button, RebuildBone, false);
|
||||
}
|
||||
|
||||
private void RebuildBone(string name)
|
||||
{
|
||||
var ragdoll = (Ragdoll)Values[0];
|
||||
var animatedModel = (AnimatedModel)ragdoll.Parent;
|
||||
|
||||
// Remove existing body
|
||||
var bodies = ragdoll.Children.Where(x => x is RigidBody && x.IsActive);
|
||||
var body = bodies.FirstOrDefault(x => x.Name == name);
|
||||
if (body != null)
|
||||
{
|
||||
var action = new Actions.DeleteActorsAction(new List<SceneGraphNode> { SceneGraphFactory.FindNode(body.ID) });
|
||||
action.Do();
|
||||
Presenter.Undo?.AddAction(action);
|
||||
}
|
||||
|
||||
// Build ragdoll
|
||||
SceneGraph.Actors.AnimatedModelNode.BuildRagdoll(animatedModel, new AnimatedModelNode.RebuildOptions(), ragdoll, name);
|
||||
}
|
||||
|
||||
private void OnRemoveBone(Button button)
|
||||
{
|
||||
PickBone(button, RemoveBone, true);
|
||||
}
|
||||
|
||||
private void RemoveBone(string name)
|
||||
{
|
||||
var ragdoll = (Ragdoll)Values[0];
|
||||
var bodies = ragdoll.Children.Where(x => x is RigidBody && x.IsActive);
|
||||
var joints = bodies.SelectMany(y => y.Children).Where(z => z is Joint && z.IsActive).Cast<Joint>();
|
||||
var body = bodies.First(x => x.Name == name);
|
||||
var replacementJoint = joints.FirstOrDefault(x => x.Parent == body && x.Target != null);
|
||||
|
||||
// Fix joints using this bone
|
||||
foreach (var joint in joints)
|
||||
{
|
||||
if (joint.Target == body)
|
||||
{
|
||||
if (replacementJoint != null)
|
||||
{
|
||||
// Swap the joint target to the parent of the removed body to keep ragdoll connected
|
||||
using (new UndoBlock(Presenter.Undo, joint, "Fix joint"))
|
||||
{
|
||||
joint.Target = replacementJoint.Target;
|
||||
joint.EnableAutoAnchor = true;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// Remove joint that will no longer be valid
|
||||
var action = new Actions.DeleteActorsAction(new List<SceneGraphNode> { SceneGraphFactory.FindNode(joint.ID) });
|
||||
action.Do();
|
||||
Presenter.Undo?.AddAction(action);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Remove body
|
||||
{
|
||||
var action = new Actions.DeleteActorsAction(new List<SceneGraphNode> { SceneGraphFactory.FindNode(body.ID) });
|
||||
action.Do();
|
||||
Presenter.Undo?.AddAction(action);
|
||||
}
|
||||
}
|
||||
|
||||
private void PickBone(Button button, Action<string> action, bool showOnlyExisting)
|
||||
{
|
||||
// Show context menu with list of bones to pick
|
||||
var cm = new ItemsListContextMenu(280);
|
||||
var ragdoll = (Ragdoll)Values[0];
|
||||
var bodies = ragdoll.Children.Where(x => x is RigidBody && x.IsActive);
|
||||
var animatedModel = (AnimatedModel)ragdoll.Parent;
|
||||
animatedModel.SkinnedModel.WaitForLoaded();
|
||||
var nodes = animatedModel.SkinnedModel.Nodes;
|
||||
var bones = animatedModel.SkinnedModel.Bones;
|
||||
foreach (var bone in bones)
|
||||
{
|
||||
var node = nodes[bone.NodeIndex];
|
||||
if (showOnlyExisting && !bodies.Any(x => x.Name == node.Name))
|
||||
continue;
|
||||
string prefix = string.Empty, tooltip = node.Name;
|
||||
var boneParentIndex = bone.ParentIndex;
|
||||
while (boneParentIndex != -1)
|
||||
{
|
||||
prefix += " ";
|
||||
tooltip = nodes[bones[boneParentIndex].NodeIndex].Name + " > " + tooltip;
|
||||
boneParentIndex = bones[boneParentIndex].ParentIndex;
|
||||
}
|
||||
var item = new ItemsListContextMenu.Item
|
||||
{
|
||||
Name = prefix + node.Name,
|
||||
TooltipText = tooltip,
|
||||
Tag = node.Name,
|
||||
};
|
||||
if (!showOnlyExisting && !bodies.Any(x => x.Name == node.Name))
|
||||
item.TintColor = new Color(1, 0.8f, 0.8f, 0.6f);
|
||||
cm.AddItem(item);
|
||||
}
|
||||
cm.ItemClicked += item => action((string)item.Tag);
|
||||
cm.SortChildren();
|
||||
cm.Show(button.Parent, button.BottomLeft);
|
||||
}
|
||||
|
||||
private void OnSelectAllJoints()
|
||||
{
|
||||
var ragdoll = (Ragdoll)Values[0];
|
||||
var bodies = ragdoll.Children.Where(x => x is RigidBody && x.IsActive);
|
||||
var joints = bodies.SelectMany(y => y.Children).Where(z => z is Joint && z.IsActive);
|
||||
Select(joints);
|
||||
}
|
||||
|
||||
private void OnSelectAllColliders()
|
||||
{
|
||||
var ragdoll = (Ragdoll)Values[0];
|
||||
var bodies = ragdoll.Children.Where(x => x is RigidBody && x.IsActive);
|
||||
var colliders = bodies.SelectMany(y => y.Children).Where(z => z is Collider && z.IsActive);
|
||||
Select(colliders);
|
||||
}
|
||||
|
||||
private void OnSelectAllBodies()
|
||||
{
|
||||
var ragdoll = (Ragdoll)Values[0];
|
||||
var bodies = ragdoll.Children.Where(x => x is RigidBody && x.IsActive);
|
||||
Select(bodies);
|
||||
}
|
||||
|
||||
private void Select(IEnumerable<Actor> list)
|
||||
{
|
||||
var selection = new List<SceneGraphNode>();
|
||||
foreach (var e in list)
|
||||
{
|
||||
var node = SceneGraphFactory.FindNode(e.ID);
|
||||
if (node != null)
|
||||
selection.Add(node);
|
||||
}
|
||||
if (Presenter.Owner is Windows.PropertiesWindow propertiesWindow)
|
||||
propertiesWindow.Editor.SceneEditing.Select(selection);
|
||||
else if (Presenter.Owner is Windows.Assets.PrefabWindow prefabWindow)
|
||||
prefabWindow.Select(selection);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user