File indexing completed on 2024-05-05 05:40:31

 /***************************************************************************
  *  Copyright (C) 2022 by Renaud Guezennec                               *
  *   http://www.rolisteam.org/contact                                      *
  *                                                                         *
  *   This software 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; either version 2 of the License, or     *
  *   (at your option) any later version.                                   *
  *                                                                         *
  *   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.,                                       *
  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  ***************************************************************************/
 #include "model/statemodel.h"
 #include "model/nonplayablecharactermodel.h"
 #include <QSet>
 
 StateModel::StateModel(QObject* parent) : QAbstractListModel(parent) {}
 
 int StateModel::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() + 1;
 }
 
 QVariant StateModel::data(const QModelIndex& index, int role) const
 {
     if(!index.isValid())
         return QVariant();
 
     auto i= index.row();
 
     if(i == 0 && role == Qt::DisplayRole)
         return tr("Any State");
     else
         return {};
 
     return m_data[i - 1];
 }
 
 void StateModel::setSourceModel(const campaign::NonPlayableCharacterModel* source)
 {
 
     connect(source, &campaign::NonPlayableCharacterModel::dataChanged, this,
             [this, &source](const QModelIndex& start, const QModelIndex& end, QList<int> roles)
             {
                 if(roles.contains(campaign::NonPlayableCharacterModel::RoleColor))
                 {
                     auto index= start;
                     for(int i= start.row(); i < end.row(); ++i)
                     {
                         auto stateId= source->data(index, campaign::NonPlayableCharacterModel::RoleState).toString();
 
                         if(m_data.contains(stateId))
                             return;
 
                         beginInsertRows(QModelIndex(), rowCount(), rowCount());
                         m_data.append(stateId);
                         endInsertRows();
                     }
                 }
             });
     connect(source, &campaign::NonPlayableCharacterModel::rowsInserted, this,
             [this, source](const QModelIndex& parent, int start, int end)
             {
                 for(int i= start; i < end; ++i)
                 {
                     if(i < source->rowCount())
                     {
                         auto stateId
                             = source->data(source->index(i, 0, parent), campaign::NonPlayableCharacterModel::RoleState)
                                   .toString();
 
                         if(m_data.contains(stateId))
                             return;
 
                         beginInsertRows(QModelIndex(), rowCount(), rowCount());
                         m_data.append(stateId);
                         endInsertRows();
                     }
                 }
             });
 
     QSet<QString> ids;
     auto const& npcs= source->npcList();
 
     std::for_each(std::begin(npcs), std::end(npcs),
                   [&ids](const std::unique_ptr<campaign::NonPlayableCharacter>& npc)
                   {
                       ids.insert(npc->stateId());
                   });
 
     beginResetModel();
     m_data= ids.values();
     endResetModel();
 }