using Assets.Domain.Agents;
using Assets.Message.Decoding;
using System.Collections.Generic;
using System.Linq;
using Unity.Collections;
using Unity.Jobs;
using UnityEngine;

namespace Assets.GAMA.net.Message.Jobs
{
    public class GamaMessageDecodingJobHandler : BaseMessageDecodingJobHandler
    {
        private const int MaxAgentsPerMessage = 1000;
        private GamaMessageDecodingJob Job;
        private JobHandle Handle;

        private NativeArray<byte> _messageBytes;
        public NativeArray<float> _headings;
        public NativeArray<float> _speeds;
        public NativeArray<bool> _hasToBeRemoved;
        public NativeArray<Vector3> _positions;
        public NativeArray<FixedByteString> _names;
        public NativeArray<FixedByteString> _species;
        private NativeArray<int> _agentCount;
        private NativeArray<long> _experimentCycle;
        private NativeArray<long> _timestampTicks;
        public override string Topic { get; }

        public GamaMessageDecodingJobHandler(byte[] message, string topic)
        {
            Topic = topic;
            _messageBytes = new NativeArray<byte>(message, Allocator.Persistent);
            _agentCount = new NativeArray<int>(1, Allocator.Persistent);
            _experimentCycle = new NativeArray<long>(1, Allocator.Persistent);
            _timestampTicks = new NativeArray<long>(1, Allocator.Persistent);
            _headings = new NativeArray<float>(MaxAgentsPerMessage, Allocator.Persistent);
            _speeds = new NativeArray<float>(MaxAgentsPerMessage, Allocator.Persistent);
            _positions = new NativeArray<Vector3>(MaxAgentsPerMessage, Allocator.Persistent);
            _names = new NativeArray<FixedByteString>(MaxAgentsPerMessage, Allocator.Persistent);
            _species = new NativeArray<FixedByteString>(MaxAgentsPerMessage, Allocator.Persistent);
            _hasToBeRemoved = new NativeArray<bool>(MaxAgentsPerMessage, Allocator.Persistent);
        }

        public override void Start()
        {
            Job = new GamaMessageDecodingJob()
            {
                MessageBytes = _messageBytes,
                AgentCount = _agentCount,
                ExperimentCycle = _experimentCycle,
                Names = _names,
                Headings = _headings,
                Speeds = _speeds,
                Positions = _positions,
                HasToBeRemoved = _hasToBeRemoved,
                TimestampTicks = _timestampTicks,
                Species = _species
            };
            Handle = Job.Schedule();
        }

        protected override DecodedPublishedMessage DoHarvest()
        {
            Handle.Complete();
            
            List<SimulationAgent> agents = new List<SimulationAgent>();
            long experimentCycle = _experimentCycle.First();
            long timestampTicks = _timestampTicks.First();
            for (int i = 0; i < _agentCount[0]; i++)
            {
                agents.Add(new PeopleSimulationAgent()
                {
                    Position = _positions[i],
                    Species = _species[i].GetString(),
                    Heading = _headings[i],
                    Name = _names[i].GetString(),
                    Speed = _speeds[i],
                    HasToBeRemoved = _hasToBeRemoved[i]
                });
            }

            _headings.Dispose();
            _names.Dispose();
            _species.Dispose();
            _speeds.Dispose();
            _positions.Dispose();
            _messageBytes.Dispose();
            _agentCount.Dispose();
            _experimentCycle.Dispose();
            _hasToBeRemoved.Dispose();
            _timestampTicks.Dispose();

            return new DecodedPublishedMessage(experimentCycle, timestampTicks, agents);
        }

        public override bool IsCompleted()
        {
            return Handle.IsCompleted;
        }
    }
}