Hopefully working TDV implementation

handler.py

@@ -13,7 +13,8 @@ regs = {
         r"cdn-1.matterport.com": "matterportsimple",
         r"youtube.com": "youtubesteal",
         r"l\d_[frblud]_\d\d_\d\d.jpg": "giraffesteal",
-        r"=x\d-y\d-z\d": "googlesteal"
+        r"=x\d-y\d-z\d": "googlesteal",
+        r"[brflud]\/\d\/\d_\d\.jpg": "tdvsteal",
        }
 
 

tdvsteal/__init__.py

@@ -0,0 +1,135 @@
+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
+
+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
+    '''
+
+    l = 0
+
+    while True:
+        try:
+            url = schema % ("f", l+1, 0, 0)
+            with urllib.request.urlopen(url) as res:
+                assert res.getcode() == 200
+                l += 1
+        except:
+            return l
+
+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:
+                    res = urllib.request.urlopen(schema % (tile, maxzoom, y, x))
+                    assert res.getcode() == 200
+                    fo = io.BytesIO(res.read())
+                    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