2020-10-28 05:16:30 +00:00
# include "cvm.h"
# include "cstate.h"
# include "cdebug.h"
# include "cmem.h"
# include <stdarg.h>
# include <string.h>
2020-11-13 18:54:06 +00:00
void cosmoV_error ( CState * state , const char * format , . . . ) {
2020-10-28 05:16:30 +00:00
if ( state - > panic )
return ;
// print stack trace
for ( int i = 0 ; i < state - > frameCount ; i + + ) {
CCallFrame * frame = & state - > callFrame [ i ] ;
CObjFunction * function = frame - > closure - > function ;
CChunk * chunk = & function - > chunk ;
int line = chunk - > lineInfo [ frame - > pc - chunk - > buf - 1 ] ;
if ( i = = state - > frameCount - 1 ) { // it's the last call frame, prepare for the objection to be printed
2020-12-09 18:23:16 +00:00
fprintf ( stderr , " Objection in %.*s on [line %d] in " , function - > module - > length , function - > module - > str , line ) ;
2020-10-28 05:16:30 +00:00
if ( function - > name = = NULL ) { // unnamed chunk
fprintf ( stderr , " %s \n \t " , UNNAMEDCHUNK ) ;
} else {
fprintf ( stderr , " %.*s() \n \t " , function - > name - > length , function - > name - > str ) ;
}
} else {
fprintf ( stderr , " [line %d] in " , line ) ;
if ( function - > name = = NULL ) { // unnamed chunk
fprintf ( stderr , " %s \n " , UNNAMEDCHUNK ) ;
} else {
fprintf ( stderr , " %.*s() \n " , function - > name - > length , function - > name - > str ) ;
}
}
}
va_list args ;
va_start ( args , format ) ;
vfprintf ( stderr , format , args ) ;
va_end ( args ) ;
fputs ( " \n " , stderr ) ;
// TODO: push error onto the stack :P
state - > panic = true ;
}
CObjUpval * captureUpvalue ( CState * state , CValue * local ) {
CObjUpval * prev = NULL ;
CObjUpval * upvalue = state - > openUpvalues ;
while ( upvalue ! = NULL & & upvalue - > val > local ) { // while upvalue exists and is higher on the stack than local
prev = upvalue ;
upvalue = upvalue - > next ;
}
if ( upvalue ! = NULL & & upvalue - > val = = local ) { // we found the local we were going to capture
return upvalue ;
}
CObjUpval * newUpval = cosmoO_newUpvalue ( state , local ) ;
newUpval - > next = upvalue ;
// the list is sorted, so insert it at our found upvalue
if ( prev = = NULL ) {
state - > openUpvalues = newUpval ;
} else {
prev - > next = newUpval ;
}
return newUpval ;
}
void closeUpvalues ( CState * state , CValue * local ) {
while ( state - > openUpvalues ! = NULL & & state - > openUpvalues - > val > = local ) { // for every upvalue that points to the local or anything above it
CObjUpval * upvalue = state - > openUpvalues ;
upvalue - > closed = * upvalue - > val ;
upvalue - > val = & upvalue - > closed ; // upvalue now points to itself :P
state - > openUpvalues = upvalue - > next ;
}
}
void pushCallFrame ( CState * state , CObjClosure * closure , int args ) {
CCallFrame * frame = & state - > callFrame [ state - > frameCount + + ] ;
frame - > base = state - > top - args - 1 ; // - 1 for the function
frame - > pc = closure - > function - > chunk . buf ;
frame - > closure = closure ;
}
2020-11-13 23:39:47 +00:00
// offset is the offset of the callframe base we set the state->top back too (useful for passing values in the stack as arguments, like methods)
void popCallFrame ( CState * state , int offset ) {
2020-10-28 05:16:30 +00:00
closeUpvalues ( state , state - > callFrame [ state - > frameCount - 1 ] . base ) ; // close any upvalue still open
2020-11-13 23:39:47 +00:00
state - > top = state - > callFrame [ state - > frameCount - 1 ] . base + offset ; // resets the stack
2020-10-28 05:16:30 +00:00
state - > frameCount - - ;
}
CObjString * cosmoV_concat ( CState * state , CObjString * strA , CObjString * strB ) {
size_t sz = strA - > length + strB - > length ;
char * buf = cosmoM_xmalloc ( state , sz + 1 ) ; // +1 for null terminator
memcpy ( buf , strA - > str , strA - > length ) ;
memcpy ( buf + strA - > length , strB - > str , strB - > length ) ;
buf [ sz ] = ' \0 ' ;
return cosmoO_takeString ( state , buf , sz ) ;
}
2020-11-10 01:44:12 +00:00
bool cosmoV_execute ( CState * state ) ;
2020-10-28 05:16:30 +00:00
2020-11-13 23:39:47 +00:00
static inline void callCFunction ( CState * state , CosmoCFunction cfunc , int args , int offset ) {
StkPtr savedBase = cosmoV_getTop ( state , args ) ;
cosmoM_freezeGC ( state ) ; // we don't want a GC event during c api because we don't actually trust the user to know how to evade the GC
CValue res = cfunc ( state , args , savedBase + 1 ) ;
cosmoM_unfreezeGC ( state ) ;
state - > top = savedBase + offset ;
cosmoV_pushValue ( state , res ) ;
}
bool call ( CState * state , CObjClosure * closure , int args , int offset ) {
2020-11-10 01:44:12 +00:00
// missmatched args, thats an obvious user error, so error.
if ( args ! = closure - > function - > args ) {
2020-11-13 18:54:06 +00:00
cosmoV_error ( state , " Expected %d parameters for %s, got %d! " , closure - > function - > args , closure - > function - > name = = NULL ? UNNAMEDCHUNK : closure - > function - > name - > str , args ) ;
2020-11-10 01:44:12 +00:00
return false ;
}
// load function into callframe
pushCallFrame ( state , closure , closure - > function - > args ) ;
// execute
if ( ! cosmoV_execute ( state ) )
return false ;
// remember where the return value is
CValue * result = cosmoV_getTop ( state , 0 ) ;
// pop the callframe and return result :)
2020-11-13 23:39:47 +00:00
popCallFrame ( state , offset ) ;
2020-11-10 01:44:12 +00:00
// push the return value back onto the stack
cosmoV_pushValue ( state , * result ) ;
return true ;
2020-10-28 05:16:30 +00:00
}
2020-11-17 21:07:56 +00:00
bool invokeMethod ( CState * state , CObjObject * obj , CValue func , int args ) {
// first, set the first argument to the object
StkPtr temp = cosmoV_getTop ( state , args ) ;
* temp = cosmoV_newObj ( obj ) ;
if ( IS_CFUNCTION ( func ) ) {
callCFunction ( state , cosmoV_readCFunction ( func ) , args + 1 , 1 ) ;
} else if ( IS_CLOSURE ( func ) ) {
call ( state , cosmoV_readClosure ( func ) , args + 1 , 1 ) ; // offset = 1 so our stack is properly reset
2020-11-13 05:04:09 +00:00
} else {
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Cannot invoke non-function type %s! " , cosmoV_typeStr ( func ) ) ;
2020-11-13 05:04:09 +00:00
}
return true ;
}
2020-11-10 01:44:12 +00:00
// args = # of pass parameters
COSMOVMRESULT cosmoV_call ( CState * state , int args ) {
2020-10-28 05:16:30 +00:00
StkPtr val = cosmoV_getTop ( state , args ) ; // function will always be right above the args
2020-12-04 06:04:14 +00:00
if ( GET_TYPE ( * val ) ! = COSMO_TOBJ ) {
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Cannot call non-function type %s! " , cosmoV_typeStr ( * val ) ) ;
2020-10-28 05:16:30 +00:00
return COSMOVM_RUNTIME_ERR ;
}
2020-12-04 06:04:14 +00:00
switch ( cosmoV_readObj ( * val ) - > type ) {
2020-10-28 05:16:30 +00:00
case COBJ_CLOSURE : {
2020-12-04 06:04:14 +00:00
CObjClosure * closure = ( CObjClosure * ) cosmoV_readObj ( * val ) ;
2020-11-13 23:39:47 +00:00
if ( ! call ( state , closure , args , 0 ) ) {
2020-11-10 01:44:12 +00:00
return COSMOVM_RUNTIME_ERR ;
}
break ;
}
case COBJ_METHOD : {
2020-12-04 06:04:14 +00:00
CObjMethod * method = ( CObjMethod * ) cosmoV_readObj ( * val ) ;
2020-11-17 21:07:56 +00:00
invokeMethod ( state , method - > obj , method - > func , args ) ;
2020-11-10 01:44:12 +00:00
break ;
}
2020-12-05 23:55:09 +00:00
case COBJ_OBJECT : { // object is being instantiated, making another object
2020-12-04 06:04:14 +00:00
CObjObject * protoObj = ( CObjObject * ) cosmoV_readObj ( * val ) ;
2020-11-10 01:44:12 +00:00
CObjObject * newObj = cosmoO_newObject ( state ) ;
2020-11-15 18:22:11 +00:00
newObj - > proto = protoObj ;
2020-11-10 01:44:12 +00:00
CValue ret ;
// check if they defined an initalizer
2020-11-17 21:07:56 +00:00
if ( cosmoO_getIString ( state , protoObj , ISTRING_INIT , & ret ) ) {
invokeMethod ( state , newObj , ret , args ) ;
2020-11-10 01:44:12 +00:00
} else {
// no default initalizer
if ( args ! = 0 ) {
2020-11-13 18:54:06 +00:00
cosmoV_error ( state , " Expected 0 parameters, got %d! " , args ) ;
2020-11-10 01:44:12 +00:00
return COSMOVM_RUNTIME_ERR ;
}
2020-10-28 05:16:30 +00:00
}
2020-11-17 21:07:56 +00:00
cosmoV_pop ( state ) ; // pops the return value, it's unused
2020-11-10 01:44:12 +00:00
cosmoV_pushValue ( state , cosmoV_newObj ( newObj ) ) ;
2020-10-28 05:16:30 +00:00
break ;
}
case COBJ_CFUNCTION : {
// it's a C function, so call it
2020-12-04 06:04:14 +00:00
CosmoCFunction cfunc = ( ( CObjCFunction * ) cosmoV_readObj ( * val ) ) - > cfunc ;
2020-11-13 23:39:47 +00:00
callCFunction ( state , cfunc , args , 0 ) ;
2020-10-28 05:16:30 +00:00
break ;
}
default :
2020-11-13 18:54:06 +00:00
cosmoV_error ( state , " Cannot call non-function value! " ) ;
2020-10-28 05:16:30 +00:00
return COSMOVM_RUNTIME_ERR ;
}
return state - > panic ? COSMOVM_RUNTIME_ERR : COSMOVM_OK ;
}
static inline bool isFalsey ( StkPtr val ) {
2020-12-04 06:04:14 +00:00
return IS_NIL ( * val ) | | ( IS_BOOLEAN ( * val ) & & ! cosmoV_readBoolean ( * val ) ) ;
2020-10-28 05:16:30 +00:00
}
2020-12-05 23:55:09 +00:00
COSMO_API void cosmoV_makeObject ( CState * state , int pairs ) {
2020-11-12 23:17:41 +00:00
StkPtr key , val ;
2020-11-30 18:50:55 +00:00
CObjObject * newObj = cosmoO_newObject ( state ) ;
2020-11-12 23:17:41 +00:00
cosmoV_pushValue ( state , cosmoV_newObj ( newObj ) ) ; // so our GC doesn't free our new object
for ( int i = 0 ; i < pairs ; i + + ) {
val = cosmoV_getTop ( state , ( i * 2 ) + 1 ) ;
key = cosmoV_getTop ( state , ( i * 2 ) + 2 ) ;
// set key/value pair
CValue * newVal = cosmoT_insert ( state , & newObj - > tbl , * key ) ;
* newVal = * val ;
}
// once done, pop everything off the stack + push new object
2020-11-30 18:50:55 +00:00
cosmoV_setTop ( state , ( pairs * 2 ) + 1 ) ; // + 1 for our object
2020-11-12 23:17:41 +00:00
cosmoV_pushValue ( state , cosmoV_newObj ( newObj ) ) ;
}
2020-12-10 02:32:42 +00:00
COSMO_API void cosmoV_makeDictionary ( CState * state , int pairs ) {
StkPtr key , val ;
CObjDict * newObj = cosmoO_newDictionary ( state ) ;
cosmoV_pushValue ( state , cosmoV_newObj ( newObj ) ) ; // so our GC doesn't free our new dictionary
for ( int i = 0 ; i < pairs ; i + + ) {
val = cosmoV_getTop ( state , ( i * 2 ) + 1 ) ;
key = cosmoV_getTop ( state , ( i * 2 ) + 2 ) ;
// set key/value pair
CValue * newVal = cosmoT_insert ( state , & newObj - > tbl , * key ) ;
* newVal = * val ;
}
// once done, pop everything off the stack + push new dictionary
cosmoV_setTop ( state , ( pairs * 2 ) + 1 ) ; // + 1 for our dictionary
cosmoV_pushValue ( state , cosmoV_newObj ( newObj ) ) ;
}
2020-11-13 05:04:09 +00:00
COSMO_API bool cosmoV_getObject ( CState * state , CObjObject * object , CValue key , CValue * val ) {
if ( cosmoO_getObject ( state , object , key , val ) ) {
2020-12-04 06:04:14 +00:00
if ( IS_OBJ ( * val ) ) {
if ( cosmoV_readObj ( * val ) - > type = = COBJ_CLOSURE ) { // is it a function? if so, make it a method to the current object
CObjMethod * method = cosmoO_newMethod ( state , ( CObjClosure * ) cosmoV_readObj ( * val ) , object ) ;
2020-11-13 05:04:09 +00:00
* val = cosmoV_newObj ( method ) ;
2020-12-04 06:04:14 +00:00
} else if ( cosmoV_readObj ( * val ) - > type = = COBJ_CFUNCTION ) {
CObjMethod * method = cosmoO_newCMethod ( state , ( CObjCFunction * ) cosmoV_readObj ( * val ) , object ) ;
2020-11-13 05:04:09 +00:00
* val = cosmoV_newObj ( method ) ;
}
}
return true ;
}
return false ;
}
2020-11-30 18:50:55 +00:00
# define NUMBEROP(typeConst, op) \
2020-10-28 05:16:30 +00:00
StkPtr valA = cosmoV_getTop ( state , 1 ) ; \
StkPtr valB = cosmoV_getTop ( state , 0 ) ; \
2020-12-04 06:04:14 +00:00
if ( IS_NUMBER ( * valA ) & & IS_NUMBER ( * valB ) ) { \
2020-10-28 05:16:30 +00:00
cosmoV_setTop ( state , 2 ) ; /* pop the 2 values */ \
2020-12-04 06:04:14 +00:00
cosmoV_pushValue ( state , typeConst ( cosmoV_readNumber ( * valA ) op cosmoV_readNumber ( * valB ) ) ) ; \
2020-10-28 05:16:30 +00:00
} else { \
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Expected numbers, got %s and %s! " , cosmoV_typeStr ( * valA ) , cosmoV_typeStr ( * valB ) ) ; \
2020-10-28 05:16:30 +00:00
} \
2020-11-10 01:44:12 +00:00
// returns false if panic
bool cosmoV_execute ( CState * state ) {
2020-10-28 05:16:30 +00:00
CCallFrame * frame = & state - > callFrame [ state - > frameCount - 1 ] ; // grabs the current frame
CValue * constants = frame - > closure - > function - > chunk . constants . values ; // cache the pointer :)
# define READBYTE() *frame->pc++
# define READUINT() (frame->pc += 2, *(uint16_t*)(&frame->pc[-2]))
while ( ! state - > panic ) {
/*disasmInstr(&frame->closure->function->chunk, frame->pc - frame->closure->function->chunk.buf, 0);
printf ( " \n " ) ; */
switch ( READBYTE ( ) ) {
case OP_LOADCONST : { // push const[uint] to stack
uint16_t indx = READUINT ( ) ;
cosmoV_pushValue ( state , constants [ indx ] ) ;
break ;
}
case OP_SETGLOBAL : {
uint16_t indx = READUINT ( ) ;
CValue ident = constants [ indx ] ; // grabs identifier
CValue * val = cosmoT_insert ( state , & state - > globals , ident ) ;
* val = * cosmoV_pop ( state ) ; // sets the value in the hash table
break ;
}
case OP_GETGLOBAL : {
uint16_t indx = READUINT ( ) ;
CValue ident = constants [ indx ] ; // grabs identifier
CValue val ; // to hold our value
cosmoT_get ( & state - > globals , ident , & val ) ;
cosmoV_pushValue ( state , val ) ; // pushes the value to the stack
break ;
}
case OP_SETLOCAL : {
2020-11-17 19:17:23 +00:00
uint8_t indx = READBYTE ( ) ;
frame - > base [ indx ] = * cosmoV_pop ( state ) ;
2020-10-28 05:16:30 +00:00
break ;
}
case OP_GETLOCAL : {
2020-11-17 19:17:23 +00:00
uint8_t indx = READBYTE ( ) ;
cosmoV_pushValue ( state , frame - > base [ indx ] ) ;
2020-10-28 05:16:30 +00:00
break ;
}
case OP_GETUPVAL : {
uint8_t indx = READBYTE ( ) ;
cosmoV_pushValue ( state , * frame - > closure - > upvalues [ indx ] - > val ) ;
break ;
}
case OP_SETUPVAL : {
uint8_t indx = READBYTE ( ) ;
* frame - > closure - > upvalues [ indx ] - > val = * cosmoV_pop ( state ) ;
break ;
}
2020-11-30 18:50:55 +00:00
case OP_PEJMP : { // pop equality jump
2020-10-28 05:16:30 +00:00
uint16_t offset = READUINT ( ) ;
if ( isFalsey ( cosmoV_pop ( state ) ) ) { // pop, if the condition is false, jump!
frame - > pc + = offset ;
}
break ;
}
2020-11-30 18:50:55 +00:00
case OP_EJMP : { // equality jump
2020-10-28 05:16:30 +00:00
uint16_t offset = READUINT ( ) ;
if ( isFalsey ( cosmoV_getTop ( state , 0 ) ) ) { // if the condition is false, jump!
frame - > pc + = offset ;
}
break ;
}
2020-11-30 18:50:55 +00:00
case OP_JMP : { // jump
2020-10-28 05:16:30 +00:00
uint16_t offset = READUINT ( ) ;
frame - > pc + = offset ;
break ;
}
case OP_JMPBACK : {
uint16_t offset = READUINT ( ) ;
frame - > pc - = offset ;
break ;
}
case OP_POP : { // pops value off the stack
cosmoV_setTop ( state , READBYTE ( ) ) ;
break ;
}
case OP_CALL : {
uint8_t args = READBYTE ( ) ;
2020-11-10 01:44:12 +00:00
COSMOVMRESULT result = cosmoV_call ( state , args ) ;
2020-10-28 05:16:30 +00:00
if ( result ! = COSMOVM_OK ) {
return result ;
}
break ;
}
case OP_CLOSURE : {
uint16_t index = READUINT ( ) ;
CObjFunction * func = cosmoV_readFunction ( constants [ index ] ) ;
CObjClosure * closure = cosmoO_newClosure ( state , func ) ;
cosmoV_pushValue ( state , cosmoV_newObj ( ( CObj * ) closure ) ) ;
for ( int i = 0 ; i < closure - > upvalueCount ; i + + ) {
uint8_t encoding = READBYTE ( ) ;
uint8_t index = READBYTE ( ) ;
if ( encoding = = OP_GETUPVAL ) {
// capture upvalue from current frame's closure
closure - > upvalues [ i ] = frame - > closure - > upvalues [ index ] ;
} else {
// capture local
closure - > upvalues [ i ] = captureUpvalue ( state , frame - > base + index ) ;
}
}
break ;
}
case OP_CLOSE : {
closeUpvalues ( state , state - > top - 1 ) ;
cosmoV_pop ( state ) ;
break ;
}
2020-12-10 02:32:42 +00:00
case OP_NEWDICT : {
uint16_t pairs = READUINT ( ) ;
cosmoV_makeDictionary ( state , pairs ) ;
break ;
}
case OP_INDEX : {
StkPtr key = cosmoV_getTop ( state , 0 ) ; // key should be the top of the stack
StkPtr temp = cosmoV_getTop ( state , 1 ) ; // after that should be the dictionary
// sanity check
if ( IS_OBJ ( * temp ) ) {
CValue val ; // to hold our value
if ( cosmoV_readObj ( * temp ) - > type = = COBJ_DICT ) {
CObjDict * dict = ( CObjDict * ) cosmoV_readObj ( * temp ) ;
cosmoT_get ( & dict - > tbl , * key , & val ) ;
} else if ( cosmoV_readObj ( * temp ) - > type = = COBJ_OBJECT ) { // check for __index!
CObjObject * object = ( CObjObject * ) cosmoV_readObj ( * temp ) ;
if ( ! cosmoO_indexObject ( state , object , * key , & val ) )
break ;
} else {
cosmoV_error ( state , " Couldn't index type %s! " , cosmoV_typeStr ( * temp ) ) ;
break ;
}
cosmoV_setTop ( state , 2 ) ; // pops the object & the key
cosmoV_pushValue ( state , val ) ; // pushes the field result
} else {
cosmoV_error ( state , " Couldn't index type %s! " , cosmoV_typeStr ( * temp ) ) ;
}
break ;
}
case OP_NEWINDEX : {
StkPtr value = cosmoV_getTop ( state , 0 ) ; // value is at the top of the stack
StkPtr key = cosmoV_getTop ( state , 1 ) ;
StkPtr temp = cosmoV_getTop ( state , 2 ) ; // object is after the key
// sanity check
if ( IS_OBJ ( * temp ) ) {
if ( cosmoV_readObj ( * temp ) - > type = = COBJ_DICT ) {
CObjDict * dict = ( CObjDict * ) cosmoV_readObj ( * temp ) ;
CValue * newVal = cosmoT_insert ( state , & dict - > tbl , * key ) ;
* newVal = * value ; // set the index
} else if ( cosmoV_readObj ( * temp ) - > type = = COBJ_OBJECT ) { // check for __newindex!
CObjObject * object = ( CObjObject * ) cosmoV_readObj ( * temp ) ;
if ( ! cosmoO_newIndexObject ( state , object , * key , * value ) )
break ;
} else {
cosmoV_error ( state , " Couldn't set index with type %s! " , cosmoV_typeStr ( * temp ) ) ;
break ;
}
// pop everything off the stack
cosmoV_setTop ( state , 3 ) ;
} else {
cosmoV_error ( state , " Couldn't set index with type %s! " , cosmoV_typeStr ( * temp ) ) ;
}
break ;
}
2020-11-04 04:10:51 +00:00
case OP_NEWOBJECT : {
2020-11-18 20:35:58 +00:00
uint16_t pairs = READUINT ( ) ;
2020-12-05 23:55:09 +00:00
cosmoV_makeObject ( state , pairs ) ;
2020-11-04 04:10:51 +00:00
break ;
}
case OP_GETOBJECT : {
StkPtr key = cosmoV_getTop ( state , 0 ) ; // key should be the top of the stack
StkPtr temp = cosmoV_getTop ( state , 1 ) ; // after that should be the object
// sanity check
2020-12-04 06:04:14 +00:00
if ( ! IS_OBJ ( * temp ) | | cosmoV_readObj ( * temp ) - > type ! = COBJ_OBJECT ) {
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Couldn't get from type %s! " , cosmoV_typeStr ( * temp ) ) ;
2020-11-04 04:10:51 +00:00
break ;
}
2020-12-04 06:04:14 +00:00
CObjObject * object = ( CObjObject * ) cosmoV_readObj ( * temp ) ;
2020-11-04 04:10:51 +00:00
CValue val ; // to hold our value
2020-11-13 05:04:09 +00:00
cosmoV_getObject ( state , object , * key , & val ) ;
2020-11-04 04:10:51 +00:00
cosmoV_setTop ( state , 2 ) ; // pops the object & the key
cosmoV_pushValue ( state , val ) ; // pushes the field result
break ;
}
case OP_SETOBJECT : {
StkPtr value = cosmoV_getTop ( state , 0 ) ; // value is at the top of the stack
StkPtr key = cosmoV_getTop ( state , 1 ) ;
StkPtr temp = cosmoV_getTop ( state , 2 ) ; // object is after the key
// sanity check
2020-12-04 06:04:14 +00:00
if ( ! IS_OBJ ( * temp ) | | cosmoV_readObj ( * temp ) - > type ! = COBJ_OBJECT ) {
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Couldn't set a field on type %s! " , cosmoV_typeStr ( * temp ) ) ;
2020-11-04 04:10:51 +00:00
break ;
}
2020-12-04 06:04:14 +00:00
CObjObject * object = ( CObjObject * ) cosmoV_readObj ( * temp ) ;
2020-11-06 01:53:55 +00:00
cosmoO_setObject ( state , object , * key , * value ) ;
2020-11-04 04:10:51 +00:00
// pop everything off the stack
cosmoV_setTop ( state , 3 ) ;
break ;
}
2020-11-13 23:45:14 +00:00
case OP_INVOKE : { // this is an optimization made by the parser, instead of allocating a CObjMethod every time we want to invoke a method, we shrink it down into one minimal instruction!
2020-11-13 23:39:47 +00:00
uint8_t args = READBYTE ( ) ;
StkPtr key = cosmoV_getTop ( state , args ) ; // grabs key from stack
StkPtr temp = cosmoV_getTop ( state , args + 1 ) ; // grabs object from stack
// sanity check
2020-12-04 06:04:14 +00:00
if ( ! IS_OBJ ( * temp ) | | cosmoV_readObj ( * temp ) - > type ! = COBJ_OBJECT ) {
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Couldn't get from non-object type %s! " , cosmoV_typeStr ( * temp ) ) ;
2020-11-13 23:39:47 +00:00
break ;
}
2020-12-04 06:04:14 +00:00
CObjObject * object = ( CObjObject * ) cosmoV_readObj ( * temp ) ;
2020-11-13 23:39:47 +00:00
CValue val ; // to hold our value
2020-11-13 23:45:14 +00:00
cosmoO_getObject ( state , object , * key , & val ) ; // we use cosmoO_getObject instead of the cosmoV_getObject wrapper so we get the raw value from the object instead of the CObjMethod wrapper
2020-11-13 23:39:47 +00:00
2020-11-17 21:07:56 +00:00
// now invoke the method!
invokeMethod ( state , object , val , args ) ;
2020-11-13 23:39:47 +00:00
// moves return value & resets stack (key now points to the stack location of our return value)
* temp = * key ;
state - > top = key ;
break ;
}
2020-10-28 05:16:30 +00:00
case OP_ADD : { // pop 2 values off the stack & try to add them together
2020-11-30 18:50:55 +00:00
NUMBEROP ( cosmoV_newNumber , + ) ;
2020-10-28 05:16:30 +00:00
break ;
}
case OP_SUB : { // pop 2 values off the stack & try to subtracts them
2020-11-30 18:50:55 +00:00
NUMBEROP ( cosmoV_newNumber , - )
2020-10-28 05:16:30 +00:00
break ;
}
case OP_MULT : { // pop 2 values off the stack & try to multiplies them together
2020-11-30 18:50:55 +00:00
NUMBEROP ( cosmoV_newNumber , * )
2020-10-28 05:16:30 +00:00
break ;
}
case OP_DIV : { // pop 2 values off the stack & try to divides them
2020-11-30 18:50:55 +00:00
NUMBEROP ( cosmoV_newNumber , / )
2020-10-28 05:16:30 +00:00
break ;
}
case OP_NOT : {
2020-11-30 18:40:36 +00:00
cosmoV_pushBoolean ( state , isFalsey ( cosmoV_pop ( state ) ) ) ;
2020-10-28 05:16:30 +00:00
break ;
}
case OP_NEGATE : { // pop 1 value off the stack & try to negate
StkPtr val = cosmoV_getTop ( state , 0 ) ;
2020-12-04 06:04:14 +00:00
if ( IS_NUMBER ( * val ) ) {
2020-10-28 05:16:30 +00:00
cosmoV_pop ( state ) ;
2020-12-04 06:04:14 +00:00
cosmoV_pushNumber ( state , - ( cosmoV_readNumber ( * val ) ) ) ;
2020-10-28 05:16:30 +00:00
} else {
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Expected number, got %s! " , cosmoV_typeStr ( * val ) ) ;
2020-10-28 05:16:30 +00:00
}
break ;
}
2020-12-12 23:34:54 +00:00
case OP_COUNT : { // pop 1 value off the stack & if it's a dictionary return the ammount of active entries it has
2020-11-30 18:32:04 +00:00
StkPtr temp = cosmoV_getTop ( state , 0 ) ;
2020-12-12 23:34:54 +00:00
if ( ! IS_OBJ ( * temp ) | | cosmoV_readObj ( * temp ) - > type ! = COBJ_DICT ) {
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Expected object, got %s! " , cosmoV_typeStr ( * temp ) ) ;
break ;
}
2020-12-12 23:34:54 +00:00
CObjDict * dict = ( CObjDict * ) cosmoV_readObj ( * temp ) ;
2020-11-30 18:32:04 +00:00
cosmoV_pop ( state ) ;
2020-12-12 23:34:54 +00:00
cosmoV_pushNumber ( state , cosmoT_count ( & dict - > tbl ) ) ; // pushes the count onto the stack
2020-11-30 18:32:04 +00:00
break ;
}
2020-10-28 05:16:30 +00:00
case OP_CONCAT : {
uint8_t vals = READBYTE ( ) ;
StkPtr start = state - > top - vals ;
StkPtr end = cosmoV_getTop ( state , 0 ) ;
CObjString * result = cosmoV_toString ( state , * start ) ;
for ( StkPtr current = start + 1 ; current < = end ; current + + ) {
cosmoV_pushValue ( state , cosmoV_newObj ( result ) ) ; // so our GC can find our current result string
CObjString * otherStr = cosmoV_toString ( state , * current ) ;
cosmoV_pushValue ( state , cosmoV_newObj ( otherStr ) ) ; // also so our GC won't free otherStr
result = cosmoV_concat ( state , result , otherStr ) ;
cosmoV_setTop ( state , 2 ) ; // pop result & otherStr off the stack
}
state - > top = start ;
cosmoV_pushValue ( state , cosmoV_newObj ( result ) ) ;
break ;
}
2020-11-19 20:41:21 +00:00
case OP_INCLOCAL : { // this leaves the value on the stack
int8_t inc = READBYTE ( ) - 128 ; // ammount we're incrementing by
uint8_t indx = READBYTE ( ) ;
StkPtr val = & frame - > base [ indx ] ;
// check that it's a number value
2020-12-04 06:04:14 +00:00
if ( IS_NUMBER ( * val ) ) {
2020-11-19 20:41:21 +00:00
cosmoV_pushValue ( state , * val ) ; // pushes old value onto the stack :)
2020-12-04 06:04:14 +00:00
* val = cosmoV_newNumber ( cosmoV_readNumber ( * val ) + inc ) ;
2020-11-19 20:41:21 +00:00
} else {
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Expected number, got %s! " , cosmoV_typeStr ( * val ) ) ;
2020-11-19 20:41:21 +00:00
}
break ;
}
case OP_INCGLOBAL : {
int8_t inc = READBYTE ( ) - 128 ; // ammount we're incrementing by
uint16_t indx = READUINT ( ) ;
CValue ident = constants [ indx ] ; // grabs identifier
CValue * val = cosmoT_insert ( state , & state - > globals , ident ) ;
// check that it's a number value
2020-12-04 06:04:14 +00:00
if ( IS_NUMBER ( * val ) ) {
2020-11-19 20:41:21 +00:00
cosmoV_pushValue ( state , * val ) ; // pushes old value onto the stack :)
2020-12-04 06:04:14 +00:00
* val = cosmoV_newNumber ( cosmoV_readNumber ( * val ) + inc ) ;
2020-11-19 20:41:21 +00:00
} else {
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Expected number, got %s! " , cosmoV_typeStr ( * val ) ) ;
2020-11-19 20:41:21 +00:00
}
break ;
}
case OP_INCUPVAL : {
int8_t inc = READBYTE ( ) - 128 ; // ammount we're incrementing by
uint8_t indx = READBYTE ( ) ;
CValue * val = frame - > closure - > upvalues [ indx ] - > val ;
// check that it's a number value
2020-12-04 06:04:14 +00:00
if ( IS_NUMBER ( * val ) ) {
2020-11-19 20:41:21 +00:00
cosmoV_pushValue ( state , * val ) ; // pushes old value onto the stack :)
2020-12-04 06:04:14 +00:00
* val = cosmoV_newNumber ( cosmoV_readNumber ( * val ) + inc ) ;
2020-11-19 20:41:21 +00:00
} else {
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Expected number, got %s! " , cosmoV_typeStr ( * val ) ) ;
2020-11-19 20:41:21 +00:00
}
break ;
}
2020-12-10 02:32:42 +00:00
case OP_INCINDEX : {
int8_t inc = READBYTE ( ) - 128 ; // ammount we're incrementing by
StkPtr temp = cosmoV_getTop ( state , 1 ) ; // object should be above the key
StkPtr key = cosmoV_getTop ( state , 0 ) ; // grabs key
if ( IS_OBJ ( * temp ) ) {
if ( cosmoV_readObj ( * temp ) - > type = = COBJ_DICT ) {
CObjDict * dict = ( CObjDict * ) cosmoV_readObj ( * temp ) ;
CValue * val = cosmoT_insert ( state , & dict - > tbl , * key ) ;
// pops dict & key from stack
cosmoV_setTop ( state , 2 ) ;
if ( IS_NUMBER ( * val ) ) {
cosmoV_pushValue ( state , * val ) ; // pushes old value onto the stack :)
* val = cosmoV_newNumber ( cosmoV_readNumber ( * val ) + inc ) ;
} else {
cosmoV_error ( state , " Expected number, got %s! " , cosmoV_typeStr ( * val ) ) ;
break ;
}
} else if ( cosmoV_readObj ( * temp ) - > type = = COBJ_OBJECT ) { // check for __newindex!
CObjObject * object = ( CObjObject * ) cosmoV_readObj ( * temp ) ;
CValue val ;
// call __index
if ( cosmoO_indexObject ( state , object , * key , & val ) ) {
if ( IS_NUMBER ( val ) ) {
cosmoV_pushValue ( state , val ) ; // pushes old value onto the stack :)
// call __newindex
cosmoO_newIndexObject ( state , object , * key , cosmoV_newNumber ( cosmoV_readNumber ( val ) + inc ) ) ;
} else {
cosmoV_error ( state , " Expected number, got %s! " , cosmoV_typeStr ( val ) ) ;
break ;
}
}
} else {
cosmoV_error ( state , " Couldn't set index with type %s! " , cosmoV_typeStr ( * temp ) ) ;
break ;
}
} else {
cosmoV_error ( state , " Couldn't set index with type %s! " , cosmoV_typeStr ( * temp ) ) ;
}
break ;
}
2020-11-19 20:41:21 +00:00
case OP_INCOBJECT : {
int8_t inc = READBYTE ( ) - 128 ; // ammount we're incrementing by
uint16_t indx = READUINT ( ) ;
StkPtr temp = cosmoV_getTop ( state , 0 ) ; // object should be at the top of the stack
CValue ident = constants [ indx ] ; // grabs identifier
// sanity check
2020-12-04 06:04:14 +00:00
if ( ! IS_OBJ ( * temp ) | | cosmoV_readObj ( * temp ) - > type ! = COBJ_OBJECT ) {
2020-11-30 18:32:04 +00:00
cosmoV_error ( state , " Couldn't set a field on non-object type %s! " , cosmoV_typeStr ( * temp ) ) ;
2020-11-19 20:41:21 +00:00
break ;
}
2020-12-04 06:04:14 +00:00
CObjObject * object = ( CObjObject * ) cosmoV_readObj ( * temp ) ;
2020-12-10 02:32:42 +00:00
CValue val ;
cosmoO_getObject ( state , object , ident , & val ) ;
2020-11-19 20:41:21 +00:00
// pop the object off the stack
cosmoV_pop ( state ) ;
// check that it's a number value
2020-12-10 02:32:42 +00:00
if ( IS_NUMBER ( val ) ) {
cosmoV_pushValue ( state , val ) ; // pushes old value onto the stack :)
cosmoO_setObject ( state , object , ident , cosmoV_newNumber ( cosmoV_readNumber ( val ) + inc ) ) ;
2020-11-19 20:41:21 +00:00
} else {
2020-12-10 02:32:42 +00:00
cosmoV_error ( state , " Expected number, got %s! " , cosmoV_typeStr ( val ) ) ;
2020-11-19 20:41:21 +00:00
}
break ;
}
2020-10-28 05:16:30 +00:00
case OP_EQUAL : {
// pop vals
StkPtr valB = cosmoV_pop ( state ) ;
StkPtr valA = cosmoV_pop ( state ) ;
// compare & push
2020-11-30 18:40:36 +00:00
cosmoV_pushBoolean ( state , cosmoV_equal ( * valA , * valB ) ) ;
2020-10-28 05:16:30 +00:00
break ;
}
case OP_GREATER : {
2020-11-30 18:50:55 +00:00
NUMBEROP ( cosmoV_newBoolean , > )
2020-10-28 05:16:30 +00:00
break ;
}
case OP_LESS : {
2020-11-30 18:50:55 +00:00
NUMBEROP ( cosmoV_newBoolean , < )
2020-10-28 05:16:30 +00:00
break ;
}
case OP_GREATER_EQUAL : {
2020-11-30 18:50:55 +00:00
NUMBEROP ( cosmoV_newBoolean , > = )
2020-10-28 05:16:30 +00:00
break ;
}
case OP_LESS_EQUAL : {
2020-11-30 18:50:55 +00:00
NUMBEROP ( cosmoV_newBoolean , < = )
2020-10-28 05:16:30 +00:00
break ;
}
2020-11-30 18:40:36 +00:00
case OP_TRUE : cosmoV_pushBoolean ( state , true ) ; break ;
case OP_FALSE : cosmoV_pushBoolean ( state , false ) ; break ;
2020-10-28 05:16:30 +00:00
case OP_NIL : cosmoV_pushValue ( state , cosmoV_newNil ( ) ) ; break ;
case OP_RETURN : {
2020-11-10 01:44:12 +00:00
return true ;
2020-10-28 05:16:30 +00:00
}
default :
CERROR ( " unknown opcode! " ) ;
exit ( 0 ) ;
}
//cosmoV_printStack(state);
}
# undef READBYTE
# undef READUINT
// we'll only reach this is state->panic is true
2020-11-10 01:44:12 +00:00
return false ;
2020-10-28 05:16:30 +00:00
}
2020-11-30 18:50:55 +00:00
# undef NUMBEROP