1 files changed, 11 insertions, 1 deletions

diff --git a/shemov.cpp b/shemov.cpp
index 7a380d5..2703a56 100644
--- a/shemov.cpp
+++ b/shemov.cpp
@@ -132,6 +132,7 @@ SheMov::SheMov(QWidget *parent, Qt::WindowFlags flags) : QMainWindow(parent, fla
 	readSettings();
 	mFSWidget->fileView()->setFocus(Qt::ActiveWindowFocusReason);
 	mATree->filesWidget()->filesTree()->header()->resizeSections(QHeaderView::ResizeToContents);
+    mPicWidget->readSettings();
     Helper::centerWidget(picViewer);
 }
 
@@ -489,10 +490,15 @@ void SheMov::createActions(){
     mShowTreeGroup->addAction(mShowAllFilesA);
 	connect(mShowAllFilesA, SIGNAL(toggled(bool)), mATree, SLOT(showAllFiles(bool)));
 
-	//Tree view headers
+    //headers
     mFilesTreeHeadersGroup = new QActionGroup(this);
     QSignalMapper *fileTreeHeaderMapper = createHeaderMapper(filesModel, mFilesTreeHeadersGroup);
     connect(fileTreeHeaderMapper, SIGNAL(mapped(QObject*)), mATree->filesWidget()->filesTree(), SLOT(toggleHeader(QObject*)));
+    mPicsTreeHeaderGroup = new QActionGroup(this);
+    SmTreeModel *picFilesModel = static_cast<SmTreeModel*>(SmGlobals::instance()->model("PicFiles"));
+    QSignalMapper *picsTreeHeaderMapper = createHeaderMapper(picFilesModel, mPicsTreeHeaderGroup);
+    connect(picsTreeHeaderMapper, SIGNAL(mapped(QObject*)), mPicWidget->picView(), SLOT(toggleHeader(QObject*)));
+    mPicWidget->picView()->setHeaderGroup(mPicsTreeHeaderGroup);
 
 	//Tree view hover
 	mHoverTreeGroup = new QActionGroup(this);
@@ -792,6 +798,10 @@ void SheMov::createMenus(){
     mPicViewMenu->addAction(mPVAddA);
     mPicViewMenu->addAction(mPVReplaceA);
     mPicViewMenu->addSeparator();
+    QMenu *headerMenu = new QMenu(tr("Show headers"), this);
+    headerMenu->addActions(mPicsTreeHeaderGroup->actions());
+    mPicViewMenu->addMenu(headerMenu);
+    mPicViewMenu->addSeparator();
     mPicViewMenu->addAction(mPVShuffleA);
     mPicViewMenu->addAction(mPWRefreshA);
     mViewPicMenuA = menuBar()->addMenu(mPicViewMenu);