|
Packit |
64f477 |
/*
|
|
Packit |
64f477 |
* a52dec.c
|
|
Packit |
64f477 |
* Copyright (C) 2000-2002 Michel Lespinasse <walken@zoy.org>
|
|
Packit |
64f477 |
* Copyright (C) 1999-2000 Aaron Holtzman <aholtzma@ess.engr.uvic.ca>
|
|
Packit |
64f477 |
*
|
|
Packit |
64f477 |
* This file is part of a52dec, a free ATSC A-52 stream decoder.
|
|
Packit |
64f477 |
* See http://liba52.sourceforge.net/ for updates.
|
|
Packit |
64f477 |
*
|
|
Packit |
64f477 |
* a52dec is free software; you can redistribute it and/or modify
|
|
Packit |
64f477 |
* it under the terms of the GNU General Public License as published by
|
|
Packit |
64f477 |
* the Free Software Foundation; either version 2 of the License, or
|
|
Packit |
64f477 |
* (at your option) any later version.
|
|
Packit |
64f477 |
*
|
|
Packit |
64f477 |
* a52dec is distributed in the hope that it will be useful,
|
|
Packit |
64f477 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit |
64f477 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
Packit |
64f477 |
* GNU General Public License for more details.
|
|
Packit |
64f477 |
*
|
|
Packit |
64f477 |
* You should have received a copy of the GNU General Public License
|
|
Packit |
64f477 |
* along with this program; if not, write to the Free Software
|
|
Packit |
64f477 |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
Packit |
64f477 |
*/
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#include "config.h"
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#include <stdio.h>
|
|
Packit |
64f477 |
#include <stdlib.h>
|
|
Packit |
64f477 |
#include <string.h>
|
|
Packit |
64f477 |
#include <errno.h>
|
|
Packit |
64f477 |
#include <getopt.h>
|
|
Packit |
64f477 |
#include <math.h>
|
|
Packit |
64f477 |
#include <signal.h>
|
|
Packit |
64f477 |
#ifdef HAVE_IO_H
|
|
Packit |
64f477 |
#include <fcntl.h>
|
|
Packit |
64f477 |
#include <io.h>
|
|
Packit |
64f477 |
#endif
|
|
Packit |
64f477 |
#include <inttypes.h>
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#include "a52.h"
|
|
Packit |
64f477 |
#include "audio_out.h"
|
|
Packit |
64f477 |
#include "mm_accel.h"
|
|
Packit |
64f477 |
#include "gettimeofday.h"
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#define BUFFER_SIZE 4096
|
|
Packit |
64f477 |
static uint8_t buffer[BUFFER_SIZE];
|
|
Packit |
64f477 |
static FILE * in_file;
|
|
Packit |
64f477 |
static int demux_track = 0;
|
|
Packit |
64f477 |
static int demux_pid = 0;
|
|
Packit |
64f477 |
static int disable_accel = 0;
|
|
Packit |
64f477 |
static int disable_dynrng = 0;
|
|
Packit |
64f477 |
static int disable_adjust = 0;
|
|
Packit |
64f477 |
static sample_t gain = 1;
|
|
Packit |
64f477 |
static ao_open_t * output_open = NULL;
|
|
Packit |
64f477 |
static ao_instance_t * output;
|
|
Packit |
64f477 |
static a52_state_t * state;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#ifdef HAVE_GETTIMEOFDAY
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
static void print_fps (int final);
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
static RETSIGTYPE signal_handler (int sig)
|
|
Packit |
64f477 |
{
|
|
Packit |
64f477 |
print_fps (1);
|
|
Packit |
64f477 |
signal (sig, SIG_DFL);
|
|
Packit |
64f477 |
raise (sig);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
static void print_fps (int final)
|
|
Packit |
64f477 |
{
|
|
Packit |
64f477 |
static uint32_t frame_counter = 0;
|
|
Packit |
64f477 |
static struct timeval tv_beg, tv_start;
|
|
Packit |
64f477 |
static int total_elapsed;
|
|
Packit |
64f477 |
static int last_count = 0;
|
|
Packit |
64f477 |
struct timeval tv_end;
|
|
Packit |
64f477 |
float fps, tfps;
|
|
Packit |
64f477 |
int frames, elapsed;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
gettimeofday (&tv_end, NULL);
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
if (!frame_counter) {
|
|
Packit |
64f477 |
tv_start = tv_beg = tv_end;
|
|
Packit |
64f477 |
signal (SIGINT, signal_handler);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
elapsed = (tv_end.tv_sec - tv_beg.tv_sec) * 100 +
|
|
Packit |
64f477 |
(tv_end.tv_usec - tv_beg.tv_usec) / 10000;
|
|
Packit |
64f477 |
total_elapsed = (tv_end.tv_sec - tv_start.tv_sec) * 100 +
|
|
Packit |
64f477 |
(tv_end.tv_usec - tv_start.tv_usec) / 10000;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
if (final) {
|
|
Packit |
64f477 |
if (total_elapsed)
|
|
Packit |
64f477 |
tfps = frame_counter * 100.0 / total_elapsed;
|
|
Packit |
64f477 |
else
|
|
Packit |
64f477 |
tfps = 0;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
fprintf (stderr,"\n%d frames decoded in %.2f seconds (%.2f fps)\n",
|
|
Packit |
64f477 |
frame_counter, total_elapsed / 100.0, tfps);
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
return;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
frame_counter++;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
if (elapsed < 50) /* only display every 0.50 seconds */
|
|
Packit |
64f477 |
return;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
tv_beg = tv_end;
|
|
Packit |
64f477 |
frames = frame_counter - last_count;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
fps = frames * 100.0 / elapsed;
|
|
Packit |
64f477 |
tfps = frame_counter * 100.0 / total_elapsed;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
fprintf (stderr, "%d frames in %.2f sec (%.2f fps), "
|
|
Packit |
64f477 |
"%d last %.2f sec (%.2f fps)\033[K\r", frame_counter,
|
|
Packit |
64f477 |
total_elapsed / 100.0, tfps, frames, elapsed / 100.0, fps);
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
last_count = frame_counter;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#else /* !HAVE_GETTIMEOFDAY */
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
static void print_fps (int final)
|
|
Packit |
64f477 |
{
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#endif
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
static void print_usage (char ** argv)
|
|
Packit |
64f477 |
{
|
|
Packit |
64f477 |
int i;
|
|
Packit |
64f477 |
ao_driver_t * drivers;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
fprintf (stderr, "usage: "
|
|
Packit |
64f477 |
"%s [-o <mode>] [-s [<track>]] [-t <pid>] [-c] [-r] [-a] \\\n"
|
|
Packit |
64f477 |
"\t\t[-g <gain>] <file>\n"
|
|
Packit |
64f477 |
"\t-s\tuse program stream demultiplexer, track 0-7 or 0x80-0x87\n"
|
|
Packit |
64f477 |
"\t-t\tuse transport stream demultiplexer, pid 0x10-0x1ffe\n"
|
|
Packit |
64f477 |
"\t-c\tuse c implementation, disables all accelerations\n"
|
|
Packit |
64f477 |
"\t-r\tdisable dynamic range compression\n"
|
|
Packit |
64f477 |
"\t-a\tdisable level adjustment based on output mode\n"
|
|
Packit |
64f477 |
"\t-g\tadd specified gain in decibels, -96.0 to +96.0\n"
|
|
Packit |
64f477 |
"\t-o\taudio output mode\n", argv[0]);
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
drivers = ao_drivers ();
|
|
Packit |
64f477 |
for (i = 0; drivers[i].name; i++)
|
|
Packit |
64f477 |
fprintf (stderr, "\t\t\t%s\n", drivers[i].name);
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
exit (1);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
static void handle_args (int argc, char ** argv)
|
|
Packit |
64f477 |
{
|
|
Packit |
64f477 |
int c;
|
|
Packit |
64f477 |
ao_driver_t * drivers;
|
|
Packit |
64f477 |
int i;
|
|
Packit |
64f477 |
char * s;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
drivers = ao_drivers ();
|
|
Packit |
64f477 |
while ((c = getopt (argc, argv, "s::t:crag:o:")) != -1)
|
|
Packit |
64f477 |
switch (c) {
|
|
Packit |
64f477 |
case 'o':
|
|
Packit |
64f477 |
for (i = 0; drivers[i].name != NULL; i++)
|
|
Packit |
64f477 |
if (strcmp (drivers[i].name, optarg) == 0)
|
|
Packit |
64f477 |
output_open = drivers[i].open;
|
|
Packit |
64f477 |
if (output_open == NULL) {
|
|
Packit |
64f477 |
fprintf (stderr, "Invalid video driver: %s\n", optarg);
|
|
Packit |
64f477 |
print_usage (argv);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
case 's':
|
|
Packit |
64f477 |
demux_track = 0x80;
|
|
Packit |
64f477 |
if (optarg != NULL) {
|
|
Packit |
64f477 |
demux_track = strtol (optarg, &s, 16);
|
|
Packit |
64f477 |
if (demux_track < 0x80)
|
|
Packit |
64f477 |
demux_track += 0x80;
|
|
Packit |
64f477 |
if ((demux_track < 0x80) || (demux_track > 0x87) || (*s)) {
|
|
Packit |
64f477 |
fprintf (stderr, "Invalid track number: %s\n", optarg);
|
|
Packit |
64f477 |
print_usage (argv);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
case 't':
|
|
Packit |
64f477 |
demux_pid = strtol (optarg, &s, 16);
|
|
Packit |
64f477 |
if ((demux_pid < 0x10) || (demux_pid > 0x1ffe) || (*s)) {
|
|
Packit |
64f477 |
fprintf (stderr, "Invalid pid: %s\n", optarg);
|
|
Packit |
64f477 |
print_usage (argv);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
case 'c':
|
|
Packit |
64f477 |
disable_accel = 1;
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
case 'r':
|
|
Packit |
64f477 |
disable_dynrng = 1;
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
case 'a':
|
|
Packit |
64f477 |
disable_adjust = 1;
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
case 'g':
|
|
Packit |
64f477 |
gain = strtod (optarg, &s);
|
|
Packit |
64f477 |
if ((gain < -96) || (gain > 96) || (*s)) {
|
|
Packit |
64f477 |
fprintf (stderr, "Invalid gain: %s\n", optarg);
|
|
Packit |
64f477 |
print_usage (argv);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
gain = pow (2, gain / 6);
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
default:
|
|
Packit |
64f477 |
print_usage (argv);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
/* -o not specified, use a default driver */
|
|
Packit |
64f477 |
if (output_open == NULL)
|
|
Packit |
64f477 |
output_open = drivers[0].open;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
if (optind < argc) {
|
|
Packit |
64f477 |
in_file = fopen (argv[optind], "rb");
|
|
Packit |
64f477 |
if (!in_file) {
|
|
Packit |
64f477 |
fprintf (stderr, "%s - could not open file %s\n", strerror (errno),
|
|
Packit |
64f477 |
argv[optind]);
|
|
Packit |
64f477 |
exit (1);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
} else
|
|
Packit |
64f477 |
in_file = stdin;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
void a52_decode_data (uint8_t * start, uint8_t * end)
|
|
Packit |
64f477 |
{
|
|
Packit |
64f477 |
static uint8_t buf[3840];
|
|
Packit |
64f477 |
static uint8_t * bufptr = buf;
|
|
Packit |
64f477 |
static uint8_t * bufpos = buf + 7;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
/*
|
|
Packit |
64f477 |
* sample_rate and flags are static because this routine could
|
|
Packit |
64f477 |
* exit between the a52_syncinfo() and the ao_setup(), and we want
|
|
Packit |
64f477 |
* to have the same values when we get back !
|
|
Packit |
64f477 |
*/
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
static int sample_rate;
|
|
Packit |
64f477 |
static int flags;
|
|
Packit |
64f477 |
int bit_rate;
|
|
Packit |
64f477 |
int len;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
while (1) {
|
|
Packit |
64f477 |
len = end - start;
|
|
Packit |
64f477 |
if (!len)
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
if (len > bufpos - bufptr)
|
|
Packit |
64f477 |
len = bufpos - bufptr;
|
|
Packit |
64f477 |
memcpy (bufptr, start, len);
|
|
Packit |
64f477 |
bufptr += len;
|
|
Packit |
64f477 |
start += len;
|
|
Packit |
64f477 |
if (bufptr == bufpos) {
|
|
Packit |
64f477 |
if (bufpos == buf + 7) {
|
|
Packit |
64f477 |
int length;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
length = a52_syncinfo (buf, &flags, &sample_rate, &bit_rate);
|
|
Packit |
64f477 |
if (!length) {
|
|
Packit |
64f477 |
fprintf (stderr, "skip\n");
|
|
Packit |
64f477 |
for (bufptr = buf; bufptr < buf + 6; bufptr++)
|
|
Packit |
64f477 |
bufptr[0] = bufptr[1];
|
|
Packit |
64f477 |
continue;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
bufpos = buf + length;
|
|
Packit |
64f477 |
} else {
|
|
Packit |
64f477 |
sample_t level, bias;
|
|
Packit |
64f477 |
int i;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
if (ao_setup (output, sample_rate, &flags, &level, &bias))
|
|
Packit |
64f477 |
goto error;
|
|
Packit |
64f477 |
if (!disable_adjust)
|
|
Packit |
64f477 |
flags |= A52_ADJUST_LEVEL;
|
|
Packit |
64f477 |
level *= gain;
|
|
Packit |
64f477 |
if (a52_frame (state, buf, &flags, &level, bias))
|
|
Packit |
64f477 |
goto error;
|
|
Packit |
64f477 |
if (disable_dynrng)
|
|
Packit |
64f477 |
a52_dynrng (state, NULL, NULL);
|
|
Packit |
64f477 |
for (i = 0; i < 6; i++) {
|
|
Packit |
64f477 |
if (a52_block (state))
|
|
Packit |
64f477 |
goto error;
|
|
Packit |
64f477 |
if (ao_play (output, flags, a52_samples (state)))
|
|
Packit |
64f477 |
goto error;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
bufptr = buf;
|
|
Packit |
64f477 |
bufpos = buf + 7;
|
|
Packit |
64f477 |
print_fps (0);
|
|
Packit |
64f477 |
continue;
|
|
Packit |
64f477 |
error:
|
|
Packit |
64f477 |
fprintf (stderr, "error\n");
|
|
Packit |
64f477 |
bufptr = buf;
|
|
Packit |
64f477 |
bufpos = buf + 7;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#define DEMUX_PAYLOAD_START 1
|
|
Packit |
64f477 |
static int demux (uint8_t * buf, uint8_t * end, int flags)
|
|
Packit |
64f477 |
{
|
|
Packit |
64f477 |
static int mpeg1_skip_table[16] = {
|
|
Packit |
64f477 |
0, 0, 4, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
|
|
Packit |
64f477 |
};
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
/*
|
|
Packit |
64f477 |
* the demuxer keeps some state between calls:
|
|
Packit |
64f477 |
* if "state" = DEMUX_HEADER, then "head_buf" contains the first
|
|
Packit |
64f477 |
* "bytes" bytes from some header.
|
|
Packit |
64f477 |
* if "state" == DEMUX_DATA, then we need to copy "bytes" bytes
|
|
Packit |
64f477 |
* of ES data before the next header.
|
|
Packit |
64f477 |
* if "state" == DEMUX_SKIP, then we need to skip "bytes" bytes
|
|
Packit |
64f477 |
* of data before the next header.
|
|
Packit |
64f477 |
*
|
|
Packit |
64f477 |
* NEEDBYTES makes sure we have the requested number of bytes for a
|
|
Packit |
64f477 |
* header. If we dont, it copies what we have into head_buf and returns,
|
|
Packit |
64f477 |
* so that when we come back with more data we finish decoding this header.
|
|
Packit |
64f477 |
*
|
|
Packit |
64f477 |
* DONEBYTES updates "buf" to point after the header we just parsed.
|
|
Packit |
64f477 |
*/
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#define DEMUX_HEADER 0
|
|
Packit |
64f477 |
#define DEMUX_DATA 1
|
|
Packit |
64f477 |
#define DEMUX_SKIP 2
|
|
Packit |
64f477 |
static int state = DEMUX_SKIP;
|
|
Packit |
64f477 |
static int state_bytes = 0;
|
|
Packit |
64f477 |
static uint8_t head_buf[268];
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
uint8_t * header;
|
|
Packit |
64f477 |
int bytes;
|
|
Packit |
64f477 |
int len;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#define NEEDBYTES(x) \
|
|
Packit |
64f477 |
do { \
|
|
Packit |
64f477 |
int missing; \
|
|
Packit |
64f477 |
\
|
|
Packit |
64f477 |
missing = (x) - bytes; \
|
|
Packit |
64f477 |
if (missing > 0) { \
|
|
Packit |
64f477 |
if (header == head_buf) { \
|
|
Packit |
64f477 |
if (missing <= end - buf) { \
|
|
Packit |
64f477 |
memcpy (header + bytes, buf, missing); \
|
|
Packit |
64f477 |
buf += missing; \
|
|
Packit |
64f477 |
bytes = (x); \
|
|
Packit |
64f477 |
} else { \
|
|
Packit |
64f477 |
memcpy (header + bytes, buf, end - buf); \
|
|
Packit |
64f477 |
state_bytes = bytes + end - buf; \
|
|
Packit |
64f477 |
return 0; \
|
|
Packit |
64f477 |
} \
|
|
Packit |
64f477 |
} else { \
|
|
Packit |
64f477 |
memcpy (head_buf, header, bytes); \
|
|
Packit |
64f477 |
state = DEMUX_HEADER; \
|
|
Packit |
64f477 |
state_bytes = bytes; \
|
|
Packit |
64f477 |
return 0; \
|
|
Packit |
64f477 |
} \
|
|
Packit |
64f477 |
} \
|
|
Packit |
64f477 |
} while (0)
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#define DONEBYTES(x) \
|
|
Packit |
64f477 |
do { \
|
|
Packit |
64f477 |
if (header != head_buf) \
|
|
Packit |
64f477 |
buf = header + (x); \
|
|
Packit |
64f477 |
} while (0)
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
if (flags & DEMUX_PAYLOAD_START)
|
|
Packit |
64f477 |
goto payload_start;
|
|
Packit |
64f477 |
switch (state) {
|
|
Packit |
64f477 |
case DEMUX_HEADER:
|
|
Packit |
64f477 |
if (state_bytes > 0) {
|
|
Packit |
64f477 |
header = head_buf;
|
|
Packit |
64f477 |
bytes = state_bytes;
|
|
Packit |
64f477 |
goto continue_header;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
case DEMUX_DATA:
|
|
Packit |
64f477 |
if (demux_pid || (state_bytes > end - buf)) {
|
|
Packit |
64f477 |
a52_decode_data (buf, end);
|
|
Packit |
64f477 |
state_bytes -= end - buf;
|
|
Packit |
64f477 |
return 0;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
a52_decode_data (buf, buf + state_bytes);
|
|
Packit |
64f477 |
buf += state_bytes;
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
case DEMUX_SKIP:
|
|
Packit |
64f477 |
if (demux_pid || (state_bytes > end - buf)) {
|
|
Packit |
64f477 |
state_bytes -= end - buf;
|
|
Packit |
64f477 |
return 0;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
buf += state_bytes;
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
while (1) {
|
|
Packit |
64f477 |
if (demux_pid) {
|
|
Packit |
64f477 |
state = DEMUX_SKIP;
|
|
Packit |
64f477 |
return 0;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
payload_start:
|
|
Packit |
64f477 |
header = buf;
|
|
Packit |
64f477 |
bytes = end - buf;
|
|
Packit |
64f477 |
continue_header:
|
|
Packit |
64f477 |
NEEDBYTES (4);
|
|
Packit |
64f477 |
if (header[0] || header[1] || (header[2] != 1)) {
|
|
Packit |
64f477 |
if (demux_pid) {
|
|
Packit |
64f477 |
state = DEMUX_SKIP;
|
|
Packit |
64f477 |
return 0;
|
|
Packit |
64f477 |
} else if (header != head_buf) {
|
|
Packit |
64f477 |
buf++;
|
|
Packit |
64f477 |
goto payload_start;
|
|
Packit |
64f477 |
} else {
|
|
Packit |
64f477 |
header[0] = header[1];
|
|
Packit |
64f477 |
header[1] = header[2];
|
|
Packit |
64f477 |
header[2] = header[3];
|
|
Packit |
64f477 |
bytes = 3;
|
|
Packit |
64f477 |
goto continue_header;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
if (demux_pid && (header[3] != 0xbd)) {
|
|
Packit |
64f477 |
fprintf (stderr, "bad stream id %x\n", header[3]);
|
|
Packit |
64f477 |
exit (1);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
switch (header[3]) {
|
|
Packit |
64f477 |
case 0xb9: /* program end code */
|
|
Packit |
64f477 |
/* DONEBYTES (4); */
|
|
Packit |
64f477 |
/* break; */
|
|
Packit |
64f477 |
return 1;
|
|
Packit |
64f477 |
case 0xba: /* pack header */
|
|
Packit |
64f477 |
NEEDBYTES (12);
|
|
Packit |
64f477 |
if ((header[4] & 0xc0) == 0x40) { /* mpeg2 */
|
|
Packit |
64f477 |
NEEDBYTES (14);
|
|
Packit |
64f477 |
len = 14 + (header[13] & 7);
|
|
Packit |
64f477 |
NEEDBYTES (len);
|
|
Packit |
64f477 |
DONEBYTES (len);
|
|
Packit |
64f477 |
/* header points to the mpeg2 pack header */
|
|
Packit |
64f477 |
} else if ((header[4] & 0xf0) == 0x20) { /* mpeg1 */
|
|
Packit |
64f477 |
DONEBYTES (12);
|
|
Packit |
64f477 |
/* header points to the mpeg1 pack header */
|
|
Packit |
64f477 |
} else {
|
|
Packit |
64f477 |
fprintf (stderr, "weird pack header\n");
|
|
Packit |
64f477 |
exit (1);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
case 0xbd: /* private stream 1 */
|
|
Packit |
64f477 |
NEEDBYTES (7);
|
|
Packit |
64f477 |
if ((header[6] & 0xc0) == 0x80) { /* mpeg2 */
|
|
Packit |
64f477 |
NEEDBYTES (9);
|
|
Packit |
64f477 |
len = 10 + header[8];
|
|
Packit |
64f477 |
NEEDBYTES (len);
|
|
Packit |
64f477 |
/* header points to the mpeg2 pes header */
|
|
Packit |
64f477 |
} else { /* mpeg1 */
|
|
Packit |
64f477 |
len = 7;
|
|
Packit |
64f477 |
while ((header-1)[len] == 0xff) {
|
|
Packit |
64f477 |
len++;
|
|
Packit |
64f477 |
NEEDBYTES (len);
|
|
Packit |
64f477 |
if (len == 23) {
|
|
Packit |
64f477 |
fprintf (stderr, "too much stuffing\n");
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
if (((header-1)[len] & 0xc0) == 0x40) {
|
|
Packit |
64f477 |
len += 2;
|
|
Packit |
64f477 |
NEEDBYTES (len);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
len += mpeg1_skip_table[(header - 1)[len] >> 4] + 1;
|
|
Packit |
64f477 |
NEEDBYTES (len);
|
|
Packit |
64f477 |
/* header points to the mpeg1 pes header */
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
if ((!demux_pid) && ((header-1)[len] != demux_track)) {
|
|
Packit |
64f477 |
DONEBYTES (len);
|
|
Packit |
64f477 |
bytes = 6 + (header[4] << 8) + header[5] - len;
|
|
Packit |
64f477 |
if (bytes <= 0)
|
|
Packit |
64f477 |
continue;
|
|
Packit |
64f477 |
goto skip;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
len += 3;
|
|
Packit |
64f477 |
NEEDBYTES (len);
|
|
Packit |
64f477 |
DONEBYTES (len);
|
|
Packit |
64f477 |
bytes = 6 + (header[4] << 8) + header[5] - len;
|
|
Packit |
64f477 |
if (demux_pid || (bytes > end - buf)) {
|
|
Packit |
64f477 |
a52_decode_data (buf, end);
|
|
Packit |
64f477 |
state = DEMUX_DATA;
|
|
Packit |
64f477 |
state_bytes = bytes - (end - buf);
|
|
Packit |
64f477 |
return 0;
|
|
Packit |
64f477 |
} else if (bytes <= 0)
|
|
Packit |
64f477 |
continue;
|
|
Packit |
64f477 |
a52_decode_data (buf, buf + bytes);
|
|
Packit |
64f477 |
buf += bytes;
|
|
Packit |
64f477 |
break;
|
|
Packit |
64f477 |
default:
|
|
Packit |
64f477 |
if (header[3] < 0xb9) {
|
|
Packit |
64f477 |
fprintf (stderr,
|
|
Packit |
64f477 |
"looks like a video stream, not system stream\n");
|
|
Packit |
64f477 |
exit (1);
|
|
Packit |
64f477 |
} else {
|
|
Packit |
64f477 |
NEEDBYTES (6);
|
|
Packit |
64f477 |
DONEBYTES (6);
|
|
Packit |
64f477 |
bytes = (header[4] << 8) + header[5];
|
|
Packit |
64f477 |
skip:
|
|
Packit |
64f477 |
if (bytes > end - buf) {
|
|
Packit |
64f477 |
state = DEMUX_SKIP;
|
|
Packit |
64f477 |
state_bytes = bytes - (end - buf);
|
|
Packit |
64f477 |
return 0;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
buf += bytes;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
static void ps_loop (void)
|
|
Packit |
64f477 |
{
|
|
Packit |
64f477 |
uint8_t * end;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
do {
|
|
Packit |
64f477 |
end = buffer + fread (buffer, 1, BUFFER_SIZE, in_file);
|
|
Packit |
64f477 |
if (demux (buffer, end, 0))
|
|
Packit |
64f477 |
break; /* hit program_end_code */
|
|
Packit |
64f477 |
} while (end == buffer + BUFFER_SIZE);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
static void ts_loop (void)
|
|
Packit |
64f477 |
{
|
|
Packit |
64f477 |
#define PACKETS (BUFFER_SIZE / 188)
|
|
Packit |
64f477 |
uint8_t * buf;
|
|
Packit |
64f477 |
uint8_t * data;
|
|
Packit |
64f477 |
uint8_t * end;
|
|
Packit |
64f477 |
int packets;
|
|
Packit |
64f477 |
int i;
|
|
Packit |
64f477 |
int pid;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
do {
|
|
Packit |
64f477 |
packets = fread (buffer, 188, PACKETS, in_file);
|
|
Packit |
64f477 |
for (i = 0; i < packets; i++) {
|
|
Packit |
64f477 |
buf = buffer + i * 188;
|
|
Packit |
64f477 |
end = buf + 188;
|
|
Packit |
64f477 |
if (buf[0] != 0x47) {
|
|
Packit |
64f477 |
fprintf (stderr, "bad sync byte\n");
|
|
Packit |
64f477 |
exit (1);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
pid = ((buf[1] << 8) + buf[2]) & 0x1fff;
|
|
Packit |
64f477 |
if (pid != demux_pid)
|
|
Packit |
64f477 |
continue;
|
|
Packit |
64f477 |
data = buf + 4;
|
|
Packit |
64f477 |
if (buf[3] & 0x20) { /* buf contains an adaptation field */
|
|
Packit |
64f477 |
data = buf + 5 + buf[4];
|
|
Packit |
64f477 |
if (data > end)
|
|
Packit |
64f477 |
continue;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
if (buf[3] & 0x10)
|
|
Packit |
64f477 |
demux (data, end, (buf[1] & 0x40) ? DEMUX_PAYLOAD_START : 0);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
} while (packets == PACKETS);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
static void es_loop (void)
|
|
Packit |
64f477 |
{
|
|
Packit |
64f477 |
int size;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
do {
|
|
Packit |
64f477 |
size = fread (buffer, 1, BUFFER_SIZE, in_file);
|
|
Packit |
64f477 |
a52_decode_data (buffer, buffer + size);
|
|
Packit |
64f477 |
} while (size == BUFFER_SIZE);
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
int main (int argc, char ** argv)
|
|
Packit |
64f477 |
{
|
|
Packit |
64f477 |
uint32_t accel;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
#ifdef HAVE_IO_H
|
|
Packit |
64f477 |
setmode (fileno (stdout), O_BINARY);
|
|
Packit |
64f477 |
#endif
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
fprintf (stderr, PACKAGE"-"VERSION
|
|
Packit |
64f477 |
" - by Michel Lespinasse <walken@zoy.org> and Aaron Holtzman\n");
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
handle_args (argc, argv);
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
accel = disable_accel ? 0 : MM_ACCEL_DJBFFT;
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
output = ao_open (output_open);
|
|
Packit |
64f477 |
if (output == NULL) {
|
|
Packit |
64f477 |
fprintf (stderr, "Can not open output\n");
|
|
Packit |
64f477 |
return 1;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
state = a52_init (accel);
|
|
Packit |
64f477 |
if (state == NULL) {
|
|
Packit |
64f477 |
fprintf (stderr, "A52 init failed\n");
|
|
Packit |
64f477 |
return 1;
|
|
Packit |
64f477 |
}
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
if (demux_pid)
|
|
Packit |
64f477 |
ts_loop ();
|
|
Packit |
64f477 |
else if (demux_track)
|
|
Packit |
64f477 |
ps_loop ();
|
|
Packit |
64f477 |
else
|
|
Packit |
64f477 |
es_loop ();
|
|
Packit |
64f477 |
|
|
Packit |
64f477 |
a52_free (state);
|
|
Packit |
64f477 |
print_fps (1);
|
|
Packit |
64f477 |
ao_close (output);
|
|
Packit |
64f477 |
return 0;
|
|
Packit |
64f477 |
}
|