/*
 * 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/>.
 */

/*
 * QtElementList.cpp
 *
 *  Created on: Mar 6, 2012
 *      Author: ankele
 */

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

#include "Views/Qt4/QtElementList.hpp"

#include <iostream>

#include "CodePatterns/MemDebug.hpp"

#include "Atom/atom.hpp"
#include "Atom/AtomObserver.hpp"

#include "Element/element.hpp"
#include "Element/periodentafel.hpp"
#include "Descriptors/AtomTypeDescriptor.hpp"
#include <QAbstractItemView>
#include "Actions/SelectionAction/Atoms/AtomByElementAction.hpp"
#include "Actions/SelectionAction/Atoms/NotAtomByElementAction.hpp"

using namespace std;

const int QtElementList::COLUMNCOUNT = COLUMNTYPES_MAX;
const char *QtElementList::COLUMNNAMES[QtElementList::COLUMNCOUNT]={"Number","Name","Symbol","Mass","Occurrence"};

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

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

	dirty = true;

	AtomObserver::getInstance().signOn(this, atom::ElementChanged);

  connect(this,SIGNAL(itemSelectionChanged()),this,SLOT(rowSelected()));
  connect(this,SIGNAL(needsRefill(const atomId_t)),this,SLOT(refill(const atomId_t)), Qt::QueuedConnection);
//  connect(this,SIGNAL(changed()),this,SLOT(update()));

//  emit needsRefill();
}

QtElementList::~QtElementList()
{
  AtomObserver::getInstance().signOff(this, atom::ElementChanged);
}

void QtElementList::update(Observable *publisher)
{
  ASSERT(0, "QtElementList::update() - is not enlisted to any general update.");
}


void QtElementList::recieveNotification(Observable *publisher, Notification_ptr notification)
{
  if (notification->getChannelNo() == atom::ElementChanged) {
//    updateElement(static_cast<atom &>(*publisher));
    dirty = true;

    emit needsRefill(dynamic_cast<atom *>(publisher)->getId());
  } else
    ASSERT(0, "QtElementList::recieveNotification() - we are not enlisted to any other instance's channel.");
}

void QtElementList::updateElement(const atom &_atom)
{
  // this function has not been tested and is not used because it is
  // incompatible with the current refill scheme
  AtomElementMap_t::iterator iter = AtomElementMap.find(_atom.getId());
  if (iter == AtomElementMap.end()) {
    AtomElementMap.insert( std::make_pair(_atom.getId(), -1));
    iter = AtomElementMap.find(_atom.getId());
  }
  const atomicNumber_t oldelement = iter->second;
  const atomicNumber_t newelement = _atom.getElementNo();
  iter->second = newelement;
  QTreeWidgetItem *oldtreeItem = NULL;
  if (oldelement != (atomicNumber_t)-1)
    oldtreeItem = topLevelItem(oldelement-1);
  QTreeWidgetItem *newtreeItem = topLevelItem(newelement-1);
  ASSERT( (newtreeItem != NULL),
      "QtElementList::updateElement() - missing new element item.");
  if (oldelement != (atomicNumber_t)-1) {
    const int count_old = oldtreeItem->text(OCCURRENCE).toInt();
    setOccurrence(*oldtreeItem, count_old-1);
  }
  const int count_new = newtreeItem->text(OCCURRENCE).toInt();
  setOccurrence(*newtreeItem, count_new+1);
}

void QtElementList::setOccurrence(QTreeWidgetItem &_item, const int count)
{
  ASSERT (count >= 0,
      "QtElementList::setOccurrence() - count for an elment < 0.");

//    // something must be very wrong, rather refill
//    dirty = true;
//    refill();
//  } else {
    if (count > 0) {
      _item.setText(OCCURRENCE, QString::number(count));
    } else {
      _item.setText(OCCURRENCE, "none");
      _item.setDisabled(true);
    }
//  }
}

void QtElementList::refill(const atomId_t _atomid)
{
  refill_mutex.lock();

//  periodentafel *&periode = World::getInstance().getPeriode();
//
//  elementSelection.clear();
//
//  int i;
//
//  clear();
//  periodentafel::const_iterator iter;
//  for(iter = periode->begin(),i=0;
//      iter != periode->end();
//      ++i,++iter) {
//    const element *e = iter->second;
//    int count = 0;
//    count = const_cast<const World &>(World::getInstance()).
//        getAllAtoms(AtomByType(e)).size();
//
//    QTreeWidgetItem *treeItem = new QTreeWidgetItem(this);
//    treeItem->setText(NUMBER, QString::number(e->getAtomicNumber()));
//    treeItem->setText(NAME, QString(e->getName().c_str()));
//    treeItem->setText(SYMBOL, QString(e->getSymbol().c_str()));
//    treeItem->setText(MASS, QString::number(e->getMass()));
//    setOccurrence(*treeItem, count);
//    elementSelection.push_back(treeItem->isSelected());
//  }
//  dirty = false
  const atom * const walker = const_cast<const World &>(World::getInstance()).
      getAtom(AtomById(_atomid));
  if (walker != NULL)
    updateElement(*walker);
  else {
    AtomElementMap_t::iterator iter = AtomElementMap.find(_atomid);
    QTreeWidgetItem *oldtreeItem = topLevelItem(iter->second-1);
    const int count_old = oldtreeItem->text(OCCURRENCE).toInt();
    setOccurrence(*oldtreeItem, count_old-1);

    AtomElementMap.erase(iter);
  }

  refill_mutex.unlock();
}

void QtElementList::paintEvent(QPaintEvent * event)
{
//  if (dirty)
//    refill();
  QTreeWidget::paintEvent(event);
}

void QtElementList::subjectKilled(Observable *publisher) {
}


void QtElementList::rowSelected()
{
  //std::cout << "rowSelected\n";
  periodentafel *&periode = World::getInstance().getPeriode();

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

      int atomicNumber = item->text(NUMBER).toInt();
      const element *e = periode->FindElement(atomicNumber);
      if (newSelection)
        MoleCuilder::SelectionAtomByElement(e);
      else
        MoleCuilder::SelectionNotAtomByElement(e);
    }
  }
}