Initial commit

Lts/Tools/SqlBrowser/AddModel.cpp

@@ -0,0 +1,152 @@
+#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;
+}