# 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. ```cs var entityWrapper = new EntityWrapper(entity) .SetComponentData(new Translation { Value = float3.zero }); .AddBuffer(); .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: ```cs EntityWrapper.Wrap(entity, PostUpdateCommands) .AddBuffer() .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: ```cs EntityWrapper.CreateEntity(EntityManager) .AddBuffer() .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: 1. EntityManager ```cs EntityManagerWrapper.Default ``` 2. EntityCommandBuffer (PostUpdateCommands) ```cs EntityManagerWrapper.FromCommandBuffer(PostUpdateCommands) ``` 3. EntityCommandBuffer.Concurrent (Jobs) ```cs EntityManagerWrapper.FromJobCommandBuffer(commandBuffer, threadId); ``` Usage example: ```cs 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: ```cs 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 ```cs public class ItemBuilder : EntityBuilder { public static ItemBuilder Create(ItemData itemData) { return new ItemBuilder(itemData); } protected override ItemBuilder Self => this; internal ItemBuilder(ItemData itemData) { var contentItem = DependencyProvider.Resolve().GetByKey(itemData.itemId); CreateFromArchetype(); this.SetSpriteRenderer(contentItem.icon); SetVariable(new ZIndexVariable(SpriteLayers.OBJECT)); SetComponentData(itemData); SetComponentData(new Scale {Value = 0.5f}); } } ``` Usage: ```cs var itemEntity = ItemBuilder .Create(contentObject.gatheredResource) .SetPosition(EntityManager.GetComponentData(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`: ```cs 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: ```cs CreateFromArchetype(); ``` ### Initialising Archetypes As you can only create archetypes in main thread by using EntityManager (not possible via EntityCommandBuffer), you can initialise your archetypes beforehand: ```cs 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: ```cs [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: ```cs 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: ```cs [Archetype(WorldType.SERVER)] public class ServerPlayerArchetype : IArchetypeDescriptor ``` ### Extending the EntityWrapper/EntityBuilder You can easily add your own methods to EntityWrapper: ```cs 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: ```cs public static class EntityBuilder2DExtensions { public static T SetPosition2D(this EntityBuilder 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: ```cs 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. ```cs 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. ```cs public class ZIndexVariable : IEntityVariable { public int zIndex; public ZIndexVariable(int zIndex) { this.zIndex = zIndex; } } ``` And then extend the EntityBuilder: ```cs 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()?.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)); } } ``` Usage: ```cs ObjectBuilder.Create(6) .SetVariable(new ZIndexVariable(10)); .SetPosition2D(new int2 {x = -7, y = 0}) .Build(); ```