Mercurial > audlegacy
view Plugins/Input/aac/src/libmp4.c @ 1024:3b1dec2c5da0 trunk
[svn] - fix infinite loop issue
author | nenolod |
---|---|
date | Wed, 10 May 2006 16:01:35 -0700 |
parents | 0b03a1d25c44 |
children | 2b42e48a688e |
line wrap: on
line source
#include <glib.h> #include <glib/gi18n.h> #include <gtk/gtk.h> #include <string.h> #include <stdlib.h> #include "faad.h" #include "mp4ff.h" #include "tagging.h" #include <audacious/plugin.h> #include <audacious/output.h> #include <libaudacious/util.h> #include <libaudacious/titlestring.h> #include <libaudacious/vfs.h> #define MP4_VERSION VERSION #define BUFFER_SIZE FAAD_MIN_STREAMSIZE*64 const char *audmp4_id3_genres[GENRE_MAX] = { N_("Blues"), N_("Classic Rock"), N_("Country"), N_("Dance"), N_("Disco"), N_("Funk"), N_("Grunge"), N_("Hip-Hop"), N_("Jazz"), N_("Metal"), N_("New Age"), N_("Oldies"), N_("Other"), N_("Pop"), N_("R&B"), N_("Rap"), N_("Reggae"), N_("Rock"), N_("Techno"), N_("Industrial"), N_("Alternative"), N_("Ska"), N_("Death Metal"), N_("Pranks"), N_("Soundtrack"), N_("Euro-Techno"), N_("Ambient"), N_("Trip-Hop"), N_("Vocal"), N_("Jazz+Funk"), N_("Fusion"), N_("Trance"), N_("Classical"), N_("Instrumental"), N_("Acid"), N_("House"), N_("Game"), N_("Sound Clip"), N_("Gospel"), N_("Noise"), N_("AlternRock"), N_("Bass"), N_("Soul"), N_("Punk"), N_("Space"), N_("Meditative"), N_("Instrumental Pop"), N_("Instrumental Rock"), N_("Ethnic"), N_("Gothic"), N_("Darkwave"), N_("Techno-Industrial"), N_("Electronic"), N_("Pop-Folk"), N_("Eurodance"), N_("Dream"), N_("Southern Rock"), N_("Comedy"), N_("Cult"), N_("Gangsta Rap"), N_("Top 40"), N_("Christian Rap"), N_("Pop/Funk"), N_("Jungle"), N_("Native American"), N_("Cabaret"), N_("New Wave"), N_("Psychedelic"), N_("Rave"), N_("Showtunes"), N_("Trailer"), N_("Lo-Fi"), N_("Tribal"), N_("Acid Punk"), N_("Acid Jazz"), N_("Polka"), N_("Retro"), N_("Musical"), N_("Rock & Roll"), N_("Hard Rock"), N_("Folk"), N_("Folk/Rock"), N_("National Folk"), N_("Swing"), N_("Fast-Fusion"), N_("Bebob"), N_("Latin"), N_("Revival"), N_("Celtic"), N_("Bluegrass"), N_("Avantgarde"), N_("Gothic Rock"), N_("Progressive Rock"), N_("Psychedelic Rock"), N_("Symphonic Rock"), N_("Slow Rock"), N_("Big Band"), N_("Chorus"), N_("Easy Listening"), N_("Acoustic"), N_("Humour"), N_("Speech"), N_("Chanson"), N_("Opera"), N_("Chamber Music"), N_("Sonata"), N_("Symphony"), N_("Booty Bass"), N_("Primus"), N_("Porn Groove"), N_("Satire"), N_("Slow Jam"), N_("Club"), N_("Tango"), N_("Samba"), N_("Folklore"), N_("Ballad"), N_("Power Ballad"), N_("Rhythmic Soul"), N_("Freestyle"), N_("Duet"), N_("Punk Rock"), N_("Drum Solo"), N_("A Cappella"), N_("Euro-House"), N_("Dance Hall"), N_("Goa"), N_("Drum & Bass"), N_("Club-House"), N_("Hardcore"), N_("Terror"), N_("Indie"), N_("BritPop"), N_("Negerpunk"), N_("Polsk Punk"), N_("Beat"), N_("Christian Gangsta Rap"), N_("Heavy Metal"), N_("Black Metal"), N_("Crossover"), N_("Contemporary Christian"), N_("Christian Rock"), N_("Merengue"), N_("Salsa"), N_("Thrash Metal"), N_("Anime"), N_("JPop"), N_("Synthpop") }; #define AAC_MAGIC (unsigned char [4]) { 0xFF, 0xF9, 0x5C, 0x80 } static void mp4_init(void); static void mp4_about(void); static void mp4_play(char *); static void mp4_stop(void); static void mp4_pause(short); static void mp4_seek(int); static int mp4_getTime(void); static void mp4_cleanup(void); static int mp4_IsOurFile(char *); static void mp4_getSongTitle(char *filename, char **, int *); static void* mp4Decode(void *); void audmp4_file_info_box(gchar *); gboolean buffer_playing; InputPlugin mp4_ip = { 0, // handle 0, // filename "MP4 Audio Plugin", mp4_init, mp4_about, 0, // configuration mp4_IsOurFile, 0, //scandir mp4_play, mp4_stop, mp4_pause, mp4_seek, 0, // set equalizer mp4_getTime, 0, // get volume 0, mp4_cleanup, 0, // obsolete 0, // send visualisation data 0, // set player window info 0, // set song title text mp4_getSongTitle, // get song title text audmp4_file_info_box, // info box 0, // to output plugin }; typedef struct _mp4cfg{ gshort file_type; #define FILE_UNKNOW 0 #define FILE_MP4 1 #define FILE_AAC 2 } Mp4Config; static Mp4Config mp4cfg; static GThread *decodeThread; GStaticMutex mutex = G_STATIC_MUTEX_INIT; static int seekPosition = -1; void getMP4info(char*); int getAACTrack(MP4FileHandle); uint32_t mp4_read_callback(void *data, void *buffer, uint32_t len) { if (data == NULL) return -1; return vfs_fread(buffer, 1, len, (VFSFile *) data); } uint32_t mp4_seek_callback(void *data, uint64_t pos) { if (data == NULL) return -1; return vfs_fseek((VFSFile *) data, pos, SEEK_SET); } /* * Function extname (filename) * * Return pointer within filename to its extenstion, or NULL if * filename has no extension. * */ static gchar * extname(const char *filename) { gchar *ext = strrchr(filename, '.'); if (ext != NULL) ++ext; return ext; } InputPlugin *get_iplugin_info(void) { return(&mp4_ip); } static void mp4_init(void) { mp4cfg.file_type = FILE_UNKNOW; seekPosition = -1; return; } static void mp4_play(char *filename) { buffer_playing = TRUE; decodeThread = g_thread_create((GThreadFunc)mp4Decode, g_strdup(filename), TRUE, NULL); return; } static void mp4_stop(void) { if(buffer_playing){ buffer_playing = FALSE; g_thread_join(decodeThread); mp4_ip.output->close_audio(); } } static int mp4_IsOurFile(char *filename) { VFSFile *file; gchar* extension; gchar magic[8]; extension = strrchr(filename, '.'); if ((file = vfs_fopen(filename, "rb"))) { vfs_fread(magic, 1, 8, file); if (!memcmp(magic, AAC_MAGIC, 4)) { vfs_fclose(file); return 1; } if (!memcmp(magic, "ID3", 3)) { // ID3 tag bolted to the front, obfuscated magic bytes vfs_fclose(file); if (extension &&( !strcasecmp(extension, ".mp4") || // official extension !strcasecmp(extension, ".m4a") || // Apple mp4 extension !strcasecmp(extension, ".aac") // old MPEG2/4-AAC extension )) return 1; else return 0; } if (!memcmp(&magic[4], "ftyp", 4)) { vfs_fclose(file); return 1; } vfs_fclose(file); } return 0; } static void mp4_about(void) { static GtkWidget *aboutbox; if(aboutbox!=NULL) return; aboutbox = xmms_show_message("About MP4 AAC player plugin", "Using libfaad2-" FAAD2_VERSION " for decoding.\n" "Copyright (c) 2005-2006 Audacious team", "Ok", FALSE, NULL, NULL); gtk_signal_connect(GTK_OBJECT(aboutbox), "destroy", GTK_SIGNAL_FUNC(gtk_widget_destroyed), &aboutbox); } static void mp4_pause(short flag) { mp4_ip.output->pause(flag); } static void mp4_seek(int time) { seekPosition = time; while(buffer_playing && seekPosition!=-1) xmms_usleep(10000); } static int mp4_getTime(void) { if(!buffer_playing) return (-1); else return (mp4_ip.output->output_time()); } static void mp4_cleanup(void) { } #if 0 static void mp4_getSongInfo(char *filename) { if(mp4cfg.file_type == FILE_MP4) getMP4info(filename); else if(mp4cfg.file_type == FILE_AAC) ; } #endif static gchar *mp4_get_song_title(char *filename) { mp4ff_callback_t *mp4cb = g_malloc0(sizeof(mp4ff_callback_t)); VFSFile *mp4fh; mp4ff_t *mp4file; gchar *title = NULL; mp4fh = vfs_fopen(filename, "rb"); mp4cb->read = mp4_read_callback; mp4cb->seek = mp4_seek_callback; mp4cb->user_data = mp4fh; if (!(mp4file = mp4ff_open_read(mp4cb))) { g_free(mp4cb); vfs_fclose(mp4fh); } else { TitleInput *input; gchar *tmpval; input = bmp_title_input_new(); mp4ff_meta_get_title(mp4file, &input->track_name); mp4ff_meta_get_album(mp4file, &input->album_name); mp4ff_meta_get_artist(mp4file, &input->performer); mp4ff_meta_get_date(mp4file, &tmpval); mp4ff_meta_get_genre(mp4file, &input->genre); if (tmpval) { input->year = atoi(tmpval); free(tmpval); } input->file_name = g_path_get_basename(filename); input->file_path = g_path_get_dirname(filename); input->file_ext = extname(filename); title = xmms_get_titlestring(xmms_get_gentitle_format(), input); free (input->track_name); free (input->album_name); free (input->performer); free (input->genre); free (input->file_name); free (input->file_path); free (input); free (mp4cb); vfs_fclose(mp4fh); } if (!title) { title = g_path_get_basename(filename); if (extname(title)) *(extname(title) - 1) = '\0'; } return title; } static void mp4_getSongTitle(char *filename, char **title_real, int *len_real) { (*title_real) = mp4_get_song_title(filename); (*len_real) = -1; } static int my_decode_mp4( char *filename, mp4ff_t *mp4file ) { // We are reading a MP4 file gint mp4track= getAACTrack(mp4file); if ( mp4track < 0 ) { //TODO: check here for others Audio format..... g_print("Unsupported Audio track type\n"); } else { faacDecHandle decoder; unsigned char *buffer = NULL; guint bufferSize = 0; gulong samplerate; guchar channels; //guint avgBitrate; gulong duration; gulong msDuration; gulong numSamples; gulong sampleID = 1; unsigned int framesize = 1024; mp4AudioSpecificConfig mp4ASC; gchar *xmmstitle = NULL; gchar *ext = strrchr(filename, '.'); xmmstitle = getMP4title(mp4file, filename); if(xmmstitle == NULL) xmmstitle = g_strdup(filename); mp4_ip.set_info(xmmstitle, -1, -1, samplerate, channels); decoder = faacDecOpen(); mp4ff_get_decoder_config(mp4file, mp4track, &buffer, &bufferSize); if ( !buffer ) { faacDecClose(decoder); return FALSE; } if ( faacDecInit2(decoder, buffer, bufferSize, &samplerate, &channels) < 0 ) { faacDecClose(decoder); return FALSE; } /* Add some hacks for SBR profile */ if (AudioSpecificConfig(buffer, bufferSize, &mp4ASC) >= 0) { if (mp4ASC.frameLengthFlag == 1) framesize = 960; if (mp4ASC.sbr_present_flag == 1) framesize *= 2; } g_free(buffer); if( !channels ) { g_print("Number of Channels not supported\n"); faacDecClose(decoder); return FALSE; } // duration= mp4ff_get_track_duration(mp4file, mp4track); numSamples = mp4ff_num_samples(mp4file, mp4track); msDuration = ((float)numSamples * (float)(framesize - 1.0)/(float)samplerate) * 1000; mp4_ip.output->open_audio(FMT_S16_NE, samplerate, channels); mp4_ip.output->flush(0); mp4_ip.set_info(xmmstitle, msDuration, mp4ff_get_avg_bitrate( mp4file, mp4track ), samplerate,channels); while ( buffer_playing ) { void* sampleBuffer; faacDecFrameInfo frameInfo; gint rc; /* Seek if seek position has changed */ if ( seekPosition!=-1 ) { sampleID = (float)seekPosition*(float)samplerate/(float)(framesize - 1.0); mp4_ip.output->flush(seekPosition*1000); seekPosition = -1; } /* Otherwise continue playing */ buffer=NULL; bufferSize=0; /* If we've run to the end of the file, we're done. */ if(sampleID > numSamples){ /* Finish playing before we close the output. */ while ( mp4_ip.output->buffer_playing() ) { xmms_usleep(10000); } mp4_ip.output->flush(seekPosition*1000); mp4_ip.output->close_audio(); faacDecClose(decoder); return FALSE; } rc= mp4ff_read_sample(mp4file, mp4track, sampleID++, &buffer, &bufferSize); /*g_print(":: %d/%d\n", sampleID-1, numSamples);*/ /* If we can't read the file, we're done. */ if((!rc) || (buffer== NULL)){ g_print("MP4: read error\n"); sampleBuffer = NULL; sampleID=0; mp4_ip.output->buffer_free(); mp4_ip.output->close_audio(); faacDecClose(decoder); return FALSE; } sampleBuffer= faacDecDecode(decoder, &frameInfo, buffer, bufferSize); /* If there was an error decoding, we're done. */ if(frameInfo.error > 0){ g_print("MP4: %s\n", faacDecGetErrorMessage(frameInfo.error)); mp4_ip.output->close_audio(); faacDecClose(decoder); return FALSE; } if(buffer){ g_free(buffer); buffer=NULL; bufferSize=0; } produce_audio(mp4_ip.output->written_time(), FMT_S16_NE, channels, frameInfo.samples<<1, sampleBuffer, &buffer_playing); } mp4_ip.output->close_audio(); faacDecClose(decoder); } return TRUE; } static void my_decode_aac( char *filename ) { // WE ARE READING AN AAC FILE VFSFile *file = NULL; faacDecHandle decoder = 0; guchar *buffer = 0; gulong bufferconsumed = 0; gulong samplerate = 0; guchar channels; gulong buffervalid = 0; TitleInput* input; gchar *temp = g_strdup(filename); gchar *ext = strrchr(temp, '.'); gchar *xmmstitle = NULL; faacDecConfigurationPtr config; if((file = vfs_fopen(filename, "rb")) == 0){ g_print("AAC: can't find file %s\n", filename); buffer_playing = FALSE; g_static_mutex_unlock(&mutex); g_thread_exit(NULL); } if((decoder = faacDecOpen()) == NULL){ g_print("AAC: Open Decoder Error\n"); vfs_fclose(file); buffer_playing = FALSE; g_static_mutex_unlock(&mutex); g_thread_exit(NULL); } config = faacDecGetCurrentConfiguration(decoder); config->useOldADTSFormat = 0; faacDecSetConfiguration(decoder, config); if((buffer = g_malloc(BUFFER_SIZE)) == NULL){ g_print("AAC: error g_malloc\n"); vfs_fclose(file); buffer_playing = FALSE; faacDecClose(decoder); g_static_mutex_unlock(&mutex); g_thread_exit(NULL); } if((buffervalid = vfs_fread(buffer, 1, BUFFER_SIZE, file))==0){ g_print("AAC: Error reading file\n"); g_free(buffer); vfs_fclose(file); buffer_playing = FALSE; faacDecClose(decoder); g_static_mutex_unlock(&mutex); g_thread_exit(NULL); } XMMS_NEW_TITLEINPUT(input); input->file_name = (char*)g_basename(temp); input->file_ext = ext ? ext+1 : NULL; input->file_path = temp; if(!strncmp((char*)buffer, "ID3", 3)){ gint size = 0; vfs_fseek(file, 0, SEEK_SET); size = (buffer[6]<<21) | (buffer[7]<<14) | (buffer[8]<<7) | buffer[9]; size+=10; vfs_fread(buffer, 1, size, file); buffervalid = vfs_fread(buffer, 1, BUFFER_SIZE, file); } xmmstitle = xmms_get_titlestring(xmms_get_gentitle_format(), input); if(xmmstitle == NULL) xmmstitle = g_strdup(input->file_name); if(temp) g_free(temp); if(input->performer) g_free(input->performer); if(input->album_name) g_free(input->album_name); if(input->track_name) g_free(input->track_name); if(input->genre) g_free(input->genre); g_free(input); bufferconsumed = faacDecInit(decoder, buffer, buffervalid, &samplerate, &channels); if(mp4_ip.output->open_audio(FMT_S16_NE,samplerate,channels) == FALSE){ g_print("AAC: Output Error\n"); g_free(buffer); buffer=0; faacDecClose(decoder); vfs_fclose(file); mp4_ip.output->close_audio(); g_free(xmmstitle); buffer_playing = FALSE; g_static_mutex_unlock(&mutex); g_thread_exit(NULL); } mp4_ip.set_info(xmmstitle, -1, -1, samplerate, channels); mp4_ip.output->flush(0); while(buffer_playing && buffervalid > 0){ faacDecFrameInfo finfo; unsigned long samplesdecoded; char* sample_buffer = NULL; if(bufferconsumed > 0){ memmove(buffer, &buffer[bufferconsumed], buffervalid-bufferconsumed); buffervalid -= bufferconsumed; buffervalid += vfs_fread(&buffer[buffervalid], 1, BUFFER_SIZE-buffervalid, file); bufferconsumed = 0; } sample_buffer = faacDecDecode(decoder, &finfo, buffer, buffervalid); if(finfo.error){ config = faacDecGetCurrentConfiguration(decoder); if(config->useOldADTSFormat != 1){ faacDecClose(decoder); decoder = faacDecOpen(); config = faacDecGetCurrentConfiguration(decoder); config->useOldADTSFormat = 1; faacDecSetConfiguration(decoder, config); finfo.bytesconsumed=0; finfo.samples = 0; faacDecInit(decoder, buffer, buffervalid, &samplerate, &channels); }else{ g_print("FAAD2 Warning %s\n", faacDecGetErrorMessage(finfo.error)); buffervalid = 0; } } bufferconsumed += finfo.bytesconsumed; samplesdecoded = finfo.samples; if((samplesdecoded<=0) && !sample_buffer){ g_print("AAC: error sample decoding\n"); continue; } produce_audio(mp4_ip.output->written_time(), FMT_S16_LE, channels, samplesdecoded<<1, sample_buffer, &buffer_playing); } mp4_ip.output->buffer_free(); mp4_ip.output->close_audio(); buffer_playing = FALSE; g_free(buffer); faacDecClose(decoder); g_free(xmmstitle); vfs_fclose(file); seekPosition = -1; buffer_playing = FALSE; g_static_mutex_unlock(&mutex); g_thread_exit(NULL); } static void *mp4Decode( void *args ) { mp4ff_callback_t *mp4cb = g_malloc0(sizeof(mp4ff_callback_t)); VFSFile *mp4fh; mp4ff_t *mp4file; char* url= (char*)args; char filename[255]; memset( filename, '\0', 255 ); #if 0 /* If we have a URL-style string, de-URLise it... */ if ( !strncmp( url, "file://", 7 ) ) { char *output= curl_unescape( url, 0 ); char *tmp= output+7; strncpy( filename, tmp, 254 ); curl_free( output ); } else { strncpy( filename, url, 254 ); } #endif strncpy( filename, url, 254 ); mp4fh = vfs_fopen(filename, "rb"); mp4cb->read = mp4_read_callback; mp4cb->seek = mp4_seek_callback; mp4cb->user_data = mp4fh; g_static_mutex_lock(&mutex); seekPosition= -1; buffer_playing= TRUE; mp4file= mp4ff_open_read(mp4cb); if( !mp4file ) { mp4cfg.file_type = FILE_AAC; vfs_fclose(mp4fh); g_free(mp4cb); } else { mp4cfg.file_type = FILE_MP4; } if ( mp4cfg.file_type == FILE_MP4 ) { my_decode_mp4( filename, mp4file ); g_free(args); vfs_fclose(mp4fh); buffer_playing = FALSE; g_static_mutex_unlock(&mutex); g_thread_exit(NULL); } else { my_decode_aac( filename ); } return NULL; }