neonucleus/src/computer.c
2025-07-28 19:03:22 +02:00

649 lines
20 KiB
C

#include "computer.h"
#include "component.h"
#include "universe.h"
#include "neonucleus.h"
#include "resource.h"
nn_computer *nn_newComputer(nn_universe *universe, nn_address address, nn_architecture *arch, void *userdata, nn_size_t memoryLimit, nn_size_t componentLimit) {
nn_Alloc *alloc = &universe->ctx.allocator;
nn_computer *c = nn_alloc(alloc, sizeof(nn_computer));
c->components = nn_alloc(alloc, sizeof(nn_component) * componentLimit);
if(c->components == NULL) {
nn_dealloc(alloc, c, sizeof(nn_computer));
return NULL;
}
c->address = nn_strdup(alloc, address);
if(c->address == NULL) {
nn_dealloc(alloc, c->components, sizeof(nn_component) * componentLimit);
nn_dealloc(alloc, c, sizeof(nn_computer));
return NULL;
}
c->lock = nn_newGuard(&universe->ctx);
if(c->lock == NULL) {
nn_deallocStr(alloc, c->address);
nn_dealloc(alloc, c->components, sizeof(nn_component) * componentLimit);
nn_dealloc(alloc, c, sizeof(nn_computer));
return NULL;
}
c->timeOffset = nn_getTime(universe);
c->supportedArchCount = 0;
c->argc = 0;
c->retc = 0;
c->err = NULL;
c->allocatedError = false;
c->state = NN_STATE_SETUP;
c->componentLen = 0;
c->componentCap = componentLimit;
c->userCount = 0;
c->maxEnergy = 5000;
c->signalCount = 0;
c->universe = universe;
c->arch = arch;
c->nextArch = arch;
c->userdata = userdata;
c->memoryTotal = memoryLimit;
c->tmpAddress = NULL;
c->temperature = 30;
c->roomTemperature = 30;
c->temperatureCoefficient = 1;
c->callCost = 0;
c->callBudget = 256;
// Setup Architecture
c->archState = c->arch->setup(c, c->arch->userdata);
if(c->archState == NULL) {
nn_deleteGuard(&universe->ctx, c->lock);
nn_deallocStr(alloc, c->address);
nn_dealloc(alloc, c->components, sizeof(nn_component) * componentLimit);
nn_dealloc(alloc, c, sizeof(nn_computer));
return NULL;
}
c->rid = NN_NULL_RESOURCE;
for(nn_size_t i = 0; i < NN_MAX_CONCURRENT_RESOURCES; i++) {
c->resources[i].id = NN_NULL_RESOURCE;
}
return c;
}
nn_universe *nn_getUniverse(nn_computer *computer) {
return computer->universe;
}
void nn_setTmpAddress(nn_computer *computer, nn_address tmp) {
nn_deallocStr(&computer->universe->ctx.allocator, computer->tmpAddress);
computer->tmpAddress = nn_strdup(&computer->universe->ctx.allocator, tmp);
}
nn_address nn_getComputerAddress(nn_computer *computer) {
return computer->address;
}
nn_address nn_getTmpAddress(nn_computer *computer) {
return computer->tmpAddress;
}
int nn_tickComputer(nn_computer *computer) {
computer->callCost = 0;
computer->state = NN_STATE_RUNNING;
nn_clearError(computer);
computer->arch->tick(computer, computer->archState, computer->arch->userdata);
return nn_getState(computer);
}
double nn_getUptime(nn_computer *computer) {
return nn_getTime(computer->universe) - computer->timeOffset;
}
nn_size_t nn_getComputerMemoryUsed(nn_computer *computer) {
return computer->arch->getMemoryUsage(computer, computer->archState, computer->arch->userdata);
}
nn_size_t nn_getComputerMemoryTotal(nn_computer *computer) {
return computer->memoryTotal;
}
void *nn_getComputerUserData(nn_computer *computer) {
return computer->userdata;
}
void nn_addSupportedArchitecture(nn_computer *computer, nn_architecture *arch) {
if(computer->supportedArchCount == NN_MAX_ARCHITECTURES) return;
computer->supportedArch[computer->supportedArchCount] = arch;
computer->supportedArchCount++;
}
nn_architecture *nn_getSupportedArchitecture(nn_computer *computer, nn_size_t idx) {
if(idx >= computer->supportedArchCount) return NULL;
return computer->supportedArch[idx];
}
nn_architecture *nn_getArchitecture(nn_computer *computer) {
return computer->arch;
}
nn_architecture *nn_getNextArchitecture(nn_computer *computer) {
return computer->nextArch;
}
void nn_setNextArchitecture(nn_computer *computer, nn_architecture *arch) {
computer->nextArch = arch;
}
void nn_deleteComputer(nn_computer *computer) {
nn_clearError(computer);
nn_resetCall(computer);
while(computer->signalCount > 0) {
nn_popSignal(computer);
}
nn_Alloc *a = &computer->universe->ctx.allocator;
for(nn_size_t i = 0; i < computer->userCount; i++) {
nn_deallocStr(a, computer->users[i]);
}
for(nn_size_t i = 0; i < NN_MAX_CONCURRENT_RESOURCES; i++) {
if(computer->resources[i].id != NN_NULL_RESOURCE) {
nn_resource_release(computer, computer->resources[i].id);
}
}
computer->arch->teardown(computer, computer->archState, computer->arch->userdata);
nn_deleteGuard(&computer->universe->ctx, computer->lock);
nn_deallocStr(a, computer->address);
nn_deallocStr(a, computer->tmpAddress);
nn_dealloc(a, computer->components, sizeof(nn_component) * computer->componentCap);
nn_dealloc(a, computer, sizeof(nn_computer));
}
const char *nn_pushSignal(nn_computer *computer, nn_value *values, nn_size_t len) {
if(len > NN_MAX_SIGNAL_VALS) return "too many values";
if(len == 0) return "missing event";
// no OOM for you hehe
if(nn_measurePacketSize(values, len) > NN_MAX_SIGNAL_SIZE) {
return "too big";
}
if(computer->signalCount == NN_MAX_SIGNALS) return "too many signals";
computer->signals[computer->signalCount].len = len;
for(nn_size_t i = 0; i < len; i++) {
computer->signals[computer->signalCount].values[i] = values[i];
}
computer->signalCount++;
return NULL;
}
nn_value nn_fetchSignalValue(nn_computer *computer, nn_size_t index) {
if(computer->signalCount == 0) return nn_values_nil();
nn_signal *p = computer->signals;
if(index >= p->len) return nn_values_nil();
return p->values[index];
}
nn_size_t nn_signalSize(nn_computer *computer) {
if(computer->signalCount == 0) return 0;
return computer->signals[0].len;
}
void nn_popSignal(nn_computer *computer) {
if(computer->signalCount == 0) return;
nn_signal *p = computer->signals;
for(nn_size_t i = 0; i < p->len; i++) {
nn_values_drop(p->values[i]);
}
for(nn_size_t i = 1; i < computer->signalCount; i++) {
computer->signals[i-1] = computer->signals[i];
}
computer->signalCount--;
}
const char *nn_addUser(nn_computer *computer, const char *name) {
if(computer->userCount == NN_MAX_USERS) return "too many users";
char *user = nn_strdup(&computer->universe->ctx.allocator, name);
if(user == NULL) return "out of memory";
computer->users[computer->userCount] = user;
computer->userCount++;
return NULL;
}
void nn_deleteUser(nn_computer *computer, const char *name) {
nn_size_t j = 0;
for(nn_size_t i = 0; i < computer->userCount; i++) {
char *user = computer->users[i];
if(nn_strcmp(user, name) == 0) {
nn_deallocStr(&computer->universe->ctx.allocator, user);
} else {
computer->users[j] = user;
j++;
}
}
computer->userCount = j;
}
const char *nn_indexUser(nn_computer *computer, nn_size_t idx) {
if(idx >= computer->userCount) return NULL;
return computer->users[idx];
}
nn_bool_t nn_isUser(nn_computer *computer, const char *name) {
if(computer->userCount == 0) return true;
for(nn_size_t i = 0; i < computer->userCount; i++) {
if(nn_strcmp(computer->users[i], name) == 0) return true;
}
return false;
}
void nn_setCallBudget(nn_computer *computer, double callBudget) {
computer->callBudget = callBudget;
}
double nn_getCallBudget(nn_computer *computer) {
return computer->callBudget;
}
void nn_callCost(nn_computer *computer, double cost) {
computer->callCost += cost;
if(computer->callCost >= computer->callBudget) nn_triggerIndirect(computer);
}
double nn_getCallCost(nn_computer *computer) {
return computer->callCost;
}
nn_bool_t nn_isOverworked(nn_computer *computer) {
return computer->state == NN_STATE_OVERWORKED;
}
void nn_triggerIndirect(nn_computer *computer) {
computer->state = NN_STATE_OVERWORKED;
}
int nn_getState(nn_computer *computer) {
return computer->state;
}
void nn_setState(nn_computer *computer, int state) {
computer->state = state;
}
void nn_setEnergyInfo(nn_computer *computer, double energy, double capacity) {
computer->energy = energy;
computer->maxEnergy = capacity;
}
double nn_getEnergy(nn_computer *computer) {
return computer->energy;
}
double nn_getMaxEnergy(nn_computer *computer) {
return computer->maxEnergy;
}
void nn_removeEnergy(nn_computer *computer, double energy) {
if(computer->energy < energy) {
// blackout
computer->energy = 0;
computer->state = NN_STATE_BLACKOUT;
return;
}
computer->energy -= energy;
}
void nn_addEnergy(nn_computer *computer, double amount) {
if(computer->maxEnergy - computer->energy < amount) {
computer->energy = computer->maxEnergy;
return;
}
computer->energy += amount;
}
double nn_getTemperature(nn_computer *computer) {
return computer->temperature;
}
double nn_getThermalCoefficient(nn_computer *computer) {
return computer->temperatureCoefficient;
}
double nn_getRoomTemperature(nn_computer *computer) {
return computer->roomTemperature;
}
void nn_setTemperature(nn_computer *computer, double temperature) {
computer->temperature = temperature;
if(computer->temperature < computer->roomTemperature) computer->temperature = computer->roomTemperature;
}
void nn_setTemperatureCoefficient(nn_computer *computer, double coefficient) {
computer->temperatureCoefficient = coefficient;
}
void nn_setRoomTemperature(nn_computer *computer, double roomTemperature) {
computer->roomTemperature = roomTemperature;
if(computer->temperature < computer->roomTemperature) computer->temperature = computer->roomTemperature;
}
void nn_addHeat(nn_computer *computer, double heat) {
computer->temperature += heat * computer->temperatureCoefficient;
if(computer->temperature < computer->roomTemperature) computer->temperature = computer->roomTemperature;
}
void nn_removeHeat(nn_computer *computer, double heat) {
computer->temperature -= heat;
if(computer->temperature < computer->roomTemperature) computer->temperature = computer->roomTemperature;
}
nn_bool_t nn_isOverheating(nn_computer *computer) {
return computer->temperature > NN_OVERHEAT_MIN;
}
const char *nn_getError(nn_computer *computer) {
return computer->err;
}
void nn_clearError(nn_computer *computer) {
if(computer->allocatedError) {
nn_deallocStr(&computer->universe->ctx.allocator, computer->err);
}
computer->err = NULL;
computer->allocatedError = false;
}
void nn_setError(nn_computer *computer, const char *err) {
nn_clearError(computer);
char *copy = nn_strdup(&computer->universe->ctx.allocator, err);
if(copy == NULL) {
nn_setCError(computer, "out of memory");
return;
}
computer->err = copy;
computer->allocatedError = true;
}
void nn_setCError(nn_computer *computer, const char *err) {
nn_clearError(computer);
// we pinky promise this is safe
computer->err = (char *)err;
computer->allocatedError = false;
}
nn_component *nn_newComponent(nn_computer *computer, nn_address address, int slot, nn_componentTable *table, void *userdata) {
nn_component *c = NULL;
for(nn_size_t i = 0; i < computer->componentLen; i++) {
if(computer->components[i].address == NULL) {
c = computer->components + i;
break;
}
}
if(c == NULL) {
if(computer->componentLen == computer->componentCap) return NULL; // too many
c = computer->components + computer->componentLen;
computer->componentLen++;
}
if(address == NULL) {
c->address = nn_randomUUID(&computer->universe->ctx);
} else {
c->address = nn_strdup(&computer->universe->ctx.allocator, address);
}
if(c->address == NULL) return NULL;
c->table = table;
c->slot = slot;
c->computer = computer;
if(table->constructor == NULL) {
c->statePtr = userdata;
} else {
c->statePtr = table->constructor(table->userdata, userdata);
}
return c;
}
void nn_removeComponent(nn_computer *computer, nn_address address) {
for(nn_size_t i = 0; i < computer->componentLen; i++) {
if(nn_strcmp(computer->components[i].address, address) == 0) {
nn_destroyComponent(computer->components + i);
}
}
}
void nn_destroyComponent(nn_component *component) {
nn_deallocStr(&component->computer->universe->ctx.allocator, component->address);
if(component->table->destructor != NULL) {
component->table->destructor(component->table->userdata, component, component->statePtr);
}
component->address = NULL; // marks component as freed
}
nn_component *nn_findComponent(nn_computer *computer, nn_address address) {
for(nn_size_t i = 0; i < computer->componentLen; i++) {
if(computer->components[i].address == NULL) continue; // empty slot
if(nn_strcmp(computer->components[i].address, address) == 0) {
return computer->components + i;
}
}
return NULL;
}
nn_component *nn_iterComponent(nn_computer *computer, nn_size_t *internalIndex) {
for(nn_size_t i = *internalIndex; i < computer->componentLen; i++) {
if(computer->components[i].address == NULL) continue;
*internalIndex = i+1;
return computer->components + i;
}
return NULL;
}
void nn_resetCall(nn_computer *computer) {
for(nn_size_t i = 0; i < computer->argc; i++) {
nn_values_drop(computer->args[i]);
}
for(nn_size_t i = 0; i < computer->retc; i++) {
nn_values_drop(computer->rets[i]);
}
computer->argc = 0;
computer->retc = 0;
}
void nn_addArgument(nn_computer *computer, nn_value arg) {
if(computer->argc == NN_MAX_ARGS) return;
computer->args[computer->argc] = arg;
computer->argc++;
}
void nn_return(nn_computer *computer, nn_value val) {
if(computer->retc == NN_MAX_RETS) return;
computer->rets[computer->retc] = val;
computer->retc++;
}
nn_value nn_getArgument(nn_computer *computer, nn_size_t idx) {
if(idx >= computer->argc) return nn_values_nil();
return computer->args[idx];
}
nn_value nn_getReturn(nn_computer *computer, nn_size_t idx) {
if(idx >= computer->retc) return nn_values_nil();
return computer->rets[idx];
}
nn_size_t nn_getArgumentCount(nn_computer *computer) {
return computer->argc;
}
nn_size_t nn_getReturnCount(nn_computer *computer) {
return computer->retc;
}
char *nn_serializeProgram(nn_computer *computer, nn_Alloc *alloc, nn_size_t *len) {
return computer->arch->serialize(computer, alloc, computer->archState, computer->arch->userdata, len);
}
void nn_deserializeProgram(nn_computer *computer, const char *memory, nn_size_t len) {
computer->arch->deserialize(computer, memory, len, computer->archState, computer->arch->userdata);
}
nn_Context *nn_getComputerContext(nn_computer *computer) {
return &computer->universe->ctx;
}
nn_guard *nn_getComputerLock(nn_computer *computer) {
return computer->lock;
}
void nn_return_nil(nn_computer *computer) {
nn_return(computer, nn_values_nil());
}
void nn_return_integer(nn_computer *computer, nn_intptr_t integer) {
nn_return(computer, nn_values_integer(integer));
}
void nn_return_number(nn_computer *computer, double number) {
nn_return(computer, nn_values_number(number));
}
void nn_return_boolean(nn_computer *computer, nn_bool_t boolean) {
nn_return(computer, nn_values_boolean(boolean));
}
void nn_return_cstring(nn_computer *computer, const char *cstr) {
nn_return(computer, nn_values_cstring(cstr));
}
void nn_return_string(nn_computer *computer, const char *str, nn_size_t len) {
nn_value val = nn_values_string(&computer->universe->ctx.allocator, str, len);
if(val.tag == NN_VALUE_NIL) {
nn_setCError(computer, "out of memory");
}
nn_return(computer, val);
}
nn_value nn_return_array(nn_computer *computer, nn_size_t len) {
nn_value val = nn_values_array(&computer->universe->ctx.allocator, len);
if(val.tag == NN_VALUE_NIL) {
nn_setCError(computer, "out of memory");
}
nn_return(computer, val);
return val;
}
nn_value nn_return_table(nn_computer *computer, nn_size_t len) {
nn_value val = nn_values_table(&computer->universe->ctx.allocator, len);
if(val.tag == NN_VALUE_NIL) {
nn_setCError(computer, "out of memory");
}
nn_return(computer, val);
return val;
}
void nn_return_resource(nn_computer *computer, nn_size_t userdata) {
nn_return(computer, nn_values_resource(userdata));
}
nn_bool_t nn_wakeupMatches(nn_value *values, nn_size_t valueLen, const char *wakeUp, nn_bool_t fuzzy) {
if(valueLen == 0) return false;
nn_value header = values[0];
const char *headerStr = nn_toCString(header);
if(fuzzy) {
return nn_strbegin(headerStr, wakeUp);
} else {
return nn_strcmp(headerStr, wakeUp) == 0;
}
}
const char *nn_pushNetworkMessage(nn_computer *computer, nn_address receiver, nn_address sender, nn_size_t port, double distance, nn_value *values, nn_size_t valueLen) {
nn_Alloc *alloc = &computer->universe->ctx.allocator;
nn_value buffer[valueLen + 5];
buffer[0] = nn_values_cstring("modem_message");
buffer[1] = nn_values_string(alloc, receiver, nn_strlen(receiver));
buffer[2] = nn_values_string(alloc, sender, nn_strlen(sender));
buffer[3] = nn_values_integer(port);
buffer[4] = nn_values_number(distance);
for(nn_size_t i = 0; i < valueLen; i++) {
buffer[i + 5] = nn_values_retain(values[i]);
}
return nn_pushSignal(computer, buffer, valueLen + 5);
}
static nn_resource_t *nn_resource_find(nn_computer *computer, nn_size_t id) {
for(nn_size_t i = 0; i < NN_MAX_CONCURRENT_RESOURCES; i++) {
if(computer->resources[i].id == id) {
return computer->resources + i;
}
}
return NULL;
}
nn_size_t nn_resource_allocate(nn_computer *computer, void *userdata, nn_resourceTable_t *table) {
nn_size_t i = 0;
for(nn_size_t j = 0; j < NN_MAX_CONCURRENT_RESOURCES; j++) {
if(computer->resources[j].id == NN_NULL_RESOURCE) {
i = j;
goto slotFound;
}
}
return NN_NULL_RESOURCE;
slotFound:
computer->rid++;
nn_size_t rid = computer->rid;
computer->resources[i] = (nn_resource_t) {
.id = rid,
.ptr = userdata,
.table = table,
};
return rid;
}
void nn_resource_release(nn_computer *computer, nn_size_t id) {
nn_resource_t *res = nn_resource_find(computer, id);
if(res == NULL) return;
res->id = NN_NULL_RESOURCE;
if(res->table->dtor != NULL) {
res->table->dtor(res->ptr);
}
}
nn_resourceTable_t *nn_resource_fetchTable(nn_computer *computer, nn_size_t resourceID) {
nn_resource_t *res = nn_resource_find(computer, resourceID);
if(res == NULL) return NULL;
return res->table;
}
nn_bool_t nn_resource_invoke(nn_computer *computer, nn_size_t resourceID, const char *method) {
nn_resource_t *res = nn_resource_find(computer, resourceID);
if(res == NULL) return false;
nn_resourceTable_t *t = res->table;
for(nn_size_t i = 0; i < t->methodCount; i++) {
nn_resourceMethod_t m = t->methods[i];
if(nn_strcmp(m.name, method) != 0) continue;
if(m.condition != NULL) {
if(!m.condition(res->ptr, m.userdata)) continue;
}
m.callback(res->ptr, m.userdata, computer);
return true;
}
return false;
}
// returns the name, and NULL for out of bounds
const char *nn_resource_nextMethodInfo(nn_computer *computer, nn_size_t id, const char **doc, nn_size_t *idx) {
nn_resource_t *res = nn_resource_find(computer, id);
if(res == NULL) return false;
nn_resourceTable_t *t = res->table;
for(nn_size_t i = *idx; i < t->methodCount; i++) {
nn_resourceMethod_t method = t->methods[i];
if(method.condition != NULL) {
if(!method.condition(res->ptr, method.userdata)) {
continue;
}
}
*idx = i + 1;
*doc = method.doc;
return method.name;
}
return NULL;
}