Merge remote-tracking branch 'boock/master'
[paraslash.git] / ogg_afh.c
index a18c4d7efb7695584bbf0c7de89e491fdc899419..ef42c497062c228ddab10497c3388ffa2b8de90b 100644 (file)
--- a/ogg_afh.c
+++ b/ogg_afh.c
 /*
- * Copyright (C) 2004-2007 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 2004-2011 Andre Noll <maan@systemlinux.org>
  *
- *     This program is free software; you can redistribute it and/or modify
- *     it under the terms of the GNU General Public License as published by
- *     the Free Software Foundation; either version 2 of the License, or
- *     (at your option) any later version.
- *
- *     This program is distributed in the hope that it will be useful,
- *     but WITHOUT ANY WARRANTY; without even the implied warranty of
- *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- *     GNU General Public License for more details.
- *
- *     You should have received a copy of the GNU General Public License
- *     along with this program; if not, write to the Free Software
- *     Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
+ * Licensed under the GPL v2. For licencing details see COPYING.
  */
-/** \file ogg_afh.c para_server's ogg vorbis audio format handler */
 
-#include <inttypes.h>
-#include <ogg/ogg.h>
+/** \file ogg_afh.c Audio format handler for ogg/vorbis files. */
+
 #include <vorbis/codec.h>
-#include <vorbis/vorbisfile.h>
+#include <regex.h>
 
-#include "server.cmdline.h"
-#include "server.h"
-#include "vss.h"
+#include "para.h"
 #include "afh.h"
 #include "error.h"
 #include "string.h"
-#include "fd.h"
+#include "ogg_afh_common.h"
 
