Line data Source code
1 : /******************************************************************************
2 : *
3 : * Project: GDAL MBTiles driver
4 : * Purpose: Implement GDAL MBTiles support using OGR SQLite driver
5 : * Author: Even Rouault, Even Rouault <even.rouault at spatialys.com>
6 : *
7 : **********************************************************************
8 : * Copyright (c) 2012-2016, Even Rouault <even.rouault at spatialys.com>
9 : *
10 : * SPDX-License-Identifier: MIT
11 : ****************************************************************************/
12 :
13 : #if defined(HAVE_SQLITE) && defined(HAVE_GEOS)
14 : // Needed by mvtutils.h
15 : #define HAVE_MVT_WRITE_SUPPORT
16 : #endif
17 :
18 : #include "gdal_frmts.h"
19 : #include "gdal_pam.h"
20 : #include "ogr_api.h"
21 : #include "cpl_json.h"
22 : #include "cpl_vsil_curl_priv.h"
23 : #include "gpkgmbtilescommon.h"
24 : #include "gdal_utils.h"
25 : #include "gdalwarper.h"
26 : #include "mvtutils.h"
27 : #include "ogrsqlitevfs.h"
28 : #include "ogrsqlitebase.h"
29 :
30 : #include "zlib.h"
31 : #include "ogrlibjsonutils.h"
32 :
33 : #include <math.h>
34 : #include <algorithm>
35 : #include <memory>
36 : #include <vector>
37 :
38 : static const char *const apszAllowedDrivers[] = {"JPEG", "PNG", "WEBP",
39 : nullptr};
40 :
41 : #define SRS_EPSG_3857 \
42 : "PROJCS[\"WGS 84 / Pseudo-Mercator\",GEOGCS[\"WGS " \
43 : "84\",DATUM[\"WGS_1984\",SPHEROID[\"WGS " \
44 : "84\",6378137,298.257223563,AUTHORITY[\"EPSG\",\"7030\"]],AUTHORITY[" \
45 : "\"EPSG\",\"6326\"]],PRIMEM[\"Greenwich\",0,AUTHORITY[\"EPSG\",\"8901\"]]" \
46 : ",UNIT[\"degree\",0.0174532925199433,AUTHORITY[\"EPSG\",\"9122\"]]," \
47 : "AUTHORITY[\"EPSG\",\"4326\"]],PROJECTION[\"Mercator_1SP\"],PARAMETER[" \
48 : "\"central_meridian\",0],PARAMETER[\"scale_factor\",1],PARAMETER[\"false_" \
49 : "easting\",0],PARAMETER[\"false_northing\",0],UNIT[\"metre\",1,AUTHORITY[" \
50 : "\"EPSG\",\"9001\"]],AXIS[\"X\",EAST],AXIS[\"Y\",NORTH],EXTENSION[" \
51 : "\"PROJ4\",\"+proj=merc +a=6378137 +b=6378137 +lat_ts=0.0 +lon_0=0.0 " \
52 : "+x_0=0.0 +y_0=0 +k=1.0 +units=m +nadgrids=@null +wktext " \
53 : "+no_defs\"],AUTHORITY[\"EPSG\",\"3857\"]]"
54 :
55 : #define SPHERICAL_RADIUS 6378137.0
56 : #define MAX_GM (SPHERICAL_RADIUS * M_PI) // 20037508.342789244
57 :
58 : // TileMatrixSet origin : caution this is in GeoPackage / WMTS convention ! That
59 : // is upper-left corner
60 : #define TMS_ORIGIN_X -MAX_GM
61 : #define TMS_ORIGIN_Y MAX_GM
62 :
63 : #if defined(DEBUG) || defined(FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION) || \
64 : defined(ALLOW_FORMAT_DUMPS)
65 : // Enable accepting a SQL dump (starting with a "-- SQL MBTILES" line) as a
66 : // valid file. This makes fuzzer life easier
67 : #define ENABLE_SQL_SQLITE_FORMAT
68 : #endif
69 :
70 : constexpr int knDEFAULT_BLOCK_SIZE = 256;
71 :
72 : class MBTilesBand;
73 :
74 : /************************************************************************/
75 : /* MBTILESOpenSQLiteDB() */
76 : /************************************************************************/
77 :
78 77 : static GDALDatasetH MBTILESOpenSQLiteDB(const char *pszFilename,
79 : GDALAccess eAccess)
80 : {
81 77 : const char *l_apszAllowedDrivers[] = {"SQLITE", nullptr};
82 154 : return GDALOpenEx((CPLString("SQLITE:") + pszFilename).c_str(),
83 : GDAL_OF_VECTOR | GDAL_OF_INTERNAL |
84 : ((eAccess == GA_Update) ? GDAL_OF_UPDATE : 0),
85 154 : l_apszAllowedDrivers, nullptr, nullptr);
86 : }
87 :
88 : /************************************************************************/
89 : /* ==================================================================== */
90 : /* MBTilesDataset */
91 : /* ==================================================================== */
92 : /************************************************************************/
93 :
94 : class MBTilesDataset final : public GDALPamDataset,
95 : public GDALGPKGMBTilesLikePseudoDataset
96 : {
97 : friend class MBTilesBand;
98 : friend class MBTilesVectorLayer;
99 :
100 : public:
101 : MBTilesDataset();
102 :
103 : virtual ~MBTilesDataset();
104 :
105 : virtual CPLErr GetGeoTransform(double *padfGeoTransform) override;
106 : virtual CPLErr SetGeoTransform(double *padfGeoTransform) override;
107 : const OGRSpatialReference *GetSpatialRef() const override;
108 : CPLErr SetSpatialRef(const OGRSpatialReference *poSRS) override;
109 :
110 : virtual char **GetMetadataDomainList() override;
111 : virtual char **GetMetadata(const char *pszDomain = "") override;
112 : virtual const char *GetMetadataItem(const char *pszName,
113 : const char *pszDomain = "") override;
114 :
115 : virtual CPLErr IBuildOverviews(const char *pszResampling, int nOverviews,
116 : const int *panOverviewList, int nBandsIn,
117 : const int * /* panBandList */,
118 : GDALProgressFunc pfnProgress,
119 : void *pProgressData,
120 : CSLConstList papszOptions) override;
121 :
122 1166 : virtual int GetLayerCount() override
123 : {
124 1166 : return static_cast<int>(m_apoLayers.size());
125 : }
126 :
127 : virtual OGRLayer *GetLayer(int) override;
128 :
129 : static GDALDataset *Open(GDALOpenInfo *);
130 : static int Identify(GDALOpenInfo *);
131 : static GDALDataset *Create(const char *pszFilename, int nXSize, int nYSize,
132 : int nBandsIn, GDALDataType eDT,
133 : char **papszOptions);
134 : static GDALDataset *CreateCopy(const char *pszFilename,
135 : GDALDataset *poSrcDS, int bStrict,
136 : char **papszOptions,
137 : GDALProgressFunc pfnProgress,
138 : void *pProgressData);
139 :
140 : char *FindKey(int iPixel, int iLine);
141 :
142 : bool HasNonEmptyGrids();
143 :
144 : private:
145 : bool m_bWriteBounds;
146 : CPLString m_osBounds;
147 : CPLString m_osCenter;
148 : bool m_bWriteMinMaxZoom;
149 : MBTilesDataset *poMainDS;
150 : bool m_bGeoTransformValid;
151 : double m_adfGeoTransform[6];
152 : int m_nMinZoomLevel = 0;
153 : OGRSpatialReference m_oSRS{};
154 :
155 : int m_nOverviewCount;
156 : MBTilesDataset **m_papoOverviewDS;
157 :
158 : GDALDatasetH hDS;
159 : sqlite3 *hDB;
160 :
161 : sqlite3_vfs *pMyVFS;
162 :
163 : bool bFetchedMetadata;
164 : CPLStringList aosList;
165 :
166 : int nHasNonEmptyGrids;
167 :
168 : bool m_bInFlushCache;
169 :
170 : CPLString m_osMetadataMemFilename;
171 : CPLString m_osClip;
172 : std::vector<std::unique_ptr<OGRLayer>> m_apoLayers;
173 :
174 : void ParseCompressionOptions(char **papszOptions);
175 : CPLErr FinalizeRasterRegistration();
176 : void ComputeTileAndPixelShifts();
177 : bool InitRaster(MBTilesDataset *poParentDS, int nZoomLevel, int nBandCount,
178 : int nTileSize, double dfGDALMinX, double dfGDALMinY,
179 : double dfGDALMaxX, double dfGDALMaxY);
180 :
181 : bool CreateInternal(const char *pszFilename, int nXSize, int nYSize,
182 : int nBandsIn, GDALDataType eDT, char **papszOptions);
183 : void InitVector(double dfMinX, double dfMinY, double dfMaxX, double dfMaxY,
184 : bool bZoomLevelFromSpatialFilter, bool bJsonField);
185 :
186 : protected:
187 : // Coming from GDALGPKGMBTilesLikePseudoDataset
188 :
189 : virtual CPLErr IFlushCacheWithErrCode(bool bAtClosing) override;
190 :
191 1963 : virtual int IGetRasterCount() override
192 : {
193 1963 : return nBands;
194 : }
195 :
196 3310 : virtual GDALRasterBand *IGetRasterBand(int nBand) override
197 : {
198 3310 : return GetRasterBand(nBand);
199 : }
200 :
201 546 : virtual sqlite3 *IGetDB() override
202 : {
203 546 : return hDB;
204 : }
205 :
206 943 : virtual bool IGetUpdate() override
207 : {
208 943 : return eAccess == GA_Update;
209 : }
210 :
211 : virtual bool ICanIWriteBlock() override;
212 : virtual OGRErr IStartTransaction() override;
213 : virtual OGRErr ICommitTransaction() override;
214 :
215 22 : virtual const char *IGetFilename() override
216 : {
217 22 : return GetDescription();
218 : }
219 :
220 : virtual int GetRowFromIntoTopConvention(int nRow) override;
221 : };
222 :
223 : /************************************************************************/
224 : /* ==================================================================== */
225 : /* MBTilesVectorLayer */
226 : /* ==================================================================== */
227 : /************************************************************************/
228 :
229 : class MBTilesVectorLayer final : public OGRLayer
230 : {
231 : MBTilesDataset *m_poDS = nullptr;
232 : OGRFeatureDefn *m_poFeatureDefn = nullptr;
233 : OGRLayerH m_hTileIteratorLyr = nullptr;
234 : bool m_bEOF = false;
235 : CPLString m_osTmpFilename;
236 : GDALDatasetH m_hTileDS = nullptr;
237 : GIntBig m_nFeatureCount = -1;
238 : int m_nX = 0;
239 : int m_nY = 0;
240 : OGREnvelope m_sExtent;
241 : int m_nFilterMinX = 0;
242 : int m_nFilterMinY = 0;
243 : int m_nFilterMaxX = 0;
244 : int m_nFilterMaxY = 0;
245 : int m_nZoomLevel = 0;
246 : bool m_bZoomLevelAuto = false;
247 : bool m_bJsonField = false;
248 :
249 : OGRFeature *GetNextRawFeature();
250 : OGRFeature *GetNextSrcFeature();
251 : OGRFeature *CreateFeatureFrom(OGRFeature *poSrcFeature);
252 :
253 : public:
254 : MBTilesVectorLayer(MBTilesDataset *poDS, const char *pszLayerName,
255 : const CPLJSONObject &oFields,
256 : const CPLJSONArray &oAttributesFromTileStats,
257 : bool bJsonField, double dfMinX, double dfMinY,
258 : double dfMaxX, double dfMaxY,
259 : OGRwkbGeometryType eGeomType,
260 : bool bZoomLevelFromSpatialFilter);
261 : ~MBTilesVectorLayer();
262 :
263 : virtual void ResetReading() override;
264 : virtual OGRFeature *GetNextFeature() override;
265 :
266 856 : virtual OGRFeatureDefn *GetLayerDefn() override
267 : {
268 856 : return m_poFeatureDefn;
269 : }
270 :
271 : virtual GIntBig GetFeatureCount(int bForce) override;
272 : virtual int TestCapability(const char *) override;
273 :
274 : OGRErr GetExtent(OGREnvelope *psExtent, int bForce) override;
275 :
276 4 : virtual OGRErr GetExtent(int iGeomField, OGREnvelope *psExtent,
277 : int bForce) override
278 : {
279 4 : return OGRLayer::GetExtent(iGeomField, psExtent, bForce);
280 : }
281 :
282 : virtual void SetSpatialFilter(OGRGeometry *) override;
283 :
284 13 : virtual void SetSpatialFilter(int iGeomField, OGRGeometry *poGeom) override
285 : {
286 13 : OGRLayer::SetSpatialFilter(iGeomField, poGeom);
287 13 : }
288 :
289 : virtual OGRFeature *GetFeature(GIntBig nFID) override;
290 : };
291 :
292 : /************************************************************************/
293 : /* ==================================================================== */
294 : /* MBTilesBand */
295 : /* ==================================================================== */
296 : /************************************************************************/
297 :
298 : class MBTilesBand final : public GDALGPKGMBTilesLikeRasterBand
299 : {
300 : friend class MBTilesDataset;
301 :
302 : CPLString osLocationInfo;
303 :
304 : public:
305 : explicit MBTilesBand(MBTilesDataset *poDS, int nTileSize);
306 :
307 : virtual int GetOverviewCount() override;
308 : virtual GDALRasterBand *GetOverview(int nLevel) override;
309 :
310 : virtual char **GetMetadataDomainList() override;
311 : virtual const char *GetMetadataItem(const char *pszName,
312 : const char *pszDomain = "") override;
313 : };
314 :
315 : /************************************************************************/
316 : /* MBTilesBand() */
317 : /************************************************************************/
318 :
319 734 : MBTilesBand::MBTilesBand(MBTilesDataset *poDSIn, int nTileSize)
320 734 : : GDALGPKGMBTilesLikeRasterBand(poDSIn, nTileSize, nTileSize)
321 : {
322 734 : }
323 :
324 : /************************************************************************/
325 : /* utf8decode() */
326 : /************************************************************************/
327 :
328 0 : static unsigned utf8decode(const char *p, const char *end, int *len)
329 : {
330 0 : unsigned char c = *(unsigned char *)p;
331 0 : if (c < 0x80)
332 : {
333 0 : *len = 1;
334 0 : return c;
335 : }
336 0 : else if (c < 0xc2)
337 : {
338 0 : goto FAIL;
339 : }
340 0 : if (p + 1 >= end || (p[1] & 0xc0) != 0x80)
341 0 : goto FAIL;
342 0 : if (c < 0xe0)
343 : {
344 0 : *len = 2;
345 0 : return ((p[0] & 0x1f) << 6) + ((p[1] & 0x3f));
346 : }
347 0 : else if (c == 0xe0)
348 : {
349 0 : if (((unsigned char *)p)[1] < 0xa0)
350 0 : goto FAIL;
351 0 : goto UTF8_3;
352 : }
353 : #if STRICT_RFC3629
354 : else if (c == 0xed)
355 : {
356 : // RFC 3629 says surrogate chars are illegal.
357 : if (((unsigned char *)p)[1] >= 0xa0)
358 : goto FAIL;
359 : goto UTF8_3;
360 : }
361 : else if (c == 0xef)
362 : {
363 : // 0xfffe and 0xffff are also illegal characters
364 : if (((unsigned char *)p)[1] == 0xbf && ((unsigned char *)p)[2] >= 0xbe)
365 : goto FAIL;
366 : goto UTF8_3;
367 : }
368 : #endif
369 0 : else if (c < 0xf0)
370 : {
371 0 : UTF8_3:
372 0 : if (p + 2 >= end || (p[2] & 0xc0) != 0x80)
373 0 : goto FAIL;
374 0 : *len = 3;
375 0 : return ((p[0] & 0x0f) << 12) + ((p[1] & 0x3f) << 6) + ((p[2] & 0x3f));
376 : }
377 0 : else if (c == 0xf0)
378 : {
379 0 : if (((unsigned char *)p)[1] < 0x90)
380 0 : goto FAIL;
381 0 : goto UTF8_4;
382 : }
383 0 : else if (c < 0xf4)
384 : {
385 0 : UTF8_4:
386 0 : if (p + 3 >= end || (p[2] & 0xc0) != 0x80 || (p[3] & 0xc0) != 0x80)
387 0 : goto FAIL;
388 0 : *len = 4;
389 : #if STRICT_RFC3629
390 : // RFC 3629 says all codes ending in fffe or ffff are illegal:
391 : if ((p[1] & 0xf) == 0xf && ((unsigned char *)p)[2] == 0xbf &&
392 : ((unsigned char *)p)[3] >= 0xbe)
393 : goto FAIL;
394 : #endif
395 0 : return ((p[0] & 0x07) << 18) + ((p[1] & 0x3f) << 12) +
396 0 : ((p[2] & 0x3f) << 6) + ((p[3] & 0x3f));
397 : }
398 0 : else if (c == 0xf4)
399 : {
400 0 : if (((unsigned char *)p)[1] > 0x8f)
401 0 : goto FAIL; // after 0x10ffff
402 0 : goto UTF8_4;
403 : }
404 : else
405 : {
406 0 : FAIL:
407 0 : *len = 1;
408 0 : return 0xfffd; // Unicode REPLACEMENT CHARACTER
409 : }
410 : }
411 :
412 : /************************************************************************/
413 : /* HasNonEmptyGrids() */
414 : /************************************************************************/
415 :
416 0 : bool MBTilesDataset::HasNonEmptyGrids()
417 : {
418 : OGRLayerH hSQLLyr;
419 : OGRFeatureH hFeat;
420 :
421 0 : if (poMainDS)
422 0 : return poMainDS->HasNonEmptyGrids();
423 :
424 0 : if (nHasNonEmptyGrids >= 0)
425 0 : return nHasNonEmptyGrids != FALSE;
426 :
427 0 : nHasNonEmptyGrids = false;
428 :
429 0 : if (GDALDatasetGetLayerByName(hDS, "grids") == nullptr)
430 0 : return false;
431 :
432 0 : const char *pszSQL = "SELECT type FROM sqlite_master WHERE name = 'grids'";
433 0 : CPLDebug("MBTILES", "%s", pszSQL);
434 0 : hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, nullptr, nullptr);
435 0 : if (hSQLLyr == nullptr)
436 0 : return false;
437 :
438 0 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
439 0 : if (hFeat == nullptr || !OGR_F_IsFieldSetAndNotNull(hFeat, 0))
440 : {
441 0 : OGR_F_Destroy(hFeat);
442 0 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
443 0 : return false;
444 : }
445 :
446 0 : bool bGridsIsView = strcmp(OGR_F_GetFieldAsString(hFeat, 0), "view") == 0;
447 :
448 0 : OGR_F_Destroy(hFeat);
449 0 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
450 :
451 0 : nHasNonEmptyGrids = TRUE;
452 :
453 : /* In the case 'grids' is a view (and a join between the 'map' and
454 : * 'grid_utfgrid' layers */
455 : /* the cost of evaluating a join is very long, even if grid_utfgrid is empty
456 : */
457 : /* so check it is not empty */
458 0 : if (bGridsIsView)
459 : {
460 : OGRLayerH hGridUTFGridLyr;
461 0 : hGridUTFGridLyr = GDALDatasetGetLayerByName(hDS, "grid_utfgrid");
462 0 : if (hGridUTFGridLyr != nullptr)
463 : {
464 0 : OGR_L_ResetReading(hGridUTFGridLyr);
465 0 : hFeat = OGR_L_GetNextFeature(hGridUTFGridLyr);
466 0 : OGR_F_Destroy(hFeat);
467 :
468 0 : nHasNonEmptyGrids = hFeat != nullptr;
469 : }
470 : }
471 :
472 0 : return nHasNonEmptyGrids != FALSE;
473 : }
474 :
475 : /************************************************************************/
476 : /* FindKey() */
477 : /************************************************************************/
478 :
479 0 : char *MBTilesDataset::FindKey(int iPixel, int iLine)
480 : {
481 : int nBlockXSize;
482 : int nBlockYSize;
483 0 : GetRasterBand(1)->GetBlockSize(&nBlockXSize, &nBlockYSize);
484 :
485 : // Compute shift between GDAL origin and TileMatrixSet origin
486 : // Caution this is in GeoPackage / WMTS convention ! That is upper-left
487 : // corner
488 0 : const int nShiftXPixels = (int)floor(
489 0 : 0.5 + (m_adfGeoTransform[0] - TMS_ORIGIN_X) / m_adfGeoTransform[1]);
490 0 : const int nShiftYPixelsFromGPKGOrigin = (int)floor(
491 0 : 0.5 + (m_adfGeoTransform[3] - TMS_ORIGIN_Y) / m_adfGeoTransform[5]);
492 :
493 0 : const int iLineFromGPKGOrigin = iLine + nShiftYPixelsFromGPKGOrigin;
494 0 : const int iLineFromMBTilesOrigin =
495 0 : m_nTileMatrixHeight * nBlockYSize - 1 - iLineFromGPKGOrigin;
496 0 : const int iPixelFromMBTilesOrigin = iPixel + nShiftXPixels;
497 :
498 0 : const int nTileColumn = iPixelFromMBTilesOrigin / nBlockXSize;
499 0 : const int nTileRow = iLineFromMBTilesOrigin / nBlockYSize;
500 0 : int nColInTile = iPixelFromMBTilesOrigin % nBlockXSize;
501 0 : int nRowInTile = nBlockYSize - 1 - (iLineFromMBTilesOrigin % nBlockYSize);
502 :
503 0 : char *pszKey = nullptr;
504 :
505 : OGRLayerH hSQLLyr;
506 : OGRFeatureH hFeat;
507 0 : json_object *poGrid = nullptr;
508 : int i;
509 :
510 : /* See https://github.com/mapbox/utfgrid-spec/blob/master/1.0/utfgrid.md */
511 : /* for the explanation of the following process */
512 : const char *pszSQL =
513 0 : CPLSPrintf("SELECT grid FROM grids WHERE "
514 : "zoom_level = %d AND tile_column = %d AND tile_row = %d",
515 : m_nZoomLevel, nTileColumn, nTileRow);
516 0 : CPLDebug("MBTILES", "%s", pszSQL);
517 0 : hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, nullptr, nullptr);
518 0 : if (hSQLLyr == nullptr)
519 0 : return nullptr;
520 :
521 0 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
522 0 : if (hFeat == nullptr || !OGR_F_IsFieldSetAndNotNull(hFeat, 0))
523 : {
524 0 : OGR_F_Destroy(hFeat);
525 0 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
526 0 : return nullptr;
527 : }
528 :
529 0 : int nDataSize = 0;
530 0 : GByte *pabyData = OGR_F_GetFieldAsBinary(hFeat, 0, &nDataSize);
531 :
532 0 : int nUncompressedSize = nBlockXSize * nBlockYSize;
533 0 : GByte *pabyUncompressed = (GByte *)VSIMalloc(nUncompressedSize + 1);
534 0 : if (pabyUncompressed == nullptr)
535 : {
536 0 : OGR_F_Destroy(hFeat);
537 0 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
538 0 : return nullptr;
539 : }
540 :
541 : z_stream sStream;
542 0 : memset(&sStream, 0, sizeof(sStream));
543 0 : if (inflateInit(&sStream) != Z_OK)
544 : {
545 0 : OGR_F_Destroy(hFeat);
546 0 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
547 0 : CPLFree(pabyUncompressed);
548 0 : return nullptr;
549 : }
550 0 : sStream.next_in = pabyData;
551 0 : sStream.avail_in = nDataSize;
552 0 : sStream.next_out = pabyUncompressed;
553 0 : sStream.avail_out = nUncompressedSize;
554 0 : int nStatus = inflate(&sStream, Z_FINISH);
555 0 : inflateEnd(&sStream);
556 0 : if (nStatus != Z_OK && nStatus != Z_STREAM_END)
557 : {
558 0 : CPLDebug("MBTILES", "Error unzipping grid");
559 0 : nUncompressedSize = 0;
560 0 : pabyUncompressed[nUncompressedSize] = 0;
561 : }
562 : else
563 : {
564 0 : nUncompressedSize -= sStream.avail_out;
565 0 : pabyUncompressed[nUncompressedSize] = 0;
566 : // CPLDebug("MBTILES", "Grid size = %d", nUncompressedSize);
567 : // CPLDebug("MBTILES", "Grid value = %s", (const
568 : // char*)pabyUncompressed);
569 : }
570 :
571 0 : json_object *jsobj = nullptr;
572 :
573 0 : if (nUncompressedSize == 0)
574 : {
575 0 : goto end;
576 : }
577 :
578 0 : if (!OGRJSonParse((const char *)pabyUncompressed, &jsobj, true))
579 : {
580 0 : goto end;
581 : }
582 :
583 0 : if (json_object_is_type(jsobj, json_type_object))
584 : {
585 0 : poGrid = CPL_json_object_object_get(jsobj, "grid");
586 : }
587 0 : if (poGrid != nullptr && json_object_is_type(poGrid, json_type_array))
588 : {
589 : int nFactor;
590 : json_object *poRow;
591 0 : char *pszRow = nullptr;
592 :
593 0 : const int nLines = static_cast<int>(json_object_array_length(poGrid));
594 0 : if (nLines == 0)
595 0 : goto end;
596 :
597 0 : nFactor = nBlockXSize / nLines;
598 0 : nRowInTile /= nFactor;
599 0 : nColInTile /= nFactor;
600 :
601 0 : poRow = json_object_array_get_idx(poGrid, nRowInTile);
602 :
603 : /* Extract line of interest in grid */
604 0 : if (poRow != nullptr && json_object_is_type(poRow, json_type_string))
605 : {
606 0 : pszRow = CPLStrdup(json_object_get_string(poRow));
607 : }
608 :
609 0 : if (pszRow == nullptr)
610 0 : goto end;
611 :
612 : /* Unapply JSON encoding */
613 0 : for (i = 0; pszRow[i] != '\0'; i++)
614 : {
615 0 : unsigned char c = ((GByte *)pszRow)[i];
616 0 : if (c >= 93)
617 0 : c--;
618 0 : if (c >= 35)
619 0 : c--;
620 0 : if (c < 32)
621 : {
622 0 : CPLDebug("MBTILES", "Invalid character at byte %d", i);
623 0 : break;
624 : }
625 0 : c -= 32;
626 0 : ((GByte *)pszRow)[i] = c;
627 : }
628 :
629 0 : if (pszRow[i] == '\0')
630 : {
631 0 : char *pszEnd = pszRow + i;
632 :
633 0 : int iCol = 0;
634 0 : i = 0;
635 0 : int nKey = -1;
636 0 : while (pszRow + i < pszEnd)
637 : {
638 0 : int len = 0;
639 0 : unsigned int res = utf8decode(pszRow + i, pszEnd, &len);
640 :
641 : /* Invalid UTF8 ? */
642 0 : if (res > 127 && len == 1)
643 0 : break;
644 :
645 0 : if (iCol == nColInTile)
646 : {
647 0 : nKey = (int)res;
648 : // CPLDebug("MBTILES", "Key index = %d", nKey);
649 0 : break;
650 : }
651 0 : i += len;
652 0 : iCol++;
653 : }
654 :
655 : /* Find key */
656 0 : json_object *poKeys = CPL_json_object_object_get(jsobj, "keys");
657 0 : if (nKey >= 0 && poKeys != nullptr &&
658 0 : json_object_is_type(poKeys, json_type_array) &&
659 0 : nKey < static_cast<int>(json_object_array_length(poKeys)))
660 : {
661 0 : json_object *poKey = json_object_array_get_idx(poKeys, nKey);
662 0 : if (poKey != nullptr &&
663 0 : json_object_is_type(poKey, json_type_string))
664 : {
665 0 : pszKey = CPLStrdup(json_object_get_string(poKey));
666 : }
667 : }
668 : }
669 :
670 0 : CPLFree(pszRow);
671 : }
672 :
673 0 : end:
674 0 : if (jsobj)
675 0 : json_object_put(jsobj);
676 0 : VSIFree(pabyUncompressed);
677 0 : if (hFeat)
678 0 : OGR_F_Destroy(hFeat);
679 0 : if (hSQLLyr)
680 0 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
681 :
682 0 : return pszKey;
683 : }
684 :
685 : /************************************************************************/
686 : /* GetMetadataDomainList() */
687 : /************************************************************************/
688 :
689 0 : char **MBTilesBand::GetMetadataDomainList()
690 : {
691 0 : return CSLAddString(GDALPamRasterBand::GetMetadataDomainList(),
692 0 : "LocationInfo");
693 : }
694 :
695 : /************************************************************************/
696 : /* GetMetadataItem() */
697 : /************************************************************************/
698 :
699 42 : const char *MBTilesBand::GetMetadataItem(const char *pszName,
700 : const char *pszDomain)
701 : {
702 42 : MBTilesDataset *poGDS = (MBTilesDataset *)poDS;
703 :
704 : /* ==================================================================== */
705 : /* LocationInfo handling. */
706 : /* ==================================================================== */
707 42 : if (poGDS->hDS != nullptr && pszDomain != nullptr &&
708 20 : EQUAL(pszDomain, "LocationInfo") &&
709 0 : (STARTS_WITH_CI(pszName, "Pixel_") ||
710 0 : STARTS_WITH_CI(pszName, "GeoPixel_")))
711 : {
712 : int iPixel, iLine;
713 :
714 0 : if (!poGDS->HasNonEmptyGrids())
715 0 : return nullptr;
716 :
717 : /* --------------------------------------------------------------------
718 : */
719 : /* What pixel are we aiming at? */
720 : /* --------------------------------------------------------------------
721 : */
722 0 : if (STARTS_WITH_CI(pszName, "Pixel_"))
723 : {
724 0 : if (sscanf(pszName + 6, "%d_%d", &iPixel, &iLine) != 2)
725 0 : return nullptr;
726 : }
727 0 : else if (STARTS_WITH_CI(pszName, "GeoPixel_"))
728 : {
729 : double adfGeoTransform[6];
730 : double adfInvGeoTransform[6];
731 : double dfGeoX, dfGeoY;
732 :
733 0 : dfGeoX = CPLAtof(pszName + 9);
734 0 : const char *pszUnderscore = strchr(pszName + 9, '_');
735 0 : if (!pszUnderscore)
736 0 : return nullptr;
737 0 : dfGeoY = CPLAtof(pszUnderscore + 1);
738 :
739 0 : if (GetDataset() == nullptr)
740 0 : return nullptr;
741 :
742 0 : if (GetDataset()->GetGeoTransform(adfGeoTransform) != CE_None)
743 0 : return nullptr;
744 :
745 0 : if (!GDALInvGeoTransform(adfGeoTransform, adfInvGeoTransform))
746 0 : return nullptr;
747 :
748 0 : iPixel = (int)floor(adfInvGeoTransform[0] +
749 0 : adfInvGeoTransform[1] * dfGeoX +
750 0 : adfInvGeoTransform[2] * dfGeoY);
751 0 : iLine = (int)floor(adfInvGeoTransform[3] +
752 0 : adfInvGeoTransform[4] * dfGeoX +
753 0 : adfInvGeoTransform[5] * dfGeoY);
754 : }
755 : else
756 0 : return nullptr;
757 :
758 0 : if (iPixel < 0 || iLine < 0 || iPixel >= GetXSize() ||
759 0 : iLine >= GetYSize())
760 0 : return nullptr;
761 :
762 0 : char *pszKey = poGDS->FindKey(iPixel, iLine);
763 :
764 0 : if (pszKey != nullptr)
765 : {
766 : // CPLDebug("MBTILES", "Key = %s", pszKey);
767 :
768 0 : osLocationInfo = "<LocationInfo>";
769 0 : osLocationInfo += "<Key>";
770 : char *pszXMLEscaped =
771 0 : CPLEscapeString(pszKey, -1, CPLES_XML_BUT_QUOTES);
772 0 : osLocationInfo += pszXMLEscaped;
773 0 : CPLFree(pszXMLEscaped);
774 0 : osLocationInfo += "</Key>";
775 :
776 0 : if (GDALDatasetGetLayerByName(poGDS->hDS, "grid_data") != nullptr &&
777 0 : strchr(pszKey, '\'') == nullptr)
778 : {
779 : OGRLayerH hSQLLyr;
780 : OGRFeatureH hFeat;
781 :
782 : const char *pszSQL =
783 0 : CPLSPrintf("SELECT key_json FROM keymap WHERE "
784 : "key_name = '%s'",
785 : pszKey);
786 0 : CPLDebug("MBTILES", "%s", pszSQL);
787 : hSQLLyr =
788 0 : GDALDatasetExecuteSQL(poGDS->hDS, pszSQL, nullptr, nullptr);
789 0 : if (hSQLLyr)
790 : {
791 0 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
792 0 : if (hFeat != nullptr &&
793 0 : OGR_F_IsFieldSetAndNotNull(hFeat, 0))
794 : {
795 0 : const char *pszJSon = OGR_F_GetFieldAsString(hFeat, 0);
796 : // CPLDebug("MBTILES", "JSon = %s", pszJSon);
797 :
798 0 : osLocationInfo += "<JSon>";
799 : #ifdef CPLES_XML_BUT_QUOTES
800 : pszXMLEscaped =
801 0 : CPLEscapeString(pszJSon, -1, CPLES_XML_BUT_QUOTES);
802 : #else
803 : pszXMLEscaped = CPLEscapeString(pszJSon, -1, CPLES_XML);
804 : #endif
805 0 : osLocationInfo += pszXMLEscaped;
806 0 : CPLFree(pszXMLEscaped);
807 0 : osLocationInfo += "</JSon>";
808 : }
809 0 : OGR_F_Destroy(hFeat);
810 : }
811 0 : GDALDatasetReleaseResultSet(poGDS->hDS, hSQLLyr);
812 : }
813 :
814 0 : osLocationInfo += "</LocationInfo>";
815 :
816 0 : CPLFree(pszKey);
817 :
818 0 : return osLocationInfo.c_str();
819 : }
820 :
821 0 : return nullptr;
822 : }
823 : else
824 42 : return GDALPamRasterBand::GetMetadataItem(pszName, pszDomain);
825 : }
826 :
827 : /************************************************************************/
828 : /* GetOverviewCount() */
829 : /************************************************************************/
830 :
831 6 : int MBTilesBand::GetOverviewCount()
832 : {
833 6 : MBTilesDataset *poGDS = (MBTilesDataset *)poDS;
834 :
835 6 : if (poGDS->m_nOverviewCount >= 1)
836 5 : return poGDS->m_nOverviewCount;
837 : else
838 1 : return GDALPamRasterBand::GetOverviewCount();
839 : }
840 :
841 : /************************************************************************/
842 : /* GetOverview() */
843 : /************************************************************************/
844 :
845 7 : GDALRasterBand *MBTilesBand::GetOverview(int nLevel)
846 : {
847 7 : MBTilesDataset *poGDS = (MBTilesDataset *)poDS;
848 :
849 7 : if (poGDS->m_nOverviewCount == 0)
850 0 : return GDALPamRasterBand::GetOverview(nLevel);
851 :
852 7 : if (nLevel < 0 || nLevel >= poGDS->m_nOverviewCount)
853 0 : return nullptr;
854 :
855 7 : GDALDataset *poOvrDS = poGDS->m_papoOverviewDS[nLevel];
856 7 : if (poOvrDS)
857 7 : return poOvrDS->GetRasterBand(nBand);
858 : else
859 0 : return nullptr;
860 : }
861 :
862 : /************************************************************************/
863 : /* MBTilesDataset() */
864 : /************************************************************************/
865 :
866 276 : MBTilesDataset::MBTilesDataset()
867 : {
868 276 : m_bWriteBounds = true;
869 276 : m_bWriteMinMaxZoom = true;
870 276 : poMainDS = nullptr;
871 276 : m_nOverviewCount = 0;
872 276 : hDS = nullptr;
873 276 : m_papoOverviewDS = nullptr;
874 276 : bFetchedMetadata = false;
875 276 : nHasNonEmptyGrids = -1;
876 276 : hDB = nullptr;
877 276 : pMyVFS = nullptr;
878 :
879 276 : m_bGeoTransformValid = false;
880 276 : m_adfGeoTransform[0] = 0.0;
881 276 : m_adfGeoTransform[1] = 1.0;
882 276 : m_adfGeoTransform[2] = 0.0;
883 276 : m_adfGeoTransform[3] = 0.0;
884 276 : m_adfGeoTransform[4] = 0.0;
885 276 : m_adfGeoTransform[5] = 1.0;
886 276 : m_bInFlushCache = false;
887 :
888 276 : m_osRasterTable = "tiles";
889 276 : m_eTF = GPKG_TF_PNG;
890 :
891 276 : m_oSRS.SetAxisMappingStrategy(OAMS_TRADITIONAL_GIS_ORDER);
892 276 : m_oSRS.importFromEPSG(3857);
893 276 : }
894 :
895 : /************************************************************************/
896 : /* ~MBTilesDataset() */
897 : /************************************************************************/
898 :
899 552 : MBTilesDataset::~MBTilesDataset()
900 : {
901 : // Need to explicitly clear it before close hDS
902 276 : m_apoLayers.clear();
903 :
904 276 : FlushCache(true);
905 :
906 276 : if (poMainDS == nullptr)
907 : {
908 101 : if (m_papoOverviewDS)
909 : {
910 235 : for (int i = 0; i < m_nOverviewCount; i++)
911 175 : delete m_papoOverviewDS[i];
912 60 : CPLFree(m_papoOverviewDS);
913 : }
914 :
915 101 : if (hDS != nullptr)
916 : {
917 60 : GDALClose(hDS);
918 60 : hDB = nullptr;
919 : }
920 101 : if (hDB != nullptr)
921 : {
922 41 : sqlite3_close(hDB);
923 :
924 41 : if (pMyVFS)
925 : {
926 23 : sqlite3_vfs_unregister(pMyVFS);
927 23 : CPLFree(pMyVFS->pAppData);
928 23 : CPLFree(pMyVFS);
929 : }
930 : }
931 : }
932 :
933 276 : if (!m_osMetadataMemFilename.empty())
934 : {
935 32 : VSIUnlink(m_osMetadataMemFilename);
936 : }
937 552 : }
938 :
939 : /************************************************************************/
940 : /* IStartTransaction() */
941 : /************************************************************************/
942 :
943 24 : OGRErr MBTilesDataset::IStartTransaction()
944 : {
945 24 : char *pszErrMsg = nullptr;
946 24 : const int rc = sqlite3_exec(hDB, "BEGIN", nullptr, nullptr, &pszErrMsg);
947 24 : if (rc != SQLITE_OK)
948 : {
949 0 : CPLError(CE_Failure, CPLE_AppDefined, "%s transaction failed: %s",
950 : "BEGIN", pszErrMsg);
951 0 : sqlite3_free(pszErrMsg);
952 0 : return OGRERR_FAILURE;
953 : }
954 :
955 24 : return OGRERR_NONE;
956 : }
957 :
958 : /************************************************************************/
959 : /* ICommitTransaction() */
960 : /************************************************************************/
961 :
962 24 : OGRErr MBTilesDataset::ICommitTransaction()
963 : {
964 24 : char *pszErrMsg = nullptr;
965 24 : const int rc = sqlite3_exec(hDB, "COMMIT", nullptr, nullptr, &pszErrMsg);
966 24 : if (rc != SQLITE_OK)
967 : {
968 0 : CPLError(CE_Failure, CPLE_AppDefined, "%s transaction failed: %s",
969 : "COMMIT", pszErrMsg);
970 0 : sqlite3_free(pszErrMsg);
971 0 : return OGRERR_FAILURE;
972 : }
973 :
974 24 : return OGRERR_NONE;
975 : }
976 :
977 : /************************************************************************/
978 : /* ICanIWriteBlock() */
979 : /************************************************************************/
980 :
981 155 : bool MBTilesDataset::ICanIWriteBlock()
982 : {
983 155 : if (eAccess != GA_Update)
984 : {
985 0 : CPLError(
986 : CE_Failure, CPLE_NotSupported,
987 : "IWriteBlock() not supported on dataset opened in read-only mode");
988 0 : return false;
989 : }
990 :
991 155 : if (!m_bGeoTransformValid)
992 : {
993 0 : CPLError(CE_Failure, CPLE_NotSupported,
994 : "IWriteBlock() not supported if georeferencing not set");
995 0 : return false;
996 : }
997 155 : return true;
998 : }
999 :
1000 : /************************************************************************/
1001 : /* IFlushCacheWithErrCode() */
1002 : /************************************************************************/
1003 :
1004 3240 : CPLErr MBTilesDataset::IFlushCacheWithErrCode(bool bAtClosing)
1005 :
1006 : {
1007 3240 : if (m_bInFlushCache)
1008 2474 : return CE_None;
1009 766 : m_bInFlushCache = true;
1010 : // Short circuit GDALPamDataset to avoid serialization to .aux.xml
1011 766 : GDALDataset::FlushCache(bAtClosing);
1012 :
1013 766 : CPLErr eErr = FlushTiles();
1014 :
1015 766 : m_bInFlushCache = false;
1016 766 : return eErr;
1017 : }
1018 :
1019 : /************************************************************************/
1020 : /* ICanIWriteBlock() */
1021 : /************************************************************************/
1022 :
1023 546 : int MBTilesDataset::GetRowFromIntoTopConvention(int nRow)
1024 : {
1025 546 : return m_nTileMatrixHeight - 1 - nRow;
1026 : }
1027 :
1028 : /************************************************************************/
1029 : /* GetGeoTransform() */
1030 : /************************************************************************/
1031 :
1032 31 : CPLErr MBTilesDataset::GetGeoTransform(double *padfGeoTransform)
1033 : {
1034 31 : memcpy(padfGeoTransform, m_adfGeoTransform, 6 * sizeof(double));
1035 31 : return (m_bGeoTransformValid) ? CE_None : CE_Failure;
1036 : }
1037 :
1038 : /************************************************************************/
1039 : /* SphericalMercatorToLongLat() */
1040 : /************************************************************************/
1041 :
1042 69 : static void SphericalMercatorToLongLat(double *x, double *y)
1043 : {
1044 69 : double lng = *x / SPHERICAL_RADIUS / M_PI * 180;
1045 69 : double lat = 2 * (atan(exp(*y / SPHERICAL_RADIUS)) - M_PI / 4) / M_PI * 180;
1046 69 : *x = lng;
1047 69 : *y = lat;
1048 69 : }
1049 :
1050 : /************************************************************************/
1051 : /* LongLatToSphericalMercator() */
1052 : /************************************************************************/
1053 :
1054 112 : static void LongLatToSphericalMercator(double *x, double *y)
1055 : {
1056 112 : double X = SPHERICAL_RADIUS * (*x) / 180 * M_PI;
1057 112 : double Y = SPHERICAL_RADIUS * log(tan(M_PI / 4 + 0.5 * (*y) / 180 * M_PI));
1058 112 : *x = X;
1059 112 : *y = Y;
1060 112 : }
1061 :
1062 : /************************************************************************/
1063 : /* SetGeoTransform() */
1064 : /************************************************************************/
1065 :
1066 28 : CPLErr MBTilesDataset::SetGeoTransform(double *padfGeoTransform)
1067 : {
1068 28 : if (eAccess != GA_Update)
1069 : {
1070 1 : CPLError(CE_Failure, CPLE_NotSupported,
1071 : "SetGeoTransform() not supported on read-only dataset");
1072 1 : return CE_Failure;
1073 : }
1074 27 : if (m_bGeoTransformValid)
1075 : {
1076 1 : CPLError(CE_Failure, CPLE_NotSupported,
1077 : "Cannot modify geotransform once set");
1078 1 : return CE_Failure;
1079 : }
1080 26 : if (padfGeoTransform[2] != 0.0 || padfGeoTransform[4] != 0 ||
1081 26 : padfGeoTransform[5] > 0.0)
1082 : {
1083 1 : CPLError(CE_Failure, CPLE_NotSupported,
1084 : "Only north-up non rotated geotransform supported");
1085 1 : return CE_Failure;
1086 : }
1087 :
1088 25 : if (m_bWriteBounds)
1089 : {
1090 46 : CPLString osBounds(m_osBounds);
1091 23 : if (osBounds.empty())
1092 : {
1093 23 : double minx = padfGeoTransform[0];
1094 23 : double miny =
1095 23 : padfGeoTransform[3] + nRasterYSize * padfGeoTransform[5];
1096 23 : double maxx =
1097 23 : padfGeoTransform[0] + nRasterXSize * padfGeoTransform[1];
1098 23 : double maxy = padfGeoTransform[3];
1099 :
1100 23 : SphericalMercatorToLongLat(&minx, &miny);
1101 23 : SphericalMercatorToLongLat(&maxx, &maxy);
1102 23 : if (fabs(minx + 180) < 1e-7)
1103 : {
1104 3 : minx = -180.0;
1105 : }
1106 23 : if (fabs(maxx - 180) < 1e-7)
1107 : {
1108 3 : maxx = 180.0;
1109 : }
1110 :
1111 : // Clamp latitude so that when transformed back to EPSG:3857, we
1112 : // don't have too big northings
1113 23 : double tmpx = 0.0;
1114 23 : double ok_maxy = MAX_GM;
1115 23 : SphericalMercatorToLongLat(&tmpx, &ok_maxy);
1116 23 : if (maxy > ok_maxy)
1117 0 : maxy = ok_maxy;
1118 23 : if (miny < -ok_maxy)
1119 0 : miny = -ok_maxy;
1120 :
1121 23 : osBounds.Printf("%.17g,%.17g,%.17g,%.17g", minx, miny, maxx, maxy);
1122 : }
1123 :
1124 23 : char *pszSQL = sqlite3_mprintf(
1125 : "INSERT INTO metadata (name, value) VALUES ('bounds', '%q')",
1126 : osBounds.c_str());
1127 23 : sqlite3_exec(hDB, pszSQL, nullptr, nullptr, nullptr);
1128 23 : sqlite3_free(pszSQL);
1129 :
1130 23 : if (!m_osCenter.empty())
1131 : {
1132 0 : pszSQL = sqlite3_mprintf(
1133 : "INSERT INTO metadata (name, value) VALUES ('center', '%q')",
1134 : m_osCenter.c_str());
1135 0 : sqlite3_exec(hDB, pszSQL, nullptr, nullptr, nullptr);
1136 0 : sqlite3_free(pszSQL);
1137 : }
1138 : }
1139 :
1140 : int nBlockXSize;
1141 : int nBlockYSize;
1142 25 : GetRasterBand(1)->GetBlockSize(&nBlockXSize, &nBlockYSize);
1143 25 : const double dfPixelXSizeZoomLevel0 = 2 * MAX_GM / nBlockXSize;
1144 25 : const double dfPixelYSizeZoomLevel0 = 2 * MAX_GM / nBlockYSize;
1145 181 : for (m_nZoomLevel = 0; m_nZoomLevel < 25; m_nZoomLevel++)
1146 : {
1147 180 : double dfExpectedPixelXSize =
1148 180 : dfPixelXSizeZoomLevel0 / (1 << m_nZoomLevel);
1149 180 : double dfExpectedPixelYSize =
1150 180 : dfPixelYSizeZoomLevel0 / (1 << m_nZoomLevel);
1151 180 : if (fabs(padfGeoTransform[1] - dfExpectedPixelXSize) <
1152 180 : 1e-8 * dfExpectedPixelXSize &&
1153 24 : fabs(fabs(padfGeoTransform[5]) - dfExpectedPixelYSize) <
1154 24 : 1e-8 * dfExpectedPixelYSize)
1155 : {
1156 24 : break;
1157 : }
1158 : }
1159 25 : if (m_nZoomLevel == 25)
1160 : {
1161 1 : m_nZoomLevel = -1;
1162 1 : CPLError(CE_Failure, CPLE_NotSupported,
1163 : "Could not find an appropriate zoom level that matches raster "
1164 : "pixel size");
1165 1 : return CE_Failure;
1166 : }
1167 :
1168 24 : memcpy(m_adfGeoTransform, padfGeoTransform, 6 * sizeof(double));
1169 24 : m_bGeoTransformValid = true;
1170 :
1171 24 : return FinalizeRasterRegistration();
1172 : }
1173 :
1174 : /************************************************************************/
1175 : /* ComputeTileAndPixelShifts() */
1176 : /************************************************************************/
1177 :
1178 259 : void MBTilesDataset::ComputeTileAndPixelShifts()
1179 : {
1180 : int nTileWidth, nTileHeight;
1181 259 : GetRasterBand(1)->GetBlockSize(&nTileWidth, &nTileHeight);
1182 :
1183 : // Compute shift between GDAL origin and TileMatrixSet origin
1184 : // Caution this is in GeoPackage / WMTS convention ! That is upper-left
1185 : // corner
1186 259 : int nShiftXPixels = (int)floor(0.5 + (m_adfGeoTransform[0] - TMS_ORIGIN_X) /
1187 259 : m_adfGeoTransform[1]);
1188 259 : m_nShiftXTiles = (int)floor(1.0 * nShiftXPixels / nTileWidth);
1189 259 : m_nShiftXPixelsMod =
1190 259 : ((nShiftXPixels % nTileWidth) + nTileWidth) % nTileWidth;
1191 259 : int nShiftYPixels = (int)floor(0.5 + (m_adfGeoTransform[3] - TMS_ORIGIN_Y) /
1192 259 : m_adfGeoTransform[5]);
1193 259 : m_nShiftYTiles = (int)floor(1.0 * nShiftYPixels / nTileHeight);
1194 259 : m_nShiftYPixelsMod =
1195 259 : ((nShiftYPixels % nTileHeight) + nTileHeight) % nTileHeight;
1196 259 : }
1197 :
1198 : /************************************************************************/
1199 : /* FinalizeRasterRegistration() */
1200 : /************************************************************************/
1201 :
1202 24 : CPLErr MBTilesDataset::FinalizeRasterRegistration()
1203 : {
1204 24 : m_nTileMatrixWidth = (1 << m_nZoomLevel);
1205 24 : m_nTileMatrixHeight = (1 << m_nZoomLevel);
1206 :
1207 24 : ComputeTileAndPixelShifts();
1208 :
1209 24 : double dfGDALMinX = m_adfGeoTransform[0];
1210 24 : double dfGDALMinY =
1211 24 : m_adfGeoTransform[3] + nRasterYSize * m_adfGeoTransform[5];
1212 24 : double dfGDALMaxX =
1213 24 : m_adfGeoTransform[0] + nRasterXSize * m_adfGeoTransform[1];
1214 24 : double dfGDALMaxY = m_adfGeoTransform[3];
1215 :
1216 24 : m_nOverviewCount = m_nZoomLevel;
1217 48 : m_papoOverviewDS = (MBTilesDataset **)CPLCalloc(sizeof(MBTilesDataset *),
1218 24 : m_nOverviewCount);
1219 :
1220 24 : if (m_bWriteMinMaxZoom)
1221 : {
1222 24 : char *pszSQL = sqlite3_mprintf(
1223 : "INSERT INTO metadata (name, value) VALUES ('minzoom', '%d')",
1224 : m_nZoomLevel);
1225 24 : sqlite3_exec(hDB, pszSQL, nullptr, nullptr, nullptr);
1226 24 : sqlite3_free(pszSQL);
1227 24 : pszSQL = sqlite3_mprintf(
1228 : "INSERT INTO metadata (name, value) VALUES ('maxzoom', '%d')",
1229 : m_nZoomLevel);
1230 24 : sqlite3_exec(hDB, pszSQL, nullptr, nullptr, nullptr);
1231 24 : sqlite3_free(pszSQL);
1232 : }
1233 :
1234 155 : for (int i = 0; i < m_nOverviewCount; i++)
1235 : {
1236 131 : MBTilesDataset *poOvrDS = new MBTilesDataset();
1237 131 : poOvrDS->ShareLockWithParentDataset(this);
1238 : int nBlockSize;
1239 131 : GetRasterBand(1)->GetBlockSize(&nBlockSize, &nBlockSize);
1240 131 : poOvrDS->InitRaster(this, i, nBands, nBlockSize, dfGDALMinX, dfGDALMinY,
1241 : dfGDALMaxX, dfGDALMaxY);
1242 :
1243 131 : m_papoOverviewDS[m_nZoomLevel - 1 - i] = poOvrDS;
1244 : }
1245 :
1246 24 : return CE_None;
1247 : }
1248 :
1249 : /************************************************************************/
1250 : /* InitRaster() */
1251 : /************************************************************************/
1252 :
1253 235 : bool MBTilesDataset::InitRaster(MBTilesDataset *poParentDS, int nZoomLevel,
1254 : int nBandCount, int nTileSize,
1255 : double dfGDALMinX, double dfGDALMinY,
1256 : double dfGDALMaxX, double dfGDALMaxY)
1257 : {
1258 235 : m_nZoomLevel = nZoomLevel;
1259 235 : m_nTileMatrixWidth = 1 << nZoomLevel;
1260 235 : m_nTileMatrixHeight = 1 << nZoomLevel;
1261 :
1262 235 : const int nTileWidth = nTileSize;
1263 235 : const int nTileHeight = nTileSize;
1264 235 : const double dfPixelXSize = 2 * MAX_GM / nTileWidth / (1 << nZoomLevel);
1265 235 : const double dfPixelYSize = 2 * MAX_GM / nTileHeight / (1 << nZoomLevel);
1266 :
1267 235 : m_bGeoTransformValid = true;
1268 235 : m_adfGeoTransform[0] = dfGDALMinX;
1269 235 : m_adfGeoTransform[1] = dfPixelXSize;
1270 235 : m_adfGeoTransform[3] = dfGDALMaxY;
1271 235 : m_adfGeoTransform[5] = -dfPixelYSize;
1272 235 : double dfRasterXSize = 0.5 + (dfGDALMaxX - dfGDALMinX) / dfPixelXSize;
1273 235 : double dfRasterYSize = 0.5 + (dfGDALMaxY - dfGDALMinY) / dfPixelYSize;
1274 235 : if (dfRasterXSize > INT_MAX || dfRasterYSize > INT_MAX)
1275 0 : return false;
1276 235 : nRasterXSize = (int)dfRasterXSize;
1277 235 : nRasterYSize = (int)dfRasterYSize;
1278 :
1279 235 : m_pabyCachedTiles =
1280 235 : (GByte *)VSI_MALLOC3_VERBOSE(4 * 4, nTileWidth, nTileHeight);
1281 235 : if (m_pabyCachedTiles == nullptr)
1282 : {
1283 0 : return false;
1284 : }
1285 :
1286 235 : if (poParentDS)
1287 : {
1288 175 : eAccess = poParentDS->eAccess;
1289 : }
1290 :
1291 914 : for (int i = 1; i <= nBandCount; i++)
1292 679 : SetBand(i, new MBTilesBand(this, nTileSize));
1293 :
1294 235 : ComputeTileAndPixelShifts();
1295 :
1296 235 : GDALDataset::SetMetadataItem("INTERLEAVE", "PIXEL", "IMAGE_STRUCTURE");
1297 235 : GDALDataset::SetMetadataItem("ZOOM_LEVEL", CPLSPrintf("%d", m_nZoomLevel));
1298 :
1299 235 : if (poParentDS)
1300 : {
1301 175 : m_poParentDS = poParentDS;
1302 175 : poMainDS = poParentDS;
1303 175 : hDS = poParentDS->hDS;
1304 175 : hDB = poParentDS->hDB;
1305 175 : m_eTF = poParentDS->m_eTF;
1306 175 : m_nQuality = poParentDS->m_nQuality;
1307 175 : m_nZLevel = poParentDS->m_nZLevel;
1308 175 : m_bDither = poParentDS->m_bDither;
1309 175 : m_osWHERE = poParentDS->m_osWHERE;
1310 175 : SetDescription(CPLSPrintf("%s - zoom_level=%d",
1311 175 : poParentDS->GetDescription(), m_nZoomLevel));
1312 : }
1313 :
1314 235 : return true;
1315 : }
1316 :
1317 : /************************************************************************/
1318 : /* GetSpatialRef() */
1319 : /************************************************************************/
1320 :
1321 3 : const OGRSpatialReference *MBTilesDataset::GetSpatialRef() const
1322 : {
1323 3 : return &m_oSRS;
1324 : }
1325 :
1326 : /************************************************************************/
1327 : /* SetSpatialRef() */
1328 : /************************************************************************/
1329 :
1330 3 : CPLErr MBTilesDataset::SetSpatialRef(const OGRSpatialReference *poSRS)
1331 : {
1332 3 : if (eAccess != GA_Update)
1333 : {
1334 1 : CPLError(CE_Failure, CPLE_NotSupported,
1335 : "SetSpatialRef() not supported on read-only dataset");
1336 1 : return CE_Failure;
1337 : }
1338 :
1339 2 : if (poSRS == nullptr || poSRS->GetAuthorityName(nullptr) == nullptr ||
1340 1 : !EQUAL(poSRS->GetAuthorityName(nullptr), "EPSG") ||
1341 5 : poSRS->GetAuthorityCode(nullptr) == nullptr ||
1342 1 : !EQUAL(poSRS->GetAuthorityCode(nullptr), "3857"))
1343 : {
1344 1 : CPLError(CE_Failure, CPLE_NotSupported,
1345 : "Only EPSG:3857 supported on MBTiles dataset");
1346 1 : return CE_Failure;
1347 : }
1348 1 : return CE_None;
1349 : }
1350 :
1351 : /************************************************************************/
1352 : /* GetMetadataDomainList() */
1353 : /************************************************************************/
1354 :
1355 0 : char **MBTilesDataset::GetMetadataDomainList()
1356 : {
1357 0 : return BuildMetadataDomainList(GDALDataset::GetMetadataDomainList(), TRUE,
1358 0 : "", nullptr);
1359 : }
1360 :
1361 : /************************************************************************/
1362 : /* GetMetadata() */
1363 : /************************************************************************/
1364 :
1365 104 : char **MBTilesDataset::GetMetadata(const char *pszDomain)
1366 : {
1367 104 : if (hDS == nullptr || (pszDomain != nullptr && !EQUAL(pszDomain, "")))
1368 29 : return GDALPamDataset::GetMetadata(pszDomain);
1369 :
1370 75 : if (bFetchedMetadata)
1371 15 : return aosList.List();
1372 :
1373 60 : bFetchedMetadata = true;
1374 60 : aosList = CPLStringList(GDALPamDataset::GetMetadata(), FALSE);
1375 :
1376 60 : OGRLayerH hSQLLyr = GDALDatasetExecuteSQL(
1377 : hDS, "SELECT name, value FROM metadata WHERE name != 'json' LIMIT 1000",
1378 : nullptr, nullptr);
1379 60 : if (hSQLLyr == nullptr)
1380 0 : return nullptr;
1381 :
1382 60 : if (OGR_FD_GetFieldCount(OGR_L_GetLayerDefn(hSQLLyr)) != 2)
1383 : {
1384 0 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
1385 0 : return nullptr;
1386 : }
1387 :
1388 : OGRFeatureH hFeat;
1389 577 : while ((hFeat = OGR_L_GetNextFeature(hSQLLyr)) != nullptr)
1390 : {
1391 1034 : if (OGR_F_IsFieldSetAndNotNull(hFeat, 0) &&
1392 517 : OGR_F_IsFieldSetAndNotNull(hFeat, 1))
1393 : {
1394 1034 : CPLString osName = OGR_F_GetFieldAsString(hFeat, 0);
1395 1034 : CPLString osValue = OGR_F_GetFieldAsString(hFeat, 1);
1396 1034 : if (osName[0] != '\0' && !STARTS_WITH(osValue, "function(") &&
1397 517 : strstr(osValue, "<img ") == nullptr &&
1398 517 : strstr(osValue, "<p>") == nullptr &&
1399 1551 : strstr(osValue, "</p>") == nullptr &&
1400 517 : strstr(osValue, "<div") == nullptr)
1401 : {
1402 517 : aosList.AddNameValue(osName, osValue);
1403 : }
1404 : }
1405 517 : OGR_F_Destroy(hFeat);
1406 : }
1407 60 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
1408 :
1409 60 : return aosList.List();
1410 : }
1411 :
1412 : /************************************************************************/
1413 : /* GetMetadataItem() */
1414 : /************************************************************************/
1415 :
1416 89 : const char *MBTilesDataset::GetMetadataItem(const char *pszName,
1417 : const char *pszDomain)
1418 : {
1419 89 : if (pszDomain == nullptr || EQUAL(pszDomain, ""))
1420 : {
1421 83 : const char *pszValue = CSLFetchNameValue(GetMetadata(), pszName);
1422 83 : if (pszValue)
1423 65 : return pszValue;
1424 : }
1425 24 : return GDALPamDataset::GetMetadataItem(pszName, pszDomain);
1426 : }
1427 :
1428 : /************************************************************************/
1429 : /* GetLayer() */
1430 : /************************************************************************/
1431 :
1432 59 : OGRLayer *MBTilesDataset::GetLayer(int iLayer)
1433 :
1434 : {
1435 59 : if (iLayer < 0 || iLayer >= GetLayerCount())
1436 2 : return nullptr;
1437 57 : return m_apoLayers[iLayer].get();
1438 : }
1439 :
1440 : /************************************************************************/
1441 : /* MBTilesVectorLayer() */
1442 : /************************************************************************/
1443 :
1444 37 : MBTilesVectorLayer::MBTilesVectorLayer(
1445 : MBTilesDataset *poDS, const char *pszLayerName,
1446 : const CPLJSONObject &oFields, const CPLJSONArray &oAttributesFromTileStats,
1447 : bool bJsonField, double dfMinX, double dfMinY, double dfMaxX, double dfMaxY,
1448 37 : OGRwkbGeometryType eGeomType, bool bZoomLevelFromSpatialFilter)
1449 37 : : m_poDS(poDS), m_poFeatureDefn(new OGRFeatureDefn(pszLayerName)),
1450 74 : m_bJsonField(bJsonField)
1451 : {
1452 37 : SetDescription(pszLayerName);
1453 37 : m_poFeatureDefn->SetGeomType(eGeomType);
1454 37 : OGRSpatialReference *poSRS = new OGRSpatialReference();
1455 37 : poSRS->SetFromUserInput(SRS_EPSG_3857);
1456 37 : m_poFeatureDefn->GetGeomFieldDefn(0)->SetSpatialRef(poSRS);
1457 37 : poSRS->Release();
1458 37 : m_poFeatureDefn->Reference();
1459 :
1460 37 : if (m_bJsonField)
1461 : {
1462 2 : OGRFieldDefn oFieldDefnId("mvt_id", OFTInteger64);
1463 1 : m_poFeatureDefn->AddFieldDefn(&oFieldDefnId);
1464 : }
1465 : else
1466 : {
1467 36 : OGRMVTInitFields(m_poFeatureDefn, oFields, oAttributesFromTileStats);
1468 : }
1469 :
1470 37 : m_sExtent.MinX = dfMinX;
1471 37 : m_sExtent.MinY = dfMinY;
1472 37 : m_sExtent.MaxX = dfMaxX;
1473 37 : m_sExtent.MaxY = dfMaxY;
1474 :
1475 37 : m_nZoomLevel = m_poDS->m_nZoomLevel;
1476 37 : m_bZoomLevelAuto = bZoomLevelFromSpatialFilter;
1477 37 : MBTilesVectorLayer::SetSpatialFilter(nullptr);
1478 :
1479 : // If the metadata contains an empty fields object, this may be a sign
1480 : // that it doesn't know the schema. In that case check if a tile has
1481 : // attributes, and in that case create a json field.
1482 37 : if (!m_bJsonField && oFields.IsValid() && oFields.GetChildren().empty())
1483 : {
1484 8 : m_bJsonField = true;
1485 8 : OGRFeature *poSrcFeature = GetNextSrcFeature();
1486 8 : m_bJsonField = false;
1487 :
1488 8 : if (poSrcFeature)
1489 : {
1490 : // There is at least the mvt_id field
1491 8 : if (poSrcFeature->GetFieldCount() > 1)
1492 : {
1493 1 : m_bJsonField = true;
1494 : }
1495 8 : delete poSrcFeature;
1496 : }
1497 8 : MBTilesVectorLayer::ResetReading();
1498 : }
1499 :
1500 37 : if (m_bJsonField)
1501 : {
1502 4 : OGRFieldDefn oFieldDefn("json", OFTString);
1503 2 : m_poFeatureDefn->AddFieldDefn(&oFieldDefn);
1504 : }
1505 37 : }
1506 :
1507 : /************************************************************************/
1508 : /* ~MBTilesVectorLayer() */
1509 : /************************************************************************/
1510 :
1511 74 : MBTilesVectorLayer::~MBTilesVectorLayer()
1512 : {
1513 37 : m_poFeatureDefn->Release();
1514 37 : if (m_hTileIteratorLyr)
1515 15 : GDALDatasetReleaseResultSet(m_poDS->hDS, m_hTileIteratorLyr);
1516 37 : if (!m_osTmpFilename.empty())
1517 : {
1518 15 : VSIUnlink(m_osTmpFilename);
1519 : }
1520 37 : if (m_hTileDS)
1521 11 : GDALClose(m_hTileDS);
1522 74 : }
1523 :
1524 : /************************************************************************/
1525 : /* TestCapability() */
1526 : /************************************************************************/
1527 :
1528 35 : int MBTilesVectorLayer::TestCapability(const char *pszCap)
1529 : {
1530 35 : if (EQUAL(pszCap, OLCStringsAsUTF8) ||
1531 23 : EQUAL(pszCap, OLCFastSpatialFilter) || EQUAL(pszCap, OLCFastGetExtent))
1532 : {
1533 14 : return TRUE;
1534 : }
1535 21 : return FALSE;
1536 : }
1537 :
1538 : /************************************************************************/
1539 : /* GetExtent() */
1540 : /************************************************************************/
1541 :
1542 4 : OGRErr MBTilesVectorLayer::GetExtent(OGREnvelope *psExtent, int)
1543 : {
1544 4 : *psExtent = m_sExtent;
1545 4 : return OGRERR_NONE;
1546 : }
1547 :
1548 : /************************************************************************/
1549 : /* ResetReading() */
1550 : /************************************************************************/
1551 :
1552 109 : void MBTilesVectorLayer::ResetReading()
1553 : {
1554 109 : if (m_hTileDS)
1555 44 : GDALClose(m_hTileDS);
1556 109 : m_hTileDS = nullptr;
1557 109 : m_bEOF = false;
1558 109 : if (m_hTileIteratorLyr)
1559 94 : GDALDatasetReleaseResultSet(m_poDS->hDS, m_hTileIteratorLyr);
1560 109 : CPLString osSQL;
1561 : osSQL.Printf("SELECT tile_column, tile_row, tile_data FROM tiles "
1562 : "WHERE zoom_level = %d "
1563 : "AND tile_column BETWEEN %d AND %d "
1564 : "AND tile_row BETWEEN %d AND %d",
1565 : m_nZoomLevel, m_nFilterMinX, m_nFilterMaxX, m_nFilterMinY,
1566 109 : m_nFilterMaxY);
1567 109 : m_hTileIteratorLyr =
1568 109 : GDALDatasetExecuteSQL(m_poDS->hDS, osSQL.c_str(), nullptr, nullptr);
1569 109 : }
1570 :
1571 : /************************************************************************/
1572 : /* SetSpatialFilter() */
1573 : /************************************************************************/
1574 :
1575 62 : void MBTilesVectorLayer::SetSpatialFilter(OGRGeometry *poGeomIn)
1576 : {
1577 62 : OGRLayer::SetSpatialFilter(poGeomIn);
1578 :
1579 62 : if (m_poFilterGeom != nullptr && m_sFilterEnvelope.MinX <= -MAX_GM &&
1580 2 : m_sFilterEnvelope.MinY <= -MAX_GM && m_sFilterEnvelope.MaxX >= MAX_GM &&
1581 2 : m_sFilterEnvelope.MaxY >= MAX_GM)
1582 : {
1583 2 : if (m_bZoomLevelAuto)
1584 : {
1585 0 : m_nZoomLevel = m_poDS->m_nMinZoomLevel;
1586 : }
1587 2 : m_nFilterMinX = 0;
1588 2 : m_nFilterMinY = 0;
1589 2 : m_nFilterMaxX = (1 << m_nZoomLevel) - 1;
1590 2 : m_nFilterMaxY = (1 << m_nZoomLevel) - 1;
1591 : }
1592 60 : else if (m_poFilterGeom != nullptr &&
1593 6 : m_sFilterEnvelope.MinX >= -10 * MAX_GM &&
1594 6 : m_sFilterEnvelope.MinY >= -10 * MAX_GM &&
1595 6 : m_sFilterEnvelope.MaxX <= 10 * MAX_GM &&
1596 6 : m_sFilterEnvelope.MaxY <= 10 * MAX_GM)
1597 : {
1598 6 : if (m_bZoomLevelAuto)
1599 : {
1600 : double dfExtent =
1601 0 : std::min(m_sFilterEnvelope.MaxX - m_sFilterEnvelope.MinX,
1602 0 : m_sFilterEnvelope.MaxY - m_sFilterEnvelope.MinY);
1603 0 : m_nZoomLevel = std::max(
1604 0 : m_poDS->m_nMinZoomLevel,
1605 0 : std::min(static_cast<int>(0.5 + log(2 * MAX_GM / dfExtent) /
1606 : log(2.0)),
1607 0 : m_poDS->m_nZoomLevel));
1608 0 : CPLDebug("MBTILES", "Zoom level = %d", m_nZoomLevel);
1609 : }
1610 6 : const double dfTileDim = 2 * MAX_GM / (1 << m_nZoomLevel);
1611 6 : m_nFilterMinX =
1612 12 : std::max(0, static_cast<int>(floor(
1613 6 : (m_sFilterEnvelope.MinX + MAX_GM) / dfTileDim)));
1614 6 : m_nFilterMinY =
1615 12 : std::max(0, static_cast<int>(floor(
1616 6 : (m_sFilterEnvelope.MinY + MAX_GM) / dfTileDim)));
1617 6 : m_nFilterMaxX =
1618 12 : std::min(static_cast<int>(
1619 6 : ceil((m_sFilterEnvelope.MaxX + MAX_GM) / dfTileDim)),
1620 6 : (1 << m_nZoomLevel) - 1);
1621 6 : m_nFilterMaxY =
1622 12 : std::min(static_cast<int>(
1623 6 : ceil((m_sFilterEnvelope.MaxY + MAX_GM) / dfTileDim)),
1624 6 : (1 << m_nZoomLevel) - 1);
1625 : }
1626 : else
1627 : {
1628 54 : if (m_bZoomLevelAuto)
1629 : {
1630 0 : m_nZoomLevel = m_poDS->m_nZoomLevel;
1631 : }
1632 54 : m_nFilterMinX = 0;
1633 54 : m_nFilterMinY = 0;
1634 54 : m_nFilterMaxX = (1 << m_nZoomLevel) - 1;
1635 54 : m_nFilterMaxY = (1 << m_nZoomLevel) - 1;
1636 : }
1637 62 : }
1638 :
1639 : /************************************************************************/
1640 : /* GetNextFeature() */
1641 : /************************************************************************/
1642 :
1643 151 : OGRFeature *MBTilesVectorLayer::GetNextFeature()
1644 : {
1645 : while (true)
1646 : {
1647 151 : OGRFeature *poFeature = GetNextRawFeature();
1648 151 : if (poFeature == nullptr)
1649 29 : return nullptr;
1650 :
1651 286 : if ((m_poFilterGeom == nullptr ||
1652 231 : FilterGeometry(poFeature->GetGeometryRef())) &&
1653 109 : (m_poAttrQuery == nullptr || m_poAttrQuery->Evaluate(poFeature)))
1654 : {
1655 81 : return poFeature;
1656 : }
1657 :
1658 41 : delete poFeature;
1659 41 : }
1660 : }
1661 :
1662 : /************************************************************************/
1663 : /* GetFeatureCount() */
1664 : /************************************************************************/
1665 :
1666 15 : GIntBig MBTilesVectorLayer::GetFeatureCount(int bForce)
1667 : {
1668 15 : if (m_poFilterGeom == nullptr && m_poAttrQuery == nullptr)
1669 : {
1670 9 : if (m_nFeatureCount < 0)
1671 : {
1672 1 : m_nFeatureCount = 0;
1673 1 : ResetReading();
1674 5 : while (m_hTileIteratorLyr != nullptr)
1675 : {
1676 5 : OGRFeatureH hFeat = OGR_L_GetNextFeature(m_hTileIteratorLyr);
1677 5 : if (hFeat == nullptr)
1678 : {
1679 1 : break;
1680 : }
1681 4 : m_nX = OGR_F_GetFieldAsInteger(hFeat, 0);
1682 : // MBTiles y origin is bottom based, whereas MVT directory
1683 : // is top based
1684 4 : m_nY =
1685 4 : (1 << m_nZoomLevel) - 1 - OGR_F_GetFieldAsInteger(hFeat, 1);
1686 4 : int nDataSize = 0;
1687 4 : GByte *pabyData = OGR_F_GetFieldAsBinary(hFeat, 2, &nDataSize);
1688 4 : GByte *pabyDataDup = static_cast<GByte *>(CPLMalloc(nDataSize));
1689 4 : memcpy(pabyDataDup, pabyData, nDataSize);
1690 4 : OGR_F_Destroy(hFeat);
1691 :
1692 4 : if (!m_osTmpFilename.empty())
1693 : {
1694 3 : VSIUnlink(m_osTmpFilename);
1695 : }
1696 : m_osTmpFilename = VSIMemGenerateHiddenFilename(
1697 4 : CPLSPrintf("mvt_%d_%d.pbf", m_nX, m_nY));
1698 4 : VSIFCloseL(VSIFileFromMemBuffer(m_osTmpFilename, pabyDataDup,
1699 : nDataSize, true));
1700 :
1701 4 : const char *l_apszAllowedDrivers[] = {"MVT", nullptr};
1702 4 : if (m_hTileDS)
1703 0 : GDALClose(m_hTileDS);
1704 4 : char **papszOpenOptions = nullptr;
1705 : papszOpenOptions =
1706 4 : CSLSetNameValue(papszOpenOptions, "METADATA_FILE",
1707 4 : m_poDS->m_osMetadataMemFilename.c_str());
1708 4 : m_hTileDS =
1709 4 : GDALOpenEx(("MVT:" + m_osTmpFilename).c_str(),
1710 : GDAL_OF_VECTOR | GDAL_OF_INTERNAL,
1711 : l_apszAllowedDrivers, papszOpenOptions, nullptr);
1712 4 : CSLDestroy(papszOpenOptions);
1713 4 : if (m_hTileDS)
1714 : {
1715 : OGRLayerH hLayer =
1716 4 : GDALDatasetGetLayerByName(m_hTileDS, GetName());
1717 4 : if (hLayer)
1718 : {
1719 4 : m_nFeatureCount += OGR_L_GetFeatureCount(hLayer, true);
1720 : }
1721 4 : GDALClose(m_hTileDS);
1722 4 : m_hTileDS = nullptr;
1723 : }
1724 : }
1725 1 : ResetReading();
1726 : }
1727 9 : return m_nFeatureCount;
1728 : }
1729 6 : return OGRLayer::GetFeatureCount(bForce);
1730 : }
1731 :
1732 : /************************************************************************/
1733 : /* GetNextSrcFeature() */
1734 : /************************************************************************/
1735 :
1736 159 : OGRFeature *MBTilesVectorLayer::GetNextSrcFeature()
1737 : {
1738 159 : if (m_bEOF)
1739 : {
1740 1 : return nullptr;
1741 : }
1742 158 : if (m_hTileIteratorLyr == nullptr)
1743 : {
1744 14 : ResetReading();
1745 14 : if (m_hTileIteratorLyr == nullptr)
1746 : {
1747 0 : return nullptr;
1748 : }
1749 : }
1750 :
1751 158 : OGRFeatureH hTileFeat = nullptr;
1752 261 : if (m_hTileDS == nullptr ||
1753 103 : (hTileFeat = OGR_L_GetNextFeature(
1754 103 : GDALDatasetGetLayerByName(m_hTileDS, GetName()))) == nullptr)
1755 : {
1756 : while (true)
1757 : {
1758 167 : OGRFeatureH hFeat = OGR_L_GetNextFeature(m_hTileIteratorLyr);
1759 167 : if (hFeat == nullptr)
1760 : {
1761 28 : m_bEOF = true;
1762 28 : return nullptr;
1763 : }
1764 139 : m_nX = OGR_F_GetFieldAsInteger(hFeat, 0);
1765 : // MBTiles y origin is bottom based, whereas MVT directory
1766 : // is top based
1767 139 : m_nY = (1 << m_nZoomLevel) - 1 - OGR_F_GetFieldAsInteger(hFeat, 1);
1768 139 : CPLDebug("MBTiles", "X=%d, Y=%d", m_nX, m_nY);
1769 :
1770 139 : int nDataSize = 0;
1771 139 : GByte *pabyData = OGR_F_GetFieldAsBinary(hFeat, 2, &nDataSize);
1772 139 : GByte *pabyDataDup = static_cast<GByte *>(CPLMalloc(nDataSize));
1773 139 : memcpy(pabyDataDup, pabyData, nDataSize);
1774 139 : OGR_F_Destroy(hFeat);
1775 :
1776 139 : if (!m_osTmpFilename.empty())
1777 : {
1778 125 : VSIUnlink(m_osTmpFilename);
1779 : }
1780 : m_osTmpFilename = VSIMemGenerateHiddenFilename(
1781 139 : CPLSPrintf("mvt_%d_%d.pbf", m_nX, m_nY));
1782 139 : VSIFCloseL(VSIFileFromMemBuffer(m_osTmpFilename, pabyDataDup,
1783 : nDataSize, true));
1784 :
1785 139 : const char *l_apszAllowedDrivers[] = {"MVT", nullptr};
1786 139 : if (m_hTileDS)
1787 72 : GDALClose(m_hTileDS);
1788 139 : char **papszOpenOptions = nullptr;
1789 : papszOpenOptions =
1790 139 : CSLSetNameValue(papszOpenOptions, "X", CPLSPrintf("%d", m_nX));
1791 : papszOpenOptions =
1792 139 : CSLSetNameValue(papszOpenOptions, "Y", CPLSPrintf("%d", m_nY));
1793 139 : papszOpenOptions = CSLSetNameValue(papszOpenOptions, "Z",
1794 : CPLSPrintf("%d", m_nZoomLevel));
1795 139 : papszOpenOptions = CSLSetNameValue(
1796 : papszOpenOptions, "METADATA_FILE",
1797 139 : m_bJsonField ? "" : m_poDS->m_osMetadataMemFilename.c_str());
1798 139 : if (!m_poDS->m_osClip.empty())
1799 : {
1800 : papszOpenOptions =
1801 3 : CSLSetNameValue(papszOpenOptions, "CLIP", m_poDS->m_osClip);
1802 : }
1803 139 : m_hTileDS =
1804 139 : GDALOpenEx(("MVT:" + m_osTmpFilename).c_str(),
1805 : GDAL_OF_VECTOR | GDAL_OF_INTERNAL,
1806 : l_apszAllowedDrivers, papszOpenOptions, nullptr);
1807 139 : CSLDestroy(papszOpenOptions);
1808 139 : if (m_hTileDS)
1809 : {
1810 139 : if (GDALDatasetGetLayerByName(m_hTileDS, GetName()))
1811 : {
1812 131 : hTileFeat = OGR_L_GetNextFeature(
1813 131 : GDALDatasetGetLayerByName(m_hTileDS, GetName()));
1814 131 : if (hTileFeat)
1815 127 : break;
1816 : }
1817 12 : GDALClose(m_hTileDS);
1818 12 : m_hTileDS = nullptr;
1819 : }
1820 12 : }
1821 : }
1822 :
1823 130 : return reinterpret_cast<OGRFeature *>(hTileFeat);
1824 : }
1825 :
1826 : /************************************************************************/
1827 : /* CreateFeatureFrom() */
1828 : /************************************************************************/
1829 :
1830 124 : OGRFeature *MBTilesVectorLayer::CreateFeatureFrom(OGRFeature *poSrcFeature)
1831 : {
1832 :
1833 124 : return OGRMVTCreateFeatureFrom(poSrcFeature, m_poFeatureDefn, m_bJsonField,
1834 248 : GetSpatialRef());
1835 : }
1836 :
1837 : /************************************************************************/
1838 : /* GetNextRawFeature() */
1839 : /************************************************************************/
1840 :
1841 151 : OGRFeature *MBTilesVectorLayer::GetNextRawFeature()
1842 : {
1843 151 : OGRFeature *poSrcFeat = GetNextSrcFeature();
1844 151 : if (poSrcFeat == nullptr)
1845 29 : return nullptr;
1846 :
1847 122 : const GIntBig nFIDBase =
1848 122 : (static_cast<GIntBig>(m_nY) << m_nZoomLevel) | m_nX;
1849 122 : OGRFeature *poFeature = CreateFeatureFrom(poSrcFeat);
1850 122 : poFeature->SetFID((poSrcFeat->GetFID() << (2 * m_nZoomLevel)) | nFIDBase);
1851 122 : delete poSrcFeat;
1852 :
1853 122 : return poFeature;
1854 : }
1855 :
1856 : /************************************************************************/
1857 : /* GetFeature() */
1858 : /************************************************************************/
1859 :
1860 5 : OGRFeature *MBTilesVectorLayer::GetFeature(GIntBig nFID)
1861 : {
1862 5 : const int nZ = m_nZoomLevel;
1863 5 : const int nX = static_cast<int>(nFID & ((1 << nZ) - 1));
1864 5 : const int nY = static_cast<int>((nFID >> nZ) & ((1 << nZ) - 1));
1865 5 : const GIntBig nTileFID = nFID >> (2 * nZ);
1866 :
1867 10 : CPLString osSQL;
1868 : osSQL.Printf("SELECT tile_data FROM tiles "
1869 : "WHERE zoom_level = %d AND "
1870 : "tile_column = %d AND tile_row = %d",
1871 5 : m_nZoomLevel, nX, (1 << nZ) - 1 - nY);
1872 : auto hSQLLyr =
1873 5 : GDALDatasetExecuteSQL(m_poDS->hDS, osSQL.c_str(), nullptr, nullptr);
1874 5 : if (hSQLLyr == nullptr)
1875 0 : return nullptr;
1876 5 : auto hFeat = OGR_L_GetNextFeature(hSQLLyr);
1877 5 : if (hFeat == nullptr)
1878 : {
1879 0 : GDALDatasetReleaseResultSet(m_poDS->hDS, hSQLLyr);
1880 0 : return nullptr;
1881 : }
1882 5 : int nDataSize = 0;
1883 5 : GByte *pabyData = OGR_F_GetFieldAsBinary(hFeat, 0, &nDataSize);
1884 5 : GByte *pabyDataDup = static_cast<GByte *>(CPLMalloc(nDataSize));
1885 5 : memcpy(pabyDataDup, pabyData, nDataSize);
1886 5 : OGR_F_Destroy(hFeat);
1887 5 : GDALDatasetReleaseResultSet(m_poDS->hDS, hSQLLyr);
1888 :
1889 : const CPLString osTmpFilename = VSIMemGenerateHiddenFilename(
1890 5 : CPLSPrintf("mvt_get_feature_%d_%d.pbf", m_nX, m_nY));
1891 5 : VSIFCloseL(
1892 : VSIFileFromMemBuffer(osTmpFilename, pabyDataDup, nDataSize, true));
1893 :
1894 5 : const char *l_apszAllowedDrivers[] = {"MVT", nullptr};
1895 5 : char **papszOpenOptions = nullptr;
1896 : papszOpenOptions =
1897 5 : CSLSetNameValue(papszOpenOptions, "X", CPLSPrintf("%d", nX));
1898 : papszOpenOptions =
1899 5 : CSLSetNameValue(papszOpenOptions, "Y", CPLSPrintf("%d", nY));
1900 : papszOpenOptions =
1901 5 : CSLSetNameValue(papszOpenOptions, "Z", CPLSPrintf("%d", m_nZoomLevel));
1902 5 : papszOpenOptions = CSLSetNameValue(
1903 : papszOpenOptions, "METADATA_FILE",
1904 5 : m_bJsonField ? "" : m_poDS->m_osMetadataMemFilename.c_str());
1905 5 : if (!m_poDS->m_osClip.empty())
1906 : {
1907 : papszOpenOptions =
1908 0 : CSLSetNameValue(papszOpenOptions, "CLIP", m_poDS->m_osClip);
1909 : }
1910 5 : auto hTileDS = GDALOpenEx(("MVT:" + osTmpFilename).c_str(),
1911 : GDAL_OF_VECTOR | GDAL_OF_INTERNAL,
1912 : l_apszAllowedDrivers, papszOpenOptions, nullptr);
1913 5 : CSLDestroy(papszOpenOptions);
1914 :
1915 5 : OGRFeature *poFeature = nullptr;
1916 5 : if (hTileDS)
1917 : {
1918 5 : OGRLayerH hLayer = GDALDatasetGetLayerByName(hTileDS, GetName());
1919 5 : if (hLayer)
1920 : {
1921 : OGRFeature *poUnderlyingFeature = reinterpret_cast<OGRFeature *>(
1922 5 : OGR_L_GetFeature(hLayer, nTileFID));
1923 5 : if (poUnderlyingFeature)
1924 : {
1925 2 : poFeature = CreateFeatureFrom(poUnderlyingFeature);
1926 2 : poFeature->SetFID(nFID);
1927 : }
1928 5 : delete poUnderlyingFeature;
1929 : }
1930 : }
1931 5 : GDALClose(hTileDS);
1932 :
1933 5 : VSIUnlink(osTmpFilename);
1934 :
1935 5 : return poFeature;
1936 : }
1937 :
1938 : /************************************************************************/
1939 : /* InitVector() */
1940 : /************************************************************************/
1941 :
1942 32 : void MBTilesDataset::InitVector(double dfMinX, double dfMinY, double dfMaxX,
1943 : double dfMaxY, bool bZoomLevelFromSpatialFilter,
1944 : bool bJsonField)
1945 : {
1946 32 : const char *pszSQL = "SELECT value FROM metadata WHERE name = 'json'";
1947 32 : CPLDebug("MBTILES", "%s", pszSQL);
1948 64 : CPLJSONDocument oJsonDoc;
1949 64 : CPLJSONDocument oDoc;
1950 32 : auto hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, nullptr, nullptr);
1951 32 : if (hSQLLyr)
1952 : {
1953 32 : auto hFeat = OGR_L_GetNextFeature(hSQLLyr);
1954 32 : if (hFeat)
1955 : {
1956 32 : auto pszJson = OGR_F_GetFieldAsString(hFeat, 0);
1957 32 : oDoc.GetRoot().Add("json", pszJson);
1958 32 : CPL_IGNORE_RET_VAL(
1959 32 : oJsonDoc.LoadMemory(reinterpret_cast<const GByte *>(pszJson)));
1960 32 : OGR_F_Destroy(hFeat);
1961 : }
1962 32 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
1963 : }
1964 :
1965 : m_osMetadataMemFilename =
1966 32 : VSIMemGenerateHiddenFilename("mbtiles_metadata.json");
1967 32 : oDoc.Save(m_osMetadataMemFilename);
1968 :
1969 64 : CPLJSONArray oVectorLayers;
1970 32 : oVectorLayers.Deinit();
1971 :
1972 64 : CPLJSONArray oTileStatLayers;
1973 32 : oTileStatLayers.Deinit();
1974 :
1975 32 : oVectorLayers = oJsonDoc.GetRoot().GetArray("vector_layers");
1976 :
1977 32 : oTileStatLayers = oJsonDoc.GetRoot().GetArray("tilestats/layers");
1978 :
1979 69 : for (int i = 0; i < oVectorLayers.Size(); i++)
1980 : {
1981 111 : CPLJSONObject oId = oVectorLayers[i].GetObj("id");
1982 37 : if (oId.IsValid() && oId.GetType() == CPLJSONObject::Type::String)
1983 : {
1984 37 : OGRwkbGeometryType eGeomType = wkbUnknown;
1985 37 : if (oTileStatLayers.IsValid())
1986 : {
1987 36 : eGeomType = OGRMVTFindGeomTypeFromTileStat(
1988 72 : oTileStatLayers, oId.ToString().c_str());
1989 : }
1990 :
1991 111 : CPLJSONObject oFields = oVectorLayers[i].GetObj("fields");
1992 : CPLJSONArray oAttributesFromTileStats =
1993 : OGRMVTFindAttributesFromTileStat(oTileStatLayers,
1994 74 : oId.ToString().c_str());
1995 37 : m_apoLayers.push_back(
1996 74 : std::unique_ptr<OGRLayer>(new MBTilesVectorLayer(
1997 74 : this, oId.ToString().c_str(), oFields,
1998 : oAttributesFromTileStats, bJsonField, dfMinX, dfMinY,
1999 37 : dfMaxX, dfMaxY, eGeomType, bZoomLevelFromSpatialFilter)));
2000 : }
2001 : }
2002 32 : }
2003 :
2004 : /************************************************************************/
2005 : /* Identify() */
2006 : /************************************************************************/
2007 :
2008 62949 : int MBTilesDataset::Identify(GDALOpenInfo *poOpenInfo)
2009 : {
2010 : #ifdef ENABLE_SQL_SQLITE_FORMAT
2011 62949 : if (poOpenInfo->pabyHeader &&
2012 11930 : STARTS_WITH((const char *)poOpenInfo->pabyHeader, "-- SQL MBTILES"))
2013 : {
2014 2 : return TRUE;
2015 : }
2016 : #endif
2017 :
2018 62947 : if ((poOpenInfo->IsExtensionEqualToCI("MBTILES") ||
2019 : // Allow direct Amazon S3 signed URLs that contains .mbtiles in the
2020 : // middle of the URL
2021 62658 : strstr(poOpenInfo->pszFilename, ".mbtiles") != nullptr) &&
2022 125752 : poOpenInfo->nHeaderBytes >= 1024 && poOpenInfo->pabyHeader &&
2023 153 : STARTS_WITH_CI((const char *)poOpenInfo->pabyHeader, "SQLite Format 3"))
2024 : {
2025 153 : return TRUE;
2026 : }
2027 :
2028 62788 : return FALSE;
2029 : }
2030 :
2031 : /************************************************************************/
2032 : /* MBTilesGetMinMaxZoomLevel() */
2033 : /************************************************************************/
2034 :
2035 77 : static int MBTilesGetMinMaxZoomLevel(GDALDatasetH hDS, int bHasMap,
2036 : int &nMinLevel, int &nMaxLevel)
2037 : {
2038 : OGRLayerH hSQLLyr;
2039 : OGRFeatureH hFeat;
2040 77 : int bHasMinMaxLevel = FALSE;
2041 :
2042 77 : const char *pszSQL =
2043 : "SELECT value FROM metadata WHERE name = 'minzoom' UNION ALL "
2044 : "SELECT value FROM metadata WHERE name = 'maxzoom'";
2045 77 : CPLDebug("MBTILES", "%s", pszSQL);
2046 77 : hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, nullptr, nullptr);
2047 77 : if (hSQLLyr)
2048 : {
2049 77 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
2050 77 : if (hFeat)
2051 : {
2052 73 : int bHasMinLevel = FALSE;
2053 73 : if (OGR_F_IsFieldSetAndNotNull(hFeat, 0))
2054 : {
2055 73 : nMinLevel = OGR_F_GetFieldAsInteger(hFeat, 0);
2056 73 : bHasMinLevel = TRUE;
2057 : }
2058 73 : OGR_F_Destroy(hFeat);
2059 :
2060 73 : if (bHasMinLevel)
2061 : {
2062 73 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
2063 73 : if (hFeat)
2064 : {
2065 73 : if (OGR_F_IsFieldSetAndNotNull(hFeat, 0))
2066 : {
2067 73 : nMaxLevel = OGR_F_GetFieldAsInteger(hFeat, 0);
2068 73 : bHasMinMaxLevel = TRUE;
2069 : }
2070 73 : OGR_F_Destroy(hFeat);
2071 : }
2072 : }
2073 : }
2074 :
2075 77 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2076 : }
2077 :
2078 77 : if (!bHasMinMaxLevel)
2079 : {
2080 : #define OPTIMIZED_FOR_VSICURL
2081 : #ifdef OPTIMIZED_FOR_VSICURL
2082 : int iLevel;
2083 40 : for (iLevel = 0; nMinLevel < 0 && iLevel <= 32; iLevel++)
2084 : {
2085 36 : pszSQL = CPLSPrintf(
2086 : "SELECT zoom_level FROM %s WHERE zoom_level = %d LIMIT 1",
2087 : (bHasMap) ? "map" : "tiles", iLevel);
2088 36 : CPLDebug("MBTILES", "%s", pszSQL);
2089 36 : hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, nullptr, nullptr);
2090 36 : if (hSQLLyr)
2091 : {
2092 36 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
2093 36 : if (hFeat)
2094 : {
2095 3 : nMinLevel = iLevel;
2096 3 : OGR_F_Destroy(hFeat);
2097 : }
2098 36 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2099 : }
2100 : }
2101 :
2102 4 : if (nMinLevel < 0)
2103 1 : return FALSE;
2104 :
2105 99 : for (iLevel = 32; nMaxLevel < 0 && iLevel >= nMinLevel; iLevel--)
2106 : {
2107 96 : pszSQL = CPLSPrintf(
2108 : "SELECT zoom_level FROM %s WHERE zoom_level = %d LIMIT 1",
2109 : (bHasMap) ? "map" : "tiles", iLevel);
2110 96 : CPLDebug("MBTILES", "%s", pszSQL);
2111 96 : hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, nullptr, nullptr);
2112 96 : if (hSQLLyr)
2113 : {
2114 96 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
2115 96 : if (hFeat)
2116 : {
2117 3 : nMaxLevel = iLevel;
2118 3 : bHasMinMaxLevel = TRUE;
2119 3 : OGR_F_Destroy(hFeat);
2120 : }
2121 96 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2122 : }
2123 : }
2124 : #else
2125 : pszSQL = "SELECT min(zoom_level), max(zoom_level) FROM tiles";
2126 : CPLDebug("MBTILES", "%s", pszSQL);
2127 : hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, NULL, nullptr);
2128 : if (hSQLLyr == NULL)
2129 : {
2130 : return FALSE;
2131 : }
2132 :
2133 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
2134 : if (hFeat == NULL)
2135 : {
2136 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2137 : return FALSE;
2138 : }
2139 :
2140 : if (OGR_F_IsFieldSetAndNotNull(hFeat, 0) &&
2141 : OGR_F_IsFieldSetAndNotNull(hFeat, 1))
2142 : {
2143 : nMinLevel = OGR_F_GetFieldAsInteger(hFeat, 0);
2144 : nMaxLevel = OGR_F_GetFieldAsInteger(hFeat, 1);
2145 : bHasMinMaxLevel = TRUE;
2146 : }
2147 :
2148 : OGR_F_Destroy(hFeat);
2149 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2150 : #endif
2151 : }
2152 :
2153 76 : return bHasMinMaxLevel;
2154 : }
2155 :
2156 : /************************************************************************/
2157 : /* MBTilesTileCoordToWorldCoord() */
2158 : /************************************************************************/
2159 :
2160 16 : static double MBTilesTileCoordToWorldCoord(double dfTileCoord, int nZoomLevel)
2161 : {
2162 16 : return -MAX_GM + 2 * MAX_GM * (dfTileCoord / (1 << nZoomLevel));
2163 : }
2164 :
2165 : /************************************************************************/
2166 : /* MBTilesWorldCoordToTileCoord() */
2167 : /************************************************************************/
2168 :
2169 240 : static double MBTilesWorldCoordToTileCoord(double dfWorldCoord, int nZoomLevel)
2170 : {
2171 240 : return (dfWorldCoord + MAX_GM) / (2 * MAX_GM) * (1 << nZoomLevel);
2172 : }
2173 :
2174 : /************************************************************************/
2175 : /* MBTilesGetBounds() */
2176 : /************************************************************************/
2177 :
2178 76 : static bool MBTilesGetBounds(GDALDatasetH hDS, bool bUseBounds, int nMaxLevel,
2179 : double &minX, double &minY, double &maxX,
2180 : double &maxY)
2181 : {
2182 76 : bool bHasBounds = false;
2183 : OGRLayerH hSQLLyr;
2184 : OGRFeatureH hFeat;
2185 :
2186 76 : if (bUseBounds)
2187 : {
2188 75 : const char *pszSQL = "SELECT value FROM metadata WHERE name = 'bounds'";
2189 75 : CPLDebug("MBTILES", "%s", pszSQL);
2190 75 : hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, nullptr, nullptr);
2191 75 : if (hSQLLyr)
2192 : {
2193 75 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
2194 75 : if (hFeat != nullptr)
2195 : {
2196 73 : const char *pszBounds = OGR_F_GetFieldAsString(hFeat, 0);
2197 73 : char **papszTokens = CSLTokenizeString2(pszBounds, ",", 0);
2198 73 : if (CSLCount(papszTokens) != 4 ||
2199 72 : fabs(CPLAtof(papszTokens[0])) > 180 ||
2200 56 : fabs(CPLAtof(papszTokens[1])) >= 89.99 ||
2201 56 : fabs(CPLAtof(papszTokens[2])) > 180 ||
2202 56 : fabs(CPLAtof(papszTokens[3])) >= 89.99 ||
2203 201 : CPLAtof(papszTokens[0]) > CPLAtof(papszTokens[2]) ||
2204 56 : CPLAtof(papszTokens[1]) > CPLAtof(papszTokens[3]))
2205 : {
2206 17 : CPLError(CE_Warning, CPLE_AppDefined,
2207 : "Invalid value for 'bounds' metadata. Ignoring it "
2208 : "and fall back to present tile extent");
2209 : }
2210 : else
2211 : {
2212 56 : minX = CPLAtof(papszTokens[0]);
2213 56 : minY = CPLAtof(papszTokens[1]);
2214 56 : maxX = CPLAtof(papszTokens[2]);
2215 56 : maxY = CPLAtof(papszTokens[3]);
2216 56 : LongLatToSphericalMercator(&minX, &minY);
2217 56 : LongLatToSphericalMercator(&maxX, &maxY);
2218 :
2219 : // Clamp northings
2220 56 : if (maxY > MAX_GM)
2221 0 : maxY = MAX_GM;
2222 56 : if (minY < -MAX_GM)
2223 8 : minY = -MAX_GM;
2224 :
2225 56 : bHasBounds = true;
2226 : }
2227 :
2228 73 : CSLDestroy(papszTokens);
2229 :
2230 73 : OGR_F_Destroy(hFeat);
2231 : }
2232 75 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2233 : }
2234 : }
2235 :
2236 76 : if (!bHasBounds)
2237 : {
2238 : const char *pszSQL =
2239 20 : CPLSPrintf("SELECT min(tile_column), max(tile_column), "
2240 : "min(tile_row), max(tile_row) FROM tiles "
2241 : "WHERE zoom_level = %d",
2242 : nMaxLevel);
2243 20 : CPLDebug("MBTILES", "%s", pszSQL);
2244 20 : hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, nullptr, nullptr);
2245 20 : if (hSQLLyr == nullptr)
2246 : {
2247 0 : return false;
2248 : }
2249 :
2250 20 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
2251 20 : if (hFeat == nullptr)
2252 : {
2253 0 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2254 0 : return false;
2255 : }
2256 :
2257 20 : if (OGR_F_IsFieldSetAndNotNull(hFeat, 0) &&
2258 4 : OGR_F_IsFieldSetAndNotNull(hFeat, 1) &&
2259 28 : OGR_F_IsFieldSetAndNotNull(hFeat, 2) &&
2260 4 : OGR_F_IsFieldSetAndNotNull(hFeat, 3))
2261 : {
2262 4 : int nMinTileCol = OGR_F_GetFieldAsInteger(hFeat, 0);
2263 4 : int nMaxTileCol = OGR_F_GetFieldAsInteger(hFeat, 1);
2264 4 : int nMinTileRow = OGR_F_GetFieldAsInteger(hFeat, 2);
2265 4 : int nMaxTileRow = OGR_F_GetFieldAsInteger(hFeat, 3);
2266 4 : if (nMaxTileCol < INT_MAX && nMaxTileRow < INT_MAX)
2267 : {
2268 4 : minX = MBTilesTileCoordToWorldCoord(nMinTileCol, nMaxLevel);
2269 4 : minY = MBTilesTileCoordToWorldCoord(nMinTileRow, nMaxLevel);
2270 4 : maxX = MBTilesTileCoordToWorldCoord(nMaxTileCol + 1, nMaxLevel);
2271 4 : maxY = MBTilesTileCoordToWorldCoord(nMaxTileRow + 1, nMaxLevel);
2272 4 : bHasBounds = true;
2273 : }
2274 : }
2275 :
2276 20 : OGR_F_Destroy(hFeat);
2277 20 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2278 : }
2279 :
2280 76 : return bHasBounds;
2281 : }
2282 :
2283 : /************************************************************************/
2284 : /* MBTilesCurlReadCbk() */
2285 : /************************************************************************/
2286 :
2287 : typedef struct
2288 : {
2289 : int nBands;
2290 : int nSize;
2291 : } TileProperties;
2292 :
2293 : /* We spy the data received by CURL for the initial request where we try */
2294 : /* to get a first tile to see its characteristics. We just need the header */
2295 : /* to determine that, so let's make VSICurl stop reading after we have found it
2296 : */
2297 :
2298 1 : static int MBTilesCurlReadCbk(CPL_UNUSED VSILFILE *fp, void *pabyBuffer,
2299 : size_t nBufferSize, void *pfnUserData)
2300 : {
2301 1 : TileProperties *psTP = static_cast<TileProperties *>(pfnUserData);
2302 :
2303 1 : const GByte abyPNGSig[] = {0x89, 0x50, 0x4E, 0x47,
2304 : 0x0D, 0x0A, 0x1A, 0x0A, /* PNG signature */
2305 : 0x00, 0x00, 0x00, 0x0D, /* IHDR length */
2306 : 0x49, 0x48, 0x44, 0x52 /* IHDR chunk */};
2307 :
2308 : /* JPEG SOF0 (Start Of Frame 0) marker */
2309 1 : const GByte abyJPEG1CompSig[] = {0xFF, 0xC0, /* marker */
2310 : 0x00, 0x0B, /* data length = 8 + 1 * 3 */
2311 : 0x08, /* depth : 8 bit */};
2312 1 : const GByte abyJPEG3CompSig[] = {0xFF, 0xC0, /* marker */
2313 : 0x00, 0x11, /* data length = 8 + 3 * 3 */
2314 : 0x08, /* depth : 8 bit */};
2315 :
2316 : int i;
2317 16369 : for (i = 0; i < (int)nBufferSize - (int)sizeof(abyPNGSig); i++)
2318 : {
2319 16368 : if (memcmp(((GByte *)pabyBuffer) + i, abyPNGSig, sizeof(abyPNGSig)) ==
2320 0 : 0 &&
2321 0 : i + sizeof(abyPNGSig) + 4 + 4 + 1 + 1 < nBufferSize)
2322 : {
2323 0 : GByte *ptr = ((GByte *)(pabyBuffer)) + i + (int)sizeof(abyPNGSig);
2324 :
2325 : int nWidth;
2326 0 : memcpy(&nWidth, ptr, 4);
2327 0 : CPL_MSBPTR32(&nWidth);
2328 0 : ptr += 4;
2329 :
2330 : int nHeight;
2331 0 : memcpy(&nHeight, ptr, 4);
2332 0 : CPL_MSBPTR32(&nHeight);
2333 0 : ptr += 4;
2334 :
2335 0 : GByte nDepth = *ptr;
2336 0 : ptr += 1;
2337 :
2338 0 : GByte nColorType = *ptr;
2339 0 : CPLDebug("MBTILES",
2340 : "PNG: nWidth=%d nHeight=%d depth=%d nColorType=%d", nWidth,
2341 : nHeight, nDepth, nColorType);
2342 :
2343 0 : psTP->nBands = -2;
2344 0 : psTP->nSize = nWidth;
2345 0 : if (nWidth == nHeight && nDepth == 8)
2346 : {
2347 0 : if (nColorType == 0)
2348 0 : psTP->nBands = 1; /* Gray */
2349 0 : else if (nColorType == 2)
2350 0 : psTP->nBands = 3; /* RGB */
2351 0 : else if (nColorType == 3)
2352 : {
2353 : /* This might also be a color table with transparency */
2354 : /* but we cannot tell ! */
2355 0 : psTP->nBands = -1;
2356 0 : return TRUE;
2357 : }
2358 0 : else if (nColorType == 4)
2359 0 : psTP->nBands = 2; /* Gray + alpha */
2360 0 : else if (nColorType == 6)
2361 0 : psTP->nBands = 4; /* RGB + alpha */
2362 : }
2363 :
2364 0 : return FALSE;
2365 : }
2366 : }
2367 :
2368 16375 : for (i = 0; i < (int)nBufferSize - ((int)sizeof(abyJPEG1CompSig) + 5); i++)
2369 : {
2370 16374 : if (memcmp(((GByte *)pabyBuffer) + i, abyJPEG1CompSig,
2371 0 : sizeof(abyJPEG1CompSig)) == 0 &&
2372 0 : ((GByte *)pabyBuffer)[sizeof(abyJPEG1CompSig) + 4] == 1)
2373 : {
2374 : GUInt16 nWidth;
2375 0 : memcpy(&nWidth, &(((GByte *)pabyBuffer)[sizeof(abyJPEG1CompSig)]),
2376 : 2);
2377 0 : CPL_MSBPTR16(&nWidth);
2378 : GUInt16 nHeight;
2379 0 : memcpy(&nHeight,
2380 0 : &(((GByte *)pabyBuffer)[sizeof(abyJPEG1CompSig) + 2]), 2);
2381 0 : CPL_MSBPTR16(&nHeight);
2382 :
2383 0 : CPLDebug("MBTILES", "JPEG: nWidth=%d nHeight=%d depth=%d nBands=%d",
2384 : nWidth, nHeight, 8, 1);
2385 :
2386 0 : psTP->nBands = -2;
2387 0 : if (nWidth == nHeight)
2388 : {
2389 0 : psTP->nSize = nWidth;
2390 0 : psTP->nBands = 1;
2391 : }
2392 :
2393 0 : return FALSE;
2394 : }
2395 16374 : else if (memcmp(((GByte *)pabyBuffer) + i, abyJPEG3CompSig,
2396 3 : sizeof(abyJPEG3CompSig)) == 0 &&
2397 3 : ((GByte *)pabyBuffer)[sizeof(abyJPEG3CompSig) + 4] == 3)
2398 : {
2399 : GUInt16 nWidth;
2400 0 : memcpy(&nWidth, &(((GByte *)pabyBuffer)[sizeof(abyJPEG3CompSig)]),
2401 : 2);
2402 0 : CPL_MSBPTR16(&nWidth);
2403 : GUInt16 nHeight;
2404 0 : memcpy(&nHeight,
2405 0 : &(((GByte *)pabyBuffer)[sizeof(abyJPEG3CompSig) + 2]), 2);
2406 0 : CPL_MSBPTR16(&nHeight);
2407 :
2408 0 : CPLDebug("MBTILES", "JPEG: nWidth=%d nHeight=%d depth=%d nBands=%d",
2409 : nWidth, nHeight, 8, 3);
2410 :
2411 0 : psTP->nBands = -2;
2412 0 : if (nWidth == nHeight)
2413 : {
2414 0 : psTP->nSize = nWidth;
2415 0 : psTP->nBands = 3;
2416 : }
2417 :
2418 0 : return FALSE;
2419 : }
2420 : }
2421 :
2422 1 : return TRUE;
2423 : }
2424 :
2425 : /************************************************************************/
2426 : /* MBTilesGetBandCountAndTileSize() */
2427 : /************************************************************************/
2428 :
2429 60 : static int MBTilesGetBandCountAndTileSize(bool bIsVSICURL, GDALDatasetH &hDS,
2430 : int nMaxLevel, int nMinTileRow,
2431 : int nMaxTileRow, int nMinTileCol,
2432 : int nMaxTileCol, int &nTileSize)
2433 : {
2434 : OGRLayerH hSQLLyr;
2435 : OGRFeatureH hFeat;
2436 60 : VSILFILE *fpCURLOGR = nullptr;
2437 60 : int bFirstSelect = TRUE;
2438 :
2439 60 : int nBands = -1;
2440 60 : nTileSize = 0;
2441 :
2442 : /* Get the VSILFILE associated with the OGR SQLite DB */
2443 120 : CPLString osDSName(GDALGetDescription(hDS));
2444 60 : if (bIsVSICURL)
2445 : {
2446 3 : auto poDS = dynamic_cast<OGRSQLiteBaseDataSource *>(
2447 6 : GDALDataset::FromHandle(hDS));
2448 3 : CPLAssert(poDS);
2449 3 : if (poDS)
2450 : {
2451 3 : fpCURLOGR = poDS->GetVSILFILE();
2452 : }
2453 : }
2454 :
2455 : const char *pszSQL =
2456 120 : CPLSPrintf("SELECT tile_data FROM tiles WHERE "
2457 : "tile_column = %d AND tile_row = %d AND zoom_level = %d",
2458 60 : nMinTileCol / 2 + nMaxTileCol / 2,
2459 60 : nMinTileRow / 2 + nMaxTileRow / 2, nMaxLevel);
2460 60 : CPLDebug("MBTILES", "%s", pszSQL);
2461 :
2462 60 : if (fpCURLOGR)
2463 : {
2464 : /* Install a spy on the file connection that will intercept */
2465 : /* PNG or JPEG headers, to interrupt their downloading */
2466 : /* once the header is found. Speeds up dataset opening. */
2467 3 : CPLErrorReset();
2468 : TileProperties tp;
2469 3 : tp.nBands = -1;
2470 3 : tp.nSize = 0;
2471 3 : VSICurlInstallReadCbk(fpCURLOGR, MBTilesCurlReadCbk, &tp, TRUE);
2472 3 : nBands = tp.nBands;
2473 3 : nTileSize = tp.nSize;
2474 :
2475 3 : CPLErrorReset();
2476 3 : CPLPushErrorHandler(CPLQuietErrorHandler);
2477 3 : hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, nullptr, nullptr);
2478 3 : CPLPopErrorHandler();
2479 :
2480 3 : VSICurlUninstallReadCbk(fpCURLOGR);
2481 :
2482 : /* Did the spy intercept something interesting ? */
2483 : // cppcheck-suppress knownConditionTrueFalse
2484 3 : if (nBands != -1)
2485 : {
2486 0 : CPLErrorReset();
2487 :
2488 0 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2489 0 : hSQLLyr = nullptr;
2490 :
2491 : // Re-open OGR SQLite DB, because with our spy we have simulated an
2492 : // I/O error that SQLite will have difficulties to recover within
2493 : // the existing connection. This will be fast because
2494 : // the /vsicurl/ cache has cached the already read blocks.
2495 0 : GDALClose(hDS);
2496 0 : hDS = MBTILESOpenSQLiteDB(osDSName.c_str(), GA_ReadOnly);
2497 0 : if (hDS == nullptr)
2498 0 : return -1;
2499 :
2500 : /* Unrecognized form of PNG. Error out */
2501 0 : if (nBands <= 0)
2502 0 : return -1;
2503 :
2504 0 : return nBands;
2505 : }
2506 3 : else if (CPLGetLastErrorType() == CE_Failure)
2507 : {
2508 0 : CPLError(CE_Failure, CPLGetLastErrorNo(), "%s",
2509 : CPLGetLastErrorMsg());
2510 : }
2511 : }
2512 : else
2513 : {
2514 57 : hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, nullptr, nullptr);
2515 : }
2516 :
2517 : while (true)
2518 : {
2519 73 : if (hSQLLyr == nullptr && bFirstSelect)
2520 : {
2521 13 : bFirstSelect = FALSE;
2522 13 : pszSQL = CPLSPrintf("SELECT tile_data FROM tiles WHERE "
2523 : "zoom_level = %d LIMIT 1",
2524 : nMaxLevel);
2525 13 : CPLDebug("MBTILES", "%s", pszSQL);
2526 13 : hSQLLyr = GDALDatasetExecuteSQL(hDS, pszSQL, nullptr, nullptr);
2527 13 : if (hSQLLyr == nullptr)
2528 0 : return -1;
2529 : }
2530 :
2531 73 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
2532 73 : if (hFeat == nullptr)
2533 : {
2534 18 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2535 18 : hSQLLyr = nullptr;
2536 18 : if (!bFirstSelect)
2537 5 : return -1;
2538 : }
2539 : else
2540 55 : break;
2541 : }
2542 :
2543 110 : const CPLString osMemFileName(VSIMemGenerateHiddenFilename("mvt_temp.db"));
2544 :
2545 55 : int nDataSize = 0;
2546 55 : GByte *pabyData = OGR_F_GetFieldAsBinary(hFeat, 0, &nDataSize);
2547 :
2548 55 : VSIFCloseL(VSIFileFromMemBuffer(osMemFileName.c_str(), pabyData, nDataSize,
2549 : FALSE));
2550 :
2551 55 : GDALDatasetH hDSTile = GDALOpenEx(osMemFileName.c_str(), GDAL_OF_RASTER,
2552 : apszAllowedDrivers, nullptr, nullptr);
2553 55 : if (hDSTile == nullptr)
2554 : {
2555 28 : VSIUnlink(osMemFileName.c_str());
2556 28 : OGR_F_Destroy(hFeat);
2557 28 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2558 28 : return -1;
2559 : }
2560 :
2561 27 : nBands = GDALGetRasterCount(hDSTile);
2562 :
2563 14 : if ((nBands != 1 && nBands != 2 && nBands != 3 && nBands != 4) ||
2564 68 : GDALGetRasterXSize(hDSTile) != GDALGetRasterYSize(hDSTile) ||
2565 27 : GDALGetRasterDataType(GDALGetRasterBand(hDSTile, 1)) != GDT_Byte)
2566 : {
2567 0 : CPLError(CE_Failure, CPLE_NotSupported,
2568 : "Unsupported tile characteristics");
2569 0 : GDALClose(hDSTile);
2570 0 : VSIUnlink(osMemFileName.c_str());
2571 0 : OGR_F_Destroy(hFeat);
2572 0 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2573 0 : return -1;
2574 : }
2575 :
2576 27 : nTileSize = GDALGetRasterXSize(hDSTile);
2577 : GDALColorTableH hCT =
2578 27 : GDALGetRasterColorTable(GDALGetRasterBand(hDSTile, 1));
2579 27 : if (nBands == 1 && hCT != nullptr)
2580 : {
2581 8 : nBands = 3;
2582 8 : if (GDALGetColorEntryCount(hCT) > 0)
2583 : {
2584 : /* Typical of paletted PNG with transparency */
2585 8 : const GDALColorEntry *psEntry = GDALGetColorEntry(hCT, 0);
2586 8 : if (psEntry->c4 == 0)
2587 0 : nBands = 4;
2588 : }
2589 : }
2590 :
2591 27 : GDALClose(hDSTile);
2592 27 : VSIUnlink(osMemFileName.c_str());
2593 27 : OGR_F_Destroy(hFeat);
2594 27 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2595 :
2596 27 : return nBands;
2597 : }
2598 :
2599 : /************************************************************************/
2600 : /* Open() */
2601 : /************************************************************************/
2602 :
2603 77 : GDALDataset *MBTilesDataset::Open(GDALOpenInfo *poOpenInfo)
2604 : {
2605 154 : CPLString osFileName;
2606 154 : CPLString osTableName;
2607 :
2608 77 : if (!Identify(poOpenInfo))
2609 0 : return nullptr;
2610 :
2611 77 : if ((poOpenInfo->nOpenFlags & GDAL_OF_VECTOR) != 0 &&
2612 52 : (poOpenInfo->nOpenFlags & GDAL_OF_RASTER) == 0 &&
2613 15 : (poOpenInfo->nOpenFlags & GDAL_OF_UPDATE) != 0)
2614 : {
2615 0 : return nullptr;
2616 : }
2617 :
2618 : /* -------------------------------------------------------------------- */
2619 : /* Open underlying OGR DB */
2620 : /* -------------------------------------------------------------------- */
2621 :
2622 : GDALDatasetH hDS =
2623 77 : MBTILESOpenSQLiteDB(poOpenInfo->pszFilename, poOpenInfo->eAccess);
2624 :
2625 77 : MBTilesDataset *poDS = nullptr;
2626 :
2627 77 : if (hDS == nullptr)
2628 0 : goto end;
2629 :
2630 : /* -------------------------------------------------------------------- */
2631 : /* Build dataset */
2632 : /* -------------------------------------------------------------------- */
2633 : {
2634 77 : CPLString osMetadataTableName, osRasterTableName;
2635 77 : CPLString osSQL;
2636 : OGRLayerH hMetadataLyr, hRasterLyr;
2637 : OGRFeatureH hFeat;
2638 : int nBands;
2639 77 : OGRLayerH hSQLLyr = nullptr;
2640 77 : int nMinLevel = -1;
2641 77 : int nMaxLevel = -1;
2642 77 : int bHasMinMaxLevel = FALSE;
2643 : int bHasMap;
2644 :
2645 77 : osMetadataTableName = "metadata";
2646 :
2647 : hMetadataLyr =
2648 77 : GDALDatasetGetLayerByName(hDS, osMetadataTableName.c_str());
2649 77 : if (hMetadataLyr == nullptr)
2650 0 : goto end;
2651 :
2652 77 : osRasterTableName += "tiles";
2653 :
2654 77 : hRasterLyr = GDALDatasetGetLayerByName(hDS, osRasterTableName.c_str());
2655 77 : if (hRasterLyr == nullptr)
2656 0 : goto end;
2657 :
2658 77 : bHasMap = GDALDatasetGetLayerByName(hDS, "map") != nullptr;
2659 77 : if (bHasMap)
2660 : {
2661 0 : bHasMap = FALSE;
2662 :
2663 0 : hSQLLyr = GDALDatasetExecuteSQL(
2664 : hDS, "SELECT type FROM sqlite_master WHERE name = 'tiles'",
2665 : nullptr, nullptr);
2666 0 : if (hSQLLyr != nullptr)
2667 : {
2668 0 : hFeat = OGR_L_GetNextFeature(hSQLLyr);
2669 0 : if (hFeat)
2670 : {
2671 0 : if (OGR_F_IsFieldSetAndNotNull(hFeat, 0))
2672 : {
2673 0 : bHasMap = strcmp(OGR_F_GetFieldAsString(hFeat, 0),
2674 0 : "view") == 0;
2675 0 : if (!bHasMap)
2676 : {
2677 0 : CPLDebug("MBTILES", "Weird! 'tiles' is not a view, "
2678 : "but 'map' exists");
2679 : }
2680 : }
2681 0 : OGR_F_Destroy(hFeat);
2682 : }
2683 0 : GDALDatasetReleaseResultSet(hDS, hSQLLyr);
2684 : }
2685 : }
2686 :
2687 : /* --------------------------------------------------------------------
2688 : */
2689 : /* Get minimum and maximum zoom levels */
2690 : /* --------------------------------------------------------------------
2691 : */
2692 :
2693 : bHasMinMaxLevel =
2694 77 : MBTilesGetMinMaxZoomLevel(hDS, bHasMap, nMinLevel, nMaxLevel);
2695 :
2696 : const char *pszZoomLevel =
2697 77 : CSLFetchNameValue(poOpenInfo->papszOpenOptions, "ZOOM_LEVEL");
2698 77 : if (pszZoomLevel != nullptr)
2699 0 : nMaxLevel = atoi(pszZoomLevel);
2700 :
2701 77 : if (bHasMinMaxLevel && (nMinLevel < 0 || nMinLevel > nMaxLevel))
2702 : {
2703 0 : CPLError(CE_Failure, CPLE_AppDefined,
2704 : "Inconsistent values : min(zoom_level) = %d, "
2705 : "max(zoom_level) = %d",
2706 : nMinLevel, nMaxLevel);
2707 0 : goto end;
2708 : }
2709 :
2710 77 : if (bHasMinMaxLevel && nMaxLevel > 22)
2711 : {
2712 0 : CPLError(CE_Failure, CPLE_NotSupported,
2713 : "zoom_level > 22 not supported");
2714 0 : goto end;
2715 : }
2716 :
2717 77 : if (!bHasMinMaxLevel)
2718 : {
2719 1 : CPLError(CE_Failure, CPLE_AppDefined,
2720 : "Cannot find min and max zoom_level");
2721 1 : goto end;
2722 : }
2723 :
2724 : /* --------------------------------------------------------------------
2725 : */
2726 : /* Get bounds */
2727 : /* --------------------------------------------------------------------
2728 : */
2729 76 : double dfMinX = 0.0;
2730 76 : double dfMinY = 0.0;
2731 76 : double dfMaxX = 0.0;
2732 76 : double dfMaxY = 0.0;
2733 152 : bool bUseBounds = CPLFetchBool(
2734 76 : const_cast<const char **>(poOpenInfo->papszOpenOptions),
2735 : "USE_BOUNDS", true);
2736 : const char *pszMinX =
2737 76 : CSLFetchNameValue(poOpenInfo->papszOpenOptions, "MINX");
2738 : const char *pszMinY =
2739 76 : CSLFetchNameValue(poOpenInfo->papszOpenOptions, "MINY");
2740 : const char *pszMaxX =
2741 76 : CSLFetchNameValue(poOpenInfo->papszOpenOptions, "MAXX");
2742 : const char *pszMaxY =
2743 76 : CSLFetchNameValue(poOpenInfo->papszOpenOptions, "MAXY");
2744 : bool bHasBounds;
2745 76 : if (pszMinX != nullptr && pszMinY != nullptr && pszMaxX != nullptr &&
2746 : pszMaxY != nullptr)
2747 : {
2748 0 : bHasBounds = true;
2749 : }
2750 : else
2751 : {
2752 76 : bHasBounds = MBTilesGetBounds(hDS, bUseBounds, nMaxLevel, dfMinX,
2753 : dfMinY, dfMaxX, dfMaxY);
2754 : }
2755 76 : if (!bHasBounds)
2756 : {
2757 16 : CPLError(CE_Failure, CPLE_AppDefined,
2758 : "Cannot find min and max tile numbers");
2759 16 : goto end;
2760 : }
2761 60 : if (pszMinX != nullptr)
2762 0 : dfMinX = CPLAtof(pszMinX);
2763 60 : if (pszMinY != nullptr)
2764 0 : dfMinY = CPLAtof(pszMinY);
2765 60 : if (pszMaxX != nullptr)
2766 0 : dfMaxX = CPLAtof(pszMaxX);
2767 60 : if (pszMaxY != nullptr)
2768 0 : dfMaxY = CPLAtof(pszMaxY);
2769 :
2770 : /* --------------------------------------------------------------------
2771 : */
2772 : /* Get number of bands */
2773 : /* --------------------------------------------------------------------
2774 : */
2775 : int nMinTileCol =
2776 60 : static_cast<int>(MBTilesWorldCoordToTileCoord(dfMinX, nMaxLevel));
2777 : int nMinTileRow =
2778 60 : static_cast<int>(MBTilesWorldCoordToTileCoord(dfMinY, nMaxLevel));
2779 : int nMaxTileCol =
2780 60 : static_cast<int>(MBTilesWorldCoordToTileCoord(dfMaxX, nMaxLevel));
2781 : int nMaxTileRow =
2782 60 : static_cast<int>(MBTilesWorldCoordToTileCoord(dfMaxY, nMaxLevel));
2783 60 : int nTileSize = 0;
2784 120 : nBands = MBTilesGetBandCountAndTileSize(
2785 60 : STARTS_WITH_CI(poOpenInfo->pszFilename, "/vsicurl/"), hDS,
2786 : nMaxLevel, nMinTileRow, nMaxTileRow, nMinTileCol, nMaxTileCol,
2787 : nTileSize);
2788 60 : bool bFoundRasterTile = nBands > 0;
2789 60 : if (!bFoundRasterTile)
2790 33 : nTileSize = knDEFAULT_BLOCK_SIZE;
2791 :
2792 : // Force 4 bands by default (see #6119)
2793 60 : nBands = 4;
2794 :
2795 60 : const char *pszBandCount = CSLFetchNameValueDef(
2796 60 : poOpenInfo->papszOpenOptions, "BAND_COUNT",
2797 : CPLGetConfigOption("MBTILES_BAND_COUNT", nullptr));
2798 60 : if (pszBandCount)
2799 : {
2800 1 : int nTmpBands = atoi(pszBandCount);
2801 1 : if (nTmpBands >= 1 && nTmpBands <= 4)
2802 1 : nBands = nTmpBands;
2803 : }
2804 :
2805 60 : if (poOpenInfo->eAccess == GA_Update)
2806 : {
2807 : // So that we can edit all potential overviews
2808 4 : nMinLevel = 0;
2809 : }
2810 :
2811 : /* --------------------------------------------------------------------
2812 : */
2813 : /* Set dataset attributes */
2814 : /* --------------------------------------------------------------------
2815 : */
2816 :
2817 60 : poDS = new MBTilesDataset();
2818 60 : poDS->eAccess = poOpenInfo->eAccess;
2819 60 : poDS->hDS = hDS;
2820 60 : poDS->hDB = (sqlite3 *)GDALGetInternalHandle((GDALDatasetH)hDS,
2821 : "SQLITE_HANDLE");
2822 60 : CPLAssert(poDS->hDB != nullptr);
2823 :
2824 : /* poDS will release it from now */
2825 60 : hDS = nullptr;
2826 :
2827 : poDS->m_osClip =
2828 60 : CSLFetchNameValueDef(poOpenInfo->papszOpenOptions, "CLIP", "");
2829 60 : poDS->m_nMinZoomLevel = nMinLevel;
2830 60 : bool bRasterOK = poDS->InitRaster(nullptr, nMaxLevel, nBands, nTileSize,
2831 : dfMinX, dfMinY, dfMaxX, dfMaxY);
2832 :
2833 60 : const char *pszFormat = poDS->GetMetadataItem("format");
2834 60 : if (pszFormat != nullptr && EQUAL(pszFormat, "pbf"))
2835 : {
2836 33 : if ((poOpenInfo->nOpenFlags & GDAL_OF_VECTOR) == 0)
2837 : {
2838 1 : CPLDebug("MBTiles", "This files contain vector tiles, "
2839 : "but driver open in raster-only mode");
2840 1 : delete poDS;
2841 1 : return nullptr;
2842 : }
2843 64 : poDS->InitVector(dfMinX, dfMinY, dfMaxX, dfMaxY,
2844 32 : CPLFetchBool(poOpenInfo->papszOpenOptions,
2845 : "ZOOM_LEVEL_AUTO",
2846 32 : CPLTestBool(CPLGetConfigOption(
2847 : "MVT_ZOOM_LEVEL_AUTO", "NO"))),
2848 32 : CPLFetchBool(poOpenInfo->papszOpenOptions,
2849 : "JSON_FIELD", false));
2850 : }
2851 27 : else if ((pszFormat != nullptr && !EQUAL(pszFormat, "pbf")) ||
2852 : bFoundRasterTile)
2853 : {
2854 27 : if ((poOpenInfo->nOpenFlags & GDAL_OF_RASTER) == 0)
2855 : {
2856 1 : CPLDebug("MBTiles", "This files contain raster tiles, "
2857 : "but driver open in vector-only mode");
2858 1 : delete poDS;
2859 1 : return nullptr;
2860 : }
2861 : }
2862 :
2863 58 : if ((pszFormat == nullptr || !EQUAL(pszFormat, "pbf")) && !bRasterOK)
2864 : {
2865 0 : delete poDS;
2866 0 : return nullptr;
2867 : }
2868 :
2869 58 : if (poDS->eAccess == GA_Update)
2870 : {
2871 4 : if (pszFormat != nullptr &&
2872 4 : (EQUAL(pszFormat, "jpg") || EQUAL(pszFormat, "jpeg")))
2873 : {
2874 0 : poDS->m_eTF = GPKG_TF_JPEG;
2875 : }
2876 :
2877 : const char *pszTF =
2878 4 : CSLFetchNameValue(poOpenInfo->papszOpenOptions, "TILE_FORMAT");
2879 4 : if (pszTF)
2880 : {
2881 0 : poDS->m_eTF = GDALGPKGMBTilesGetTileFormat(pszTF);
2882 0 : if ((pszFormat != nullptr &&
2883 0 : (EQUAL(pszFormat, "jpg") || EQUAL(pszFormat, "jpeg")) &&
2884 0 : poDS->m_eTF != GPKG_TF_JPEG) ||
2885 0 : (pszFormat != nullptr && EQUAL(pszFormat, "png") &&
2886 0 : poDS->m_eTF == GPKG_TF_JPEG))
2887 : {
2888 0 : CPLError(CE_Warning, CPLE_AppDefined,
2889 : "Format metadata = '%s', but TILE_FORMAT='%s'",
2890 : pszFormat, pszTF);
2891 : }
2892 : }
2893 :
2894 4 : poDS->ParseCompressionOptions(poOpenInfo->papszOpenOptions);
2895 : }
2896 :
2897 : /* --------------------------------------------------------------------
2898 : */
2899 : /* Add overview levels as internal datasets */
2900 : /* --------------------------------------------------------------------
2901 : */
2902 71 : for (int iLevel = nMaxLevel - 1; iLevel >= nMinLevel; iLevel--)
2903 : {
2904 44 : MBTilesDataset *poOvrDS = new MBTilesDataset();
2905 44 : poOvrDS->ShareLockWithParentDataset(poDS);
2906 44 : poOvrDS->InitRaster(poDS, iLevel, nBands, nTileSize, dfMinX, dfMinY,
2907 : dfMaxX, dfMaxY);
2908 :
2909 88 : poDS->m_papoOverviewDS = (MBTilesDataset **)CPLRealloc(
2910 44 : poDS->m_papoOverviewDS,
2911 44 : sizeof(MBTilesDataset *) * (poDS->m_nOverviewCount + 1));
2912 44 : poDS->m_papoOverviewDS[poDS->m_nOverviewCount++] = poOvrDS;
2913 :
2914 75 : if (poOvrDS->GetRasterXSize() < 256 &&
2915 31 : poOvrDS->GetRasterYSize() < 256)
2916 : {
2917 31 : break;
2918 : }
2919 : }
2920 :
2921 : /* --------------------------------------------------------------------
2922 : */
2923 : /* Initialize any PAM information. */
2924 : /* --------------------------------------------------------------------
2925 : */
2926 58 : poDS->SetDescription(poOpenInfo->pszFilename);
2927 :
2928 58 : if (!STARTS_WITH_CI(poOpenInfo->pszFilename, "/vsicurl/"))
2929 55 : poDS->TryLoadXML();
2930 : else
2931 : {
2932 3 : poDS->SetPamFlags(poDS->GetPamFlags() & ~GPF_DIRTY);
2933 : }
2934 : }
2935 :
2936 75 : end:
2937 75 : if (hDS)
2938 17 : GDALClose(hDS);
2939 :
2940 75 : return poDS;
2941 : }
2942 :
2943 : /************************************************************************/
2944 : /* Create() */
2945 : /************************************************************************/
2946 :
2947 82 : GDALDataset *MBTilesDataset::Create(const char *pszFilename, int nXSize,
2948 : int nYSize, int nBandsIn, GDALDataType eDT,
2949 : char **papszOptions)
2950 : {
2951 : #ifdef HAVE_MVT_WRITE_SUPPORT
2952 82 : if (nXSize == 0 && nYSize == 0 && nBandsIn == 0 && eDT == GDT_Unknown)
2953 : {
2954 41 : char **papszOptionsMod = CSLDuplicate(papszOptions);
2955 41 : papszOptionsMod = CSLSetNameValue(papszOptionsMod, "FORMAT", "MBTILES");
2956 41 : GDALDataset *poRet = OGRMVTWriterDatasetCreate(
2957 : pszFilename, nXSize, nYSize, nBandsIn, eDT, papszOptionsMod);
2958 41 : CSLDestroy(papszOptionsMod);
2959 41 : return poRet;
2960 : }
2961 : #endif
2962 :
2963 41 : MBTilesDataset *poDS = new MBTilesDataset();
2964 41 : if (!poDS->CreateInternal(pszFilename, nXSize, nYSize, nBandsIn, eDT,
2965 : papszOptions))
2966 : {
2967 13 : delete poDS;
2968 13 : poDS = nullptr;
2969 : }
2970 41 : return poDS;
2971 : }
2972 :
2973 : /************************************************************************/
2974 : /* CreateInternal() */
2975 : /************************************************************************/
2976 :
2977 41 : bool MBTilesDataset::CreateInternal(const char *pszFilename, int nXSize,
2978 : int nYSize, int nBandsIn, GDALDataType eDT,
2979 : char **papszOptions)
2980 : {
2981 41 : if (eDT != GDT_Byte)
2982 : {
2983 0 : CPLError(CE_Failure, CPLE_NotSupported, "Only Byte supported");
2984 0 : return false;
2985 : }
2986 41 : if (nBandsIn != 1 && nBandsIn != 2 && nBandsIn != 3 && nBandsIn != 4)
2987 : {
2988 0 : CPLError(CE_Failure, CPLE_NotSupported,
2989 : "Only 1 (Grey/ColorTable), 2 (Grey+Alpha), 3 (RGB) or 4 "
2990 : "(RGBA) band dataset supported");
2991 0 : return false;
2992 : }
2993 :
2994 : // for test/debug purposes only. true is the nominal value
2995 41 : m_bPNGSupports2Bands =
2996 41 : CPLTestBool(CPLGetConfigOption("MBTILES_PNG_SUPPORTS_2BANDS", "TRUE"));
2997 41 : m_bPNGSupportsCT =
2998 41 : CPLTestBool(CPLGetConfigOption("MBTILES_PNG_SUPPORTS_CT", "TRUE"));
2999 41 : m_bWriteBounds = CPLFetchBool(const_cast<const char **>(papszOptions),
3000 : "WRITE_BOUNDS", true);
3001 41 : m_bWriteMinMaxZoom = CPLFetchBool(const_cast<const char **>(papszOptions),
3002 : "WRITE_MINMAXZOOM", true);
3003 : int nBlockSize = std::max(
3004 41 : 64, std::min(8192, atoi(CSLFetchNameValueDef(
3005 : papszOptions, "BLOCKSIZE",
3006 41 : CPLSPrintf("%d", knDEFAULT_BLOCK_SIZE)))));
3007 41 : m_osBounds = CSLFetchNameValueDef(papszOptions, "BOUNDS", "");
3008 41 : m_osCenter = CSLFetchNameValueDef(papszOptions, "CENTER", "");
3009 :
3010 41 : VSIUnlink(pszFilename);
3011 41 : SetDescription(pszFilename);
3012 :
3013 : int rc;
3014 41 : if (STARTS_WITH(pszFilename, "/vsi"))
3015 : {
3016 23 : pMyVFS = OGRSQLiteCreateVFS(nullptr, nullptr);
3017 23 : sqlite3_vfs_register(pMyVFS, 0);
3018 23 : rc = sqlite3_open_v2(pszFilename, &hDB,
3019 : SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE,
3020 23 : pMyVFS->zName);
3021 : }
3022 : else
3023 : {
3024 18 : rc = sqlite3_open(pszFilename, &hDB);
3025 : }
3026 :
3027 41 : if (rc != SQLITE_OK)
3028 : {
3029 3 : CPLError(CE_Failure, CPLE_FileIO, "Cannot create %s", pszFilename);
3030 3 : return false;
3031 : }
3032 :
3033 38 : sqlite3_exec(hDB, "PRAGMA synchronous = OFF", nullptr, nullptr, nullptr);
3034 :
3035 38 : rc = sqlite3_exec(hDB,
3036 : "CREATE TABLE tiles ("
3037 : "zoom_level INTEGER NOT NULL,"
3038 : "tile_column INTEGER NOT NULL,"
3039 : "tile_row INTEGER NOT NULL,"
3040 : "tile_data BLOB NOT NULL,"
3041 : "UNIQUE (zoom_level, tile_column, tile_row) )",
3042 : nullptr, nullptr, nullptr);
3043 38 : if (rc != SQLITE_OK)
3044 : {
3045 8 : CPLError(CE_Failure, CPLE_FileIO, "Cannot create tiles table");
3046 8 : return false;
3047 : }
3048 :
3049 30 : rc = sqlite3_exec(hDB, "CREATE TABLE metadata (name TEXT, value TEXT)",
3050 : nullptr, nullptr, nullptr);
3051 30 : if (rc != SQLITE_OK)
3052 : {
3053 2 : CPLError(CE_Failure, CPLE_FileIO, "Cannot create metadata table");
3054 2 : return false;
3055 : }
3056 :
3057 : const std::string osName = CSLFetchNameValueDef(
3058 84 : papszOptions, "NAME", CPLGetBasenameSafe(pszFilename).c_str());
3059 28 : char *pszSQL = sqlite3_mprintf(
3060 : "INSERT INTO metadata (name, value) VALUES ('name', '%q')",
3061 : osName.c_str());
3062 28 : sqlite3_exec(hDB, pszSQL, nullptr, nullptr, nullptr);
3063 28 : sqlite3_free(pszSQL);
3064 :
3065 28 : const char *pszType = CSLFetchNameValueDef(papszOptions, "TYPE", "overlay");
3066 28 : pszSQL = sqlite3_mprintf(
3067 : "INSERT INTO metadata (name, value) VALUES ('type', '%q')", pszType);
3068 28 : sqlite3_exec(hDB, pszSQL, nullptr, nullptr, nullptr);
3069 28 : sqlite3_free(pszSQL);
3070 :
3071 : const std::string osDescription = CSLFetchNameValueDef(
3072 84 : papszOptions, "DESCRIPTION", CPLGetBasenameSafe(pszFilename).c_str());
3073 28 : pszSQL = sqlite3_mprintf(
3074 : "INSERT INTO metadata (name, value) VALUES ('description', '%q')",
3075 : osDescription.c_str());
3076 28 : sqlite3_exec(hDB, pszSQL, nullptr, nullptr, nullptr);
3077 28 : sqlite3_free(pszSQL);
3078 :
3079 28 : const char *pszTF = CSLFetchNameValue(papszOptions, "TILE_FORMAT");
3080 28 : if (pszTF)
3081 3 : m_eTF = GDALGPKGMBTilesGetTileFormat(pszTF);
3082 :
3083 28 : const char *pszVersion = CSLFetchNameValueDef(
3084 28 : papszOptions, "VERSION", (m_eTF == GPKG_TF_WEBP) ? "1.3" : "1.1");
3085 28 : pszSQL = sqlite3_mprintf(
3086 : "INSERT INTO metadata (name, value) VALUES ('version', '%q')",
3087 : pszVersion);
3088 28 : sqlite3_exec(hDB, pszSQL, nullptr, nullptr, nullptr);
3089 28 : sqlite3_free(pszSQL);
3090 :
3091 28 : const char *pszFormat = CSLFetchNameValueDef(
3092 : papszOptions, "FORMAT", GDALMBTilesGetTileFormatName(m_eTF));
3093 28 : pszSQL = sqlite3_mprintf(
3094 : "INSERT INTO metadata (name, value) VALUES ('format', '%q')",
3095 : pszFormat);
3096 28 : sqlite3_exec(hDB, pszSQL, nullptr, nullptr, nullptr);
3097 28 : sqlite3_free(pszSQL);
3098 :
3099 28 : m_bNew = true;
3100 28 : eAccess = GA_Update;
3101 28 : nRasterXSize = nXSize;
3102 28 : nRasterYSize = nYSize;
3103 :
3104 28 : m_pabyCachedTiles =
3105 28 : (GByte *)VSI_MALLOC3_VERBOSE(4 * 4, nBlockSize, nBlockSize);
3106 28 : if (m_pabyCachedTiles == nullptr)
3107 : {
3108 0 : return false;
3109 : }
3110 :
3111 83 : for (int i = 1; i <= nBandsIn; i++)
3112 55 : SetBand(i, new MBTilesBand(this, nBlockSize));
3113 :
3114 28 : ParseCompressionOptions(papszOptions);
3115 :
3116 28 : return true;
3117 : }
3118 :
3119 : /************************************************************************/
3120 : /* CreateCopy() */
3121 : /************************************************************************/
3122 :
3123 : typedef struct
3124 : {
3125 : const char *pszName;
3126 : GDALResampleAlg eResampleAlg;
3127 : } WarpResamplingAlg;
3128 :
3129 : static const WarpResamplingAlg asResamplingAlg[] = {
3130 : {"NEAREST", GRA_NearestNeighbour},
3131 : {"BILINEAR", GRA_Bilinear},
3132 : {"CUBIC", GRA_Cubic},
3133 : {"CUBICSPLINE", GRA_CubicSpline},
3134 : {"LANCZOS", GRA_Lanczos},
3135 : {"MODE", GRA_Mode},
3136 : {"AVERAGE", GRA_Average},
3137 : {"RMS", GRA_RMS},
3138 : };
3139 :
3140 38 : GDALDataset *MBTilesDataset::CreateCopy(const char *pszFilename,
3141 : GDALDataset *poSrcDS, int /*bStrict*/,
3142 : char **papszOptions,
3143 : GDALProgressFunc pfnProgress,
3144 : void *pProgressData)
3145 : {
3146 :
3147 38 : int nBands = poSrcDS->GetRasterCount();
3148 38 : if (nBands != 1 && nBands != 2 && nBands != 3 && nBands != 4)
3149 : {
3150 2 : CPLError(CE_Failure, CPLE_NotSupported,
3151 : "Only 1 (Grey/ColorTable), 2 (Grey+Alpha), 3 (RGB) or 4 "
3152 : "(RGBA) band dataset supported");
3153 2 : return nullptr;
3154 : }
3155 :
3156 36 : char **papszTO = CSLSetNameValue(nullptr, "DST_SRS", SRS_EPSG_3857);
3157 :
3158 36 : void *hTransformArg = nullptr;
3159 :
3160 : // Hack to compensate for GDALSuggestedWarpOutput2() failure (or not
3161 : // ideal suggestion with PROJ 8) when reprojecting latitude = +/- 90 to
3162 : // EPSG:3857.
3163 36 : double adfSrcGeoTransform[6] = {0, 0, 0, 0, 0, 0};
3164 36 : std::unique_ptr<GDALDataset> poTmpDS;
3165 36 : bool bModifiedMaxLat = false;
3166 36 : bool bModifiedMinLat = false;
3167 36 : const auto poSrcSRS = poSrcDS->GetSpatialRef();
3168 36 : if (poSrcDS->GetGeoTransform(adfSrcGeoTransform) == CE_None &&
3169 72 : adfSrcGeoTransform[2] == 0 && adfSrcGeoTransform[4] == 0 &&
3170 36 : adfSrcGeoTransform[5] < 0)
3171 : {
3172 36 : if (poSrcSRS && poSrcSRS->IsGeographic())
3173 : {
3174 30 : double maxLat = adfSrcGeoTransform[3];
3175 30 : double minLat = adfSrcGeoTransform[3] +
3176 30 : poSrcDS->GetRasterYSize() * adfSrcGeoTransform[5];
3177 : // Corresponds to the latitude of MAX_GM
3178 30 : constexpr double MAX_LAT = 85.0511287798066;
3179 30 : if (maxLat > MAX_LAT)
3180 : {
3181 3 : maxLat = MAX_LAT;
3182 3 : bModifiedMaxLat = true;
3183 : }
3184 30 : if (minLat < -MAX_LAT)
3185 : {
3186 3 : minLat = -MAX_LAT;
3187 3 : bModifiedMinLat = true;
3188 : }
3189 30 : if (bModifiedMaxLat || bModifiedMinLat)
3190 : {
3191 6 : CPLStringList aosOptions;
3192 3 : aosOptions.AddString("-of");
3193 3 : aosOptions.AddString("VRT");
3194 3 : aosOptions.AddString("-projwin");
3195 : aosOptions.AddString(
3196 3 : CPLSPrintf("%.17g", adfSrcGeoTransform[0]));
3197 3 : aosOptions.AddString(CPLSPrintf("%.17g", maxLat));
3198 : aosOptions.AddString(
3199 3 : CPLSPrintf("%.17g", adfSrcGeoTransform[0] +
3200 3 : poSrcDS->GetRasterXSize() *
3201 3 : adfSrcGeoTransform[1]));
3202 3 : aosOptions.AddString(CPLSPrintf("%.17g", minLat));
3203 : auto psOptions =
3204 3 : GDALTranslateOptionsNew(aosOptions.List(), nullptr);
3205 3 : poTmpDS.reset(GDALDataset::FromHandle(GDALTranslate(
3206 : "", GDALDataset::ToHandle(poSrcDS), psOptions, nullptr)));
3207 3 : GDALTranslateOptionsFree(psOptions);
3208 3 : if (poTmpDS)
3209 : {
3210 3 : hTransformArg = GDALCreateGenImgProjTransformer2(
3211 3 : GDALDataset::FromHandle(poTmpDS.get()), nullptr,
3212 : papszTO);
3213 : }
3214 : }
3215 : }
3216 : }
3217 36 : if (hTransformArg == nullptr)
3218 : {
3219 : hTransformArg =
3220 33 : GDALCreateGenImgProjTransformer2(poSrcDS, nullptr, papszTO);
3221 : }
3222 36 : if (hTransformArg == nullptr)
3223 : {
3224 0 : CSLDestroy(papszTO);
3225 0 : return nullptr;
3226 : }
3227 :
3228 36 : GDALTransformerInfo *psInfo = (GDALTransformerInfo *)hTransformArg;
3229 : double adfGeoTransform[6];
3230 : double adfExtent[4];
3231 : int nXSize, nYSize;
3232 :
3233 36 : if (GDALSuggestedWarpOutput2(poSrcDS, psInfo->pfnTransform, hTransformArg,
3234 : adfGeoTransform, &nXSize, &nYSize, adfExtent,
3235 36 : 0) != CE_None)
3236 : {
3237 0 : CSLDestroy(papszTO);
3238 0 : GDALDestroyGenImgProjTransformer(hTransformArg);
3239 0 : return nullptr;
3240 : }
3241 :
3242 36 : GDALDestroyGenImgProjTransformer(hTransformArg);
3243 36 : hTransformArg = nullptr;
3244 36 : poTmpDS.reset();
3245 :
3246 36 : if (bModifiedMaxLat || bModifiedMinLat)
3247 : {
3248 3 : if (bModifiedMaxLat)
3249 : {
3250 3 : const double maxNorthing = MAX_GM;
3251 3 : adfGeoTransform[3] = maxNorthing;
3252 3 : adfExtent[3] = maxNorthing;
3253 : }
3254 3 : if (bModifiedMinLat)
3255 : {
3256 3 : const double minNorthing = -MAX_GM;
3257 3 : adfExtent[1] = minNorthing;
3258 : }
3259 :
3260 3 : if (poSrcSRS && poSrcSRS->IsGeographic())
3261 : {
3262 6 : if (adfSrcGeoTransform[0] +
3263 3 : poSrcDS->GetRasterXSize() * adfSrcGeoTransform[1] ==
3264 : 180)
3265 : {
3266 3 : adfExtent[2] = MAX_GM;
3267 : }
3268 : }
3269 : }
3270 :
3271 : int nZoomLevel;
3272 36 : double dfComputedRes = adfGeoTransform[1];
3273 36 : double dfPrevRes = 0.0;
3274 36 : double dfRes = 0.0;
3275 : int nBlockSize = std::max(
3276 36 : 64, std::min(8192, atoi(CSLFetchNameValueDef(
3277 : papszOptions, "BLOCKSIZE",
3278 36 : CPLSPrintf("%d", knDEFAULT_BLOCK_SIZE)))));
3279 36 : const double dfPixelXSizeZoomLevel0 = 2 * MAX_GM / nBlockSize;
3280 242 : for (nZoomLevel = 0; nZoomLevel < 25; nZoomLevel++)
3281 : {
3282 242 : dfRes = dfPixelXSizeZoomLevel0 / (1 << nZoomLevel);
3283 242 : if (dfComputedRes > dfRes)
3284 36 : break;
3285 206 : dfPrevRes = dfRes;
3286 : }
3287 36 : if (nZoomLevel == 25)
3288 : {
3289 0 : CPLError(CE_Failure, CPLE_AppDefined,
3290 : "Could not find an appropriate zoom level");
3291 0 : CSLDestroy(papszTO);
3292 0 : return nullptr;
3293 : }
3294 :
3295 : const char *pszZoomLevelStrategy =
3296 36 : CSLFetchNameValueDef(papszOptions, "ZOOM_LEVEL_STRATEGY", "AUTO");
3297 36 : if (fabs(dfComputedRes - dfRes) / dfRes > 1e-8)
3298 : {
3299 36 : if (EQUAL(pszZoomLevelStrategy, "LOWER"))
3300 : {
3301 0 : if (nZoomLevel > 0)
3302 0 : nZoomLevel--;
3303 : }
3304 36 : else if (EQUAL(pszZoomLevelStrategy, "UPPER"))
3305 : {
3306 : /* do nothing */
3307 : }
3308 36 : else if (nZoomLevel > 0)
3309 : {
3310 35 : if (dfPrevRes / dfComputedRes < dfComputedRes / dfRes)
3311 33 : nZoomLevel--;
3312 : }
3313 : }
3314 :
3315 36 : dfRes = dfPixelXSizeZoomLevel0 / (1 << nZoomLevel);
3316 :
3317 36 : double dfMinX = adfExtent[0];
3318 36 : double dfMinY = adfExtent[1];
3319 36 : double dfMaxX = adfExtent[2];
3320 36 : double dfMaxY = adfExtent[3];
3321 :
3322 36 : nXSize = (int)(0.5 + (dfMaxX - dfMinX) / dfRes);
3323 36 : nYSize = (int)(0.5 + (dfMaxY - dfMinY) / dfRes);
3324 36 : adfGeoTransform[1] = dfRes;
3325 36 : adfGeoTransform[5] = -dfRes;
3326 :
3327 36 : int nTargetBands = nBands;
3328 : /* For grey level or RGB, if there's reprojection involved, add an alpha */
3329 : /* channel */
3330 68 : if ((nBands == 1 &&
3331 36 : poSrcDS->GetRasterBand(1)->GetColorTable() == nullptr) ||
3332 : nBands == 3)
3333 : {
3334 64 : OGRSpatialReference oSrcSRS;
3335 32 : oSrcSRS.SetFromUserInput(poSrcDS->GetProjectionRef());
3336 32 : oSrcSRS.AutoIdentifyEPSG();
3337 39 : if (oSrcSRS.GetAuthorityCode(nullptr) == nullptr ||
3338 7 : atoi(oSrcSRS.GetAuthorityCode(nullptr)) != 3857)
3339 : {
3340 32 : nTargetBands++;
3341 : }
3342 : }
3343 :
3344 36 : GDALResampleAlg eResampleAlg = GRA_Bilinear;
3345 36 : const char *pszResampling = CSLFetchNameValue(papszOptions, "RESAMPLING");
3346 36 : if (pszResampling)
3347 : {
3348 4 : for (size_t iAlg = 0;
3349 4 : iAlg < sizeof(asResamplingAlg) / sizeof(asResamplingAlg[0]);
3350 : iAlg++)
3351 : {
3352 4 : if (EQUAL(pszResampling, asResamplingAlg[iAlg].pszName))
3353 : {
3354 4 : eResampleAlg = asResamplingAlg[iAlg].eResampleAlg;
3355 4 : break;
3356 : }
3357 : }
3358 : }
3359 :
3360 32 : if (nBands == 1 && poSrcDS->GetRasterBand(1)->GetColorTable() != nullptr &&
3361 68 : eResampleAlg != GRA_NearestNeighbour && eResampleAlg != GRA_Mode)
3362 : {
3363 0 : CPLError(
3364 : CE_Warning, CPLE_AppDefined,
3365 : "Input dataset has a color table, which will likely lead to "
3366 : "bad results when using a resampling method other than "
3367 : "nearest neighbour or mode. Converting the dataset to 24/32 bit "
3368 : "(e.g. with gdal_translate -expand rgb/rgba) is advised.");
3369 : }
3370 :
3371 36 : GDALDataset *poDS = Create(pszFilename, nXSize, nYSize, nTargetBands,
3372 : GDT_Byte, papszOptions);
3373 36 : if (poDS == nullptr)
3374 : {
3375 13 : CSLDestroy(papszTO);
3376 13 : return nullptr;
3377 : }
3378 23 : poDS->SetGeoTransform(adfGeoTransform);
3379 25 : if (nTargetBands == 1 && nBands == 1 &&
3380 2 : poSrcDS->GetRasterBand(1)->GetColorTable() != nullptr)
3381 : {
3382 4 : poDS->GetRasterBand(1)->SetColorTable(
3383 2 : poSrcDS->GetRasterBand(1)->GetColorTable());
3384 : }
3385 :
3386 23 : hTransformArg = GDALCreateGenImgProjTransformer2(poSrcDS, poDS, papszTO);
3387 23 : CSLDestroy(papszTO);
3388 23 : if (hTransformArg == nullptr)
3389 : {
3390 0 : CPLError(CE_Failure, CPLE_AppDefined,
3391 : "GDALCreateGenImgProjTransformer2 failed");
3392 0 : delete poDS;
3393 0 : return nullptr;
3394 : }
3395 :
3396 : /* -------------------------------------------------------------------- */
3397 : /* Warp the transformer with a linear approximator */
3398 : /* -------------------------------------------------------------------- */
3399 23 : hTransformArg = GDALCreateApproxTransformer(GDALGenImgProjTransform,
3400 : hTransformArg, 0.125);
3401 23 : GDALApproxTransformerOwnsSubtransformer(hTransformArg, TRUE);
3402 :
3403 : /* -------------------------------------------------------------------- */
3404 : /* Setup warp options. */
3405 : /* -------------------------------------------------------------------- */
3406 23 : GDALWarpOptions *psWO = GDALCreateWarpOptions();
3407 :
3408 23 : psWO->papszWarpOptions = CSLSetNameValue(nullptr, "OPTIMIZE_SIZE", "YES");
3409 23 : psWO->eWorkingDataType = GDT_Byte;
3410 :
3411 23 : psWO->eResampleAlg = eResampleAlg;
3412 :
3413 23 : psWO->hSrcDS = poSrcDS;
3414 23 : psWO->hDstDS = poDS;
3415 :
3416 23 : psWO->pfnTransformer = GDALApproxTransform;
3417 23 : psWO->pTransformerArg = hTransformArg;
3418 :
3419 23 : psWO->pfnProgress = pfnProgress;
3420 23 : psWO->pProgressArg = pProgressData;
3421 :
3422 : /* -------------------------------------------------------------------- */
3423 : /* Setup band mapping. */
3424 : /* -------------------------------------------------------------------- */
3425 :
3426 23 : if (nBands == 2 || nBands == 4)
3427 2 : psWO->nBandCount = nBands - 1;
3428 : else
3429 21 : psWO->nBandCount = nBands;
3430 :
3431 23 : psWO->panSrcBands = (int *)CPLMalloc(psWO->nBandCount * sizeof(int));
3432 23 : psWO->panDstBands = (int *)CPLMalloc(psWO->nBandCount * sizeof(int));
3433 :
3434 52 : for (int i = 0; i < psWO->nBandCount; i++)
3435 : {
3436 29 : psWO->panSrcBands[i] = i + 1;
3437 29 : psWO->panDstBands[i] = i + 1;
3438 : }
3439 :
3440 23 : if (nBands == 2 || nBands == 4)
3441 : {
3442 2 : psWO->nSrcAlphaBand = nBands;
3443 : }
3444 23 : if (nTargetBands == 2 || nTargetBands == 4)
3445 : {
3446 21 : psWO->nDstAlphaBand = nTargetBands;
3447 : }
3448 :
3449 : /* -------------------------------------------------------------------- */
3450 : /* Initialize and execute the warp. */
3451 : /* -------------------------------------------------------------------- */
3452 23 : GDALWarpOperation oWO;
3453 :
3454 23 : CPLErr eErr = oWO.Initialize(psWO);
3455 23 : if (eErr == CE_None)
3456 : {
3457 : /*if( bMulti )
3458 : eErr = oWO.ChunkAndWarpMulti( 0, 0, nXSize, nYSize );
3459 : else*/
3460 23 : eErr = oWO.ChunkAndWarpImage(0, 0, nXSize, nYSize);
3461 : }
3462 23 : if (eErr != CE_None)
3463 : {
3464 0 : delete poDS;
3465 0 : poDS = nullptr;
3466 : }
3467 :
3468 23 : GDALDestroyTransformer(hTransformArg);
3469 23 : GDALDestroyWarpOptions(psWO);
3470 :
3471 23 : return poDS;
3472 : }
3473 :
3474 : /************************************************************************/
3475 : /* ParseCompressionOptions() */
3476 : /************************************************************************/
3477 :
3478 32 : void MBTilesDataset::ParseCompressionOptions(char **papszOptions)
3479 : {
3480 32 : const char *pszZLevel = CSLFetchNameValue(papszOptions, "ZLEVEL");
3481 32 : if (pszZLevel)
3482 0 : m_nZLevel = atoi(pszZLevel);
3483 :
3484 32 : const char *pszQuality = CSLFetchNameValue(papszOptions, "QUALITY");
3485 32 : if (pszQuality)
3486 2 : m_nQuality = atoi(pszQuality);
3487 :
3488 32 : const char *pszDither = CSLFetchNameValue(papszOptions, "DITHER");
3489 32 : if (pszDither)
3490 1 : m_bDither = CPLTestBool(pszDither);
3491 32 : }
3492 :
3493 : /************************************************************************/
3494 : /* IBuildOverviews() */
3495 : /************************************************************************/
3496 :
3497 4 : static int GetFloorPowerOfTwo(int n)
3498 : {
3499 4 : int p2 = 1;
3500 11 : while ((n = n >> 1) > 0)
3501 : {
3502 7 : p2 <<= 1;
3503 : }
3504 4 : return p2;
3505 : }
3506 :
3507 4 : CPLErr MBTilesDataset::IBuildOverviews(
3508 : const char *pszResampling, int nOverviews, const int *panOverviewList,
3509 : int nBandsIn, const int * /*panBandList*/, GDALProgressFunc pfnProgress,
3510 : void *pProgressData, CSLConstList papszOptions)
3511 : {
3512 4 : if (GetAccess() != GA_Update)
3513 : {
3514 0 : CPLError(CE_Failure, CPLE_NotSupported,
3515 : "Overview building not supported on a database opened in "
3516 : "read-only mode");
3517 0 : return CE_Failure;
3518 : }
3519 4 : if (m_poParentDS != nullptr)
3520 : {
3521 0 : CPLError(CE_Failure, CPLE_NotSupported,
3522 : "Overview building not supported on overview dataset");
3523 0 : return CE_Failure;
3524 : }
3525 :
3526 4 : if (nOverviews == 0)
3527 : {
3528 2 : for (int i = 0; i < m_nOverviewCount; i++)
3529 1 : m_papoOverviewDS[i]->FlushCache(false);
3530 1 : char *pszSQL = sqlite3_mprintf(
3531 : "DELETE FROM 'tiles' WHERE zoom_level < %d", m_nZoomLevel);
3532 1 : char *pszErrMsg = nullptr;
3533 1 : int ret = sqlite3_exec(hDB, pszSQL, nullptr, nullptr, &pszErrMsg);
3534 1 : sqlite3_free(pszSQL);
3535 1 : if (ret != SQLITE_OK)
3536 : {
3537 0 : CPLError(CE_Failure, CPLE_AppDefined, "Failure: %s",
3538 0 : pszErrMsg ? pszErrMsg : "");
3539 0 : sqlite3_free(pszErrMsg);
3540 0 : return CE_Failure;
3541 : }
3542 :
3543 1 : int nRows = 0;
3544 1 : int nCols = 0;
3545 1 : char **papszResult = nullptr;
3546 1 : sqlite3_get_table(
3547 : hDB, "SELECT * FROM metadata WHERE name = 'minzoom' LIMIT 2",
3548 : &papszResult, &nRows, &nCols, nullptr);
3549 1 : sqlite3_free_table(papszResult);
3550 1 : if (nRows == 1)
3551 : {
3552 1 : pszSQL = sqlite3_mprintf(
3553 : "UPDATE metadata SET value = %d WHERE name = 'minzoom'",
3554 : m_nZoomLevel);
3555 1 : sqlite3_exec(hDB, pszSQL, nullptr, nullptr, nullptr);
3556 1 : sqlite3_free(pszSQL);
3557 : }
3558 :
3559 1 : return CE_None;
3560 : }
3561 :
3562 3 : if (nBandsIn != nBands)
3563 : {
3564 0 : CPLError(CE_Failure, CPLE_NotSupported,
3565 : "Generation of overviews only"
3566 : "supported when operating on all bands.");
3567 0 : return CE_Failure;
3568 : }
3569 :
3570 3 : if (m_nOverviewCount == 0)
3571 : {
3572 0 : CPLError(CE_Failure, CPLE_AppDefined,
3573 : "Image too small to support overviews");
3574 0 : return CE_Failure;
3575 : }
3576 :
3577 3 : FlushCache(false);
3578 :
3579 24 : const auto GetOverviewIndex = [](int nVal)
3580 : {
3581 24 : int iOvr = -1;
3582 66 : while (nVal > 1)
3583 : {
3584 42 : nVal >>= 1;
3585 42 : iOvr++;
3586 : }
3587 24 : return iOvr;
3588 : };
3589 :
3590 7 : for (int i = 0; i < nOverviews; i++)
3591 : {
3592 4 : if (panOverviewList[i] < 2)
3593 : {
3594 0 : CPLError(CE_Failure, CPLE_IllegalArg,
3595 0 : "Overview factor '%d' must be >= 2", panOverviewList[i]);
3596 0 : return CE_Failure;
3597 : }
3598 :
3599 4 : if (GetFloorPowerOfTwo(panOverviewList[i]) != panOverviewList[i])
3600 : {
3601 0 : CPLError(CE_Failure, CPLE_IllegalArg,
3602 : "Overview factor '%d' is not a power of 2",
3603 0 : panOverviewList[i]);
3604 0 : return CE_Failure;
3605 : }
3606 4 : const int iOvr = GetOverviewIndex(panOverviewList[i]);
3607 4 : if (iOvr >= m_nOverviewCount)
3608 : {
3609 1 : CPLDebug("MBTILES",
3610 : "Requested overview factor %d leads to too small overview "
3611 : "and will be ignored",
3612 1 : panOverviewList[i]);
3613 : }
3614 : }
3615 :
3616 : GDALRasterBand ***papapoOverviewBands =
3617 3 : (GDALRasterBand ***)CPLCalloc(sizeof(void *), nBands);
3618 3 : int iCurOverview = 0;
3619 15 : for (int iBand = 0; iBand < nBands; iBand++)
3620 : {
3621 24 : papapoOverviewBands[iBand] =
3622 12 : (GDALRasterBand **)CPLCalloc(sizeof(void *), nOverviews);
3623 12 : iCurOverview = 0;
3624 28 : for (int i = 0; i < nOverviews; i++)
3625 : {
3626 16 : const int iOvr = GetOverviewIndex(panOverviewList[i]);
3627 16 : if (iOvr < m_nOverviewCount)
3628 : {
3629 12 : MBTilesDataset *poODS = m_papoOverviewDS[iOvr];
3630 24 : papapoOverviewBands[iBand][iCurOverview] =
3631 12 : poODS->GetRasterBand(iBand + 1);
3632 12 : iCurOverview++;
3633 : }
3634 : }
3635 : }
3636 :
3637 6 : CPLErr eErr = GDALRegenerateOverviewsMultiBand(
3638 3 : nBands, papoBands, iCurOverview, papapoOverviewBands, pszResampling,
3639 : pfnProgress, pProgressData, papszOptions);
3640 :
3641 15 : for (int iBand = 0; iBand < nBands; iBand++)
3642 : {
3643 12 : CPLFree(papapoOverviewBands[iBand]);
3644 : }
3645 3 : CPLFree(papapoOverviewBands);
3646 :
3647 3 : if (eErr == CE_None)
3648 : {
3649 : // Determine new minzoom value from the existing one and the new
3650 : // requested overview levels
3651 3 : int nMinZoom = m_nZoomLevel;
3652 3 : bool bHasMinZoomMetadata = false;
3653 3 : int nRows = 0;
3654 3 : int nCols = 0;
3655 3 : char **papszResult = nullptr;
3656 3 : sqlite3_get_table(
3657 : hDB, "SELECT value FROM metadata WHERE name = 'minzoom' LIMIT 2",
3658 : &papszResult, &nRows, &nCols, nullptr);
3659 3 : if (nRows == 1 && nCols == 1 && papszResult[1])
3660 : {
3661 3 : bHasMinZoomMetadata = true;
3662 3 : nMinZoom = atoi(papszResult[1]);
3663 : }
3664 3 : sqlite3_free_table(papszResult);
3665 3 : if (bHasMinZoomMetadata)
3666 : {
3667 7 : for (int i = 0; i < nOverviews; i++)
3668 : {
3669 4 : const int iOvr = GetOverviewIndex(panOverviewList[i]);
3670 4 : if (iOvr < m_nOverviewCount)
3671 : {
3672 3 : const MBTilesDataset *poODS = m_papoOverviewDS[iOvr];
3673 3 : nMinZoom = std::min(nMinZoom, poODS->m_nZoomLevel);
3674 : }
3675 : }
3676 :
3677 3 : char *pszSQL = sqlite3_mprintf(
3678 : "UPDATE metadata SET value = '%d' WHERE name = 'minzoom'",
3679 : nMinZoom);
3680 3 : sqlite3_exec(hDB, pszSQL, nullptr, nullptr, nullptr);
3681 3 : sqlite3_free(pszSQL);
3682 : }
3683 : }
3684 :
3685 3 : return eErr;
3686 : }
3687 :
3688 : /************************************************************************/
3689 : /* GDALRegister_MBTiles() */
3690 : /************************************************************************/
3691 :
3692 1682 : void GDALRegister_MBTiles()
3693 :
3694 : {
3695 1682 : if (!GDAL_CHECK_VERSION("MBTiles driver"))
3696 0 : return;
3697 :
3698 1682 : if (GDALGetDriverByName("MBTiles") != nullptr)
3699 301 : return;
3700 :
3701 1381 : GDALDriver *poDriver = new GDALDriver();
3702 :
3703 1381 : poDriver->SetDescription("MBTiles");
3704 1381 : poDriver->SetMetadataItem(GDAL_DCAP_RASTER, "YES");
3705 1381 : poDriver->SetMetadataItem(GDAL_DCAP_VECTOR, "YES");
3706 1381 : poDriver->SetMetadataItem(GDAL_DMD_LONGNAME, "MBTiles");
3707 1381 : poDriver->SetMetadataItem(GDAL_DMD_HELPTOPIC,
3708 1381 : "drivers/raster/mbtiles.html");
3709 1381 : poDriver->SetMetadataItem(GDAL_DMD_EXTENSION, "mbtiles");
3710 1381 : poDriver->SetMetadataItem(GDAL_DMD_CREATIONDATATYPES, "Byte");
3711 :
3712 : #define COMPRESSION_OPTIONS \
3713 : " <Option name='TILE_FORMAT' scope='raster' type='string-select' " \
3714 : "description='Format to use to create tiles' default='PNG'>" \
3715 : " <Value>PNG</Value>" \
3716 : " <Value>PNG8</Value>" \
3717 : " <Value>JPEG</Value>" \
3718 : " <Value>WEBP</Value>" \
3719 : " </Option>" \
3720 : " <Option name='QUALITY' scope='raster' type='int' min='1' max='100' " \
3721 : "description='Quality for JPEG and WEBP tiles' default='75'/>" \
3722 : " <Option name='ZLEVEL' scope='raster' type='int' min='1' max='9' " \
3723 : "description='DEFLATE compression level for PNG tiles' default='6'/>" \
3724 : " <Option name='DITHER' scope='raster' type='boolean' " \
3725 : "description='Whether to apply Floyd-Steinberg dithering (for " \
3726 : "TILE_FORMAT=PNG8)' default='NO'/>"
3727 :
3728 1381 : poDriver->SetMetadataItem(
3729 : GDAL_DMD_OPENOPTIONLIST,
3730 : "<OpenOptionList>"
3731 : " <Option name='ZOOM_LEVEL' scope='raster,vector' type='integer' "
3732 : "description='Zoom level of full resolution. If not specified, maximum "
3733 : "non-empty zoom level'/>"
3734 : " <Option name='BAND_COUNT' scope='raster' type='string-select' "
3735 : "description='Number of raster bands' default='AUTO'>"
3736 : " <Value>AUTO</Value>"
3737 : " <Value>1</Value>"
3738 : " <Value>2</Value>"
3739 : " <Value>3</Value>"
3740 : " <Value>4</Value>"
3741 : " </Option>"
3742 : " <Option name='MINX' scope='raster,vector' type='float' "
3743 : "description='Minimum X of area of interest'/>"
3744 : " <Option name='MINY' scope='raster,vector' type='float' "
3745 : "description='Minimum Y of area of interest'/>"
3746 : " <Option name='MAXX' scope='raster,vector' type='float' "
3747 : "description='Maximum X of area of interest'/>"
3748 : " <Option name='MAXY' scope='raster,vector' type='float' "
3749 : "description='Maximum Y of area of interest'/>"
3750 : " <Option name='USE_BOUNDS' scope='raster,vector' type='boolean' "
3751 : "description='Whether to use the bounds metadata, when available, to "
3752 : "determine the AOI' default='YES'/>" COMPRESSION_OPTIONS
3753 : " <Option name='CLIP' scope='vector' type='boolean' "
3754 : "description='Whether to clip geometries to tile extent' "
3755 : "default='YES'/>"
3756 : " <Option name='ZOOM_LEVEL_AUTO' scope='vector' type='boolean' "
3757 : "description='Whether to auto-select the zoom level for vector layers "
3758 : "according to spatial filter extent. Only for display purpose' "
3759 : "default='NO'/>"
3760 : " <Option name='JSON_FIELD' scope='vector' type='boolean' "
3761 : "description='For vector layers, "
3762 : "whether to put all attributes as a serialized JSon dictionary'/>"
3763 1381 : "</OpenOptionList>");
3764 :
3765 1381 : poDriver->SetMetadataItem(
3766 : GDAL_DMD_CREATIONOPTIONLIST,
3767 : "<CreationOptionList>"
3768 : " <Option name='NAME' scope='raster,vector' type='string' "
3769 : "description='Tileset name'/>"
3770 : " <Option name='DESCRIPTION' scope='raster,vector' type='string' "
3771 : "description='A description of the layer'/>"
3772 : " <Option name='TYPE' scope='raster,vector' type='string-select' "
3773 : "description='Layer type' default='overlay'>"
3774 : " <Value>overlay</Value>"
3775 : " <Value>baselayer</Value>"
3776 : " </Option>"
3777 : " <Option name='VERSION' scope='raster' type='string' "
3778 : "description='The version of the tileset, as a plain number' "
3779 : "default='1.1'/>"
3780 : " <Option name='BLOCKSIZE' scope='raster' type='int' "
3781 : "description='Block size in pixels' default='256' min='64' "
3782 : "max='8192'/>" COMPRESSION_OPTIONS
3783 : " <Option name='ZOOM_LEVEL_STRATEGY' scope='raster' "
3784 : "type='string-select' description='Strategy to determine zoom level.' "
3785 : "default='AUTO'>"
3786 : " <Value>AUTO</Value>"
3787 : " <Value>LOWER</Value>"
3788 : " <Value>UPPER</Value>"
3789 : " </Option>"
3790 : " <Option name='RESAMPLING' scope='raster' type='string-select' "
3791 : "description='Resampling algorithm.' default='BILINEAR'>"
3792 : " <Value>NEAREST</Value>"
3793 : " <Value>BILINEAR</Value>"
3794 : " <Value>CUBIC</Value>"
3795 : " <Value>CUBICSPLINE</Value>"
3796 : " <Value>LANCZOS</Value>"
3797 : " <Value>MODE</Value>"
3798 : " <Value>AVERAGE</Value>"
3799 : " </Option>"
3800 : " <Option name='WRITE_BOUNDS' scope='raster' type='boolean' "
3801 : "description='Whether to write the bounds metadata' default='YES'/>"
3802 : " <Option name='WRITE_MINMAXZOOM' scope='raster' type='boolean' "
3803 : "description='Whether to write the minzoom and maxzoom metadata' "
3804 : "default='YES'/>"
3805 : " <Option name='BOUNDS' scope='raster,vector' type='string' "
3806 : "description='Override default value for bounds metadata item'/>"
3807 : " <Option name='CENTER' scope='raster,vector' type='string' "
3808 : "description='Override default value for center metadata item'/>"
3809 : #ifdef HAVE_MVT_WRITE_SUPPORT
3810 : MVT_MBTILES_COMMON_DSCO
3811 : #endif
3812 1381 : "</CreationOptionList>");
3813 1381 : poDriver->SetMetadataItem(GDAL_DCAP_VIRTUALIO, "YES");
3814 :
3815 : #ifdef HAVE_MVT_WRITE_SUPPORT
3816 1381 : poDriver->SetMetadataItem(GDAL_DCAP_CREATE_FIELD, "YES");
3817 1381 : poDriver->SetMetadataItem(GDAL_DMD_CREATIONFIELDDATATYPES,
3818 1381 : "Integer Integer64 Real String");
3819 1381 : poDriver->SetMetadataItem(GDAL_DMD_CREATIONFIELDDATASUBTYPES,
3820 1381 : "Boolean Float32");
3821 :
3822 1381 : poDriver->SetMetadataItem(GDAL_DS_LAYER_CREATIONOPTIONLIST, MVT_LCO);
3823 : #endif
3824 :
3825 : #ifdef ENABLE_SQL_SQLITE_FORMAT
3826 1381 : poDriver->SetMetadataItem("ENABLE_SQL_SQLITE_FORMAT", "YES");
3827 : #endif
3828 1381 : poDriver->SetMetadataItem(GDAL_DMD_SUPPORTED_SQL_DIALECTS, "SQLITE OGRSQL");
3829 :
3830 1381 : poDriver->pfnOpen = MBTilesDataset::Open;
3831 1381 : poDriver->pfnIdentify = MBTilesDataset::Identify;
3832 1381 : poDriver->pfnCreateCopy = MBTilesDataset::CreateCopy;
3833 1381 : poDriver->pfnCreate = MBTilesDataset::Create;
3834 :
3835 1381 : GetGDALDriverManager()->RegisterDriver(poDriver);
3836 : }
|