tommy/libpostal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
722904ce5973fa5cd4ff3ec606fb744db8a41f5a
libpostal/src/geonames.c

568 lines
16 KiB
C
Raw Blame History

#include "geonames.h"
#define DEFAULT_BUFFER_SIZE 4096
#define GEONAMES_NAME_DEFAULT_LENGTH 255
#define GEONAMES_ISO_LANGUAGE_DEFAULT_LENGTH 10
#define GEONAMES_FEATURE_CODE_DEFAULT_LENGTH 5
#define GEONAMES_COUNTRY_CODE_DEFAULT_LENGTH 3
#define GEONAMES_ADMIN1_CODE_DEFAULT_LENGTH 32
#define GEONAMES_ADMIN2_CODE_DEFAULT_LENGTH 64
#define GEONAMES_ADMIN3_CODE_DEFAULT_LENGTH 32
#define GEONAMES_ADMIN4_CODE_DEFAULT_LENGTH 32
#define GEONAMES_POSTAL_CODE_DEFAULT_LENGTH 64
#define GEONAMES_POSTAL_ADMIN1_IDS_DEFAULT_LENGTH 10
#define GEONAMES_POSTAL_ADMIN2_IDS_DEFAULT_LENGTH 20
#define GEONAMES_POSTAL_ADMIN3_IDS_DEFAULT_LENGTH 20
/* To save on malloc calls, create just one of these and call geoname_clear
* (which does not deallocate) and geoname_deserialize
*/
geoname_t *geoname_new(void) {
geoname_t *self = malloc(sizeof(geoname_t));
self->name = char_array_new_size(GEONAMES_NAME_DEFAULT_LENGTH);
self->canonical = char_array_new_size(GEONAMES_NAME_DEFAULT_LENGTH);
self->iso_language = char_array_new_size(GEONAMES_ISO_LANGUAGE_DEFAULT_LENGTH);
self->feature_code = char_array_new_size(GEONAMES_FEATURE_CODE_DEFAULT_LENGTH);
self->country_code = char_array_new_size(GEONAMES_COUNTRY_CODE_DEFAULT_LENGTH);
self->admin1_code = char_array_new_size(GEONAMES_ADMIN1_CODE_DEFAULT_LENGTH);
self->admin2_code = char_array_new_size(GEONAMES_ADMIN2_CODE_DEFAULT_LENGTH);
self->admin3_code = char_array_new_size(GEONAMES_ADMIN3_CODE_DEFAULT_LENGTH);
self->admin4_code = char_array_new_size(GEONAMES_ADMIN4_CODE_DEFAULT_LENGTH);
return self;
}
void geoname_clear(geoname_t *self) {
char_array_clear(self->name);
char_array_clear(self->canonical);
char_array_clear(self->iso_language);
char_array_clear(self->feature_code);
char_array_clear(self->country_code);
char_array_clear(self->admin1_code);
char_array_clear(self->admin2_code);
char_array_clear(self->admin3_code);
char_array_clear(self->admin4_code);
}
void geoname_destroy(geoname_t *self) {
if (!self)
return;
if (self->name) {
char_array_destroy(self->name);
}
if (self->canonical) {
char_array_destroy(self->canonical);
}
if (self->iso_language) {
char_array_destroy(self->iso_language);
}
if (self->feature_code) {
char_array_destroy(self->feature_code);
}
if (self->country_code) {
char_array_destroy(self->country_code);
}
if (self->admin1_code) {
char_array_destroy(self->admin1_code);
}
if (self->admin2_code) {
char_array_destroy(self->admin2_code);
}
if (self->admin3_code) {
char_array_destroy(self->admin3_code);
}
if (self->admin4_code) {
char_array_destroy(self->admin4_code);
}
free(self);
}
static char_array *read_string(char_array *str, size_t len, buffer_t *buffer) {
char_array_clear(str);
char_array_cat_len(str, buffer->data->a + buffer->offset, len);
buffer->offset += len;
return str;
}
static bool bytes_reader(cmp_ctx_t *ctx, void *data, size_t size) {
buffer_t *buffer = ctx->buf;
if (buffer->offset + size > char_array_len(buffer->data)) {
return false;
}
memcpy(data, buffer->data->a + buffer->offset, size);
buffer->offset += size;
return true;
}
static size_t bytes_writer(cmp_ctx_t *ctx, const void *data, size_t count) {
buffer_t *buffer = (buffer_t *)ctx->buf;
char_array_cat_len(buffer->data, (char *)data, count);
buffer->offset += count;
return count;
}
bool cmp_read_str_size_or_nil(cmp_ctx_t *ctx, char_array **str, uint32_t *size) {
cmp_object_t obj;
if (!cmp_read_object(ctx, &obj)) {
return false;
}
if (cmp_object_is_str(&obj)) {
*size = obj.as.str_size;
*str = read_string(*str, *size, ctx->buf);
} else if (cmp_object_is_nil(&obj)) {
*size = 0;
char_array_clear(*str);
}
return true;
}
bool cmp_write_str_or_nil(cmp_ctx_t *ctx, char_array *str) {
if (str != NULL && char_array_len(str) > 0) {
return cmp_write_str(ctx, str->a, char_array_len(str));
} else {
return cmp_write_nil(ctx);
}
}
bool cmp_write_uint_vector(cmp_ctx_t *ctx, uint32_array *array) {
size_t n = array->n;
if (!cmp_write_array(ctx, n)) {
return false;
}
for (size_t i = 0; i < n; i++) {
if (!cmp_write_uint(ctx, array->a[i])) {
return false;
}
}
return true;
}
bool geoname_deserialize_ctx(geoname_t *self, cmp_ctx_t *ctx) {
uint32_t len;
if (!cmp_read_uint(ctx, &self->geonames_id)) {
return false;
}
if (!cmp_read_str_size_or_nil(ctx, &self->name, &len)) {
return false;
}
if (!cmp_read_str_size_or_nil(ctx, &self->canonical, &len)) {
return false;
}
if (!cmp_read_uint(ctx, &self->type)) {
return false;
}
if (!cmp_read_str_size_or_nil(ctx, &self->iso_language, &len)) {
return false;
}
if (!cmp_read_bool(ctx, &self->has_wikipedia_entry)) {
return false;
}
if (!cmp_read_bool(ctx, &self->is_preferred_name)) {
return false;
}
if (!cmp_read_bool(ctx, &self->is_short_name)) {
return false;
}
if (!cmp_read_bool(ctx, &self->is_colloquial)) {
return false;
}
if (!cmp_read_bool(ctx, &self->is_historical)) {
return false;
}
if (!cmp_read_uint(ctx, &self->population)) {
return false;
}
if (!cmp_read_double(ctx, &self->latitude)) {
return false;
}
if (!cmp_read_double(ctx, &self->longitude)) {
return false;
}
if (!cmp_read_str_size_or_nil(ctx, &self->feature_code, &len)) {
return false;
}
if (!cmp_read_str_size_or_nil(ctx, &self->country_code, &len)) {
return false;
}
if (!cmp_read_uint(ctx, &self->country_geonames_id)) {
return false;
}
if (!cmp_read_str_size_or_nil(ctx, &self->admin1_code, &len)) {
return false;
}
if (!cmp_read_uint(ctx, &self->admin1_geonames_id)) {
return false;
}
if (!cmp_read_str_size_or_nil(ctx, &self->admin2_code, &len)) {
return false;
}
if (!cmp_read_uint(ctx, &self->admin2_geonames_id)) {
return false;
}
if (!cmp_read_str_size_or_nil(ctx, &self->admin3_code, &len)) {
return false;
}
if (!cmp_read_uint(ctx, &self->admin3_geonames_id)) {
return false;
}
if (!cmp_read_str_size_or_nil(ctx, &self->admin4_code, &len)) {
return false;
}
if (!cmp_read_uint(ctx, &self->admin4_geonames_id)) {
return false;
}
return true;
}
bool geoname_deserialize(geoname_t *self, char_array *str) {
cmp_ctx_t ctx;
buffer_t buffer = (buffer_t){str, 0};
cmp_init(&ctx, &buffer, bytes_reader, bytes_writer);
return geoname_deserialize_ctx(self, &ctx);
}
bool geoname_serialize_ctx(geoname_t *self, cmp_ctx_t *ctx) {
if (!cmp_write_uint(ctx, self->geonames_id) ||
!cmp_write_str_or_nil(ctx, self->name) ||
!cmp_write_str_or_nil(ctx, self->canonical) ||
!cmp_write_uint(ctx, self->type) ||
!cmp_write_str_or_nil(ctx, self->iso_language) ||
!cmp_write_bool(ctx, self->has_wikipedia_entry) ||
!cmp_write_bool(ctx, self->is_preferred_name) ||
!cmp_write_bool(ctx, self->is_short_name) ||
!cmp_write_bool(ctx, self->is_colloquial) ||
!cmp_write_bool(ctx, self->is_historical) ||
!cmp_write_uint(ctx, self->population) ||
!cmp_write_double(ctx, self->latitude) ||
!cmp_write_double(ctx, self->longitude) ||
!cmp_write_str_or_nil(ctx, self->feature_code) ||
!cmp_write_str_or_nil(ctx, self->country_code) ||
!cmp_write_uint(ctx, self->country_geonames_id) ||
!cmp_write_str_or_nil(ctx, self->admin1_code) ||
!cmp_write_uint(ctx, self->admin1_geonames_id) ||
!cmp_write_str_or_nil(ctx, self->admin2_code) ||
!cmp_write_uint(ctx, self->admin2_geonames_id) ||
!cmp_write_str_or_nil(ctx, self->admin3_code) ||
!cmp_write_uint(ctx, self->admin3_geonames_id) ||
!cmp_write_str_or_nil(ctx, self->admin4_code) ||
!cmp_write_uint(ctx, self->admin4_geonames_id)
) { return false; }
return true;
}
bool geoname_serialize(geoname_t *self, char_array *str) {
cmp_ctx_t ctx;
buffer_t buffer = (buffer_t){str, 0};
cmp_init(&ctx, &buffer, bytes_reader, bytes_writer);
return geoname_serialize_ctx(self, &ctx);
}
void geoname_print(geoname_t *self) {
printf("geonames_id: %d\n", self->geonames_id);
printf("name: %s\n", char_array_get_string(self->name));
printf("canonical: %s\n", char_array_get_string(self->canonical));
printf("type: %d\n", self->type);
printf("iso_language: %s\n", char_array_get_string(self->iso_language));
printf("has_wikipedia_entry: %d\n", self->has_wikipedia_entry);
printf("is_preferred: %d\n", self->is_preferred_name);
printf("is_short_name: %d\n", self->is_short_name);
printf("is_colloquial: %d\n", self->is_colloquial);
printf("is_historical: %d\n", self->is_historical);
printf("population: %d\n", self->population);
printf("latitude: %f\n", self->latitude);
printf("longitude: %f\n", self->longitude);
printf("feature_code: %s\n", char_array_get_string(self->feature_code));
printf("country_code: %s\n", char_array_get_string(self->country_code));
printf("country_geonames_id: %d\n", self->country_geonames_id);
printf("admin1_code: %s\n", char_array_get_string(self->admin1_code));
printf("admin1_geonames_id: %d\n", self->admin1_geonames_id);
printf("admin2_code: %s\n", char_array_get_string(self->admin2_code));
printf("admin2_geonames_id: %d\n", self->admin2_geonames_id);
printf("admin3_code: %s\n", char_array_get_string(self->admin3_code));
printf("admin3_geonames_id: %d\n", self->admin3_geonames_id);
printf("admin4_code: %s\n", char_array_get_string(self->admin4_code));
printf("admin4_geonames_id: %d\n", self->admin4_geonames_id);
}
gn_postal_code_t *gn_postal_code_new(void) {
gn_postal_code_t *self = malloc(sizeof(gn_postal_code_t));
self->postal_code = char_array_new_size(GEONAMES_POSTAL_CODE_DEFAULT_LENGTH);
self->country_code = char_array_new_size(GEONAMES_COUNTRY_CODE_DEFAULT_LENGTH);
self->containing_geoname = char_array_new_size(GEONAMES_NAME_DEFAULT_LENGTH);
self->admin1_ids = uint32_array_new_size(GEONAMES_POSTAL_ADMIN1_IDS_DEFAULT_LENGTH);
self->admin2_ids = uint32_array_new_size(GEONAMES_POSTAL_ADMIN2_IDS_DEFAULT_LENGTH);
self->admin3_ids = uint32_array_new_size(GEONAMES_POSTAL_ADMIN3_IDS_DEFAULT_LENGTH);
return self;
}
void gn_postal_code_clear(gn_postal_code_t *self) {
char_array_clear(self->postal_code);
char_array_clear(self->country_code);
char_array_clear(self->containing_geoname);
uint32_array_clear(self->admin1_ids);
uint32_array_clear(self->admin2_ids);
uint32_array_clear(self->admin3_ids);
}
void gn_postal_code_destroy(gn_postal_code_t *self) {
if (!self)
return;
if (self->postal_code) {
char_array_destroy(self->postal_code);
}
if (self->country_code) {
char_array_destroy(self->country_code);
}
if (self->containing_geoname) {
char_array_destroy(self->containing_geoname);
}
if (self->admin1_ids) {
uint32_array_destroy(self->admin1_ids);
}
if (self->admin2_ids) {
uint32_array_destroy(self->admin2_ids);
}
if (self->admin3_ids) {
uint32_array_destroy(self->admin3_ids);
}
free(self);
}
bool gn_postal_code_deserialize_ctx(gn_postal_code_t *self, cmp_ctx_t *ctx) {
uint32_t len;
if (!cmp_read_str_size_or_nil(ctx, &self->postal_code, &len)) {
return false;
}
if (!cmp_read_str_size_or_nil(ctx, &self->country_code, &len)) {
return false;
}
if (!cmp_read_uint(ctx, &self->country_geonames_id)) {
return false;
}
if (!cmp_read_bool(ctx, &self->have_containing_geoname)) {
return false;
}
if (self->have_containing_geoname) {
if (!cmp_read_str_size_or_nil(ctx, &self->containing_geoname, &len))
return false;
if (!cmp_read_uint(ctx, &self->containing_geonames_id))
return false;
}
uint32_t array_size;
if (!cmp_read_array(ctx, &array_size)) {
return false;
}
if (array_size > 0) {
uint32_t admin1_id;
for ( ;array_size; array_size--) {
if (!cmp_read_uint(ctx, &admin1_id)) {
return false;
}
uint32_array_push(self->admin1_ids, admin1_id);
}
}
if (!cmp_read_array(ctx, &array_size)) {
return false;
}
if (array_size > 0) {
uint32_t admin2_id;
for (; array_size > 0; array_size--) {
if (!cmp_read_uint(ctx, &admin2_id)) {
return false;
}
uint32_array_push(self->admin2_ids, admin2_id);
}
}
if (!cmp_read_array(ctx, &array_size)) {
return false;
}
if (array_size > 0) {
uint32_t admin3_id;
for (; array_size > 0; array_size--) {
if (!cmp_read_uint(ctx, &admin3_id)) {
return false;
}
uint32_array_push(self->admin3_ids, admin3_id);
}
}
return true;
}
bool gn_postal_code_deserialize(gn_postal_code_t *self, char_array *str) {
cmp_ctx_t ctx;
buffer_t buffer = (buffer_t){str, 0};
cmp_init(&ctx, &buffer, bytes_reader, bytes_writer);
return gn_postal_code_deserialize_ctx(self, &ctx);
}
static bool gn_postal_code_serialize_ctx(gn_postal_code_t *self, cmp_ctx_t *ctx) {
if (!cmp_write_str_or_nil(ctx, self->postal_code) ||
!cmp_write_str_or_nil(ctx, self->country_code) ||
!cmp_write_uint(ctx, self->country_geonames_id) ||
!cmp_write_bool(ctx, self->have_containing_geoname) ||
(self->have_containing_geoname &&
(!cmp_write_str_or_nil(ctx, self->containing_geoname) ||
!cmp_write_uint(ctx, self->containing_geonames_id)
)
) ||
!cmp_write_uint_vector(ctx, self->admin1_ids) ||
!cmp_write_uint_vector(ctx, self->admin2_ids) ||
!cmp_write_uint_vector(ctx, self->admin3_ids)
) { return false; }
return true;
}
bool gn_postal_code_serialize(gn_postal_code_t *self, char_array *str) {
cmp_ctx_t ctx;
buffer_t buffer = (buffer_t){str, 0};
cmp_init(&ctx, &buffer, bytes_reader, bytes_writer);
return gn_postal_code_serialize_ctx(self, &ctx);
}
void gn_postal_code_print(gn_postal_code_t *self) {
int i;
printf("postal_code: %s\n", char_array_get_string(self->postal_code));
printf("country_code: %s\n", char_array_get_string(self->country_code));
printf("have_containing_geoname: %d\n", self->have_containing_geoname);
if (self->have_containing_geoname) {
printf("containing_geoname: %s\n", char_array_get_string(self->containing_geoname));
printf("containing_geonames_id: %d\n", self->containing_geonames_id);
}
printf("admin1_ids: [ ");
for (i = 0; i < self->admin1_ids->n; i++) {
printf("%d ", self->admin1_ids->a[i]);
}
printf("]\n");
printf("admin2_ids: [ ");
for (i = 0; i < self->admin2_ids->n; i++) {
printf("%d ", self->admin2_ids->a[i]);
}
printf("]\n");
printf("admin3_ids: [ ");
for (i = 0; i < self->admin3_ids->n; i++) {
printf("%d ", self->admin3_ids->a[i]);
}
printf("]\n");
}
bool geonames_generic_serialize(geonames_generic_t *gn, char_array *str) {
cmp_ctx_t ctx;
buffer_t buffer = (buffer_t){str, 0};
cmp_init(&ctx, &buffer, bytes_reader, bytes_writer);
if (!cmp_write_u8(&ctx, (uint8_t)gn->type)) {
return false;
}
if (gn->type == GEONAMES_PLACE) {
return geoname_serialize_ctx(gn->geoname, &ctx);
} else if (gn->type == GEONAMES_POSTAL_CODE) {
return gn_postal_code_serialize_ctx(gn->postal_code, &ctx);
}
return false;
}
bool geonames_generic_deserialize(gn_type *type, geoname_t *geoname, gn_postal_code_t *postal_code, char_array *str) {
cmp_ctx_t ctx;
buffer_t buffer = (buffer_t){str, 0};
cmp_init(&ctx, &buffer, bytes_reader, bytes_writer);
uint8_t geonames_type;
if (!cmp_read_u8(&ctx, &geonames_type)) {
return false;
}
*type = geonames_type;
if (geonames_type == GEONAMES_PLACE) {
return geoname_deserialize_ctx(geoname, &ctx);
} else if (geonames_type == GEONAMES_POSTAL_CODE) {
return gn_postal_code_deserialize_ctx(postal_code, &ctx);
}
return false;
}
Reference in New Issue View Git Blame Copy Permalink