summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMichał Klichowicz <techniczna@wmwprojekt.pl>2019-04-17 10:38:54 +0200
committerMichał Klichowicz <techniczna@wmwprojekt.pl>2019-04-17 10:38:54 +0200
commitf88a1fd59f439339ba4da151bb3955f988bb7ac9 (patch)
tree247b3d551ad84ea5080e7bb183d1855564c6f6ed
Initial scriptHEADmaster
-rw-r--r--get-wms.py112
1 files changed, 112 insertions, 0 deletions
diff --git a/get-wms.py b/get-wms.py
new file mode 100644
index 0000000..e199d89
--- /dev/null
+++ b/get-wms.py
@@ -0,0 +1,112 @@
+from math import ceil
+import os, sys
+
+from owslib.wms import WebMapService
+from pyproj import Proj, transform
+from PIL import Image
+
+wms = WebMapService(sys.argv[1], version='1.1.1')
+
+if len(sys.argv) < 3:
+ first_layer = None
+ for layer in wms.contents:
+ if first_layer is None:
+ first_layer = layer
+ print layer
+ print wms[first_layer].boundingBox
+ print wms[first_layer].boundingBoxWGS84
+ sys.exit()
+else:
+ layers = sys.argv[8:]
+ bbox = [float(b) for b in sys.argv[2:6]]
+ coord_sys = wms[layers[0]].boundingBox[4]
+ wgs = Proj(init='epsg:4326')
+ out_proj = Proj(init=coord_sys)
+ result_bbox = transform(wgs, out_proj, bbox[0], bbox[1]) + transform(wgs, out_proj, bbox[2], bbox[3])
+
+ image_size = [int(sys.argv[6]), int(sys.argv[7])]
+ bbox_size = [abs(result_bbox[2] - result_bbox[0]), abs(result_bbox[3] - result_bbox[1])]
+ bbox_ratio = bbox_size[0] / bbox_size[1]
+ image_size_corrected = [
+ int(max(image_size[0], image_size[1]*bbox_ratio)),
+ int(max(image_size[1], image_size[0]/bbox_ratio))]
+ print 'Correcting image size to %dx%d due to coordinates box ratio' % (
+ image_size_corrected[0], image_size_corrected[1])
+
+ image_size = [int(ceil(i/400.0))*400 for i in image_size_corrected]
+ print 'Correcting image size to %dx%d due to tile size (400x400)' % (
+ image_size[0], image_size[1])
+
+ min_long = min(result_bbox[0], result_bbox[2])
+ max_long = max(result_bbox[0], result_bbox[2])
+ long_span = max_long - min_long
+ min_lat = min(result_bbox[1], result_bbox[3])
+ max_lat = max(result_bbox[1], result_bbox[3])
+ lat_span = max_lat - min_lat
+ tile_long_span = long_span * 400.0 / image_size_corrected[0]
+ tile_lat_span = lat_span * 400.0 / image_size_corrected[1]
+
+ tiles = []
+ current_lat = min_lat
+ current_long = min_long
+ while current_lat < max_lat:
+ current_row = []
+ while current_long < max_long:
+ current_row.append([
+ current_long, current_lat,
+ current_long+tile_long_span, current_lat+tile_lat_span])
+ current_long += tile_long_span
+ if len(current_row):
+ tiles.append(current_row)
+ current_lat += tile_lat_span
+ current_long = min_long
+
+ files = []
+ x = 0
+ y = 0
+ for row in tiles:
+ filerow = []
+ for tile in row:
+ print 'Fetching tile %dx%d: %s' % (x, y, tile)
+ img = wms.getmap(layers=layers,
+ srs=coord_sys,
+ bbox=tile,
+ size=(400,400),
+ format='image/png',
+ transparent=True)
+ filename = 'out-%d-%d.png' % (x, y)
+ out = open(filename, 'wb')
+ out.write(img.read())
+ out.close()
+ filerow.append(filename)
+ y += 1
+ files.append(filerow)
+ x += 1
+ y = 0
+
+ strips = []
+ r = 0
+ for row in files:
+ tiles = [Image.open(f) for f in row]
+ row_image = Image.new('RGBA', (400*len(tiles), 400))
+ offset = 0
+ for tile in tiles:
+ row_image.paste(tile, (offset, 0))
+ offset += 400
+ strip_filename = 'out-%d.png' % (r)
+ row_image.save(strip_filename)
+ strips.append(strip_filename)
+ for f in row:
+ os.remove(f)
+ r += 1
+
+ strips.reverse()
+ strip_images = [Image.open(f) for f in strips]
+ final_image = Image.new('RGBA', (strip_images[0].width, 400*len(strip_images)))
+ offset = 0
+ for strip_image in strip_images:
+ final_image.paste(strip_image, (0, offset))
+ offset += 400
+ final_image.save('out.png')
+ for s in strips:
+ os.remove(s)