mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-02 06:03:43 +08:00
Jason Wen
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
"""
|
|
Copyright (c) 2021-, Haibin Wen, sunnypilot, and a number of other contributors.
|
|
|
|
This file is part of sunnypilot and is licensed under the MIT License.
|
|
See the LICENSE.md file in the root directory for more details.
|
|
"""
|
|
from abc import abstractmethod, ABC
|
|
|
|
from cereal import messaging
|
|
from openpilot.common.gps import get_gps_location_service
|
|
from openpilot.common.params import Params
|
|
from openpilot.sunnypilot.navd.helpers import coordinate_from_param
|
|
|
|
|
|
class BaseMapData(ABC):
|
|
def __init__(self):
|
|
self.params = Params()
|
|
|
|
self.gps_location_service = get_gps_location_service(self.params)
|
|
gps_packets = [self.gps_location_service]
|
|
self.sm = messaging.SubMaster(['livePose'] + gps_packets, ignore_alive=gps_packets, ignore_avg_freq=gps_packets,
|
|
ignore_valid=gps_packets, poll='livePose')
|
|
self.pm = messaging.PubMaster(['liveMapDataSP'])
|
|
|
|
self.last_bearing = None
|
|
self.last_position = coordinate_from_param("LastGPSPosition", self.params)
|
|
|
|
@abstractmethod
|
|
def update_location(self) -> None:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_current_speed_limit(self) -> float:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_next_speed_limit_and_distance(self) -> tuple[float, float]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_current_road_name(self) -> str:
|
|
pass
|
|
|
|
def publish(self) -> None:
|
|
speed_limit = self.get_current_speed_limit()
|
|
next_speed_limit, next_speed_limit_distance = self.get_next_speed_limit_and_distance()
|
|
|
|
mapd_sp_send = messaging.new_message('liveMapDataSP')
|
|
mapd_sp_send.valid = self.sm.all_checks(['livePose'])
|
|
live_map_data = mapd_sp_send.liveMapDataSP
|
|
|
|
live_map_data.speedLimitValid = bool(speed_limit > 0)
|
|
live_map_data.speedLimit = speed_limit
|
|
live_map_data.speedLimitAheadValid = bool(next_speed_limit > 0)
|
|
live_map_data.speedLimitAhead = next_speed_limit
|
|
live_map_data.speedLimitAheadDistance = next_speed_limit_distance
|
|
live_map_data.roadName = self.get_current_road_name()
|
|
|
|
self.pm.send('liveMapDataSP', mapd_sp_send)
|
|
|
|
def tick(self) -> None:
|
|
self.sm.update(0)
|
|
self.update_location()
|
|
self.publish()
|