Force fingerprint function

selfdrive/frogpilot/ui/vehicle_settings.cc

@@ -1,7 +1,103 @@
+#include <QDir>
+#include <QRegularExpression>
+#include <QTextStream>
+
 #include "selfdrive/frogpilot/ui/vehicle_settings.h"
 #include "selfdrive/ui/ui.h"
 
+QStringList getCarNames(const QString &carMake) {
+  QMap<QString, QString> makeMap;
+  makeMap["acura"] = "honda";
+  makeMap["audi"] = "volkswagen";
+  makeMap["buick"] = "gm";
+  makeMap["cadillac"] = "gm";
+  makeMap["chevrolet"] = "gm";
+  makeMap["chrysler"] = "chrysler";
+  makeMap["dodge"] = "chrysler";
+  makeMap["ford"] = "ford";
+  makeMap["gm"] = "gm";
+  makeMap["gmc"] = "gm";
+  makeMap["genesis"] = "hyundai";
+  makeMap["honda"] = "honda";
+  makeMap["hyundai"] = "hyundai";
+  makeMap["infiniti"] = "nissan";
+  makeMap["jeep"] = "chrysler";
+  makeMap["kia"] = "hyundai";
+  makeMap["lexus"] = "toyota";
+  makeMap["lincoln"] = "ford";
+  makeMap["man"] = "volkswagen";
+  makeMap["mazda"] = "mazda";
+  makeMap["nissan"] = "nissan";
+  makeMap["ram"] = "chrysler";
+  makeMap["seat"] = "volkswagen";
+  makeMap["subaru"] = "subaru";
+  makeMap["tesla"] = "tesla";
+  makeMap["toyota"] = "toyota";
+  makeMap["volkswagen"] = "volkswagen";
+  makeMap["skoda"] = "volkswagen";
+
+  QString dirPath = "../../selfdrive/car";
+  QDir dir(dirPath);
+  QString targetFolder = makeMap.value(carMake, carMake);
+  QStringList names;
+
+  foreach (const QString &folder, dir.entryList(QDir::Dirs | QDir::NoDotAndDotDot)) {
+    if (folder == targetFolder) {
+      QFile file(dirPath + "/" + folder + "/values.py");
+      if (file.open(QIODevice::ReadOnly | QIODevice::Text)) {
+        QRegularExpression regex("class CAR\\(StrEnum\\):([\\s\\S]*?)(?=^\\w)", QRegularExpression::MultilineOption);
+        QRegularExpressionMatch match = regex.match(QTextStream(&file).readAll());
+        file.close();
+
+        if (match.hasMatch()) {
+          QRegularExpression nameRegex("=\\s*\"([^\"]+)\"");
+          QRegularExpressionMatchIterator it = nameRegex.globalMatch(match.captured(1));
+          while (it.hasNext()) {
+            names << it.next().captured(1);
+          }
+        }
+      }
+    }
+  }
+
+  std::sort(names.begin(), names.end());
+  return names;
+}
+
 FrogPilotVehiclesPanel::FrogPilotVehiclesPanel(SettingsWindow *parent) : FrogPilotListWidget(parent) {
+  selectMakeButton = new ButtonControl(tr("Select Make"), tr("SELECT"));
+  QObject::connect(selectMakeButton, &ButtonControl::clicked, [this]() {
+    std::string currentMake = params.get("CarMake");
+    QStringList makes = {
+      "Acura", "Audi", "BMW", "Buick", "Cadillac", "Chevrolet", "Chrysler", "Dodge", "Ford", "GM", "GMC",
+      "Genesis", "Honda", "Hyundai", "Infiniti", "Jeep", "Kia", "Lexus", "Lincoln", "MAN", "Mazda",
+      "Mercedes", "Nissan", "Ram", "SEAT", "Subaru", "Tesla", "Toyota", "Volkswagen", "Volvo", "Škoda",
+    };
+
+    QString newMakeSelection = MultiOptionDialog::getSelection(tr("Select a Make"), makes, QString::fromStdString(currentMake), this);
+    if (!newMakeSelection.isEmpty()) {
+      carMake = newMakeSelection;
+      params.put("CarMake", carMake.toStdString());
+      selectMakeButton->setValue(newMakeSelection);
+      setModels();
+    }
+  });
+  addItem(selectMakeButton);
+
+  selectModelButton = new ButtonControl(tr("Select Model"), tr("SELECT"));
+  QString modelSelection = QString::fromStdString(params.get("CarModel"));
+  QObject::connect(selectModelButton, &ButtonControl::clicked, [this]() {
+    std::string currentModel = params.get("CarModel");
+    QString newModelSelection = MultiOptionDialog::getSelection(tr("Select a Model"), models, QString::fromStdString(currentModel), this);
+    if (!newModelSelection.isEmpty()) {
+      params.put("CarModel", newModelSelection.toStdString());
+      selectModelButton->setValue(newModelSelection);
+    }
+  });
+  selectModelButton->setValue(modelSelection);
+  addItem(selectModelButton);
+  selectModelButton->setVisible(false);
+
   std::vector<std::tuple<QString, QString, QString, QString>> vehicleToggles {
     {"GasRegenCmd", "Gas Regen Cmd", "", ""},
   };
@@ -29,6 +125,23 @@ FrogPilotVehiclesPanel::FrogPilotVehiclesPanel(SettingsWindow *parent) : FrogPil
       }
     });
   }
+
+  QObject::connect(uiState(), &UIState::offroadTransition, this, [this](bool offroad) {
+    if (!offroad) {
+      std::thread([this]() {
+        while (carMake.isEmpty()) {
+          std::this_thread::sleep_for(std::chrono::seconds(1));
+          carMake = QString::fromStdString(params.get("CarMake"));
+        }
+        setModels();
+      }).detach();
+    }
+  });
+
+  carMake = QString::fromStdString(params.get("CarMake"));
+  if (!carMake.isEmpty()) {
+    setModels();
+  }
 }
 
 void FrogPilotVehiclesPanel::updateToggles() {
@@ -39,9 +152,17 @@ void FrogPilotVehiclesPanel::updateToggles() {
   }).detach();
 }
 
+void FrogPilotVehiclesPanel::setModels() {
+  models = getCarNames(carMake.toLower());
+  setToggles();
+}
+
 void FrogPilotVehiclesPanel::setToggles() {
-  bool gm = false;
-  bool toyota = false;
+  selectMakeButton->setValue(carMake);
+  selectModelButton->setVisible(!carMake.isEmpty());
+
+  bool gm = carMake == "Buick" || carMake == "Cadillac" || carMake == "Chevrolet" || carMake == "GM" || carMake == "GMC";
+  bool toyota = carMake == "Lexus" || carMake == "Toyota";
 
   for (auto &[key, toggle] : toggles) {
     toggle->setVisible(false);