"Fix" conveyors (#6301)

This commit is contained in:
metalgearsloth
2022-01-31 01:30:15 +11:00
committed by GitHub
parent f392997597
commit 9a5a8a55e1
4 changed files with 93 additions and 80 deletions

View File

@@ -21,7 +21,6 @@ namespace Content.Server.Conveyor
public class ConveyorSystem : EntitySystem
{
[Dependency] private StunSystem _stunSystem = default!;
[Dependency] private IEntityLookup _entityLookup = default!;
public override void Initialize()
{
@@ -111,55 +110,5 @@ namespace Content.Server.Conveyor
return true;
}
/// <summary>
/// Calculates the angle in which entities on top of this conveyor
/// belt are pushed in
/// </summary>
/// <returns>
/// The angle when taking into account if the conveyor is reversed
/// </returns>
public Angle GetAngle(ConveyorComponent component)
{
var adjustment = component.State == ConveyorState.Reversed ? MathHelper.Pi/2 : -MathHelper.Pi/2;
var radians = MathHelper.DegreesToRadians(component.Angle);
return new Angle(EntityManager.GetComponent<TransformComponent>(component.Owner).LocalRotation.Theta + radians + adjustment);
}
public IEnumerable<(EntityUid, IPhysBody)> GetEntitiesToMove(ConveyorComponent comp)
{
//todo uuuhhh cache this
foreach (var entity in _entityLookup.GetEntitiesIntersecting(comp.Owner, flags: LookupFlags.Approximate))
{
if (Deleted(entity))
{
continue;
}
if (entity == comp.Owner)
{
continue;
}
if (!EntityManager.TryGetComponent(entity, out IPhysBody? physics) ||
physics.BodyType == BodyType.Static || physics.BodyStatus == BodyStatus.InAir || entity.IsWeightless())
{
continue;
}
if (EntityManager.HasComponent<IMapGridComponent>(entity))
{
continue;
}
if (entity.IsInContainer())
{
continue;
}
yield return (entity, physics);
}
}
}
}

View File

@@ -256,7 +256,7 @@ namespace Content.Server.MachineLinking.System
new SignalReceivedEvent(receiverPort, link.Transmitterport.Signal));
entity.PopupMessageCursor(Loc.GetString("signal-linker-component-linked-port", ("port", receiverPort),
("machine", receiver)));
("machine", receiver.Owner)));
}
}

View File

