riversong-code-showcase/Source/Riversong/Game/World/FertilityMap/UpdateFertilityMapSystem.cs

using System;
using Cysharp.Threading.Tasks;
using Unity.Burst;
using Unity.Collections;
using Unity.Jobs;
using Unity.Mathematics;
using UnityEngine;

namespace DanieleMarotta.RiversongCodeShowcase
{
    [RequiresWorldReadyForUpdate]
    [GameSystemGroup(typeof(EarlyGameSystemGroup))]
    public class UpdateFertilityMapSystem : GameSystem, IInitializable, IDisposable, IUpdatable
    {
        [InjectService]
        private GameConfig _config;

        [InjectService]
        private World _world;

        [InjectService]
        private ISignalBus _signalBus;

        public UpdateFertilityMapSystem(IServiceLocator serviceLocator) : base(serviceLocator)
        {
        }

        public UniTask InitializeAsync()
        {
            _signalBus.Subscribe<BuildingPlacementAnimationStartedSignal>(OnBuildingPlacementAnimationStarted);
            _signalBus.Subscribe<RoadTileUpdatedSignal>(OnRoadTileUpdated);

            return UniTask.CompletedTask;
        }

        public void Dispose()
        {
            _signalBus.Unsubscribe<BuildingPlacementAnimationStartedSignal>(OnBuildingPlacementAnimationStarted);
            _signalBus.Unsubscribe<RoadTileUpdatedSignal>(OnRoadTileUpdated);
        }

        public void Update()
        {
            var blockMap = _world.BlockMap.GetNativeArray();
            var fertility = _world.Fertility.GetNativeArray();
            var delta = _config.Terrain.FertilityRegenarationRate * Time.deltaTime;

            new UpdateFertilityMapJob
                {
                    BlockMap = blockMap,
                    Fertility = fertility,
                    Delta = delta
                }.Schedule(fertility.Length, _world.Size.x)
                .Complete();
        }

        private void OnBuildingPlacementAnimationStarted(BuildingPlacementAnimationStartedSignal signal)
        {
            foreach (var tile in TileRange.From(signal.Rect)) _world.Fertility.GetValueRW(tile).CurrentFertility = 0;
        }

        private void OnRoadTileUpdated(RoadTileUpdatedSignal signal)
        {
            if (!signal.RoadTileAdded) return;

            _world.Fertility.GetValueRW(signal.Tile).CurrentFertility = 0;
        }

        [BurstCompile]
        private struct UpdateFertilityMapJob : IJobParallelFor
        {
            [ReadOnly]
            public NativeArray<BlockReason> BlockMap;

            public NativeArray<FertilityMapValue> Fertility;

            public float Delta;

            [BurstCompile]
            public void Execute(int index)
            {
                var value = Fertility[index];

                if ((BlockMap[index] & BlockReason.ClearGrassMask) != 0 || value.CurrentFertility >= value.MaxFertility) return;

                value.CurrentFertility = math.min(value.CurrentFertility + Delta, value.MaxFertility);

                Fertility[index] = value;
            }
        }
    }
}