freeso/TSOClient/tso.simantics/Primitives/VMFindBestObjectForFunction.cs

using System;
using System.Collections.Generic;
using FSO.Files.Utils;
using FSO.SimAntics.Model;
using FSO.LotView.Model;
using System.IO;
using FSO.SimAntics.Model.TSOPlatform;

namespace FSO.SimAntics.Engine.Primitives
{

    public class VMFindBestObjectForFunction : VMPrimitiveHandler
    {
        public static HashSet<uint> SurfaceFunctions = new HashSet<uint>() {
            14, 18, 20, 25
        };

        public static uint[] FunctionToEntryPoint = {
            18, //prepare food
            19, //cook food
            20, //flat surface
            21, //dispose
            22, //eat
            23, //pick up from slot
            24, //wash dish
            25, //eating surface
            26, //sit
            27, //stand
            14, //serving surface
            28, //clean
            16, //gardening
            17, //wash hands
            29 //repair
        };

        public static VMStackObjectVariable[] ScoreVar =
        {
            VMStackObjectVariable.PrepValue,
            VMStackObjectVariable.CookValue,
            VMStackObjectVariable.SurfaceValue,
            VMStackObjectVariable.DisposeValue,
            VMStackObjectVariable.Invalid, //hunger  value?
            VMStackObjectVariable.Invalid,
            VMStackObjectVariable.WashDishValue,
            VMStackObjectVariable.EatingSurfaceValue,
            VMStackObjectVariable.Invalid, //sit, may score using comfort value?
            VMStackObjectVariable.Invalid,
            VMStackObjectVariable.ServingSurfaceValue,
            VMStackObjectVariable.DirtyLevel,
            VMStackObjectVariable.GardeningValue,
            VMStackObjectVariable.WashHandsValue,
            VMStackObjectVariable.RepairState
        };

        public static Dictionary<VMStackObjectVariable, short> Thresholds = new Dictionary<VMStackObjectVariable, short>()
        {
            { VMStackObjectVariable.DirtyLevel, 800 },
            { VMStackObjectVariable.RepairState, 600 },
            { VMStackObjectVariable.GardeningValue, 15 }
        };

        public override VMPrimitiveExitCode Execute(VMStackFrame context, VMPrimitiveOperand args)
        {
            var operand = (VMFindBestObjectForFunctionOperand)args;

            var entities = context.VM.Entities;

            int bestScore = int.MinValue;
            VMEntity bestObj = null;
            var funcVar = ScoreVar[operand.Function];

            var entry = FunctionToEntryPoint[operand.Function];
            for (int i=0; i<entities.Count; i++) {
                var ent = entities[i];

                if (ent.GetValue(VMStackObjectVariable.LockoutCount) > 0
                    || (ent is VMGameObject && ((VMGameObject)ent).Disabled > 0)
                    || ent.Position == LotTilePos.OUT_OF_WORLD
                    || (!context.VM.TS1 && funcVar != VMStackObjectVariable.RepairState && ((ent.MultitileGroup.BaseObject.TSOState as VMTSOObjectState)?.Broken ?? false)))
                {
                    continue; //this object is not important!!!
                }

                if (ent.EntryPoints[entry].ActionFunction != 0) {
                    bool Execute;

                    int score = 0;
                    if (ScoreVar[operand.Function] != VMStackObjectVariable.Invalid)
                    {
                        score = ent.GetValue(funcVar);
                        short threshold;
                        if (context.VM.TS1 || funcVar != VMStackObjectVariable.RepairState)
                        {
                            if (score <= 0) continue; // lots of invalid functions with 0 score. just ignore them.
                            if (Thresholds.TryGetValue(funcVar, out threshold) && score < threshold) continue;
                        }
                        else if (ent is VMAvatar || !((VMTSOObjectState)ent.MultitileGroup.BaseObject.TSOState).Broken) continue;
                    }

                    if (ent.EntryPoints[entry].ConditionFunction != 0) {

                        var Behavior = ent.GetRoutineWithOwner(ent.EntryPoints[entry].ConditionFunction, context.VM.Context);
                        if (Behavior != null)
                        {
                            var test = VMThread.EvaluateCheck(context.VM.Context, context.Caller, new VMStackFrame()
                            {
                                Caller = context.Caller,
                                Callee = ent,
                                CodeOwner = Behavior.owner,
                                StackObject = ent,
                                Routine = Behavior.routine,
                                Args = new short[4]
                            });

                            Execute = (test == VMPrimitiveExitCode.RETURN_TRUE);
                        } else Execute = true;

                    } else if (SurfaceFunctions.Contains(entry)) {
                        //ts1: surface functions that have no check tree rely on the engine to check there is nothing in slot 0.
                        Execute = ent.GetSlot(0) == null;
                    } else {
                        Execute = true;
                    }

                    if (Execute)
                    {
                        if (ent.IsInUse(context.VM.Context, true)) continue; //this object is in use. this check is more expensive than check trees, so do it last.
                        //calculate the score for this object.

                        LotTilePos posDiff = ent.Position - context.Caller.Position;
                        score -= (int)Math.Sqrt(posDiff.x*posDiff.x+posDiff.y*posDiff.y+(posDiff.Level*posDiff.Level*900*256))/3;

                        if (score > bestScore)
                        {
                            bestScore = score;
                            bestObj = ent;
                        }
                    }
                }
            }

            if (bestObj != null)
            {
                context.StackObject = bestObj;
                return VMPrimitiveExitCode.GOTO_TRUE;
            } else return VMPrimitiveExitCode.GOTO_FALSE; //couldn't find an object! :'(
        }

    }

    public class VMFindBestObjectForFunctionOperand : VMPrimitiveOperand
    {
        public ushort Function { get; set; }

        #region VMPrimitiveOperand Members
        public void Read(byte[] bytes)
        {
            using (var io = IoBuffer.FromBytes(bytes, ByteOrder.LITTLE_ENDIAN))
            {
                Function = io.ReadUInt16();
            }
        }

        public void Write(byte[] bytes) {
            using (var io = new BinaryWriter(new MemoryStream(bytes)))
            {
                io.Write(Function);
            }
        }
        #endregion
    }
}