src/qt/transactionfilterproxy.cpp

   1 #include "transactionfilterproxy.h"
   2 #include "transactiontablemodel.h"
   3
   4 #include <QDateTime>
   5 #include <QDebug>
   6
   7 // Earliest date that can be represented (far in the past)
   8 const QDateTime TransactionFilterProxy::MIN_DATE = QDateTime::fromTime_t(0);
   9 // Last date that can be represented (far in the future)
  10 const QDateTime TransactionFilterProxy::MAX_DATE = QDateTime::fromTime_t(0xFFFFFFFF);
  11
  12 TransactionFilterProxy::TransactionFilterProxy(QObject *parent) :
  13     QSortFilterProxyModel(parent),
  14     dateFrom(MIN_DATE),
  15     dateTo(MAX_DATE),
  16     addrPrefix(),
  17     typeFilter(ALL_TYPES),
  18     minAmount(0),
  19     limitRows(-1)
  20 {
  21 }
  22
  23 bool TransactionFilterProxy::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const
  24 {
  25     QModelIndex index = sourceModel()->index(sourceRow, 0, sourceParent);
  26
  27     int type = index.data(TransactionTableModel::TypeRole).toInt();
  28     QDateTime datetime = index.data(TransactionTableModel::DateRole).toDateTime();
  29     QString address = index.data(TransactionTableModel::AddressRole).toString();
  30     QString label = index.data(TransactionTableModel::LabelRole).toString();
  31     qint64 amount = index.data(TransactionTableModel::AbsoluteAmountRole).toLongLong();
  32
  33     if(!(TYPE(type) & typeFilter))
  34         return false;
  35     if(datetime < dateFrom || datetime > dateTo)
  36         return false;
  37     if(!address.startsWith(addrPrefix) && !label.startsWith(addrPrefix))
  38         return false;
  39     if(amount < minAmount)
  40         return false;
  41
  42     return true;
  43 }
  44
  45 void TransactionFilterProxy::setDateRange(const QDateTime &from, const QDateTime &to)
  46 {
  47     this->dateFrom = from;
  48     this->dateTo = to;
  49     invalidateFilter();
  50 }
  51
  52 void TransactionFilterProxy::setAddressPrefix(const QString &addrPrefix)
  53 {
  54     this->addrPrefix = addrPrefix;
  55     invalidateFilter();
  56 }
  57
  58 void TransactionFilterProxy::setTypeFilter(quint32 modes)
  59 {
  60     this->typeFilter = modes;
  61     invalidateFilter();
  62 }
  63
  64 void TransactionFilterProxy::setMinAmount(qint64 minimum)
  65 {
  66     this->minAmount = minimum;
  67     invalidateFilter();
  68 }
  69
  70 void TransactionFilterProxy::setLimit(int limit)
  71 {
  72     this->limitRows = limit;
  73 }
  74
  75 int TransactionFilterProxy::rowCount(const QModelIndex &parent) const
  76 {
  77     if(limitRows != -1)
  78     {
  79         return std::min(QSortFilterProxyModel::rowCount(parent), limitRows);
  80     }
  81     else
  82     {
  83         return QSortFilterProxyModel::rowCount(parent);
  84     }
  85 }