private void OnActivated(Entity<GravityGeneratorComponent> ent, ref ChargedMachineActivatedEvent args)
{
ent.Comp.GravityActive = true;
- if (TryComp<TransformComponent>(ent, out var xform) &&
- TryComp(xform.ParentUid, out GravityComponent? gravity))
+
+ var xform = Transform(ent);
+
+ if (TryComp(xform.ParentUid, out GravityComponent? gravity))
{
_gravitySystem.EnableGravity(xform.ParentUid, gravity);
}
private void OnDeactivated(Entity<GravityGeneratorComponent> ent, ref ChargedMachineDeactivatedEvent args)
{
ent.Comp.GravityActive = false;
- if (TryComp<TransformComponent>(ent, out var xform) &&
- TryComp(xform.ParentUid, out GravityComponent? gravity))
+
+ var xform = Transform(ent);
+
+ if (TryComp(xform.ParentUid, out GravityComponent? gravity))
{
_gravitySystem.RefreshGravity(xform.ParentUid, gravity);
}