86 lines
2.7 KiB
C++
86 lines
2.7 KiB
C++
#ifndef REMOTEREPLAYLIST_TREEWIDGET_H
|
|
#define REMOTEREPLAYLIST_TREEWIDGET_H
|
|
|
|
#include <QAbstractItemModel>
|
|
#include <QDateTime>
|
|
#include <QTreeView>
|
|
#include "pb/serverinfo_replay.pb.h"
|
|
#include "pb/serverinfo_replay_match.pb.h"
|
|
|
|
class Response;
|
|
class AbstractClient;
|
|
class QSortFilterProxyModel;
|
|
|
|
class RemoteReplayList_TreeModel : public QAbstractItemModel {
|
|
Q_OBJECT
|
|
private:
|
|
class MatchNode;
|
|
class ReplayNode;
|
|
class Node {
|
|
protected:
|
|
QString name;
|
|
public:
|
|
Node(const QString &_name)
|
|
: name(_name) { }
|
|
virtual ~Node() { };
|
|
QString getName() const { return name; }
|
|
};
|
|
class MatchNode : public Node, public QList<ReplayNode *> {
|
|
private:
|
|
ServerInfo_ReplayMatch matchInfo;
|
|
public:
|
|
MatchNode(const ServerInfo_ReplayMatch &_matchInfo);
|
|
~MatchNode();
|
|
void clearTree();
|
|
const ServerInfo_ReplayMatch &getMatchInfo() { return matchInfo; }
|
|
};
|
|
class ReplayNode : public Node {
|
|
private:
|
|
MatchNode *parent;
|
|
ServerInfo_Replay replayInfo;
|
|
public:
|
|
ReplayNode(const ServerInfo_Replay &_replayInfo, MatchNode *_parent)
|
|
: Node(QString::fromStdString(_replayInfo.replay_name())), parent(_parent), replayInfo(_replayInfo) { }
|
|
MatchNode *getParent() const { return parent; }
|
|
const ServerInfo_Replay &getReplayInfo() { return replayInfo; }
|
|
};
|
|
|
|
AbstractClient *client;
|
|
QList<MatchNode *> replayMatches;
|
|
|
|
QIcon dirIcon, fileIcon;
|
|
void clearTree();
|
|
signals:
|
|
void treeRefreshed();
|
|
private slots:
|
|
void replayListFinished(const Response &r);
|
|
public:
|
|
RemoteReplayList_TreeModel(AbstractClient *_client, QObject *parent = 0);
|
|
~RemoteReplayList_TreeModel();
|
|
int rowCount(const QModelIndex &parent = QModelIndex()) const;
|
|
int columnCount(const QModelIndex &/*parent*/ = QModelIndex()) const;
|
|
QVariant data(const QModelIndex &index, int role) const;
|
|
QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;
|
|
QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const;
|
|
QModelIndex parent(const QModelIndex &index) const;
|
|
Qt::ItemFlags flags(const QModelIndex &index) const;
|
|
void refreshTree();
|
|
ServerInfo_Replay const* getReplay(const QModelIndex &index) const;
|
|
ServerInfo_ReplayMatch const* getReplayMatch(const QModelIndex &index) const;
|
|
void addMatchInfo(const ServerInfo_ReplayMatch &matchInfo);
|
|
};
|
|
|
|
class RemoteReplayList_TreeWidget : public QTreeView {
|
|
private:
|
|
RemoteReplayList_TreeModel *treeModel;
|
|
QSortFilterProxyModel *proxyModel;
|
|
ServerInfo_Replay const *getNode(const QModelIndex &ind) const;
|
|
public:
|
|
RemoteReplayList_TreeWidget(AbstractClient *_client, QWidget *parent = 0);
|
|
ServerInfo_Replay const *getCurrentReplay() const;
|
|
ServerInfo_ReplayMatch const *getCurrentReplayMatch() const;
|
|
void refreshTree();
|
|
void addMatchInfo(const ServerInfo_ReplayMatch &matchInfo);
|
|
};
|
|
|
|
#endif
|