static void defineMethod(WrenVM* vm, const char* className, const char* methodName, int numParams, WrenForeignMethodFn methodFn, bool isStatic) { ASSERT(className != NULL, "Must provide class name."); int length = (int)strlen(methodName); ASSERT(methodName != NULL, "Must provide method name."); ASSERT(strlen(methodName) < MAX_METHOD_NAME, "Method name too long."); ASSERT(numParams >= 0, "numParams cannot be negative."); ASSERT(numParams <= MAX_PARAMETERS, "Too many parameters."); ASSERT(methodFn != NULL, "Must provide method function."); // Find or create the class to bind the method to. int classSymbol = wrenSymbolTableFind(&vm->globalNames, className, strlen(className)); ObjClass* classObj; if (classSymbol != -1) { // TODO: Handle name is not class. classObj = AS_CLASS(vm->globals.data[classSymbol]); } else { // The class doesn't already exist, so create it. size_t length = strlen(className); ObjString* nameString = AS_STRING(wrenNewString(vm, className, length)); WREN_PIN(vm, nameString); // TODO: Allow passing in name for superclass? classObj = wrenNewClass(vm, vm->objectClass, 0, nameString); wrenDefineGlobal(vm, className, length, OBJ_VAL(classObj)); WREN_UNPIN(vm); } // Create a name for the method, including its arity. char name[MAX_METHOD_SIGNATURE]; strncpy(name, methodName, length); for (int i = 0; i < numParams; i++) { name[length++] = ' '; } name[length] = '\0'; // Bind the method. int methodSymbol = wrenSymbolTableEnsure(vm, &vm->methodNames, name, length); Method method; method.type = METHOD_FOREIGN; method.fn.foreign = methodFn; if (isStatic) classObj = classObj->obj.classObj; wrenBindMethod(vm, classObj, methodSymbol, method); }
int wrenSymbolTableAdd(WrenVM* vm, SymbolTable* symbols, const char* name, size_t length) { // If already present, return an error. if (wrenSymbolTableFind(symbols, name, length) != -1) return -1; return addSymbol(vm, symbols, name, length); }
int wrenSymbolTableEnsure(WrenVM* vm, SymbolTable* symbols, const char* name, size_t length) { // See if the symbol is already defined. int existing = wrenSymbolTableFind(symbols, name, length); if (existing != -1) return existing; // New symbol, so add it. return addSymbol(vm, symbols, name, length); }
// Returns the global variable named [name]. static Value findGlobal(WrenVM* vm, const char* name) { int symbol = wrenSymbolTableFind(&vm->globalNames, name, strlen(name)); return vm->globals.data[symbol]; }