RobustToolbox/Robust.Client/GameObjects/ClientEntityManager.Spawn.cs at 6011c3263bcbaf13939d3c668ee0899b5edeb281 · Space-Wizards-Federation/RobustToolbox · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
using System.Diagnostics.CodeAnalysis;
using System.Numerics;
using System.Runtime.CompilerServices;
using Robust.Shared.Containers;
using Robust.Shared.GameObjects;
using Robust.Shared.Map;
using Robust.Shared.Maths;
using Robust.Shared.Prototypes;
using Robust.Shared.Utility;

namespace Robust.Client.GameObjects;

public sealed partial class ClientEntityManager
{
    public override EntityUid PredictedSpawnAttachedTo(string? protoName, EntityCoordinates coordinates, ComponentRegistry? overrides = null, Angle rotation = default)
    {
        var ent = SpawnAttachedTo(protoName, coordinates, overrides, rotation);
        FlagPredicted(ent);
        return ent;
    }

    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    public override EntityUid PredictedSpawn(string? protoName = null, ComponentRegistry? overrides = null, bool doMapInit = true)
    {
        var ent = Spawn(protoName, overrides, doMapInit);
        FlagPredicted(ent);
        return ent;
    }

    public override EntityUid PredictedSpawn(string? protoName, MapCoordinates coordinates, ComponentRegistry? overrides = null, Angle rotation = default!)
    {
        var ent = Spawn(protoName, coordinates, overrides, rotation);
        FlagPredicted(ent);
        return ent;
    }

    public override EntityUid PredictedSpawnAtPosition(string? protoName, EntityCoordinates coordinates, ComponentRegistry? overrides = null)
    {
        var ent = SpawnAtPosition(protoName, coordinates, overrides);
        FlagPredicted(ent);
        return ent;
    }

    public override bool PredictedTrySpawnNextTo(
        string? protoName,
        EntityUid target,
        [NotNullWhen(true)] out EntityUid? uid,
        TransformComponent? xform = null,
        ComponentRegistry? overrides = null)
    {
        if (!TrySpawnNextTo(protoName, target, out uid, xform, overrides))
            return false;

        FlagPredicted(uid.Value);
        return true;
    }

    public override bool PredictedTrySpawnInContainer(
        string? protoName,
        EntityUid containerUid,
        string containerId,
        [NotNullWhen(true)] out EntityUid? uid,
        ContainerManagerComponent? containerComp = null,
        ComponentRegistry? overrides = null)
    {
        if (!TrySpawnInContainer(protoName, containerUid, containerId, out uid, containerComp, overrides))
            return false;

        FlagPredicted(uid.Value);
        return true;
    }

    public override EntityUid PredictedSpawnNextToOrDrop(EntProtoId? protoName, Entity<TransformComponent?> target, Vector2 offset = default, ComponentRegistry? overrides = null)
    {
        var ent = SpawnNextToOrDrop(protoName, target, offset, overrides);
        FlagPredicted(ent);
        return ent;
    }

    public override EntityUid PredictedSpawnInContainerOrDrop(
        string? protoName,
        EntityUid containerUid,
        string containerId,
        TransformComponent? xform = null,
        ContainerManagerComponent? containerComp = null,
        ComponentRegistry? overrides = null)
    {
        var ent = SpawnInContainerOrDrop(protoName, containerUid, containerId, xform, containerComp, overrides);
        FlagPredicted(ent);
        return ent;
    }

    public override EntityUid PredictedSpawnInContainerOrDrop(
        string? protoName,
        EntityUid containerUid,
        string containerId,
        out bool inserted,
        TransformComponent? xform = null,
        ContainerManagerComponent? containerComp = null,
        ComponentRegistry? overrides = null)
    {
        var ent = SpawnInContainerOrDrop(protoName,
            containerUid,
            containerId,
            out inserted,
            xform,
            containerComp,
            overrides);

        FlagPredicted(ent);
        return ent;
    }

    public override void FlagPredicted(Entity<MetaDataComponent?> ent)
    {
        if (!MetaQuery.Resolve(ent.Owner, ref ent.Comp))
            return;

        DebugTools.Assert(IsClientSide(ent.Owner, ent.Comp));
        EnsureComponent<PredictedSpawnComponent>(ent.Owner);

        // TODO: Need to map call site or something, needs to be consistent between client and server.
    }
}