86 lines
2.4 KiB
C++
86 lines
2.4 KiB
C++
#include "transactionfilterproxy.h"
|
|
#include "transactiontablemodel.h"
|
|
|
|
#include <QDateTime>
|
|
|
|
#include <cstdlib>
|
|
|
|
// Earliest date that can be represented (far in the past)
|
|
const QDateTime TransactionFilterProxy::MIN_DATE = QDateTime::fromTime_t(0);
|
|
// Last date that can be represented (far in the future)
|
|
const QDateTime TransactionFilterProxy::MAX_DATE = QDateTime::fromTime_t(0xFFFFFFFF);
|
|
|
|
TransactionFilterProxy::TransactionFilterProxy(QObject *parent) :
|
|
QSortFilterProxyModel(parent),
|
|
dateFrom(MIN_DATE),
|
|
dateTo(MAX_DATE),
|
|
addrPrefix(),
|
|
typeFilter(ALL_TYPES),
|
|
minAmount(0),
|
|
limitRows(-1)
|
|
{
|
|
}
|
|
|
|
bool TransactionFilterProxy::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const
|
|
{
|
|
QModelIndex index = sourceModel()->index(sourceRow, 0, sourceParent);
|
|
|
|
int type = index.data(TransactionTableModel::TypeRole).toInt();
|
|
QDateTime datetime = index.data(TransactionTableModel::DateRole).toDateTime();
|
|
QString address = index.data(TransactionTableModel::AddressRole).toString();
|
|
QString label = index.data(TransactionTableModel::LabelRole).toString();
|
|
qint64 amount = llabs(index.data(TransactionTableModel::AmountRole).toLongLong());
|
|
|
|
if(!(TYPE(type) & typeFilter))
|
|
return false;
|
|
if(datetime < dateFrom || datetime > dateTo)
|
|
return false;
|
|
if (!address.contains(addrPrefix, Qt::CaseInsensitive) && !label.contains(addrPrefix, Qt::CaseInsensitive))
|
|
return false;
|
|
if(amount < minAmount)
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
void TransactionFilterProxy::setDateRange(const QDateTime &from, const QDateTime &to)
|
|
{
|
|
this->dateFrom = from;
|
|
this->dateTo = to;
|
|
invalidateFilter();
|
|
}
|
|
|
|
void TransactionFilterProxy::setAddressPrefix(const QString &addrPrefix)
|
|
{
|
|
this->addrPrefix = addrPrefix;
|
|
invalidateFilter();
|
|
}
|
|
|
|
void TransactionFilterProxy::setTypeFilter(quint32 modes)
|
|
{
|
|
this->typeFilter = modes;
|
|
invalidateFilter();
|
|
}
|
|
|
|
void TransactionFilterProxy::setMinAmount(qint64 minimum)
|
|
{
|
|
this->minAmount = minimum;
|
|
invalidateFilter();
|
|
}
|
|
|
|
void TransactionFilterProxy::setLimit(int limit)
|
|
{
|
|
this->limitRows = limit;
|
|
}
|
|
|
|
int TransactionFilterProxy::rowCount(const QModelIndex &parent) const
|
|
{
|
|
if(limitRows != -1)
|
|
{
|
|
return std::min(QSortFilterProxyModel::rowCount(parent), limitRows);
|
|
}
|
|
else
|
|
{
|
|
return QSortFilterProxyModel::rowCount(parent);
|
|
}
|
|
}
|