2 files changed, 137 insertions, 0 deletions

diff --git a/apps/plugins/SOURCES b/apps/plugins/SOURCES
index 4810553d8d..dc68510863 100644
--- a/apps/plugins/SOURCES
+++ b/apps/plugins/SOURCES
@@ -11,6 +11,7 @@ rockbox_flash.c
 search.c
 snow.c
 sort.c
+stats.c
 stopwatch.c
 vbrfix.c
 viewer.c
diff --git a/apps/plugins/stats.c b/apps/plugins/stats.c
new file mode 100644
index 0000000000..b0610010ec
--- /dev/null
+++ b/apps/plugins/stats.c
@@ -0,0 +1,136 @@
+/***************************************************************************
+ *             __________               __   ___.
+ *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
+ *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
+ *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
+ *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
+ *                     \/            \/     \/    \/            \/
+ * $Id$
+ *
+ * Copyright (C) 2005 Jonas Haggqvist
+ *
+ * All files in this archive are subject to the GNU General Public License.
+ * See the file COPYING in the source tree root for full license agreement.
+ *
+ * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
+ * KIND, either express or implied.
+ *
+ ****************************************************************************/
+#include "plugin.h"
+
+static struct plugin_api* rb;
+static int files, dirs;
+static int lasttick;
+static bool abort;
+#ifdef HAVE_LCD_BITMAP
+static int fontwidth, fontheight;
+#endif
+
+#if CONFIG_KEYPAD == PLAYER_PAD 
+#define STATS_STOP BUTTON_STOP
+#else
+#define STATS_STOP BUTTON_OFF
+#endif
+
+void update_screen(void)
+{
+    char buf[15];
+#ifdef HAVE_LCD_BITMAP
+    rb->lcd_clear_display();
+    rb->lcd_putsxy(0,0,"Files:");
+    rb->lcd_putsxy(0,fontheight,"Dirs:");
+    rb->snprintf(buf, sizeof(buf), "%d", files);
+    rb->lcd_putsxy(fontwidth,0,buf);
+    rb->snprintf(buf, sizeof(buf), "%d", dirs);
+    rb->lcd_putsxy(fontwidth,fontheight,buf);
+    rb->lcd_update();
+#else
+    rb->snprintf(buf, sizeof(buf), "Files:%5d", files);
+    rb->lcd_puts(0,0,buf);
+    rb->snprintf(buf, sizeof(buf), "Dirs: %5d", dirs);
+    rb->lcd_puts(0,1,buf);
+#endif
+}
+
+void traversedir(char* location, char* name)
+{
+    int button;
+    struct dirent *entry;
+    DIR* dir;
+    char fullpath[MAX_PATH];
+
+    rb->snprintf(fullpath, sizeof(fullpath), "%s/%s", location, name);
+    dir = rb->opendir(fullpath);
+    if (dir) {
+        entry = rb->readdir(dir);
+        while (entry) {
+            if (abort == true)
+                break;
+            /* Skip .. and . */
+            if (rb->strcmp(entry->d_name, ".") && rb->strcmp(entry->d_name, ".."))
+            {
+                if (entry->attribute & ATTR_DIRECTORY) {
+                    traversedir(fullpath, entry->d_name);
+                    dirs += 1;
+                }
+                else
+                    /* Might want to only count .mp3, .ogg etc. */
+                    files++;
+            }
+            if (*rb->current_tick - lasttick > (HZ/2)) {
+                update_screen();
+                lasttick = *rb->current_tick;
+                button = rb->button_get(false);
+                if (button == STATS_STOP) {
+                    abort = true;
+                    break;
+                }
+            }
+
+            entry = rb->readdir(dir);
+        }
+        rb->closedir(dir);
+    }
+}
+
+enum plugin_status plugin_start(struct plugin_api* api, void* parameter)
+{
+    int button;
+    TEST_PLUGIN_API(api);
+
+    (void)parameter;
+
+    rb = api;
+    files = 0;
+    dirs = 0;
+    abort = false;
+
+#ifdef HAVE_LCD_BITMAP
+    rb->lcd_getstringsize("Files: ", &fontwidth, &fontheight);
+#endif
+    rb->splash(HZ, true, "Counting..");
+    update_screen();
+    lasttick = *rb->current_tick;
+
+    traversedir("", "");
+    if (abort == true) {
+        rb->splash(HZ, true, "Aborted");
+        return PLUGIN_OK;
+    }
+    rb->splash(HZ, true, "Done");
+    update_screen();
+    button = rb->button_get(true);
+    while (1) {
+        switch (button) {
+            case STATS_STOP:
+                return PLUGIN_OK;
+                break;
+            default:
+                if (rb->default_event_handler(button) == SYS_USB_CONNECTED) {
+                    return PLUGIN_USB_CONNECTED;
+                }
+                break;
+        }
+        return PLUGIN_OK;
+    }
+}