sjs/gh-BertoldVdb-ZoneDetect
1
0
Fork 0
mirror of https://github.com/BertoldVdb/ZoneDetect.git synced 2026-04-27 14:57:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Format library

Browse source
This commit is contained in:
Bertold Van den Bergh 2019-08-14 18:52:24 +02:00
parent 591a75df2f
commit 79da7aae0a
2 changed files with 179 additions and 159 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

178
library/zonedetect.c
View file

@ -32,12 +32,12 @@
#include <string.h> #include <string.h>
#include <math.h> #include <math.h>
#if defined(_MSC_VER) || defined(__MINGW32__) #if defined(_MSC_VER) || defined(__MINGW32__)
#include <windows.h> #include <windows.h>
#else #else
#include <errno.h> #include <errno.h>
#include <sys/mman.h> #include <sys/mman.h>
#include <fcntl.h> #include <fcntl.h>
#include <unistd.h> #include <unistd.h>
#endif #endif
#include "zonedetect.h" #include "zonedetect.h"
@ -95,7 +95,8 @@ static int32_t ZDFloatToFixedPoint(float input, float scale, unsigned int precis
return (int32_t)(inputScaled * (float)(1 << (precision - 1))); return (int32_t)(inputScaled * (float)(1 << (precision - 1)));
} }
static float ZDFixedPointToFloat(int32_t input, float scale, unsigned int precision){ static float ZDFixedPointToFloat(int32_t input, float scale, unsigned int precision)
{
const float value = (float)input / (float)(1 << (precision - 1)); const float value = (float)input / (float)(1 << (precision - 1));
return value * scale; return value * scale;
} }
@ -143,7 +144,8 @@ static unsigned int ZDDecodeVariableLengthUnsigned(const ZoneDetect *library, ui
return i; return i;
} }
static unsigned int ZDDecodeVariableLengthUnsignedReverse(const ZoneDetect *library, uint32_t *index, uint64_t *result){ static unsigned int ZDDecodeVariableLengthUnsignedReverse(const ZoneDetect *library, uint32_t *index, uint64_t *result)
{
uint32_t i = *index; uint32_t i = *index;
if(*index >= (uint32_t)library->length) { if(*index >= (uint32_t)library->length) {
@ -154,17 +156,17 @@ static unsigned int ZDDecodeVariableLengthUnsignedReverse(const ZoneDetect *libr
__try { __try {
#endif #endif
if(library->mapping[i] & UINT8_C(0x80)){ if(library->mapping[i] & UINT8_C(0x80)) {
return 0; return 0;
} }
if(!i){ if(!i) {
return 0; return 0;
} }
i--; i--;
while(library->mapping[i] & UINT8_C(0x80)){ while(library->mapping[i] & UINT8_C(0x80)) {
if(!i){ if(!i) {
return 0; return 0;
} }
i--; i--;
@ -187,7 +189,8 @@ static unsigned int ZDDecodeVariableLengthUnsignedReverse(const ZoneDetect *libr
return ZDDecodeVariableLengthUnsigned(library, &i2, result); return ZDDecodeVariableLengthUnsigned(library, &i2, result);
} }
static int64_t ZDDecodeUnsignedToSigned(uint64_t value){ static int64_t ZDDecodeUnsignedToSigned(uint64_t value)
{
return (value & 1) ? -(int64_t)(value / 2) : (int64_t)(value / 2); return (value & 1) ? -(int64_t)(value / 2) : (int64_t)(value / 2);
} }
@ -326,19 +329,20 @@ static int ZDPointInBox(int32_t xl, int32_t x, int32_t xr, int32_t yl, int32_t y
return 0; return 0;
} }
static void ZDDecodePoint(uint64_t point, int32_t* lat, int32_t* lon){ static void ZDDecodePoint(uint64_t point, int32_t* lat, int32_t* lon)
{
uint64_t latu = 0; uint64_t latu = 0;
uint64_t lonu = 0; uint64_t lonu = 0;
for(uint64_t i=0; i<32; i++){ for(uint64_t i=0; i<32; i++) {
latu <<= 1; latu <<= 1;
lonu <<= 1; lonu <<= 1;
if((point >> (2*(31-i))) & 1){ if((point >> (2*(31-i))) & 1) {
latu |= 1; latu |= 1;
} }
if((point >> (2*(31-i)+1)) & 1){ if((point >> (2*(31-i)+1)) & 1) {
lonu |= 1; lonu |= 1;
} }
} }
@ -347,7 +351,7 @@ static void ZDDecodePoint(uint64_t point, int32_t* lat, int32_t* lon){
*lon = (int32_t)ZDDecodeUnsignedToSigned(lonu); *lon = (int32_t)ZDDecodeUnsignedToSigned(lonu);
} }
struct Reader{ struct Reader {
const ZoneDetect *library; const ZoneDetect *library;
uint32_t polygonIndex; uint32_t polygonIndex;
@ -361,7 +365,8 @@ struct Reader{
int32_t firstLat, firstLon; int32_t firstLat, firstLon;
}; };
static void ZDReaderInit(struct Reader *reader, const ZoneDetect *library, uint32_t polygonIndex){ static void ZDReaderInit(struct Reader *reader, const ZoneDetect *library, uint32_t polygonIndex)
{
memset(reader, 0, sizeof(*reader)); memset(reader, 0, sizeof(*reader));
reader->library = library; reader->library = library;
@ -370,15 +375,16 @@ static void ZDReaderInit(struct Reader *reader, const ZoneDetect *library, uint3
reader->first = 1; reader->first = 1;
} }
static int ZDReaderGetPoint(struct Reader *reader, int32_t *pointLat, int32_t *pointLon){ static int ZDReaderGetPoint(struct Reader *reader, int32_t *pointLat, int32_t *pointLon)
{
int32_t diffLat = 0, diffLon = 0; int32_t diffLat = 0, diffLon = 0;
readNewPoint: readNewPoint:
if(reader->done > 1){ if(reader->done > 1) {
return 0; return 0;
} }
if(reader->first && reader->library->version == 0){ if(reader->first && reader->library->version == 0) {
if(!ZDDecodeVariableLengthUnsigned(reader->library, &reader->polygonIndex, &reader->numVertices)) return -1; if(!ZDDecodeVariableLengthUnsigned(reader->library, &reader->polygonIndex, &reader->numVertices)) return -1;
if(!reader->numVertices) return -1; if(!reader->numVertices) return -1;
} }
@ -386,37 +392,37 @@ readNewPoint:
uint64_t point; uint64_t point;
uint8_t referenceDone = 0; uint8_t referenceDone = 0;
if(reader->library->version == 1){ if(reader->library->version == 1) {
if(!reader->referenceDirection){ if(!reader->referenceDirection) {
if(!ZDDecodeVariableLengthUnsigned(reader->library, &reader->polygonIndex, &point)) return -1; if(!ZDDecodeVariableLengthUnsigned(reader->library, &reader->polygonIndex, &point)) return -1;
}else{ } else {
if(reader->referenceDirection > 0){ if(reader->referenceDirection > 0) {
/* Read reference forward */ /* Read reference forward */
if(!ZDDecodeVariableLengthUnsigned(reader->library, &reader->referenceStart, &point)) return -1; if(!ZDDecodeVariableLengthUnsigned(reader->library, &reader->referenceStart, &point)) return -1;
if(reader->referenceStart >= reader->referenceEnd){ if(reader->referenceStart >= reader->referenceEnd) {
referenceDone = 1; referenceDone = 1;
} }
}else if(reader->referenceDirection < 0){ } else if(reader->referenceDirection < 0) {
/* Read reference backwards */ /* Read reference backwards */
if(!ZDDecodeVariableLengthUnsignedReverse(reader->library, &reader->referenceStart, &point)) return -1; if(!ZDDecodeVariableLengthUnsignedReverse(reader->library, &reader->referenceStart, &point)) return -1;
if(reader->referenceStart <= reader->referenceEnd){ if(reader->referenceStart <= reader->referenceEnd) {
referenceDone = 1; referenceDone = 1;
} }
} }
} }
if(!point){ if(!point) {
/* This is a special marker, it is not allowed in reference mode */ /* This is a special marker, it is not allowed in reference mode */
if(reader->referenceDirection){ if(reader->referenceDirection) {
return -1; return -1;
} }
uint64_t value; uint64_t value;
if(!ZDDecodeVariableLengthUnsigned(reader->library, &reader->polygonIndex, &value)) return -1; if(!ZDDecodeVariableLengthUnsigned(reader->library, &reader->polygonIndex, &value)) return -1;
if(value == 0){ if(value == 0) {
reader->done = 2; reader->done = 2;
}else if(value == 1){ } else if(value == 1) {
int32_t diff; int32_t diff;
int64_t start; int64_t start;
if(!ZDDecodeVariableLengthUnsigned(reader->library, &reader->polygonIndex, (uint64_t*)&start)) return -1; if(!ZDDecodeVariableLengthUnsigned(reader->library, &reader->polygonIndex, (uint64_t*)&start)) return -1;
@ -425,27 +431,27 @@ readNewPoint:
reader->referenceStart = reader->library->dataOffset+(uint32_t)start; reader->referenceStart = reader->library->dataOffset+(uint32_t)start;
reader->referenceEnd = reader->library->dataOffset+(uint32_t)(start + diff); reader->referenceEnd = reader->library->dataOffset+(uint32_t)(start + diff);
reader->referenceDirection = diff; reader->referenceDirection = diff;
if(diff < 0){ if(diff < 0) {
reader->referenceStart--; reader->referenceStart--;
reader->referenceEnd--; reader->referenceEnd--;
} }
goto readNewPoint; goto readNewPoint;
} }
}else{ } else {
ZDDecodePoint(point, &diffLat, &diffLon); ZDDecodePoint(point, &diffLat, &diffLon);
if(reader->referenceDirection < 0){ if(reader->referenceDirection < 0) {
diffLat = -diffLat; diffLat = -diffLat;
diffLon = -diffLon; diffLon = -diffLon;
} }
} }
} }
if(reader->library->version == 0){ if(reader->library->version == 0) {
if(!ZDDecodeVariableLengthSigned(reader->library, &reader->polygonIndex, &diffLat)) return -1; if(!ZDDecodeVariableLengthSigned(reader->library, &reader->polygonIndex, &diffLat)) return -1;
if(!ZDDecodeVariableLengthSigned(reader->library, &reader->polygonIndex, &diffLon)) return -1; if(!ZDDecodeVariableLengthSigned(reader->library, &reader->polygonIndex, &diffLon)) return -1;
} }
if(!reader->done){ if(!reader->done) {
reader->pointLat += diffLat; reader->pointLat += diffLat;
reader->pointLon += diffLon; reader->pointLon += diffLon;
if(reader->first) { if(reader->first) {
@ -461,32 +467,33 @@ readNewPoint:
reader->first = 0; reader->first = 0;
if(reader->library->version == 0){ if(reader->library->version == 0) {
reader->numVertices--; reader->numVertices--;
if(!reader->numVertices){ if(!reader->numVertices) {
reader->done = 1; reader->done = 1;
} }
if(!diffLat && !diffLon){ if(!diffLat && !diffLon) {
goto readNewPoint; goto readNewPoint;
} }
} }
if(referenceDone){ if(referenceDone) {
reader->referenceDirection = 0; reader->referenceDirection = 0;
} }
if(pointLat){ if(pointLat) {
*pointLat = reader->pointLat; *pointLat = reader->pointLat;
} }
if(pointLon){ if(pointLon) {
*pointLon = reader->pointLon; *pointLon = reader->pointLon;
} }
return 1; return 1;
} }
static int ZDFindPolygon(const ZoneDetect *library, uint32_t wantedId, uint32_t* metadataIndexPtr, uint32_t* polygonIndexPtr){ static int ZDFindPolygon(const ZoneDetect *library, uint32_t wantedId, uint32_t* metadataIndexPtr, uint32_t* polygonIndexPtr)
{
uint32_t polygonId = 0; uint32_t polygonId = 0;
uint32_t bboxIndex = library->bboxOffset; uint32_t bboxIndex = library->bboxOffset;
@ -506,12 +513,12 @@ static int ZDFindPolygon(const ZoneDetect *library, uint32_t wantedId, uint32_t*
metadataIndex += (uint32_t)metadataIndexDelta; metadataIndex += (uint32_t)metadataIndexDelta;
polygonIndex += (uint32_t)polygonIndexDelta; polygonIndex += (uint32_t)polygonIndexDelta;
if(polygonId == wantedId){ if(polygonId == wantedId) {
if(metadataIndexPtr){ if(metadataIndexPtr) {
metadataIndex += library->metadataOffset; metadataIndex += library->metadataOffset;
*metadataIndexPtr = metadataIndex; *metadataIndexPtr = metadataIndex;
} }
if(polygonIndexPtr){ if(polygonIndexPtr) {
polygonIndex += library->dataOffset; polygonIndex += library->dataOffset;
*polygonIndexPtr = polygonIndex; *polygonIndexPtr = polygonIndex;
} }
@ -524,7 +531,8 @@ static int ZDFindPolygon(const ZoneDetect *library, uint32_t wantedId, uint32_t*
return 0; return 0;
} }
static int32_t* ZDPolygonToListInternal(const ZoneDetect *library, uint32_t polygonIndex, size_t* length){ static int32_t* ZDPolygonToListInternal(const ZoneDetect *library, uint32_t polygonIndex, size_t* length)
{
struct Reader reader; struct Reader reader;
ZDReaderInit(&reader, library, polygonIndex); ZDReaderInit(&reader, library, polygonIndex);
@ -532,27 +540,27 @@ static int32_t* ZDPolygonToListInternal(const ZoneDetect *library, uint32_t poly
size_t listIndex = 0; size_t listIndex = 0;
int32_t* list = malloc(sizeof(int32_t) * listLength); int32_t* list = malloc(sizeof(int32_t) * listLength);
if(!list){ if(!list) {
goto fail; goto fail;
} }
while(1){ while(1) {
int32_t pointLat, pointLon; int32_t pointLat, pointLon;
int result = ZDReaderGetPoint(&reader, &pointLat, &pointLon); int result = ZDReaderGetPoint(&reader, &pointLat, &pointLon);
if(result < 0){ if(result < 0) {
goto fail; goto fail;
}else if(result == 0){ } else if(result == 0) {
break; break;
} }
if(listIndex >= listLength){ if(listIndex >= listLength) {
listLength *= 2; listLength *= 2;
if(listLength >= 1048576){ if(listLength >= 1048576) {
goto fail; goto fail;
} }
list = realloc(list, sizeof(int32_t) * listLength); list = realloc(list, sizeof(int32_t) * listLength);
if(!list){ if(!list) {
goto fail; goto fail;
} }
} }
@ -561,41 +569,42 @@ static int32_t* ZDPolygonToListInternal(const ZoneDetect *library, uint32_t poly
list[listIndex++] = pointLon; list[listIndex++] = pointLon;
} }
if(length){ if(length) {
*length = listIndex; *length = listIndex;
} }
return list; return list;
fail: fail:
if(list){ if(list) {
free(list); free(list);
} }
return NULL; return NULL;
} }
float* ZDPolygonToList(const ZoneDetect *library, uint32_t polygonId, size_t* lengthPtr){ float* ZDPolygonToList(const ZoneDetect *library, uint32_t polygonId, size_t* lengthPtr)
{
uint32_t polygonIndex; uint32_t polygonIndex;
int32_t* data = NULL; int32_t* data = NULL;
float* flData = NULL; float* flData = NULL;
if(!ZDFindPolygon(library, polygonId, NULL, &polygonIndex)){ if(!ZDFindPolygon(library, polygonId, NULL, &polygonIndex)) {
goto fail; goto fail;
} }
size_t length = 0; size_t length = 0;
data = ZDPolygonToListInternal(library, polygonIndex, &length); data = ZDPolygonToListInternal(library, polygonIndex, &length);
if(!data){ if(!data) {
goto fail; goto fail;
} }
flData = malloc(sizeof(float) * length); flData = malloc(sizeof(float) * length);
if(!flData){ if(!flData) {
goto fail; goto fail;
} }
for(size_t i = 0; i<length; i+= 2){ for(size_t i = 0; i<length; i+= 2) {
int32_t lat = data[i]; int32_t lat = data[i];
int32_t lon = data[i+1]; int32_t lon = data[i+1];
@ -603,17 +612,17 @@ float* ZDPolygonToList(const ZoneDetect *library, uint32_t polygonId, size_t* le
flData[i+1] = ZDFixedPointToFloat(lon, 180, library->precision); flData[i+1] = ZDFixedPointToFloat(lon, 180, library->precision);
} }
if(lengthPtr){ if(lengthPtr) {
*lengthPtr = length; *lengthPtr = length;
} }
return flData; return flData;
fail: fail:
if(data){ if(data) {
free(data); free(data);
} }
if(flData){ if(flData) {
free(flData); free(flData);
} }
return NULL; return NULL;
@ -629,11 +638,11 @@ static ZDLookupResult ZDPointInPolygon(const ZoneDetect *library, uint32_t polyg
struct Reader reader; struct Reader reader;
ZDReaderInit(&reader, library, polygonIndex); ZDReaderInit(&reader, library, polygonIndex);
while(1){ while(1) {
int result = ZDReaderGetPoint(&reader, &pointLat, &pointLon); int result = ZDReaderGetPoint(&reader, &pointLat, &pointLon);
if(result < 0){ if(result < 0) {
return ZD_LOOKUP_PARSE_ERROR; return ZD_LOOKUP_PARSE_ERROR;
}else if(result == 0){ } else if(result == 0) {
break; break;
} }
@ -1051,19 +1060,30 @@ const char *ZDLookupResultToString(ZDLookupResult result)
const char *ZDGetErrorString(int errZD) const char *ZDGetErrorString(int errZD)
{ {
switch ((enum ZDInternalError)errZD) { switch ((enum ZDInternalError)errZD) {
default: assert(0); default:
case ZD_OK : return ""; assert(0);
case ZD_E_DB_OPEN : return ZD_E_COULD_NOT("open database file"); case ZD_OK :
case ZD_E_DB_SEEK : return ZD_E_COULD_NOT("retrieve database file size"); return "";
case ZD_E_DB_MMAP : return ZD_E_COULD_NOT("map database file to system memory"); case ZD_E_DB_OPEN :
return ZD_E_COULD_NOT("open database file");
case ZD_E_DB_SEEK :
return ZD_E_COULD_NOT("retrieve database file size");
case ZD_E_DB_MMAP :
return ZD_E_COULD_NOT("map database file to system memory");
#if defined(_MSC_VER) || defined(__MINGW32__) #if defined(_MSC_VER) || defined(__MINGW32__)
case ZD_E_DB_MMAP_MSVIEW : return ZD_E_COULD_NOT("open database file view"); case ZD_E_DB_MMAP_MSVIEW :
case ZD_E_DB_MAP_EXCEPTION: return "I/O exception occurred while accessing database file view"; return ZD_E_COULD_NOT("open database file view");
case ZD_E_DB_MUNMAP_MSVIEW: return ZD_E_COULD_NOT("close database file view"); case ZD_E_DB_MAP_EXCEPTION:
return "I/O exception occurred while accessing database file view";
case ZD_E_DB_MUNMAP_MSVIEW:
return ZD_E_COULD_NOT("close database file view");
#endif #endif
case ZD_E_DB_MUNMAP : return ZD_E_COULD_NOT("unmap database"); case ZD_E_DB_MUNMAP :
case ZD_E_DB_CLOSE : return ZD_E_COULD_NOT("close database file"); return ZD_E_COULD_NOT("unmap database");
case ZD_E_PARSE_HEADER : return ZD_E_COULD_NOT("parse database header"); case ZD_E_DB_CLOSE :
return ZD_E_COULD_NOT("close database file");
case ZD_E_PARSE_HEADER :
return ZD_E_COULD_NOT("parse database header");
} }
} }

10
library/zonedetect.h
View file

@ -31,11 +31,11 @@
#define INCL_ZONEDETECT_H_ #define INCL_ZONEDETECT_H_
#if !defined(ZD_EXPORT) #if !defined(ZD_EXPORT)
#if defined(_MSC_VER) #if defined(_MSC_VER)
#define ZD_EXPORT __declspec(dllimport) #define ZD_EXPORT __declspec(dllimport)
#else #else
#define ZD_EXPORT #define ZD_EXPORT
#endif #endif
#endif #endif
typedef enum { typedef enum {