diff --git a/decode.cpp b/decode.cpp index 0727e3b..cc84daf 100644 --- a/decode.cpp +++ b/decode.cpp @@ -6,6 +6,7 @@ #include #include #include +#include #include #include #include @@ -49,7 +50,7 @@ struct lonlat { } }; -void handle(std::string message, int z, unsigned x, unsigned y, int describe) { +void handle(std::string message, int z, unsigned x, unsigned y, int describe, std::set const &to_decode) { int within = 0; mvt_tile tile; @@ -77,12 +78,17 @@ void handle(std::string message, int z, unsigned x, unsigned y, int describe) { printf(", \"features\": [\n"); + bool first_layer = true; for (size_t l = 0; l < tile.layers.size(); l++) { mvt_layer &layer = tile.layers[l]; int extent = layer.extent; + if (to_decode.size() != 0 && !to_decode.count(layer.name)) { + continue; + } + if (describe) { - if (l != 0) { + if (!first_layer) { printf(",\n"); } @@ -93,6 +99,7 @@ void handle(std::string message, int z, unsigned x, unsigned y, int describe) { printf(" }"); printf(", \"features\": [\n"); + first_layer = false; within = 0; } @@ -333,7 +340,7 @@ void handle(std::string message, int z, unsigned x, unsigned y, int describe) { printf("] }\n"); } -void decode(char *fname, int z, unsigned x, unsigned y) { +void decode(char *fname, int z, unsigned x, unsigned y, std::set const &to_decode) { sqlite3 *db; int oz = z; unsigned ox = x, oy = y; @@ -348,7 +355,7 @@ void decode(char *fname, int z, unsigned x, unsigned y) { if (strcmp(map, "SQLite format 3") != 0) { if (z >= 0) { std::string s = std::string(map, st.st_size); - handle(s, z, x, y, 1); + handle(s, z, x, y, 1, to_decode); munmap(map, st.st_size); return; } else { @@ -425,7 +432,7 @@ void decode(char *fname, int z, unsigned x, unsigned y) { ty = (1LL << tz) - 1 - ty; const char *s = (const char *) sqlite3_column_blob(stmt, 0); - handle(std::string(s, len), tz, tx, ty, 1); + handle(std::string(s, len), tz, tx, ty, 1, to_decode); } printf("] }\n"); @@ -453,7 +460,7 @@ void decode(char *fname, int z, unsigned x, unsigned y) { fprintf(stderr, "%s: Warning: using tile %d/%u/%u instead of %d/%u/%u\n", fname, z, x, y, oz, ox, oy); } - handle(std::string(s, len), z, x, y, 0); + handle(std::string(s, len), z, x, y, 0, to_decode); handled = 1; } @@ -472,7 +479,7 @@ void decode(char *fname, int z, unsigned x, unsigned y) { } void usage(char **argv) { - fprintf(stderr, "Usage: %s [-t projection] [-Z minzoom] [-z maxzoom] file.mbtiles [zoom x y]\n", argv[0]); + fprintf(stderr, "Usage: %s [-t projection] [-Z minzoom] [-z maxzoom] [-l layer ...] file.mbtiles [zoom x y]\n", argv[0]); exit(EXIT_FAILURE); } @@ -480,8 +487,9 @@ int main(int argc, char **argv) { extern int optind; extern char *optarg; int i; + std::set to_decode; - while ((i = getopt(argc, argv, "t:Z:z:")) != -1) { + while ((i = getopt(argc, argv, "t:Z:z:l:")) != -1) { switch (i) { case 't': set_projection_or_exit(optarg); @@ -495,15 +503,19 @@ int main(int argc, char **argv) { minzoom = atoi(optarg); break; + case 'l': + to_decode.insert(optarg); + break; + default: usage(argv); } } if (argc == optind + 4) { - decode(argv[optind], atoi(argv[optind + 1]), atoi(argv[optind + 2]), atoi(argv[optind + 3])); + decode(argv[optind], atoi(argv[optind + 1]), atoi(argv[optind + 2]), atoi(argv[optind + 3]), to_decode); } else if (argc == optind + 1) { - decode(argv[optind], -1, -1, -1); + decode(argv[optind], -1, -1, -1, to_decode); } else { usage(argv); }