Unity ECS EntityBuilder

This project is a wrapper around Unity ECS entities that allows one to simplify the process of creating / modifying entities.

Install

You can either just put the files into Assets/Plugins/ECSEntityBuilder or use it as a submodule:

git submodule add https://github.com/actionk/UnityECSEntityBuilder.git Assets/Plugins/ECSEntityBuilder

Usage

EntityWrapper

Wrapping an existing entity

Entity Wrapper allows you to wrap Entity with an object a modify it without the need of passing the Entity reference to the EntityManager over and over again.

var entityWrapper = new EntityWrapper(entity)
    .SetComponentData(new Translation { Value = float3.zero });
    .AddBuffer<TaskBufferElement>();
    .AddBuffer(
        new PathPositionBufferElement {position = new int2(0, 1)},
        new PathPositionBufferElement {position = new int2(0, 2)}
    );

You can also specify target entity manager / command buffer when creating a wrapper:

EntityWrapper.Wrap(entity, PostUpdateCommands)
    .AddBuffer<MyBuffer>()
    .SetComponentData(new Translation {Value = command.position})
    .SetComponentData(new Scale {Value = 1.0f});

Creating an entity

Apart of modifying, you can also create the entities using the EntityWrapper:

EntityWrapper.CreateEntity(EntityManager)
    .AddBuffer<MyBuffer>()
    .SetComponentData(new Translation {Value = command.position})
    .SetComponentData(new Scale {Value = 1.0f});

In this case, you pass the EntityManager or a command buffer into the CreateEntity method and the wrapper will do the rest for you.

EntityManagerWrapper

EntityManagerWrapper wraps around different ways of accessing EntityManager:

EntityManager

EntityManagerWrapper.Default

EntityCommandBuffer (PostUpdateCommands)

EntityManagerWrapper.FromCommandBuffer(PostUpdateCommands)

EntityCommandBuffer.Concurrent (Jobs)

EntityManagerWrapper.FromJobCommandBuffer(commandBuffer, threadId);

Usage example:

var entityWrapper = new EntityWrapper(entity);
entityWrapper.UsingWrapper(EntityManagerWrapper.FromCommandBuffer(PostUpdateCommands), wrapper =>
{
    wrapper.AddComponentData(new Rotation());
    wrapper.AddElementToBuffer(new PathPositionBufferElement {position = new int2(2, 0)});
});

Or using the builder:

var entityManagerWrapper = EntityManagerWrapper.FromJobCommandBuffer(commandBuffer, threadId);

UnitTaskBuilder
    .CreateCombined(itemPosition)
    .AddInnerTasks(
        UnitTaskBuilder.CreateInner(new PickUpItemTaskData {itemEntity = itemEntityToPickUp}).Build(entityManagerWrapper),
        UnitTaskBuilder.CreateInner(new PutItemToStorageTaskData {storageEntity = storageEntity}).Build(entityManagerWrapper)
    )
    .Build(entityManagerWrapper);

EntityBuilder

public class ItemBuilder : EntityBuilder<ItemBuilder>
{
    public static ItemBuilder Create(ItemData itemData)
    {
        return new ItemBuilder(itemData);
    }

    protected override ItemBuilder Self => this;

    internal ItemBuilder(ItemData itemData)
    {
        var contentItem = DependencyProvider.Resolve<ContentItemRepository>().GetByKey(itemData.itemId);

        CreateFromArchetype<ArchetypeItem>();
        this.SetSpriteRenderer(contentItem.icon);
        SetVariable(new ZIndexVariable(SpriteLayers.OBJECT));
        SetComponentData(itemData);
        SetComponentData(new Scale {Value = 0.5f});
    }
}

Usage:

var itemEntity = ItemBuilder
    .Create(contentObject.gatheredResource)
    .SetPosition(EntityManager.GetComponentData<Translation>(taskData.objectEntity).Value)
    .Build(PostUpdateCommands);

You can pass the EntityManagerWrapper into the Build() function.

Defining Archetypes

For defining an archetype you can just create a class implementing IArchetypeDescriptor:

public class ArchetypeItem : IArchetypeDescriptor
{
    public string Name => "Item";

    public ComponentType[] Components => new ComponentType[]
    {
        // common
        typeof(LocalToWorld),
        typeof(Translation),
        typeof(Rotation),
        typeof(Scale),

        // rendering
        typeof(MeshRendererData),

        // type-specific
        typeof(TagItem),
        typeof(ObjectData),
        typeof(ItemData)
    };
}

