Helldragon67
bdb040b756
Some checks failed
Build and push docker image / Build (pull_request) Successful in 1m56s
Run linting on the backend code / Build (pull_request) Failing after 25s
Run testing on the backend code / Build (pull_request) Failing after 49s
Build and release APK / Build APK (pull_request) Successful in 5m35s
77 lines
3.2 KiB
Python
77 lines
3.2 KiB
Python
import logging
|
|
import yaml
|
|
|
|
from .utils.landmarks_manager import LandmarkManager
|
|
from .utils.optimizer import Optimizer
|
|
from .utils.refiner import Refiner
|
|
from .structs.landmark import Landmark
|
|
from .structs.linked_landmarks import LinkedLandmarks
|
|
from .structs.preferences import Preferences, Preference
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
def test(start_coords: tuple[float, float], finish_coords: tuple[float, float] = None) -> list[Landmark]:
|
|
manager = LandmarkManager()
|
|
optimizer = Optimizer()
|
|
refiner = Refiner(optimizer=optimizer)
|
|
|
|
|
|
preferences = Preferences(
|
|
sightseeing=Preference(type='sightseeing', score = 5),
|
|
nature=Preference(type='nature', score = 5),
|
|
shopping=Preference(type='shopping', score = 5),
|
|
|
|
max_time_minute=180,
|
|
detour_tolerance_minute=30
|
|
)
|
|
|
|
# Create start and finish
|
|
if finish_coords is None :
|
|
finish_coords = start_coords
|
|
start = Landmark(name='start', type='start', location=start_coords, osm_type='', osm_id=0, attractiveness=0, n_tags = 0)
|
|
finish = Landmark(name='finish', type='finish', location=finish_coords, osm_type='', osm_id=0, attractiveness=0, n_tags = 0)
|
|
#finish = Landmark(name='finish', type=LandmarkType(landmark_type='finish'), location=(48.8777055, 2.3640967), osm_type='finish', osm_id=0, attractiveness=0, must_do=True, n_tags = 0)
|
|
#start = Landmark(name='start', type=LandmarkType(landmark_type='start'), location=(48.847132, 2.312359), osm_type='start', osm_id=0, attractiveness=0, must_do=True, n_tags = 0)
|
|
#finish = Landmark(name='finish', type=LandmarkType(landmark_type='finish'), location=(48.843185, 2.344533), osm_type='finish', osm_id=0, attractiveness=0, must_do=True, n_tags = 0)
|
|
#finish = Landmark(name='finish', type=LandmarkType(landmark_type='finish'), location=(48.847132, 2.312359), osm_type='finish', osm_id=0, attractiveness=0, must_do=True, n_tags = 0)
|
|
|
|
|
|
|
|
# Generate the landmarks from the start location
|
|
landmarks, landmarks_short = manager.generate_landmarks_list(
|
|
center_coordinates = start_coords,
|
|
preferences = preferences
|
|
)
|
|
|
|
# Store data to file for debug purposes
|
|
# write_data(landmarks, "landmarks_Strasbourg.txt")
|
|
|
|
# Insert start and finish to the landmarks list
|
|
landmarks_short.insert(0, start)
|
|
landmarks_short.append(finish)
|
|
|
|
# First stage optimization
|
|
base_tour = optimizer.solve_optimization(max_time=preferences.max_time_minute, landmarks=landmarks_short)
|
|
|
|
# Second stage using linear optimization
|
|
refined_tour = refiner.refine_optimization(all_landmarks=landmarks, base_tour=base_tour, max_time = preferences.max_time_minute, detour = preferences.detour_tolerance_minute)
|
|
|
|
linked_tour = LinkedLandmarks(refined_tour)
|
|
logger.info(f"Optimized route: {linked_tour}")
|
|
|
|
# with open('linked_tour.yaml', 'w') as f:
|
|
# yaml.dump(linked_tour.asdict(), f)
|
|
|
|
return linked_tour
|
|
|
|
|
|
#test(tuple((48.8344400, 2.3220540))) # Café Chez César
|
|
#test(tuple((48.8375946, 2.2949904))) # Point random
|
|
#test(tuple((47.377859, 8.540585))) # Zurich HB
|
|
#test(tuple((45.7576485, 4.8330241))) # Lyon Bellecour
|
|
test(tuple((48.5848435, 7.7332974))) # Strasbourg Gare
|
|
#test(tuple((48.2067858, 16.3692340))) # Vienne
|