QueryResultModel.h

#include "AbstractQueryResultModel.h"
 
namespace kDB
{
  namespace Repository
  {
    template<typename _T_>
    class QueryResultModel : public AbstractQueryResultModel
    {
    public:
      explicit QueryResultModel(QObject* parent = 0);
      explicit QueryResultModel(const _T_& _result, QObject* parent = 0);
      int rowCount(const QModelIndex &parent = QModelIndex()) const override;
      QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
      QHash<int, QByteArray> roleNames() const override;
      _T_ result() const { return m_result; }
      void setResult(const _T_& _t);
    protected:
      QStringList roleNames_() const override;
    private:
      _T_ m_result;
    };
    template<typename _T_>
    QueryResultModel<_T_>::QueryResultModel(QObject* parent) : AbstractQueryResultModel(parent)
    {
    }
    template<typename _T_>
    QueryResultModel<_T_>::QueryResultModel(const _T_& _result, QObject* parent) : AbstractQueryResultModel(parent), m_result(_result)
    {
    }
    template<typename _T_>
    QHash<int, QByteArray> QueryResultModel<_T_>::roleNames() const
    {
      QHash<int, QByteArray> roles;
      if(m_result.isValid())
      {
        for(int i = 0; i < m_result.fields(); ++i)
        {
          roles[Qt::UserRole + i] = m_result.fieldName(i).toUtf8();
        }
      }
      return roles;
    }
    template<typename _T_>
    QStringList QueryResultModel<_T_>::roleNames_() const
    {
      QStringList r;
      if(m_result.isValid())
      {
        for(int i = 0; i < m_result.fields(); ++i)
        {
          r << m_result.fieldName(i);
        }
      }
      return r;
    }
    template<typename _T_>
    int QueryResultModel<_T_>::rowCount(const QModelIndex &parent) const
    {
      Q_UNUSED(parent);
      return m_result.isValid() ? m_result.tuples() : 0;
    }
    
    template<typename _T_>
    QVariant QueryResultModel<_T_>::data(const QModelIndex &index, int role) const
    {
      if(index.isValid() and role >= Qt::UserRole and role < Qt::UserRole + m_result.fields())
      {
        return QVariant::fromValue(m_result.value(index.row(), role - Qt::UserRole));
      } else {
        return QVariant();
      }
    }
    template<typename _T_>
    void QueryResultModel<_T_>::setResult(const _T_& _t)
    {
      beginResetModel();
      m_result = _t;
      endResetModel();
    }
  }
}