-/** must be big enough to hold header */
-#define CHUNK_SIZE 32768
-static double chunk_time = 0.25;
-
-static OggVorbis_File *oggvorbis_file;
-static FILE *infile;
-static int header_len, inbuf_size, vi_channels;
-static char *header, *inbuf;
-static ssize_t *chunk_table, max_chunk_len, num_chunks;
-static struct audio_format_handler *af;
-static long vi_sampling_rate, vi_bitrate, vi_bitrate_nominal;
-
-static int ogg_compute_header_len(void)
-{
-       int ret, len, in = fileno(infile);
-       unsigned int serial;
-       char *buf;
-       ogg_page page;
-       ogg_packet packet;
-       vorbis_comment vc;
+struct private_vorbis_data {
        vorbis_info vi;
-       ogg_stream_state *stream_in = para_malloc(sizeof(ogg_stream_state));
-       ogg_stream_state *stream_out = para_malloc(sizeof(ogg_stream_state));
-       ogg_sync_state *sync_in = para_malloc(sizeof(ogg_sync_state));
-
-       ogg_sync_init(sync_in);
-       vorbis_info_init(&vi);
-       vorbis_comment_init(&vc);
-       buf = ogg_sync_buffer(sync_in, CHUNK_SIZE);
-       len = read(in, buf, CHUNK_SIZE);
-       ret = -E_OGG_READ;
-       if (len <= 0)
-               goto err1;
-       ogg_sync_wrote(sync_in, len);
-       ret = -E_SYNC_PAGEOUT;
-       if (ogg_sync_pageout(sync_in, &page) <= 0)
-               goto err1;
-       serial = ogg_page_serialno(&page);
-       ogg_stream_init(stream_in, serial);
-       ogg_stream_init(stream_out, serial);
-       ret = ogg_stream_pagein(stream_in, &page);
-       if (ret < 0) {
-               ret = E_STREAM_PAGEIN;
-               goto err2;
-       }
-       ret = ogg_stream_packetout(stream_in, &packet);
-       if (ret != 1) {
-               ret = -E_STREAM_PACKETOUT;
-               goto err2;
-       }
-       ret = -E_VORBIS;
-       if (vorbis_synthesis_headerin(&vi, &vc, &packet) < 0)
-               goto err2;
-       PARA_INFO_LOG("channels: %i, rate: %li\n", vi.channels, vi.rate);
-       ogg_stream_packetin(stream_out, &packet);
-
-       ret = ogg_sync_pageout(sync_in, &page);
-       if (ret <= 0) {
-               ret = -E_SYNC_PAGEOUT;
-               goto err2;
-       }
-       ogg_stream_pagein(stream_in, &page);
-       ogg_stream_packetout(stream_in, &packet);
-       ogg_stream_packetin(stream_out, &packet);
-
-       ret = ogg_sync_pageout(sync_in, &page);
-       if (ret <= 0) {
-               ret = -E_SYNC_PAGEOUT;
-               goto err2;
-       }
-       ogg_stream_pagein(stream_in, &page);
-       ogg_stream_packetout(stream_in, &packet);
-       ogg_stream_packetin(stream_out, &packet);
-
-       header_len = 0;
-       while (ogg_stream_flush(stream_out, &page))
-               header_len += page.body_len + page.header_len;
-       ret = len;
-       PARA_INFO_LOG("header_len = %d\n", header_len);
-err2:
-       ogg_stream_destroy(stream_in);
-       ogg_stream_destroy(stream_out);
-err1:
-       ogg_sync_destroy(sync_in);
-       vorbis_info_clear(&vi);
-       vorbis_comment_clear(&vc);
-       return ret;
-}
-
-static void tunetable(void)
-{
-       int i = 1, j = -1, lp = 1;
-       while (i < num_chunks) {
-               if (chunk_table[i] == chunk_table[lp]) {
-                       i++;
-                       continue;
-               }
-               if (j < 0)
-                       tv_scale(i, &af->chunk_tv, &af->eof_tv);
-               for (j = lp; j < i; j++)
-                       chunk_table[j] = chunk_table[i];
-               lp = i;
-       }
-#if 1
-       for (i = 2; i < num_chunks; i++)
-               if (chunk_table[i] != chunk_table[1])
-                       break;
-       lp = i;
-       for (i = 2; i < num_chunks - lp; i++)
-               chunk_table[i] = chunk_table[i + lp];
-#endif
-}
-
-
-/*
- * Alloc and fill array table of byte offsets. chunk_table[i] is the
- * offset in the current infile at which the sample containing time i *
- * CHUNK_TIME begins.
- */
-static void ogg_compute_chunk_table(double time_total)
-{
-       int i, ret, num;
-       ssize_t pos = 0, min = 0, old_pos;
-
-       old_pos = 0;
-       ret = 0;
-       num = time_total / chunk_time + 3;
-       PARA_DEBUG_LOG("chunk time: %g allocating %d chunk pointers\n",
-               chunk_time, num);
-       chunk_table = para_malloc(num * sizeof(size_t));
-       chunk_table[0] = 0;
-       max_chunk_len = 0;
-       rewind(infile);
-       for (i = 1; ret == 0; i++) {
-               ogg_int64_t diff;
-               ret = ov_time_seek(oggvorbis_file, i * chunk_time);
-               if (ret)
-                       break;
-               pos = ov_raw_tell(oggvorbis_file);
-               diff = pos - old_pos;
-               max_chunk_len = PARA_MAX(max_chunk_len, diff);
-               min = (i == 1)? diff : PARA_MIN(min, diff);
-               chunk_table[i] = pos;
-               if (i < 11 || !((i - 1) % 1000)|| i > num - 11)
-                       PARA_DEBUG_LOG("chunk #%d: %g secs, pos: %zd, "
-                               "size: %zd\n", i - 1,
-                               i * chunk_time, pos, pos - old_pos);
-               old_pos = pos;
-       }
-       num_chunks = i - 1;
-       chunk_table[i] = pos;
-       tunetable();
-       PARA_INFO_LOG("%zu chunks (%fs), max chunk: %zd, min chunk: %zd\n",
-               num_chunks, chunk_time, max_chunk_len, min);
-       rewind(infile);
-}
+       vorbis_comment vc;
+};
 
