140 lines
3.7 KiB
Python
140 lines
3.7 KiB
Python
import PIL.Image
|
|
import urllib.request
|
|
import io
|
|
import math
|
|
import subprocess
|
|
import tempfile
|
|
import pathlib
|
|
import os
|
|
|
|
from stitching import tiles_to_equirectangular_blender, multistitch
|
|
|
|
from socket import timeout
|
|
from ssl import SSLError
|
|
from urllib.error import URLError, HTTPError
|
|
|
|
def tdv_normalize(url):
|
|
'''
|
|
Takes the URL of any image in a tdv panorama and returns a string
|
|
with substitutable variables for image IDs.
|
|
|
|
:param url: URL of an image contained in a tdv panorama
|
|
:return: string with substitutable variables or False if URL invalid
|
|
'''
|
|
|
|
try:
|
|
with urllib.request.urlopen(url) as res:
|
|
assert res.getcode() == 200
|
|
|
|
parts = url.split("/")
|
|
|
|
assert "_" in parts[-1]
|
|
parts[-1] = "%i_%i.jpg"
|
|
parts[-2] = "%i"
|
|
parts[-3] = "%s"
|
|
|
|
return "/".join(parts)
|
|
|
|
except:
|
|
return False
|
|
|
|
def tdv_getmaxzoom(schema):
|
|
'''
|
|
Takes a normalized string from tdv_normalize() and returns the maximum
|
|
zoom level available.
|
|
|
|
:param schema: normalized URL format output by tdv_normalize()
|
|
:return: int value of largest available zoom level
|
|
'''
|
|
|
|
return 0
|
|
|
|
def tdv_export(schema):
|
|
'''
|
|
Takes a normalized string from tdv_normalize() and returns a list of
|
|
lists of lists containing all images fit for passing into stitch().
|
|
|
|
:param schema: normalized URL format output by tdv_normalize()
|
|
:return: list of lists of lists of PIL.Image() objects for multistitch()
|
|
'''
|
|
|
|
maxzoom = tdv_getmaxzoom(schema)
|
|
output = []
|
|
|
|
for tile in "frblud":
|
|
t_array = []
|
|
y = 0
|
|
|
|
while True:
|
|
r_array = []
|
|
x = 0
|
|
|
|
while True:
|
|
try:
|
|
print("Getting %s" % (schema % (tile, maxzoom, y, x)))
|
|
while True:
|
|
try:
|
|
res = urllib.request.urlopen(schema % (tile, maxzoom, y, x), timeout=10)
|
|
assert res.getcode() == 200
|
|
fo = io.BytesIO(res.read())
|
|
break
|
|
except (timeout, SSLError, URLError) as e:
|
|
if isinstance(e, HTTPError):
|
|
raise
|
|
continue
|
|
print("Done")
|
|
img = PIL.Image.open(fo)
|
|
r_array.append(img)
|
|
x += 1
|
|
except Exception as e:
|
|
break
|
|
|
|
if not r_array:
|
|
break
|
|
|
|
t_array.append(r_array)
|
|
y += 1
|
|
|
|
output.append(t_array)
|
|
|
|
return output
|
|
|
|
def tdv_make_tiles(url):
|
|
'''
|
|
Determines the type of processing needed to build the six tiles, then
|
|
creates and returns them.
|
|
|
|
:param urL: URL of any image in a tdv panorama
|
|
:return: list of stitched PIL.Image objects (back, right, front, left, top,
|
|
bottom)
|
|
'''
|
|
|
|
parts = url.split("/")
|
|
|
|
try:
|
|
schema = tdv_normalize(url)
|
|
images = tdv_export(schema)
|
|
return multistitch(images)
|
|
|
|
except:
|
|
raise
|
|
raise ValueError("%s does not seem to be a valid tdv URL." % url)
|
|
|
|
|
|
def tdv_to_equirectangular(url, rotation=[0,0,0], resolution=[3840,1920]):
|
|
'''
|
|
Takes the URL of any image in a tdv panorama and returns a finished
|
|
stitched image.
|
|
|
|
:param url: Image URL
|
|
:return: PIL.Image object containing the final image
|
|
'''
|
|
|
|
stitched = tdv_make_tiles(url)
|
|
function = tiles_to_equirectangular_blender
|
|
|
|
rx, ry, rz = rotation
|
|
width, height = resolution
|
|
return function(*stitched, rx=rx, ry=ry, rz=rz, width=width, height=height)
|
|
|
|
process_url = tdv_to_equirectangular
|