RecursiveSortFilterProxyModel

RecursiveSortFilterProxyModel.java

import com.trolltech.qt.core.QModelIndex;
import com.trolltech.qt.gui.*;

public class RecursiveSortFilterProxyModel extends QSortFilterProxyModel {
	protected boolean filterAcceptsRow(int row, QModelIndex parent) {
		if(super.filterAcceptsRow(row, parent)) {
			return true;
		}
		
		QStandardItemModel model = (QStandardItemModel)sourceModel();
		QStandardItem item;
		
		if(parent == null) {
			item = model.item(row);
		} else {
			item = model.itemFromIndex(parent).child(row);
		}
		
		if(item.hasChildren()) {
			QModelIndex index = item.index();
			int rowCount = item.rowCount();
			for(int i = 0; i < rowCount; i++) {
				if(filterAcceptsRow(i, index)) {
					return true;
				}
			}
		}
		
		return false;
	}
}