#include <QStandardItem>
#include <QSortFilterProxyModel>
#include <QHBoxLayout>

#include "collectionwidget.h"

CollectionWidget::CollectionWidget(QWidget *parent) : QWidget(parent){
    mView = new QTreeView;
    mModel = new QStandardItemModel(this);
    mProxy = new QSortFilterProxyModel(this);
    mProxy->setSourceModel(mModel);
    mView->setModel(mProxy);
    mView->setSortingEnabled(true);
    mView->setAlternatingRowColors(true);
    QHBoxLayout *mainLayout = new QHBoxLayout;
    mainLayout->addWidget(mView);
    setLayout(mainLayout);
}

void CollectionWidget::enableSorting(int column, Qt::SortOrder sortOrder){
    view()->setSortingEnabled(true);
    view()->sortByColumn(column, sortOrder);
}