silofilter.cpp

/*
 * silofilter.cpp
 *
 *  Created on: Dec 10, 2008
 *      Author: Vincent
 */

#include "silofilter.h"
#include "common.h"
using namespace Silo;

SiloFilter::SiloFilter(QObject *parent) :
	QSortFilterProxyModel(parent) {
	filterType = FILTER_INVALID;
	filterSize = 0;
}

void SiloFilter::setFilterDateTime(const QDateTime &dateTime,
		FilterType filterType) {
	this->filterType = filterType;
	filterDate = dateTime;
}

void SiloFilter::setFilterSize(qint64 filterSize, FilterType filterType) {
	this->filterType = filterType;
	this->filterSize = filterSize;
}

void SiloFilter::setFilterType(FilterType filterType) {
	this->filterType = filterType;
}

bool SiloFilter::filterAcceptsRow(int sourceRow,
		const QModelIndex &sourceParent) const {
	int keyColumn = filterKeyColumn();
	if (keyColumn == ID_DATE) {
		if (filterType == FILTER_INVALID) {
			return true;
		}
		QModelIndex index = sourceModel()->index(sourceRow, keyColumn,
				sourceParent);
		switch (filterType) {
		case FILTER_LESS:
			return index.data(Qt::DisplayRole).toDateTime() < filterDate;
		case FILTER_MORE:
			return index.data(Qt::DisplayRole).toDateTime() > filterDate;
		case FILTER_MATCH:
			return index.data(Qt::DisplayRole).toDateTime() == filterDate;
		}
	} else if (keyColumn == ID_SIZE) {
		if (filterType == FILTER_INVALID) {
			return true;
		}
		QModelIndex index = sourceModel()->index(sourceRow, keyColumn,
				sourceParent);
		switch (filterType) {
		case FILTER_LESS:
			return index.data(Qt::DisplayRole).toLongLong() < filterSize;
		case FILTER_MORE:
			return index.data(Qt::DisplayRole).toLongLong() > filterSize;
		case FILTER_MATCH:
			return index.data(Qt::DisplayRole).toLongLong() == filterSize;
		}
	}
	return QSortFilterProxyModel::filterAcceptsRow(sourceRow, sourceParent);
}