mirror of
https://github.com/CPunch/Cosmo.git
synced 2024-11-05 08:10:05 +00:00
added foreach()
This commit is contained in:
parent
219a87c79c
commit
ec8f5e7e4f
@ -7,6 +7,7 @@
|
|||||||
void cosmoB_loadLibrary(CState *state) {
|
void cosmoB_loadLibrary(CState *state) {
|
||||||
cosmoM_freezeGC(state);
|
cosmoM_freezeGC(state);
|
||||||
cosmoV_register(state, "print", cosmoV_newObj(cosmoO_newCFunction(state, cosmoB_print)));
|
cosmoV_register(state, "print", cosmoV_newObj(cosmoO_newCFunction(state, cosmoB_print)));
|
||||||
|
cosmoV_register(state, "foreach", cosmoV_newObj(cosmoO_newCFunction(state, cosmoB_foreach)));
|
||||||
cosmoM_unfreezeGC(state);
|
cosmoM_unfreezeGC(state);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -20,6 +21,41 @@ CValue cosmoB_print(CState *state, int nargs, CValue *args) {
|
|||||||
return cosmoV_newNil(); // print doesn't return any args
|
return cosmoV_newNil(); // print doesn't return any args
|
||||||
}
|
}
|
||||||
|
|
||||||
|
CValue cosmoB_foreach(CState *state, int nargs, CValue *args) {
|
||||||
|
if (nargs != 2) {
|
||||||
|
cosmoV_error(state, "foreach() expected 2 parameters, got %d!", nargs);
|
||||||
|
return cosmoV_newNil();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!IS_DICT(args[0])) {
|
||||||
|
cosmoV_error(state, "foreach() expected first parameter to be <dictionary>, got %s!", cosmoV_typeStr(args[0]));
|
||||||
|
return cosmoV_newNil();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!IS_CALLABLE(args[1])) {
|
||||||
|
cosmoV_error(state, "foreach() expected second parameter to be callable, got %s!", cosmoV_typeStr(args[1]));
|
||||||
|
return cosmoV_newNil();
|
||||||
|
}
|
||||||
|
|
||||||
|
// loop through dictonary table, calling args[1] on active entries
|
||||||
|
CObjDict *dict = (CObjDict*)cosmoV_readObj(args[0]);
|
||||||
|
|
||||||
|
for (int i = 0; i < dict->tbl.capacity; i++) {
|
||||||
|
CTableEntry *entry = &dict->tbl.table[i];
|
||||||
|
|
||||||
|
if (!IS_NIL(entry->key)) {
|
||||||
|
// push key & value, then call args[1]
|
||||||
|
cosmoV_pushValue(state, args[1]);
|
||||||
|
cosmoV_pushValue(state, entry->key);
|
||||||
|
cosmoV_pushValue(state, entry->val);
|
||||||
|
cosmoV_call(state, 2);
|
||||||
|
cosmoV_pop(state); // throw away the return value
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return cosmoV_newNil();
|
||||||
|
}
|
||||||
|
|
||||||
CValue cosmoB_dsetProto(CState *state, int nargs, CValue *args) {
|
CValue cosmoB_dsetProto(CState *state, int nargs, CValue *args) {
|
||||||
if (nargs == 2) {
|
if (nargs == 2) {
|
||||||
CObjObject *obj = cosmoV_readObject(args[0]); // object to set proto too
|
CObjObject *obj = cosmoV_readObject(args[0]); // object to set proto too
|
||||||
|
@ -7,5 +7,6 @@
|
|||||||
COSMO_API void cosmoB_loadLibrary(CState *state);
|
COSMO_API void cosmoB_loadLibrary(CState *state);
|
||||||
COSMO_API void cosmoB_loadDebug(CState *state);
|
COSMO_API void cosmoB_loadDebug(CState *state);
|
||||||
COSMO_API CValue cosmoB_print(CState *state, int nargs, CValue *args);
|
COSMO_API CValue cosmoB_print(CState *state, int nargs, CValue *args);
|
||||||
|
COSMO_API CValue cosmoB_foreach(CState *state, int nargs, CValue *args);
|
||||||
|
|
||||||
#endif
|
#endif
|
@ -111,6 +111,11 @@ static inline bool isObjType(CValue val, CObjType type) {
|
|||||||
return IS_OBJ(val) && cosmoV_readObj(val)->type == type;
|
return IS_OBJ(val) && cosmoV_readObj(val)->type == type;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// just protects against macro expansion
|
||||||
|
static inline bool IS_CALLABLE(CValue val) {
|
||||||
|
return IS_CLOSURE(val) || IS_CFUNCTION(val);
|
||||||
|
}
|
||||||
|
|
||||||
CObj *cosmoO_allocateBase(CState *state, size_t sz, CObjType type);
|
CObj *cosmoO_allocateBase(CState *state, size_t sz, CObjType type);
|
||||||
void cosmoO_free(CState *state, CObj* obj);
|
void cosmoO_free(CState *state, CObj* obj);
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user