summaryrefslogtreecommitdiff
path: root/src/model/Sport.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/model/Sport.cpp')
-rw-r--r--src/model/Sport.cpp159
1 files changed, 45 insertions, 114 deletions
diff --git a/src/model/Sport.cpp b/src/model/Sport.cpp
index a21e21d..a0ca649 100644
--- a/src/model/Sport.cpp
+++ b/src/model/Sport.cpp
@@ -123,60 +123,16 @@ QJsonArray filter(QJsonArray input, function<bool (QJsonObject)> eval) {
return output;
}
-// static compare function for specific attribute in competitors
-function<bool (const QJsonValue &left, const QJsonValue &right)> genCompare(QString attribute) {
- return [attribute](const QJsonValue &left, const QJsonValue &right) {
- QString l = left.toObject()[attribute].toString();
- QString r = right.toObject()[attribute].toString();
- return l.compare(r) < 0;
- };
-}
-
-// static compare function for the results of a competitor in a specific competition (also called mark)
-bool compareMark(const QJsonValue &left, const QJsonValue &right) {
- // check if one competitor has no mark
- QJsonObject l = left.toObject();
- if (!l.contains("results")) return true;
- QJsonObject r = right.toObject();
- if (!r.contains("results")) return false;
-
- QString lMark = l["results"].toObject()["mark"].toString();
- QString rMark = r["results"].toObject()["mark"].toString();
-
- // check if the marks are numerical values
- if (!lMark.contains(":") && !rMark.contains(":")) return lMark.toFloat() < rMark.toFloat();
-
- // compare time values if not numerical
- QString lTime(""), rTime("");
-
- for (QChar c : lMark) if (c.isDigit()) lTime.append(c);
- for (QChar c : rMark) if (c.isDigit()) rTime.append(c);
-
- return lTime.compare(rTime) < 0;
-}
-
-// static compare function for the amount of medals a competitor has gotten
-bool compareMedals(const QJsonValue &left, const QJsonValue &right) {
- QJsonObject lMedals = left.toObject()["medals"].toObject();
- QJsonObject rMedals = right.toObject()["medals"].toObject();
-
- int gold = lMedals["ME_GOLD"].toInt() - rMedals["ME_GOLD"].toInt();
- int silver = lMedals["ME_SILVER"].toInt() - rMedals["ME_SILVER"].toInt();
- int bronze = lMedals["ME_BRONZE"].toInt() - rMedals["ME_BRONZE"].toInt();
-
- return gold < 0 || (gold == 0 && (silver < 0 || (silver == 0 && bronze < 0)));
-}
-
/**
* @brief Sport::lastName Reduce the full name to the part that is marked in capital letters (probably last name).
* @param competitors The competitors of one category.
*/
-void Sport::lastName(QJsonArray &competitors) {
+void Sport::lastName(QList<Competitor> &competitors) {
// validate competitors
- if (competitors.isEmpty() || !competitors[0].toObject().contains("name")) return;
+ if (competitors.isEmpty()) return;
- for (int i = 0; i < competitors.size(); ++i) {
- string fullName = competitors[i].toObject()["name"].toString().toUtf8().constData();
+ for (Competitor comp : competitors) {
+ string fullName = comp.getName().toUtf8().constData();
// regex to identify names, written in CAPS
regex r("[A-Z']{2,}");
@@ -189,15 +145,10 @@ void Sport::lastName(QJsonArray &competitors) {
for (string s : m) lastName = lastName + s + " ";
// remove last space
- QJsonValue nameValue = QJsonValue(QString(lastName.substr(0, lastName.size() - 1).c_str()));
+ QString name = QString(lastName.substr(0, lastName.size() - 1).c_str());
- // create new object with replaced name
- QJsonObject comp(competitors[i].toObject());
- comp.remove("name");
- comp.insert("name", nameValue);
-
- // replace competitor in array
- competitors.replace(i, comp);
+ // replace competitor name in list
+ comp.setName(name);
}
}
@@ -236,8 +187,8 @@ set<QString> Sport::getCategories() {
* @param category The category to search in.
* @return An QJsonArray with all competitors as QJsonValueRef, which can be casted to QJsonObject.
*/
-QJsonArray Sport::getCompetitorsByCategory(QString category) {
- QJsonArray competitors;
+QList<CompetitorWithResults> Sport::getCompetitorsByCategory(QString category) {
+ QList<CompetitorWithResults> competitors;
if (!validateDiscipline()) return competitors;
@@ -252,21 +203,21 @@ QJsonArray Sport::getCompetitorsByCategory(QString category) {
// add all competitors from one unit
for (const QJsonValueRef &compRef : unit["competitors"].toArray()) {
- competitors.push_back(compRef.toObject());
+ competitors.push_back(CompetitorWithResults(compRef.toObject()));
}
}
- return QJsonArray(competitors);
+ return competitors;
}
/**
* @brief Sport::getCompetitorsWithMedal Filters all competitors, who have at least one medal. These objects are different from getCompetitorsByCategory !!!
* @return All competitors, who won at least one medal. Structure of one competitor: {code, name, noc, medals{ME_GOLD, ME_SILVER, ME_BRONZE}}
*/
-QJsonArray Sport::getCompetitorsWithMedal() {
+QList<MedalWinner> Sport::getCompetitorsWithMedal() {
map<QString, QJsonObject> competitors;
- if (!validateDiscipline()) return QJsonArray();
+ if (!validateDiscipline()) return QList<MedalWinner>();
// filter all units, which have medal events
QJsonArray units = filter(this->discipline["units"].toArray(), [](QJsonObject unit){
@@ -320,9 +271,9 @@ QJsonArray Sport::getCompetitorsWithMedal() {
}
// convert map to QJsonArray
- QJsonArray output;
+ QList<MedalWinner> output;
for (const pair<QString, QJsonObject> &competitor : competitors) {
- output.append(competitor.second);
+ output.append(MedalWinner(competitor.second));
}
return output;
@@ -361,8 +312,8 @@ QJsonObject Sport::createCompetitorWithMedals(QJsonObject comp) {
* @param competitors The competitors of one category.
* @param name The (part of the) name to search for.
*/
-void Sport::filterByName(QJsonArray &competitors, QString name) {
- filterCompetitors(competitors, QString("name"), name);
+void Sport::filterByName(QList<Competitor> &competitors, QString name) {
+ filterCompetitors(competitors, name);
}
/**
@@ -370,8 +321,8 @@ void Sport::filterByName(QJsonArray &competitors, QString name) {
* @param competitors The competitors of one category.
* @param nocShort The (part of the) national olympics comittee short name to search for.
*/
-void Sport::filterByCountry(QJsonArray &competitors, QString nocShort) {
- filterCompetitors(competitors, QString("noc"), nocShort);
+void Sport::filterByCountry(QList<Competitor> &competitors, QString nocShort) {
+ filterCompetitors(competitors, nocShort);
}
/**
@@ -380,16 +331,12 @@ void Sport::filterByCountry(QJsonArray &competitors, QString nocShort) {
* @param attribute The attribute to filter by.
* @param filter The string, which should be contained.
*/
-void Sport::filterCompetitors(QJsonArray &competitors, QString attribute, QString filter) {
- for (int i = 0; i < competitors.size(); i++) {
- QJsonObject comp = competitors[i].toObject();
-
- if (!comp.contains(attribute) || !comp[attribute].toString().contains(filter, Qt::CaseInsensitive)) {
- // remove the competitor, if the attribute does not fit the filter string
- competitors.removeAt(i);
+void Sport::filterCompetitors(QList<Competitor> &competitors, QString filter) {
+ for (qsizetype i = 0; i < competitors.size(); i++) {
+ if (competitors.value(i).getNOC().contains(filter)) {
+ competitors.remove(i);
i--;
}
-
}
}
@@ -397,52 +344,49 @@ void Sport::filterCompetitors(QJsonArray &competitors, QString attribute, QStrin
* @brief Sport::sortByName Sort the competitors by their name (alphabetical, ascending).
* @param competitors The competitors of one category.
*/
-void Sport::sortByName(QJsonArray &competitors) {
- sortCompetitors(competitors, genCompare( QString("name") ));
+void Sport::sortByName(QList<Competitor> &competitors) {
+ if (competitors.isEmpty()) return;
+ sort(competitors.begin(), competitors.end(), Competitor::compareName);
}
/**
* @brief Sport::sortByCountry Sort the competitors by their national olympic comittee short name (alphabetical, ascending).
* @param competitors The competitors of one category.
*/
-void Sport::sortByCountry(QJsonArray &competitors) {
- sortCompetitors(competitors, genCompare( QString("noc") ));
+void Sport::sortByCountry(QList<Competitor> &competitors) {
+ if (competitors.isEmpty()) return;
+ sort(competitors.begin(), competitors.end(), Competitor::compareNOC);
}
/**
* @brief Sport::sortByResult Sort the competitors by their results in one specific category (numerical, ascending).
* @param competitors The competitors of one category.
*/
-void Sport::sortByResult(QJsonArray &competitors) {
+void Sport::sortByResult(QList<CompetitorWithResults> &competitors) {
if (competitors.isEmpty()) return;
-
- QJsonObject comp = competitors[0].toObject();
-
- if (comp.contains("results")) sortCompetitors(competitors, compareMark);
- else if (comp.contains("medals")) sortCompetitors(competitors, compareMedals);
+ sort(competitors.begin(), competitors.end(), CompetitorWithResults::compare);
}
/**
- * @brief Sport::sortCompetitors Sorts the given QJsonArray according to the compare function.
+ * @brief Sport::sortByMedals Sort the competitors by their medal amounts in one specific category (numerical, ascending).
* @param competitors The competitors of one category.
- * @param compare A function to compare two competitors with each other. This defines the order.
*/
-void Sport::sortCompetitors(QJsonArray &competitors, function<bool (const QJsonValue &left, const QJsonValue &right)> compare) {
- make_heap(competitors.begin(), competitors.end(), compare);
- sort_heap(competitors.begin(), competitors.end(), compare);
+void Sport::sortByMedals(QList<MedalWinner> &competitors) {
+ if (competitors.isEmpty()) return;
+ sort(competitors.begin(), competitors.end(), MedalWinner::compare);
}
/**
* @brief Sport::reverseOrder Reverses the order of the competitors.
* @param competitors The competitors of one category.
*/
-void Sport::reverseOrder(QJsonArray &competitors) {
+void Sport::reverseOrder(QList<Competitor> &competitors) {
int iterations = competitors.size() / 2; // automatically rounds down
for (int i = 0; i < iterations; i++) {
- QJsonObject temp = competitors[i].toObject();
- competitors[i] = competitors[competitors.size() - 1 - i].toObject();
- competitors[competitors.size() - 1 - i] = temp;
+ Competitor temp = competitors.value(i);
+ competitors.replace(i, competitors.value(competitors.size() - 1 - i));
+ competitors.replace(competitors.size() - 1 - i, temp);
}
}
@@ -451,35 +395,22 @@ void Sport::reverseOrder(QJsonArray &competitors) {
* Stores the statistic in obj->results->stat for each competitor.
* @param competitors The competitors of one category.
*/
-void Sport::addRelativeToFirst(QJsonArray &competitors) {
+void Sport::addRelativeToFirst(QList<CompetitorWithResults> &competitors) {
if (competitors.isEmpty()) return;
- QJsonObject reference = competitors[0].toObject();
-
- // validate competitors
- if (!reference.contains("results")) return;
-
- QString refVal = reference["results"].toObject()["mark"].toString();
+ QString reference = competitors.value(0).getMark();
- for (int i = 0; i < competitors.size(); i++) {
- QJsonObject competitor = competitors[i].toObject();
- QJsonObject results = competitor["results"].toObject();
-
- if (results.contains("stat")) results.remove("stat");
+ for (CompetitorWithResults comp : competitors) {
+ QString results = comp.getMark();
// format relative float value to string with 2 digits after decimal point and sign
stringstream sstream;
- sstream << fixed << setprecision(2) << calcRelativeStat(refVal, results["mark"].toString());
+ sstream << fixed << setprecision(2) << calcRelativeStat(reference, results["mark"].toString());
QString stat(sstream.str().c_str());
stat.append("%");
if (stat.at(0).isNumber()) stat = QString("+").append(stat);
- results.insert("stat", stat);
-
- competitor.remove("results");
- competitor.insert("results", results);
-
- competitors.replace(i, competitor);
+ comp.setStatistic(stat);
}
}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-01 17:31:52 +0000