@@ -1,19 +1,24 @@
using System;
using System.Collections.Generic;
using Content.Server.Conveyor;
using Content.Server.Recycling.Components;
using Content.Shared.Conveyor;
using Content.Shared.Movement.Components;
using Robust.Shared.Containers;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Map;
using Robust.Shared.Maths;
using Robust.Shared.Physics;
using Robust.Shared.Physics.Controllers;
namespace Content.Server.Physics.Controllers
{
internal sealed class ConveyorController : VirtualController
public sealed class ConveyorController : VirtualController
{
[Dependency] private readonly IEntityLookup _lookup = default!;
[Dependency] private readonly IMapManager _mapManager = default!;
[Dependency] private readonly ConveyorSystem _conveyor = default!;
[Dependency] private readonly SharedContainerSystem _container = default!;
public override void Initialize()
{
@@ -25,43 +30,110 @@ namespace Content.Server.Physics.Controllers
public override void UpdateBeforeSolve(bool prediction, float frameTime)
{
base.UpdateBeforeSolve(prediction, frameTime);
foreach (var comp in EntityManager.EntityQuery<ConveyorComponent>())
var conveyed = new HashSet<EntityUid>();
// TODO: This won't work if someone wants a massive fuckoff conveyor so look at using StartCollide or something.
foreach (var (comp, xform) in EntityManager.EntityQuery<ConveyorComponent, TransformComponent>())
{
Convey(_conveyor, comp, frameTime);
Convey(comp, xform, conveyed, frameTime);
}
}
private void Convey(ConveyorSystem system, ConveyorComponent comp, float frameTime)
private void Convey(ConveyorComponent comp, TransformComponent xform, HashSet<EntityUid> conveyed, float frameTime)
{
// Use an event for conveyors to know what needs to run
if (!system.CanRun(comp))
if (!_conveyor.CanRun(comp))
{
return;
}
var direction = system.GetAngle(comp).ToVec();
var entMan = IoCManager.Resolve<IEntityManager>();
var ownerPos = entMan.GetComponent<TransformComponent>(comp.Owner).WorldPosition;
var speed = comp.Speed;
foreach (var (entity, physics) in EntitySystem.Get<ConveyorSystem>().GetEntitiesToMove(comp))
if (speed <= 0f) return;
var (conveyorPos, conveyorRot) = xform.GetWorldPositionRotation();
conveyorRot += comp.Angle;
if (comp.State == ConveyorState.Reversed)
{
var itemRelativeToConveyor = entMan.GetComponent<TransformComponent>(entity).WorldPosition - ownerPos;
physics.LinearVelocity += Convey(direction, comp.Speed, frameTime, itemRelativeToConveyor);
conveyorRot += MathF.PI;
}
var direction = conveyorRot.ToWorldVec();
foreach (var (entity, transform) in GetEntitiesToMove(comp, xform))
{
if (!conveyed.Add(entity)) continue;
var worldPos = transform.WorldPosition;
var itemRelative = conveyorPos - worldPos;
worldPos += Convey(direction, speed, frameTime, itemRelative);
transform.WorldPosition = worldPos;
}
}
private Vector2 Convey(Vector2 direction, float speed, float frameTime, Vector2 itemRelativeToConveyor)
private static Vector2 Convey(Vector2 direction, float speed, float frameTime, Vector2 itemRelative)
{
if(speed == 0 || direction.Length == 0) return Vector2.Zero;
if (speed == 0 || direction.Length == 0) return Vector2.Zero;
/* TODO: Figure out how to fix corner cuts.
direction = direction.Normalized;
var dirNormal = new Vector2(direction.Y, direction.X);
var dot = Vector2.Dot(itemRelativeToConveyor, dirNormal);
var dot = Vector2.Dot(itemRelative, dirNormal);
*/
var velocity = direction * speed * 5;
var velocity = direction * speed;
return velocity * frameTime;
/*
velocity += dirNormal * speed * -dot;
return velocity * frameTime;
*/
}
public IEnumerable<(EntityUid, TransformComponent)> GetEntitiesToMove(ConveyorComponent comp, TransformComponent xform)
{
if (!_mapManager.TryGetGrid(xform.GridID, out var grid) ||
!grid.TryGetTileRef(xform.Coordinates, out var tile)) yield break;
var tileAABB = _lookup.GetLocalBounds(tile, grid.TileSize).Enlarged(0.01f);
var gridMatrix = Transform(grid.GridEntityId).InvWorldMatrix;
foreach (var entity in _lookup.GetEntitiesIntersecting(tile))
{
if (entity == comp.Owner ||
Deleted(entity) ||
HasComp<IMapGridComponent>(entity)) continue;
if (!TryComp(entity, out PhysicsComponent? physics) ||
physics.BodyType == BodyType.Static ||
physics.BodyStatus == BodyStatus.InAir ||
entity.IsWeightless(physics, entityManager: EntityManager))
{
continue;
}
if (_container.IsEntityInContainer(entity))
{
continue;
}
// Yes there's still going to be the occasional rounding issue where it stops getting conveyed
// When you fix the corner issue that will fix this anyway.
var transform = Transform(entity);
var gridPos = gridMatrix.Transform(transform.WorldPosition);
var gridAABB = new Box2(gridPos - 0.1f, gridPos + 0.1f);
if (!tileAABB.Intersects(gridAABB)) continue;
yield return (entity, transform);
}
}
}
}

View File

@@ -5,20 +5,10 @@
placement:
mode: SnapgridCenter
components:
- type: Rotatable
rotateWhileAnchored: true
- type: Clickable
- type: InteractionOutline
- type: Physics
- type: Fixtures
fixtures:
- shape:
!type:PhysShapeAabb
bounds: "-0.49,-0.49,0.49,0.49"
hard: false
layer:
- Opaque
- Impassable
- MobImpassable
- VaultImpassable
- type: Transform
anchored: true
- type: Sprite
@@ -66,6 +56,8 @@
id: TwoWayLever
name: two way lever
description: A two way lever.
placement:
mode: SnapgridCenter
components:
- type: Clickable
- type: InteractionOutline