1 files changed, 33 insertions, 1 deletions
diff --git a/filestreewidget.cpp b/filestreewidget.cpp
index b017d7d..4ed9662 100644
--- a/filestreewidget.cpp
+++ b/filestreewidget.cpp
@@ -21,6 +21,7 @@
 #include <QSettings>
 #include <QEvent>
 #include <QSettings>
+#include <QHeaderView>
 
 #include "filestreewidget.h"
 #include "smglobals.h"
@@ -275,7 +276,16 @@ void FilesTreeWidget::fileSelectionChanged(const QItemSelection &selected, const
 
 FilesTreeView::FilesTreeView(QWidget *parent) : QTreeView(parent), mHoverWin(new HoverWindow), mHover(false){
 	setAttribute(Qt::WA_Hover);
-	readConfig();
+
+}
+
+void FilesTreeView::setModel(QAbstractItemModel *model){
+	QTreeView::setModel(model);
+	for(int i = 0; i < header()->count(); ++i){
+		header()->setSectionHidden(i, true);
+	}
+	readHeaderConfig();
+	connect(header(), SIGNAL(sectionMoved(int,int,int)), this, SLOT(sectionHasMoved(int,int,int)));
 }
 
 void FilesTreeView::readConfig(){
@@ -284,6 +294,27 @@ void FilesTreeView::readConfig(){
 	mHoverWin->setWindowOpacity(s.value("ui/hoveropacity", 10).toFloat() / 10.0);
 }
 
+void FilesTreeView::readHeaderConfig(){
+	QSettings s;
+	QByteArray headerPos = s.value("ui/headerpos").toByteArray();
+	if(!headerPos.isEmpty()){
+		header()->restoreState(headerPos);
+	}
+}
+
+void FilesTreeView::writeHeaderConfig(){
+	QSettings s;
+	s.setValue("ui/headerpos", header()->saveState());
+}
+
+void FilesTreeView::toggleHeader(QObject *action){
+	QAction *a = qobject_cast<QAction*>(action);
+	Q_ASSERT(a);
+	int logicalIndex = a->data().toInt();
+	QHeaderView *hv = header();
+	hv->setSectionHidden(logicalIndex, !a->isChecked());
+}
+
 void FilesTreeView::contextMenuEvent(QContextMenuEvent *event){
 	QMenu ctxMenu;
 	foreach(QAction *a, actions()){
@@ -350,6 +381,7 @@ bool FilesTreeView::exitHover(bool exitVal){
 
 FilesTreeSortModel::FilesTreeSortModel(QObject *parent) : QSortFilterProxyModel(parent) {}
 
+// left + right are from the sourceModel() !!!!
 bool FilesTreeSortModel::lessThan(const QModelIndex &left, const QModelIndex &right) const{
 	if(left.parent() == QModelIndex()){
 		return false;