96 lines
3.0 KiB
C#
96 lines
3.0 KiB
C#
using System.Collections.Generic;
|
|
using BenchmarkDotNet.Attributes;
|
|
using Moq;
|
|
using Robust.Shared.Exceptions;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Log;
|
|
using Robust.Shared.Map;
|
|
using Robust.Shared.Reflection;
|
|
|
|
namespace Content.Benchmarks
|
|
{
|
|
public class EntityManagerGetAllComponents
|
|
{
|
|
private IEntityManager _entityManager;
|
|
|
|
[Params(5000)] public int N { get; set; }
|
|
|
|
public static void TestRun()
|
|
{
|
|
var x = new EntityManagerGetAllComponents
|
|
{
|
|
N = 500
|
|
};
|
|
x.Setup();
|
|
x.Run();
|
|
}
|
|
|
|
[GlobalSetup]
|
|
public void Setup()
|
|
{
|
|
// Initialize component manager.
|
|
IoCManager.InitThread();
|
|
|
|
IoCManager.Register<IEntityManager, EntityManager>();
|
|
IoCManager.Register<IRuntimeLog, RuntimeLog>();
|
|
IoCManager.Register<ILogManager, LogManager>();
|
|
IoCManager.Register<IDynamicTypeFactory, DynamicTypeFactory>();
|
|
IoCManager.Register<IEntitySystemManager, EntitySystemManager>();
|
|
IoCManager.RegisterInstance<IReflectionManager>(new Mock<IReflectionManager>().Object);
|
|
|
|
var dummyReg = new Mock<IComponentRegistration>();
|
|
dummyReg.SetupGet(p => p.Name).Returns("Dummy");
|
|
dummyReg.SetupGet(p => p.Type).Returns(typeof(DummyComponent));
|
|
dummyReg.SetupGet(p => p.NetID).Returns((ushort?) null);
|
|
dummyReg.SetupGet(p => p.References).Returns(new [] {typeof(DummyComponent)});
|
|
|
|
var componentFactory = new Mock<IComponentFactory>();
|
|
componentFactory.Setup(p => p.GetComponent<DummyComponent>()).Returns(new DummyComponent());
|
|
componentFactory.Setup(p => p.GetRegistration(It.IsAny<DummyComponent>())).Returns(dummyReg.Object);
|
|
componentFactory.Setup(p => p.GetAllRefTypes()).Returns(new[] {typeof(DummyComponent)});
|
|
|
|
IoCManager.RegisterInstance<IComponentFactory>(componentFactory.Object);
|
|
|
|
IoCManager.BuildGraph();
|
|
_entityManager = IoCManager.Resolve<IEntityManager>();
|
|
_entityManager.Initialize();
|
|
|
|
// Initialize N entities with one component.
|
|
for (var i = 0; i < N; i++)
|
|
{
|
|
var entity = _entityManager.SpawnEntity(null, EntityCoordinates.Invalid);
|
|
_entityManager.AddComponent<DummyComponent>(entity);
|
|
}
|
|
}
|
|
|
|
[Benchmark]
|
|
public int Run()
|
|
{
|
|
var count = 0;
|
|
|
|
foreach (var _ in _entityManager.EntityQuery<DummyComponent>(true))
|
|
{
|
|
count += 1;
|
|
}
|
|
|
|
return count;
|
|
}
|
|
|
|
[Benchmark]
|
|
public int Noop()
|
|
{
|
|
var count = 0;
|
|
|
|
_entityManager.TryGetComponent(default, out DummyComponent _);
|
|
|
|
return count;
|
|
}
|
|
|
|
private class DummyComponent : Component
|
|
{
|
|
public override string Name => "Dummy";
|
|
}
|
|
}
|
|
}
|