mirror of
https://github.com/protomaps/PMTiles.git
synced 2026-02-04 10:51:07 +00:00
63 lines
2.5 KiB
Python
Executable File
63 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
#pmtiles to files
|
|
#pmtiles to mbtiles
|
|
#mbtiles to pmtiles
|
|
|
|
import os
|
|
import argparse
|
|
import sqlite3
|
|
from pmtiles.reader import read
|
|
from pmtiles.writer import write
|
|
|
|
parser = argparse.ArgumentParser(description='Convert between PMTiles and other archive formats.')
|
|
parser.add_argument('input',help='Input .mbtiles or .pmtiles')
|
|
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('--overwrite', help='Overwrite the existing output.',action='store_true')
|
|
args = parser.parse_args()
|
|
|
|
if os.path.exists(args.output) and not args.overwrite:
|
|
print("Output exists, use --overwrite to overwrite the output.")
|
|
exit(1)
|
|
if os.path.exists(args.output) and args.overwrite:
|
|
os.remove(args.output)
|
|
|
|
if args.input.endswith('.mbtiles') and args.output.endswith('.pmtiles'):
|
|
conn = sqlite3.connect(args.input)
|
|
cursor = conn.cursor()
|
|
|
|
with write(args.output) as writer:
|
|
for row in cursor.execute('SELECT zoom_level,tile_column,tile_row,tile_data FROM tiles WHERE zoom_level <= ? ORDER BY zoom_level,tile_column,tile_row ASC',(args.maxzoom or 99,)):
|
|
writer.write_tile(row[0],row[1],row[2],row[3])
|
|
|
|
metadata = {}
|
|
for row in cursor.execute('SELECT name,value FROM metadata'):
|
|
metadata[row[0]] = row[1]
|
|
if args.maxzoom:
|
|
metadata['maxzoom'] = str(args.maxzoom)
|
|
result = writer.finalize(metadata)
|
|
print("Num tiles:",result['num_tiles'])
|
|
print("Num unique tiles:",result['num_unique_tiles'])
|
|
print("Num leaves:",result['num_leaves'])
|
|
|
|
conn.close()
|
|
elif args.input.endswith('.pmtiles') and args.output.endswith('.mbtiles'):
|
|
conn = sqlite3.connect(args.output)
|
|
cursor = conn.cursor()
|
|
cursor.execute('CREATE TABLE metadata (name text, value text);')
|
|
cursor.execute('CREATE TABLE tiles (zoom_level integer, tile_column integer, tile_row integer, tile_data blob);')
|
|
|
|
with read(args.input) as reader:
|
|
for k,v in reader.metadata.items():
|
|
cursor.execute('INSERT INTO metadata VALUES(?,?)',(k,v))
|
|
for tile, data in reader.tiles():
|
|
cursor.execute('INSERT INTO tiles VALUES(?,?,?,?)',(tile[0],tile[1],tile[2],data))
|
|
|
|
cursor.execute('CREATE UNIQUE INDEX tile_index on tiles (zoom_level, tile_column, tile_row);')
|
|
conn.commit()
|
|
conn.close()
|
|
elif args.input.endswith(".pmtiles"):
|
|
print("PMTiles to directory not yet implemented")
|
|
else:
|
|
print("Conversion not implemented") |