#include "MainWindow.h"
#include "ui_MainWindow.h"

#include <QHBoxLayout>
#include <QTreeView>
#include <QSortFilterProxyModel>

#include "MyTestTree.h"
#include "MyHeaderView.h"

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    ui->setupUi(this);

    QHBoxLayout *mainLayout = new QHBoxLayout();
    ui->centralWidget->setLayout(mainLayout);

    MyTestTree *tree = new MyTestTree();
    QSortFilterProxyModel *filter = new QSortFilterProxyModel(); // Needed for sorting
    filter->setSourceModel(tree);
    MyHeaderView *header = new MyHeaderView(); // Needed so we can see the private variables
    QTreeView *treeView = new QTreeView();
    treeView->setHeader(header);
    treeView->setModel(filter);
    treeView->setSortingEnabled(true);

    // Hiding 10 columns means persistentHiddenSections will grow by 10 every time
    for(int i=5; i < 15; ++i) {
        treeView->setColumnHidden(i, true);
    }
    mainLayout->addWidget(treeView);
}

MainWindow::~MainWindow()
{
    delete ui;
}