// === Functions =============================================================================================
// ===========================================================================================================
private void DefineMethod()
{
ulong methodName = (ulong)ReadConstant();
GearsValue methodPtr = Peek();
GearsObjFunction method = GetObjectFromPtr <GearsObjFunction>(methodPtr);
GearsObjClass objClass = GetObjectFromPtr <GearsObjClass>(Peek(1));
objClass.Methods.Set(methodName, methodPtr);
Pop();
}
private bool BindLoxMethod(GearsObjClass classObj, ulong name)
{
if (!classObj.Methods.TryGet(name, out GearsValue method))
{
return(false);
}
int objPtr = HeapAddObject(new GearsObjBoundMethod(Peek(), HeapGetObject(method.AsObjPtr) as GearsObjFunction));
Pop();
Push(GearsValue.CreateObjPtr(objPtr));
return(true);
}
private void CallInvokeSuper()
{
int argCount = ReadByte();
// next instruction will always be OP_GET_UPVALUE (for the super class). we include it here:
if (!(ReadByte() == (int)OP_GET_UPVALUE))
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "OP_SUPER_INVOKE must be followed by OP_GET_UPVALUE.");
}
int slot = ReadShort();
GearsObjUpvalue upvalue = _OpenFrame.Function.Upvalues[slot];
GearsObjClass superclass = GetObjectFromPtr <GearsObjClass>(upvalue.IsClosed ? upvalue.Value : StackGet(upvalue.OriginalSP));
ulong methodName = (ulong)ReadConstant();
InvokeFromClass(argCount, methodName, GearsValue.NilValue, superclass);
}
private void InvokeFromClass(int argCount, ulong methodName, GearsValue receiverPtr, GearsObjClass objClass)
{
if (!objClass.Methods.TryGet(methodName, out GearsValue methodPtr))
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), $"{objClass} has no method with name '{BitString.GetBitStr(methodName)}'.");
}
if ((!methodPtr.IsObjPtr) || !(HeapGetObject(methodPtr.AsObjPtr) is GearsObjFunction method))
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), $"Could not resolve method '{BitString.GetBitStr(methodName)}' in class {objClass}.");
}
if (method.Arity != argCount)
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), $"{method} expects {method.Arity} arguments but was passed {argCount}.");
}
int ip = method.IP;
int bp = _SP - (method.Arity + 1);
if (!receiverPtr.IsNil)
{
StackSet(bp, receiverPtr); // todo: this wipes out the method object. Is this bad?
}
PushFrame(new GearsCallFrame(method, ip, bp));
}
/// <summary>
/// Runs the script from top to bottom.
/// </summary>
private bool RunOne()
{
LastReturnValue = GearsValue.NilValue;
while (true)
{
EGearsOpCode instruction = (EGearsOpCode)ReadByte();
switch (instruction)
{
case OP_LOAD_CONSTANT:
Push(ReadConstant());
break;
case OP_LOAD_STRING:
Push(GearsValue.CreateObjPtr(HeapAddObject(new GearsObjString(ReadConstantString()))));
break;
case OP_LOAD_FUNCTION: {
int arity = ReadByte();
int address = ReadShort();
int upvalueCount = ReadByte();
GearsObjFunction closure = new GearsObjFunction(Chunk, arity, upvalueCount, address);
for (int i = 0; i < upvalueCount; i++)
{
bool isLocal = ReadByte() == 1;
int index = ReadByte();
if (isLocal)
{
int location = _OpenFrame.BP + index;
closure.Upvalues[i] = CaptureUpvalue(location);
}
else
{
closure.Upvalues[i] = _OpenFrame.Function.Upvalues[index];
}
}
Push(GearsValue.CreateObjPtr(HeapAddObject(closure)));
}
break;
case OP_NIL:
Push(GearsValue.NilValue);
break;
case OP_TRUE:
Push(GearsValue.TrueValue);
break;
case OP_FALSE:
Push(GearsValue.FalseValue);
break;
case OP_POP:
Pop();
break;
case OP_GET_LOCAL: {
int slot = ReadShort();
Push(StackGet(slot + _BP));
}
break;
case OP_SET_LOCAL: {
int slot = ReadShort();
StackSet(slot + _BP, Peek());
}
break;
case OP_GET_GLOBAL: {
ulong name = (ulong)ReadConstant();
if (!Globals.TryGet(name, out GearsValue value))
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), $"Undefined variable '{BitString.GetBitStr(name)}'.");
}
Push(value);
}
break;
case OP_DEFINE_GLOBAL: {
ulong name = (ulong)ReadConstant();
Globals.Set(name, Peek());
Pop();
}
break;
case OP_SET_GLOBAL: {
ulong name = (ulong)ReadConstant();
if (!Globals.ContainsKey(name))
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), $"Undefined variable '{BitString.GetBitStr(name)}'.");
}
Globals.Set(name, Peek());
break;
}
case OP_GET_UPVALUE: {
int slot = ReadShort();
GearsObjUpvalue upvalue = _OpenFrame.Function.Upvalues[slot];
if (upvalue.IsClosed)
{
Push(upvalue.Value);
}
else
{
Push(StackGet(upvalue.OriginalSP));
}
}
break;
case OP_SET_UPVALUE: {
int slot = ReadShort();
GearsObjUpvalue upvalue = _OpenFrame.Function.Upvalues[slot];
if (upvalue.IsClosed)
{
upvalue.Value = Peek();
}
else
{
StackSet(upvalue.OriginalSP, Peek());
}
}
break;
case OP_GET_PROPERTY: {
GearsObjInstance instance = GetObjectFromPtr <GearsObjInstance>(Peek());
ulong name = (ulong)ReadConstant(); // property name
if (instance.TryGetField(name, out GearsValue value))
{
Pop(); // instance
Push(value); // property value
break;
}
if (instance is GearsObjInstanceLox loxInstance && BindLoxMethod(loxInstance.Class, name))
{
break;
}
throw new GearsRuntimeException(Chunk.LineAt(_IP), $"Undefined property or method '{name}'.");
}
case OP_SET_PROPERTY: {
GearsObjInstance instance = GetObjectFromPtr <GearsObjInstance>(Peek(1));
ulong name = (ulong)ReadConstant(); // property name
GearsValue value = Pop(); // value
instance.SetField(name, value);
Pop(); // ptr
Push(value); // value
}
break;
case OP_GET_SUPER: {
ulong name = (ulong)ReadConstant(); // method/property name
GearsObjClass superclass = GetObjectFromPtr <GearsObjClass>(Pop());
if (!BindLoxMethod(superclass, name))
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), $"Could not get {name} in superclass {superclass}");
}
}
break;
case OP_EQUAL:
Push(AreValuesEqual(Pop(), Pop()));
break;
case OP_GREATER: {
if (!Peek(0).IsNumber || !Peek(1).IsNumber)
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "Operands must be numbers.");
}
GearsValue b = Pop();
GearsValue a = Pop();
Push(a > b);
}
break;
case OP_LESS: {
if (!Peek(0).IsNumber || !Peek(1).IsNumber)
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "Operands must be numbers.");
}
GearsValue b = Pop();
GearsValue a = Pop();
Push(a < b);
}
break;
case OP_BITWISE_AND: {
if (!Peek(0).IsNumber || !Peek(1).IsNumber)
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "Operands of bitwise operators must be numbers.");
}
GearsValue b = Pop();
GearsValue a = Pop();
Push((long)((double)a) & (long)((double)b));
}
break;
case OP_BITWISE_OR: {
if (!Peek(0).IsNumber || !Peek(1).IsNumber)
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "Operands of bitwise operators must be numbers.");
}
GearsValue b = Pop();
GearsValue a = Pop();
Push((long)((double)a) | (long)((double)b));
}
break;
case OP_ADD: {
GearsValue b = Pop();
GearsValue a = Pop();
if (a.IsNumber && b.IsNumber)
{
Push(a + b);
}
else if (a.IsObjType <GearsObjString>(this) && b.IsObjType <GearsObjString>(this))
{
string sa = GetObjectFromPtr <GearsObjString>(a).Value;
string sb = GetObjectFromPtr <GearsObjString>(b).Value;
Push(GearsValue.CreateObjPtr(HeapAddObject(new GearsObjString(sa + sb))));
}
else
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "Operands of add must be numbers or strings.");
}
}
break;
case OP_SUBTRACT: {
if (!Peek(0).IsNumber || !Peek(1).IsNumber)
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "Operands of subtract must be numbers.");
}
GearsValue b = Pop();
GearsValue a = Pop();
Push(a - b);
}
break;
case OP_MULTIPLY: {
if (!Peek(0).IsNumber || !Peek(1).IsNumber)
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "Operands of multiply must be numbers.");
}
GearsValue b = Pop();
GearsValue a = Pop();
Push(a * b);
}
break;
case OP_DIVIDE: {
if (!Peek(0).IsNumber || !Peek(1).IsNumber)
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "Operands of divide must be numbers.");
}
GearsValue b = Pop();
GearsValue a = Pop();
Push(a / b);
}
break;
case OP_NOT:
Push(IsFalsey(Pop()));
break;
case OP_NEGATE: {
if (!Peek(0).IsNumber)
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "Operand of negate must be a number.");
}
Push(-Pop());
}
break;
case OP_JUMP: {
int offset = ReadShort();
ModIP(offset);
}
break;
case OP_JUMP_IF_FALSE: {
int offset = ReadShort();
if (IsFalsey(Peek()))
{
ModIP(offset);
}
}
break;
case OP_LOOP: {
int offset = ReadShort();
ModIP(-offset);
}
break;
case OP_CALL: {
Call(argCount: ReadByte());
}
break;
case OP_INVOKE: {
CallInvoke();
}
break;
case OP_SUPER_INVOKE: {
CallInvokeSuper();
}
break;
case OP_CLOSE_UPVALUE:
CloseUpvalues(_SP - 1);
Pop();
break;
case OP_RETURN: {
GearsValue result = Pop();
CloseUpvalues(_OpenFrame.BP);
if (PopFrame())
{
if (_SP != 0)
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), $"Error after final return: SP is '{_SP}', not '0'.");
}
LastReturnValue = result;
_IP = Chunk.SizeCode; // code is complete and no longer running.
return(true);
}
Push(result);
}
break;
case OP_CLASS: {
Push(GearsValue.CreateObjPtr(HeapAddObject(new GearsObjClass(ReadConstantString()))));
}
break;
case OP_INHERIT: {
if (!Peek(0).IsObjType <GearsObjClass>(this))
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "Superclass is not a class.");
}
GearsObjClass super = GetObjectFromPtr <GearsObjClass>(Peek(1));
GearsObjClass sub = GetObjectFromPtr <GearsObjClass>(Peek(0));
foreach (ulong key in super.Methods.AllKeys)
{
if (!super.Methods.TryGet(key, out GearsValue methodPtr))
{
throw new GearsRuntimeException(Chunk.LineAt(_IP), "Could not copy superclass method table.");
}
sub.Methods.Set(key, methodPtr);
}
Pop(); // pop subclass
}
break;
case OP_METHOD: {
DefineMethod();
}
break;
default:
throw new GearsRuntimeException(Chunk.LineAt(_IP), $"Unknown opcode {instruction}");
}
}
}