summaryrefslogtreecommitdiff
path: root/src/modlistbypriorityproxy.h
blob: 6ec04b9bc3e95bc8c553992b6f846234b5ca9253 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#ifndef MODLISBYPRIORITYPROXY_H
#define MODLISBYPRIORITYPROXY_H

#include <optional>
#include <vector>

#include <QAbstractProxyModel>
#include <QModelIndex>
#include <QMultiHash>
#include <QStringList>
#include <QIcon>
#include <QSet>

#include "modinfo.h"

class ModList;

class ModListByPriorityProxy : public QAbstractProxyModel
{
  Q_OBJECT

public:
  explicit ModListByPriorityProxy(ModList* modList, QObject* parent = nullptr);
  ~ModListByPriorityProxy();

  void setSourceModel(QAbstractItemModel* sourceModel) override;

  int rowCount(const QModelIndex& parent = QModelIndex()) const override;
  QModelIndex parent(const QModelIndex& child) const override;
  QModelIndex index(int row, int column, const QModelIndex& parent = QModelIndex()) const override;
  int columnCount(const QModelIndex& index) const override;
  bool hasChildren(const QModelIndex& parent) const override;

  QModelIndex mapFromSource(const QModelIndex& sourceIndex) const override;
  QModelIndex mapToSource(const QModelIndex& proxyIndex) const override;

private:

  void buildTree();

  struct ModInfoWithPriority {
    const ModInfo::Ptr mod;
    const int priority;
  };

  std::vector<ModInfoWithPriority> topLevelItems() const;
  std::vector<ModInfoWithPriority> childItems(int priority) const;
  std::optional<ModInfoWithPriority> separator(int priority) const;

private:

  ModList* m_ModList;

};

#endif //GROUPINGPROXY_H