32 #include "pgsql_compat.h"
34 #ifdef HAVE_LIBPROTOBUF
35 #include "utils/jsonb.h"
37 #include "lwgeom_wagyu.h"
39 #define uthash_fatal(msg) lwerror("uthash: fatal error (out of memory)")
40 #define uthash_malloc(sz) palloc(sz)
41 #define uthash_free(ptr,sz) pfree(ptr)
45 #include "vector_tile.pb-c.h"
47 #define FEATURES_CAPACITY_INITIAL 50
71 VectorTile__Tile__Value
value[1];
78 #define TAGS_INITIAL_CAPACITY 20
93 VectorTile__Tile__GeomType
type;
104 builder->
type = VECTOR_TILE__TILE__GEOM_TYPE__UNKNOWN;
111 VectorTile__Tile__Feature *feature = palloc(
sizeof(*feature));
112 vector_tile__tile__feature__init(feature);
114 feature->has_id = builder->
has_id;
115 feature->id = builder->
id;
116 feature->n_tags = builder->
n_tags;
117 feature->tags = builder->
tags;
118 feature->type = builder->
type;
120 feature->geometry = builder->
geometry;
126 size_t new_n_tags = builder->
n_tags + 2;
130 builder->
tags = repalloc(builder->
tags, new_capacity *
sizeof(*builder->
tags));
135 builder->
tags[builder->
n_tags + 1] = value_id;
136 builder->
n_tags = new_n_tags;
142 return (
id & 0x7) | (
count << 3);
154 int32_t dx, dy,
x,
y;
158 for (i = 0; i < pa->
npoints; i++)
201 int32_t px = 0, py = 0;
207 feature->
type = VECTOR_TILE__TILE__GEOM_TYPE__POINT;
218 feature->
type = VECTOR_TILE__TILE__GEOM_TYPE__POINT;
228 feature->
type = VECTOR_TILE__TILE__GEOM_TYPE__LINESTRING;
238 int32_t px = 0, py = 0;
239 size_t c = 0, offset = 0;
240 feature->
type = VECTOR_TILE__TILE__GEOM_TYPE__LINESTRING;
241 for (i = 0; i < lwmline->
ngeoms; i++)
244 for (i = 0; i < lwmline->
ngeoms; i++)
247 feature->
geometry + offset, &px, &py);
254 int32_t px = 0, py = 0;
255 size_t c = 0, offset = 0;
256 feature->
type = VECTOR_TILE__TILE__GEOM_TYPE__POLYGON;
257 for (i = 0; i < lwpoly->
nrings; i++)
260 for (i = 0; i < lwpoly->
nrings; i++)
263 feature->
geometry + offset, &px, &py);
270 int32_t px = 0, py = 0;
271 size_t c = 0, offset = 0;
273 feature->
type = VECTOR_TILE__TILE__GEOM_TYPE__POLYGON;
274 for (i = 0; i < lwmpoly->
ngeoms; i++)
275 for (j = 0; poly = lwmpoly->
geoms[i], j < poly->
nrings; j++)
278 for (i = 0; i < lwmpoly->
ngeoms; i++)
279 for (j = 0; poly = lwmpoly->
geoms[i], j < poly->
nrings; j++)
304 default: elog(ERROR,
"encode_feature_geometry: '%s' geometry type not supported",
311 Oid tupType = HeapTupleHeaderGetTypeId(ctx->
row);
312 int32 tupTypmod = HeapTupleHeaderGetTypMod(ctx->
row);
313 TupleDesc tupdesc = lookup_rowtype_tupdesc(tupType, tupTypmod);
329 size_t size = strlen(
name);
330 kv = palloc(
sizeof(*kv));
340 bool geom_found =
false;
342 POSTGIS_DEBUG(2,
"parse_column_keys called");
352 for (i = 0; i < natts; i++)
359 if (typoid == JSONBOID)
367 if (!geom_found && typoid == postgis_oid(GEOMETRYOID))
376 if (!geom_found && strcmp(tkey, ctx->
geom_name) == 0)
386 (strcmp(tkey, ctx->
id_name) == 0) &&
387 (typoid == INT2OID || typoid == INT4OID || typoid == INT8OID))
398 elog(ERROR,
"parse_column_keys: no geometry column found");
401 elog(ERROR,
"mvt_agg_transfn: Could not find column '%s' of integer type", ctx->
id_name);
408 char **keys = palloc(n_keys *
sizeof(*keys));
409 for (kv = ctx->
keys_hash; kv != NULL; kv=kv->
hh.next)
411 ctx->
layer->n_keys = n_keys;
412 ctx->
layer->keys = keys;
417 #define MVT_CREATE_VALUES(hash) \
419 struct mvt_kv_value *kv; \
420 for (kv = hash; kv != NULL; kv = kv->hh.next) \
422 values[kv->id] = kv->value; \
428 VectorTile__Tile__Value **values;
429 POSTGIS_DEBUG(2,
"encode_values called");
439 POSTGIS_DEBUGF(3,
"encode_values n_values: %d", ctx->
values_hash_i);
441 ctx->
layer->values = values;
450 #define MVT_PARSE_VALUE(hash, newvalue, size, pfvaluefield, pftype) \
452 POSTGIS_DEBUG(2, "MVT_PARSE_VALUE called"); \
454 struct mvt_kv_value *kv; \
456 HASH_VALUE(&newvalue, size, hashv); \
457 HASH_FIND_BYHASHVALUE(hh, ctx->hash, &newvalue, size, hashv, kv); \
460 POSTGIS_DEBUG(4, "MVT_PARSE_VALUE value not found"); \
461 kv = palloc(sizeof(*kv)); \
462 POSTGIS_DEBUGF(4, "MVT_PARSE_VALUE new hash key: %d", ctx->values_hash_i); \
463 kv->id = ctx->values_hash_i++; \
464 vector_tile__tile__value__init(kv->value); \
465 kv->value->pfvaluefield = newvalue; \
466 kv->value->test_oneof_case = pftype; \
467 HASH_ADD_KEYPTR_BYHASHVALUE(hh, ctx->hash, &kv->value->pfvaluefield, size, hashv, kv); \
469 feature_add_property(feature, k, kv->id); \
473 #define MVT_PARSE_INT_VALUE(value) \
477 uint64_t cvalue = value; \
478 MVT_PARSE_VALUE(uint_values_hash, \
482 VECTOR_TILE__TILE__VALUE__TEST_ONEOF_UINT_VALUE); \
486 int64_t cvalue = value; \
487 MVT_PARSE_VALUE(sint_values_hash, \
491 VECTOR_TILE__TILE__VALUE__TEST_ONEOF_SINT_VALUE); \
495 #define MVT_PARSE_DATUM(type, datumfunc, hash, size, pfvaluefield, pftype) \
497 type value = datumfunc(datum); \
498 MVT_PARSE_VALUE(hash, value, size, pfvaluefield, pftype); \
501 #define MVT_PARSE_INT_DATUM(type, datumfunc) \
503 type value = datumfunc(datum); \
504 MVT_PARSE_INT_VALUE(value); \
513 HASH_VALUE(
value, size, hashv);
514 POSTGIS_DEBUG(2,
"add_value_as_string called");
518 POSTGIS_DEBUG(4,
"add_value_as_string value not found");
519 kv = palloc(
sizeof(*kv));
520 POSTGIS_DEBUGF(4,
"add_value_as_string new hash key: %d",
523 vector_tile__tile__value__init(kv->
value);
525 kv->
value->test_oneof_case = VECTOR_TILE__TILE__VALUE__TEST_ONEOF_STRING_VALUE;
550 POSTGIS_DEBUG(2,
"parse_value_as_string called");
551 getTypeOutputInfo(typoid, &foutoid, &typisvarlena);
552 value = OidOutputFunctionCall(foutoid, datum);
553 POSTGIS_DEBUGF(4,
"parse_value_as_string value: %s",
value);
561 bool skipNested =
false;
562 JsonbIteratorToken
r;
565 if (!JB_ROOT_IS_OBJECT(jb))
568 it = JsonbIteratorInit(&jb->root);
570 while ((
r = JsonbIteratorNext(&it, &v, skipNested)) != WJB_DONE)
574 if (
r == WJB_KEY && v.type != jbvNull)
580 char *key = palloc(v.val.string.len + 1);
581 memcpy(key, v.val.string.val, v.val.string.len);
582 key[v.val.string.len] =
'\0';
586 r = JsonbIteratorNext(&it, &v, skipNested);
588 if (v.type == jbvString)
590 char *
value = palloc(v.val.string.len + 1);
591 memcpy(
value, v.val.string.val, v.val.string.len);
592 value[v.val.string.len] =
'\0';
595 else if (v.type == jbvBool)
599 sizeof(protobuf_c_boolean),
601 VECTOR_TILE__TILE__VALUE__TEST_ONEOF_BOOL_VALUE);
603 else if (v.type == jbvNumeric)
608 str = DatumGetCString(DirectFunctionCall1(numeric_out,
609 PointerGetDatum(v.val.numeric)));
610 d = strtod(
str, NULL);
611 l = strtol(
str, NULL, 10);
613 if (fabs(d - (
double)l) > FLT_EPSILON)
619 VECTOR_TILE__TILE__VALUE__TEST_ONEOF_DOUBLE_VALUE);
636 int64_t
value = INT64_MIN;
640 POSTGIS_DEBUG(3,
"set_feature_id: Ignored null value");
647 value = DatumGetInt16(datum);
650 value = DatumGetInt32(datum);
653 value = DatumGetInt64(datum);
656 elog(ERROR,
"set_feature_id: Feature id type does not match");
661 POSTGIS_DEBUG(3,
"set_feature_id: Ignored negative value");
666 feature->
id = (uint64_t)
value;
673 uint32_t natts = (uint32_t) cc.
tupdesc->natts;
677 POSTGIS_DEBUG(2,
"parse_values called");
680 tuple.t_len = HeapTupleHeaderGetDatumLength(ctx->
row);
681 ItemPointerSetInvalid(&(tuple.t_self));
682 tuple.t_tableOid = InvalidOid;
683 tuple.t_data = ctx->
row;
688 POSTGIS_DEBUGF(3,
"parse_values natts: %d", natts);
690 for (i = 0; i < natts; i++)
695 Datum datum = cc.
values[i];
708 POSTGIS_DEBUG(3,
"parse_values isnull detected");
712 key = TupleDescAttr(cc.
tupdesc, i)->attname.data;
717 elog(ERROR,
"parse_values: unexpectedly could not find parsed key name '%s'", key);
718 if (typoid == JSONBOID)
730 sizeof(protobuf_c_boolean),
732 VECTOR_TILE__TILE__VALUE__TEST_ONEOF_BOOL_VALUE);
749 VECTOR_TILE__TILE__VALUE__TEST_ONEOF_FLOAT_VALUE);
757 VECTOR_TILE__TILE__VALUE__TEST_ONEOF_DOUBLE_VALUE);
788 return geom->
type - 3;
795 for (i = 0; i < g->
ngeoms; i++)
803 elog(ERROR,
"%s: Invalid type (%d)", __func__, geom->
type);
830 geom_out = g->
geoms[0];
851 POSTGIS_DEBUG(3,
"mvt_geom: geometry outside clip box");
857 POSTGIS_DEBUG(3,
"mvt_geom: geometry contained fully inside the box");
880 gridspec grid = {0, 0, 0, 0, 1, 1, 0, 0};
919 clipped_lwgeom = lwgeom_wagyu_clip_by_box(lwgeom, &clip_box);
921 return clipped_lwgeom;
935 AFFINE affine = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
936 gridspec grid = {0, 0, 0, 0, 1, 1, 0, 0};
937 double width = gbox->
xmax - gbox->
xmin;
938 double height = gbox->
ymax - gbox->
ymin;
942 POSTGIS_DEBUG(2,
"mvt_geom called");
952 fy = -(extent / height);
991 VectorTile__Tile__Layer *layer;
993 POSTGIS_DEBUG(2,
"mvt_agg_init_context called");
996 elog(ERROR,
"mvt_agg_init_context: extent cannot be 0");
1014 layer = palloc(
sizeof(*layer));
1015 vector_tile__tile__layer__init(layer);
1017 layer->name = ctx->
name;
1018 layer->extent = ctx->
extent;
1033 bool isnull =
false;
1038 VectorTile__Tile__Layer *layer = ctx->
layer;
1039 POSTGIS_DEBUG(2,
"mvt_agg_transfn called");
1046 datum = GetAttributeByNum(ctx->
row, ctx->
geom_index + 1, &isnull);
1066 POSTGIS_DEBUGF(3,
"mvt_agg_transfn encoded feature count: %zd", layer->n_features);
1070 layer->features = repalloc(layer->features, new_capacity *
1071 sizeof(*layer->features));
1073 POSTGIS_DEBUGF(3,
"mvt_agg_transfn new_capacity: %zd", new_capacity);
1083 VectorTile__Tile *tile;
1087 tile = palloc(
sizeof(VectorTile__Tile));
1088 vector_tile__tile__init(tile);
1089 tile->layers = palloc(
sizeof(VectorTile__Tile__Layer*) * n_layers);
1090 tile->layers[0] = ctx->
layer;
1091 tile->n_layers = n_layers;
1110 if (ctx && ctx->
layer && ctx->
layer->n_features == 0)
1112 bytea* ba_empty = palloc(VARHDRSZ);
1113 SET_VARSIZE(ba_empty, VARHDRSZ);
1118 len = VARHDRSZ + vector_tile__tile__get_packed_size(ctx->
tile);
1120 vector_tile__tile__pack(ctx->
tile, (uint8_t*)VARDATA(ba));
1121 SET_VARSIZE(ba, len);
1133 return palloc(size);
1143 ProtobufCAllocator allocator =
1150 size_t len = VARSIZE_ANY_EXHDR(ba);
1151 VectorTile__Tile *tile = vector_tile__tile__unpack(&allocator, len, (uint8_t*)VARDATA(ba));
1165 static VectorTile__Tile__Layer *
1168 const uint32_t key_offset = layer->n_keys;
1169 const uint32_t value_offset = layer->n_values;
1170 const uint32_t feature_offset = layer->n_features;
1174 layer->keys = layer2->keys;
1175 layer->n_keys = layer2->n_keys;
1177 else if (layer2->n_keys)
1179 layer->keys = repalloc(layer->keys,
sizeof(
char *) * (layer->n_keys + layer2->n_keys));
1180 memcpy(&layer->keys[key_offset], layer2->keys,
sizeof(
char *) * layer2->n_keys);
1181 layer->n_keys += layer2->n_keys;
1184 if (!layer->n_values)
1186 layer->values = layer2->values;
1187 layer->n_values = layer2->n_values;
1189 else if (layer2->n_values)
1192 repalloc(layer->values,
sizeof(VectorTile__Tile__Value *) * (layer->n_values + layer2->n_values));
1194 &layer->values[value_offset], layer2->values,
sizeof(VectorTile__Tile__Value *) * layer2->n_values);
1195 layer->n_values += layer2->n_values;
1198 if (!layer->n_features)
1200 layer->features = layer2->features;
1201 layer->n_features = layer2->n_features;
1203 else if (layer2->n_features)
1205 layer->features = repalloc(
1206 layer->features,
sizeof(VectorTile__Tile__Feature *) * (layer->n_features + layer2->n_features));
1207 memcpy(&layer->features[feature_offset], layer2->features,
sizeof(
char *) * layer2->n_features);
1208 layer->n_features += layer2->n_features;
1210 for (uint32_t i = feature_offset; i < layer->n_features; i++)
1212 for (uint32_t t = 0; t < layer->features[i]->n_tags; t += 2)
1214 layer->features[i]->tags[t] += key_offset;
1215 layer->features[i]->tags[t + 1] += value_offset;
1224 static VectorTile__Tile *
1228 VectorTile__Tile *tile;
1231 if (tile1->n_layers == 0 && tile2->n_layers == 0)
1233 else if (tile1->n_layers == 0)
1235 else if (tile2->n_layers == 0)
1238 tile = palloc(
sizeof(VectorTile__Tile));
1239 vector_tile__tile__init(tile);
1240 tile->layers = palloc(
sizeof(
void*));
1244 for (i = 0; i < tile1->n_layers; i++)
1246 for (j = 0; j < tile2->n_layers; j++)
1248 VectorTile__Tile__Layer *l1 = tile1->layers[i];
1249 VectorTile__Tile__Layer *l2 = tile2->layers[j];
1250 if (strcmp(l1->name, l2->name)==0)
1255 tile->layers[tile->n_layers++] = layer;
1257 tile->layers = repalloc(tile->layers, (tile->n_layers+1) *
sizeof(
void*));
1268 if (ctx1 && ! ctx2)
return ctx1;
1269 if (ctx2 && ! ctx1)
return ctx2;
1270 if (ctx1 && ctx2 && ctx1->
tile && ctx2->
tile)
1279 elog(DEBUG2,
"ctx1->tile = %p", ctx1->
tile);
1280 elog(DEBUG2,
"ctx2->tile = %p", ctx2->
tile);
1281 elog(ERROR,
"%s: unable to combine contexts where tile attribute is null", __func__);
int gbox_overlaps_2d(const GBOX *g1, const GBOX *g2)
Return LW_TRUE if the GBOX overlaps on the 2d plane, LW_FALSE otherwise.
void gbox_init(GBOX *gbox)
Zero out all the entries in the GBOX.
int gbox_contains_2d(const GBOX *g1, const GBOX *g2)
Return LW_TRUE if the first GBOX contains the second on the 2d plane, LW_FALSE otherwise.
LWGEOM * lwgeom_from_gserialized(const GSERIALIZED *g)
Allocate a new LWGEOM from a GSERIALIZED.
int lwgeom_simplify_in_place(LWGEOM *igeom, double dist, int preserve_collapsed)
void lwgeom_free(LWGEOM *geom)
int lwgeom_remove_repeated_points_in_place(LWGEOM *in, double tolerance)
#define POINTTYPE
LWTYPE numbers, used internally by PostGIS.
LWCOLLECTION * lwcollection_extract(const LWCOLLECTION *col, uint32_t type)
int lwgeom_is_collection(const LWGEOM *lwgeom)
Determine whether a LWGEOM can contain sub-geometries or not.
void lwgeom_affine(LWGEOM *geom, const AFFINE *affine)
const char * lwtype_name(uint8_t type)
Return the type name string associated with a type number (e.g.
LWLINE * lwline_from_lwmpoint(int32_t srid, const LWMPOINT *mpoint)
int lwgeom_calculate_gbox(const LWGEOM *lwgeom, GBOX *gbox)
Calculate bounding box of a geometry, automatically taking into account whether it is cartesian or ge...
#define FLAGS_SET_GEODETIC(flags, value)
LWGEOM * lwgeom_clip_by_rect(const LWGEOM *geom1, double x0, double y0, double x1, double y1)
void lwgeom_grid_in_place(LWGEOM *lwgeom, const gridspec *grid)
static const POINT2D * getPoint2d_cp(const POINTARRAY *pa, uint32_t n)
Returns a POINT2D pointer into the POINTARRAY serialized_ptlist, suitable for reading from.
static uint32_t lwgeom_get_type(const LWGEOM *geom)
Return LWTYPE number.
static int lwgeom_is_empty(const LWGEOM *geom)
Return true or false depending on whether a geometry is an "empty" geometry (no vertices members)
static uint32_t get_key_index_with_size(mvt_agg_context *ctx, const char *name, size_t size)
static VectorTile__Tile__Layer * vectortile_layer_combine(VectorTile__Tile__Layer *layer, VectorTile__Tile__Layer *layer2)
Combine 2 layers.
#define MVT_PARSE_INT_VALUE(value)
static void parse_values(mvt_agg_context *ctx, struct feature_builder *feature)
static void encode_mpoly(struct feature_builder *feature, LWMPOLY *lwmpoly)
mvt_agg_context * mvt_ctx_combine(mvt_agg_context *ctx1, mvt_agg_context *ctx2)
mvt_agg_context * mvt_ctx_deserialize(const bytea *ba)
static void encode_line(struct feature_builder *feature, LWLINE *lwline)
#define MVT_PARSE_DATUM(type, datumfunc, hash, size, pfvaluefield, pftype)
LWGEOM * mvt_geom(LWGEOM *lwgeom, const GBOX *gbox, uint32_t extent, uint32_t buffer, bool clip_geom)
Transform a geometry into vector tile coordinate space.
static bool add_value_as_string_with_size(mvt_agg_context *ctx, struct feature_builder *feature, char *value, size_t size, uint32_t k)
#define FEATURES_CAPACITY_INITIAL
#define TAGS_INITIAL_CAPACITY
static void encode_keys(mvt_agg_context *ctx)
static LWGEOM * mvt_unsafe_clip_by_box(LWGEOM *lwg_in, GBOX *clip_box)
#define MVT_PARSE_INT_DATUM(type, datumfunc)
#define MVT_CREATE_VALUES(hash)
bytea * mvt_agg_finalfn(mvt_agg_context *ctx)
Finalize aggregation.
static void parse_datum_as_string(mvt_agg_context *ctx, struct feature_builder *feature, Oid typoid, Datum datum, uint32_t k)
bytea * mvt_ctx_serialize(mvt_agg_context *ctx)
static void feature_add_property(struct feature_builder *builder, uint32_t key_id, uint32_t value_id)
static void mvt_deallocator(__attribute__((__unused__)) void *data, void *ptr)
static void parse_column_keys(mvt_agg_context *ctx)
static void encode_mline(struct feature_builder *feature, LWMLINE *lwmline)
static uint32_t encode_ptarray_initial(enum mvt_type type, POINTARRAY *pa, uint32_t *buffer)
static uint32_t c_int(enum mvt_cmd_id id, uint32_t count)
static void set_feature_id(mvt_agg_context *ctx, struct feature_builder *feature, Datum datum, bool isNull)
Sets the feature id.
static void encode_poly(struct feature_builder *feature, LWPOLY *lwpoly)
static VectorTile__Tile * mvt_ctx_to_tile(mvt_agg_context *ctx)
static uint32_t add_key(mvt_agg_context *ctx, char *name)
static LWGEOM * mvt_clip_and_validate_geos(LWGEOM *lwgeom, uint8_t basic_type, uint32_t extent, uint32_t buffer, bool clip_geom)
static LWGEOM * lwgeom_to_basic_type(LWGEOM *geom, uint8_t original_type)
In place process a collection to find a concrete geometry object and expose that as the actual object...
static void * mvt_allocator(__attribute__((__unused__)) void *data, size_t size)
void mvt_agg_init_context(mvt_agg_context *ctx)
Initialize aggregation context.
static bytea * mvt_ctx_to_bytea(mvt_agg_context *ctx)
static void encode_point(struct feature_builder *feature, LWPOINT *point)
static void encode_mpoint(struct feature_builder *feature, LWMPOINT *mpoint)
static TupleDesc get_tuple_desc(mvt_agg_context *ctx)
#define MVT_PARSE_VALUE(hash, newvalue, size, pfvaluefield, pftype)
static void add_value_as_string(mvt_agg_context *ctx, struct feature_builder *feature, char *value, uint32_t k)
static uint8_t lwgeom_get_basic_type(LWGEOM *geom)
static LWGEOM * mvt_clip_and_validate(LWGEOM *lwgeom, uint8_t basic_type, uint32_t extent, uint32_t buffer, bool clip_geom)
static void parse_jsonb(mvt_agg_context *ctx, struct feature_builder *feature, Jsonb *jb)
static VectorTile__Tile * vectortile_tile_combine(VectorTile__Tile *tile1, VectorTile__Tile *tile2)
void mvt_agg_transfn(mvt_agg_context *ctx)
Aggregation step.
static void feature_init(struct feature_builder *builder)
static void encode_values(mvt_agg_context *ctx)
static uint32_t p_int(int32_t value)
static uint32_t encode_ptarray(enum mvt_type type, POINTARRAY *pa, uint32_t *buffer, int32_t *px, int32_t *py)
static VectorTile__Tile__Feature * feature_build(struct feature_builder *builder)
static void encode_feature_geometry(struct feature_builder *feature, LWGEOM *lwgeom)
Datum buffer(PG_FUNCTION_ARGS)
VectorTile__Tile__GeomType type
struct mvt_kv_value * uint_values_hash
VectorTile__Tile__Layer * layer
struct mvt_kv_key * keys_hash
struct mvt_kv_value * float_values_hash
struct mvt_kv_value * bool_values_hash
struct mvt_kv_value * string_values_hash
struct mvt_kv_value * double_values_hash
mvt_column_cache column_cache
struct mvt_kv_value * sint_values_hash
uint32_t * column_keys_index
VectorTile__Tile__Value value[1]