mirrors/PMTiles
1
0
Fork 0
mirror of https://github.com/protomaps/PMTiles.git synced 2026-02-04 10:51:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

add compression output option to pmtiles-convert

Browse Source
This commit is contained in:
Brandon Liu
2021-03-31 21:56:20 +08:00
parent 568ae4fe7f
commit da5ea8dc65
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
python/bin/pmtiles-convert
View File

@@ -2,6 +2,7 @@
#pmtiles to files #pmtiles to files
import argparse import argparse
import gzip
import json import json
import os import os
import sqlite3 import sqlite3
@@ -13,9 +14,15 @@ parser = argparse.ArgumentParser(description='Convert between PMTiles and other
parser.add_argument('input',help='Input .mbtiles or .pmtiles') parser.add_argument('input',help='Input .mbtiles or .pmtiles')
parser.add_argument('output',help='Output .mbtiles, .pmtiles, or directory') parser.add_argument('output',help='Output .mbtiles, .pmtiles, or directory')
parser.add_argument('--maxzoom', help='the maximum zoom level to include in the output.') parser.add_argument('--maxzoom', help='the maximum zoom level to include in the output.')
parser.add_argument('--gzip', help='Add gzip encoding to the output if it is not already gzipped.',action='store_true')
parser.add_argument('--overwrite', help='Overwrite the existing output.',action='store_true') parser.add_argument('--overwrite', help='Overwrite the existing output.',action='store_true')
args = parser.parse_args() args = parser.parse_args()
def may_compress(data,compress):
if compress and data[0:2] != b'\x1f\x8b':
return gzip.compress(data)
return data
if os.path.exists(args.output) and not args.overwrite: if os.path.exists(args.output) and not args.overwrite:
print("Output exists, use --overwrite to overwrite the output.") print("Output exists, use --overwrite to overwrite the output.")
exit(1) exit(1)
@@ -54,7 +61,7 @@ elif args.input.endswith('.pmtiles') and args.output.endswith('.mbtiles'):
for k,v in reader.metadata.items(): for k,v in reader.metadata.items():
cursor.execute('INSERT INTO metadata VALUES(?,?)',(k,v)) cursor.execute('INSERT INTO metadata VALUES(?,?)',(k,v))
for tile, data in reader.tiles(): for tile, data in reader.tiles():
cursor.execute('INSERT INTO tiles VALUES(?,?,?,?)',(tile[0],tile[1],tile[2],data)) cursor.execute('INSERT INTO tiles VALUES(?,?,?,?)',(tile[0],tile[1],tile[2],may_compress(data,args.gzip)))
cursor.execute('CREATE UNIQUE INDEX tile_index on tiles (zoom_level, tile_column, tile_row);') cursor.execute('CREATE UNIQUE INDEX tile_index on tiles (zoom_level, tile_column, tile_row);')
conn.commit() conn.commit()
@@ -73,7 +80,7 @@ elif args.input.endswith(".pmtiles"):
path = os.path.join(directory,str(tile[2]) + '.' + metadata['format']) path = os.path.join(directory,str(tile[2]) + '.' + metadata['format'])
os.makedirs(directory,exist_ok=True) os.makedirs(directory,exist_ok=True)
with open(path,'wb') as f: with open(path,'wb') as f:
f.write(data) f.write(may_compress(data,args.gzip))
else: else:
print("Conversion not implemented") print("Conversion not implemented")