// Created on: 2017-06-16
// Created by: Natalia ERMOLAEVA
// Copyright (c) 2017 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement. 

#include <inspector/TreeModel_ContextMenu.hxx>
#include <inspector/TreeModel_ModelBase.hxx>
#include <inspector/TreeModel_Tools.hxx>

#include <inspector/ViewControl_Tools.hxx>

#include <Standard_WarningsDisable.hxx>
#include <QHeaderView>
#include <QMenu>
#include <QTreeView>
#include <Standard_WarningsRestore.hxx>

// =======================================================================
// function : Constructor
// purpose :
// =======================================================================
TreeModel_ContextMenu::TreeModel_ContextMenu (QTreeView* theTreeView)
  : QObject (theTreeView), myTreeView (theTreeView)
{
  myTreeView->header()->setContextMenuPolicy (Qt::CustomContextMenu);
#if QT_VERSION >= 0x050000
  myTreeView->header()->setSectionsClickable (true);
#endif
  myTreeView->header()->setHighlightSections (true);
  connect (myTreeView->header(), SIGNAL (customContextMenuRequested (const QPoint&)),
           this, SLOT (onTreeViewHeaderContextMenuRequested (const QPoint&)));
}

// =======================================================================
// function : onTreeViewHeaderContextMenuRequested
// purpose :
// =======================================================================
void TreeModel_ContextMenu::onTreeViewHeaderContextMenuRequested (const QPoint& thePosition)
{
  TreeModel_ModelBase* aModel = dynamic_cast<TreeModel_ModelBase*> (myTreeView->model());
  if (!aModel)
    return;

  QMenu* aMenu = new QMenu (myTreeView);
  int aNbSections = aModel->columnCount();
  for (int aColumnId = 0; aColumnId < aNbSections; aColumnId++)
  {
    QAction* anAction = ViewControl_Tools::CreateAction (aModel->GetHeaderItem (aColumnId).GetName(),
                                                       SLOT (onColumnVisibilityChanged()), myTreeView, this);
    anAction->setCheckable (true);
    anAction->setChecked (!myTreeView->isColumnHidden (aColumnId));
    anAction->setData (aColumnId);
    aMenu->addAction (anAction);
  }

  if (aMenu->actions().isEmpty())
    return;

  QPoint aPoint = myTreeView->mapToGlobal (thePosition);
  aMenu->exec (aPoint);
}

// =======================================================================
// function : onColumnVisibilityChanged
// purpose :
// =======================================================================
void TreeModel_ContextMenu::onColumnVisibilityChanged()
{
  QAction* aClickedAction = (QAction*)sender();

  myTreeView->setColumnHidden(aClickedAction->data().toInt(), !aClickedAction->isChecked());
}