1 files changed, 147 insertions, 0 deletions
diff --git a/cpp-practicum/Record.cpp b/cpp-practicum/Record.cpp
index 1041b23..ac367df 100644
--- a/cpp-practicum/Record.cpp
+++ b/cpp-practicum/Record.cpp
@@ -1,4 +1,7 @@
 #include "Record.hpp"
+#include "QAbstractitemmodel.h"
+#include "qnamespace.h"
+#include "qvariant.h"
 
 void Record::WriteTo(QFile file) {
   file.open(QFile::ReadWrite | QFile::Text | QFile::Truncate);
@@ -36,3 +39,147 @@ void Record::ReadFrom(QFile file) {
     vendors_.push_back(std::move(vendor));
   }
 }
+
+int BookModel::rowCount(const QModelIndex &parent) const {
+  if (parent.isValid())
+    return 0;
+  return static_cast<int>(record_->GetBooks().size());
+}
+
+int BookModel::columnCount(const QModelIndex &parent) const {
+  if (parent.isValid())
+    return 0;
+  return 6;
+}
+
+QVariant BookModel::headerData(int section, Qt::Orientation orientation,
+                               int role) const {
+  if (role != Qt::DisplayRole)
+    return QVariant();
+  if (orientation == Qt::Horizontal) {
+    switch (section) {
+    case 0:
+      return QStringLiteral("ISBN");
+    case 1:
+      return QStringLiteral("标题");
+    case 2:
+      return QStringLiteral("类型");
+    case 3:
+      return QStringLiteral("作者");
+    case 4:
+      return QStringLiteral("出版社");
+    case 5:
+      return QStringLiteral("库存");
+    default:
+      return QVariant();
+    }
+  }
+  return QVariant();
+}
+
+QVariant BookModel::data(const QModelIndex &index, int role) const {
+  if (role != Qt::DisplayRole)
+    return QVariant();
+
+  if (!index.isValid())
+    return QVariant();
+
+  if (index.row() >= static_cast<int>(record_->GetBooks().size()) ||
+      index.row() < 0)
+    return QVariant();
+
+  int row = index.row();
+  const Book &book = record_->GetBooks()[row];
+
+  int col = index.column();
+  switch (col) {
+  case 0:
+    return QString::fromStdU16String(book.GetIsbn());
+  case 1:
+    return QString::fromStdU16String(book.GetTitle());
+  case 2:
+    return QString::fromStdU16String(book.GetType());
+  case 3:
+    return QString::fromStdU16String(book.GetAuthor());
+  case 4:
+    return QString::fromStdU16String(book.GetPress());
+  case 5:
+    return book.GetStockCount();
+  default:
+    return QVariant();
+  }
+}
+
+bool BookModel::setData(const QModelIndex &index, const QVariant &value,
+                        int role) {
+  if (index.isValid() && role == Qt::EditRole) {
+    int row = index.row();
+    Book &book = record_->GetBooks()[row];
+
+    int col = index.column();
+    switch (col) {
+    case 0:
+      if (!value.canConvert<QString>())
+        return false;
+      book.SetIsbn(value.toString().toStdU16String());
+      return true;
+    case 1:
+      if (!value.canConvert<QString>())
+        return false;
+      book.SetTitle(value.toString().toStdU16String());
+      emit dataChanged(index, index, {Qt::DisplayRole, Qt::EditRole});
+      return true;
+    case 2:
+      if (!value.canConvert<QString>())
+        return false;
+      book.SetType(value.toString().toStdU16String());
+      emit dataChanged(index, index, {Qt::DisplayRole, Qt::EditRole});
+      return true;
+    case 3:
+      if (!value.canConvert<QString>())
+        return false;
+      book.SetAuthor(value.toString().toStdU16String());
+      emit dataChanged(index, index, {Qt::DisplayRole, Qt::EditRole});
+      return true;
+    case 4:
+      if (!value.canConvert<QString>())
+        return false;
+      book.SetPress(value.toString().toStdU16String());
+      emit dataChanged(index, index, {Qt::DisplayRole, Qt::EditRole});
+      return true;
+    case 5:
+      if (!value.canConvert<int>())
+        return false;
+      book.SetStockCount(value.toInt());
+      emit dataChanged(index, index, {Qt::DisplayRole, Qt::EditRole});
+      return true;
+    default:
+      return false;
+    }
+  }
+  return false;
+}
+
+Qt::ItemFlags BookModel::flags(const QModelIndex &index) const {
+  if (!index.isValid())
+    return Qt::ItemIsEnabled;
+
+  return QAbstractTableModel::flags(index) | Qt::ItemIsEditable;
+}
+
+bool BookModel::insertRows(int row, int count, const QModelIndex &parent) {
+  beginInsertRows(parent, row, row + count - 1);
+  for (int i = 0; i < count; i++) {
+    record_->GetBooks().insert(record_->GetBooks().cbegin() + row, Book());
+  }
+  endInsertRows();
+  return true;
+}
+
+bool BookModel::removeRows(int row, int count, const QModelIndex &parent) {
+  beginRemoveRows(parent, row, row + count - 1);
+  record_->GetBooks().erase(record_->GetBooks().cbegin() + row,
+                            record_->GetBooks().cbegin() + row + count);
+  endRemoveRows();
+  return true;
+}