diff options
Diffstat (limited to 'debian/transcode/transcode-1.1.7/import/ts_reader.c')
| -rw-r--r-- | debian/transcode/transcode-1.1.7/import/ts_reader.c | 564 |
1 files changed, 564 insertions, 0 deletions
diff --git a/debian/transcode/transcode-1.1.7/import/ts_reader.c b/debian/transcode/transcode-1.1.7/import/ts_reader.c new file mode 100644 index 00000000..e3b5ec93 --- /dev/null +++ b/debian/transcode/transcode-1.1.7/import/ts_reader.c @@ -0,0 +1,564 @@ +/* + * ts_reader.c + * + * Copyright (C) Thomas Oestreich - December 2002 + * + * based on extract_mpeg2.c + * Copyright (C) 2000-2002 Michel Lespinasse <walken@zoy.org> + * Copyright (C) 1999-2000 Aaron Holtzman <aholtzma@ess.engr.uvic.ca> + * + * This file is part of transcode, a video stream processing tool + * + * transcode 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, or (at your option) + * any later version. + * + * transcode 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 GNU Make; see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * + */ + +#include "transcode.h" +#include "tcinfo.h" + +#include <sys/mman.h> + +#include "ioaux.h" + +#ifdef HAVE_IO_H +#include <io.h> +#endif + +#define BUFFER_SIZE 188 +#define TS_PACK BUFFER_SIZE + +static uint8_t buffer[BUFFER_SIZE]; +static int demux_track = 0xe0; +static int demux_pid = 0; +//static int demux_pva = 0; + +static int fd_in; + + + +#define TRANS_ERROR 0x80 +#define PAY_START 0x40 +#define TRANS_PRIO 0x20 +#define PID_MASK_HI 0x1F +//flags +#define TRANS_SCRMBL1 0x80 +#define TRANS_SCRMBL2 0x40 +#define ADAPT_FIELD 0x20 +#define PAYLOAD 0x10 +#define COUNT_MASK 0x0F + +// adaptation flags +#define DISCON_IND 0x80 +#define RAND_ACC_IND 0x40 +#define ES_PRI_IND 0x20 +#define PCR_FLAG 0x10 +#define OPCR_FLAG 0x08 +#define SPLICE_FLAG 0x04 +#define TRANS_PRIV 0x02 +#define ADAP_EXT_FLAG 0x01 + +// adaptation extension flags +#define LTW_FLAG 0x80 +#define PIECE_RATE 0x40 +#define SEAM_SPLICE 0x20 + +typedef struct ts_packet_{ + uint8_t pid[2]; + uint8_t flags; + uint8_t count; + uint8_t data[184]; + uint8_t adapt_length; + uint8_t adapt_flags; + uint8_t pcr[6]; + uint8_t opcr[6]; + uint8_t splice_count; + uint8_t priv_dat_len; + uint8_t *priv_dat; + uint8_t adapt_ext_len; + uint8_t adapt_eflags; + uint8_t ltw[2]; + uint8_t piece_rate[3]; + uint8_t dts[5]; + int rest; + uint8_t stuffing; +} ts_packet; + +static void init_ts(ts_packet *p) +{ + p->pid[0] = 0; + p->pid[1] = 0; + p->flags = 0; + p->count = 0; + p->adapt_length = 0; + p->adapt_flags = 0; + p->splice_count = 0; + p->priv_dat_len = 0; + p->priv_dat = NULL; + p->adapt_ext_len = 0; + p->adapt_eflags = 0; + p->rest = 0; + p->stuffing = 0; +} + + +static uint16_t get_pid(uint8_t *pid) +{ + uint16_t pp = 0; + + pp = (pid[0] & PID_MASK_HI)<<8; + pp |= pid[1]; + + return pp; +} + +// tibit +void probe_ts(info_t *ipipe) +{ + int i; + int found=0, doit=1; + uint8_t sync = 0; + ts_packet p; + ssize_t size=0; + +#define MAX_PID 20 + int pid[MAX_PID]; + int npid=0; + + // look for a syncword + while (!found && doit) { + i = tc_pread(ipipe->fd_in, (uint8_t *)&sync, 1); + if (sync == 0x47) found = 1; + if (i == 0) doit = 0; + } + + for (i=0 ; i<MAX_PID; i++) + pid[i] = -1; + + while (size < ipipe->factor*1024*1024) { + + if((i=tc_pread(ipipe->fd_in, buffer, TS_PACK-1)) != TS_PACK-1) { + tc_log_info(__FILE__, "end of stream"); + return; + } + size += i; + + init_ts (&p); + ac_memcpy (&p, buffer, 3); + + found = 0; + for (i=0;i<npid;i++){ + if ( get_pid (buffer) == pid[i] ) + found = 1; + } + + if (!found) { + tc_log_info(__FILE__, "Found pid 0x%x", get_pid (buffer)); + pid[npid] = get_pid (buffer); + npid++; + if (npid >= MAX_PID) { + tc_log_warn(__FILE__, "Too many pids"); + return; + } + } + + + found = 0; doit = 1; + // read away syncword + + while (!found && doit) { + i = tc_pread(ipipe->fd_in, (uint8_t *)&sync, 1); + if (sync == 0x47) found = 1; + if (i == 0) doit = 0; + size += i; + } + if(!doit) { + tc_log_info(__FILE__, "end of stream"); + return; + } + } + + if (!npid) + tc_log_info(__FILE__, "No pids found"); +} + +#define DEMUX_PAYLOAD_START 1 + +static int demux (uint8_t * buf, uint8_t * end, int flags) +{ + static int mpeg1_skip_table[16] = { + 0, 0, 4, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 + }; + + /* + * the demuxer keeps some state between calls: + * if "state" = DEMUX_HEADER, then "head_buf" contains the first + * "bytes" bytes from some header. + * if "state" == DEMUX_DATA, then we need to copy "bytes" bytes + * of ES data before the next header. + * if "state" == DEMUX_SKIP, then we need to skip "bytes" bytes + * of data before the next header. + * + * NEEDBYTES makes sure we have the requested number of bytes for a + * header. If we dont, it copies what we have into head_buf and returns, + * so that when we come back with more data we finish decoding this header. + * + * DONEBYTES updates "buf" to point after the header we just parsed. + */ + +#define DEMUX_HEADER 0 +#define DEMUX_DATA 1 +#define DEMUX_SKIP 2 + static int state = DEMUX_SKIP; + static int state_bytes = 0; + static uint8_t head_buf[264]; + + uint8_t * header; + int bytes; + int len; + +#define NEEDBYTES(x) \ + do { \ + int missing; \ + \ + missing = (x) - bytes; \ + if (missing > 0) { \ + if (header == head_buf) { \ + if (missing <= end - buf) { \ + ac_memcpy (header + bytes, buf, missing); \ + buf += missing; \ + bytes = (x); \ + } else { \ + ac_memcpy (header + bytes, buf, end - buf); \ + state_bytes = bytes + end - buf; \ + return 0; \ + } \ + } else { \ + ac_memcpy (head_buf, header, bytes); \ + state = DEMUX_HEADER; \ + state_bytes = bytes; \ + return 0; \ + } \ + } \ + } while (0) + +#define DONEBYTES(x) \ + do { \ + if (header != head_buf) \ + buf = header + (x); \ + } while (0) + + if (flags & DEMUX_PAYLOAD_START) + goto payload_start; + switch (state) { + case DEMUX_HEADER: + if (state_bytes > 0) { + header = head_buf; + bytes = state_bytes; + goto continue_header; + } + break; + case DEMUX_DATA: + if (demux_pid || (state_bytes > end - buf)) { + fwrite (buf, end - buf, 1, stdout); + state_bytes -= end - buf; + return 0; + } + fwrite (buf, state_bytes, 1, stdout); + buf += state_bytes; + break; + case DEMUX_SKIP: + if (demux_pid || (state_bytes > end - buf)) { + state_bytes -= end - buf; + return 0; + } + buf += state_bytes; + break; + } + + while (1) { + if (demux_pid) { + state = DEMUX_SKIP; + return 0; + } + payload_start: + header = buf; + bytes = end - buf; + continue_header: + NEEDBYTES (4); + if (header[0] || header[1] || (header[2] != 1)) { + if (demux_pid) { + state = DEMUX_SKIP; + return 0; + } else if (header != head_buf) { + buf++; + goto payload_start; + } else { + header[0] = header[1]; + header[1] = header[2]; + header[2] = header[3]; + bytes = 3; + goto continue_header; + } + } + if (demux_pid) { + if ((header[3] >= 0xe0) && (header[3] <= 0xef)) + goto pes; + tc_log_error(__FILE__, "bad stream id %x", header[3]); + exit (1); + } + switch (header[3]) { + case 0xb9: /* program end code */ + /* DONEBYTES (4); */ + /* break; */ + return 1; + case 0xba: /* pack header */ + NEEDBYTES (12); + if ((header[4] & 0xc0) == 0x40) { /* mpeg2 */ + NEEDBYTES (14); + len = 14 + (header[13] & 7); + NEEDBYTES (len); + DONEBYTES (len); + /* header points to the mpeg2 pack header */ + } else if ((header[4] & 0xf0) == 0x20) { /* mpeg1 */ + DONEBYTES (12); + /* header points to the mpeg1 pack header */ + } else { + tc_log_error(__FILE__, "weird pack header"); + exit (1); + } + break; + default: + if (header[3] == demux_track) { + pes: + NEEDBYTES (7); + if ((header[6] & 0xc0) == 0x80) { /* mpeg2 */ + NEEDBYTES (9); + len = 9 + header[8]; + NEEDBYTES (len); + /* header points to the mpeg2 pes header */ + } else { /* mpeg1 */ + len = 7; + while ((header-1)[len] == 0xff) { + len++; + NEEDBYTES (len); + if (len > 23) { + tc_log_warn(__FILE__, "too much stuffing"); + break; + } + } + if (((header-1)[len] & 0xc0) == 0x40) { + len += 2; + NEEDBYTES (len); + } + len += mpeg1_skip_table[(header - 1)[len] >> 4]; + NEEDBYTES (len); + /* header points to the mpeg1 pes header */ + } + DONEBYTES (len); + bytes = 6 + (header[4] << 8) + header[5] - len; + if (demux_pid || (bytes > end - buf)) { + fwrite (buf, end - buf, 1, stdout); + state = DEMUX_DATA; + state_bytes = bytes - (end - buf); + return 0; + } else if (bytes <= 0) + continue; + fwrite (buf, bytes, 1, stdout); + buf += bytes; + } else if (header[3] < 0xb9) { + tc_log_info(__FILE__, + "looks like a video stream, not system stream"); + DONEBYTES (4); + } else { + NEEDBYTES (6); + DONEBYTES (6); + bytes = (header[4] << 8) + header[5]; + if (bytes > end - buf) { + state = DEMUX_SKIP; + state_bytes = bytes - (end - buf); + return 0; + } + buf += bytes; + } + } + } +} + +#if 0 +static void ps_loop (void) +{ + uint8_t * end; + + do { + end = buffer + fread (buffer, 1, BUFFER_SIZE, in_file); + if (demux (buffer, end, 0)) + break; /* hit program_end_code */ + } while (end == buffer + BUFFER_SIZE); +} + +static int pva_demux (uint8_t * buf, uint8_t * end) +{ + static int state = DEMUX_SKIP; + static int state_bytes = 0; + static uint8_t head_buf[12]; + + uint8_t * header; + int bytes; + int len; + + switch (state) { + case DEMUX_HEADER: + if (state_bytes > 0) { + header = head_buf; + bytes = state_bytes; + goto continue_header; + } + break; + case DEMUX_DATA: + if (state_bytes > end - buf) { + fwrite (buf, end - buf, 1, stdout); + state_bytes -= end - buf; + return 0; + } + fwrite (buf, state_bytes, 1, stdout); + buf += state_bytes; + break; + case DEMUX_SKIP: + if (state_bytes > end - buf) { + state_bytes -= end - buf; + return 0; + } + buf += state_bytes; + break; + } + + while (1) { + payload_start: + header = buf; + bytes = end - buf; + continue_header: + NEEDBYTES (2); + if (header[0] != 0x41 || header[1] != 0x56) { + if (header != head_buf) { + buf++; + goto payload_start; + } else { + header[0] = header[1]; + bytes = 1; + goto continue_header; + } + } + NEEDBYTES (8); + if (header[2] != 1) { + DONEBYTES (8); + bytes = (header[6] << 8) + header[7]; + if (bytes > end - buf) { + state = DEMUX_SKIP; + state_bytes = bytes - (end - buf); + return 0; + } + buf += bytes; + } else { + len = 8; + if (header[5] & 0x10) { + len = 12; + NEEDBYTES (len); + } + DONEBYTES (len); + bytes = (header[6] << 8) + header[7] + 8 - len; + if (bytes > end - buf) { + fwrite (buf, end - buf, 1, stdout); + state = DEMUX_DATA; + state_bytes = bytes - (end - buf); + return 0; + } else if (bytes > 0) { + fwrite (buf, bytes, 1, stdout); + buf += bytes; + } + } + } +} +#endif + +#if 0 +static void pva_loop (void) +{ + uint8_t * end; + + do { + end = buffer + fread (buffer, 1, BUFFER_SIZE, in_file); + pva_demux (buffer, end); + } while (end == buffer + BUFFER_SIZE); +} +#endif + +static void ts_loop (void) +{ +#define PACKETS (BUFFER_SIZE / 188) + uint8_t * buf; + uint8_t * data; + uint8_t * end; + int packets; + int i; + int pid; + + do { + + if((i=tc_pread(fd_in, buffer, TS_PACK)) != TS_PACK) { + tc_log_info(__FILE__, "end of stream"); + return; + } + + + packets=1; + + for (i = 0; i < packets; i++) { + buf = buffer + i * 188; + end = buf + 188; + if (buf[0] != 0x47) { + tc_log_error(__FILE__, "bad sync byte"); + exit (1); + } + pid = ((buf[1] << 8) + buf[2]) & 0x1fff; + if (pid != demux_pid) { + //tc_log_msg(__FILE__, "0x%x", pid); + continue; + } + data = buf + 4; + if (buf[3] & 0x20) { /* buf contains an adaptation field */ + data = buf + 5 + buf[4]; + if (data > end) + continue; + } + if (buf[3] & 0x10) + demux (data, end, (buf[1] & 0x40) ? DEMUX_PAYLOAD_START : 0); + } + } while (packets == PACKETS); +} + +int ts_read(int _fd_in, int fd_out, int _demux_pid) +{ + +#ifdef HAVE_IO_H + setmode (fileno (stdout), O_BINARY); +#endif + + demux_pid = _demux_pid; + fd_in = _fd_in; + + ts_loop(); + return 0; + +} |
