2 files changed, 89 insertions, 0 deletions
diff --git a/src/discipline/Sport.cpp b/src/discipline/Sport.cpp
new file mode 100644
index 0000000..ed8331d
--- /dev/null
+++ b/src/discipline/Sport.cpp
@@ -0,0 +1,48 @@
+
+#include "Sport.h"
+#include <set>
+
+#include <QJsonObject>
+#include <QJsonArray>
+#include <QJsonValueRef>
+#include <QString>
+
+
+/**
+ * @brief Sport::getCategories Reads all possible categories (also called units).
+ * @return A set of all category names.
+ */
+set<QString> Sport::getCategories() {
+    set<QString> categoryNames;
+
+    for (const QJsonValueRef& unit : this->discipline["units"].toArray()) {
+        categoryNames.insert(unit.toObject()["eventUnitName"].toString());
+    }
+
+    return categoryNames;
+}
+
+/**
+ * @brief Sport::getCompetitorsByCategory Searches for all competitors, who took part in the given category.
+ * @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;
+
+    for (const QJsonValueRef& unitRef : this->discipline["units"].toArray()) {
+        QJsonObject unit = unitRef.toObject();
+
+        // search all units with the same category
+        if (QString::compare(unit["eventUnitName"].toString(), category, Qt::CaseSensitive) == 0) {
+
+            // add all competitors from one unit
+            for (const QJsonValueRef& comp : unit["competitors"].toArray()) {
+                competitors.push_back(comp.toObject());
+            }
+
+        }
+    }
+
+    return competitors;
+}
diff --git a/src/discipline/Sport.h b/src/discipline/Sport.h
new file mode 100644
index 0000000..9c993fb
--- /dev/null
+++ b/src/discipline/Sport.h
@@ -0,0 +1,41 @@
+
+#ifndef ITAT_CHALLANGE_OLYMPICS_SPORT_H
+#define ITAT_CHALLANGE_OLYMPICS_SPORT_H
+
+
+#include <set>
+
+#include <QJsonObject>
+#include <QJsonDocument>
+#include <QString>
+
+using namespace std;
+
+
+class Sport {
+
+public:
+
+    Sport(QJsonObject discipline) {
+        this->discipline = discipline;
+    }
+
+    set<QString> getCategories();
+    QJsonArray getCompetitorsByCategory(QString category);
+
+    // chainable
+    QJsonObject filterByName(QJsonObject discipline, QString name);
+    QJsonObject filterByCountry(QJsonObject discipline, QString name);
+
+    void setDiscipline(QJsonObject discipline) {
+        this->discipline = discipline;
+    }
+
+private:
+
+    QJsonObject discipline;
+
+};
+
+
+#endif //ITAT_CHALLANGE_OLYMPICS_SPORT_H