Add taxonomy helper functions

Right now we just want direct access to the country, I assume
we'll add more like this in the future.

Signed-off-by: Dirk Hohndel <dirk@hohndel.org>
This commit is contained in:
Dirk Hohndel 2017-10-02 22:54:24 -07:00
parent a163bbc442
commit 9fd6b3eefb
2 changed files with 39 additions and 1 deletions

View file

@ -2,6 +2,7 @@
#include "taxonomy.h" #include "taxonomy.h"
#include "gettext.h" #include "gettext.h"
#include <stdlib.h> #include <stdlib.h>
#include <stdio.h>
char *taxonomy_category_names[TC_NR_CATEGORIES] = { char *taxonomy_category_names[TC_NR_CATEGORIES] = {
QT_TRANSLATE_NOOP("gettextFromC", "None"), QT_TRANSLATE_NOOP("gettextFromC", "None"),
@ -47,3 +48,31 @@ int taxonomy_index_for_category(struct taxonomy_data *t, enum taxonomy_category
return i; return i;
return -1; 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_country(struct taxonomy_data *t, const char *country, enum taxonomy_origin origin)
{
int idx = -1;
for (int i = 0; i < t->nr; i++)
if (t->category[i].category == TC_COUNTRY) {
idx = i;
break;
}
if (idx == -1) {
if (t->nr == TC_NR_CATEGORIES - 1) {
// can't add another one
fprintf(stderr, "Error adding country taxonomy\n");
return;
}
idx = ++t->nr;
}
t->category[idx].value = country;
t->category[idx].origin = origin;
}

View file

@ -17,13 +17,20 @@ enum taxonomy_category {
TC_NR_CATEGORIES TC_NR_CATEGORIES
}; };
enum taxonomy_origin {
GEOCODED,
GEOPARSED,
GEOMANUAL,
GEOCOPIED
};
extern char *taxonomy_category_names[TC_NR_CATEGORIES]; extern char *taxonomy_category_names[TC_NR_CATEGORIES];
extern char *taxonomy_api_names[TC_NR_CATEGORIES]; extern char *taxonomy_api_names[TC_NR_CATEGORIES];
struct taxonomy { struct taxonomy {
int category; /* the category for this tag: ocean, country, admin_l1, admin_l2, localname, etc */ int category; /* the category for this tag: ocean, country, admin_l1, admin_l2, localname, etc */
const char *value; /* the value returned, parsed, or manually entered for that category */ const char *value; /* the value returned, parsed, or manually entered for that category */
enum { GEOCODED, PARSED, MANUAL, COPIED } origin; enum taxonomy_origin origin;
}; };
/* the data block contains 3 taxonomy structures - unused ones have a tag value of NONE */ /* the data block contains 3 taxonomy structures - unused ones have a tag value of NONE */
@ -35,6 +42,8 @@ struct taxonomy_data {
struct taxonomy *alloc_taxonomy(); struct taxonomy *alloc_taxonomy();
void free_taxonomy(struct taxonomy_data *t); void free_taxonomy(struct taxonomy_data *t);
int taxonomy_index_for_category(struct taxonomy_data *t, enum taxonomy_category cat); int taxonomy_index_for_category(struct taxonomy_data *t, enum taxonomy_category cat);
const char *taxonomy_get_country(struct taxonomy_data *t);
void taxonomy_set_country(struct taxonomy_data *t, const char *country, enum taxonomy_origin origin);
#ifdef __cplusplus #ifdef __cplusplus
} }