//=========================================================================
//  Copyright (c) Kitware, Inc.
//  All rights reserved.
//  See LICENSE.txt for details.
//
//  This software is distributed WITHOUT ANY WARRANTY; without even
//  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
//  PURPOSE.  See the above copyright notice for more information.
//=========================================================================
#include "smtk/extension/qt/testing/cxx/ModelBrowser.h"

#include "smtk/extension/qt/qtEntityItemModel.h"
#include "smtk/extension/qt/qtEntityItemDelegate.h"

#include "smtk/extension/qt/testing/cxx/ui_ModelBrowser.h"

#include "smtk/model/Group.h"
#include "smtk/model/Model.h"
#include "smtk/model/EntityPhrase.h"
#include "smtk/model/EntityListPhrase.h"
#include "smtk/model/Manager.h"

#include <QtGui/QPushButton>
#include <QtGui/QTreeView>

namespace Ui { class qtAttributeAssociation; }

using namespace smtk::model;

class ModelBrowser::Internals : public Ui::ModelBrowser
{
public:
  smtk::extension::QEntityItemModel* qmodel;
  smtk::extension::QEntityItemDelegate* qdelegate;
};

ModelBrowser::ModelBrowser(QWidget* p) :
  QWidget(p)
{
  this->m_p = new ModelBrowser::Internals;
  this->m_p->setupUi(this);
  QObject::connect(
    this->m_p->addGroupButton, SIGNAL(clicked()),
    this, SLOT(addGroup())
  );
  QObject::connect(
    this->m_p->removeFromGroupButton, SIGNAL(clicked()),
    this, SLOT(removeFromGroup())
  );
  QObject::connect(
    this->m_p->addToGroupButton, SIGNAL(clicked()),
    this, SLOT(addToGroup())
  );
}

ModelBrowser::~ModelBrowser()
{
  delete this->m_p;
}

QTreeView* ModelBrowser::tree() const
{
  return this->m_p->modelTree;
}

void ModelBrowser::setup(
  smtk::model::ManagerPtr manager,
  smtk::extension::QEntityItemModel* qmodel,
  smtk::extension::QEntityItemDelegate* qdelegate,
  smtk::model::DescriptivePhrasePtr root)
{
  this->m_manager = manager;
  qmodel->setRoot(root);
  this->m_p->modelTree->setModel(qmodel); // Must come after qmodel->setRoot()!
  this->m_p->modelTree->setItemDelegate(qdelegate);
  this->m_p->qmodel = qmodel;
  this->m_p->qdelegate = qdelegate;
  QObject::connect(
    this->m_p->modelTree->selectionModel(), SIGNAL(currentRowChanged(const QModelIndex&, const QModelIndex&)),
    this, SLOT(updateButtonStates(const QModelIndex&, const QModelIndex&))
  );
}

void ModelBrowser::addGroup()
{
  Group newGroup = this->m_manager->addGroup(0, "New Group");
  Models models;
  smtk::model::EntityRef::EntityRefsFromUUIDs(
    models,
    this->m_manager,
    this->m_manager->entitiesMatchingFlags(smtk::model::MODEL_ENTITY));
  if (!models.empty())
    {
    models.begin()->addGroup(newGroup);
    std::cout << "Added " << newGroup.name() << " to " << models.begin()->name() << "\n";
    }
}

// Add the entity under the entityref to the first group (ordered by UUID)
void ModelBrowser::addToGroup()
{
  QModelIndex qidx = this->m_p->modelTree->currentIndex();
  Group group;
  EntityRef item;
  Groups groups;
  EntityRef::EntityRefsFromUUIDs(
    groups, this->m_manager, this->m_manager->entitiesMatchingFlags(smtk::model::GROUP_ENTITY));
  if (groups.empty())
    return;

  group = *groups.begin();

  // Only keep the phrase alive as long as we must.
    {
    DescriptivePhrasePtr phrase = this->m_p->qmodel->getItem(qidx);
    if (!phrase)
      return;

    item = phrase->relatedEntity();
    }

  if (!item.isValid())
    return;

  // Now that our shared pointer keeping "phrase" alive is gone,
  // remove the entity from the group.
  if (group.isValid())
    {
    std::cout << "Adding " << item.name() << " to " << group.name() << "\n";
    group.addEntity(item);
    }
}

