modules/gui/qt4/components/playlist/selector.cpp

   1 /*****************************************************************************
   2  * selector.cpp : Playlist source selector
   3  ****************************************************************************
   4  * Copyright (C) 2006-2009 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Clément Stenac <zorglub@videolan.org>
   8  *          Jean-Baptiste Kempf
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23  *****************************************************************************/
  24
  25 #ifdef HAVE_CONFIG_H
  26 # include "config.h"
  27 #endif
  28
  29 #include "qt4.hpp"
  30 #include "components/playlist/selector.hpp"
  31 #include "playlist_model.hpp"                /* plMimeData */
  32 #include "input_manager.hpp"                 /* MainInputManager, for podcast */
  33
  34 #include <QInputDialog>
  35 #include <QMessageBox>
  36 #include <QMimeData>
  37 #include <QDragMoveEvent>
  38 #include <QTreeWidgetItem>
  39 #include <QHBoxLayout>
  40 #include <QPainter>
  41 #include <QPalette>
  42
  43 #include <vlc_playlist.h>
  44 #include <vlc_services_discovery.h>
  45
  46 void SelectorActionButton::paintEvent( QPaintEvent *event )
  47 {
  48     QPainter p( this );
  49     QColor color = palette().color( QPalette::HighlightedText );
  50     color.setAlpha( 80 );
  51     if( underMouse() )
  52         p.fillRect( rect(), color );
  53     p.setPen( color );
  54     int frame = style()->pixelMetric( QStyle::PM_DefaultFrameWidth, 0, this );
  55     p.drawLine( rect().topLeft() + QPoint( 0, frame ),
  56                 rect().bottomLeft() - QPoint( 0, frame ) );
  57     QFramelessButton::paintEvent( event );
  58 }
  59
  60 PLSelItem::PLSelItem ( QTreeWidgetItem *i, const QString& text )
  61     : qitem(i), lblAction( NULL)
  62 {
  63     layout = new QHBoxLayout( this );
  64     layout->setContentsMargins(0,0,0,0);
  65     layout->addSpacing( 3 );
  66
  67     lbl = new QElidingLabel( text );
  68     layout->addWidget(lbl, 1);
  69
  70     int height = qMax( 22, fontMetrics().height() + 8 );
  71     setMinimumHeight( height );
  72 }
  73
  74 void PLSelItem::addAction( ItemAction act, const QString& tooltip )
  75 {
  76     if( lblAction ) return; //might change later
  77
  78     QIcon icon;
  79
  80     switch( act )
  81     {
  82     case ADD_ACTION:
  83         icon = QIcon( ":/buttons/playlist/playlist_add" ); break;
  84     case RM_ACTION:
  85         icon = QIcon( ":/buttons/playlist/playlist_remove" ); break;
  86     }
  87
  88     lblAction = new SelectorActionButton();
  89     lblAction->setIcon( icon );
  90     lblAction->setMinimumWidth( lblAction->sizeHint().width() + 6 );
  91
  92     if( !tooltip.isEmpty() ) lblAction->setToolTip( tooltip );
  93
  94     layout->addWidget( lblAction, 0 );
  95     lblAction->hide();
  96
  97     CONNECT( lblAction, clicked(), this, triggerAction() );
  98 }
  99
 100
 101 PLSelector::PLSelector( QWidget *p, intf_thread_t *_p_intf )
 102            : QTreeWidget( p ), p_intf(_p_intf)
 103 {
 104     /* Properties */
 105     setFrameStyle( QFrame::NoFrame );
 106     setAttribute( Qt::WA_MacShowFocusRect, false );
 107     viewport()->setAutoFillBackground( false );
 108     setIconSize( QSize( 24,24 ) );
 109     setIndentation( 12 );
 110     setHeaderHidden( true );
 111     setRootIsDecorated( true );
 112     setAlternatingRowColors( false );
 113
 114     /* drops */
 115     viewport()->setAcceptDrops(true);
 116     setDropIndicatorShown(true);
 117     invisibleRootItem()->setFlags( invisibleRootItem()->flags() & ~Qt::ItemIsDropEnabled );
 118
 119     /* Podcasts */
 120     podcastsParent = NULL;
 121     podcastsParentId = -1;
 122
 123     /* Podcast connects */
 124     CONNECT( THEMIM, playlistItemAppended( int, int ),
 125              this, plItemAdded( int, int ) );
 126     CONNECT( THEMIM, playlistItemRemoved( int ),
 127              this, plItemRemoved( int ) );
 128     DCONNECT( THEMIM->getIM(), metaChanged( input_item_t *),
 129               this, inputItemUpdate( input_item_t * ) );
 130
 131     createItems();
 132
 133     /***
 134      * We need to react to both clicks and activation (enter-key) here.
 135      * We use curItem to avoid rebuilding twice.
 136      * See QStyle::SH_ItemView_ActivateItemOnSingleClick
 137      ***/
 138     curItem = NULL;
 139     CONNECT( this, itemActivated( QTreeWidgetItem *, int ),
 140              this, setSource( QTreeWidgetItem *) );
 141     CONNECT( this, itemClicked( QTreeWidgetItem *, int ),
 142              this, setSource( QTreeWidgetItem *) );
 143
 144     /* select the first item */
 145 //  view->setCurrentIndex( model->index( 0, 0, QModelIndex() ) );
 146
 147 #ifdef Q_WS_MAC
 148     setAutoFillBackground( true );
 149     QPalette palette;
 150     palette.setColor( QPalette::Window, QColor(209,215,226) );
 151     setPalette( palette );
 152 #endif
 153     setMinimumHeight( 120 );
 154 }
 155
 156 PLSelector::~PLSelector()
 157 {
 158     if( podcastsParent )
 159     {
 160         int c = podcastsParent->childCount();
 161         for( int i = 0; i < c; i++ )
 162         {
 163             QTreeWidgetItem *item = podcastsParent->child(i);
 164             input_item_t *p_input = item->data( 0, IN_ITEM_ROLE ).value<input_item_t*>();
 165             vlc_gc_decref( p_input );
 166         }
 167     }
 168 }
 169
 170 PLSelItem * putSDData( PLSelItem* item, const char* name, const char* longname )
 171 {
 172     item->treeItem()->setData( 0, NAME_ROLE, qfu( name ) );
 173     item->treeItem()->setData( 0, LONGNAME_ROLE, qfu( longname ) );
 174     return item;
 175 }
 176
 177 PLSelItem * putPLData( PLSelItem* item, playlist_item_t* plItem )
 178 {
 179     item->treeItem()->setData( 0, PL_ITEM_ROLE, QVariant::fromValue( plItem ) );
 180 /*    item->setData( 0, PL_ITEM_ID_ROLE, plItem->i_id );
 181     item->setData( 0, IN_ITEM_ROLE, QVariant::fromValue( (void*) plItem->p_input ) ); );*/
 182     return item;
 183 }
 184
 185 void PLSelector::createItems()
 186 {
 187     /* PL */
 188     PLSelItem *pl = putPLData( addItem( PL_ITEM_TYPE, N_("Playlist"), true ),
 189                               THEPL->p_playing );
 190     pl->treeItem()->setData( 0, SPECIAL_ROLE, QVariant( IS_PL ) );
 191
 192     /* ML */
 193     PLSelItem *ml = putPLData( addItem( PL_ITEM_TYPE, N_("Media Library"), true ),
 194                               THEPL->p_media_library );
 195     ml->treeItem()->setData( 0, SPECIAL_ROLE, QVariant( IS_ML ) );
 196
 197 #ifdef MEDIA_LIBRARY
 198     /* SQL ML */
 199     addItem( SQL_ML_TYPE, "SQL Media Library" )->treeItem();
 200 #endif
 201
 202     /* SD nodes */
 203     QTreeWidgetItem *mycomp = addItem( CATEGORY_TYPE, N_("My Computer") )->treeItem();
 204     QTreeWidgetItem *devices = addItem( CATEGORY_TYPE, N_("Devices") )->treeItem();
 205     QTreeWidgetItem *lan = addItem( CATEGORY_TYPE, N_("Local Network") )->treeItem();
 206     QTreeWidgetItem *internet = addItem( CATEGORY_TYPE, N_("Internet") )->treeItem();
 207
 208     /* SD subnodes */
 209     char **ppsz_longnames;
 210     int *p_categories;
 211     char **ppsz_names = vlc_sd_GetNames( THEPL, &ppsz_longnames, &p_categories );
 212     if( !ppsz_names )
 213         return;
 214
 215     char **ppsz_name = ppsz_names, **ppsz_longname = ppsz_longnames;
 216     int *p_category = p_categories;
 217     for( ; *ppsz_name; ppsz_name++, ppsz_longname++, p_category++ )
 218     {
 219         //msg_Dbg( p_intf, "Adding a SD item: %s", *ppsz_longname );
 220
 221         PLSelItem *selItem;
 222         switch( *p_category )
 223         {
 224         case SD_CAT_INTERNET:
 225             {
 226             selItem = addItem( SD_TYPE, *ppsz_longname, false, internet );
 227             if( !strncmp( *ppsz_name, "podcast", 7 ) )
 228             {
 229                 selItem->treeItem()->setData( 0, SPECIAL_ROLE, QVariant( IS_PODCAST ) );
 230                 selItem->addAction( ADD_ACTION, qtr( "Subscribe to a podcast" ) );
 231                 CONNECT( selItem, action( PLSelItem* ), this, podcastAdd( PLSelItem* ) );
 232                 podcastsParent = selItem->treeItem();
 233             }
 234             }
 235             break;
 236         case SD_CAT_DEVICES:
 237             selItem = addItem( SD_TYPE, *ppsz_longname, false, devices );
 238             break;
 239         case SD_CAT_LAN:
 240             selItem = addItem( SD_TYPE, *ppsz_longname, false, lan );
 241             break;
 242         case SD_CAT_MYCOMPUTER:
 243             selItem = addItem( SD_TYPE, *ppsz_longname, false, mycomp );
 244             break;
 245         default:
 246             selItem = addItem( SD_TYPE, *ppsz_longname );
 247         }
 248
 249         putSDData( selItem, *ppsz_name, *ppsz_longname );
 250         free( *ppsz_name );
 251         free( *ppsz_longname );
 252     }
 253     free( ppsz_names );
 254     free( ppsz_longnames );
 255     free( p_categories );
 256
 257     if( mycomp->childCount() == 0 ) delete mycomp;
 258     if( devices->childCount() == 0 ) delete devices;
 259     if( lan->childCount() == 0 ) delete lan;
 260     if( internet->childCount() == 0 ) delete internet;
 261 }
 262
 263 void PLSelector::setSource( QTreeWidgetItem *item )
 264 {
 265     if( !item || item == curItem )
 266         return;
 267
 268     curItem = item;
 269
 270     bool b_ok;
 271     int i_type = item->data( 0, TYPE_ROLE ).toInt( &b_ok );
 272     if( !b_ok || i_type == CATEGORY_TYPE )
 273         return;
 274
 275     bool sd_loaded;
 276     if( i_type == SD_TYPE )
 277     {
 278         QString qs = item->data( 0, NAME_ROLE ).toString();
 279         sd_loaded = playlist_IsServicesDiscoveryLoaded( THEPL, qtu( qs ) );
 280         if( !sd_loaded )
 281             playlist_ServicesDiscoveryAdd( THEPL, qtu( qs ) );
 282     }
 283 #ifdef MEDIA_LIBRARY
 284     else if( i_type == SQL_ML_TYPE )
 285     {
 286         emit categoryActivated( NULL, true );
 287         return;
 288     }
 289 #endif
 290
 291     /* */
 292     playlist_Lock( THEPL );
 293     playlist_item_t *pl_item = NULL;
 294
 295     /* Special case for podcast */
 296     // FIXME: simplify
 297     if( i_type == SD_TYPE )
 298     {
 299         /* Find the right item for the SD */
 300         pl_item = playlist_ChildSearchName( THEPL->p_root,
 301                       qtu( item->data(0, LONGNAME_ROLE ).toString() ) );
 302
 303         /* Podcasts */
 304         if( item->data( 0, SPECIAL_ROLE ).toInt() == IS_PODCAST )
 305         {
 306             if( pl_item && !sd_loaded )
 307             {
 308                 podcastsParentId = pl_item->i_id;
 309                 for( int i=0; i < pl_item->i_children; i++ )
 310                     addPodcastItem( pl_item->pp_children[i] );
 311             }
 312             pl_item = NULL; //to prevent activating it
 313         }
 314     }
 315     else
 316         pl_item = item->data( 0, PL_ITEM_ROLE ).value<playlist_item_t*>();
 317
 318     playlist_Unlock( THEPL );
 319
 320     /* */
 321     if( pl_item )
 322         emit categoryActivated( pl_item, false );
 323 }
 324
 325 PLSelItem * PLSelector::addItem (
 326     SelectorItemType type, const char* str, bool drop,
 327     QTreeWidgetItem* parentItem )
 328 {
 329   QTreeWidgetItem *item = parentItem ?
 330       new QTreeWidgetItem( parentItem ) : new QTreeWidgetItem( this );
 331
 332   PLSelItem *selItem = new PLSelItem( item, qtr( str ) );
 333   setItemWidget( item, 0, selItem );
 334   item->setData( 0, TYPE_ROLE, (int)type );
 335   if( !drop ) item->setFlags( item->flags() & ~Qt::ItemIsDropEnabled );
 336
 337   return selItem;
 338 }
 339
 340 PLSelItem *PLSelector::addPodcastItem( playlist_item_t *p_item )
 341 {
 342     vlc_gc_incref( p_item->p_input );
 343
 344     char *psz_name = input_item_GetName( p_item->p_input );
 345     PLSelItem *item = addItem( PL_ITEM_TYPE,  psz_name, false, podcastsParent );
 346     free( psz_name );
 347
 348     item->addAction( RM_ACTION, qtr( "Remove this podcast subscription" ) );
 349     item->treeItem()->setData( 0, PL_ITEM_ROLE, QVariant::fromValue( p_item ) );
 350     item->treeItem()->setData( 0, PL_ITEM_ID_ROLE, QVariant(p_item->i_id) );
 351     item->treeItem()->setData( 0, IN_ITEM_ROLE, QVariant::fromValue( p_item->p_input ) );
 352     CONNECT( item, action( PLSelItem* ), this, podcastRemove( PLSelItem* ) );
 353     return item;
 354 }
 355
 356 QStringList PLSelector::mimeTypes() const
 357 {
 358     QStringList types;
 359     types << "vlc/qt-input-items";
 360     return types;
 361 }
 362
 363 bool PLSelector::dropMimeData ( QTreeWidgetItem * parent, int,
 364     const QMimeData * data, Qt::DropAction )
 365 {
 366     if( !parent ) return false;
 367
 368     QVariant type = parent->data( 0, TYPE_ROLE );
 369     if( type == QVariant() ) return false;
 370
 371     int i_truth = parent->data( 0, SPECIAL_ROLE ).toInt();
 372     if( i_truth != IS_PL && i_truth != IS_ML ) return false;
 373
 374     bool to_pl = ( i_truth == IS_PL );
 375
 376     const PlMimeData *plMimeData = qobject_cast<const PlMimeData*>( data );
 377     if( !plMimeData ) return false;
 378
 379     QList<input_item_t*> inputItems = plMimeData->inputItems();
 380
 381     playlist_Lock( THEPL );
 382
 383     foreach( input_item_t *p_input, inputItems )
 384     {
 385         playlist_item_t *p_item = playlist_ItemGetByInput( THEPL, p_input );
 386         if( !p_item ) continue;
 387
 388         playlist_NodeAddCopy( THEPL, p_item,
 389                               to_pl ? THEPL->p_playing : THEPL->p_media_library,
 390                               PLAYLIST_END );
 391     }
 392
 393     playlist_Unlock( THEPL );
 394
 395     return true;
 396 }
 397
 398 void PLSelector::dragMoveEvent ( QDragMoveEvent * event )
 399 {
 400     event->setDropAction( Qt::CopyAction );
 401     QAbstractItemView::dragMoveEvent( event );
 402 }
 403
 404 void PLSelector::plItemAdded( int item, int parent )
 405 {
 406     if( parent != podcastsParentId ) return;
 407
 408     playlist_Lock( THEPL );
 409
 410     playlist_item_t *p_item = playlist_ItemGetById( THEPL, item );
 411     if( !p_item ) {
 412         playlist_Unlock( THEPL );
 413         return;
 414     }
 415
 416     int c = podcastsParent->childCount();
 417     for( int i = 0; i < c; i++ )
 418     {
 419         QTreeWidgetItem *podItem = podcastsParent->child(i);
 420         if( podItem->data( 0, PL_ITEM_ID_ROLE ).toInt() == item )
 421         {
 422           //msg_Dbg( p_intf, "Podcast already in: (%d) %s", item, p_item->p_input->psz_uri);
 423           playlist_Unlock( THEPL );
 424           return;
 425         }
 426     }
 427
 428     //msg_Dbg( p_intf, "Adding podcast: (%d) %s", item, p_item->p_input->psz_uri );
 429     addPodcastItem( p_item );
 430
 431     playlist_Unlock( THEPL );
 432
 433     podcastsParent->setExpanded( true );
 434 }
 435
 436 void PLSelector::plItemRemoved( int id )
 437 {
 438     int c = podcastsParent->childCount();
 439     for( int i = 0; i < c; i++ )
 440     {
 441         QTreeWidgetItem *item = podcastsParent->child(i);
 442         if( item->data( 0, PL_ITEM_ID_ROLE ).toInt() == id )
 443         {
 444             input_item_t *p_input = item->data( 0, IN_ITEM_ROLE ).value<input_item_t*>();
 445             //msg_Dbg( p_intf, "Removing podcast: (%d) %s", id, p_input->psz_uri );
 446             vlc_gc_decref( p_input );
 447             delete item;
 448             return;
 449         }
 450     }
 451 }
 452
 453 void PLSelector::inputItemUpdate( input_item_t *arg )
 454 {
 455     int c = podcastsParent->childCount();
 456     for( int i = 0; i < c; i++ )
 457     {
 458         QTreeWidgetItem *item = podcastsParent->child(i);
 459         input_item_t *p_input = item->data( 0, IN_ITEM_ROLE ).value<input_item_t*>();
 460         if( p_input == arg )
 461         {
 462             PLSelItem *si = itemWidget( item );
 463             char *psz_name = input_item_GetName( p_input );
 464             si->setText( qfu( psz_name ) );
 465             free( psz_name );
 466             return;
 467         }
 468     }
 469 }
 470
 471 void PLSelector::podcastAdd( PLSelItem * )
 472 {
 473     bool ok;
 474     QString url = QInputDialog::getText( this, qtr( "Subscribe" ),
 475                                          qtr( "Enter URL of the podcast to subscribe to:" ),
 476                                          QLineEdit::Normal, QString(), &ok );
 477     if( !ok || url.isEmpty() ) return;
 478
 479     setSource( podcastsParent ); //to load the SD in case it's not loaded
 480
 481     vlc_object_t *p_obj = (vlc_object_t*) vlc_object_find_name( p_intf->p_libvlc, "podcast" );
 482     if( !p_obj ) return;
 483
 484     QString request("ADD:");
 485     request += url.trimmed();
 486     var_SetString( p_obj, "podcast-request", qtu( request ) );
 487     vlc_object_release( p_obj );
 488 }
 489
 490 void PLSelector::podcastRemove( PLSelItem* item )
 491 {
 492     QString question ( qtr( "Do you really want to unsubscribe from %1?" ) );
 493     question = question.arg( item->text() );
 494     QMessageBox::StandardButton res =
 495         QMessageBox::question( this, qtr( "Unsubscribe" ), question,
 496                                QMessageBox::Ok | QMessageBox::Cancel,
 497                                QMessageBox::Cancel );
 498     if( res == QMessageBox::Cancel ) return;
 499
 500     input_item_t *input = item->treeItem()->data( 0, IN_ITEM_ROLE ).value<input_item_t*>();
 501     if( !input ) return;
 502
 503     vlc_object_t *p_obj = (vlc_object_t*) vlc_object_find_name(
 504         p_intf->p_libvlc, "podcast" );
 505     if( !p_obj ) return;
 506
 507     QString request("RM:");
 508     char *psz_uri = input_item_GetURI( input );
 509     request += qfu( psz_uri );
 510     var_SetString( p_obj, "podcast-request", qtu( request ) );
 511     vlc_object_release( p_obj );
 512     free( psz_uri );
 513 }
 514
 515 PLSelItem * PLSelector::itemWidget( QTreeWidgetItem *item )
 516 {
 517     return ( static_cast<PLSelItem*>( QTreeWidget::itemWidget( item, 0 ) ) );
 518 }
 519
 520 void PLSelector::drawBranches ( QPainter * painter, const QRect & rect, const QModelIndex & index ) const
 521 {
 522     if( !model()->hasChildren( index ) ) return;
 523     QStyleOption option;
 524     option.initFrom( this );
 525     option.rect = rect.adjusted( rect.width() - indentation(), 0, 0, 0 );
 526     style()->drawPrimitive( isExpanded( index ) ?
 527                             QStyle::PE_IndicatorArrowDown :
 528                             QStyle::PE_IndicatorArrowRight, &option, painter );
 529 }
 530
 531 void PLSelector::getCurrentSelectedItem( int* type, QString *string)
 532 {
 533     *type = currentItem()->data( 0, TYPE_ROLE ).toInt();
 534     *string = currentItem()->data( 0, NAME_ROLE ).toString();
 535 }
 536
 537 int PLSelector::getCurrentItemCategory()
 538 {
 539     return currentItem()->data( 0, SPECIAL_ROLE ).toInt();
 540 }
 541
 542 void PLSelector::wheelEvent( QWheelEvent *e )
 543 {
 544     // Accept this event in order to prevent unwanted volume up/down changes
 545     e->accept();
 546 }