1 files changed, 181 insertions, 0 deletions
diff --git a/src/Authoring/Studio/Palettes/Inspector/VariantsGroupModel.cpp b/src/Authoring/Studio/Palettes/Inspector/VariantsGroupModel.cpp
new file mode 100644
index 00000000..3a2a2a45
--- /dev/null
+++ b/src/Authoring/Studio/Palettes/Inspector/VariantsGroupModel.cpp
@@ -0,0 +1,181 @@
+/****************************************************************************
+**
+** Copyright (C) 2019 The Qt Company Ltd.
+** Contact: https://www.qt.io/licensing/
+**
+** This file is part of Qt 3D Studio.
+**
+** $QT_BEGIN_LICENSE:GPL-EXCEPT$
+** Commercial License Usage
+** Licensees holding valid commercial Qt licenses may use this file in
+** accordance with the commercial license agreement provided with the
+** Software or, alternatively, in accordance with the terms contained in
+** a written agreement between you and The Qt Company. For licensing terms
+** and conditions see https://www.qt.io/terms-conditions. For further
+** information use the contact form at https://www.qt.io/contact-us.
+**
+** GNU General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU
+** General Public License version 3 as published by the Free Software
+** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT
+** included in the packaging of this file. Please review the following
+** information to ensure the GNU General Public License requirements will
+** be met: https://www.gnu.org/licenses/gpl-3.0.html.
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#include "VariantsGroupModel.h"
+#include "VariantsTagModel.h"
+#include "StudioApp.h"
+#include "Core.h"
+#include "QDebug" // TODO: remove
+#include "Qt3DSDMStudioSystem.h"
+#include "ClientDataModelBridge.h"
+#include "IDocumentEditor.h"
+#include "VariantTagDialog.h"
+
+VariantsGroupModel::VariantsGroupModel(QObject *parent)
+    : QAbstractListModel(parent)
+{
+
+}
+
+void VariantsGroupModel::refresh()
+{
+    int instance = g_StudioApp.GetCore()->GetDoc()->GetSelectedInstance();
+    auto bridge = g_StudioApp.GetCore()->GetDoc()->GetStudioSystem()->GetClientDataModelBridge();
+
+    if (instance == 0 || !bridge->IsLayerInstance(instance)) {
+        m_instance = 0;
+        m_property = 0;
+        return;
+    }
+
+    auto propertySystem = g_StudioApp.GetCore()->GetDoc()->GetStudioSystem()->GetPropertySystem();
+    m_instance = instance;
+    m_property = propertySystem->GetAggregateInstancePropertyByName(instance, L"variants");
+
+    qt3dsdm::SValue sValue;
+    if (propertySystem->GetInstancePropertyValue(m_instance, m_property, sValue)) {
+        beginResetModel();
+        m_data.clear();
+
+        QString val = QString::fromWCharArray(
+                      qt3dsdm::get<qt3dsdm::TDataStrPtr>(sValue)->GetData());
+        // TODO: remove qDebug when the variants work is fully done.
+        qDebug() << "\x1b[42m \x1b[1m" << __FUNCTION__
+                 << ", val=" << val
+                 << "\x1b[m";
+        QHash<QString, QStringList> propTags;
+        if (!val.isEmpty()) {
+            const QStringList propTagsList = val.split(QChar(','));
+            for (auto &propTag : propTagsList) {
+                const QStringList propTagPair = propTag.split(QChar(':'));
+                propTags[propTagPair[0]].append(propTagPair[1]);
+            }
+        }
+
+        // build the variants data model
+        const auto variantsDef = g_StudioApp.GetCore()->getProjectFile().variantsDef();
+        for (auto &group : variantsDef) {
+            TagGroupData g;
+            g.m_title = group.m_title;
+            g.m_color = group.m_color;
+
+            VariantsTagModel *m = new VariantsTagModel(this);
+            QVector<std::pair<QString, bool> > tags;
+            for (int i = 0; i < group.m_tags.length(); ++i)
+                tags.append({group.m_tags[i], propTags[group.m_title].contains(group.m_tags[i])});
+
+            m->init(tags);
+            g.m_tagsModel = m;
+
+            m_data.push_back(g);
+        }
+
+        endResetModel();
+    }
+}
+
+int VariantsGroupModel::rowCount(const QModelIndex &parent) const
+{
+    // For list models only the root node (an invalid parent) should return the list's size. For all
+    // other (valid) parents, rowCount() should return 0 so that it does not become a tree model.
+    if (parent.isValid())
+        return 0;
+
+    return m_data.size();
+}
+
+QVariant VariantsGroupModel::data(const QModelIndex &index, int role) const
+{
+    if (!index.isValid())
+        return QVariant();
+
+    if (role == GroupTitleRole)
+        return m_data.at(index.row()).m_title;
+    else if (role == GroupColorRole)
+        return m_data.at(index.row()).m_color;
+    else if (role == TagRole)
+        return QVariant::fromValue(m_data.at(index.row()).m_tagsModel);
+
+    return QVariant();
+}
+
+void VariantsGroupModel::setTagState(const QString &group, const QString &tag, bool selected)
+{
+    QString val;
+    QString tagsStr;
+    bool skipFirst = false;
+    for (auto &g : qAsConst(m_data)) {
+        if (g.m_title == group)
+            g.m_tagsModel->updateTagState(tag, selected);
+
+        tagsStr = g.m_tagsModel->serialize(g.m_title);
+        if (!tagsStr.isEmpty()) {
+            if (skipFirst)
+                val.append(QChar(','));
+            val.append(tagsStr);
+            skipFirst = true;
+        }
+    }
+
+    auto sVal = std::make_shared<qt3dsdm::CDataStr>(Q3DStudio::CString::fromQString(val));
+    Q3DStudio::SCOPED_DOCUMENT_EDITOR(*g_StudioApp.GetCore()->GetDoc(), QObject::tr("Set Property"))
+            ->SetInstancePropertyValue(m_instance, m_property, sVal);
+}
+
+void VariantsGroupModel::addNewTag(const QString &group)
+{
+    VariantTagDialog dlg(VariantTagDialog::AddTag, group);
+
+    if (dlg.exec() == QDialog::Accepted)
+        g_StudioApp.GetCore()->getProjectFile().addVariantTag(group, dlg.getNames().second);
+}
+
+void VariantsGroupModel::addNewGroup()
+{
+    VariantTagDialog dlg(VariantTagDialog::AddGroup);
+
+    if (dlg.exec() == QDialog::Accepted)
+        g_StudioApp.GetCore()->getProjectFile().addVariantGroup(dlg.getNames().second);
+}
+
+QHash<int, QByteArray> VariantsGroupModel::roleNames() const
+{
+    auto names = QAbstractListModel::roleNames();
+    names.insert(GroupTitleRole, "group");
+    names.insert(GroupColorRole, "color");
+    names.insert(TagRole, "tags");
+    return names;
+}
+
+Qt::ItemFlags VariantsGroupModel::flags(const QModelIndex &index) const
+{
+    if (!index.isValid())
+        return Qt::NoItemFlags;
+
+    return Qt::ItemIsEditable;
+}