153 lines
2.9 KiB
C++
Executable File
153 lines
2.9 KiB
C++
Executable File
#include "AddModel.h"
|
|
|
|
#include <QDebug>
|
|
|
|
|
|
void AddModel::setHeaderList(const HeaderList& headerList)
|
|
{
|
|
m_headerList = headerList;
|
|
beginResetModel();
|
|
endResetModel();
|
|
}
|
|
|
|
|
|
const HeaderList& AddModel::getHeaderList() const
|
|
{
|
|
return m_headerList;
|
|
}
|
|
|
|
|
|
bool AddModel::removeRow(int row, const QModelIndex &parent)
|
|
{
|
|
//qWarning() << "removeRow";
|
|
|
|
(void)parent;
|
|
|
|
if (row >= m_dataList.count())
|
|
{
|
|
//qWarning() << "removeRows false";
|
|
return false;
|
|
}
|
|
|
|
beginRemoveRows(parent, row, row);
|
|
m_dataList.removeAt(row);
|
|
endRemoveRows();
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
void AddModel::clear()
|
|
{
|
|
beginResetModel();
|
|
m_dataList.clear();
|
|
endResetModel();
|
|
}
|
|
|
|
|
|
QVariant AddModel::data(const QModelIndex &index, int role) const
|
|
{
|
|
switch (role)
|
|
{
|
|
case Qt::EditRole:
|
|
case Qt::DisplayRole:
|
|
{
|
|
if (index.row() >= m_dataList.count())
|
|
{
|
|
return QVariant(); // probably the new line was not added yet (use setData the first time to create it)
|
|
}
|
|
|
|
return m_dataList.at(index.row()).at(index.column());
|
|
}
|
|
}
|
|
|
|
return QVariant();
|
|
}
|
|
|
|
|
|
bool AddModel::setData(const QModelIndex &index, const QVariant &value, int role)
|
|
{
|
|
if (role != Qt::EditRole)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (index.row() >= m_dataList.count()) // no row in dataList?
|
|
{
|
|
Data newData; // create new row
|
|
int headerListCount = m_headerList.count();
|
|
newData.reserve(headerListCount);
|
|
|
|
for (int i = 0; i < headerListCount; i++) // preadd empty columns
|
|
{
|
|
newData.append(QString());
|
|
}
|
|
|
|
m_dataList.append(newData); // finally add new line to dataList
|
|
}
|
|
|
|
Data data = m_dataList.at(index.row()); // get old row
|
|
data.replace(index.column(), value.toString()); // replace column in row
|
|
m_dataList.replace(index.row(), data); // finally replace row in dataList
|
|
|
|
beginResetModel();
|
|
endResetModel(); // tell the model to show the new empty line
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
QVariant::Type AddModel::getColumnType(const int column) const
|
|
{
|
|
if (column < 0 || column >= m_headerList.size())
|
|
return QVariant::Invalid;
|
|
return m_headerList.at(column).second;
|
|
}
|
|
|
|
|
|
QVariant AddModel::headerData(int section, Qt::Orientation orientation, int role) const
|
|
{
|
|
if (orientation != Qt::Horizontal)
|
|
return QVariant();
|
|
|
|
switch (role)
|
|
{
|
|
case Qt::DisplayRole:
|
|
{
|
|
return m_headerList.at(section).first;
|
|
}
|
|
}
|
|
|
|
return QVariant();
|
|
}
|
|
|
|
|
|
int AddModel::rowCount(const QModelIndex &) const
|
|
{
|
|
return m_dataList.count() + 1 /* empty row for new data */; // yes, if the user wants to add more than 1 line!
|
|
}
|
|
|
|
|
|
int AddModel::columnCount(const QModelIndex &) const
|
|
{
|
|
return m_headerList.count();
|
|
}
|
|
|
|
|
|
Qt::ItemFlags AddModel::flags(const QModelIndex &) const
|
|
{
|
|
return Qt::ItemIsSelectable | Qt::ItemIsEnabled | Qt::ItemIsEditable;
|
|
}
|
|
|
|
|
|
void AddModel::setTablename(const QString& tableName)
|
|
{
|
|
m_tableName = tableName;
|
|
}
|
|
|
|
|
|
const QString& AddModel::getTableName() const
|
|
{
|
|
return m_tableName;
|
|
}
|