path: root/aoc2019/IntCodeVM.cs

namespace aoc2019;

public class IntCodeVM
{
    public enum HaltType
    {
        Terminate,
        Waiting
    }

    private readonly Queue<long> input;
    public readonly Queue<long> Output;

    private readonly long[] program;
    private long i;
    public long[] Memory;
    private long relativeBase;

    public IntCodeVM(string tape)
    {
        i = 0;
        relativeBase = 0;
        program = tape.Split(',').Select(long.Parse).ToArray();
        Memory = program;
        input = new();
        Output = new();
    }

    public long Result => Output.Dequeue();

    public void Reset()
    {
        i = 0;
        relativeBase = 0;
        Memory = program;
        input.Clear();
        Output.Clear();
    }

    public void AddInput(params long[] values)
    {
        foreach (var v in values) AddInput(v);
    }

    public void AddInput(long value)
    {
        input.Enqueue(value);
    }

    private long MemGet(long addr)
    {
        return addr < Memory.Length ? Memory[addr] : 0;
    }

    private void MemSet(long addr, long value)
    {
        if (addr < 0) addr = 0;
        if (addr >= Memory.Length)
            Array.Resize(ref Memory, (int)addr + 1);
        Memory[addr] = value;
    }

    private long Mode(long idx)
    {
        var mode = MemGet(i) / 100;
        for (var s = 1; s < idx; s++)
            mode /= 10;
        return mode % 10;
    }

    private long Get(long idx)
    {
        var param = MemGet(i + idx);
        return Mode(idx) switch
        {
            0 => MemGet(param),
            1 => param,
            2 => MemGet(relativeBase + param),
            _ => throw new("invalid parameter mode")
        };
    }

    private void Set(long idx, long val)
    {
        var param = MemGet(i + idx);
        switch (Mode(idx))
        {
            case 0:
                MemSet(param, val);
                break;
            case 1: throw new("cannot set in immediate mode");
            case 2:
                MemSet(relativeBase + param, val);
                break;
            default: throw new("invalid parameter mode");
        }
    }

    public HaltType Run(params long[] additionalInput)
    {
        foreach (var s in additionalInput) AddInput(s);
        return Run();
    }

    public HaltType Run()
    {
        while (i < Memory.Length)
        {
            var op = MemGet(i) % 100;
            switch (op)
            {
                case 1:
                    Set(3, Get(1) + Get(2));
                    i += 4;
                    break;
                case 2:
                    Set(3, Get(1) * Get(2));
                    i += 4;
                    break;
                case 3:
                    if (!input.Any())
                        return HaltType.Waiting;
                    Set(1, input.Dequeue());
                    i += 2;
                    break;
                case 4:
                    Output.Enqueue(Get(1));
                    i += 2;
                    break;
                case 5:
                    i = Get(1) == 0 ? i + 3 : Get(2);
                    break;
                case 6:
                    i = Get(1) != 0 ? i + 3 : Get(2);
                    break;
                case 7:
                    Set(3, Get(1) < Get(2) ? 1 : 0);
                    i += 4;
                    break;
                case 8:
                    Set(3, Get(1) == Get(2) ? 1 : 0);
                    i += 4;
                    break;
                case 9:
                    relativeBase += Get(1);
                    i += 2;
                    break;
                case 99:
                    return HaltType.Terminate;
                default:
                    throw new($"unknown op {op} at {i}");
            }
        }

        return HaltType.Terminate;
    }
}