pgLab/pglablib/TypeSelectionItemModel.h

#ifndef TYPESELECTIONITEMMODEL_H
#define TYPESELECTIONITEMMODEL_H

#include <QAbstractListModel>
#include <memory>
#include <vector>

class PgTypeContainer;

class TypeSelectionItemModel : public QAbstractListModel {
	Q_OBJECT
public:
	explicit TypeSelectionItemModel(QObject *parent = 0);

	void setTypeList(std::shared_ptr<const PgTypeContainer> types);

	int rowCount(const QModelIndex &parent = QModelIndex()) const override;
	int columnCount(const QModelIndex &parent = QModelIndex()) const override;
	QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;

private:
	std::vector<QString> m_types;
};

class TypeModel : public QAbstractListModel {
	Q_OBJECT
public:
	enum e_Columns : int {
		OidCol,
		NameCol, //
		colCount
	};

	explicit TypeModel(QObject *parent = 0);

	void setTypeList(std::shared_ptr<const PgTypeContainer> types);

	int rowCount(const QModelIndex &parent = QModelIndex()) const override;
	int columnCount(const QModelIndex &parent = QModelIndex()) const override;
	QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;

private:
	std::shared_ptr<const PgTypeContainer> m_types;
};


#endif // TYPESELECTIONITEMMODEL_H