import math
import requests
import json
import pyautogui
import time
import os

# parameters
start_lat = 36.31669444         # degs
start_lng = -115.38336111       # degs

end_lat = 35.93336111          	# degs
end_lng = -114.95002778        	# degs

fov = 10                        # deg
zoom = 16    

# constants
C = 40075016.686                # meters
                              
def deg_to_num(lat_deg, lon_deg, zoom):
	lat_rad = math.radians(lat_deg)
	n = 1 << zoom
	xtile = int((lon_deg + 180.0) / 360.0 * n)
	ytile = int((1.0 - math.asinh(math.tan(lat_rad)) / math.pi) / 2.0 * n)
	return xtile, ytile

def num_to_deg(xtile, ytile, zoom):
  n = 1 << zoom
  lon_deg = xtile / n * 360.0 - 180.0
  lat_rad = math.atan(math.sinh(math.pi * (1 - 2 * ytile / n)))
  lat_deg = math.degrees(lat_rad)
  return lat_deg, lon_deg

def camera_altitude(lat_deg):
	mpp = C * math.cos(math.radians(lat_deg)) / math.pow(2, zoom + 8)
	d = mpp * 1920
	alt = d / 2 * 1 / math.tan(math.radians(fov) / 2)
	return alt

# Find the starting and ending points
start_X, start_Y = deg_to_num(start_lat, start_lng, zoom)
end_X, end_Y = deg_to_num(end_lat, end_lng, zoom)

time.sleep(2)

# Create output folder
if not os.path.exists("output"):
	os.mkdir("output")

# Start looping
n = 1
total = math.floor((end_X - start_X) / 2) * math.floor((end_Y - start_Y) / 2)
for X in range(start_X, end_X, 2):
	for Y in range(start_Y, end_Y, 2):
		# Find the center of the screen
		center_lat, center_lng = num_to_deg(X + 1, Y + 1, zoom)
		center_alt = camera_altitude(center_lat)

		# Making PUT request
		data = json.dumps({'lat': center_lat, 'lng': center_lng, 'alt': center_alt})
		r = requests.put('http://localhost:8080', data = data)

		# Take and save screenshot
		screenshot = pyautogui.screenshot()
		screenshot.save(f"output/{X + 1}_{Y + 1}_{zoom}.png")

		time.sleep(0.5)

		print(f"Shot {n} of {total}")
		n = n + 1