312 lines
12 KiB
C
312 lines
12 KiB
C
|
#pragma once
|
||
|
|
|
||
|
|
#include <deque>
|
||
|
|
#include <filesystem>
|
||
|
|
#include <QDir>
|
||
|
|
#include <QDirIterator>
|
||
|
|
#include <QFileInfo>
|
||
|
|
#include <QJsonArray>
|
||
|
|
#include <QJsonDocument>
|
||
|
|
#include <QJsonObject>
|
||
|
|
|
||
|
|
#include "selfdrive/ui/qt/offroad/settings.h"
|
||
|
|
#include "selfdrive/ui/qt/widgets/controls.h"
|
||
|
|
#include "selfdrive/ui/ui.h"
|
||
|
|
|
||
|
|
QMap<QString, QString> northeastMap = {
|
||
|
|
{"CT", "Connecticut"}, {"DE", "Delaware"}, {"MA", "Massachusetts"},
|
||
|
|
{"MD", "Maryland"}, {"ME", "Maine"}, {"NH", "New Hampshire"},
|
||
|
|
{"NJ", "New Jersey"}, {"NY", "New York"}, {"PA", "Pennsylvania"},
|
||
|
|
{"RI", "Rhode Island"}, {"VT", "Vermont"}
|
||
|
|
};
|
||
|
|
|
||
|
|
QMap<QString, QString> midwestMap = {
|
||
|
|
{"IA", "Iowa"}, {"IL", "Illinois"}, {"IN", "Indiana"},
|
||
|
|
{"KS", "Kansas"}, {"MI", "Michigan"}, {"MN", "Minnesota"},
|
||
|
|
{"MO", "Missouri"}, {"ND", "North Dakota"}, {"NE", "Nebraska"},
|
||
|
|
{"OH", "Ohio"}, {"SD", "South Dakota"}, {"WI", "Wisconsin"}
|
||
|
|
};
|
||
|
|
|
||
|
|
QMap<QString, QString> southMap = {
|
||
|
|
{"AL", "Alabama"}, {"AR", "Arkansas"}, {"FL", "Florida"},
|
||
|
|
{"GA", "Georgia"}, {"KY", "Kentucky"}, {"LA", "Louisiana"},
|
||
|
|
{"MS", "Mississippi"}, {"NC", "North Carolina"}, {"OK", "Oklahoma"},
|
||
|
|
{"SC", "South Carolina"}, {"TN", "Tennessee"}, {"TX", "Texas"},
|
||
|
|
{"VA", "Virginia"}, {"WV", "West Virginia"}
|
||
|
|
};
|
||
|
|
|
||
|
|
QMap<QString, QString> westMap = {
|
||
|
|
{"AK", "Alaska"}, {"AZ", "Arizona"}, {"CA", "California"},
|
||
|
|
{"CO", "Colorado"}, {"HI", "Hawaii"}, {"ID", "Idaho"},
|
||
|
|
{"MT", "Montana"}, {"NM", "New Mexico"}, {"NV", "Nevada"},
|
||
|
|
{"OR", "Oregon"}, {"UT", "Utah"}, {"WA", "Washington"},
|
||
|
|
{"WY", "Wyoming"}
|
||
|
|
};
|
||
|
|
|
||
|
|
QMap<QString, QString> territoriesMap = {
|
||
|
|
{"AS", "American Samoa"}, {"DC", "District of Columbia"},
|
||
|
|
{"GM", "Guam"}, {"MP", "North Mariana Islands"},
|
||
|
|
{"PR", "Puerto Rico"}, {"VI", "Virgin Islands"}
|
||
|
|
};
|
||
|
|
|
||
|
|
QMap<QString, QString> africaMap = {
|
||
|
|
{"DZ", "Algeria"}, {"AO", "Angola"}, {"BJ", "Benin"}, {"BW", "Botswana"},
|
||
|
|
{"BF", "Burkina Faso"}, {"BI", "Burundi"}, {"CM", "Cameroon"}, {"CV", "Cape Verde"},
|
||
|
|
{"CF", "Central African Republic"}, {"TD", "Chad"}, {"KM", "Comoros"}, {"CG", "Congo (Brazzaville)"},
|
||
|
|
{"CD", "Congo (Kinshasa)"}, {"CI", "Ivory Coast"}, {"DJ", "Djibouti"}, {"EG", "Egypt"},
|
||
|
|
{"GQ", "Equatorial Guinea"}, {"ER", "Eritrea"}, {"SZ", "Eswatini"}, {"ET", "Ethiopia"},
|
||
|
|
{"GA", "Gabon"}, {"GM", "Gambia"}, {"GH", "Ghana"}, {"GN", "Guinea"},
|
||
|
|
{"GW", "Guinea-Bissau"}, {"KE", "Kenya"}, {"LS", "Lesotho"}, {"LR", "Liberia"},
|
||
|
|
{"LY", "Libya"}, {"MG", "Madagascar"}, {"MW", "Malawi"}, {"ML", "Mali"},
|
||
|
|
{"MR", "Mauritania"}, {"MU", "Mauritius"}, {"MA", "Morocco"}, {"MZ", "Mozambique"},
|
||
|
|
{"NA", "Namibia"}, {"NE", "Niger"}, {"NG", "Nigeria"}, {"RW", "Rwanda"},
|
||
|
|
{"ST", "Sao Tome and Principe"}, {"SN", "Senegal"}, {"SC", "Seychelles"}, {"SL", "Sierra Leone"},
|
||
|
|
{"SO", "Somalia"}, {"ZA", "South Africa"}, {"SS", "South Sudan"}, {"SD", "Sudan"},
|
||
|
|
{"TZ", "Tanzania"}, {"TG", "Togo"}, {"TN", "Tunisia"}, {"UG", "Uganda"},
|
||
|
|
{"EH", "Western Sahara"}, {"ZM", "Zambia"}, {"ZW", "Zimbabwe"}
|
||
|
|
};
|
||
|
|
|
||
|
|
QMap<QString, QString> antarcticaMap = {
|
||
|
|
{"AQ", "Antarctica"}
|
||
|
|
};
|
||
|
|
|
||
|
|
QMap<QString, QString> asiaMap = {
|
||
|
|
{"AF", "Afghanistan"}, {"AM", "Armenia"}, {"AZ", "Azerbaijan"}, {"BH", "Bahrain"},
|
||
|
|
{"BD", "Bangladesh"}, {"BT", "Bhutan"}, {"BN", "Brunei"}, {"KH", "Cambodia"},
|
||
|
|
{"CN", "China"}, {"CY", "Cyprus"}, {"GE", "Georgia"}, {"IN", "India"},
|
||
|
|
{"ID", "Indonesia"}, {"IR", "Iran"}, {"IQ", "Iraq"}, {"IL", "Israel"},
|
||
|
|
{"JP", "Japan"}, {"JO", "Jordan"}, {"KZ", "Kazakhstan"}, {"KP", "North Korea"},
|
||
|
|
{"KR", "South Korea"}, {"KW", "Kuwait"}, {"KG", "Kyrgyzstan"}, {"LA", "Laos"},
|
||
|
|
{"LB", "Lebanon"}, {"MY", "Malaysia"}, {"MV", "Maldives"}, {"MN", "Mongolia"},
|
||
|
|
{"MM", "Myanmar"}, {"NP", "Nepal"}, {"OM", "Oman"}, {"PK", "Pakistan"},
|
||
|
|
{"PS", "Palestine"}, {"PH", "Philippines"}, {"QA", "Qatar"}, {"SA", "Saudi Arabia"},
|
||
|
|
{"SG", "Singapore"}, {"LK", "Sri Lanka"}, {"SY", "Syria"}, {"TW", "Taiwan"},
|
||
|
|
{"TJ", "Tajikistan"}, {"TH", "Thailand"}, {"TL", "Timor-Leste"}, {"TR", "Turkey"},
|
||
|
|
{"TM", "Turkmenistan"}, {"AE", "United Arab Emirates"}, {"UZ", "Uzbekistan"}, {"VN", "Vietnam"},
|
||
|
|
{"YE", "Yemen"}
|
||
|
|
};
|
||
|
|
|
||
|
|
QMap<QString, QString> europeMap = {
|
||
|
|
{"AL", "Albania"}, {"AD", "Andorra"}, {"AT", "Austria"}, {"BY", "Belarus"},
|
||
|
|
{"BE", "Belgium"}, {"BA", "Bosnia and Herzegovina"}, {"BG", "Bulgaria"}, {"HR", "Croatia"},
|
||
|
|
{"CY", "Cyprus"}, {"CZ", "Czech Republic"}, {"DK", "Denmark"}, {"EE", "Estonia"},
|
||
|
|
{"FI", "Finland"}, {"FR", "France"}, {"DE", "Germany"}, {"GR", "Greece"},
|
||
|
|
{"HU", "Hungary"}, {"IS", "Iceland"}, {"IE", "Ireland"}, {"IT", "Italy"},
|
||
|
|
{"LV", "Latvia"}, {"LI", "Liechtenstein"}, {"LT", "Lithuania"}, {"LU", "Luxembourg"},
|
||
|
|
{"MT", "Malta"}, {"MD", "Moldova"}, {"MC", "Monaco"}, {"ME", "Montenegro"},
|
||
|
|
{"NL", "Netherlands"}, {"MK", "North Macedonia"}, {"NO", "Norway"}, {"PL", "Poland"},
|
||
|
|
{"PT", "Portugal"}, {"RO", "Romania"}, {"RU", "Russia"}, {"SM", "San Marino"},
|
||
|
|
{"RS", "Serbia"}, {"SK", "Slovakia"}, {"SI", "Slovenia"}, {"ES", "Spain"},
|
||
|
|
{"SE", "Sweden"}, {"CH", "Switzerland"}, {"TR", "Turkey"}, {"UA", "Ukraine"},
|
||
|
|
{"GB", "United Kingdom"}, {"VA", "Vatican City"}
|
||
|
|
};
|
||
|
|
|
||
|
|
QMap<QString, QString> northAmericaMap = {
|
||
|
|
{"AG", "Antigua and Barbuda"}, {"BS", "Bahamas"}, {"BB", "Barbados"}, {"BZ", "Belize"},
|
||
|
|
{"CA", "Canada"}, {"CR", "Costa Rica"}, {"CU", "Cuba"}, {"DM", "Dominica"},
|
||
|
|
{"DO", "Dominican Republic"}, {"SV", "El Salvador"}, {"GD", "Grenada"}, {"GT", "Guatemala"},
|
||
|
|
{"HT", "Haiti"}, {"HN", "Honduras"}, {"JM", "Jamaica"}, {"MX", "Mexico"},
|
||
|
|
{"NI", "Nicaragua"}, {"PA", "Panama"}, {"KN", "Saint Kitts and Nevis"}, {"LC", "Saint Lucia"},
|
||
|
|
{"VC", "Saint Vincent and the Grenadines"}, {"TT", "Trinidad and Tobago"}, {"US", "United States"}
|
||
|
|
};
|
||
|
|
|
||
|
|
QMap<QString, QString> oceaniaMap = {
|
||
|
|
{"AU", "Australia"}, {"FJ", "Fiji"}, {"KI", "Kiribati"}, {"MH", "Marshall Islands"},
|
||
|
|
{"FM", "Micronesia"}, {"NR", "Nauru"}, {"NZ", "New Zealand"}, {"PW", "Palau"},
|
||
|
|
{"PG", "Papua New Guinea"}, {"WS", "Samoa"}, {"SB", "Solomon Islands"}, {"TO", "Tonga"},
|
||
|
|
{"TV", "Tuvalu"}, {"VU", "Vanuatu"}
|
||
|
|
};
|
||
|
|
|
||
|
|
QMap<QString, QString> southAmericaMap = {
|
||
|
|
{"AR", "Argentina"}, {"BO", "Bolivia"}, {"BR", "Brazil"}, {"CL", "Chile"},
|
||
|
|
{"CO", "Colombia"}, {"EC", "Ecuador"}, {"GY", "Guyana"}, {"PY", "Paraguay"},
|
||
|
|
{"PE", "Peru"}, {"SR", "Suriname"}, {"TT", "Trinidad and Tobago"}, {"UY", "Uruguay"},
|
||
|
|
{"VE", "Venezuela"}
|
||
|
|
};
|
||
|
|
|
||
|
|
class ButtonSelectionControl : public QWidget {
|
||
|
|
public:
|
||
|
|
static QString selectedStates;
|
||
|
|
static QString selectedCountries;
|
||
|
|
|
||
|
|
explicit ButtonSelectionControl(const QString &id, const QString &title, const QString &description,
|
||
|
|
const QMap<QString, QString> &map, bool isCountry, QWidget *parent = nullptr)
|
||
|
|
: QWidget(parent), country(isCountry) {
|
||
|
|
QVBoxLayout *layout = new QVBoxLayout(this);
|
||
|
|
layout->setAlignment(Qt::AlignTop);
|
||
|
|
layout->setSpacing(10);
|
||
|
|
|
||
|
|
QHBoxLayout *buttonsLayout = new QHBoxLayout();
|
||
|
|
buttonsLayout->setSpacing(10);
|
||
|
|
layout->addLayout(buttonsLayout);
|
||
|
|
|
||
|
|
int count = 0;
|
||
|
|
int max = country ? 3 : 4;
|
||
|
|
|
||
|
|
QJsonObject mapsSelected = QJsonDocument::fromJson(QString::fromStdString(Params().get("MapsSelected")).toUtf8()).object();
|
||
|
|
|
||
|
|
for (const QString &stateCode : map.keys()) {
|
||
|
|
if (count % max == 0 && count != 0) {
|
||
|
|
buttonsLayout = new QHBoxLayout();
|
||
|
|
buttonsLayout->setSpacing(10);
|
||
|
|
layout->addLayout(buttonsLayout);
|
||
|
|
}
|
||
|
|
|
||
|
|
QPushButton *button = createButton(buttonsLayout, map[stateCode], stateCode);
|
||
|
|
|
||
|
|
QString key = country ? "nations" : "states";
|
||
|
|
if (mapsSelected.contains(key)) {
|
||
|
|
QJsonArray selectedItems = mapsSelected.value(key).toArray();
|
||
|
|
button->setChecked(selectedItems.contains(stateCode));
|
||
|
|
}
|
||
|
|
|
||
|
|
count++;
|
||
|
|
}
|
||
|
|
|
||
|
|
adjustButtonWidths(buttonsLayout);
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
bool country;
|
||
|
|
|
||
|
|
const QString buttonStyle = R"(
|
||
|
|
QPushButton {
|
||
|
|
border-radius: 50px; font-size: 40px; font-weight: 500;
|
||
|
|
height: 100px; padding: 0 25 0 25; color: #E4E4E4;
|
||
|
|
background-color: #393939;
|
||
|
|
}
|
||
|
|
QPushButton:pressed, QPushButton:checked {
|
||
|
|
background-color: #4a4a4a;
|
||
|
|
}
|
||
|
|
QPushButton:checked:enabled {
|
||
|
|
background-color: #33Ab4C;
|
||
|
|
}
|
||
|
|
QPushButton:disabled {
|
||
|
|
color: #33E4E4E4;
|
||
|
|
}
|
||
|
|
)";
|
||
|
|
|
||
|
|
QPushButton *createButton(QHBoxLayout *layout, const QString &label, const QString &stateCode) {
|
||
|
|
QPushButton *button = new QPushButton(label, this);
|
||
|
|
button->setCheckable(true);
|
||
|
|
button->setStyleSheet(buttonStyle);
|
||
|
|
connect(button, &QPushButton::clicked, this, [this, button, stateCode] { updateState(stateCode, button); });
|
||
|
|
layout->addWidget(button);
|
||
|
|
return button;
|
||
|
|
}
|
||
|
|
|
||
|
|
void adjustButtonWidths(QHBoxLayout *layout) {
|
||
|
|
if (!layout || layout->count() == (country ? 3 : 4)) return;
|
||
|
|
|
||
|
|
for (int i = 0; i < layout->count(); ++i) {
|
||
|
|
QWidget *widget = layout->itemAt(i)->widget();
|
||
|
|
QPushButton *button = qobject_cast<QPushButton *>(widget);
|
||
|
|
if (button) {
|
||
|
|
button->setMinimumWidth(button->sizeHint().width());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void updateState(const QString &newState, QPushButton *button) {
|
||
|
|
QString &selectedList = country ? selectedCountries : selectedStates;
|
||
|
|
QStringList tempList = selectedList.split(',');
|
||
|
|
|
||
|
|
if (button->isChecked()) {
|
||
|
|
if (!selectedList.isEmpty()) selectedList += ",";
|
||
|
|
selectedList += newState;
|
||
|
|
} else {
|
||
|
|
tempList.removeAll(newState);
|
||
|
|
selectedList = tempList.join(',');
|
||
|
|
}
|
||
|
|
|
||
|
|
Params("/dev/shm/params").remove("OSMDownloadLocations");
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
QString ButtonSelectionControl::selectedStates = "";
|
||
|
|
QString ButtonSelectionControl::selectedCountries = "";
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
template <typename T>
|
||
|
|
T extractFromJson(const std::string &jsonData, const std::string &key, T defaultValue = 0) {
|
||
|
|
std::string::size_type pos = jsonData.find(key);
|
||
|
|
return pos != std::string::npos ? std::stol(jsonData.substr(pos + key.length())) : defaultValue;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
QString formatTime(long timeInSeconds) {
|
||
|
|
long minutes = timeInSeconds / 60;
|
||
|
|
long seconds = timeInSeconds % 60;
|
||
|
|
QString formattedTime = (minutes > 0) ? QString::number(minutes) + "m " : "";
|
||
|
|
formattedTime += QString::number(seconds) + "s";
|
||
|
|
return formattedTime;
|
||
|
|
}
|
||
|
|
|
||
|
|
QString formatDateTime(const std::chrono::time_point<std::chrono::system_clock> &timePoint) {
|
||
|
|
return QDateTime::fromTime_t(std::chrono::system_clock::to_time_t(timePoint)).toString("h:mm ap");
|
||
|
|
}
|
||
|
|
|
||
|
|
QString calculateElapsedTime(int totalFiles, int downloadedFiles, const std::chrono::steady_clock::time_point &startTime) {
|
||
|
|
using namespace std::chrono;
|
||
|
|
if (totalFiles <= 0 || downloadedFiles >= totalFiles) return "Calculating...";
|
||
|
|
|
||
|
|
long elapsed = duration_cast<seconds>(steady_clock::now() - startTime).count();
|
||
|
|
return formatTime(elapsed);
|
||
|
|
}
|
||
|
|
|
||
|
|
QString calculateETA(int totalFiles, int downloadedFiles, const std::chrono::steady_clock::time_point &startTime) {
|
||
|
|
using namespace std::chrono;
|
||
|
|
if (totalFiles <= 0 || downloadedFiles >= totalFiles) return "Calculating...";
|
||
|
|
|
||
|
|
long elapsed = duration_cast<seconds>(steady_clock::now() - startTime).count();
|
||
|
|
|
||
|
|
if (downloadedFiles == 0 || elapsed <= 0) {
|
||
|
|
return "Calculating...";
|
||
|
|
}
|
||
|
|
|
||
|
|
double averageTimePerFile = static_cast<double>(elapsed) / downloadedFiles;
|
||
|
|
int remainingFiles = totalFiles - downloadedFiles;
|
||
|
|
long estimatedTimeRemaining = static_cast<long>(averageTimePerFile * remainingFiles);
|
||
|
|
|
||
|
|
std::chrono::time_point<std::chrono::system_clock> estimatedCompletionTime = system_clock::now() + seconds(estimatedTimeRemaining);
|
||
|
|
QString estimatedTimeStr = formatDateTime(estimatedCompletionTime);
|
||
|
|
|
||
|
|
return formatTime(estimatedTimeRemaining) + " (" + estimatedTimeStr + ")";
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
QString formatDownloadStatus(int totalFiles, int downloadedFiles) {
|
||
|
|
if (totalFiles <= 0) return "Calculating...";
|
||
|
|
if (downloadedFiles >= totalFiles) return "Downloaded";
|
||
|
|
|
||
|
|
int percentage = static_cast<int>(100 * downloadedFiles / totalFiles);
|
||
|
|
return QString::asprintf("Downloading: %d/%d (%d%%)", downloadedFiles, totalFiles, percentage);
|
||
|
|
}
|
||
|
|
|
||
|
|
quint64 calculateDirectorySize(const QString &path) {
|
||
|
|
quint64 totalSize = 0;
|
||
|
|
QDirIterator it(path, QDir::Files, QDirIterator::Subdirectories);
|
||
|
|
while (it.hasNext()) {
|
||
|
|
it.next();
|
||
|
|
QFileInfo fileInfo(it.filePath());
|
||
|
|
if (fileInfo.isFile()) {
|
||
|
|
totalSize += fileInfo.size();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return totalSize;
|
||
|
|
}
|
||
|
|
|
||
|
|
QString formatSize(qint64 size) {
|
||
|
|
const qint64 kb = 1024;
|
||
|
|
const qint64 mb = 1024 * kb;
|
||
|
|
const qint64 gb = 1024 * mb;
|
||
|
|
|
||
|
|
if (size < gb) {
|
||
|
|
double sizeMB = size / static_cast<double>(mb);
|
||
|
|
return QString::number(sizeMB, 'f', 2) + " MB";
|
||
|
|
} else {
|
||
|
|
double sizeGB = size / static_cast<double>(gb);
|
||
|
|
return QString::number(sizeGB, 'f', 2) + " GB";
|
||
|
|
}
|
||
|
|
}
|