subsurface/core/taxonomy.c
Berthold Stoeger 754b4a5c9d cleanup: use taxonomy_index_for_category() in taxonomy_set_category()
Instead of recoding the "search for category" loop, reuse the already
existing functionality.

Signed-off-by: Berthold Stoeger <bstoeger@mail.tuwien.ac.at>
2020-09-06 12:59:54 -07:00

109 lines
2.7 KiB
C

// SPDX-License-Identifier: GPL-2.0
#include "taxonomy.h"
#include "gettext.h"
#include "subsurface-string.h"
#include <stdlib.h>
#include <stdio.h>
char *taxonomy_category_names[TC_NR_CATEGORIES] = {
QT_TRANSLATE_NOOP("gettextFromC", "None"),
QT_TRANSLATE_NOOP("gettextFromC", "Ocean"),
QT_TRANSLATE_NOOP("gettextFromC", "Country"),
QT_TRANSLATE_NOOP("gettextFromC", "State"),
QT_TRANSLATE_NOOP("gettextFromC", "County"),
QT_TRANSLATE_NOOP("gettextFromC", "Town"),
QT_TRANSLATE_NOOP("gettextFromC", "City")
};
// these are the names for geoname.org
char *taxonomy_api_names[TC_NR_CATEGORIES] = {
"none",
"name",
"countryName",
"adminName1",
"adminName2",
"toponymName",
"adminName3"
};
static void alloc_taxonomy_table(struct taxonomy_data *t)
{
if (!t->category)
t->category = calloc(TC_NR_CATEGORIES, sizeof(struct taxonomy));
}
void free_taxonomy(struct taxonomy_data *t)
{
if (t) {
for (int i = 0; i < t->nr; i++)
free((void *)t->category[i].value);
free(t->category);
t->category = NULL;
t->nr = 0;
}
}
void copy_taxonomy(const struct taxonomy_data *orig, struct taxonomy_data *copy)
{
if (orig->category == NULL) {
free_taxonomy(copy);
} else {
alloc_taxonomy_table(copy);
for (int i = 0; i < TC_NR_CATEGORIES; i++) {
if (i < copy->nr) {
free((void *)copy->category[i].value);
copy->category[i].value = NULL;
}
if (i < orig->nr) {
copy->category[i] = orig->category[i];
copy->category[i].value = copy_string(orig->category[i].value);
}
}
copy->nr = orig->nr;
}
}
int taxonomy_index_for_category(const struct taxonomy_data *t, enum taxonomy_category cat)
{
for (int i = 0; i < t->nr; i++) {
if (t->category[i].category == cat)
return i;
}
return -1;
}
const char *taxonomy_get_country(struct taxonomy_data *t)
{
for (int i = 0; i < t->nr; i++) {
if (t->category[i].category == TC_COUNTRY)
return t->category[i].value;
}
return NULL;
}
void taxonomy_set_category(struct taxonomy_data *t, enum taxonomy_category category, const char *value, enum taxonomy_origin origin)
{
int idx = taxonomy_index_for_category(t, category);
if (idx < 0) {
alloc_taxonomy_table(t); // make sure we have taxonomy data allocated
if (t->nr == TC_NR_CATEGORIES - 1) {
// can't add another one
fprintf(stderr, "Error adding taxonomy category\n");
return;
}
idx = t->nr++;
} else {
free((void *)t->category[idx].value);
t->category[idx].value = NULL;
}
t->category[idx].value = strdup(value);
t->category[idx].origin = origin;
t->category[idx].category = category;
}
void taxonomy_set_country(struct taxonomy_data *t, const char *country, enum taxonomy_origin origin)
{
fprintf(stderr, "%s: set the taxonomy country to %s\n", __func__, country);
taxonomy_set_category(t, TC_COUNTRY, country, origin);
}