mirror of
https://gitgud.io/AbstractConcept/rimworld-animation-studio.git
synced 2024-08-15 00:43:27 +00:00
152 lines
5.6 KiB
C#
152 lines
5.6 KiB
C#
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using UnityEngine;
|
|
using UnityEngine.EventSystems;
|
|
|
|
namespace RimWorldAnimationStudio
|
|
{
|
|
public class ActorBodyPart : MonoBehaviour, IPointerClickHandler, IDragHandler, IEndDragHandler
|
|
{
|
|
public SpriteRenderer bodyPartRenderer;
|
|
public ActorBody parent;
|
|
public string bodyPart;
|
|
|
|
private Vector3 dragDelta = new Vector3();
|
|
|
|
public void Start()
|
|
{
|
|
EventsManager.onActorBodyPartSelected.AddListener(delegate (ActorBodyPart bodyPart) { OnActorBodyPartSelected(bodyPart); });
|
|
EventsManager.onActorBodySelected.AddListener(delegate (ActorBody actorBody) { OnActorBodySelected(actorBody); });
|
|
|
|
if (Workspace.ActorID == parent.actorID)
|
|
{ parent.Activate(); }
|
|
}
|
|
|
|
public void OnActorAddonChange(ActorAddon actorAddon)
|
|
{
|
|
if (actorAddon.AddonName == bodyPart)
|
|
{ gameObject?.SetActive(actorAddon.Render); }
|
|
}
|
|
|
|
public void OnActorBodySelected(ActorBody actorBody)
|
|
{
|
|
if (actorBody == parent)
|
|
{ bodyPartRenderer.color = Constants.ColorLightGreen; }
|
|
|
|
else
|
|
{ bodyPartRenderer.color = Constants.ColorWhite; }
|
|
}
|
|
|
|
public void OnActorBodyPartSelected(ActorBodyPart bodyPart)
|
|
{
|
|
if (bodyPart == this)
|
|
{ bodyPartRenderer.color = Constants.ColorGreen; }
|
|
|
|
else if (bodyPart.parent == parent)
|
|
{ bodyPartRenderer.color = Constants.ColorLightGreen; }
|
|
|
|
else
|
|
{ bodyPartRenderer.color = Constants.ColorWhite; }
|
|
}
|
|
|
|
public void OnPointerClick(PointerEventData eventData)
|
|
{
|
|
if (eventData.pointerCurrentRaycast.gameObject.GetComponent<ActorBodyPart>() == null)
|
|
{ return; }
|
|
|
|
Activate();
|
|
}
|
|
|
|
public void OnDrag(PointerEventData eventData)
|
|
{
|
|
Activate();
|
|
|
|
PawnKeyframe keyframe = Workspace.GetCurrentPawnKeyframe(true);
|
|
Vector3 mousePosition = Camera.main.ScreenToWorldPoint(Input.mousePosition);
|
|
mousePosition = new Vector3(mousePosition.x, mousePosition.y, 0f);
|
|
|
|
if (dragDelta == Vector3.zero)
|
|
{ dragDelta = mousePosition - transform.position; }
|
|
|
|
if (bodyPart.ToLower() == "head")
|
|
{
|
|
if (Workspace.actorManipulationMode == ActorManipulationMode.Pan)
|
|
{
|
|
// It's stupid, but it works
|
|
Vector3 localPosA = transform.localPosition;
|
|
transform.position = mousePosition - dragDelta;
|
|
Vector3 localPosB = transform.localPosition;
|
|
transform.localPosition = localPosA;
|
|
|
|
keyframe.HeadBob += localPosB.y - localPosA.y;
|
|
}
|
|
|
|
else if (Workspace.actorManipulationMode == ActorManipulationMode.Rotate)
|
|
{
|
|
float angle = -Vector2.SignedAngle(Vector2.down, (Vector2)mousePosition - (Vector2)transform.position);
|
|
keyframe.HeadAngle = angle;
|
|
}
|
|
|
|
else if (Workspace.actorManipulationMode == ActorManipulationMode.Face)
|
|
{
|
|
float angle = Vector2.SignedAngle(Vector2.up, (Vector2)mousePosition - (Vector2)transform.position);
|
|
int facing = -Mathf.RoundToInt(angle / 90f);
|
|
facing = facing < 0 ? facing + 4 : facing;
|
|
|
|
keyframe.HeadFacing = facing;
|
|
}
|
|
}
|
|
|
|
else if (bodyPart.ToLower() == "appendage")
|
|
{
|
|
if (Workspace.actorManipulationMode == ActorManipulationMode.Rotate)
|
|
{
|
|
float angle = -Vector2.SignedAngle(Vector2.up, (Vector2)mousePosition - (Vector2)transform.position);
|
|
keyframe.GenitalAngle = angle;
|
|
}
|
|
}
|
|
|
|
else
|
|
{
|
|
AddonKeyframe addonKeyframe = keyframe.GetAddonKeyframe(bodyPart);
|
|
ActorAddon addon = Workspace.GetCurrentPawnAnimationClip().GetActorAddon(bodyPart);
|
|
|
|
if (Workspace.actorManipulationMode == ActorManipulationMode.Pan)
|
|
{
|
|
ActorBody anchoringActorBody = AnimationController.Instance.actorBodies.GetComponentsInChildren<ActorBody>()?.FirstOrDefault(x => x.actorID == addon.AnchoringActor);
|
|
Vector3 anchor = PawnUtility.GetBodyPartAnchor(anchoringActorBody, addon.anchorName);
|
|
transform.position = mousePosition - dragDelta;
|
|
|
|
addonKeyframe.PosX = transform.position.x - anchor.x;
|
|
addonKeyframe.PosZ = transform.position.y - anchor.y;
|
|
}
|
|
|
|
else if (Workspace.actorManipulationMode == ActorManipulationMode.Rotate)
|
|
{
|
|
float angle = -Vector2.SignedAngle(Vector2.down, (Vector2)mousePosition - (Vector2)transform.position);
|
|
addonKeyframe.Rotation = angle;
|
|
}
|
|
}
|
|
|
|
PawnAnimationClip clip = Workspace.GetPawnAnimationClip(parent.actorID);
|
|
clip.BuildSimpleCurves();
|
|
|
|
EventsManager.OnPawnKeyframeChanged(keyframe);
|
|
}
|
|
|
|
public void OnEndDrag(PointerEventData eventData)
|
|
{
|
|
Workspace.RecordEvent("Actor position / orientation");
|
|
dragDelta = Vector3.zero;
|
|
}
|
|
|
|
public void Activate()
|
|
{
|
|
Workspace.ActorID = parent.actorID;
|
|
Workspace.selectedBodyPart = this;
|
|
|
|
EventsManager.OnActorBodyPartSelected(this);
|
|
}
|
|
}
|
|
}
|