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

   1 /*****************************************************************************
   2  * sort.c : Playlist sorting functions
   3  *****************************************************************************
   4  * Copyright (C) 1999-2007 the VideoLAN team
   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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23 #include <vlc/vlc.h>
  24 #include "vlc_playlist.h"
  25 #include "playlist_internal.h"
  26
  27
  28 static int playlist_ItemArraySort( playlist_t *p_playlist, int i_items,
  29                                    playlist_item_t **pp_items, int i_mode,
  30                                    int i_type );
  31
  32 /**
  33  * Sort a node.
  34  * This function must be entered with the playlist lock !
  35  *
  36  * \param p_playlist the playlist
  37  * \param p_node the node to sort
  38  * \param i_mode: SORT_ID, SORT_TITLE, SORT_ARTIST, SORT_ALBUM, SORT_RANDOM
  39  * \param i_type: ORDER_NORMAL or ORDER_REVERSE (reversed order)
  40  * \return VLC_SUCCESS on success
  41  */
  42 static int playlist_NodeSort( playlist_t * p_playlist , playlist_item_t *p_node,
  43                               int i_mode, int i_type )
  44 {
  45     playlist_ItemArraySort( p_playlist,p_node->i_children,
  46                             p_node->pp_children, i_mode, i_type );
  47     return VLC_SUCCESS;
  48 }
  49
  50 /**
  51  * Sort a node recursively.
  52  *
  53  * This function must be entered with the playlist lock !
  54  *
  55  * \param p_playlist the playlist
  56  * \param p_node the node to sort
  57  * \param i_mode: SORT_ID, SORT_TITLE, SORT_ARTIST, SORT_ALBUM, SORT_RANDOM
  58  * \param i_type: ORDER_NORMAL or ORDER_REVERSE (reversed order)
  59  * \return VLC_SUCCESS on success
  60  */
  61 int playlist_RecursiveNodeSort( playlist_t *p_playlist, playlist_item_t *p_node,
  62                                 int i_mode, int i_type )
  63 {
  64     int i;
  65     playlist_NodeSort( p_playlist, p_node, i_mode, i_type );
  66     for( i = 0 ; i< p_node->i_children; i++ )
  67     {
  68         if( p_node->pp_children[i]->i_children != -1 )
  69         {
  70             playlist_RecursiveNodeSort( p_playlist, p_node->pp_children[i],
  71                                         i_mode,i_type );
  72         }
  73     }
  74     return VLC_SUCCESS;
  75 }
  76
  77
  78 static int playlist_ItemArraySort( playlist_t *p_playlist, int i_items,
  79                                    playlist_item_t **pp_items, int i_mode,
  80                                    int i_type )
  81 {
  82     int i , i_small , i_position;
  83     playlist_item_t *p_temp;
  84     vlc_value_t val;
  85     val.b_bool = VLC_TRUE;
  86
  87     (void)p_playlist; // a bit surprising we don't need p_playlist!
  88
  89     if( i_mode == SORT_RANDOM )
  90     {
  91         for( i_position = 0; i_position < i_items ; i_position ++ )
  92         {
  93             int i_new;
  94
  95             if( i_items > 1 )
  96                 i_new = rand() % (i_items - 1);
  97             else
  98                 i_new = 0;
  99             p_temp = pp_items[i_position];
 100             pp_items[i_position] = pp_items[i_new];
 101             pp_items[i_new] = p_temp;
 102         }
 103
 104         return VLC_SUCCESS;
 105     }
 106
 107 #define DO_META_SORT( node ) { \
 108     char *psz_a = input_item_GetMeta( pp_items[i]->p_input, vlc_meta_##node ); \
 109     char *psz_b = input_item_GetMeta( pp_items[i_small]->p_input, vlc_meta_##node ); \
 110     /* Nodes go first */ \
 111     if( pp_items[i]->i_children == -1 && pp_items[i_small]->i_children >= 0 ) \
 112         i_test = 1;\
 113     else if( pp_items[i]->i_children >= 0 &&\
 114              pp_items[i_small]->i_children == -1 ) \
 115        i_test = -1; \
 116     /* Both are nodes, sort by name */ \
 117     else if( pp_items[i]->i_children >= 0 && \
 118                pp_items[i_small]->i_children >= 0 ) \
 119     { \
 120          i_test = strcasecmp( pp_items[i]->p_input->psz_name, \
 121                               pp_items[i_small]->p_input->psz_name ); \
 122     } \
 123     /* Both are items */ \
 124     else if( psz_a == NULL && psz_b != NULL ) \
 125         i_test = 1; \
 126     else if( psz_a != NULL && psz_b == NULL ) \
 127         i_test = -1;\
 128     /* No meta, sort by name */ \
 129     else if( psz_a == NULL && psz_b == NULL ) \
 130     { \
 131         i_test = strcasecmp( pp_items[i]->p_input->psz_name, \
 132                              pp_items[i_small]->p_input->psz_name ); \
 133     } \
 134     else \
 135     { \
 136         i_test = strcmp( psz_b, psz_a ); \
 137     } \
 138     free( psz_a ); \
 139     free( psz_b ); \
 140 }
 141
 142     for( i_position = 0; i_position < i_items -1 ; i_position ++ )
 143     {
 144         i_small  = i_position;
 145         for( i = i_position + 1 ; i< i_items ; i++)
 146         {
 147             int i_test = 0;
 148
 149             if( i_mode == SORT_TITLE )
 150             {
 151                 i_test = strcasecmp( pp_items[i]->p_input->psz_name,
 152                                      pp_items[i_small]->p_input->psz_name );
 153             }
 154             else if( i_mode == SORT_TITLE_NUMERIC )
 155             {
 156                 i_test = atoi( pp_items[i]->p_input->psz_name ) -
 157                          atoi( pp_items[i_small]->p_input->psz_name );
 158             }
 159             else if( i_mode == SORT_DURATION )
 160             {
 161                 i_test = pp_items[i]->p_input->i_duration -
 162                              pp_items[i_small]->p_input->i_duration;
 163             }
 164             else if( i_mode == SORT_ARTIST )
 165             {
 166                 DO_META_SORT( Artist );
 167             }
 168             else if( i_mode == SORT_ALBUM )
 169             {
 170                 DO_META_SORT( Album );
 171             }
 172             else if( i_mode == SORT_TITLE_NODES_FIRST )
 173             {
 174                 /* Alphabetic sort, all nodes first */
 175
 176                 if( pp_items[i]->i_children == -1 &&
 177                     pp_items[i_small]->i_children >= 0 )
 178                 {
 179                     i_test = 1;
 180                 }
 181                 else if( pp_items[i]->i_children >= 0 &&
 182                          pp_items[i_small]->i_children == -1 )
 183                 {
 184                     i_test = -1;
 185                 }
 186                 else
 187                 {
 188                     i_test = strcasecmp( pp_items[i]->p_input->psz_name,
 189                                          pp_items[i_small]->p_input->psz_name );
 190                 }
 191             }
 192
 193             if( ( i_type == ORDER_NORMAL  && i_test < 0 ) ||
 194                 ( i_type == ORDER_REVERSE && i_test > 0 ) )
 195             {
 196                 i_small = i;
 197             }
 198         }
 199         p_temp = pp_items[i_position];
 200         pp_items[i_position] = pp_items[i_small];
 201         pp_items[i_small] = p_temp;
 202     }
 203     return VLC_SUCCESS;
 204 }