git.sesse.net Git - vlc/blob - src/playlist/sort.c

   1 /*****************************************************************************
   2  * sort.c : Playlist sorting functions
   3  *****************************************************************************
   4  * Copyright (C) 1999-2004 VideoLAN
   5  * $Id$
   6  *
   7  * Authors: Clément Stenac <zorglub@videolan.org>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22  *****************************************************************************/
  23 #include <stdlib.h>                                      /* free(), strtol() */
  24 #include <stdio.h>                                              /* sprintf() */
  25 #include <string.h>                                            /* strerror() */
  26
  27 #include <vlc/vlc.h>
  28 #include <vlc/input.h>
  29 #include <vlc/vout.h>
  30 #include <vlc/sout.h>
  31
  32 #include "vlc_playlist.h"
  33
  34
  35 int playlist_ItemArraySort( playlist_t *p_playlist, int i_items,
  36                 playlist_item_t **pp_items, int i_mode,
  37                 int i_type );
  38
  39
  40 /**
  41  * Sort the playlist.
  42  * \param p_playlist the playlist
  43  * \param i_mode: SORT_ID, SORT_TITLE, SORT_AUTHOR, SORT_RANDOM
  44  * \param i_type: ORDER_NORMAL or ORDER_REVERSE (reversed order)
  45  * \return VLC_SUCCESS on success
  46  */
  47 int playlist_Sort( playlist_t * p_playlist , int i_mode, int i_type )
  48 {
  49     int  i_id = -1;
  50     vlc_value_t val;
  51     val.b_bool = VLC_TRUE;
  52
  53     vlc_mutex_lock( &p_playlist->object_lock );
  54
  55     p_playlist->i_sort = i_mode;
  56     p_playlist->i_order = i_type;
  57
  58     if( p_playlist->i_index >= 0 )
  59     {
  60         i_id = p_playlist->pp_items[p_playlist->i_index]->input.i_id;
  61     }
  62
  63     playlist_ItemArraySort( p_playlist, p_playlist->i_size,
  64                     p_playlist->pp_items, i_mode, i_type );
  65
  66     if( i_id != -1 )
  67     {
  68         p_playlist->i_index = playlist_GetPositionById( p_playlist, i_id );
  69     }
  70
  71     /* ensure we are in no-view mode */
  72     p_playlist->status.i_view = -1;
  73
  74     vlc_mutex_unlock( &p_playlist->object_lock );
  75
  76     /* Notify the interfaces */
  77     var_Set( p_playlist, "intf-change", val );
  78
  79     return VLC_SUCCESS;
  80 }
  81
  82 /**
  83  * Sort a node.
  84  *
  85  * This function must be entered with the playlist lock !
  86  *
  87  * \param p_playlist the playlist
  88  * \param p_node the node to sort
  89  * \param i_mode: SORT_ID, SORT_TITLE, SORT_AUTHOR, SORT_RANDOM
  90  * \param i_type: ORDER_NORMAL or ORDER_REVERSE (reversed order)
  91  * \return VLC_SUCCESS on success
  92  */
  93 int playlist_NodeSort( playlist_t * p_playlist , playlist_item_t *p_node,
  94                        int i_mode, int i_type )
  95 {
  96
  97     playlist_ItemArraySort( p_playlist,p_node->i_children,
  98                             p_node->pp_children, i_mode, i_type );
  99
 100     p_node->i_serial++;
 101
 102     return VLC_SUCCESS;
 103 }
 104
 105 /**
 106  *
 107  * Sort a node recursively.
 108  *
 109  * This function must be entered with the playlist lock !
 110  *
 111  * \param p_playlist the playlist
 112  * \param p_node the node to sort
 113  * \param i_mode: SORT_ID, SORT_TITLE, SORT_AUTHOR, SORT_RANDOM
 114  * \param i_type: ORDER_NORMAL or ORDER_REVERSE (reversed order)
 115  * \return VLC_SUCCESS on success
 116  */
 117 int playlist_RecursiveNodeSort( playlist_t *p_playlist, playlist_item_t *p_node,
 118                                 int i_mode, int i_type )
 119 {
 120     int i;
 121
 122     playlist_NodeSort( p_playlist, p_node, i_mode, i_type );
 123     for( i = 0 ; i< p_node->i_children; i++ )
 124     {
 125         if( p_node->pp_children[i]->i_children != -1 )
 126         {
 127             playlist_RecursiveNodeSort( p_playlist, p_node->pp_children[i],
 128                                         i_mode,i_type );
 129         }
 130     }
 131
 132     return VLC_SUCCESS;
 133
 134 }
 135
 136
 137 int playlist_ItemArraySort( playlist_t *p_playlist, int i_items,
 138                 playlist_item_t **pp_items, int i_mode,
 139                 int i_type )
 140 {
 141     int i , i_small , i_position;
 142     playlist_item_t *p_temp;
 143     vlc_value_t val;
 144     val.b_bool = VLC_TRUE;
 145
 146     if( i_mode == SORT_RANDOM )
 147     {
 148         for( i_position = 0; i_position < i_items ; i_position ++ )
 149         {
 150             int i_new;
 151
 152             if( i_items > 1 )
 153                 i_new = rand() % (i_items - 1);
 154             else
 155                 i_new = 0;
 156             p_temp = pp_items[i_position];
 157             pp_items[i_position] = pp_items[i_new];
 158             pp_items[i_new] = p_temp;
 159         }
 160
 161         return VLC_SUCCESS;
 162     }
 163
 164     for( i_position = 0; i_position < i_items -1 ; i_position ++ )
 165     {
 166         i_small  = i_position;
 167         for( i = i_position + 1 ; i< i_items ; i++)
 168         {
 169             int i_test = 0;
 170
 171             if( i_mode == SORT_TITLE )
 172             {
 173                 i_test = strcasecmp( pp_items[i]->input.psz_name,
 174                                          pp_items[i_small]->input.psz_name );
 175             }
 176             else if( i_mode == SORT_DURATION )
 177             {
 178                 i_test = pp_items[i]->input.i_duration -
 179                              pp_items[i_small]->input.i_duration;
 180             }
 181             else if( i_mode == SORT_AUTHOR )
 182             {
 183                 msg_Err( p_playlist,"META SORT not implemented" );
 184             }
 185             else if( i_mode == SORT_TITLE_NODES_FIRST )
 186             {
 187                 /* Alphabetic sort, all nodes first */
 188
 189                 if( pp_items[i]->i_children == -1 &&
 190                     pp_items[i_small]->i_children >= 0 )
 191                 {
 192                     i_test = 1;
 193                 }
 194                 else if( pp_items[i]->i_children >= 0 &&
 195                          pp_items[i_small]->i_children == -1 )
 196                 {
 197                     i_test = -1;
 198                 }
 199                 else
 200                 {
 201                     i_test = strcasecmp( pp_items[i]->input.psz_name,
 202                                          pp_items[i_small]->input.psz_name );
 203                 }
 204             }
 205
 206             if( ( i_type == ORDER_NORMAL  && i_test < 0 ) ||
 207                 ( i_type == ORDER_REVERSE && i_test > 0 ) )
 208             {
 209                 i_small = i;
 210             }
 211         }
 212         p_temp = pp_items[i_position];
 213         pp_items[i_position] = pp_items[i_small];
 214         pp_items[i_small] = p_temp;
 215     }
 216     return VLC_SUCCESS;
 217 }
 218
 219
 220 int playlist_NodeGroup( playlist_t * p_playlist , int i_view,
 221                         playlist_item_t *p_root,
 222                         playlist_item_t **pp_items,int i_item,
 223                         int i_mode, int i_type )
 224 {
 225     char *psz_search = NULL;
 226     int i_nodes = 0;
 227     playlist_item_t **pp_nodes = NULL;
 228     playlist_item_t *p_node;
 229     vlc_bool_t b_found;
 230     int i,j;
 231     for( i = 0; i< i_item ; i++ )
 232     {
 233         if( psz_search ) free( psz_search );
 234         if( i_mode == SORT_TITLE )
 235         {
 236             psz_search = strdup( pp_items[i]->input.psz_name );
 237         }
 238         else if ( i_mode == SORT_AUTHOR )
 239         {
 240             psz_search = vlc_input_item_GetInfo( &pp_items[i]->input,
 241                             _("Meta-information"), _( "Artist" ) );
 242         }
 243
 244         if( psz_search && !strcmp( psz_search, "" ) )
 245         {
 246             free( psz_search );
 247             psz_search = strdup( _("Undefined") );
 248         }
 249
 250         b_found = VLC_FALSE;
 251         for( j = 0 ; j< i_nodes; j++ )
 252         {
 253            if( !strcasecmp( psz_search, pp_nodes[j]->input.psz_name ) )
 254            {
 255                 playlist_NodeAppend( p_playlist, i_view,
 256                                      pp_items[i], pp_nodes[j] );
 257                 b_found = VLC_TRUE;
 258                 break;
 259            }
 260         }
 261         if( !b_found )
 262         {
 263             p_node = playlist_NodeCreate( p_playlist, i_view,psz_search,
 264                                           NULL );
 265             INSERT_ELEM( pp_nodes, i_nodes, i_nodes, p_node );
 266             playlist_NodeAppend( p_playlist, i_view,
 267                                  pp_items[i],p_node );
 268         }
 269     }
 270
 271     /* Now, sort the nodes by name */
 272     playlist_ItemArraySort( p_playlist, i_nodes, pp_nodes, SORT_TITLE,
 273                             i_type );
 274
 275     /* Now, sort each node and append it to the root node*/
 276     for( i = 0 ; i< i_nodes ; i++ )
 277     {
 278         playlist_ItemArraySort( p_playlist, pp_nodes[i]->i_children,
 279                                 pp_nodes[i]->pp_children, SORT_TITLE, i_type );
 280
 281         playlist_NodeAppend( p_playlist, i_view,
 282                              pp_nodes[i], p_root );
 283     }
 284     return VLC_SUCCESS;
 285 }