And then to create it inside the builder:

CreateFromArchetype<ArchetypeItem>();

Initialising Archetypes

As you can only create archetypes in main thread by using EntityManager (not possible via EntityCommandBuffer), you can initialise your archetypes beforehand:

EntityArchetypeManager.Instance.InitializeArchetypes(
    // tasks
    typeof(ArchetypeCombinedUnitTask),
    typeof(ArchetypeSingleUnitTask),
    typeof(ArchetypeInnerUnitTask),

    // items
    typeof(ArchetypeItem),

    // zones
    typeof(ArchetypeZone),
    typeof(ArchetypeZoneCell),

    // objects
    typeof(ArchetypeDecoration),
    typeof(ArchetypeBuildingGround),

    // generic
    typeof(ArchetypeGenericObjectWithLocation)
);

Alternatively, you can initialize archetypes by marking those classes with [Archetype] attribute:

[Archetype]
public class ArchetypeZone : IArchetypeDescriptor
{
    public string Name => "Zone";

    public ComponentType[] Components => new ComponentType[]
    {
        ... components ...
    };
}

Then you can initialize archetypes for all the marked classes in this assembly with:

EntityArchetypeManager.Instance.InitializeArchetypes(Assembly.GetCallingAssembly());

Initialising Archetypes in specific worlds

By default all the archetypes will be initialized in World.DefaultGameObjectInjectWorld If you want to change this, you can specify the world by passing World parameter in the Attribute:

[Archetype(WorldType.SERVER)]
public class ServerPlayerArchetype : IArchetypeDescriptor

Extending the EntityWrapper/EntityBuilder

You can easily add your own methods to EntityWrapper:

public static class EntityWrapper2DExtensions
{
    public static void SetPosition2D(this EntityWrapper wrapper, int2 position)
    {
        wrapper.SetComponentData(new Translation {Value = new float3(position.x + 0.5f, position.y + 0.5f, 0)});
    }
}

Or to EntityBuilder:

public static class EntityBuilder2DExtensions
{
    public static T SetPosition2D<T>(this EntityBuilder<T> wrapper, int2 position)
    {
        return wrapper.AddStep(new SetPosition2D(position));
    }
}

public class SetPosition2D : IEntityBuilderStep
{
    private readonly int2 position;

    public SetPosition2D(int2 position)
    {
        this.position = position;
    }

    public void Process(EntityManagerWrapper wrapper, EntityVariableMap variables, ref EntityBuilderData data)
    {
        wrapper.SetComponentData(data.entity, new Translation {Value = new float3(position.x + 0.5f, position.y + 0.5f, 0)});
    }
}

Additional functionality on Build()

To add your own Build implementation and access the EntityManager that was used for building, you can overload a OnBuild method:

public override void OnBuild(EntityManagerWrapper wrapper, Entity dataEntity)
{
    // your building code
}

Mapping GameObjects with the built entity

You can create a game object inside a builder and link it with the created entity without using ConvertToEntity component.

protected override void OnPreBuild(EntityManagerWrapper wrapper)
{
    base.OnPreBuild(wrapper);

    var prefab = ResourceProvider.GetPrefab("Items/Item");
    var gameObject = GameObjectEntityManager.Instance.CreateFromPrefab(prefab);

    AddComponentData(new ManagedGameObject {instanceId = gameObject.GetInstanceID()});
}

Variables

When extending the EntityWrapper you might want to save some data in the wrapper to reuse it in your extensions. For example, when setting the 2D position, you might want to save ZIndex to be used for this exact entity.

public class ZIndexVariable : IEntityVariable
{
    public int zIndex;

    public ZIndexVariable(int zIndex)
    {
        this.zIndex = zIndex;
    }
}

And then extend the EntityBuilder:

public class SetPosition2D : IEntityBuilderStep
{
    private const float ZIndexMultiplier = 0.01f;
    private readonly int2 position;

    public SetPosition2D(int2 position)
    {
        this.position = position;
    }

    private Translation GetTranslation(EntityVariableMap variables)
    {
        var zIndex = variables.Get<ZIndexVariable>()?.zIndex ?? 0;
        return new Translation {Value = new float3(position.x + 0.5f, position.y + 0.5f, -zIndex * ZIndexMultiplier)};
    }

    public void Process(EntityManagerWrapper wrapper, EntityVariableMap variables, ref EntityBuilderData data)
    {
        wrapper.SetComponentData(data.entity, GetTranslation(variables));
    }
}