diff options
Diffstat (limited to 'debian/transcode/transcode-1.1.7/import/extract_ogm.c')
| -rw-r--r-- | debian/transcode/transcode-1.1.7/import/extract_ogm.c | 684 |
1 files changed, 684 insertions, 0 deletions
diff --git a/debian/transcode/transcode-1.1.7/import/extract_ogm.c b/debian/transcode/transcode-1.1.7/import/extract_ogm.c new file mode 100644 index 00000000..4bc91fb5 --- /dev/null +++ b/debian/transcode/transcode-1.1.7/import/extract_ogm.c @@ -0,0 +1,684 @@ +/* + ogminfo -- utility for extracting raw streams from an OGG media file + + Written by Moritz Bunkus <moritz@bunkus.org> + Integrated into transcode by Tilmann Bitterberg <transcode@tibit.org> + + Distributed under the GPL + see the file COPYING for details + or visit http://www.gnu.org/copyleft/gpl.html +*/ + +#include "transcode.h" +#include "libtc/libtc.h" +#include "tcinfo.h" + +#include <sys/types.h> + +#include "ioaux.h" +#include "tc.h" + +#define NOAUDIO 0 +#define NOVIDEO 1 +#define NOTEXT 2 + +unsigned char *xaudio = NULL; +unsigned char *xvideo = NULL; +unsigned char *xtext = NULL; +int no[3]; +int xraw = 0; + +static int verbose_flag = TC_QUIET; + +#if (HAVE_OGG && HAVE_VORBIS) + +#include <ogg/ogg.h> +#include <vorbis/codec.h> + +#include "ogmstreams.h" + +#define BLOCK_SIZE 4096 + +#define ACVORBIS 0xffff +#define ACPCM 0x0001 +#define ACMP3 0x0055 +#define ACAC3 0x2000 + +typedef struct stream_t { + int serial; + int fd; + double sample_rate; + int eos, comment; + int sno; + char stype; + ogg_stream_state instate; + struct stream_t *next; + + int acodec; + ogg_int64_t bwritten; + + ogg_stream_state outstate; + int packetno; + ogg_int64_t max_granulepos; + + int subnr; +} stream_t; + +stream_t *first; +int nastreams = 0, nvstreams = 0, ntstreams = 0, numstreams = 0; +char basename[] = "stdout"; + +static void add_stream(stream_t *ndmx) { + stream_t *cur = first; + + if (first == NULL) { + first = ndmx; + first->next = NULL; + } else { + cur = first; + while (cur->next != NULL) + cur = cur->next; + cur->next = ndmx; + ndmx->next = NULL; + } +} + +static stream_t *find_stream(int fserial) { + stream_t *cur = first; + + while (cur != NULL) { + if (cur->serial == fserial) + break; + cur = cur->next; + } + + return cur; +} + +double highest_ts = 0; + +static int extraction_requested(unsigned char *s, int stream, int type) { + int i; + + if (no[type]) + return 0; + if (strlen((char *)s) == 0) + return 1; + for (i = 0; i < strlen((char *)s); i++) + if (s[i] == stream) + return 1; + + return 0; +} + +static void flush_pages(stream_t *stream, ogg_packet *op) { + ogg_page page; + + while (ogg_stream_flush(&stream->outstate, &page)) { + int ih, ib; + + ih = tc_pwrite(stream->fd, page.header, page.header_len); + if (ih != page.header_len) { + tc_log_error(__FILE__, "error while writing page header"); + import_exit(1); /* XXX ugly */ + } + ib = tc_pwrite(stream->fd, page.body, page.body_len); + if (ib != page.body_len) { + tc_log_error(__FILE__, "error while writing page bofy"); + import_exit(1); /* XXX ugly */ + } + if (verbose_flag & TC_DEBUG) + tc_log_msg(__FILE__, "x/a%d: %d + %d written", stream->sno, ih, ib); + } +} + +static void write_pages(stream_t *stream, ogg_packet *op) { + ogg_page page; + + while (ogg_stream_pageout(&stream->outstate, &page)) { + int ih, ib; + + ih = tc_pwrite(stream->fd, page.header, page.header_len); + if (ih != page.header_len) { + tc_log_error(__FILE__, "error while writing page header"); + import_exit(1); /* XXX ugly */ + } + ib = tc_pwrite(stream->fd, page.body, page.body_len); + if (ib != page.body_len) { + tc_log_error(__FILE__, "error while writing page bofy"); + import_exit(1); /* XXX ugly */ + } + if (verbose_flag & TC_DEBUG) + tc_log_msg(__FILE__, "x/a%d: %d + %d written", stream->sno, ih, ib); + } +} + +static void handle_packet(stream_t *stream, ogg_packet *pack, ogg_page *page) { + int i, w, hdrlen, end; + long long lenbytes; + char *sub; + char out[100]; + ogg_int64_t pgp, sst; + + //tc_log_msg(__FILE__, "Going handle 1"); + if (pack->e_o_s) { + stream->eos = 1; + pack->e_o_s = 1; + } + + if (((double)pack->granulepos * 1000.0 / (double)stream->sample_rate) > + highest_ts) + highest_ts = (double)pack->granulepos * 1000.0 / + (double)stream->sample_rate; + + switch (stream->stype) { + case 'v': + if (!extraction_requested(xvideo, stream->sno, NOVIDEO)) + return; + break; + case 'a': + if (!extraction_requested(xaudio, stream->sno, NOAUDIO)) + return; + break; + case 't': + if (!extraction_requested(xtext, stream->sno, NOTEXT)) + return; + break; + } + + hdrlen = (*pack->packet & OGM_PACKET_LEN_BITS01) >> 6; + hdrlen |= (*pack->packet & OGM_PACKET_LEN_BITS2) << 1; + for (i = 0, lenbytes = 0; i < hdrlen; i++) { + lenbytes = lenbytes << 8; + lenbytes += *((unsigned char *)pack->packet + hdrlen - i); + } + + switch (stream->stype) { + case 'v': + if (((*pack->packet & 3) == OGM_PACKET_TYPE_HEADER) || + ((*pack->packet & 3) == OGM_PACKET_TYPE_COMMENT)) + return; + i = tc_pwrite(stream->fd, (char *)&pack->packet[hdrlen + 1], + pack->bytes - 1 - hdrlen); + if (i != pack->bytes - 1 - hdrlen) { + tc_log_error(__FILE__, "error while writing data"); + import_exit(1); /* XXX ugly */ + } + if (verbose_flag & TC_DEBUG) + tc_log_msg(__FILE__, "x/v%d: %d written", stream->sno, i); + break; + case 't': + if (((*pack->packet & 3) == OGM_PACKET_TYPE_HEADER) || + ((*pack->packet & 3) == OGM_PACKET_TYPE_COMMENT)) + return; + + if (xraw) { + i = tc_pwrite(stream->fd, (char *)&pack->packet[hdrlen + 1], + pack->bytes - 1 - hdrlen); + if (i != pack->bytes - 1 - hdrlen) { + tc_log_error(__FILE__, "error while writing data"); + import_exit(1); /* XXX ugly */ + } + if (verbose_flag & TC_DEBUG) + tc_log_msg(__FILE__, "x/t%d: %d written", stream->sno, i); + return; + } + + sub = (char *)&pack->packet[hdrlen + 1]; + if ((strlen(sub) > 1) || (*sub != ' ')) { + sst = (pack->granulepos / stream->sample_rate) * 1000; + pgp = sst + lenbytes; + tc_snprintf(out,sizeof(out), "%d\r\n%02d:%02d:%02d,%03d --> " \ + "%02d:%02d:%02d,%03d\r\n", stream->subnr + 1, + (int)(sst / 3600000), + (int)(sst / 60000) % 60, + (int)(sst / 1000) % 60, + (int)(sst % 1000), + (int)(pgp / 3600000), + (int)(pgp / 60000) % 60, + (int)(pgp / 1000) % 60, + (int)(pgp % 1000)); + i = tc_pwrite(stream->fd, out, strlen(out)); + if (i != strlen(out)) { + tc_log_error(__FILE__, "error while writing data"); + import_exit(1); /* XXX ugly */ + } + end = strlen(sub) - 1; + while ((end >= 0) && ((sub[end] == '\n') || (sub[end] == '\r'))) { + sub[end] = 0; + end--; + } + w = tc_pwrite(stream->fd, sub, strlen(sub)); + if (i == strlen(sub)) { + i += w; + } else { + tc_log_error(__FILE__, "error while writing data"); + import_exit(1); + } + w = tc_pwrite(stream->fd, "\r\n\r\n", 4); + if (w == 4) { + i += w; + } else { + tc_log_error(__FILE__, "error while writing data"); + import_exit(1); + } + stream->subnr++; + if (verbose_flag & TC_DEBUG) + tc_log_msg(__FILE__, "x/t%d: %d written", stream->sno, i); + } + break; + case 'a': + switch (stream->acodec) { + case ACVORBIS: + if (xraw) { + if (stream->packetno == 0) { + i = tc_pwrite(stream->fd, (char *)pack->packet, pack->bytes); + if (i != pack->bytes) { + tc_log_error(__FILE__, "error while writing data"); + import_exit(1); /* XXX ugly */ + } + } else { + i = tc_pwrite(stream->fd, (char *)&pack->packet[1], + pack->bytes - 1); + if (i != pack->bytes - 1) { + tc_log_error(__FILE__, "error while writing data"); + import_exit(1); /* XXX ugly */ + } + } + if (verbose_flag & TC_DEBUG) + tc_log_msg(__FILE__, "x/a%d: %d written", stream->sno, i); + return; + } + stream->max_granulepos = (pack->granulepos > stream->max_granulepos ? + pack->granulepos : stream->max_granulepos); + if ((stream->packetno == 0) || (stream->packetno == 2)) { + ogg_stream_packetin(&stream->outstate, pack); + flush_pages(stream, pack); + } else { + ogg_stream_packetin(&stream->outstate, pack); + write_pages(stream, pack); + } + stream->packetno++; + break; + default: + if (((*pack->packet & 3) == OGM_PACKET_TYPE_HEADER) || + ((*pack->packet & 3) == OGM_PACKET_TYPE_COMMENT)) + return; + + i = tc_pwrite(stream->fd, pack->packet + 1 + hdrlen, + pack->bytes - 1 - hdrlen); + if (i != pack->bytes - 1 - hdrlen) { + tc_log_error(__FILE__, "error while writing data"); + import_exit(1); /* XXX ugly */ + } + stream->bwritten += i; + if (verbose_flag & TC_DEBUG) + tc_log_msg(__FILE__, "x/a%d: %d written", stream->sno, i); + break; + } + break; + } +} + +static void process_ogm(int fdin, int fdout) +{ + ogg_sync_state sync; + ogg_page page; + ogg_packet pack; + vorbis_info vi; + vorbis_comment vc; + char *buf, *new_name; + int nread, np, sno; + int endofstream = 0; + stream_t *stream = NULL; + + ogg_sync_init(&sync); + while (1) { + np = ogg_sync_pageseek(&sync, &page); + if (np < 0) { + tc_log_error(__FILE__, "ogg_sync_pageseek failed"); + return; + } + if (np == 0) { + buf = ogg_sync_buffer(&sync, BLOCK_SIZE); + if (!buf) { + tc_log_error(__FILE__, "ogg_sync_buffer failed"); + return; + } + if ((nread = read(fdin, buf, BLOCK_SIZE)) <= 0) { + if (verbose_flag & TC_INFO) + tc_log_info(__FILE__, "end of stream 1"); + return; + } + ogg_sync_wrote(&sync, nread); + continue; + } + + if (!ogg_page_bos(&page)) { + break; + } else { + ogg_stream_state sstate; + sno = ogg_page_serialno(&page); + if (ogg_stream_init(&sstate, sno)) { + tc_log_error(__FILE__, "ogg_stream_init failed"); + return; + } + ogg_stream_pagein(&sstate, &page); + ogg_stream_packetout(&sstate, &pack); + + if ((pack.bytes >= 7) && ! strncmp(&pack.packet[1], "vorbis", 6)) { + + stream = tc_malloc(sizeof(stream_t)); + if (stream == NULL) { + tc_log_error(__FILE__, "malloc failed."); + exit(1); + } + + memset(stream, 0, sizeof(stream_t)); + if (verbose_flag & TC_INFO) { + vorbis_info_init(&vi); + vorbis_comment_init(&vc); + if (vorbis_synthesis_headerin(&vi, &vc, &pack) >= 0) { + tc_log_info(__FILE__, "(a%d/%d) Vorbis audio (channels %d " + "rate %ld)", nastreams + 1, numstreams + 1, + vi.channels, vi.rate); + stream->sample_rate = vi.rate; + } else + tc_log_warn(__FILE__, "(a%d/%d) Vorbis audio stream indicated " + "but no Vorbis stream header found.", + nastreams + 1, numstreams + 1); + } + stream->serial = sno; + stream->acodec = ACVORBIS; + stream->sample_rate = -1; + stream->sno = nastreams + 1; + stream->stype = 'a'; + ac_memcpy(&stream->instate, &sstate, sizeof(sstate)); + if (extraction_requested(xaudio, nastreams + 1, NOAUDIO)) { + stream->fd = fdout; + if (stream->fd == -1) { + tc_log_error(__FILE__, "Failed to create \"%s\" (%d, %s).", + "new_name", errno, strerror(errno)); + exit(1); + } + if (!xraw) + ogg_stream_init(&stream->outstate, rand()); + if (verbose_flag & TC_INFO) + tc_log_info(__FILE__, "Extracting a%d to \"%s\".", + nastreams + 1, "new_name"); + do + handle_packet(stream, &pack, &page); + while (ogg_stream_packetout(&stream->instate, &pack) == 1); + } + add_stream(stream); + nastreams++; + numstreams++; + } else if ((pack.bytes >= 142) && + !strncmp(&pack.packet[1],"Direct Show Samples embedded in Ogg", + 35) ) { + if ((*(int32_t*)(pack.packet+96) == 0x05589f80) && + (pack.bytes >= 184)) { + tc_log_warn(__FILE__, "(v%d/%d) Found old video header. Not " + "supported.", nvstreams + 1, numstreams + 1); + } else if (*(int32_t*)pack.packet+96 == 0x05589F81) { + tc_log_warn(__FILE__, "(a%d/%d) Found old audio header. Not " + "supported.", nastreams + 1, numstreams + 1); + } else { + if (verbose_flag & TC_INFO) + tc_log_warn(__FILE__, "OGG stream %d has an old header with an " + "unknown type.", numstreams + 1); + } + } else if (((*pack.packet & OGM_PACKET_TYPE_BITS ) == OGM_PACKET_TYPE_HEADER) && + (pack.bytes >= (int)(sizeof(ogm_stream_header) + 1 - sizeof(int)))) { + ogm_stream_header *sth = (ogm_stream_header *)(pack.packet + 1); + if (!strncmp(sth->streamtype, "video", 5)) { + unsigned long codec; + char ccodec[5]; + strncpy(ccodec, sth->subtype, 4); + ccodec[4] = 0; + codec = (sth->subtype[0] << 24) + + (sth->subtype[1] << 16) + (sth->subtype[2] << 8) + sth->subtype[3]; + if (verbose_flag & TC_INFO) + tc_log_info(__FILE__, "(v%d/%d) fps: %.3f width height: %dx%d " + "codec: %p (%s)", nvstreams + 1, numstreams + 1, + (double)10000000 / (double)sth->time_unit, + sth->sh.video.width, sth->sh.video.height, + (void *)codec, ccodec); + stream = tc_malloc(sizeof(stream_t)); + if (stream == NULL) { + tc_log_error(__FILE__, "malloc failed."); + exit(1); + } + stream->stype = 'v'; + stream->serial = sno; + stream->sample_rate = (double)10000000 / (double)sth->time_unit; + stream->sno = nvstreams + 1; + ac_memcpy(&stream->instate, &sstate, sizeof(sstate)); + if (extraction_requested(xvideo, nvstreams + 1, NOVIDEO)) { + stream->fd = fdout; + + if (verbose_flag & TC_INFO) + tc_log_info(__FILE__, "Extracting v%d to \"%s\".", + nvstreams + 1, "new_name"); + do { + handle_packet(stream, &pack, &page); + } while (ogg_stream_packetout(&stream->instate, &pack) == 1); + } + add_stream(stream); + nvstreams++; + numstreams++; + } else if (!strncmp(sth->streamtype, "audio", 5)) { + int codec; + char buf[5]; + ac_memcpy(buf, sth->subtype, 4); + buf[4] = 0; + codec = strtoul(buf, NULL, 16); + if (verbose_flag & TC_INFO) { + tc_log_info(__FILE__, "(a%d/%d) codec: %d (0x%04x) (%s), bits per " + "sample: %d channels: %hd samples per second: %lld " + " avgbytespersec: %hd blockalign: %d", + nastreams + 1, numstreams + 1, codec, codec, + codec == ACPCM ? "PCM" : codec == 55 ? "MP3" : + codec == ACMP3 ? "MP3" : + codec == ACAC3 ? "AC3" : "unknown", + sth->bits_per_sample, sth->sh.audio.channels, + (long long)sth->samples_per_unit, + sth->sh.audio.avgbytespersec, sth->sh.audio.blockalign); + } + stream = tc_malloc(sizeof(stream_t)); + if (stream == NULL) { + tc_log_error(__FILE__, "malloc failed."); + exit(1); + } + stream->sno = nastreams + 1; + stream->stype = 'a'; + stream->sample_rate = sth->samples_per_unit * + sth->sh.audio.channels; + stream->serial = sno; + stream->acodec = codec; + ac_memcpy(&stream->instate, &sstate, sizeof(sstate)); + if (extraction_requested(xaudio, nastreams + 1, NOAUDIO)) { + + /* + codec == ACPCM ? "wav" : + codec == ACMP3 ? "mp3" : + codec == ACAC3 ? "ac3" : "audio"); + */ + stream->fd = fdout; + if (verbose_flag & TC_INFO) + tc_log_info(__FILE__, "Extracting a%d to \"%s\".", + nastreams + 1, "new_name"); + do { + handle_packet(stream, &pack, &page); + } while (ogg_stream_packetout(&stream->instate, &pack) == 1); + } + add_stream(stream); + nastreams++; + numstreams++; + } else if (!strncmp(sth->streamtype, "text", 4)) { + if (verbose_flag & TC_INFO) + tc_log_info(__FILE__, "(t%d/%d) text/subtitle stream", + ntstreams + 1, numstreams + 1); + stream = tc_malloc(sizeof(stream_t)); + if (stream == NULL) { + tc_log_error(__FILE__, "malloc failed."); + exit(1); + } + stream->sno = ntstreams + 1; + stream->stype = 't'; + stream->sample_rate = (double)10000000 / (double)sth->time_unit; + stream->serial = sno; + ac_memcpy(&stream->instate, &sstate, sizeof(sstate)); + if (extraction_requested(xtext, ntstreams + 1, NOTEXT)) { + new_name = tc_malloc(strlen(basename) + 20); + if (!new_name) { + tc_log_error(__FILE__, "Failed to allocate %d bytes.", + (int)strlen(basename) + 20); + exit(1); + } + if (!xraw) + tc_snprintf(new_name, strlen(basename) + 20, "%s-t%d.srt", basename, ntstreams + 1); + else + tc_snprintf(new_name, strlen(basename) + 20, "%s-t%d.raw", basename, ntstreams + 1); + //stream->fd = open(new_name, O_WRONLY | O_CREAT | O_TRUNC, + // S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH); + stream->fd = fdout; + if (stream->fd == -1) { + tc_log_error(__FILE__, "Failed to create \"%s\" (%d, %s).", + new_name, errno, strerror(errno)); + exit(1); + } + if (verbose_flag & TC_INFO) + tc_log_info(__FILE__, "Extracting t%d to \"%s\".", + ntstreams + 1, new_name); + do + handle_packet(stream, &pack, &page); + while (ogg_stream_packetout(&stream->instate, &pack) == 1); + } + add_stream(stream); + ntstreams++; + numstreams++; + } else { + tc_log_warn(__FILE__, "(%d) found new header of unknown/" + "unsupported type\n", numstreams + 1); + } + + } else { + tc_log_warn(__FILE__, "OGG stream %d is of an unknown type " + "(bad header?)\n", numstreams + 1); + } + } + } + + endofstream = 0; + while (!endofstream) { + sno = ogg_page_serialno(&page); + stream = find_stream(sno); + if (stream == NULL) { + if (verbose_flag & TC_DEBUG) + tc_log_warn(__FILE__, "Encountered packet for an unknown serial " + "%d !?\n", sno); + } else { + if (verbose_flag & TC_DEBUG) + tc_log_msg(__FILE__, "%c%d: NEW PAGE", stream->stype, stream->sno); + + ogg_stream_pagein(&stream->instate, &page); + while (ogg_stream_packetout(&stream->instate, &pack) == 1) + handle_packet(stream, &pack, &page); + } + + while (ogg_sync_pageseek(&sync, &page) <= 0) { + buf = ogg_sync_buffer(&sync, BLOCK_SIZE); + nread = read(fdin, buf, BLOCK_SIZE); + if (nread <= 0) { + stream = first; + while (stream != NULL) { + switch (stream->stype) { + case 'v': + close(stream->fd); + break; + case 't': + if (stream->fd > 0) + close(stream->fd); + break; + case 'a': + if ((stream->fd > 0) && !xraw) { + switch (stream->acodec) { + case ACVORBIS: + if (!stream->eos) { + pack.b_o_s = 0; + pack.e_o_s = 1; + pack.packet = NULL; + pack.bytes = 0; + pack.granulepos = stream->max_granulepos; + pack.packetno = stream->packetno; + ogg_stream_packetin(&stream->outstate, &pack); + } + flush_pages(stream, &pack); + ogg_stream_clear(&stream->outstate); + break; + } + close(stream->fd); + } else if (stream->fd > 0) + close(stream->fd); + break; + } + stream = stream->next; + } + if (verbose_flag & TC_INFO) + tc_log_info(__FILE__, "end of stream"); + endofstream = 1; + break; + } else + ogg_sync_wrote(&sync, nread); + } + } +} +#endif /* have OGG */ + +void extract_ogm (info_t *ipipe) +{ + // track + + no[NOTEXT] = 1; + no[NOAUDIO] = 0; + no[NOVIDEO] = 0; + xraw = 1; + + xvideo = tc_zalloc (16); xaudio = tc_zalloc (16); xtext = tc_zalloc (16); + + verbose_flag = ipipe->verbose; + + if (ipipe->select == TC_VIDEO) { + + no[NOAUDIO] = 1; + xvideo[0] = (unsigned char)(ipipe->track+1); + + } + + if (ipipe->select == TC_AUDIO) { + + no[NOVIDEO] = 1; + xaudio[0] = (unsigned char)(ipipe->track+1); + + // we need !xraw because no tool seems to be able to handle + // raw vorbis streams -- tibit + + if (ipipe->codec == TC_CODEC_VORBIS) { + xraw = 0; + } + + } + +#if (HAVE_OGG && HAVE_VORBIS) + process_ogm(ipipe->fd_in, ipipe->fd_out); +#else + tc_log_error(__FILE__, "No support for Ogg/Vorbis compiled in"); + import_exit(1); +#endif +} + + + +/* vim: sw=2 + */ |