main.py

import webbrowser
from OSM import OSM
from ShapesStructure import image_processing as ip
from definitions import *
import route_optimizer as ro
import time



def main():
	moods_dir = ROOT_DIR + '\\ShapesStructure\\Moods'
	moods = [f for f in os.listdir(moods_dir) if os.path.isdir(os.path.join(moods_dir, f))]

	for mood_dir in moods:
		print(mood_dir)

	is_using_route_opt = input('\nWould you like SAM to use route optimization? (Y/N): ').lower()

	if is_using_route_opt == 'y' or is_using_route_opt == 'yes':
		is_using_route_opt = True
	else:
		is_using_route_opt = False

	mood = ''
	while mood not in MOODS:
		mood = input('\nEnter a mood from above for SAM to create an image on a map of Portland: ').lower()
		if mood not in MOODS:
			print('Sorry, \'' + mood + '\' is not a supported mood. Please enter a valid mood.')
		else:
			image = ip.load_shape_from_mood(mood)
			# ranges for middle chunk of Portland roughly...
			geo_image = ip.points_to_lat_long(image, 45.53353, 45.54680, -122.65153, -122.635)

	SLC = OSM.obtain_map('Salt Lake City')
	Portland = OSM.obtain_map('Portland')
	portland_square = OSM.obtain_square_portion(45.4, 45.9, -122.7, -122.2)

	print('\nSAM is creating the route for you! This might take a minute or two...')

	start_time = time.time()

	# TODO: implement route optimization techniques (in route_optimizer.py) through rotation, scaling, and comparing to road coords from OSM using Hausdorff
	if is_using_route_opt:
		geo_image = ro.optimize_route(geo_image, portland_square)

	end_time = time.time()
	print('Time taken to create route: ' + str(round(end_time - start_time, 2)) + ' seconds')

	# construct coords to be placed into URL

	# TODO: still need to optimize which points are cut from each image (in ip.trim_points)
	if len(geo_image) > 25:
		print('Trimming image down to 25 points...')
		ip.trim_points(geo_image)

	url_end = ''
	for point in geo_image:
		url_end += str(point[0]) + ',' + str(point[1]) + '/'

	print('\nOpening SAM\'s sketch in Google Maps...')
	# showing route that will look like an apple in Google maps...
	webbrowser.open('https://www.google.com/maps/dir/' + url_end)


if __name__ == '__main__':
	main()