source: src/UIElements/Views/Qt4/QtShapeList.cpp@ e9f307

/*
 * Project: MoleCuilder
 * Description: creates and alters molecular systems
 * Copyright (C)  2010-2012 University of Bonn. All rights reserved.
 *
 *
 *   This file is part of MoleCuilder.
 *
 *    MoleCuilder is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation, either version 2 of the License, or
 *    (at your option) any later version.
 *
 *    MoleCuilder is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with MoleCuilder.  If not, see <http://www.gnu.org/licenses/>.
 */

/*
 * QtShapeList.cpp
 *
 *  Created on: Sep 14, 2012
 *      Author: ankele
 */

// include config.h
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include "Views/Qt4/QtShapeList.hpp"

#include <iostream>

//#include "CodePatterns/MemDebug.hpp"

#include "CodePatterns/Observer/Notification.hpp"

#include "Shapes/ShapeFactory.hpp"
#include "Shapes/ShapeRegistry.hpp"

#include <QtGui/QAbstractItemView>
#include "Actions/SelectionAction/Shapes/ShapeByNameAction.hpp"
#include "Actions/SelectionAction/Shapes/NotShapeByNameAction.hpp"

using namespace std;

const int QtShapeList::COLUMNCOUNT = COLUMNTYPES_MAX;
const char *QtShapeList::COLUMNNAMES[QtShapeList::COLUMNCOUNT]={"Name","Type","Center","Radius"};

QtShapeList::QtShapeList(QWidget * _parent) :
    QTreeWidget (_parent),
    Observer("QtShapeList"),
    shaperegistry_enabled(false)
{
  setColumnCount(COLUMNCOUNT);
  setSelectionMode(QAbstractItemView::MultiSelection);

  QStringList header;
        for(int i=0; i<COLUMNCOUNT;++i)
          header << COLUMNNAMES[i];
        setHeaderLabels(header);

        selecting = false;

        refill(NULL);

        ShapeRegistry::getInstance().signOn(this);
  ShapeRegistry::getInstance().signOn(this, ShapeRegistry::ShapeInserted);
  ShapeRegistry::getInstance().signOn(this, ShapeRegistry::ShapeRemoved);
  ShapeRegistry::getInstance().signOn(this, ShapeRegistry::SelectionChanged);
  shaperegistry_enabled = true;

  connect(this,SIGNAL(itemSelectionChanged()),this,SLOT(rowSelected()));
}

QtShapeList::~QtShapeList()
{
  if (shaperegistry_enabled) {
    ShapeRegistry::getInstance().signOff(this);
    ShapeRegistry::getInstance().signOff(this, ShapeRegistry::ShapeInserted);
    ShapeRegistry::getInstance().signOff(this, ShapeRegistry::ShapeRemoved);
    ShapeRegistry::getInstance().signOff(this, ShapeRegistry::SelectionChanged);
  }
}

void QtShapeList::update(Observable *publisher) {}

void QtShapeList::recieveNotification(Observable *publisher, Notification_ptr notification)
{
  if (static_cast<ShapeRegistry*>(publisher) == ShapeRegistry::getPointer()) {
    switch (notification->getChannelNo()) {
      case ShapeRegistry::ShapeInserted:
      {
        refill(NULL);
        break;
      }
      case ShapeRegistry::ShapeRemoved:
      {
        refill(ShapeRegistry::getInstance().lastChanged());
        break;
      }
      case ShapeRegistry::SelectionChanged:
      {
        refill(NULL);
        break;
      }
      default:
        ASSERT(0, "QtShapeList::recieveNotification() - we cannot get here.");
        break;
    }
  }
}

void QtShapeList::refill(::Shape *ignore)
{
  boost::recursive_mutex::scoped_lock lock(refill_mutex);

  clear();
  shapeSelection.clear();

  ShapeRegistry &reg = ShapeRegistry::getInstance();
  ShapeFactory &factory = ShapeFactory::getInstance();

  ShapeRegistry::const_iterator iter;
  for (iter = reg.getBeginIter(); iter != reg.getEndIter(); iter ++){
    ::Shape *s = iter->second;
    if (s == ignore)
      continue;

    QTreeWidgetItem *treeItem = new QTreeWidgetItem(this);
    treeItem->setText(NAME, QString(s->getName().c_str()));
    treeItem->setText(TYPE, QString(factory.getShapeName(s->getType()).c_str()));
    treeItem->setText(CENTER, QString(toString(s->getCenter()).c_str()));
    treeItem->setText(RADIUS, QString::number(s->getRadius()));
    treeItem->setSelected(reg.isSelected(s));

    shapeSelection.push_back(reg.isSelected(s));
  }
}

void QtShapeList::paintEvent(QPaintEvent * event)
{
  boost::recursive_mutex::scoped_lock lock(refill_mutex);
  /*if (dirty)
    refill();*/
  QTreeWidget::paintEvent(event);
}

void QtShapeList::subjectKilled(Observable *publisher)
{
  // as a new instance should always already be present ... just sign on
  if (static_cast<ShapeRegistry *>(publisher) == ShapeRegistry::getPointer()) {
    shaperegistry_enabled = false;
  }
}

void QtShapeList::rowSelected()
{
  boost::recursive_mutex::scoped_lock lock(refill_mutex);

  if (selecting)
    return;
  //std::cout << "rowSelected\n";
  selecting = true;

  for (int i=0;i<topLevelItemCount();i++){
    QTreeWidgetItem *item = topLevelItem(i);
    bool newSelection = item->isSelected();
    if (newSelection != shapeSelection[i]){
      shapeSelection[i] = newSelection;

      std::string shapeName = item->text(NAME).toStdString();
      if (newSelection)
        MoleCuilder::SelectionShapeByName(shapeName);
      else
        MoleCuilder::SelectionNotShapeByName(shapeName);
    }
  }
  selecting = false;
}