mirror of
https://git.rwth-aachen.de/acs/public/villas/node/
synced 2025-03-16 00:00:02 +01:00
151 lines
3.2 KiB
C
151 lines
3.2 KiB
C
/** Node type: Wrapper around RSCAD CBuilder model
|
|
*
|
|
* @author Steffen Vogel <stvogel@eonerc.rwth-aachen.de>
|
|
* @copyright 2017, Steffen Vogel
|
|
**********************************************************************************/
|
|
|
|
#include "node.h"
|
|
#include "log.h"
|
|
#include "plugin.h"
|
|
#include "utils.h"
|
|
|
|
#include "nodes/cbuilder.h"
|
|
|
|
int cbuilder_parse(struct node *n, json_t *cfg)
|
|
{
|
|
struct cbuilder *cb = n->_vd;
|
|
json_t *cfg_param, *cfg_params = NULL;
|
|
|
|
const char *model;
|
|
|
|
int ret;
|
|
size_t index;
|
|
json_error_t err;
|
|
|
|
ret = json_unpack_ex(cfg, &err, 0, "{ s: f, s: s, s: b }",
|
|
"timestep", &cb->timestep,
|
|
"model", &model,
|
|
"parameters", &cfg_params
|
|
);
|
|
if (ret)
|
|
jerror(&err, "Failed to parse configuration of node %s", node_name(n));
|
|
|
|
cb->model = (struct cbuilder_model *) plugin_lookup(PLUGIN_TYPE_MODEL_CBUILDER, model);
|
|
if (!cb->model)
|
|
error("Unknown model '%s' of node %s", model, node_name(n));
|
|
|
|
if (cfg_params) {
|
|
if (!json_is_array(cfg_params))
|
|
error("Setting 'parameters' of node %s must be an JSON array of numbers!", node_name(n));
|
|
|
|
cb->paramlen = json_array_size(cfg_params);
|
|
cb->params = alloc(cb->paramlen * sizeof(double));
|
|
|
|
json_array_foreach(cfg_params, index, cfg_param) {
|
|
if (json_is_number(cfg_param))
|
|
error("Setting 'parameters' of node %s must be an JSON array of numbers!", node_name(n));
|
|
|
|
cb->params[index] = json_number_value(cfg_params);
|
|
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int cbuilder_start(struct node *n)
|
|
{
|
|
int ret;
|
|
struct cbuilder *cb = n->_vd;
|
|
|
|
/* Initialize mutex and cv */
|
|
pthread_mutex_init(&cb->mtx, NULL);
|
|
pthread_cond_init(&cb->cv, NULL);
|
|
|
|
/* Currently only a single timestep per model / instance is supported */
|
|
cb->step = 0;
|
|
cb->read = 0;
|
|
|
|
ret = cb->model->init(cb);
|
|
if (ret)
|
|
error("Failed to intialize CBuilder model %s", node_name(n));
|
|
|
|
cb->model->ram();
|
|
|
|
return 0;
|
|
}
|
|
|
|
int cbuilder_stop(struct node *n)
|
|
{
|
|
struct cbuilder *cb = n->_vd;
|
|
|
|
pthread_mutex_destroy(&cb->mtx);
|
|
pthread_cond_destroy(&cb->cv);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int cbuilder_read(struct node *n, struct sample *smps[], unsigned cnt)
|
|
{
|
|
struct cbuilder *cb = n->_vd;
|
|
struct sample *smp = smps[0];
|
|
|
|
/* Wait for completion of step */
|
|
pthread_mutex_lock(&cb->mtx);
|
|
while (cb->read >= cb->step)
|
|
pthread_cond_wait(&cb->cv, &cb->mtx);
|
|
|
|
float data[smp->capacity];
|
|
|
|
smp->length = cb->model->read(data, smp->capacity);
|
|
|
|
/* Cast float -> double */
|
|
for (int i = 0; i < smp->length; i++)
|
|
smp->data[i].f = data[i];
|
|
|
|
smp->sequence = cb->step;
|
|
|
|
cb->read = cb->step;
|
|
|
|
pthread_mutex_unlock(&cb->mtx);
|
|
|
|
return 1;
|
|
}
|
|
|
|
int cbuilder_write(struct node *n, struct sample *smps[], unsigned cnt)
|
|
{
|
|
struct cbuilder *cb = n->_vd;
|
|
struct sample *smp = smps[0];
|
|
|
|
pthread_mutex_lock(&cb->mtx);
|
|
|
|
float flt = smp->data[0].f;
|
|
|
|
cb->model->write(&flt, smp->length);
|
|
cb->model->code();
|
|
|
|
cb->step++;
|
|
|
|
pthread_cond_signal(&cb->cv);
|
|
pthread_mutex_unlock(&cb->mtx);
|
|
|
|
return 1;
|
|
}
|
|
|
|
static struct plugin p = {
|
|
.name = "cbuilder",
|
|
.description = "RTDS CBuilder model",
|
|
.type = PLUGIN_TYPE_NODE,
|
|
.node = {
|
|
.vectorize = 1,
|
|
.size = sizeof(struct cbuilder),
|
|
.parse = cbuilder_parse,
|
|
.start = cbuilder_start,
|
|
.stop = cbuilder_stop,
|
|
.read = cbuilder_read,
|
|
.write = cbuilder_write
|
|
}
|
|
};
|
|
|
|
REGISTER_PLUGIN(&p)
|
|
LIST_INIT_STATIC(&p.node.instances)
|