bceobjweightstablemodel.hpp

// This file is part of the BCESolve library for games of incomplete
// information
// Copyright (C) 2022 Benjamin A. Brooks
// 
// BCESolve 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.
// 
// BCESolve 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 BCEOBJWEIGHTSTABLEMODEL_HPP
#define BCEOBJWEIGHTSTABLEMODEL_HPP
 
#include <QtWidgets>
#include <QAbstractTableModel>
#include <QTableView>
#include "bcegame.hpp"
#include "bceenumeration.hpp"
 
 
class BCEObjWeightsTableModel : public QAbstractTableModel
{
  Q_OBJECT
 
private:
  vector<double> mainObjWeights;
  vector<double> secondaryObjWeights;
  void setWeightData(vector<int> cellLocation,double value) {
    if (cellLocation[1]==0)
      mainObjWeights[cellLocation[0]] = value;
    else
      secondaryObjWeights[cellLocation[0]] = value;
  }
  vector<string> objectiveLabels;
 
public:
 
  BCEObjWeightsTableModel(BCEGame * _game):
    game(_game)
  {
 
    mainObjWeights = vector<double>(game->getNumObjectives(),0);
    secondaryObjWeights = vector<double>(game->getNumObjectives(),0);
 
    mainObjWeights[0] = 1;
    secondaryObjWeights[1] = 1;
 
    objectiveLabels = game->getObjLabels();
 
    if (objectiveLabels.size() < game->getNumObjectives()) {
      int currLabelIndex = objectiveLabels.size();
      while (currLabelIndex < game->getNumObjectives() + 1) {
    objectiveLabels.push_back(string());
    currLabelIndex++;
      }
    }
  }
 
 
  Qt::ItemFlags flags(const QModelIndex & index) const 
  { return Qt::ItemIsEditable | Qt::ItemIsEnabled | Qt::ItemIsSelectable; }
 
  int rowCount(const QModelIndex & parent = QModelIndex()) const Q_DECL_OVERRIDE
  { return game->getNumObjectives(); }
  int columnCount(const QModelIndex & parent = QModelIndex()) const Q_DECL_OVERRIDE
  { return 2; }
 
  void emitLayoutChanged()
  { emit layoutChanged(); }
 
 
  QVariant data(const QModelIndex & index,
        int role) const Q_DECL_OVERRIDE;
    
 
  bool setData(const QModelIndex & index, const QVariant & value, int role);
 
 
  QVariant headerData(int section,
              Qt::Orientation orientation,
              int role) const Q_DECL_OVERRIDE;
 
  void addObjective(int position) {
    mainObjWeights.insert(mainObjWeights.begin()+position,0);
    secondaryObjWeights.insert(secondaryObjWeights.begin()+position,0);
    objectiveLabels.insert(objectiveLabels.begin()+position,
               game->getObjLabels(position));
  }
 
 
  void removeObjective(int position) {
    mainObjWeights.erase(mainObjWeights.begin() + position);
    secondaryObjWeights.erase(secondaryObjWeights.begin() + position);
    objectiveLabels.erase(objectiveLabels.begin() + position);
  }
 
  const vector<double> getSolverData() {
    return mainObjWeights;
  }
 
  const vector<vector<double> > getMapBoundaryData() {
    vector<vector<double> > weightData;
    weightData.push_back(mainObjWeights);
    weightData.push_back(secondaryObjWeights);
    return weightData;
  }
 
  void resetObjectiveWeights() {
  mainObjWeights.clear();
  secondaryObjWeights.clear();
  mainObjWeights.resize(game->getNumObjectives());
  secondaryObjWeights.resize(game->getNumObjectives());
 
  mainObjWeights[0] = 1;
  secondaryObjWeights[1] = 1;
  }
  
protected:
  BCEGame * game;
 
 
};
 
 
#endif