Merge pull request #605 from GriffinRichards/encounter-graph

Add encounter charts

include/ui/encountertablemodel.h

@@ -28,7 +28,9 @@ public:
         Slot, Group, Species, MinLevel, MaxLevel, EncounterChance, SlotRatio, EncounterRate, Count
     };
 
-    WildMonInfo encounterData();
+    WildMonInfo encounterData() const { return this->monInfo; }
+    EncounterField encounterField() const { return this->encounterFields.at(this->fieldIndex); }
+    QList<double> percentages() const { return this->slotPercentages; }
     void resize(int rows, int cols);
 
 private:

include/ui/wildmonchart.h

@@ -0,0 +1,98 @@
+#ifndef WILDMONCHART_H
+#define WILDMONCHART_H
+
+#include "encountertablemodel.h"
+
+#include <QWidget>
+
+#if __has_include(<QtCharts>)
+#include <QtCharts>
+
+namespace Ui {
+class WildMonChart;
+}
+
+class WildMonChart : public QWidget
+{
+    Q_OBJECT
+public:
+    explicit WildMonChart(QWidget *parent, const EncounterTableModel *table);
+    ~WildMonChart();
+
+    virtual void closeEvent(QCloseEvent *event) override;
+
+public slots:
+    void setTable(const EncounterTableModel *table);
+    void clearTable();
+    void refresh();
+
+private:
+    Ui::WildMonChart *ui;
+    const EncounterTableModel *table;
+
+    QStringList groupNames;
+    QStringList groupNamesReversed;
+    QStringList speciesInLegendOrder;
+    QMap<int, QString> tableIndexToGroupName;
+
+    struct LevelRange {
+        int min;
+        int max;
+    };
+    QMap<QString, LevelRange> groupedLevelRanges;
+
+    struct Summary {
+        double speciesFrequency = 0.0;
+        QMap<int, double> levelFrequencies;
+    };
+    typedef QMap<QString, Summary> GroupedData;
+
+    QMap<QString, GroupedData> speciesToGroupedData;
+    QMap<QString, QColor> speciesToColor;
+
+
+    QStringList getSpeciesNamesAlphabetical() const;
+    double getSpeciesFrequency(const QString&, const QString&) const;
+    QMap<int, double> getLevelFrequencies(const QString &, const QString &) const;
+    LevelRange getLevelRange(const QString &, const QString &) const;
+    bool usesGroupLabels() const;
+
+    void clearTableData();
+    void readTable();
+    QChart* createSpeciesDistributionChart();
+    QChart* createLevelDistributionChart();
+    QBarSet* createLevelDistributionBarSet(const QString &, const QString &, bool);
+    void refreshSpeciesDistributionChart();
+    void refreshLevelDistributionChart();
+
+    void saveSpeciesColors(const QList<QBarSet*> &);
+    void applySpeciesColors(const QList<QBarSet*> &);
+    QChart::ChartTheme currentTheme() const;
+    void updateTheme();
+    void limitChartAnimation(QChart*);
+
+    void showHelpDialog();
+};
+
+#else
+
+// As of writing our static Qt build for Windows doesn't include the QtCharts module, so we dummy the class out here.
+// The charts module is additionally excluded from Windows in porymap.pro
+#define DISABLE_CHARTS_MODULE
+
+class WildMonChart : public QWidget
+{
+    Q_OBJECT
+public:
+    explicit WildMonChart(QWidget *, const EncounterTableModel *) {};
+    ~WildMonChart() {};
+
+public slots:
+    void setTable(const EncounterTableModel *) {};
+    void clearTable() {};
+    void refresh() {};
+};
+
+#endif // __has_include(<QtCharts>)
+
+#endif // WILDMONCHART_H