pilotclient/src/blackgui/models/actionmodel.cpp

/* Copyright (C) 2015
 * swift project Community / Contributors
 *
 * This file is part of swift project. It is subject to the license terms in the LICENSE file found in the top-level
 * directory of this distribution and at http://www.swift-project.org/license.html. No part of swift project,
 * including this file, may be copied, modified, propagated, or distributed except according to the terms
 * contained in the LICENSE file.
 */

#include "blackcore/inputmanager.h"
#include "blackgui/models/actionitem.h"
#include "blackgui/models/actionmodel.h"
#include "blackmisc/icons.h"

#include <QString>
#include <QStringList>
#include <QStringBuilder>
#include <QtGlobal>

using namespace BlackMisc;
using namespace BlackCore;

namespace BlackGui
{
    namespace Models
    {
        CActionModel::CActionModel(QObject *parent) :
            QAbstractItemModel(parent),
            m_rootItem(new CActionItem(QString(), QString()))
        {
            setupModelData();
        }

        CActionModel::~CActionModel()
        { }

        int CActionModel::columnCount(const QModelIndex &parent) const
        {
            return parent.isValid() ?
                   static_cast<CActionItem *>(parent.internalPointer())->getColumnCount() :
                   m_rootItem->getColumnCount();
        }

        QVariant CActionModel::data(const QModelIndex &index, int role) const
        {
            if (!index.isValid()) { return QVariant(); }

            const CActionItem *item = static_cast<CActionItem *>(index.internalPointer());
            Q_ASSERT_X(item, Q_FUNC_INFO, "Missing item");

            if (role == ActionRole) { return item->getAction(); }
            if (role == Qt::DisplayRole) { return item->getActionName(); }
            if (role == Qt::DecorationRole) { return item->getIcon(); }

            return {};
        }

        Qt::ItemFlags CActionModel::flags(const QModelIndex &index) const
        {
            if (!index.isValid()) { return 0; }
            const CActionItem *item = static_cast<CActionItem *>(index.internalPointer());
            const Qt::ItemFlags flags = QAbstractItemModel::flags(index);
            const bool selectable = item && !item->hasChildren(); // only leafs are selectable
            return selectable ? flags | Qt::ItemIsSelectable : flags & ~Qt::ItemIsSelectable;
        }

        QModelIndex CActionModel::index(int row, int column, const QModelIndex &parent) const
        {
            if (!hasIndex(row, column, parent)) { return QModelIndex(); }

            const CActionItem *parentItem = parent.isValid() ?
                                            static_cast<CActionItem *>(parent.internalPointer()) :
                                            m_rootItem.data();

            CActionItem *childItem = parentItem->getChildByRow(row);
            return childItem ?
                   createIndex(row, column, childItem) :
                   QModelIndex();
        }

        QModelIndex CActionModel::parent(const QModelIndex &index) const
        {
            if (!index.isValid()) { return {}; }

            CActionItem *childItem = static_cast<CActionItem *>(index.internalPointer());
            CActionItem *parentItem = childItem->getParentItem();

            if (parentItem == m_rootItem.data()) { return {}; }

            return createIndex(parentItem->getRow(), 0, parentItem);
        }

        int CActionModel::rowCount(const QModelIndex &parent) const
        {
            if (parent.column() > 0) { return 0; }
            const CActionItem *parentItem = parent.isValid() ?
                                            static_cast<CActionItem *>(parent.internalPointer()) :
                                            m_rootItem.data();
            return parentItem->getChildCount();
        }

        void CActionModel::setupModelData()
        {
            m_rootItem.reset(new CActionItem(QString(), QString()));

            const QMap<QString, QPixmap> availableActionsAndIcons = CInputManager::instance()->allAvailableActionsAndIcons();
            QStringList keys = availableActionsAndIcons.keys();
            keys.sort();
            for (const QString &actionPath : as_const(keys))
            {
                QString currentPath;
                const QStringList tokens = actionPath.split("/", QString::SkipEmptyParts);
                CActionItem *parentItem = m_rootItem.data();
                for (const auto &token : tokens)
                {
                    currentPath += QLatin1Char('/') % token;
                    CActionItem *child = parentItem->findChildByName(token);
                    if (child == nullptr)
                    {
                        const bool isAction = currentPath == actionPath; // action istelf, or just a node?
                        const QPixmap icon = isAction ? availableActionsAndIcons[actionPath] : CIcons::empty16();
                        child = new CActionItem(isAction ? actionPath : "", token, icon, parentItem);
                        parentItem->appendChild(child);
                    }
                    Q_ASSERT(child);
                    parentItem = child;
                }
            }
        }
    } // ns
} // ns