transactiontablemodel.cpp

   1 #include "transactiontablemodel.h"
   2
   3 #include <QLocale>
   4
   5 const QString TransactionTableModel::Sent = "s";
   6 const QString TransactionTableModel::Received = "r";
   7 const QString TransactionTableModel::Generated = "g";
   8
   9 /* Credit and Debit columns are right-aligned as they contain numbers */
  10 static int column_alignments[] = {
  11         Qt::AlignLeft|Qt::AlignVCenter,
  12         Qt::AlignLeft|Qt::AlignVCenter,
  13         Qt::AlignLeft|Qt::AlignVCenter,
  14         Qt::AlignRight|Qt::AlignVCenter,
  15         Qt::AlignRight|Qt::AlignVCenter,
  16         Qt::AlignLeft|Qt::AlignVCenter
  17     };
  18
  19 TransactionTableModel::TransactionTableModel(QObject *parent):
  20         QAbstractTableModel(parent)
  21 {
  22     columns << tr("Status") << tr("Date") << tr("Description") << tr("Debit") << tr("Credit") << tr("Type");
  23 }
  24
  25 int TransactionTableModel::rowCount(const QModelIndex &parent) const
  26 {
  27     Q_UNUSED(parent);
  28     return 5;
  29 }
  30
  31 int TransactionTableModel::columnCount(const QModelIndex &parent) const
  32 {
  33     Q_UNUSED(parent);
  34     return columns.length();
  35 }
  36
  37 QVariant TransactionTableModel::data(const QModelIndex &index, int role) const
  38 {
  39     if(!index.isValid())
  40         return QVariant();
  41
  42     if(role == Qt::DisplayRole)
  43     {
  44         /* index.row(), index.column() */
  45         /* Return QString */
  46         return QLocale::system().toString(index.row());
  47     } else if (role == Qt::TextAlignmentRole)
  48     {
  49         return column_alignments[index.column()];
  50     } else if (role == Qt::UserRole)
  51     {
  52         /* user role: transaction type for filtering
  53            "s" (sent)
  54            "r" (received)
  55            "g" (generated)
  56         */
  57         switch(index.row() % 3)
  58         {
  59         case 0: return QString("s");
  60         case 1: return QString("r");
  61         case 2: return QString("o");
  62         }
  63     }
  64     return QVariant();
  65 }
  66
  67 QVariant TransactionTableModel::headerData(int section, Qt::Orientation orientation, int role) const
  68 {
  69     if(role == Qt::DisplayRole)
  70     {
  71         if(orientation == Qt::Horizontal)
  72         {
  73             return columns[section];
  74         }
  75     } else if (role == Qt::TextAlignmentRole)
  76     {
  77         return column_alignments[section];
  78     }
  79     return QVariant();
  80 }
  81
  82 Qt::ItemFlags TransactionTableModel::flags(const QModelIndex &index) const
  83 {
  84     if (!index.isValid())
  85         return Qt::ItemIsEnabled;
  86
  87     return QAbstractTableModel::flags(index);
  88 }