mirror of
https://github.com/suchmememanyskill/TegraExplorer.git
synced 2024-11-26 05:42:07 +00:00
296 lines
No EOL
8.1 KiB
C
296 lines
No EOL
8.1 KiB
C
#include "genericClass.h"
|
|
#include "model.h"
|
|
#include "intClass.h"
|
|
#include "compat.h"
|
|
#include "eval.h"
|
|
#include <string.h>
|
|
#include "garbageCollector.h"
|
|
#include "StringClass.h"
|
|
#include "arrayClass.h"
|
|
#include "arrayReferenceClass.h"
|
|
#include "functionClass.h"
|
|
#include "scriptError.h"
|
|
#include "saveClass.h"
|
|
#include "unsolvedArrayClass.h"
|
|
#include "else.h"
|
|
#include "dictionaryClass.h"
|
|
|
|
Variable_t* copyVariableToPtr(Variable_t var) {
|
|
Variable_t* a = malloc(sizeof(Variable_t));
|
|
*a = var;
|
|
a->tagCount = 0;
|
|
addPendingReference(a);
|
|
return a;
|
|
}
|
|
|
|
MemberGetters_t memberGetters[] = {
|
|
{IntClass, getIntegerMember},
|
|
{StringClass, getStringMember},
|
|
{IntArrayClass, getArrayMember},
|
|
{StringArrayClass, getArrayMember},
|
|
{ByteArrayClass, getArrayMember},
|
|
{SolvedArrayReferenceClass, getArrayReferenceMember},
|
|
{UnresolvedArrayClass, getUnsolvedArrayMember},
|
|
{ElseClass, getElseMember},
|
|
{DictionaryClass, getDictMember},
|
|
#ifndef WIN32
|
|
{SaveClass, getSaveMember},
|
|
#endif
|
|
};
|
|
|
|
Variable_t* genericGet(Variable_t* var, CallArgs_t* ref) {
|
|
if (ref->extraAction == ActionExtraMemberName) {
|
|
for (u32 i = 0; i < ARRAY_SIZE(memberGetters); i++) {
|
|
if (var->variableType == memberGetters[i].classType) {
|
|
Variable_t member = memberGetters[i].func(var, ref->extra);
|
|
if (member.variableType == None)
|
|
return NULL;
|
|
|
|
if (member.variableType == ReferenceType) {
|
|
return member.referenceType;
|
|
}
|
|
|
|
addPendingReference(var); // So caller doesn't fall out of scope. Don't forget to free!
|
|
return copyVariableToPtr(member);
|
|
}
|
|
}
|
|
|
|
SCRIPT_FATAL_ERR("Did not find member '%s'", ref->extra);
|
|
}
|
|
else if (ref->extraAction == ActionExtraArrayIndex) {
|
|
Function_t* idx = ref->extra;
|
|
Variable_t *solvedIdx = eval(idx->operations.data, idx->operations.count, 1);
|
|
|
|
if (solvedIdx->variableType != IntClass) {
|
|
SCRIPT_FATAL_ERR("Index is not an integer");
|
|
}
|
|
|
|
Variable_t* res = callMemberFunctionDirect(var, "get", &solvedIdx, 1);
|
|
removePendingReference(solvedIdx);
|
|
return res;
|
|
}
|
|
|
|
SCRIPT_FATAL_ERR("???");
|
|
}
|
|
|
|
Variable_t* genericCallDirect(Variable_t* var, Variable_t** args, u8 len) {
|
|
if (var->variableType != FunctionClass){
|
|
SCRIPT_FATAL_ERR("Call on non function class");
|
|
}
|
|
|
|
if (var->function.builtIn) {
|
|
for (u32 i = 0; i < var->function.len; i++) {
|
|
if (var->function.builtInPtr[i].argCount == len || var->function.builtInPtr[i].argCount == VARARGCOUNT) {
|
|
int valid = 1;
|
|
if (var->function.builtInPtr[i].argCount != VARARGCOUNT) {
|
|
for (u32 j = 0; j < var->function.builtInPtr[i].argCount; j++) {
|
|
if (var->function.builtInPtr[i].argTypes[j] != args[j]->variableType && var->function.builtInPtr[i].argTypes[j] != VARARGCOUNT) {
|
|
valid = 0;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (valid) {
|
|
return var->function.builtInPtr[i].func(var->function.origin, args, len);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
Variable_t *ret = eval(var->function.function.operations.data, var->function.function.operations.count, 1);
|
|
if (ret == NULL)
|
|
return NULL;
|
|
removePendingReference(ret);
|
|
return &emptyClass;
|
|
}
|
|
|
|
SCRIPT_FATAL_ERR("Arguments do not match function defenition(s)");
|
|
}
|
|
|
|
Variable_t* genericCall(Variable_t* var, CallArgs_t* ref) {
|
|
if (var->variableType != FunctionClass){
|
|
SCRIPT_FATAL_ERR("Call on non function class");
|
|
}
|
|
|
|
if (var->function.builtIn) {
|
|
// TODO: implement arg handling
|
|
|
|
Function_t* f = ref->extra;
|
|
if (f->operations.count == 0) {
|
|
return genericCallDirect(var, NULL, 0);
|
|
}
|
|
else {
|
|
//Vector_t argsHolder = newVec(sizeof(Variable_t*), 1);
|
|
Variable_t** argsHolder = NULL;
|
|
if (var->function.builtInPtr->argCount != 0)
|
|
argsHolder = malloc(sizeof(Variable_t*) * var->function.builtInPtr->argCount);
|
|
int argCount = 0;
|
|
int lasti = 0;
|
|
Operator_t* ops = f->operations.data;
|
|
|
|
int tooManyArgs = 0;
|
|
|
|
// Loops trough the function to get all args out
|
|
for (int i = 0; i < f->operations.count; i++) {
|
|
if (ops[i].token == EquationSeperator || i + 1 == f->operations.count) {
|
|
if (i + 1 == f->operations.count)
|
|
i++;
|
|
|
|
if (argCount == var->function.builtInPtr->argCount) {
|
|
tooManyArgs = 1;
|
|
break;
|
|
}
|
|
|
|
if (var->function.firstArgAsFunction && argCount == 0) {
|
|
Function_t f = { .operations = vecFromArray(&ops[lasti], i - lasti, sizeof(Operator_t)) };
|
|
Variable_t var = newFunctionVariable(createFunctionClass(f, NULL));
|
|
var.reference = 1;
|
|
Variable_t* varPtr = copyVariableToPtr(var);
|
|
//vecAdd(&argsHolder, varPtr);
|
|
argsHolder[argCount++] = varPtr;
|
|
}
|
|
else {
|
|
Variable_t* var = eval(&ops[lasti], i - lasti, 1);
|
|
if (var == NULL)
|
|
return NULL; // maybe free first?
|
|
|
|
//vecAdd(&argsHolder, var);
|
|
argsHolder[argCount++] = var;
|
|
}
|
|
|
|
lasti = i + 1;
|
|
}
|
|
}
|
|
|
|
Variable_t* res = NULL;
|
|
|
|
if (!tooManyArgs)
|
|
res = genericCallDirect(var, argsHolder, argCount);
|
|
else {
|
|
SCRIPT_FATAL_ERR("Too many args provided (got more than %d)", argCount);
|
|
}
|
|
|
|
|
|
for (int i = 0; i < argCount; i++)
|
|
removePendingReference(argsHolder[i]);
|
|
|
|
//vecForEach(Variable_t**, tofree, (&argsHolder))
|
|
// removePendingReference(*tofree);
|
|
|
|
FREE(argsHolder);
|
|
|
|
return res;
|
|
}
|
|
}
|
|
else {
|
|
Variable_t *ret = eval(var->function.function.operations.data, var->function.function.operations.count, 1);
|
|
if (ret == NULL)
|
|
return NULL;
|
|
removePendingReference(ret);
|
|
return &emptyClass;
|
|
}
|
|
}
|
|
|
|
// TODO: add staticStorage
|
|
Variable_t getGenericFunctionMember(Variable_t* var, char* memberName, ClassFunctionTableEntry_t* entries, u8 len) {
|
|
Variable_t newVar = {.readOnly = 1, .variableType = FunctionClass};
|
|
newVar.function.origin = var;
|
|
newVar.function.builtIn = 1;
|
|
for (u32 i = 0; i < len; i++) {
|
|
if (!strcmp(entries[i].name, memberName)) {
|
|
newVar.function.builtInPtr = &entries[i];
|
|
|
|
u32 j = i;
|
|
for (; j < len && !strcmp(entries[j].name, memberName); j++);
|
|
newVar.function.len = j - i;
|
|
|
|
return newVar;
|
|
}
|
|
}
|
|
|
|
return (Variable_t){ 0 };
|
|
}
|
|
|
|
Variable_t* callMemberFunction(Variable_t* var, char* memberName, CallArgs_t* args) {
|
|
for (u32 i = 0; i < ARRAY_SIZE(memberGetters); i++) {
|
|
if (var->variableType == memberGetters[i].classType) {
|
|
Variable_t funcRef = memberGetters[i].func(var, memberName);
|
|
if (funcRef.variableType == None) {
|
|
SCRIPT_FATAL_ERR("Did not find member '%s'", memberName);
|
|
}
|
|
|
|
Variable_t* ptr = &funcRef;
|
|
if (funcRef.variableType == ReferenceType) {
|
|
ptr = funcRef.referenceType;
|
|
}
|
|
|
|
return genericCall(ptr, args);
|
|
}
|
|
}
|
|
|
|
SCRIPT_FATAL_ERR("Could not find function table for given type");
|
|
}
|
|
|
|
Variable_t* callMemberFunctionDirect(Variable_t* var, char* memberName, Variable_t** args, u8 argsLen) {
|
|
for (u32 i = 0; i < ARRAY_SIZE(memberGetters); i++) {
|
|
if (var->variableType == memberGetters[i].classType) {
|
|
Variable_t funcRef = memberGetters[i].func(var, memberName);
|
|
if (funcRef.variableType == None) {
|
|
SCRIPT_FATAL_ERR("Did not find member '%s'", memberName);
|
|
}
|
|
|
|
Variable_t* ptr = &funcRef;
|
|
if (funcRef.variableType == ReferenceType) {
|
|
ptr = funcRef.referenceType;
|
|
}
|
|
|
|
return genericCallDirect(ptr, args, argsLen);
|
|
}
|
|
}
|
|
|
|
SCRIPT_FATAL_ERR("Could not find function table for given type");
|
|
}
|
|
|
|
void freeVariableInternal(Variable_t* referencedTarget) {
|
|
switch (referencedTarget->variableType) {
|
|
case StringClass:
|
|
FREE(referencedTarget->string.value);
|
|
break;
|
|
case StringArrayClass:
|
|
vecForEach(char**, stringsInArray, (&referencedTarget->solvedArray.vector)) {
|
|
FREE(*stringsInArray);
|
|
}
|
|
case ByteArrayClass:
|
|
case IntArrayClass:
|
|
vecFree(referencedTarget->solvedArray.vector);
|
|
break;
|
|
case DictionaryClass:;
|
|
vecForEach(Dict_t*, dict, (&referencedTarget->dictionary.vector)) {
|
|
modReference(dict->var, 0);
|
|
free(dict->name);
|
|
}
|
|
FREE(referencedTarget->dictionary.vector.data);
|
|
break;
|
|
case SaveClass:;
|
|
#ifndef WIN32
|
|
save_free_contexts(&referencedTarget->save->saveCtx);
|
|
f_close(&referencedTarget->save->saveFile);
|
|
FREE(referencedTarget->save);
|
|
#endif // !WIN32
|
|
break;
|
|
}
|
|
}
|
|
|
|
void freeVariable(Variable_t** target) {
|
|
// Add specific freeing logic here
|
|
Variable_t* referencedTarget = *target;
|
|
|
|
if (!referencedTarget->reference) {
|
|
freeVariableInternal(referencedTarget);
|
|
}
|
|
|
|
|
|
FREE(referencedTarget);
|
|
*target = NULL;
|
|
} |