void ModelBrowser::removeFromGroup()
{
  QModelIndex qidx = this->m_p->modelTree->currentIndex();
  Group group;
  if ((group = this->groupParentOfIndex(qidx)).isValid())
    {
    EntityRef relEnt;
      {
      DescriptivePhrasePtr phrase = this->m_p->qmodel->getItem(qidx);
      if (phrase)
        relEnt = phrase->relatedEntity();
      }
    // Now that our shared pointer keeping "phrase" alive is gone,
    // remove the entity from the group.
    if (relEnt.isValid())
      {
      // Deselect the current row, select another.
      // This helps un-confuse Qt's treeview.
      QModelIndex sidx = qidx.parent();
      int nsiblings = sidx.model()->rowCount(sidx);
      if (nsiblings)
        {
        if (qidx.row() < nsiblings - 1)
          sidx = qidx.sibling(qidx.row() + 1, 0);
        else if (qidx.row() > 0)
          sidx = qidx.sibling(qidx.row() - 1, 0);
        }
      if (sidx.model() != qidx.model())
        {
        std::cout << "Erp! Models differ: q " << qidx.model() << " s " << sidx.model() << "\n";
        }
      if (sidx.parent() != qidx.parent())
        {
        std::cout
          << "Erp! Model parents"
          << "   s\"" << sidx.model()->data(sidx.parent(), smtk::extension::QEntityItemModel::TitleTextRole).toString().toStdString()
          << "\" q\"" << sidx.model()->data(qidx.parent(), smtk::extension::QEntityItemModel::TitleTextRole).toString().toStdString()
          << "\" differ\n";
        }
      this->m_p->modelTree->selectionModel()->select(
        sidx, QItemSelectionModel::Columns | QItemSelectionModel::SelectCurrent);
      // Removing from the group emits a signal that
      // m_p->qmodel listens for, causing m_p->modelTree redraw.
      group.removeEntity(relEnt);
      }
    }
}

void ModelBrowser::updateButtonStates(const QModelIndex& curr, const QModelIndex&)
{
  this->m_p->removeFromGroupButton->setEnabled(
    groupParentOfIndex(curr).isValid());
}

/**\brief Does \a qidx refer to an entity that is displayed as the child of a group?
  *
  * Note that a group (EntityPhrase with a EntityRef whose isGroup() is true)
  * may contain an EntityListPhrase, each entry of which is in the group.
  * We must test for this 1 level of indirection as well as for direct
  * children.
  */
smtk::model::Group ModelBrowser::groupParentOfIndex(const QModelIndex& qidx)
{
  smtk::model::Group group;
  DescriptivePhrasePtr phrase = this->m_p->qmodel->getItem(qidx);
  if (phrase)
    {
    EntityPhrasePtr ephrase = smtk::dynamic_pointer_cast<EntityPhrase>(phrase);
    if (ephrase && ephrase->relatedEntity().isValid())
      {
      phrase = ephrase->parent();
      if (phrase)
        {
        ephrase = smtk::dynamic_pointer_cast<EntityPhrase>(phrase);
        if (ephrase && (group = ephrase->relatedEntity().as<smtk::model::Group>()).isValid())
          return group; // direct child of a Group's summary phrase.
        EntityListPhrasePtr lphrase = smtk::dynamic_pointer_cast<EntityListPhrase>(phrase);
        if (lphrase)
          {
          ephrase = smtk::dynamic_pointer_cast<EntityPhrase>(lphrase->parent());
          if (ephrase && (group = ephrase->relatedEntity().as<smtk::model::Group>()).isValid())
            return group; // member of a list inside a Group's summary.
          }
        }
      }
    }
  return group;
}
