- class QtTreeWidgetItem;

- virtual void show(QtTreeWidgetItem* item) = 0;

-#include "Swift/Controllers/UIEvents/RemoveItemRosterAction.h"

-#include "Swift/QtUI/Roster/QtTreeWidgetItem.h"

-void QtRosterContextMenu::show(QtTreeWidgetItem* item) {

- if (!item->isContact()) {

- item_->performUserAction(boost::shared_ptr<UserRosterAction>(new RemoveItemRosterAction()));

- void show(QtTreeWidgetItem* item);

- QtTreeWidgetItem* item_;

-#include "Roster/QtTreeWidgetFactory.h"

-QtChatWindow::QtChatWindow(const QString &contact, QtTreeWidgetFactory *treeWidgetFactory) : QtTabbable(), contact_(contact), previousMessageWasSelf_(false), previousMessageWasSystem_(false) {

- treeWidget_ = dynamic_cast<QtTreeWidget*>(treeWidgetFactory->createTreeWidget());

-TreeWidget* QtChatWindow::getTreeWidget() {

- return treeWidget_;

- QtChatWindow(const QString &contact, QtTreeWidgetFactory* treeWidgetFactory);

- TreeWidget *getTreeWidget();

-#include "Roster/QtTreeWidgetFactory.h"

-QtChatWindowFactory::QtChatWindowFactory(QtTreeWidgetFactory *treeWidgetFactory, QSplitter* splitter, QtSettingsProvider* settings, QtChatTabs* tabs) : treeWidgetFactory_(treeWidgetFactory) {

-ChatWindow* QtChatWindowFactory::createChatWindow(const JID &contact) {

- QtChatWindow *chatWindow = new QtChatWindow(P2QSTRING(contact.toString()), treeWidgetFactory_);

- //chatWindow->show();

-#ifndef SWIFT_QtChatWindowFactory_H

-#define SWIFT_QtChatWindowFactory_H

- class QtTreeWidgetFactory;

- QtChatWindowFactory(QtTreeWidgetFactory *treeWidgetFactory, QSplitter* splitter, QtSettingsProvider* settings, QtChatTabs* tabs);

- ChatWindow* createChatWindow(const JID &contact);

- QtTreeWidgetFactory* treeWidgetFactory_;

-#endif

-#include "Roster/QtTreeWidgetFactory.h"

-QtMainWindow::QtMainWindow(UIEventStream* uiEventStream, QtTreeWidgetFactory *treeWidgetFactory) : QWidget() {

- treeWidget_ = dynamic_cast<QtTreeWidget*>(treeWidgetFactory->createTreeWidget());

-TreeWidget* QtMainWindow::getTreeWidget() {

- return treeWidget_;

-}

-

- QtMainWindow(UIEventStream* eventStream, QtTreeWidgetFactory *treeWidgetFactory);

- TreeWidget* getTreeWidget();

-#include "Roster/QtTreeWidgetFactory.h"

-QtMainWindowFactory::QtMainWindowFactory(QtTreeWidgetFactory *treeWidgetFactory) : treeWidgetFactory_(treeWidgetFactory) {

- lastWindow_ = new QtMainWindow(eventStream, treeWidgetFactory_);

- QtMainWindowFactory(QtTreeWidgetFactory *treeWidgetFactory);

- QtTreeWidgetFactory *treeWidgetFactory_;

-#include "Roster/QtTreeWidgetFactory.h"

- treeWidgetFactory_ = new QtTreeWidgetFactory();

- chatWindowFactory_ = new QtChatWindowFactory(treeWidgetFactory_, splitter_, settings_, tabs_);

- rosterWindowFactory_ = new QtMainWindowFactory(treeWidgetFactory_);

- mainController_ = new MainController(chatWindowFactory_, rosterWindowFactory_, loginWindowFactory_, treeWidgetFactory_, eventWindowFactory_, settings_, application_, systemTray_, soundPlayer_, xmlConsoleWidgetFactory_, chatListWindowFactory_, options.count("latency-debug") > 0);

- delete treeWidgetFactory_;

- class QtTreeWidgetFactory;

- QtTreeWidgetFactory *treeWidgetFactory_;

-#include "Swiften/Roster/OpenChatRosterAction.h"

-QtTreeWidget::QtTreeWidget(QWidget* parent) : QTreeView(parent) {

- treeRoot_ = new QtTreeWidgetItem(NULL);

- model_ = new RosterModel();

- model_->setRoot(treeRoot_);

- delegate_ = new RosterDelegate();

- connect(model_, SIGNAL(itemExpanded(const QModelIndex&, bool)), this, SLOT(handleModelItemExpanded(const QModelIndex&, bool)));

- connect(model_, SIGNAL(dataChanged(const QModelIndex&, const QModelIndex&)), this, SLOT(handleDataChanged(const QModelIndex&, const QModelIndex&)));

- connect(this, SIGNAL(expanded(const QModelIndex&)), this, SLOT(handleExpanded(const QModelIndex&)));

- connect(this, SIGNAL(collapsed(const QModelIndex&)), this, SLOT(handleCollapsed(const QModelIndex&)));

- QtTreeWidgetItem* qtItem = static_cast<QtTreeWidgetItem*>(index.internalPointer());

- if (qtItem) {

- qtItem->performUserAction(boost::shared_ptr<UserRosterAction>(new OpenChatRosterAction()));

- QtTreeWidgetItem* qtItem = index.isValid() ? static_cast<QtTreeWidgetItem*>(index.internalPointer()) : NULL;

- if (qtItem) {

- contextMenu_->show(qtItem);

- }

-}

-

-void QtTreeWidget::handleExpanded(const QModelIndex& index) {

- QtTreeWidgetItem* qtItem = static_cast<QtTreeWidgetItem*>(index.internalPointer());

- if (qtItem) {

- qtItem->setExpanded(true);

- }

-}

-

-void QtTreeWidget::handleCollapsed(const QModelIndex& index) {

- QtTreeWidgetItem* qtItem = static_cast<QtTreeWidgetItem*>(index.internalPointer());

- if (qtItem) {

- qtItem->setExpanded(false);

- }

-}

-

-void QtTreeWidget::handleModelItemExpanded(const QModelIndex& index, bool shouldExpand) {

- if (this->isExpanded(index) == shouldExpand) {

- return;

- }

- //setExpanded(index, shouldExpand);

- if (shouldExpand) {

- expand(index);

- emit expanded(index);

- } else {

- collapse(index);

- emit collapsed(index);

-void QtTreeWidget::handleDataChanged(const QModelIndex& topLeft, const QModelIndex& bottomRight) {

- Q_UNUSED(bottomRight);

- //in our model, this is only thrown with topLeft == bottomRight

- if (!topLeft.isValid()) {

- return;

- }

- QtTreeWidgetItem* qtItem = static_cast<QtTreeWidgetItem*>(topLeft.internalPointer());

- if (qtItem) {

- setExpanded(topLeft, qtItem->isExpanded());

- //qDebug() << "Item changed, passing expanded state to view: " << qtItem->isExpanded() << " giving an expanded state of " << isExpanded(topLeft);

- }

-

-}

-#include "Swiften/Roster/TreeWidgetFactory.h"

-#include "Swiften/Roster/TreeWidget.h"

-#include "Swiften/Roster/TreeWidgetItem.h"

-#include "Swift/QtUI/Roster/QtTreeWidgetItem.h"

-class QtTreeWidget : public QTreeView, public TreeWidget {

- QtTreeWidget(QWidget* parent = 0);

- void handleModelItemExpanded(const QModelIndex&, bool expanded);

- void handleExpanded(const QModelIndex&);

- void handleCollapsed(const QModelIndex&);

- void handleDataChanged(const QModelIndex& topLeft, const QModelIndex& bottomRight);

-/*

- * Copyright (c) 2010 Kevin Smith

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-/*

- * Copyright (c) 2010 Kevin Smith

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#ifndef SWIFT_QtTreeWidgetFactory_H

-#define SWIFT_QtTreeWidgetFactory_H

-

-#include "Swiften/Roster/TreeWidgetFactory.h"

-#include "Swiften/Roster/TreeWidget.h"

-#include "Swiften/Roster/TreeWidgetItem.h"

-#include "Swift/QtUI/Roster/QtTreeWidgetItem.h"

-#include "Swift/QtUI/Roster/QtTreeWidget.h"

-

-namespace Swift {

-

-class QtTreeWidgetFactory : public TreeWidgetFactory {

- public:

- QtTreeWidgetFactory() {

- }

-

- TreeWidget* createTreeWidget() {

- return new QtTreeWidget();

- }

-

- TreeWidgetItem* createTreeWidgetItem(TreeWidgetItem* item) {

- QtTreeWidgetItem* qtItem = dynamic_cast<QtTreeWidgetItem*>(item);

- assert(qtItem);

- QtTreeWidgetItem* newItem = new QtTreeWidgetItem(qtItem);

- qtItem->addChild(newItem);

- return newItem;

- }

-

- TreeWidgetItem* createTreeWidgetItem(TreeWidget* item) {

- QtTreeWidget* treeItem = dynamic_cast<QtTreeWidget*>(item);

- assert(treeItem);

- QtTreeWidgetItem* qtItem = treeItem->getRoot();

- QtTreeWidgetItem* newItem = new QtTreeWidgetItem(qtItem);

- //qtItem->setItemWidget(newItem, 0, newItem->getCollapsedRosterWidget());

- qtItem->addChild(newItem);

- newItem->setExpanded(true);

- return newItem;

- }

-};

-

-}

-#endif

-

-#include "QtTreeWidgetItem.h"

-RosterDelegate::RosterDelegate() {

- QtTreeWidgetItem* item = static_cast<QtTreeWidgetItem*>(index.internalPointer());

- if (!item || !item->isContact()) {

-QSize RosterDelegate::contactSizeHint(const QStyleOptionViewItem& option, const QModelIndex& index ) const {

- Q_UNUSED(option);

- Q_UNUSED(index);

- QtTreeWidgetItem* item = static_cast<QtTreeWidgetItem*>(index.internalPointer());

- if (item && !item->isContact()) {

- QtTreeWidgetItem* item = index.isValid() ? static_cast<QtTreeWidgetItem*>(index.internalPointer()) : NULL;

- if (item) {

- groupDelegate_->paint(painter, option, index.data(Qt::DisplayRole).toString(), item->rowCount(), item->isExpanded());

-

-

-

- class QtTreeWidgetItem;

- RosterDelegate();

-RosterModel::RosterModel() {

- delete tree_;

-void RosterModel::setRoot(QtTreeWidgetItem* root) {

- tree_ = root;

- connect(tree_, SIGNAL(changed(QtTreeWidgetItem*)), this, SLOT(handleItemChanged(QtTreeWidgetItem*)));

-

-void RosterModel::handleItemChanged(QtTreeWidgetItem* item) {

- if (!item->isShown()) {

- return;

- }

- Q_ASSERT(modelIndex.isValid());

- emit itemExpanded(modelIndex, item->isExpanded());

- emit dataChanged(modelIndex, modelIndex);

- emit dataChanged(parent(modelIndex), parent(modelIndex));

- emit layoutChanged();

-int RosterModel::columnCount(const QModelIndex& parent) const {

- Q_UNUSED(parent);

- QtTreeWidgetItem* item = index.isValid() ? static_cast<QtTreeWidgetItem*>(index.internalPointer()) : NULL;

- return item ? item->data(role) : QVariant();

-QModelIndex RosterModel::index(int row, int column, const QModelIndex& parent) const {

- QtTreeWidgetItem* parentItem = parent.isValid() ? static_cast<QtTreeWidgetItem*>(parent.internalPointer()) : tree_;

- Q_ASSERT(parentItem);

-

- return row < parentItem->rowCount() ? createIndex(row, column, parentItem->getItem(row)) : QModelIndex();

-QModelIndex RosterModel::index(QtTreeWidgetItem* item) const {

- return createIndex(item->row(), 0, item);

-QModelIndex RosterModel::parent(const QModelIndex& index) const {

- if (!index.isValid()) {

- return QModelIndex();

- QtTreeWidgetItem* item = static_cast<QtTreeWidgetItem*>(index.internalPointer());

- Q_ASSERT(item);

- QtTreeWidgetItem* parentItem = item->getParentItem();

- /* parentItem_ == NULL can happen during destruction.*/

- return parentItem == tree_ || parentItem == NULL ? QModelIndex() : createIndex(parentItem->row(), 0, parentItem);

- QtTreeWidgetItem* item = parent.isValid() ? static_cast<QtTreeWidgetItem*>(parent.internalPointer()) : tree_;

-

- return item->rowCount();

-#include "Swift/QtUI/Roster/QtTreeWidgetItem.h"

-class RosterModel : public QAbstractItemModel {

-Q_OBJECT

-public:

- RosterModel();

- ~RosterModel();

- void setRoot(QtTreeWidgetItem* tree);

- int columnCount(const QModelIndex& parent = QModelIndex()) const;

- QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const;

- QModelIndex index(int row, int column, const QModelIndex & parent = QModelIndex()) const;

- QModelIndex index(QtTreeWidgetItem* item) const;

- QModelIndex parent(const QModelIndex& index) const;

- int rowCount(const QModelIndex& parent = QModelIndex()) const;

-signals:

- void itemExpanded(const QModelIndex& item, bool expanded);

-private slots:

- void handleItemChanged(QtTreeWidgetItem* item);

-private:

- QtTreeWidgetItem* tree_;

-};

- "Roster/QtTreeWidgetItem.cpp",