dragonpilot/selfdrive/ui/qt/maps/map_helpers.cc

#include "selfdrive/ui/qt/maps/map_helpers.h"

#include <algorithm>
#include <string>
#include <utility>

#include <QJsonDocument>
#include <QJsonObject>

#include "common/params.h"
#include "system/hardware/hw.h"
#include "selfdrive/ui/qt/api.h"

QString get_mapbox_token() {
  // Valid for 4 weeks since we can't swap tokens on the fly
  return MAPBOX_TOKEN.isEmpty() ? CommaApi::create_jwt({}, 4 * 7 * 24 * 3600) : MAPBOX_TOKEN;
}

QMapLibre::Settings get_mapbox_settings() {
  QMapLibre::Settings settings;
  settings.setProviderTemplate(QMapLibre::Settings::ProviderTemplate::MapboxProvider);

  if (!Hardware::PC()) {
    settings.setCacheDatabasePath(MAPS_CACHE_PATH);
    settings.setCacheDatabaseMaximumSize(100 * 1024 * 1024);
  }
  settings.setApiBaseUrl(MAPS_HOST);
  settings.setApiKey(get_mapbox_token());

  return settings;
}

QGeoCoordinate to_QGeoCoordinate(const QMapLibre::Coordinate &in) {
  return QGeoCoordinate(in.first, in.second);
}

QMapLibre::CoordinatesCollections model_to_collection(
  const cereal::LiveLocationKalman::Measurement::Reader &calibratedOrientationECEF,
  const cereal::LiveLocationKalman::Measurement::Reader &positionECEF,
  const cereal::XYZTData::Reader &line){

  Eigen::Vector3d ecef(positionECEF.getValue()[0], positionECEF.getValue()[1], positionECEF.getValue()[2]);
  Eigen::Vector3d orient(calibratedOrientationECEF.getValue()[0], calibratedOrientationECEF.getValue()[1], calibratedOrientationECEF.getValue()[2]);
  Eigen::Matrix3d ecef_from_local = euler2rot(orient);

  QMapLibre::Coordinates coordinates;
  auto x = line.getX();
  auto y = line.getY();
  auto z = line.getZ();
  for (int i = 0; i < x.size(); i++) {
    Eigen::Vector3d point_ecef = ecef_from_local * Eigen::Vector3d(x[i], y[i], z[i]) + ecef;
    Geodetic point_geodetic = ecef2geodetic((ECEF){.x = point_ecef[0], .y = point_ecef[1], .z = point_ecef[2]});
    coordinates.push_back({point_geodetic.lat, point_geodetic.lon});
  }

  return {QMapLibre::CoordinatesCollection{coordinates}};
}

QMapLibre::CoordinatesCollections coordinate_to_collection(const QMapLibre::Coordinate &c) {
  QMapLibre::Coordinates coordinates{c};
  return {QMapLibre::CoordinatesCollection{coordinates}};
}

QMapLibre::CoordinatesCollections capnp_coordinate_list_to_collection(const capnp::List<cereal::NavRoute::Coordinate>::Reader& coordinate_list) {
  QMapLibre::Coordinates coordinates;
  for (auto const &c : coordinate_list) {
    coordinates.push_back({c.getLatitude(), c.getLongitude()});
  }
  return {QMapLibre::CoordinatesCollection{coordinates}};
}

QMapLibre::CoordinatesCollections coordinate_list_to_collection(const QList<QGeoCoordinate> &coordinate_list) {
  QMapLibre::Coordinates coordinates;
  for (auto &c : coordinate_list) {
    coordinates.push_back({c.latitude(), c.longitude()});
  }
  return {QMapLibre::CoordinatesCollection{coordinates}};
}

QList<QGeoCoordinate> polyline_to_coordinate_list(const QString &polylineString) {
  QList<QGeoCoordinate> path;
  if (polylineString.isEmpty())
      return path;

  QByteArray data = polylineString.toLatin1();

  bool parsingLatitude = true;

  int shift = 0;
  int value = 0;

  QGeoCoordinate coord(0, 0);

  for (int i = 0; i < data.length(); ++i) {
      unsigned char c = data.at(i) - 63;

      value |= (c & 0x1f) << shift;
      shift += 5;

      // another chunk
      if (c & 0x20)
          continue;

      int diff = (value & 1) ? ~(value >> 1) : (value >> 1);

      if (parsingLatitude) {
          coord.setLatitude(coord.latitude() + (double)diff/1e6);
      } else {
          coord.setLongitude(coord.longitude() + (double)diff/1e6);
          path.append(coord);
      }

      parsingLatitude = !parsingLatitude;

      value = 0;
      shift = 0;
  }

  return path;
}

std::optional<QMapLibre::Coordinate> coordinate_from_param(const std::string &param) {
  QString json_str = QString::fromStdString(Params().get(param));
  if (json_str.isEmpty()) return {};

  QJsonDocument doc = QJsonDocument::fromJson(json_str.toUtf8());
  if (doc.isNull()) return {};

  QJsonObject json = doc.object();
  if (json["latitude"].isDouble() && json["longitude"].isDouble()) {
    QMapLibre::Coordinate coord(json["latitude"].toDouble(), json["longitude"].toDouble());
    return coord;
  } else {
    return {};
  }
}

// return {distance, unit}
std::pair<QString, QString> map_format_distance(float d, bool is_metric) {
  auto round_distance = [](float d) -> float {
    return (d > 10) ? std::nearbyint(d) : std::nearbyint(d * 10) / 10.0;
  };

  d = std::max(d, 0.0f);
  if (is_metric) {
    return (d > 500) ? std::pair{QString::number(round_distance(d / 1000)), QObject::tr("km")}
                     : std::pair{QString::number(50 * std::nearbyint(d / 50)), QObject::tr("m")};
  } else {
    float feet = d * METER_TO_FOOT;
    return (feet > 500) ? std::pair{QString::number(round_distance(d * METER_TO_MILE)), QObject::tr("mi")}
                        : std::pair{QString::number(50 * std::nearbyint(d / 50)), QObject::tr("ft")};
  }
}