sgstatecombomodel.hpp

// This file is part of the SGSolve library for stochastic games
// Copyright (C) 2019 Benjamin A. Brooks
// 
// SGSolve 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 3 of the License, or
// (at your option) any later version.
// 
// SGSolve 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, see
// <http://www.gnu.org/licenses/>.
// 
// Benjamin A. Brooks
// ben@benjaminbrooks.net
// Chicago, IL
 
#ifndef SGSTATECOMBOMODEL_HPP
#define SGSTATECOMBOMODEL_HPP
 
#include "sgplotcontroller.hpp"
#include <QAbstractListModel>
#include <QComboBox>
 
 
class SGStateComboModel : public QAbstractListModel
{
  Q_OBJECT;
private:
  SGPlotController * controller;
 
public:
  SGStateComboModel(SGPlotController * _controller):
    controller(_controller)
  {
    connect(controller,SIGNAL(solutionChanged()),
            this,SLOT(changeLayout()));
  }
 
  int rowCount(const QModelIndex & parent) const
  {
    if (controller->hasSolution())
      return controller->getCurrentIter()->getActions().size()+1;
    else
      return 1;
  } // rowCount
 
  QVariant data(const QModelIndex & index, int role) const
  {
    if (index.row()>0)
      {
    QString dataString = QString("S")+QString::number(index.row()-1);
    // if ( (index.row()-1) == controller->getCurrentIter()->getBestState())
    //   dataString += QString("*");
    return dataString;
      }
    else if (index.row()==0)
      return QString("-");
  }
 
public slots:
  void changeState(int index)
  { controller->setState(index-1); }
 
  void changeLayout()
  { 
    emit layoutChanged(); 
  }
 
}; // SGStateComboModel_V2
 
#endif