sgprobabilitytablemodel.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 SGPROBABILITYTABLEMODEL_H
#define SGPROBABILITYTABLEMODEL_H
 
#include <QtWidgets>
#include <QAbstractTableModel>
#include <QTableView>
#include "sggame.hpp"
#include "sgpayofftablemodel.hpp"
 
 
class SGProbabilityTableModel : public SGPayoffTableModel
{
  Q_OBJECT
 
private:
 
  int nextState;
 
public:
 
  SGProbabilityTableModel(SGGame * _game,
              int _state, int _nextState):
    SGPayoffTableModel(_game,_state), nextState(_nextState)
  { }
 
 
  QVariant data(const QModelIndex & index,
        int role) const Q_DECL_OVERRIDE;
    
 
  bool setData(const QModelIndex & index, const QVariant & value, int role);
 
  bool setNextState(int newState)
  {
    if (newState < 0 || newState > game->getNumStates())
      return false;
    nextState = newState;
    return true;
  }
  
 
}; // SGProbabilityTableModel
 
#endif