Reorganization of pgLab project

pglab/catalog/models/IndexModel.cpp

@@ -0,0 +1,142 @@
+#include "IndexModel.h"
+#include "catalog/PgDatabaseCatalog.h"
+#include "catalog/PgIndexContainer.h"
+#include "Pgsql_oids.h"
+#include "ScopeGuard.h"
+#include "CustomDataRole.h"
+
+void IndexModel::setData(std::shared_ptr<const PgDatabaseCatalog> cat, const std::optional<PgClass> &table)
+{
+	if (cat != m_catalog) {
+		m_catalog = cat;
+		refreshConnection = connect(m_catalog.get(), &PgDatabaseCatalog::refreshed, this, &IndexModel::refresh);
+	}
+	m_table = table;
+	refresh();
+}
+
+void IndexModel::refresh()
+{
+	beginResetModel();
+	SCOPE_EXIT { endResetModel(); };
+
+	if (m_table)
+		m_indexes = m_catalog->indexes()->getIndexesForTable(m_table->oid());
+	else
+		m_indexes.clear();
+}
+
+int IndexModel::rowCount(const QModelIndex &/*parent*/) const
+{
+	return (int)m_indexes.size();
+}
+
+int IndexModel::columnCount(const QModelIndex &/*parent*/) const
+{
+	return colCount;
+}
+
+Oid IndexModel::getType(int column) const
+{
+    switch (column) {
+    case SizeCol:
+        return Pgsql::int8_oid;
+    case ExplicitIndexCol:
+        return Pgsql::bool_oid;
+    default:
+        return Pgsql::varchar_oid;
+    }
+}
+
+QVariant IndexModel::headerData(int section, Qt::Orientation orientation, int role) const
+{
+	QVariant v;
+	if (orientation == Qt::Horizontal) {
+		if (role == Qt::DisplayRole) {
+			QString c;
+			switch (section) {
+			case TypeCol:
+				c = tr("Type");
+				break;
+			case NameCol:
+				c = tr("Name");
+				break;
+			case AmCol:
+				c = tr("AM");
+				break;
+			case ColumnsCol:
+				c = tr("On");
+				break;
+			case ConditionCol:
+				c = tr("Condition");
+				break;
+            case SizeCol:
+                c = tr("Size");
+                break;
+            case ExplicitIndexCol:
+                c = tr("Explicit");
+                break;
+//			case DefinitionCol:
+//				c = tr("Definition");
+//				break;
+			}
+			v = c;
+		}
+	}
+	return v;
+
+}
+
+QVariant IndexModel::getData(const QModelIndex &index) const
+{
+	QVariant v;
+	int rij = index.row();
+	const auto &dat = m_indexes[rij];
+	switch (index.column()) {
+	case TypeCol:
+		if (dat.isprimary) v = ":/icons/constraints/primarykey.png";
+		else if (dat.isunique) v = ":/icons/constraints/unique.png";
+		else v = ":/icons/constraints/index.png";
+		break;
+
+	case NameCol:
+		v = dat.objectName(); // getIndexDisplayString(*m_catalog, dat.indexrelid);
+		break;
+
+	case AmCol:
+		v = dat.getAm();
+		break;
+
+	case ColumnsCol:
+		break;
+
+	case ConditionCol:
+		break;
+
+    case SizeCol:
+        v = dat.sizeBytes;
+        break;
+    case ExplicitIndexCol:
+        v = !dat.isSupportingIndex();
+        break;
+	}
+	return v;
+}
+
+QVariant IndexModel::data(const QModelIndex &index, int role) const
+{
+	QVariant v;
+	if (role == Qt::DisplayRole)
+		v = getData(index);
+	else if (role == CustomDataTypeRole)
+		v = getType(index.column());
+    else if (role == CustomDataMeaningRole) {
+        switch (index.column()) {
+        case SizeCol:
+            return static_cast<int>(DataMeaningBytes);
+        default:
+            return static_cast<int>(DataMeaningNormal);
+        }
+    }
+    return v;
+}