apps/plugins/sort.c

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  * $Id$
   9  *
  10  * Copyright (C) 2002 Linus Nielsen Feltzing
  11  *
  12  * This program is free software; you can redistribute it and/or
  13  * modify it under the terms of the GNU General Public License
  14  * as published by the Free Software Foundation; either version 2
  15  * of the License, or (at your option) any later version.
  16  *
  17  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  18  * KIND, either express or implied.
  19  *
  20  ****************************************************************************/
  21 #include "plugin.h"
  22
  23 /****************************************************************************
  24  * Buffer handling:
  25  *
  26  * We allocate the MP3 buffer for storing the text to be sorted, and then
  27  * search the buffer for newlines and store an array of character pointers
  28  * to the strings.
  29  *
  30  * The pointer array grows from the top of the buffer and downwards:
  31  *
  32  * |-------------|
  33  * | pointers[2] |--------|
  34  * | pointers[1] |------| |
  35  * | pointers[0] |----| | |
  36  * |-------------|    | | |
  37  * |             |    | | |
  38  * |             |    | | |
  39  * | free space  |    | | |
  40  * |             |    | | |
  41  * |             |    | | |
  42  * |-------------|    | | |
  43  * |             |    | | |
  44  * | line 3\0    |<---| | |
  45  * | line 2\0    |<-----| |
  46  * | line 1\0    |<-------|
  47  * |-------------|
  48  *
  49  * The advantage of this method is that we optimize the buffer usage.
  50  *
  51  * The disadvantage is that the string pointers will be loaded in reverse
  52  * order. We therefore sort the strings in reverse order as well, so we
  53  * don't have to sort an already sorted buffer.
  54  ****************************************************************************/
  55
  56 /***************************************************************************
  57  * TODO: Implement a merge sort for files larger than the buffer
  58  ****************************************************************************/
  59
  60 PLUGIN_HEADER
  61
  62 ssize_t buf_size;
  63 static char *filename;
  64 static int num_entries;
  65 static char **pointers;
  66 static char *stringbuffer;
  67 static char crlf[2] = "\r\n";
  68
  69 /* Compare function for sorting backwards */
  70 static int compare(const void* p1, const void* p2)
  71 {
  72     char *s1 = *(char **)p1;
  73     char *s2 = *(char **)p2;
  74
  75     return rb->strcasecmp(s2, s1);
  76 }
  77
  78 static void sort_buffer(void)
  79 {
  80     rb->qsort(pointers, num_entries, sizeof(char *), compare);
  81 }
  82
  83 int read_buffer(int offset)
  84 {
  85     int fd;
  86     char *buf_ptr;
  87     char *tmp_ptr;
  88     int readsize;
  89
  90     fd = rb->open(filename, O_RDONLY);
  91     if(fd < 0)
  92         return 10 * fd - 1;
  93
  94     /* Fill the buffer from the file */
  95     rb->lseek(fd, offset, SEEK_SET);
  96     readsize = rb->read(fd, stringbuffer, buf_size);
  97     rb->close(fd);
  98
  99     if(readsize < 0)
 100         return readsize * 10 - 2;
 101
 102     /* Temporary fix until we can do merged sorting */
 103     if(readsize == buf_size)
 104         return buf_size; /* File too big */
 105
 106     buf_ptr = stringbuffer;
 107     num_entries = 0;
 108
 109     do {
 110         tmp_ptr = buf_ptr;
 111         while(*buf_ptr != '\n' && buf_ptr < (char *)pointers) {
 112             /* Terminate the string with CR... */
 113             if(*buf_ptr == '\r')
 114                 *buf_ptr = 0;
 115             buf_ptr++;
 116         }
 117         /* ...and with LF */
 118         if(*buf_ptr == '\n')
 119             *buf_ptr = 0;
 120         else {
 121             return tmp_ptr - stringbuffer; /* Buffer is full, return
 122                                               the point to resume at */
 123         }
 124
 125         pointers--;
 126         *pointers = tmp_ptr;
 127         num_entries++;
 128         buf_ptr++;
 129     } while(buf_ptr < stringbuffer + readsize);
 130
 131     return 0;
 132 }
 133
 134 static int write_file(void)
 135 {
 136     char tmpfilename[MAX_PATH+1];
 137     int fd;
 138     int i;
 139     int rc;
 140
 141     /* Create a temporary file */
 142     rb->snprintf(tmpfilename, MAX_PATH+1, "%s.tmp", filename);
 143     fd = rb->creat(tmpfilename);
 144     if(fd < 0)
 145         return 10 * fd - 1;
 146
 147     /* Write the sorted strings, with appended CR/LF, to the temp file,
 148        in reverse order */
 149     for(i = num_entries-1;i >= 0;i--) {
 150         rc = rb->write(fd, pointers[i], rb->strlen(pointers[i]));
 151         if(rc < 0) {
 152             rb->close(fd);
 153             return 10 * rc - 2;
 154         }
 155
 156         rc = rb->write(fd, crlf, 2);
 157         if(rc < 0) {
 158             rb->close(fd);
 159             return 10 * rc - 3;
 160         }
 161     }
 162
 163     rb->close(fd);
 164
 165     /* Remove the original file */
 166     rc = rb->remove(filename);
 167     if(rc < 0) {
 168         return 10 * rc - 4;
 169     }
 170
 171     /* Replace the old file with the new */
 172     rc = rb->rename(tmpfilename, filename);
 173     if(rc < 0) {
 174         return 10 * rc - 5;
 175     }
 176     return 0;
 177 }
 178
 179 enum plugin_status plugin_start(const void* parameter)
 180 {
 181     char *buf;
 182     int rc;
 183
 184     filename = (char *)parameter;
 185
 186     buf = rb->plugin_get_audio_buffer((size_t *)&buf_size); /* start munching memory */
 187
 188     stringbuffer = buf;
 189     pointers = (char **)(buf + buf_size - sizeof(int));
 190
 191     rb->lcd_clear_display();
 192     rb->splash(0, "Loading...");
 193
 194     rc = read_buffer(0);
 195     if(rc == 0) {
 196         rb->lcd_clear_display();
 197         rb->splash(0, "Sorting...");
 198         sort_buffer();
 199
 200         rb->lcd_clear_display();
 201         rb->splash(0, "Writing...");
 202
 203         rc = write_file();
 204         if(rc < 0) {
 205             rb->lcd_clear_display();
 206             rb->splashf(HZ, "Can't write file: %d", rc);
 207         } else {
 208             rb->lcd_clear_display();
 209             rb->splash(HZ, "Done");
 210         }
 211     } else {
 212         if(rc < 0) {
 213             rb->lcd_clear_display();
 214             rb->splashf(HZ, "Can't read file: %d", rc);
 215         } else {
 216             rb->lcd_clear_display();
 217             rb->splash(HZ, "The file is too big");
 218         }
 219     }
 220
 221     return PLUGIN_OK;
 222 }