QtCollider/widgets/QcTreeWidget.cpp

   1 /************************************************************************
   2 *
   3 * Copyright 2011-2012 Jakob Leben (jakob.leben@gmail.com)
   4 *
   5 * This file is part of SuperCollider Qt GUI.
   6 *
   7 * This program is free software: you can redistribute it and/or modify
   8 * it under the terms of the GNU General Public License as published by
   9 * the Free Software Foundation, either version 3 of the License, or
  10 * (at your option) any later version.
  11 *
  12 * This program is distributed in the hope that it will be useful,
  13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 * GNU General Public License for more details.
  16 *
  17 * You should have received a copy of the GNU General Public License
  18 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 *
  20 ************************************************************************/
  21
  22 #include "QcTreeWidget.h"
  23 #include "../QcWidgetFactory.h"
  24
  25 #include <PyrKernel.h>
  26
  27 #include <QKeyEvent>
  28
  29 class QcTreeWidgetFactory : public QcWidgetFactory<QcTreeWidget>
  30 {
  31   void initialize( QWidgetProxy *p, QcTreeWidget *w ) {
  32     p->setMouseEventWidget( w->viewport() );
  33   }
  34 };
  35
  36 QC_DECLARE_FACTORY( QcTreeWidget, QcTreeWidgetFactory );
  37
  38 QcTreeWidget::QcTreeWidget()
  39 {
  40   // Forward signals to argument-less versions connectable from SC.
  41   connect( this, SIGNAL( itemActivated(QTreeWidgetItem*, int) ),
  42            this, SIGNAL( action() ) );
  43   connect( this, SIGNAL( itemPressed(QTreeWidgetItem*, int) ),
  44            this, SIGNAL( itemPressedAction() ) );
  45   connect( this, SIGNAL( currentItemChanged(QTreeWidgetItem*, QTreeWidgetItem*) ),
  46            this, SIGNAL( currentItemChanged() ) );
  47
  48 }
  49
  50 VariantList QcTreeWidget::columns() const
  51 {
  52   VariantList varList;
  53   QTreeWidgetItem *header = headerItem();
  54   if( header ) {
  55     for( int i = 0; i < header->columnCount(); ++i ) {
  56       varList.data << header->text(i);
  57     }
  58   }
  59   return varList;
  60 }
  61
  62 void QcTreeWidget::setColumns( const VariantList & varList )
  63 {
  64   int count = varList.data.size();
  65   setColumnCount( count );
  66   QStringList labels;
  67   Q_FOREACH( QVariant var, varList.data )
  68     labels << var.toString();
  69   setHeaderLabels( labels );
  70 }
  71
  72 QcTreeWidget::ItemPtr QcTreeWidget::currentItem() const
  73 {
  74   QTreeWidgetItem *item = QTreeWidget::currentItem();
  75   return QcTreeWidget::Item::safePtr( item );
  76 }
  77
  78 void QcTreeWidget::setCurrentItem( const ItemPtr &item )
  79 {
  80   _emitAction = false;
  81   QTreeWidget::setCurrentItem( item );
  82   _emitAction = true;
  83 }
  84
  85 QcTreeWidget::ItemPtr QcTreeWidget::item( const QcTreeWidget::ItemPtr & parent, int index  )
  86 {
  87   QTreeWidgetItem *item =
  88     parent ? parent->child(index) : QTreeWidget::topLevelItem(index);
  89
  90   return QcTreeWidget::Item::safePtr( item );
  91 }
  92
  93 QcTreeWidget::ItemPtr QcTreeWidget::parentItem( const QcTreeWidget::ItemPtr & item )
  94 {
  95   if( !item ) return QcTreeWidget::ItemPtr();
  96   return QcTreeWidget::Item::safePtr( item->parent() );
  97 }
  98
  99 int QcTreeWidget::indexOfItem( const QcTreeWidget::ItemPtr & item )
 100 {
 101   if( !item ) return -1;
 102   QTreeWidgetItem *parent = item->parent();
 103   if( parent )
 104     return parent->indexOfChild( item );
 105   else
 106     return indexOfTopLevelItem( item );
 107 }
 108
 109 QcTreeWidget::ItemPtr QcTreeWidget::addItem (
 110   const QcTreeWidget::ItemPtr & parent, const VariantList & varList )
 111 {
 112   QStringList strings;
 113   for( int i = 0; i < varList.data.count(); ++i )
 114     strings << varList.data[i].toString();
 115
 116   Item *item = new Item( strings );
 117   if( !parent ) addTopLevelItem( item );
 118   else parent->addChild( item );
 119
 120   return item->safePtr();
 121 }
 122
 123 QcTreeWidget::ItemPtr QcTreeWidget::insertItem (
 124   const QcTreeWidget::ItemPtr & parent, int index, const VariantList & varList )
 125 {
 126   int itemCount = topLevelItemCount();
 127   if( index < 0 || index > itemCount ) return ItemPtr();
 128
 129   QStringList strings;
 130   for( int i = 0; i < varList.data.count(); ++i )
 131     strings << varList.data[i].toString();
 132
 133   Item *item = new Item( strings );
 134   if( !parent ) insertTopLevelItem( index, item );
 135   else parent->insertChild( index, item );
 136
 137   if( !item->treeWidget() ) {
 138     delete item;
 139     return ItemPtr();
 140   }
 141
 142   return item->safePtr();
 143 }
 144
 145 void QcTreeWidget::removeItem( const QcTreeWidget::ItemPtr & item )
 146 {
 147   delete item.ptr();
 148 }
 149
 150 VariantList QcTreeWidget::strings( const QcTreeWidget::ItemPtr & item )
 151 {
 152   VariantList varList;
 153   if( !item ) return varList;
 154   for( int i = 0; i < item->columnCount(); ++i ) {
 155     varList.data << item->text(i);
 156   }
 157   return varList;
 158 }
 159
 160 void QcTreeWidget::setText( const QcTreeWidget::ItemPtr &item, int column, const QString & text )
 161 {
 162   if( item ) item->setText( column, text );
 163 }
 164
 165 void QcTreeWidget::setColor( const QcTreeWidget::ItemPtr &item, int column, const QColor & color )
 166 {
 167   if( item ) item->setBackground( column, color );
 168 }
 169
 170 void QcTreeWidget::setTextColor( const QcTreeWidget::ItemPtr & item, int column, const QColor & color )
 171 {
 172   if( item ) item->setData( column, Qt::ForegroundRole, color );
 173 }
 174
 175 QWidget * QcTreeWidget::itemWidget( const QcTreeWidget::ItemPtr &item, int column )
 176 {
 177   return item ? QTreeWidget::itemWidget( item, column ) : 0;
 178 }
 179
 180 void QcTreeWidget::setItemWidget( const QcTreeWidget::ItemPtr &item, int column, QObjectProxy *o )
 181 {
 182   if( !item ) return;
 183
 184   QWidget *w = qobject_cast<QWidget*>(o->object());
 185   if( !w ) return;
 186
 187   QTreeWidget::setItemWidget( item, column, w );
 188 }
 189
 190 void QcTreeWidget::removeItemWidget( const QcTreeWidget::ItemPtr &item, int column )
 191 {
 192   if( item ) QTreeWidget::removeItemWidget( item, column );
 193 }
 194
 195 void QcTreeWidget::sort( int column, bool descending )
 196 {
 197   sortItems( column, descending ? Qt::DescendingOrder : Qt::AscendingOrder );
 198 }
 199
 200 void QcTreeWidget::keyPressEvent( QKeyEvent *e )
 201 {
 202   QTreeWidget::keyPressEvent( e );
 203
 204   switch (e->key())
 205   {
 206   case Qt::Key_Up:
 207   case Qt::Key_Down:
 208   case Qt::Key_Left:
 209   case Qt::Key_Right:
 210   case Qt::Key_PageUp:
 211   case Qt::Key_PageDown:
 212   case Qt::Key_Home:
 213   case Qt::Key_End:
 214     // Prevent propagating to parent when scroller reaches minimum or maximum:
 215     e->accept();
 216   default: break;
 217   }
 218 }
 219
 220 //////////////////////////////// Item //////////////////////////////////////
 221
 222 QcTreeWidget::ItemPtr QcTreeWidget::Item::safePtr( QTreeWidgetItem * item )
 223 {
 224   if( item && item->type() == Item::Type )
 225     return static_cast<Item*>(item)->safePtr();
 226   else
 227     return ItemPtr();
 228 }
 229
 230 void QcTreeWidget::Item::initialize (
 231   VMGlobals *g, PyrObject *obj, const QcTreeWidget::ItemPtr &ptr )
 232 {
 233   Q_ASSERT( isKindOf( obj, SC_CLASS(QTreeViewItem) ) );
 234   if( ptr.id() ) {
 235     // store the SafePtr
 236     QcTreeWidget::ItemPtr *newPtr = new QcTreeWidget::ItemPtr( ptr );
 237     SetPtr( obj->slots+0, newPtr );
 238     // store the id for equality comparison
 239     SetPtr( obj->slots+1, ptr.id() );
 240     // install finalizer
 241   }
 242   else {
 243     SetNil( obj->slots+0 );
 244     SetNil( obj->slots+1 );
 245   }
 246   InstallFinalizer( g, obj, 2, &QcTreeWidget::Item::finalize );
 247 }
 248
 249 int QcTreeWidget::Item::finalize( VMGlobals *g, PyrObject *obj )
 250 {
 251   qcDebugMsg(1,"finalizing QTreeViewItem!");
 252   if( IsPtr( obj->slots+0 ) ) {
 253     QcTreeWidget::ItemPtr *ptr = static_cast<QcTreeWidget::ItemPtr*>( slotRawPtr(obj->slots+0) );
 254     delete ptr;
 255   }
 256   return errNone;
 257 }
 258
 259 bool QcTreeWidget::Item::operator< (const QTreeWidgetItem &other) const
 260 {
 261   int column = treeWidget()->sortColumn();
 262   return text(column).toLower() < other.text(column).toLower();
 263 }