-static void ogg_close_audio_file(void)
+static int vorbis_packet_callback(ogg_packet *packet, int packet_num,
+               struct afh_info *afhi, void *private_data)
 {
-       if (oggvorbis_file) {
-               PARA_DEBUG_LOG("%s", "ov_clear\n");
-               ov_clear(oggvorbis_file);
-               free(oggvorbis_file);
-               oggvorbis_file = NULL;
+       struct private_vorbis_data *pvd = private_data;
+
+       if (vorbis_synthesis_headerin(&pvd->vi, &pvd->vc, packet) < 0)
+               return -E_VORBIS;
+       if (packet_num == 0) {
+               if (pvd->vi.rate == 0)
+                       return -E_VORBIS;
+               afhi->channels = pvd->vi.channels;
+               afhi->frequency = pvd->vi.rate;
+               afhi->bitrate = pvd->vi.bitrate_nominal / 1000;
+               PARA_DEBUG_LOG("channels: %i, sampling rate: %i, bitrate: %i\n",
+                       afhi->channels, afhi->frequency, afhi->bitrate);
+               return 1;
        }
-       free(header);
-       header = NULL;
-       header_len = 0;
-       free(chunk_table);
-       chunk_table = NULL;
-       num_chunks = 0;
-       free(inbuf);
-       inbuf = NULL;
-       inbuf_size = 0;
+       if (packet_num == 1)
+               return 1; /* we also want to have packet #2 */
+       afhi->tags.artist = para_strdup(vorbis_comment_query(&pvd->vc, "artist", 0));
+       afhi->tags.title = para_strdup(vorbis_comment_query(&pvd->vc, "title", 0));
+       afhi->tags.album = para_strdup(vorbis_comment_query(&pvd->vc, "album", 0));
+       afhi->tags.year = para_strdup(vorbis_comment_query(&pvd->vc, "year", 0));
+       afhi->tags.comment = para_strdup(vorbis_comment_query(&pvd->vc, "comment", 0));
+       return 0;
 }
 
-static int ogg_save_header(FILE *file, int len)
+static int ogg_vorbis_get_file_info(char *map, size_t numbytes, __a_unused int fd,
+               struct afh_info *afhi)
 {
        int ret;
-
-       header = para_malloc(len);
-       rewind(file);
-       ret = read(fileno(file), header, len);
-       if (ret != len)
-               return -E_OGG_READ;
-       return 1;
-}
-
-/*
- * Init oggvorbis file and write some tech data to given pointers.
- */
-static int ogg_get_file_info(FILE *file, char *info_str, long unsigned *frames,
-       int *seconds, size_t **vss_chunk_table)
-{
-       int ret;
-       double time_total;
-       vorbis_info *vi;
-       ogg_int64_t raw_total;
-
-       infile = file;
-       if (!file)
-               return -E_OGG_NO_FILE;
-       ret = ogg_compute_header_len();
-       if (ret < 0)
-               return ret;
-       ret = ogg_save_header(file, header_len);
-       if (ret < 0)
-               return ret;
-       rewind(file);
-       oggvorbis_file = para_malloc(sizeof(OggVorbis_File));
-       ret = ov_open(file, oggvorbis_file, NULL, 0);
-       if (ret < 0) {
-               free(oggvorbis_file);
-               free(header);
-               return -E_OGG_OPEN;
-       }
-       ret = -E_OGG_INFO;
-       vi = ov_info(oggvorbis_file, 0);
-       if (!vi)
-               goto err;
-       time_total = ov_time_total(oggvorbis_file, -1);
-       raw_total = ov_raw_total(oggvorbis_file, -1);
-       *seconds = time_total;
-       vi_sampling_rate = vi->rate;
-       vi_bitrate = ov_bitrate(oggvorbis_file, 0);
-       vi_bitrate_nominal = vi->bitrate_nominal;
-       vi_channels = vi->channels;
-       ogg_compute_chunk_table(time_total);
-       *frames = num_chunks;
-       *vss_chunk_table = chunk_table;
-       sprintf(info_str, "audio_file_info1:%zu x %lu, %ldkHz, %d channels, %ldkbps\n"
-               "audio_file_info2: \n"
-               "audio_file_info3: \n",
-               num_chunks, (long unsigned) (chunk_time * 1000 * 1000),
-               vi_sampling_rate / 1000, vi_channels, vi_bitrate / 1000
-               );
-       rewind(file);
-       return 1;
-err:
-       ogg_close_audio_file();
+       struct private_vorbis_data pvd;
+       struct ogg_afh_callback_info vorbis_callback_info = {
+               .packet_callback = vorbis_packet_callback,
+               .private_data = &pvd,
+       };
+
+       vorbis_info_init(&pvd.vi);
+       vorbis_comment_init(&pvd.vc);
+       ret = ogg_get_file_info(map, numbytes, afhi, &vorbis_callback_info);
+       vorbis_info_clear(&pvd.vi);
+       vorbis_comment_clear(&pvd.vc);
        return ret;
 }
 
-static char *ogg_get_header_info(int *len)
-{
-       *len = header_len;
-       return header;
-}
-
 static const char* ogg_suffixes[] = {"ogg", NULL};
 
 /**
- * the init function of the ogg vorbis audio format handler
+ * The init function of the ogg vorbis audio format handler.
  *
- * \param p pointer to the struct to initialize
+ * \param afh Pointer to the struct to initialize.
  */
-void ogg_init(struct audio_format_handler *p)
+void ogg_init(struct audio_format_handler *afh)
 {
-       af = p;
-       af->get_file_info = ogg_get_file_info,
-       af->close_audio_file = ogg_close_audio_file;
-       af->get_header_info = ogg_get_header_info;
-       af->chunk_tv.tv_sec = 0;
-       af->chunk_tv.tv_usec = 250 * 1000;
-       tv_scale(3, &af->chunk_tv, &af->eof_tv);
-       af->suffixes = ogg_suffixes;
+       afh->get_file_info = ogg_vorbis_get_file_info,
+       afh->suffixes = ogg_suffixes;
 }