Example #1
0
        public override void OnFrame(Bot bot)
        {
            if (baseWorkers == null)
            {
                baseWorkers = new List <BaseWorkers>();
                foreach (Base b in bot.BaseManager.Bases)
                {
                    baseWorkers.Add(new BaseWorkers()
                    {
                        Base = b
                    });
                }
            }

            List <BaseWorkers> myBases = new List <BaseWorkers>();

            foreach (BaseWorkers workers in baseWorkers)
            {
                if (workers.Base.BaseLocation.MineralFields.Count == 0 || workers.Base.ResourceCenter == null || workers.Base.ResourceCenter.Unit.BuildProgress < 0.9)
                {
                    foreach (Agent agent in workers.MineralWorkers)
                    {
                        unassignedWorkers.Add(agent);
                    }
                    workers.MineralWorkers = new List <Agent>();
                }
                else if (!workers.Base.UnderAttack || StopTransfers)
                {
                    myBases.Add(workers);
                }
            }

            if (myBases.Count > 0 && EvacuateThreatenedBases)
            {
                foreach (BaseWorkers workers in baseWorkers)
                {
                    if (workers.Base.Evacuate)
                    {
                        foreach (Agent agent in workers.MineralWorkers)
                        {
                            unassignedWorkers.Add(agent);
                        }
                        workers.MineralWorkers = new List <Agent>();
                    }
                }
            }

            if (!StopTransfers && myBases.Count > 0)
            {
                TransferWorkers(myBases);
            }
            else
            {
                for (int i = unassignedWorkers.Count - 1; i >= 0; i--)
                {
                    BaseWorkers closest  = null;
                    float       distance = 1000 * 1000;
                    foreach (BaseWorkers workers in myBases)
                    {
                        float newDist = unassignedWorkers[i].DistanceSq(workers.Base.BaseLocation.Pos);
                        if (newDist < distance)
                        {
                            distance = newDist;
                            closest  = workers;
                        }
                    }
                    if (closest != null)
                    {
                        closest.Add(unassignedWorkers[i]);
                        unassignedWorkers.RemoveAt(i);
                    }
                }
            }

            foreach (BaseWorkers workers in myBases)
            {
                workers.OnFrame(bot);
            }
        }
Example #2
0
        private void TransferWorkers(List <BaseWorkers> myBases)
        {
            if (myBases.Count > 0)
            {
                while (unassignedWorkers.Count > 0)
                {
                    float       dist   = 10000000;
                    BaseWorkers picked = null;
                    foreach (BaseWorkers myBase in myBases)
                    {
                        float newDist = unassignedWorkers[unassignedWorkers.Count - 1].DistanceSq(myBase.Base.BaseLocation.Pos);
                        if (newDist >= dist)
                        {
                            continue;
                        }
                        dist   = newDist;
                        picked = myBase;
                    }
                    picked.Add(unassignedWorkers[unassignedWorkers.Count - 1]);
                    unassignedWorkers.RemoveAt(unassignedWorkers.Count - 1);
                }
            }

            int workersPerBase;

            if (myBases.Count == 0)
            {
                workersPerBase = 0;
            }
            else
            {
                bool done = false;
                int  totalMineralWorkers = units.Count;
                workersPerBase = totalMineralWorkers / myBases.Count;
                while (!done)
                {
                    int notFull   = 0;
                    int remaining = totalMineralWorkers;
                    foreach (BaseWorkers workers in myBases)
                    {
                        if (workersPerBase < workers.Base.BaseLocation.MineralFields.Count * 2)
                        {
                            notFull++;
                            remaining -= workersPerBase;
                        }
                        else
                        {
                            remaining -= workers.Base.BaseLocation.MineralFields.Count * 2;
                        }
                    }
                    if (notFull == 0)
                    {
                        done = true;
                    }
                    else if (remaining > 0)
                    {
                        workersPerBase += Math.Max(1, remaining / notFull);
                    }
                    else
                    {
                        done = true;
                    }
                }
            }

            bool saturated = true;

            foreach (BaseWorkers workers in myBases)
            {
                if (workers.Count < workers.Base.BaseLocation.MineralFields.Count * 2)
                {
                    saturated = false;
                    break;
                }
            }

            foreach (BaseWorkers workers in myBases)
            {
                while (workers.MineralWorkers.Count > 0 &&
                       (workers.Count > workersPerBase + 2 ||
                        (workers.Count > workers.Base.BaseLocation.MineralFields.Count * 2 && !saturated)))
                {
                    unassignedWorkers.Add(workers.MineralWorkers[workers.Count - 1]);
                    workers.MineralWorkers.RemoveAt(workers.Count - 1);
                }
            }

            foreach (BaseWorkers workers in myBases)
            {
                while (workers.Count < workersPerBase + 1 &&
                       unassignedWorkers.Count > 0 &&
                       workers.Count < workers.Base.BaseLocation.MineralFields.Count * 2)
                {
                    workers.Add(unassignedWorkers[unassignedWorkers.Count - 1]);
                    unassignedWorkers.RemoveAt(unassignedWorkers.Count - 1);
                }
            }

            /*
             * for (int i = 0; myBases.Count > 0 && unassignedWorkers.Count > 0; i++)
             * {
             *  myBases[i % myBases.Count].Add(unassignedWorkers[unassignedWorkers.Count - 1]);
             *  unassignedWorkers.RemoveAt(unassignedWorkers.Count - 1);
             * }
             */
        }