31 lines
878 B
Python
31 lines
878 B
Python
|
import sys
|
||
|
|
import os
|
||
|
|
from urllib import request
|
||
|
|
from urllib import parse
|
||
|
|
import subprocess
|
||
|
|
import math
|
||
|
|
import io
|
||
|
|
import PIL.Image
|
||
|
|
from stitching import multistitch, tiles_to_equirectangular_blender
|
||
|
|
from matterportsteal import dl_tiles
|
||
|
|
|
||
|
|
def get_url_data(url):
|
||
|
|
return url.rsplit("/", 1)[0] + "/2k_face$i_$j_$k.jpg?" + url.rsplit("?")[-1]
|
||
|
|
|
||
|
|
def matterport_make_tiles(url):
|
||
|
|
dl_url = get_url_data(url)
|
||
|
|
images = dl_tiles(dl_url)
|
||
|
|
return multistitch(images)
|
||
|
|
|
||
|
|
def matterport_to_equirectangular(url, rotation=[0,0,0], resolution=[3840,1920]):
|
||
|
|
stitched = matterport_make_tiles(url)
|
||
|
|
function = tiles_to_equirectangular_blender
|
||
|
|
|
||
|
|
ordered = stitched[1:5] + [stitched[0], stitched[5]]
|
||
|
|
|
||
|
|
rx, ry, rz = rotation
|
||
|
|
width, height = resolution
|
||
|
|
return function(*ordered, rx=rx, ry=ry, rz=rz, width=width, height=height)
|
||
|
|
|
||
|
|
process_url = matterport_to_equirectangular
|