pilotclient/src/blackgui/models/aircraftcategorytreemodel.cpp

/* Copyright (C) 2019
 * 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. 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 "blackgui/models/aircraftcategorytreemodel.h"
#include "blackgui/models/columnformatters.h"
#include "blackgui/models/columns.h"
#include "blackmisc/compare.h"
#include "blackmisc/icon.h"
#include "blackmisc/threadutils.h"
#include "blackmisc/variant.h"
#include "blackmisc/propertyindex.h"

#include <QList>
#include <QMap>
#include <QStandardItem>
#include <QStandardItemModel>
#include <QString>
#include <Qt>
#include <QtDebug>
#include <QtGlobal>

using namespace BlackMisc;
using namespace BlackMisc::PhysicalQuantities;
using namespace BlackMisc::Aviation;

namespace BlackGui
{
    namespace Models
    {
        CAircraftCategoryTreeModel::CAircraftCategoryTreeModel(QObject *parent) : QStandardItemModel(parent)
        {
            m_columns.addColumn(CColumn::standardString("id", CAircraftCategory::IndexDbIntegerKey));
            m_columns.addColumn(CColumn::standardString("description", CAircraftCategory::IndexDescription));
            m_columns.addColumn(CColumn::standardString("changed", CAircraftCategory::IndexUtcTimestampFormattedYmdhms));
        }

        void CAircraftCategoryTreeModel::updateContainer(const CAircraftCategoryList &categories)
        {
            Q_ASSERT_X(CThreadUtils::isCurrentThreadObjectThread(this), Q_FUNC_INFO, "Wrong thread");

            this->clear();
            if (categories.isEmpty())  { return; }

            m_categories = categories;
            m_categories.sortByLevel();
            QMap<int, QStandardItem *> items;
            this->setColumnCount(m_columns.size() + 1);

            for (const CAircraftCategory &category : m_categories)
            {
                QList<QStandardItem *> categoryRow;

                // ownership of QStandardItem is taken by model
                QStandardItem *si = new QStandardItem(
                    category.isAssignable() ? CIcons::paperPlane16() : CIcons::folder16(),
                    category.getLevelAndName()
                );
                si->setEditable(false);
                categoryRow.push_back(si);

                // add all clumns
                for (const CColumn &column : m_columns.columns())
                {
                    const CPropertyIndex i(column.getPropertyIndex());
                    const CVariant v(category.propertyByIndex(i));
                    const CVariant f = column.getFormatter()->displayRole(v);
                    si = new QStandardItem(f.toQString(true));
                    si->setEditable(false); // make not editable
                    categoryRow.push_back(si);
                } // columns

                // add all items
                if (categoryRow.isEmpty()) { continue; }

                QStandardItem *parent = categoryRow.first();
                if (category.isFirstLevel())
                {
                    this->invisibleRootItem()->appendRow(categoryRow);
                }
                else
                {
                    const int p = category.getDepth() - 1;
                    Q_ASSERT_X(items[p], Q_FUNC_INFO, "No parent item");
                    items[p]->appendRow(categoryRow);
                }
                items.insert(category.getDepth(), parent);
            }
        }

        void CAircraftCategoryTreeModel::clear()
        {
            m_categories.clear();
            QStandardItemModel::clear();
        }
    }  // namespace
} // namespace