src/qt/transactionfilterproxy.cpp

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