src/filelist.c

   1 /* ncmpc (Ncurses MPD Client)
   2  * (c) 2004-2010 The Music Player Daemon Project
   3  * Project homepage: http://musicpd.org
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18  */
  19
  20 #include "filelist.h"
  21
  22 #include <mpd/client.h>
  23
  24 #include <stdbool.h>
  25 #include <string.h>
  26 #include <assert.h>
  27
  28 struct filelist *
  29 filelist_new(void)
  30 {
  31         struct filelist *filelist = g_malloc(sizeof(*filelist));
  32
  33         filelist->entries = g_ptr_array_new();
  34
  35         return filelist;
  36 }
  37
  38 void
  39 filelist_free(struct filelist *filelist)
  40 {
  41         guint i;
  42
  43         for (i = 0; i < filelist_length(filelist); ++i) {
  44                 struct filelist_entry *entry = filelist_get(filelist, i);
  45
  46                 if (entry->entity)
  47                         mpd_entity_free(entry->entity);
  48
  49                 g_slice_free(struct filelist_entry, entry);
  50         }
  51
  52         g_ptr_array_free(filelist->entries, TRUE);
  53         g_free(filelist);
  54 }
  55
  56 struct filelist_entry *
  57 filelist_append(struct filelist *filelist, struct mpd_entity *entity)
  58 {
  59         struct filelist_entry *entry = g_slice_new(struct filelist_entry);
  60
  61         entry->flags = 0;
  62         entry->entity = entity;
  63
  64         g_ptr_array_add(filelist->entries, entry);
  65
  66         return entry;
  67 }
  68
  69 void
  70 filelist_move(struct filelist *filelist, struct filelist *from)
  71 {
  72         guint i;
  73
  74         for (i = 0; i < filelist_length(from); ++i)
  75                 g_ptr_array_add(filelist->entries,
  76                                 g_ptr_array_index(from->entries, i));
  77
  78         g_ptr_array_set_size(from->entries, 0);
  79 }
  80
  81 static gint
  82 filelist_compare_indirect(gconstpointer ap, gconstpointer bp, gpointer data)
  83 {
  84         GCompareFunc compare_func = data;
  85         gconstpointer a = *(const gconstpointer*)ap;
  86         gconstpointer b = *(const gconstpointer*)bp;
  87
  88         return compare_func(a, b);
  89 }
  90
  91 gint
  92 compare_filelist_entry_path(gconstpointer filelist_entry1,
  93                             gconstpointer filelist_entry2)
  94 {
  95         const struct mpd_entity *e1, *e2;
  96         int n = 0;
  97
  98         e1 = ((const struct filelist_entry *)filelist_entry1)->entity;
  99         e2 = ((const struct filelist_entry *)filelist_entry2)->entity;
 100
 101         if (e1 != NULL && e2 != NULL &&
 102             mpd_entity_get_type(e1) == mpd_entity_get_type(e2)) {
 103                 switch (mpd_entity_get_type(e1)) {
 104                 case MPD_ENTITY_TYPE_UNKNOWN:
 105                         break;
 106                 case MPD_ENTITY_TYPE_DIRECTORY:
 107                         n = g_utf8_collate(mpd_directory_get_path(mpd_entity_get_directory(e1)),
 108                                            mpd_directory_get_path(mpd_entity_get_directory(e2)));
 109                         break;
 110                 case MPD_ENTITY_TYPE_SONG:
 111                         break;
 112                 case MPD_ENTITY_TYPE_PLAYLIST:
 113                         n = g_utf8_collate(mpd_playlist_get_path(mpd_entity_get_playlist(e1)),
 114                                            mpd_playlist_get_path(mpd_entity_get_playlist(e2)));
 115                 }
 116         }
 117         return n;
 118 }
 119
 120 /* Sorts the whole filelist, at the moment used by filelist_search */
 121 void
 122 filelist_sort_all(struct filelist *filelist, GCompareFunc compare_func)
 123 {
 124         g_ptr_array_sort_with_data(filelist->entries,
 125                         filelist_compare_indirect,
 126                         compare_func);
 127 }
 128
 129
 130 /* Only sorts the directories and playlist files.
 131  * The songs stay in the order it came from mpd. */
 132 void
 133 filelist_sort_dir_play(struct filelist *filelist, GCompareFunc compare_func)
 134 {
 135         unsigned first, last;
 136         const struct mpd_entity *iter;
 137
 138         assert(filelist && filelist->entries);
 139
 140         if (filelist->entries->len < 2)
 141                 return;
 142
 143         /* If the first entry is NULL, skip it, because NULL stands for "[..]" */
 144         iter = ((struct filelist_entry*) g_ptr_array_index(filelist->entries, 0))->entity;
 145         first = (iter == NULL)? 1 : 0;
 146
 147         /* find the last directory entry */
 148         for (last = first+1; last < filelist->entries->len; last++) {
 149                 iter = ((struct filelist_entry*) g_ptr_array_index(filelist->entries, last))->entity;
 150                 if (mpd_entity_get_type(iter) != MPD_ENTITY_TYPE_DIRECTORY)
 151                         break;
 152         }
 153         if (last == filelist->entries->len - 1)
 154                 last++;
 155         /* sort the directories */
 156         if (last - first > 1)
 157                 g_qsort_with_data(filelist->entries->pdata + first,
 158                                 last - first, sizeof(gpointer),
 159                                 filelist_compare_indirect, compare_func);
 160         /* find the first playlist entry */
 161         for (first = last; first < filelist->entries->len; first++) {
 162                 iter = ((struct filelist_entry*) g_ptr_array_index(filelist->entries, first))->entity;
 163                 if (mpd_entity_get_type(iter) == MPD_ENTITY_TYPE_PLAYLIST)
 164                         break;
 165         }
 166         /* sort the playlist entries */
 167         if (filelist->entries->len - first > 1)
 168                 g_qsort_with_data(filelist->entries->pdata + first,
 169                                 filelist->entries->len - first, sizeof(gpointer),
 170                                 filelist_compare_indirect, compare_func);
 171 }
 172
 173 void
 174 filelist_no_duplicates(struct filelist *filelist)
 175 {
 176         for (int i = filelist_length(filelist) - 1; i >= 0; --i) {
 177                 struct filelist_entry *entry = filelist_get(filelist, i);
 178                 const struct mpd_song *song;
 179
 180                 if (entry->entity == NULL ||
 181                     mpd_entity_get_type(entry->entity) != MPD_ENTITY_TYPE_SONG)
 182                         continue;
 183
 184                 song = mpd_entity_get_song(entry->entity);
 185                 if (filelist_find_song(filelist, song) < i) {
 186                         g_ptr_array_remove_index(filelist->entries, i);
 187                         mpd_entity_free(entry->entity);
 188                         g_slice_free(struct filelist_entry, entry);
 189                 }
 190         }
 191 }
 192
 193 static bool
 194 same_song(const struct mpd_song *a, const struct mpd_song *b)
 195 {
 196         return strcmp(mpd_song_get_uri(a), mpd_song_get_uri(b)) == 0;
 197 }
 198
 199 int
 200 filelist_find_song(struct filelist *fl, const struct mpd_song *song)
 201 {
 202         guint i;
 203
 204         assert(song != NULL);
 205
 206         for (i = 0; i < filelist_length(fl); ++i) {
 207                 struct filelist_entry *entry = filelist_get(fl, i);
 208                 const struct mpd_entity *entity  = entry->entity;
 209
 210                 if (entity != NULL &&
 211                     mpd_entity_get_type(entity) == MPD_ENTITY_TYPE_SONG) {
 212                         const struct mpd_song *song2 =
 213                                 mpd_entity_get_song(entity);
 214
 215                         if (same_song(song, song2))
 216                                 return i;
 217                 }
 218         }
 219
 220         return -1;
 221 }
 222
 223 int
 224 filelist_find_directory(struct filelist *filelist, const char *name)
 225 {
 226         guint i;
 227
 228         assert(name != NULL);
 229
 230         for (i = 0; i < filelist_length(filelist); ++i) {
 231                 struct filelist_entry *entry = filelist_get(filelist, i);
 232                 const struct mpd_entity *entity  = entry->entity;
 233
 234                 if (entity != NULL &&
 235                     mpd_entity_get_type(entity) == MPD_ENTITY_TYPE_DIRECTORY &&
 236                     strcmp(mpd_directory_get_path(mpd_entity_get_directory(entity)),
 237                            name) == 0)
 238                         return i;
 239         }
 240
 241         return -1;
 242 }
 243
 244 void
 245 filelist_recv(struct filelist *filelist, struct mpd_connection *connection)
 246 {
 247         struct mpd_entity *entity;
 248
 249         while ((entity = mpd_recv_entity(connection)) != NULL)
 250                 filelist_append(filelist, entity);
 251 }
 252
 253 struct filelist *
 254 filelist_new_recv(struct mpd_connection *connection)
 255 {
 256         struct filelist *filelist = filelist_new();
 257         filelist_recv(filelist, connection);
 258         return filelist;
 259 }