Difference between revisions of "Qt/ModelView/AbstractListModelCheckable"

#include <QApplication>
 
#include "form.h"
 
int main(int argc, char *argv[])
{
  QApplication app(argc, argv);
  Form form;
 
  form.show();
  return app.exec();
}

#ifndef AbstractListModelCheckable_H
#define AbstractListModelCheckable_H
 
#include <QAbstractListModel>
 
#include <QVector>
 
class Item
{
public:
  Item(const QString& name = QString(), const Qt::CheckState displayed = Qt::Unchecked) :
  Name(name), Displayed(displayed){}
  QString Name;
  Qt::CheckState Displayed;
};
 
class AbstractListModelCheckable : public QAbstractListModel
{
public:
  AbstractListModelCheckable();
 
  QVariant data(const QModelIndex& index, int role) const;
 
  bool setData (const QModelIndex &index, const QVariant &value, int role);
 
  int rowCount(const QModelIndex  &parent=QModelIndex() ) const;
 
  int setItems(const QVector<Item>& items);
 
  Qt::ItemFlags flags (const QModelIndex  &index ) const;
 
protected:
  QVector<Item> Items;
};
 
#endif

#include "AbstractListModelCheckable.h"
 
#include <iostream>
 
int AbstractListModelCheckable::setItems(const QVector<Item>& items)
{
  emit beginResetModel();
  this->Items = items;
  emit endResetModel();
}
 
Qt::ItemFlags AbstractListModelCheckable::flags (const QModelIndex  &index ) const
{
  if (index.row() < 0 || index.row() >= rowCount() || !index.isValid())
    {
    return Qt::NoItemFlags;
    }
  return Qt::ItemIsUserCheckable | Qt::ItemIsEditable | Qt::ItemIsEnabled;
}
 
AbstractListModelCheckable::AbstractListModelCheckable() : QAbstractListModel()
{
 
}
 
int AbstractListModelCheckable::rowCount(const QModelIndex& parent) const
{
  return this->Items.size();
}
 
QVariant AbstractListModelCheckable::data (const QModelIndex  &index , int role ) const
{
  if (index.row() < 0 || index.row() >= rowCount() || !index.isValid())
    {
    return QVariant();
    }
 
  if (role == Qt::DisplayRole || role == Qt::EditRole)
    {
    return this->Items.at(index.row()).Name;
    }
 
  if(role == Qt::CheckStateRole)
    {
    return this->Items[index.row()].Displayed;
    }
 
  return QVariant();
}
 
bool AbstractListModelCheckable::setData (const QModelIndex &index, const QVariant &value, int role)
{
  if (index.row() < 0 || index.row() >= rowCount() || !index.isValid())
    {
    return false;
    }
 
  if(role == Qt::CheckStateRole)
    {
    this->Items[index.row()].Displayed = static_cast<Qt::CheckState>(value.toUInt());
    }
 
  emit dataChanged(index, index);
  return true;
}

#ifndef FORM_H
#define FORM_H
 
#include "ui_form.h"
 
class Form : public QWidget, public Ui::Form
{
    Q_OBJECT
 
public:
    Form(QWidget *parent = 0);
 
  public slots:
 
 
};
 
#endif

#include <QAbstractTableModel>
#include <QtGui>
 
#include <iostream>
#include <sstream>
 
#include "form.h"
#include "AbstractListModelCheckable.h"
 
Form::Form(QWidget *parent)
    : QWidget(parent)
{
  setupUi(this);
 
  AbstractListModelCheckable* model = new AbstractListModelCheckable;
 
  QVector<Item> items;
  items.push_back(Item("Test0", Qt::Unchecked));
  items.push_back(Item("Test1", Qt::Checked));
  items.push_back(Item("Test2", Qt::Unchecked));
  model->setItems(items);
  this->tableView->setModel(model);
}