65 lines
2.8 KiB
Python
65 lines
2.8 KiB
Python
from .landmark import Landmark
|
|
from utils.get_time_separation import get_time
|
|
|
|
class LinkedLandmarks:
|
|
"""
|
|
A list of landmarks that are linked together, e.g. in a route.
|
|
Each landmark serves as a node in the linked list, but since we expect these to be consumed through the rest API, a pythonic reference to the next landmark is not well suited. Instead we use the uuid of the next landmark to reference the next landmark in the list. This is not very efficient, but appropriate for the expected use case ("short" trips with onyl few landmarks).
|
|
"""
|
|
|
|
_landmarks = list[Landmark]
|
|
total_time: int = 0
|
|
|
|
def __init__(self, data: list[Landmark] = None) -> None:
|
|
"""
|
|
Initialize a new LinkedLandmarks object. This expects an ORDERED list of landmarks, where the first landmark is the starting point and the last landmark is the end point.
|
|
|
|
Args:
|
|
data (list[Landmark], optional): The list of landmarks that are linked together. Defaults to None.
|
|
"""
|
|
self._landmarks = data if data else []
|
|
self._link_landmarks()
|
|
|
|
|
|
def _link_landmarks(self) -> None:
|
|
"""
|
|
Create the links between the landmarks in the list by setting their .next_uuid and the .time_to_next attributes.
|
|
"""
|
|
|
|
# Mark secondary landmarks as such
|
|
self.update_secondary_landmarks()
|
|
|
|
|
|
for i, landmark in enumerate(self._landmarks[:-1]):
|
|
landmark.next_uuid = self._landmarks[i + 1].uuid
|
|
time_to_next = get_time(landmark.location, self._landmarks[i + 1].location)
|
|
landmark.time_to_reach_next = time_to_next
|
|
self.total_time += time_to_next
|
|
|
|
self._landmarks[-1].next_uuid = None
|
|
self._landmarks[-1].time_to_reach_next = 0
|
|
|
|
def update_secondary_landmarks(self) -> None:
|
|
# Extract the attractiveness scores and sort them in descending order
|
|
scores = sorted([landmark.attractiveness for landmark in self._landmarks], reverse=True)
|
|
|
|
# Determine the 10th highest score
|
|
if len(scores) >= 10:
|
|
threshold_score = scores[9]
|
|
else:
|
|
# If there are fewer than 10 landmarks, use the lowest score in the list as the threshold
|
|
threshold_score = min(scores) if scores else 0
|
|
|
|
# Update 'is_secondary' for landmarks with attractiveness below the threshold score
|
|
for landmark in self._landmarks:
|
|
if landmark.attractiveness < threshold_score and landmark.type not in ["start", "finish"]:
|
|
landmark.is_secondary = True
|
|
|
|
|
|
def __getitem__(self, index: int) -> Landmark:
|
|
return self._landmarks[index]
|
|
|
|
|
|
def __str__(self) -> str:
|
|
return f"LinkedLandmarks [{' ->'.join([str(landmark) for landmark in self._landmarks])}]"
|