diff options
Diffstat (limited to 'src/util.c')
| -rw-r--r-- | src/util.c | 80 |
1 files changed, 80 insertions, 0 deletions
@@ -5,6 +5,9 @@ #include <getopt.h> #include "util.h" +#define BUFFER_SIZE 1024 + +static char ** config_read_values(size_t *n_out_keys, char *first_value, char **strtok_ptr); void die(const char *fmt, ...) { @@ -33,6 +36,13 @@ void * ecalloc(size_t nmemb, size_t size) return p; } +char *e_strdup(const char *s) +{ + char *out = strdup(s); + if (out == NULL) die("strdup() Error:"); + return out; +} + void version() { @@ -121,3 +131,73 @@ void util_free_config(struct Configs *ml) free(ml->label_keys); } } + +void util_load_config(struct Configs *ml, char *filepath) +{ + int line_number = 1; + char line_buffer[BUFFER_SIZE], line_buffer_original[BUFFER_SIZE]; + char token_buffer[BUFFER_SIZE]; + FILE *fp = fopen(filepath, "r"); + if (fp == NULL) die("util_load_config() Error:"); + + while (fgets(line_buffer, BUFFER_SIZE, fp)) { + int ret = sscanf(line_buffer, "[%[-_a-zA-Z0-9]]", token_buffer); + if (ret >= 1) continue; + + strcpy(line_buffer_original, line_buffer); + + char *line_ptr = line_buffer; + while (*line_ptr == ' ') line_ptr++; // delete whitespaces + + /* if the line start with comments or is a blank line ignore it */ + if (*line_ptr == ';' + || *line_ptr == '#' + || *line_ptr == '\n') continue; + + /* Verify that each line starts with [a-zA-Z] */ + if ((*line_ptr < 0x41 && *line_ptr > 0x5A) + || (*line_ptr < 0x61 && *line_ptr > 0x7A)) + goto util_load_config_error; + + /* Load Key Value */ + char *key, *value; + char *ptr_buffer; + strtok_r(line_buffer, ";#", &ptr_buffer); // omit comments + key = strtok_r(line_buffer, " =", &ptr_buffer); + value = strtok_r(NULL, "= ,\n", &ptr_buffer); + if (value == NULL) goto util_load_config_error; + + if (!strcmp(key, "weights_path")) ml->weights_filepath = e_strdup(value); + else if (!strcmp(key, "loss")) ml->loss = e_strdup(value); + else if (!strcmp(key, "epochs")) ml->epochs = (size_t)atol(value); + else if (!strcmp(key, "alpha")) ml->alpha = (double)atof(value); + else if (!strcmp(key, "inputs")) ml->input_keys = config_read_values(&(ml->n_input_keys), value, &ptr_buffer); + else if (!strcmp(key, "labels")) ml->label_keys = config_read_values(&(ml->n_label_keys), value, &ptr_buffer); + else die("util_load_config() Error: Unknown parameter '%s' on line '%d'", key, line_number); + ptr_buffer = NULL; + value = NULL; + + } + fclose(fp); + return; + +util_load_config_error: + die("util_load_config() Error: line %d has invalid format '%s'", + line_number, line_buffer_original); +} + +char ** config_read_values(size_t *n_out_keys, char *first_value, char **strtok_ptr) +{ + *n_out_keys = 1; + char **out_keys = ecalloc(1, sizeof(char *)); + out_keys[0] = e_strdup(first_value); + + char *value; + while ((value = strtok_r(NULL, ", \n", strtok_ptr)) != NULL) { + out_keys = realloc(out_keys, sizeof(char *) * (*n_out_keys + 1)); + out_keys[*n_out_keys] = e_strdup(value); + (*n_out_keys)++; + } + return out_keys; +} +#undef BUFFER_SIZE |