apps/plugins/wavview.c

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  * $Id$
   9  *
  10  * Copyright (C) 2007 Peter D'Hoye
  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 PLUGIN_HEADER
  24
  25 /* temp byte buffer */
  26 uint8_t samples[10 * 1024]; /* read 10KB at the time */
  27
  28 static struct wav_header
  29 {
  30     int8_t   chunkid[4];
  31     int32_t  chunksize;
  32     int8_t   format[4];
  33     int8_t   formatchunkid[4];
  34     int32_t  formatchunksize;
  35     int16_t  audioformat;
  36     int16_t  numchannels;
  37     uint32_t samplerate;
  38     uint32_t byterate;
  39     uint16_t blockalign;
  40     uint16_t bitspersample;
  41     int8_t   datachunkid[4];
  42     int32_t  datachunksize;
  43 } header;
  44
  45 #define WAV_HEADER_FORMAT "4L44LSSLLSS4L"
  46
  47 struct peakstruct
  48 {
  49     int lmin;
  50     int lmax;
  51     int rmin;
  52     int rmax;
  53 };
  54
  55 /* TO DO: limit used LCD height, so the waveform isn't streched vertically? */
  56
  57 #define LEFTZERO    (LCD_HEIGHT / 4)
  58 #define RIGHTZERO   (3 * (LCD_HEIGHT / 4))
  59 #define YSCALE      ((0x8000 / (LCD_HEIGHT / 4)) + 1)
  60
  61 /* global vars */
  62 static char *audiobuf;
  63 static ssize_t audiobuflen;
  64 static uint32_t mempeakcount = 0;
  65 static uint32_t filepeakcount = 0;
  66 static uint32_t fppmp = 0; /* file peaks per mem peaks */
  67 static uint32_t zoomlevel = 1;
  68 static uint32_t leftmargin = 0;
  69 static uint32_t center = 0;
  70 static uint32_t ppp = 1;
  71
  72 /* helper function copied from libwavpack bits.c */
  73 void little_endian_to_native (void *data, char *format)
  74 {
  75     unsigned char *cp = (unsigned char *) data;
  76
  77     while (*format) {
  78         switch (*format) {
  79             case 'L':
  80                 *(long *)cp = letoh32(*(long *)cp);
  81                 cp += 4;
  82                 break;
  83
  84             case 'S':
  85                 *(short *)cp = letoh16(*(short *)cp);
  86                 cp += 2;
  87                 break;
  88
  89             default:
  90                 if (*format >= '0' && *format <= '9')
  91                     cp += *format - '0';
  92
  93                 break;
  94         }
  95         format++;
  96     }
  97 }
  98 /* --- */
  99
 100 /*  read WAV file
 101     display some info about it
 102     store peak info in aufiobuf for display routine */
 103 static int readwavpeaks(const char *filename)
 104 {
 105     register uint32_t bytes_read;
 106     register uint32_t fppmp_count;
 107     register int16_t sampleval;
 108     register uint16_t* sampleshort = NULL;
 109
 110     int file;
 111     uint32_t total_bytes_read = 0;
 112     char tstr[128];
 113     int hours;
 114     int minutes;
 115     int seconds;
 116     uint32_t peakcount = 0;
 117     struct peakstruct* peak = NULL;
 118
 119     if(rb->strcasecmp (filename + rb->strlen (filename) - 3, "wav"))
 120     {
 121         rb->splash(HZ*2, "Only for wav files!");
 122         return -1;
 123     }
 124
 125     file = rb->open(filename, O_RDONLY);
 126
 127     if(file < 0)
 128     {
 129         rb->splash(HZ*2, "Could not open file!");
 130         return -1;
 131     }
 132
 133     if(rb->read(file, &header, sizeof (header)) != sizeof (header))
 134     {
 135         rb->splash(HZ*2, "Could not read file!");
 136         rb->close (file);
 137         return -1;
 138     }
 139
 140     total_bytes_read += sizeof (header);
 141     little_endian_to_native(&header, WAV_HEADER_FORMAT);
 142
 143         if (rb->strncmp(header.chunkid, "RIFF", 4) ||
 144             rb->strncmp(header.formatchunkid, "fmt ", 4) ||
 145             rb->strncmp(header.datachunkid, "data", 4) ||
 146         (header.bitspersample != 16) ||
 147         header.audioformat != 1)
 148     {
 149             rb->splash(HZ*2, "Incompatible wav file!");
 150             rb->close (file);
 151             return -1;
 152     }
 153
 154     rb->lcd_clear_display();
 155     rb->lcd_puts(0, 0, "Viewing file:");
 156     rb->lcd_puts_scroll(0, 1, (unsigned char *)filename);
 157     rb->lcd_update();
 158
 159     rb->snprintf(tstr,127, "Channels: %s",
 160                            header.numchannels == 1 ? "mono" : "stereo");
 161     rb->lcd_puts(0, 2, tstr);
 162
 163     rb->snprintf(tstr,127, "Bits/sample: %d", header.bitspersample);
 164     rb->lcd_puts(0, 3, tstr);
 165
 166     rb->snprintf(tstr,127, "Samplerate: %d Hz", (int)(header.samplerate));
 167     rb->lcd_puts(0, 4, tstr);
 168
 169     seconds = header.datachunksize / header.byterate;
 170     hours = seconds / 3600;
 171     seconds -= hours * 3600;
 172     minutes = seconds / 60;
 173     seconds -= minutes * 60;
 174     rb->snprintf(tstr,127, "Length (hh:mm:ss): %02d:%02d:%02d", hours,
 175                                                                 minutes,
 176                                                                 seconds);
 177     rb->lcd_puts(0, 5, tstr);
 178     rb->lcd_puts(0, 6, "Searching for peaks... ");
 179     rb->lcd_update();
 180
 181     /* calculate room for peaks */
 182     filepeakcount = header.datachunksize /
 183                     (header.numchannels * (header.bitspersample / 8));
 184     mempeakcount = audiobuflen / sizeof(struct peakstruct);
 185     fppmp = (filepeakcount / mempeakcount) + 1;
 186     peak = (struct peakstruct*)audiobuf;
 187
 188     fppmp_count = fppmp;
 189     mempeakcount = 0;
 190     while(total_bytes_read < (header.datachunksize +
 191                               sizeof(struct wav_header)))
 192     {
 193         bytes_read = rb->read(file, &samples, sizeof(samples));
 194         total_bytes_read += bytes_read;
 195
 196         if(0 == bytes_read)
 197         {
 198             rb->splash(HZ*2, "File read error!");
 199             rb->close (file);
 200             return -1;
 201         }
 202         if(((bytes_read/4)*4) != bytes_read)
 203         {
 204             rb->splashf(HZ*2, "bytes_read/*4 err: %ld",(long int)bytes_read);
 205             rb->close (file);
 206             return -1;
 207         }
 208
 209         sampleshort = (int16_t*)samples;
 210         sampleval = letoh16(*sampleshort);
 211         peak->lmin = sampleval;
 212         peak->lmax = sampleval;
 213         sampleval = letoh16(*(sampleshort+1));
 214         peak->rmin = sampleval;
 215         peak->rmax = sampleval;
 216
 217         while(bytes_read)
 218         {
 219             sampleval = letoh16(*sampleshort++);
 220             if(sampleval < peak->lmin)
 221                 peak->lmin = sampleval;
 222             else if (sampleval > peak->lmax)
 223                 peak->lmax = sampleval;
 224
 225             sampleval = letoh16(*sampleshort++);
 226             if(sampleval < peak->rmin)
 227                 peak->rmin = sampleval;
 228             else if (sampleval > peak->rmax)
 229                 peak->rmax = sampleval;
 230
 231             bytes_read -= 4;
 232             peakcount++;
 233             fppmp_count--;
 234             if(!fppmp_count)
 235             {
 236                 peak++;
 237                 mempeakcount++;
 238                 fppmp_count = fppmp;
 239                 sampleval = letoh16(*sampleshort);
 240                 peak->lmin = sampleval;
 241                 peak->lmax = sampleval;
 242                 sampleval = letoh16(*(sampleshort+1));
 243                 peak->rmin = sampleval;
 244                 peak->rmax = sampleval;
 245             }
 246         }
 247
 248         /* update progress */
 249         rb->snprintf(tstr,127, "Searching for peaks... %d%%",(int)
 250             (total_bytes_read / ((header.datachunksize +
 251                                  sizeof(struct wav_header)) / 100)));
 252         rb->lcd_puts(0, 6, tstr);
 253         rb->lcd_update();
 254
 255         /* allow user to abort */
 256         if(ACTION_KBD_ABORT == rb->get_action(CONTEXT_KEYBOARD,TIMEOUT_NOBLOCK))
 257         {
 258             rb->splash(HZ*2, "ABORTED");
 259             rb->close (file);
 260             return -1;
 261         }
 262     }
 263
 264     rb->lcd_puts(0, 6, "Searching for peaks... done");
 265     rb->lcd_update();
 266
 267     rb->close (file);
 268
 269     return 0;
 270 }
 271
 272 int displaypeaks(void)
 273 {
 274     register int x = 0;
 275     register int lymin = INT_MAX;
 276     register int lymax = INT_MIN;
 277     register int rymin = INT_MAX;
 278     register int rymax = INT_MIN;
 279     register unsigned int peakcount = 0;
 280     struct peakstruct* peak = (struct peakstruct*)audiobuf + leftmargin;
 281
 282 #if LCD_DEPTH > 1
 283     unsigned org_forecolor = rb->lcd_get_foreground();
 284     rb->lcd_set_foreground(LCD_LIGHTGRAY);
 285 #endif
 286
 287     if(!zoomlevel) zoomlevel = 1;
 288     ppp = (mempeakcount / LCD_WIDTH) / zoomlevel;  /* peaks per pixel */
 289
 290     rb->lcd_clear_display();
 291
 292     rb->lcd_hline(0, LCD_WIDTH-1, LEFTZERO - (0x8000 / YSCALE));
 293     rb->lcd_hline(0, LCD_WIDTH-1, LEFTZERO);
 294     rb->lcd_hline(0, LCD_WIDTH-1, LEFTZERO + (0x8000 / YSCALE));
 295     rb->lcd_hline(0, LCD_WIDTH-1, RIGHTZERO - (0x8000 / YSCALE));
 296     rb->lcd_hline(0, LCD_WIDTH-1, RIGHTZERO);
 297     rb->lcd_hline(0, LCD_WIDTH-1, RIGHTZERO + (0x8000 / YSCALE));
 298
 299 #if LCD_DEPTH > 1
 300     rb->lcd_set_foreground(LCD_BLACK);
 301 #endif
 302
 303     /* draw zoombar */
 304     rb->lcd_hline(leftmargin / (mempeakcount / LCD_WIDTH),
 305                   (leftmargin / (mempeakcount / LCD_WIDTH)) +
 306                         (LCD_WIDTH / zoomlevel),
 307                   LCD_HEIGHT / 2);
 308
 309     while((x < LCD_WIDTH) && (peakcount < mempeakcount))
 310     {
 311         if(peak->lmin < lymin)
 312             lymin = peak->lmin;
 313         if(peak->lmax > lymax)
 314             lymax = peak->lmax;
 315         if(peak->rmin < rymin)
 316             rymin = peak->rmin;
 317         if(peak->rmax > rymax)
 318             rymax = peak->rmax;
 319         peak++;
 320         if(0 == (peakcount % ppp))
 321         {
 322             /* drawing time */
 323             rb->lcd_vline(x, LEFTZERO - (lymax / YSCALE),
 324                           LEFTZERO - (lymin / YSCALE));
 325             rb->lcd_vline(x, RIGHTZERO - (rymax / YSCALE),
 326                           RIGHTZERO - (rymin / YSCALE));
 327             lymin = INT_MAX;
 328             lymax = INT_MIN;
 329             rymin = INT_MAX;
 330             rymax = INT_MIN;
 331             x++;
 332             rb->lcd_update();
 333         }
 334         peakcount++;
 335     }
 336
 337 #if LCD_DEPTH > 1
 338     rb->lcd_set_foreground(org_forecolor);
 339 #endif
 340
 341     return 0;
 342 }
 343
 344 void show_help(void)
 345 {
 346     rb->lcd_clear_display();
 347     rb->lcd_puts(0, 0, "WAVVIEW USAGE:");
 348     rb->lcd_puts(0, 2, "up/down: zoom out/in");
 349     rb->lcd_puts(0, 3, "left/right: pan left/right");
 350     rb->lcd_puts(0, 4, "select: refresh/continue");
 351     rb->lcd_puts(0, 5, "stop/off: quit");
 352     rb->lcd_update();
 353 }
 354
 355 enum plugin_status plugin_start(const void *parameter)
 356 {
 357     unsigned int quit = 0;
 358     unsigned int action = 0;
 359     unsigned int dodisplay = 1;
 360     int retval;
 361
 362     if (!parameter)
 363         return PLUGIN_ERROR;
 364
 365     audiobuf = rb->plugin_get_audio_buffer((size_t *)&audiobuflen);
 366
 367     if (!audiobuf)
 368     {
 369         rb->splash(HZ*2, "unable to get audio buffer!");
 370         return PLUGIN_ERROR;
 371     }
 372
 373 #ifdef HAVE_ADJUSTABLE_CPU_FREQ
 374     rb->cpu_boost(true);
 375 #endif
 376
 377     retval = readwavpeaks(parameter); /* read WAV file and create peaks array */
 378
 379 #ifdef HAVE_ADJUSTABLE_CPU_FREQ
 380     rb->cpu_boost(false);
 381 #endif
 382
 383     if(retval)
 384         return 0;
 385
 386     /* press any key to continue */
 387     while(1)
 388     {
 389         retval = rb->get_action(CONTEXT_KEYBOARD,TIMEOUT_BLOCK);
 390         if(ACTION_KBD_ABORT == retval)
 391             return 0;
 392         else if(ACTION_KBD_SELECT == retval)
 393             break;
 394     }
 395
 396     /* start with the overview */
 397     zoomlevel = 1;
 398     leftmargin = 0;
 399
 400     while(!quit)
 401     {
 402         if(!center)
 403             center = mempeakcount / 2;
 404         if(zoomlevel <= 1)
 405         {
 406             zoomlevel = 1;
 407             leftmargin = 0;
 408         }
 409         else
 410         {
 411             if(center < (mempeakcount / (zoomlevel * 2)))
 412                 center = mempeakcount / (zoomlevel * 2);
 413             if(center > (((zoomlevel * 2) - 1) * (mempeakcount /
 414                                                   (zoomlevel * 2))))
 415                 center = ((zoomlevel * 2) - 1) * (mempeakcount /
 416                                                   (zoomlevel * 2));
 417             leftmargin = center - (mempeakcount / (zoomlevel * 2));
 418         }
 419
 420 #ifdef HAVE_ADJUSTABLE_CPU_FREQ
 421         rb->cpu_boost(true);
 422 #endif
 423         if(dodisplay)
 424             displaypeaks();
 425         dodisplay = 1;
 426
 427 #ifdef HAVE_ADJUSTABLE_CPU_FREQ
 428         rb->cpu_boost(false);
 429 #endif
 430
 431         action = rb->get_action(CONTEXT_KEYBOARD, TIMEOUT_BLOCK);
 432         switch(action)
 433         {
 434         case ACTION_KBD_UP:
 435             /* zoom out */
 436             if(zoomlevel > 1)
 437                 zoomlevel /= 2;
 438             rb->splashf(HZ/2, "ZOOM: %dx",(int)zoomlevel);
 439             break;
 440         case ACTION_KBD_DOWN:
 441             if(zoomlevel < (mempeakcount / LCD_WIDTH / 2))
 442                 zoomlevel *= 2;
 443             rb->splashf(HZ/2, "ZOOM: %dx",(int)zoomlevel);
 444             break;
 445         case ACTION_KBD_LEFT:
 446             center -= 10 * (mempeakcount / LCD_WIDTH) / zoomlevel;
 447             break;
 448         case ACTION_KBD_RIGHT:
 449             center += 10 * (mempeakcount / LCD_WIDTH) / zoomlevel;
 450             break;
 451         case ACTION_KBD_ABORT:
 452             quit = 1;
 453             break;
 454         case ACTION_KBD_SELECT:
 455             /* refresh */
 456             break;
 457         case ACTION_KBD_PAGE_FLIP:
 458             /* menu key shows help */
 459             show_help();
 460             while(1)
 461             {
 462                 retval = rb->get_action(CONTEXT_KEYBOARD,TIMEOUT_BLOCK);
 463                 if((ACTION_KBD_SELECT == retval) ||
 464                    (ACTION_KBD_ABORT == retval))
 465                    break;
 466             }
 467             break;
 468         default:
 469             /* eat it */
 470             dodisplay = 0;
 471             break;
 472         }
 473     }
 474
 475     return 0;
 476 }