Reworked Sport class to work with QObjects instead of JSON objects.

1 files changed, 45 insertions, 114 deletions

diff --git a/src/model/Sport.cpp b/src/model/Sport.cpp
index c7a9c72..4b822fa 100644
--- a/src/model/Sport.cpp
+++ b/src/model/Sport.cpp
@@ -124,60 +124,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,}");
@@ -190,15 +146,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);
     }
 }
 
@@ -237,8 +188,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;
 
@@ -253,21 +204,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){
@@ -321,9 +272,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;
@@ -362,8 +313,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);
 }
 
 /**
@@ -371,8 +322,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);
 }
 
 /**
@@ -381,16 +332,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--;
         }
-
     }
 }
 
@@ -398,52 +345,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);
     }
 }
 
@@ -452,35 +396,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);
     }
 
 }