+
−/*
+
− * Hedgewars, a free turn based strategy game
+
− * Copyright (c) 2004-2015 Andrey Korotaev <unC0Rr@gmail.com>
+
− *
+
− * This program is free software; you can redistribute it and/or modify
+
− * it under the terms of the GNU General Public License as published by
+
− * the Free Software Foundation; version 2 of the License
+
− *
+
− * This program is distributed in the hope that it will be useful,
+
− * but WITHOUT ANY WARRANTY; without even the implied warranty of
+
− * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+
− * GNU General Public License for more details.
+
− *
+
− * You should have received a copy of the GNU General Public License
+
− * along with this program; if not, write to the Free Software
+
− * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
− */
+
−
+
−/**
+
− * @file
+
− * @brief RoomsListModel class implementation
+
− */
+
−
+
−#include <QBrush>
+
−#include <QColor>
+
−#include <QIcon>
+
−
+
−#include "roomslistmodel.h"
+
−#include "MapModel.h"
+
−
+
−RoomsListModel::RoomsListModel(QObject *parent) :
+
−    QAbstractTableModel(parent),
+
−    c_nColumns(9)
+
−{
+
−    m_headerData =
+
−    QStringList()
+
−     << tr("In progress")
+
−     << tr("Room Name")
+
−     << tr("C")
+
−     << tr("T")
+
−     << tr("Owner")
+
−     << tr("Map")
+
−     << tr("Script")
+
−     << tr("Rules")
+
−     << tr("Weapons");
+
−
+
−    m_staticMapModel = DataManager::instance().staticMapModel();
+
−    m_missionMapModel = DataManager::instance().missionMapModel();
+
−}
+
−
+
−
+
−QVariant RoomsListModel::headerData(int section, Qt::Orientation orientation, int role) const
+
−{
+
−    if(orientation == Qt::Vertical || role != Qt::DisplayRole)
+
−        return QVariant();
+
−    else
+
−        return QVariant(m_headerData.at(section));
+
−}
+
−
+
−
+
−int RoomsListModel::rowCount(const QModelIndex & parent) const
+
−{
+
−    if(parent.isValid())
+
−        return 0;
+
−    else
+
−        return m_data.size();
+
−}
+
−
+
−
+
−int RoomsListModel::columnCount(const QModelIndex & parent) const
+
−{
+
−    if(parent.isValid())
+
−        return 0;
+
−    else
+
−        return c_nColumns;
+
−}
+
−
+
−
+
−QVariant RoomsListModel::data(const QModelIndex &index, int role) const
+
−{
+
−    int column = index.column();
+
−    int row = index.row();
+
−
+
−    // invalid index
+
−    if (!index.isValid())
+
−        return QVariant();
+
−
+
−    // invalid row
+
−    if ((row < 0) || (row >= m_data.size()))
+
−        return QVariant();
+
−
+
−    // invalid column
+
−    if ((column < 0) || (column >= c_nColumns))
+
−        return QVariant();
+
−
+
−    // not a role we have data for
+
−    if (role != Qt::DisplayRole)
+
−        // only custom-align counters
+
−        if ((role != Qt::TextAlignmentRole)
+
−            || ((column != PlayerCountColumn) && (column != TeamCountColumn)))
+
−                // only decorate name column
+
−                if ((role != Qt::DecorationRole) || (column != NameColumn))
+
−                    // only dye map column
+
−                    if ((role != Qt::ForegroundRole) || (column != MapColumn))
+
−                        return QVariant();
+
−
+
−    // decorate room name based on room state
+
−    if (role == Qt::DecorationRole)
+
−    {
+
−        const QIcon roomBusyIcon(":/res/iconDamage.png");
+
−        const QIcon roomBusyIconGreen(":/res/iconDamageLockG.png");
+
−        const QIcon roomBusyIconRed(":/res/iconDamageLockR.png");
+
−        const QIcon roomWaitingIcon(":/res/iconTime.png");
+
−        const QIcon roomWaitingIconGreen(":/res/iconTimeLockG.png");
+
−        const QIcon roomWaitingIconRed(":/res/iconTimeLockR.png");
+
−
+
−        QString flags = m_data.at(row).at(StateColumn);
+
−
+
−        if (flags.contains("g"))
+
−        {
+
−            if (flags.contains("j"))
+
−                return QVariant(roomBusyIconRed);
+
−            else if (flags.contains("p"))
+
−                return QVariant(roomBusyIconGreen);
+
−            else
+
−                return QVariant(roomBusyIcon);
+
−        }
+
−        else
+
−        {
+
−            if (flags.contains("j"))
+
−                return QVariant(roomWaitingIconRed);
+
−            else if (flags.contains("p"))
+
−                return QVariant(roomWaitingIconGreen);
+
−            else
+
−                return QVariant(roomWaitingIcon);
+
−        }
+
−    }
+
−
+
−    QString content = m_data.at(row).at(column);
+
−
+
−    if (role == Qt::DisplayRole)
+
−    {
+
−        // display room names
+
−        if (column == 5)
+
−        {
+
−            // special names
+
−            if (content[0] == '+')
+
−            {
+
−                if (content == "+rnd+") return tr("Random Map");
+
−                if (content == "+maze+") return tr("Random Maze");
+
−                if (content == "+perlin+") return tr("Random Perlin");
+
−                if (content == "+drawn+") return tr("Hand-drawn");
+
−                if (content == "+forts+") return tr("Forts");
+
−            }
+
−
+
−            // prefix ? if map not available
+
−            if (!m_staticMapModel->mapExists(content) &&
+
−                !m_missionMapModel->mapExists(content))
+
−                return QString ("? %1").arg(content);
+
−        }
+
−
+
−        return content;
+
−    }
+
−
+
−    // dye map names red if map not available
+
−    if (role == Qt::ForegroundRole)
+
−    {
+
−        if (content == "+rnd+" ||
+
−            content == "+maze+" ||
+
−            content == "+perlin+" ||
+
−            content == "+drawn+" ||
+
−            content == "+forts+" ||
+
−            m_staticMapModel->mapExists(content) ||
+
−            m_missionMapModel->mapExists(content))
+
−            return QVariant();
+
−        else
+
−            return QBrush(QColor("darkred"));
+
−    }
+
−
+
−    if (role == Qt::TextAlignmentRole)
+
−    {
+
−        return (int)(Qt::AlignHCenter | Qt::AlignVCenter);
+
−    }
+
−
+
−    Q_ASSERT(false);
+
−    return QVariant();
+
−}
+
−
+
−
+
−void RoomsListModel::setRoomsList(const QStringList & rooms)
+
−{
+
−    beginResetModel();
+
−
+
−    m_data.clear();
+
−
+
−    int nRooms = rooms.size();
+
−
+
−    for (int i = 0; i < nRooms; i += c_nColumns)
+
−    {
+
−        QStringList l;
+
−
+
−#if QT_VERSION >= QT_VERSION_CHECK(4, 7, 0)
+
−        l.reserve(c_nColumns);  // small optimisation not supported in old Qt
+
−#endif
+
−
+
−        for (int t = 0; t < c_nColumns; t++)
+
−        {
+
−            l.append(rooms[i + t]);
+
−        }
+
−
+
−        m_data.append(l);
+
−    }
+
−
+
−    endResetModel();
+
−}
+
−
+
−
+
−void RoomsListModel::addRoom(const QStringList & info)
+
−{
+
−    beginInsertRows(QModelIndex(), 0, 0);
+
−
+
−    m_data.prepend(info);
+
−
+
−    endInsertRows();
+
−}
+
−
+
−
+
−int RoomsListModel::rowOfRoom(const QString & name)
+
−{
+
−    int size = m_data.size();
+
−
+
−    if (size < 1)
+
−        return -1;
+
−
+
−    int i = 0;
+
−
+
−    // search for record with matching room name
+
−    while(m_data[i].at(NameColumn) != name)
+
−    {
+
−        i++;
+
−        if(i >= size)
+
−            return -1;
+
−    }
+
−
+
−    return i;
+
−}
+
−
+
−
+
−void RoomsListModel::removeRoom(const QString & name)
+
−{
+
−    int i = rowOfRoom(name);
+
−
+
−    if (i < 0)
+
−        return;
+
−
+
−    beginRemoveRows(QModelIndex(), i, i);
+
−
+
−    m_data.removeAt(i);
+
−
+
−    endRemoveRows();
+
−}
+
−
+
−
+
−void RoomsListModel::updateRoom(const QString & name, const QStringList & info)
+
−{
+
−    int i = rowOfRoom(name);
+
−
+
−    if (i < 0)
+
−        return;
+
−
+
−    m_data[i] = info;
+
−
+
−    emit dataChanged(index(i, 0), index(i, columnCount(QModelIndex()) - 1));
+
−}