Files
HauntedBloodlines/Assets/Obi/Scripts/Common/Collisions/ObiRigidbody2D.cs
2025-05-29 22:31:40 +03:00

88 lines
3.0 KiB
C#

using UnityEngine;
using System;
using System.Collections;
namespace Obi
{
/**
* Small helper class that lets you specify Obi-only properties for rigidbodies.
*/
[ExecuteInEditMode]
[RequireComponent(typeof(Rigidbody2D))]
public class ObiRigidbody2D : ObiRigidbodyBase
{
public Rigidbody2D unityRigidbody { get; private set; }
public Vector2 position => unityRigidbody.position;
public float rotation => unityRigidbody.rotation;
public Vector2 linearVelocity { get; protected set; }
public float angularVelocity { get; protected set; }
private Quaternion prevRotation;
private Vector3 prevPosition;
public override void OnEnable()
{
unityRigidbody = GetComponent<Rigidbody2D>();
prevPosition = transform.position;
prevRotation = transform.rotation;
linearVelocity = unityRigidbody.linearVelocity;
angularVelocity = unityRigidbody.angularVelocity;
base.OnEnable();
}
private void UpdateKinematicVelocities(float stepTime)
{
// differentiate positions/orientations to get our own velocites for kinematic objects.
// when calling Physics.Simulate, MovePosition/Rotation do not work correctly. Also useful for animations.
if (unityRigidbody.isKinematic)
{
// differentiate positions to obtain linear velocity:
linearVelocity = (transform.position - prevPosition) / stepTime;
// differentiate rotations to obtain angular velocity:
Quaternion delta = transform.rotation * Quaternion.Inverse(prevRotation);
angularVelocity = delta.z * Mathf.Rad2Deg * 2.0f / stepTime;
}
else
{
// if the object is non-kinematic, just copy velocities.
linearVelocity = unityRigidbody.linearVelocity;
angularVelocity = unityRigidbody.angularVelocity;
}
prevPosition = transform.position;
prevRotation = transform.rotation;
}
public override void UpdateIfNeeded(float stepTime)
{
UpdateKinematicVelocities(stepTime);
var world = ObiColliderWorld.GetInstance();
var rb = world.rigidbodies[handle.index];
rb.FromRigidbody(this);
world.rigidbodies[handle.index] = rb;
}
/**
* Reads velocities back from the solver.
*/
public override void UpdateVelocities(Vector3 linearDelta, Vector3 angularDelta)
{
// kinematic rigidbodies are passed to Obi with zero velocity, so we must ignore the new velocities calculated by the solver:
if (Application.isPlaying && !(unityRigidbody.isKinematic || kinematicForParticles))
{
unityRigidbody.linearVelocity += new Vector2(linearDelta.x, linearDelta.y);
unityRigidbody.angularVelocity += angularDelta[2] * Mathf.Rad2Deg;
}
}
}
}