#!/usr/bin/python3
# find all routes between all pointes in positions.csv
# max 100 points
# default by time, param -l use length

import requests
import pickle
from os.path import isfile
import fast_tsp
import gpxpy
import gpxpy.gpx
import sys

class Person:
    name: str
    position: str
    index: int
    def __init__(self, index: int, line: str):
        line.strip()
        line = line[:-1]
        data = line.rsplit(",")
        self.name = data[0]
        self.position = ",".join(data[-2:])
        self.index = index

index: int = 0
persons: list[Person] = []
key: str
matrix: list[list[dict[str, int]]] = []
byWhat: str

if len(sys.argv) == 2 and sys.argv[1] == "-l":
    byWhat = 'length'
else:
    byWhat = 'duration'


with open("key.api", "r") as f:
    key = f.readline()
    key = key[:-1]


with open("positions.csv", "r") as pos:
    for l in pos:
        persons.append(Person(index, l))
        index += 1
    p = [o.__dict__ for o in persons]

if not isfile('data.pkl'):
    url = 'https://api.mapy.cz/v1/routing/matrix-m'
    headers = {'accept': 'application/json'}
    for p in persons:
        payload = {}
        payload['starts'] = p.position
        payload['ends'] = []
        for e in persons:
            payload['ends'].append(e.position)
        payload['routeType'] = 'bike_road'
        payload['apikey'] = key
        r = requests.get(url=url, headers=headers, params=payload)
        mat = r.json()
        matrix.append([])
        for res in mat['matrix'][0]:
            matrix[p.index].append(res)
# Serialize the object to a binary format
    with open('data.pkl', 'wb') as file:
        pickle.dump(matrix, file)
else:
# Deserialize the object from the binary file
    with open('data.pkl', 'rb') as file:
        matrix = pickle.load(file)

computeMatrix = {}
computeMatrix['length'] = [[c['length'] for c in r] for r in matrix]
computeMatrix['duration'] = [[c['duration'] for c in r] for r in matrix]

tour = fast_tsp.find_tour(computeMatrix[byWhat], 60)
print('Parametry cesty:')
print('Délka:', fast_tsp.compute_cost(tour, computeMatrix['length'])/1000, 'km')
duration = fast_tsp.compute_cost(tour, computeMatrix['duration'])
print('Doba:', duration//3600, 'h', (duration%3600)//60, 'm', duration%60, 's')

# compute path
gpx = gpxpy.gpx.GPX()
points: list[gpxpy.gpx.GPXTrackPoint] = []
for i in range(0, len(tour)):
    s = tour[i]
    e = tour[(i+1)%len(tour)]
    url = 'https://api.mapy.cz/v1/routing/route'
    headers = {'accept': 'application/json'}
    payload = {}
    payload['start'] = persons[s].position
    payload['end'] = persons[e].position
    payload['routeType'] = 'bike_road'
    payload['apikey'] = key
    payload['format'] = 'geojson'
    r = requests.get(url=url, headers=headers, params=payload)
    route = r.json()
    Jpoints = route['geometry']['geometry']['coordinates']
    gpx.waypoints.append(gpxpy.gpx.GPXWaypoint(Jpoints[0][1], Jpoints[0][0], name=persons[s].name))
    for p in Jpoints:
        points.append(gpxpy.gpx.GPXTrackPoint(p[1], p[0]))

track = gpxpy.gpx.GPXTrack("Hlas")
gpx.tracks.append(track)
segment = gpxpy.gpx.GPXTrackSegment(points)
track.segments.append(segment)
with open("route.gpx", "w") as r:
    r.write(gpx.to_xml())