FluentUI/src/FluTableSortProxyModel.cpp

#include "FluTableSortProxyModel.h"

#include <QJSValueList>

FluTableSortProxyModel::FluTableSortProxyModel(QSortFilterProxyModel *parent)
    : QSortFilterProxyModel {parent}
{
    _model = nullptr;
    connect(this,&FluTableSortProxyModel::modelChanged,this,[=]{
        setSourceModel(this->model());
    });
}

bool FluTableSortProxyModel::filterAcceptsRow(int source_row, const QModelIndex &source_parent) const{
    QJSValue filter = _filter;
    if(filter.isUndefined()){
        return true;
    }
    QJSValueList data;
    data<<source_row;
    return filter.call(data).toBool();
}

bool FluTableSortProxyModel::filterAcceptsColumn(int source_column, const QModelIndex &source_parent) const{
    return true;
}

bool FluTableSortProxyModel::lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const{
    QJSValue comparator = _comparator;
    if(comparator.isUndefined()){
        return true;
    }
    QJSValueList data;
    data<<source_left.row();
    data<<source_right.row();
    bool flag = comparator.call(data).toBool();
    if(sortOrder()==Qt::AscendingOrder){
        return !flag;
    }else{
        return flag;
    }
}

void FluTableSortProxyModel::setComparator(QJSValue comparator){
    int column = 0;
    if(comparator.isUndefined()){
        column = -1;
    }
    this->_comparator = comparator;
    if(sortOrder()==Qt::AscendingOrder){
        sort(column,Qt::DescendingOrder);
    }else{
        sort(column,Qt::AscendingOrder);
    }
}

void FluTableSortProxyModel::setFilter(QJSValue filter){
    this->_filter = filter;
    invalidateFilter();
}

QVariant FluTableSortProxyModel::getRow(int rowIndex){
    QVariant result;
    QMetaObject::invokeMethod(_model, "getRow",Q_RETURN_ARG(QVariant, result),Q_ARG(int, mapToSource(index(rowIndex,0)).row()));
    return result;
}

void FluTableSortProxyModel::setRow(int rowIndex,QVariant val){
    QMetaObject::invokeMethod(_model, "setRow",Q_ARG(int, mapToSource(index(rowIndex,0)).row()),Q_ARG(QVariant,val));
}

void FluTableSortProxyModel::removeRow(int rowIndex,int rows){
    QMetaObject::invokeMethod(_model, "removeRow",Q_ARG(int, mapToSource(index(rowIndex,0)).row()),Q_ARG(int,rows));
}