diff options
Diffstat (limited to 'gst/rtp/gstrtph263pay.c')
| -rw-r--r-- | gst/rtp/gstrtph263pay.c | 1809 |
1 files changed, 1629 insertions, 180 deletions
diff --git a/gst/rtp/gstrtph263pay.c b/gst/rtp/gstrtph263pay.c index 32773e6f5..3b8dad642 100644 --- a/gst/rtp/gstrtph263pay.c +++ b/gst/rtp/gstrtph263pay.c @@ -1,5 +1,7 @@ /* GStreamer * Copyright (C) <2005> Wim Taymans <wim.taymans@gmail.com> + * Copyright (C) <2008> Dejan Sakelsak <dejan.sakelsak@marand.si> + * Copyright (C) <2009> Janin Kolenc <janin.kolenc@marand.si> * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Library General Public @@ -15,6 +17,7 @@ * License along with this library; if not, write to the * Free Software Foundation, Inc., 59 Temple Place - Suite 330, * Boston, MA 02111-1307, USA. + * */ #ifdef HAVE_CONFIG_H @@ -22,133 +25,333 @@ #endif #include <string.h> - +#include <math.h> #include <gst/rtp/gstrtpbuffer.h> #include "gstrtph263pay.h" -#define GST_RFC2190A_HEADER_LEN 4 - -typedef struct _GstRFC2190AHeader -{ -#if G_BYTE_ORDER == G_LITTLE_ENDIAN - unsigned int ebit:3; /* End position */ - unsigned int sbit:3; /* Start position */ - unsigned int p:1; /* PB-frames mode */ - unsigned int f:1; /* flag bit */ - - unsigned int r1:1; /* Reserved */ - unsigned int a:1; /* Advanced Prediction */ - unsigned int s:1; /* syntax based arithmetic coding */ - unsigned int u:1; /* Unrestricted motion vector */ - unsigned int i:1; /* Picture coding type */ - unsigned int src:3; /* Source format */ - - unsigned int trb:3; /* Temporal ref for B frame */ - unsigned int dbq:2; /* Differential Quantisation parameter */ - unsigned int r2:3; /* Reserved */ -#elif G_BYTE_ORDER == G_BIG_ENDIAN - unsigned int f:1; /* flag bit */ - unsigned int p:1; /* PB-frames mode */ - unsigned int sbit:3; /* Start position */ - unsigned int ebit:3; /* End position */ - - unsigned int src:3; /* Source format */ - unsigned int i:1; /* Picture coding type */ - unsigned int u:1; /* Unrestricted motion vector */ - unsigned int s:1; /* syntax based arithmetic coding */ - unsigned int a:1; /* Advanced Prediction */ - unsigned int r1:1; /* Reserved */ - - unsigned int r2:3; /* Reserved */ - unsigned int dbq:2; /* Differential Quantisation parameter */ - unsigned int trb:3; /* Temporal ref for B frame */ -#else -#error "G_BYTE_ORDER should be big or little endian." -#endif - unsigned int tr:8; /* Temporal ref for P frame */ -} GstRFC2190AHeader; - - -#define GST_RFC2190A_HEADER_F(buf) (((GstRFC2190AHeader *)(buf))->f) -#define GST_RFC2190A_HEADER_P(buf) (((GstRFC2190AHeader *)(buf))->p) -#define GST_RFC2190A_HEADER_SBIT(buf) (((GstRFC2190AHeader *)(buf))->sbit) -#define GST_RFC2190A_HEADER_EBIT(buf) (((GstRFC2190AHeader *)(buf))->ebit) -#define GST_RFC2190A_HEADER_SRC(buf) (((GstRFC2190AHeader *)(buf))->src) -#define GST_RFC2190A_HEADER_I(buf) (((GstRFC2190AHeader *)(buf))->i) -#define GST_RFC2190A_HEADER_U(buf) (((GstRFC2190AHeader *)(buf))->u) -#define GST_RFC2190A_HEADER_S(buf) (((GstRFC2190AHeader *)(buf))->s) -#define GST_RFC2190A_HEADER_A(buf) (((GstRFC2190AHeader *)(buf))->a) -#define GST_RFC2190A_HEADER_R1(buf) (((GstRFC2190AHeader *)(buf))->r1) -#define GST_RFC2190A_HEADER_R2(buf) (((GstRFC2190AHeader *)(buf))->r2) -#define GST_RFC2190A_HEADER_DBQ(buf) (((GstRFC2190AHeader *)(buf))->dbq) -#define GST_RFC2190A_HEADER_TRB(buf) (((GstRFC2190AHeader *)(buf))->trb) -#define GST_RFC2190A_HEADER_TR(buf) (((GstRFC2190AHeader *)(buf))->tr) - - -typedef struct _GstH263PictureLayer -{ -#if G_BYTE_ORDER == G_LITTLE_ENDIAN - unsigned int psc1:16; - - unsigned int tr1:2; - unsigned int psc2:6; - - unsigned int ptype_263:1; - unsigned int ptype_start:1; - unsigned int tr2:6; - - unsigned int ptype_umvmode:1; - unsigned int ptype_pictype:1; - unsigned int ptype_srcformat:3; - unsigned int ptype_freeze:1; - unsigned int ptype_camera:1; - unsigned int ptype_split:1; - - unsigned int chaff:5; - unsigned int ptype_pbmode:1; - unsigned int ptype_apmode:1; - unsigned int ptype_sacmode:1; -#elif G_BYTE_ORDER == G_BIG_ENDIAN - unsigned int psc1:16; - - unsigned int psc2:6; - unsigned int tr1:2; - - unsigned int tr2:6; - unsigned int ptype_start:1; - unsigned int ptype_263:1; - - unsigned int ptype_split:1; - unsigned int ptype_camera:1; - unsigned int ptype_freeze:1; - unsigned int ptype_srcformat:3; - unsigned int ptype_pictype:1; - unsigned int ptype_umvmode:1; - - unsigned int ptype_sacmode:1; - unsigned int ptype_apmode:1; - unsigned int ptype_pbmode:1; - unsigned int chaff:5; -#else -#error "G_BYTE_ORDER should be big or little endian." -#endif -} GstH263PictureLayer; - -#define GST_H263_PICTURELAYER_PLSRC(buf) (((GstH263PictureLayer *)(buf))->ptype_srcformat) -#define GST_H263_PICTURELAYER_PLTYPE(buf) (((GstH263PictureLayer *)(buf))->ptype_pictype) -#define GST_H263_PICTURELAYER_PLUMV(buf) (((GstH263PictureLayer *)(buf))->ptype_umvmode) -#define GST_H263_PICTURELAYER_PLSAC(buf) (((GstH263PictureLayer *)(buf))->ptype_sacmode) -#define GST_H263_PICTURELAYER_PLAP(buf) (((GstH263PictureLayer *)(buf))->ptype_apmode) - +typedef enum +{ + GST_H263_FRAME_TYPE_I = 0, + GST_H263_FRAME_TYPE_P = 1, + GST_H263_FRAME_TYPE_PB = 2 +} GstRtpH263PayFrameType; +typedef enum +{ + GST_RTP_H263_PAYLOAD_PICTURE_FORMAT_RES1 = 0, + GST_RTP_H263_PAYLOAD_PICTURE_FORMAT_SQCIF = 1, + GST_RTP_H263_PAYLOAD_PICTURE_FORMAT_QCIF = 2, + GST_RTP_H263_PAYLOAD_PICTURE_FORMAT_CIF = 3, + GST_RTP_H263_PAYLOAD_PICTURE_FORMAT_4CIF = 4, + GST_RTP_H263_PAYLOAD_PICTURE_FORMAT_16CIF = 5, + GST_RTP_H263_PAYLOAD_PICTURE_FORMAT_RES2 = 6, + GST_H263_PAYLOAD_PICTURE_FORMAT_PLUS = 7 +} GstRtpH263PayPictureFormat; + +static guint format_props[8][2] = { {254, 254}, +{6, 8}, +{9, 11}, +{18, 22}, +{18, 88}, +{18, 352}, +{254, 254}, +{255, 255} +}; + +/* + * I-frame MCBPC table: code, mask, nbits, cb, cr, mb type -> 10 = undefined (because we have guint) + */ +#define MCBPC_I_LEN 9 +#define MCBPC_I_WID 6 +static guint32 mcbpc_I[9][6] = { + {0x8000, 0x8000, 1, 0, 0, 3}, + {0x2000, 0xe000, 3, 0, 1, 3}, + {0x4000, 0xe000, 3, 1, 0, 3}, + {0x6000, 0xe000, 3, 1, 1, 3}, + {0x1000, 0xf000, 4, 0, 0, 4}, + {0x0400, 0xfc00, 6, 0, 1, 4}, + {0x0800, 0xfc00, 6, 1, 0, 4}, + {0x0c00, 0xfc00, 6, 1, 1, 4}, + {0x0080, 0xff80, 9, 10, 10, 10} +}; + +/* + * P-frame MCBPC table: code, mask, nbits, cb, cr, mb type -> 10 = undefined (because we have guint) + */ +#define MCBPC_P_LEN 21 +#define MCBPC_P_WID 6 +static guint16 mcbpc_P[21][6] = { + {0x8000, 0x8000, 1, 0, 0, 0}, + {0x3000, 0xf000, 4, 0, 1, 0}, + {0x2000, 0xf000, 4, 1, 0, 0}, + {0x1400, 0xfc00, 6, 1, 1, 0}, + {0x6000, 0xe000, 3, 0, 0, 1}, + {0x0e00, 0xfe00, 7, 0, 1, 1}, + {0x0c00, 0xfe00, 7, 1, 0, 1}, + {0x0280, 0xff80, 9, 1, 1, 1}, + {0x4000, 0xe000, 3, 0, 0, 2}, + {0x0a00, 0xfe00, 7, 0, 1, 2}, + {0x0800, 0xfe00, 7, 1, 0, 2}, + {0x0500, 0xff00, 8, 1, 1, 2}, + {0x1800, 0xf800, 5, 0, 0, 3}, + {0x0400, 0xff00, 8, 0, 1, 3}, + {0x0300, 0xff00, 8, 1, 0, 3}, + {0x0600, 0xfe00, 7, 1, 1, 3}, + {0x1000, 0xfc00, 6, 0, 0, 4}, + {0x0200, 0xff80, 9, 0, 1, 4}, + {0x0180, 0xff80, 9, 1, 0, 4}, + {0x0100, 0xff80, 9, 1, 1, 4}, + {0x0080, 0xff80, 9, 10, 10, 10} +}; + +/* + * I-frame CBPY (code, mask, nbits, Y0, Y1, Y2, Y3) + */ +#define CBPY_LEN 16 +#define CBPY_WID 7 +static guint8 cbpy_I[16][7] = { + {0x30, 0xf0, 4, 0, 0, 0, 0}, + {0x28, 0xf8, 5, 0, 0, 0, 1}, + {0x20, 0xf8, 5, 0, 0, 1, 0}, + {0x90, 0xf0, 4, 0, 0, 1, 1}, + {0x18, 0xf8, 5, 0, 1, 0, 0}, + {0x70, 0xf0, 4, 0, 1, 0, 1}, + {0x08, 0xfc, 6, 0, 1, 1, 0}, + {0xb0, 0xf0, 4, 0, 1, 1, 1}, + {0x10, 0xf8, 5, 1, 0, 0, 0}, + {0x0c, 0xfc, 6, 1, 0, 0, 1}, + {0x50, 0xf0, 4, 1, 0, 1, 0}, + {0xa0, 0xf0, 4, 1, 0, 1, 1}, + {0x40, 0xf0, 4, 1, 1, 0, 0}, + {0x80, 0xf0, 4, 1, 1, 0, 1}, + {0x60, 0xf0, 4, 1, 1, 1, 0}, + {0xc0, 0xc0, 2, 1, 1, 1, 1} +}; + +/* + * P-frame CBPY (code, mask, nbits, Y0, Y1, Y2, Y3) + */ +static guint8 cbpy_P[16][7] = { + {0x30, 0xf0, 4, 1, 1, 1, 1}, + {0x28, 0xf8, 5, 1, 1, 1, 0}, + {0x20, 0xf8, 5, 1, 1, 0, 1}, + {0x90, 0xf0, 4, 1, 1, 0, 0}, + {0x18, 0xf8, 5, 1, 0, 1, 1}, + {0x70, 0xf0, 4, 1, 0, 1, 0}, + {0x08, 0xfc, 6, 1, 0, 0, 1}, + {0xb0, 0xf0, 4, 1, 0, 0, 0}, + {0x10, 0xf8, 5, 0, 1, 1, 1}, + {0x0c, 0xfc, 6, 0, 1, 1, 0}, + {0x50, 0xf0, 4, 0, 1, 0, 1}, + {0xa0, 0xf0, 4, 0, 1, 0, 0}, + {0x40, 0xf0, 4, 0, 0, 1, 1}, + {0x80, 0xf0, 4, 0, 0, 1, 0}, + {0x60, 0xf0, 4, 0, 0, 0, 1}, + {0xc0, 0xc0, 2, 0, 0, 0, 0} +}; + +/* + * Block TCOEF table (code, mask, nbits, LAST, RUN, LEVEL) + */ +#define TCOEF_LEN 103 +#define TCOEF_WID 6 +static guint16 tcoef[103][6] = { + {0x8000, 0xc000, 3, 0, 0, 1}, + {0xf000, 0xf000, 5, 0, 0, 2}, + {0x5400, 0xfc00, 7, 0, 0, 3}, + {0x2e00, 0xfe00, 8, 0, 0, 4}, + {0x1f00, 0xff00, 9, 0, 0, 5}, + {0x1280, 0xff80, 10, 0, 0, 6}, + {0x1200, 0xff80, 10, 0, 0, 7}, + {0x0840, 0xffc0, 11, 0, 0, 8}, + {0x0800, 0xffc0, 11, 0, 0, 9}, + {0x00e0, 0xffe0, 12, 0, 0, 10}, //10 + {0x00c0, 0xffe0, 12, 0, 0, 11}, + {0x0400, 0xffe0, 12, 0, 0, 12}, + {0xc000, 0xe000, 4, 0, 1, 1}, + {0x5000, 0xfc00, 7, 0, 1, 2}, + {0x1e00, 0xff00, 9, 0, 1, 3}, + {0x03c0, 0xffc0, 11, 0, 1, 4}, + {0x0420, 0xffe0, 12, 0, 1, 5}, + {0x0500, 0xfff0, 13, 0, 1, 6}, + {0xe000, 0xf000, 5, 0, 2, 1}, + {0x1d00, 0xff00, 9, 0, 2, 2}, //20 + {0x0380, 0xffc0, 11, 0, 2, 3}, + {0x0510, 0xfff0, 13, 0, 2, 4}, + {0x6800, 0xf800, 6, 0, 3, 1}, + {0x1180, 0xff80, 10, 0, 3, 2}, + {0x0340, 0xffc0, 11, 0, 3, 3}, + {0x6000, 0xf800, 6, 0, 4, 1}, + {0x1100, 0xff80, 10, 0, 4, 2}, + {0x0520, 0xfff0, 13, 0, 4, 3}, + {0x5800, 0xf800, 6, 0, 5, 1}, + {0x0300, 0xffc0, 11, 0, 5, 2}, // 30 + {0x0530, 0xfff0, 13, 0, 5, 3}, + {0x4c00, 0xfc00, 7, 0, 6, 1}, + {0x02c0, 0xffc0, 11, 0, 6, 2}, + {0x0540, 0xfff0, 13, 0, 6, 3}, + {0x4800, 0xfc00, 7, 0, 7, 1}, + {0x0280, 0xffc0, 11, 0, 7, 2}, + {0x4400, 0xfc00, 7, 0, 8, 1}, + {0x0240, 0xffc0, 11, 0, 8, 2}, + {0x4000, 0xfc00, 7, 0, 9, 1}, + {0x0200, 0xffc0, 11, 0, 9, 2}, // 40 + {0x2c00, 0xfe00, 8, 0, 10, 1}, + {0x0550, 0xfff0, 13, 0, 10, 2}, + {0x2a00, 0xfe00, 8, 0, 11, 1}, + {0x2800, 0xfe00, 8, 0, 12, 1}, + {0x1c00, 0xff00, 9, 0, 13, 1}, + {0x1b00, 0xff00, 9, 0, 14, 1}, + {0x1080, 0xff80, 10, 0, 15, 1}, + {0x1000, 0xff80, 10, 0, 16, 1}, + {0x0f80, 0xff80, 10, 0, 17, 1}, + {0x0f00, 0xff80, 10, 0, 18, 1}, // 50 + {0x0e80, 0xff80, 10, 0, 19, 1}, + {0x0e00, 0xff80, 10, 0, 20, 1}, + {0x0d80, 0xff80, 10, 0, 21, 1}, + {0x0d00, 0xff80, 10, 0, 22, 1}, + {0x0440, 0xffe0, 12, 0, 23, 1}, + {0x0460, 0xffe0, 12, 0, 24, 1}, + {0x0560, 0xfff0, 13, 0, 25, 1}, + {0x0570, 0xfff0, 13, 0, 26, 1}, + {0x7000, 0xf000, 5, 1, 0, 1}, + {0x0c80, 0xff80, 10, 1, 0, 2}, // 60 + {0x00a0, 0xffe0, 12, 1, 0, 3}, + {0x3c00, 0xfc00, 7, 1, 1, 1}, + {0x0080, 0xffe0, 12, 1, 1, 2}, + {0x3800, 0xfc00, 7, 1, 2, 1}, + {0x3400, 0xfc00, 7, 1, 3, 1}, + {0x3000, 0xfc00, 7, 1, 4, 1}, + {0x2600, 0xfe00, 8, 1, 5, 1}, + {0x2400, 0xfe00, 8, 1, 6, 1}, + {0x2200, 0xfe00, 8, 1, 7, 1}, + {0x2000, 0xfe00, 8, 1, 8, 1}, // 70 + {0x1a00, 0xff00, 9, 1, 9, 1}, + {0x1900, 0xff00, 9, 1, 10, 1}, + {0x1800, 0xff00, 9, 1, 11, 1}, + {0x1700, 0xff00, 9, 1, 12, 1}, + {0x1600, 0xff00, 9, 1, 13, 1}, + {0x1500, 0xff00, 9, 1, 14, 1}, + {0x1400, 0xff00, 9, 1, 15, 1}, + {0x1300, 0xff00, 9, 1, 16, 1}, + {0x0c00, 0xff80, 10, 1, 17, 1}, + {0x0b80, 0xff80, 10, 1, 18, 1}, // 80 + {0x0b00, 0xff80, 10, 1, 19, 1}, + {0x0a80, 0xff80, 10, 1, 20, 1}, + {0x0a00, 0xff80, 10, 1, 21, 1}, + {0x0980, 0xff80, 10, 1, 22, 1}, + {0x0900, 0xff80, 10, 1, 23, 1}, + {0x0880, 0xff80, 10, 1, 24, 1}, + {0x01c0, 0xffc0, 11, 1, 25, 1}, + {0x0180, 0xffc0, 11, 1, 26, 1}, + {0x0140, 0xffc0, 11, 1, 27, 1}, + {0x0100, 0xffc0, 11, 1, 28, 1}, // 90 + {0x0480, 0xffe0, 12, 1, 29, 1}, + {0x04a0, 0xffe0, 12, 1, 30, 1}, + {0x04c0, 0xffe0, 12, 1, 31, 1}, + {0x04e0, 0xffe0, 12, 1, 32, 1}, + {0x0580, 0xfff0, 13, 1, 33, 1}, + {0x0590, 0xfff0, 13, 1, 34, 1}, + {0x05a0, 0xfff0, 13, 1, 35, 1}, + {0x05b0, 0xfff0, 13, 1, 36, 1}, + {0x05c0, 0xfff0, 13, 1, 37, 1}, + {0x05d0, 0xfff0, 13, 1, 38, 1}, // 100 + {0x05e0, 0xfff0, 13, 1, 39, 1}, + {0x05f0, 0xfff0, 13, 1, 40, 1}, + {0x0600, 0xfe00, 7, 0, 0xffff, 0xffff} +}; + +/* + * Motion vector code table (Code, mask, nbits, vector (halfpixel, two's complement), diff (halfpixel, two's complement)) + */ +#define MVD_LEN 64 +#define MVD_WID 5 +static guint16 mvd[64][5] = { + {0x0028, 0xfff8, 13, 0x0060, 0x0020}, + {0x0038, 0xfff8, 13, 0x0061, 0x0021}, + {0x0050, 0xfff0, 12, 0x0062, 0x0022}, + {0x0070, 0xfff0, 12, 0x0063, 0x0023}, + {0x0090, 0xfff0, 12, 0x0064, 0x0024}, + {0x00b0, 0xfff0, 12, 0x0065, 0x0025}, + {0x00d0, 0xfff0, 12, 0x0066, 0x0026}, + {0x00f0, 0xfff0, 12, 0x0067, 0x0027}, + {0x0120, 0xffe0, 11, 0x0068, 0x0028}, + {0x0160, 0xffe0, 11, 0x0069, 0x0029}, + {0x01a0, 0xffe0, 11, 0x006a, 0x002a}, + {0x01e0, 0xffe0, 11, 0x006b, 0x002b}, + {0x0220, 0xffe0, 11, 0x006c, 0x002c}, + {0x0260, 0xffe0, 11, 0x006d, 0x002d}, + {0x02a0, 0xffe0, 11, 0x006e, 0x002e}, + {0x02e0, 0xffe0, 11, 0x006f, 0x002f}, + {0x0320, 0xffe0, 11, 0x0070, 0x0030}, + {0x0360, 0xffe0, 11, 0x0071, 0x0031}, + {0x03a0, 0xffe0, 11, 0x0072, 0x0032}, + {0x03e0, 0xffe0, 11, 0x0073, 0x0033}, + {0x0420, 0xffe0, 11, 0x0074, 0x0034}, + {0x0460, 0xffe0, 11, 0x0075, 0x0035}, + {0x04c0, 0xffc0, 10, 0x0076, 0x0036}, + {0x0540, 0xffc0, 10, 0x0077, 0x0037}, + {0x05c0, 0xffc0, 10, 0x0078, 0x0038}, + {0x0700, 0xff00, 8, 0x0079, 0x0039}, + {0x0900, 0xff00, 8, 0x007a, 0x003a}, + {0x0b00, 0xff00, 8, 0x007b, 0x003b}, + {0x0e00, 0xfe00, 7, 0x007c, 0x003c}, + {0x1800, 0xf800, 5, 0x007d, 0x003d}, + {0x3000, 0xf000, 4, 0x007e, 0x003e}, + {0x6000, 0xe000, 3, 0x007f, 0x003f}, + {0x8000, 0x8000, 1, 0x0000, 0x0000}, + {0x4000, 0xe000, 3, 0x0001, 0x0041}, + {0x2000, 0xf000, 4, 0x0002, 0x0042}, + {0x1000, 0xf800, 5, 0x0003, 0x0043}, + {0x0c00, 0xfe00, 7, 0x0004, 0x0044}, + {0x0a00, 0xff00, 8, 0x0005, 0x0045}, + {0x0800, 0xff00, 8, 0x0006, 0x0046}, + {0x0600, 0xff00, 8, 0x0007, 0x0047}, + {0x0580, 0xffc0, 10, 0x0008, 0x0048}, + {0x0500, 0xffc0, 10, 0x0009, 0x0049}, + {0x0480, 0xffc0, 10, 0x000a, 0x004a}, + {0x0440, 0xffe0, 11, 0x000b, 0x004b}, + {0x0400, 0xffe0, 11, 0x000c, 0x004c}, + {0x03c0, 0xffe0, 11, 0x000d, 0x004d}, + {0x0380, 0xffe0, 11, 0x000e, 0x004e}, + {0x0340, 0xffe0, 11, 0x000f, 0x004f}, + {0x0300, 0xffe0, 11, 0x0010, 0x0050}, + {0x02c0, 0xffe0, 11, 0x0011, 0x0051}, + {0x0280, 0xffe0, 11, 0x0012, 0x0052}, + {0x0240, 0xffe0, 11, 0x0013, 0x0053}, + {0x0200, 0xffe0, 11, 0x0014, 0x0054}, + {0x01c0, 0xffe0, 11, 0x0015, 0x0055}, + {0x0180, 0xffe0, 11, 0x0016, 0x0056}, + {0x0140, 0xffe0, 11, 0x0017, 0x0057}, + {0x0100, 0xffe0, 11, 0x0018, 0x0058}, + {0x00e0, 0xfff0, 12, 0x0019, 0x0059}, + {0x00c0, 0xfff0, 12, 0x001a, 0x005a}, + {0x00a0, 0xfff0, 12, 0x001b, 0x005b}, + {0x0080, 0xfff0, 12, 0x001c, 0x005c}, + {0x0060, 0xfff0, 12, 0x001d, 0x005d}, + {0x0040, 0xfff0, 12, 0x001e, 0x005e}, + {0x0030, 0xfff8, 13, 0x001f, 0x005f} +}; + +GST_DEBUG_CATEGORY_STATIC (rtph263pay_debug); +#define GST_CAT_DEFAULT (rtph263pay_debug) + +#define GST_RTP_HEADER_LEN 12 + +enum +{ + PROP_0, + PROP_MODE_A_ONLY +}; /* elementfactory information */ static const GstElementDetails gst_rtp_h263pay_details = -GST_ELEMENT_DETAILS ("RTP H263 payloader", +GST_ELEMENT_DETAILS ("RTP H263 packet payloader", "Codec/Payloader/Network", "Payload-encodes H263 video in RTP packets (RFC 2190)", - "Neil Stratford <neils@vipadia.com>"); + "Neil Stratford <neils@vipadia.com>" + "Dejan Sakelsak <dejan.sakelsak@marand.si>"); static GstStaticPadTemplate gst_rtp_h263_pay_sink_template = GST_STATIC_PAD_TEMPLATE ("sink", @@ -179,9 +382,31 @@ static void gst_rtp_h263_pay_finalize (GObject * object); static gboolean gst_rtp_h263_pay_setcaps (GstBaseRTPPayload * payload, GstCaps * caps); +static void gst_rtp_h263_pay_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec); +static void gst_rtp_h263_pay_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec); static GstFlowReturn gst_rtp_h263_pay_handle_buffer (GstBaseRTPPayload * payload, GstBuffer * buffer); +static GstRtpH263PayBoundry *gst_rtp_h263_pay_boundry_new (guint8 * start, + guint8 * end, guint8 sbit, guint8 ebit); +static GstRtpH263PayGob *gst_rtp_h263_pay_gob_new (GstRtpH263PayBoundry * + boundry, guint gobn); +static GstRtpH263PayMB *gst_rtp_h263_pay_mb_new (GstRtpH263PayBoundry * boundry, + guint mba); +static GstRtpH263PayPackage *gst_rtp_h263_pay_package_new_empty (); +static GstRtpH263PayPackage *gst_rtp_h263_pay_package_new (guint8 * start, + guint8 * end, guint length, guint8 sbit, guint8 ebit, GstBuffer * outbuf, + gboolean marker); + +static void gst_rtp_h263_pay_boundry_destroy (GstRtpH263PayBoundry * boundry); +static void gst_rtp_h263_pay_mb_destroy (GstRtpH263PayMB * mb); +static void gst_rtp_h263_pay_gob_destroy (GstRtpH263PayGob * gob, guint ind); +static void gst_rtp_h263_pay_context_destroy (GstRtpH263PayContext * context, + guint ind); +static void gst_rtp_h263_pay_package_destroy (GstRtpH263PayPackage * pack); + static GstBaseRTPPayloadClass *parent_class = NULL; static GType @@ -239,6 +464,18 @@ gst_rtp_h263_pay_class_init (GstRtpH263PayClass * klass) gstbasertppayload_class->set_caps = gst_rtp_h263_pay_setcaps; gstbasertppayload_class->handle_buffer = gst_rtp_h263_pay_handle_buffer; + gobject_class->set_property = gst_rtp_h263_pay_set_property; + gobject_class->get_property = gst_rtp_h263_pay_get_property; +#if 0 + g_object_class_install_property (G_OBJECT_CLASS (klass), + PROP_MODE_A_ONLY, g_param_spec_boolean ("modea-only", + "Fragment packets in mode A Only", + "Disable packetization modes B and C", DEFAULT_MODE_A, + G_PARAM_READWRITE)); +#endif + GST_DEBUG_CATEGORY_INIT (rtph263pay_debug, "rtph263pay", 0, + "H263 RTP Payloader"); + } static void @@ -265,102 +502,1312 @@ gst_rtp_h263_pay_setcaps (GstBaseRTPPayload * payload, GstCaps * caps) { gboolean res; + payload->pt = GST_RTP_PAYLOAD_H263; gst_basertppayload_set_options (payload, "video", TRUE, "H263", 90000); res = gst_basertppayload_set_outcaps (payload, NULL); return res; } -static guint -gst_rtp_h263_pay_gobfiner (guint8 * data, guint len, guint curpos) +static void +gst_rtp_h263_pay_context_destroy (GstRtpH263PayContext * context, guint ind) +{ + if (!context) + return; + + if (context->gobs) { + guint i; + + for (i = 0; i < format_props[ind][0]; i++) { + if (context->gobs[i]) { + gst_rtp_h263_pay_gob_destroy (context->gobs[i], ind); + } + } + + g_free (context->gobs); + } + + g_free (context); +} + +static void +gst_rtp_h263_pay_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec) +{ + GstRtpH263Pay *rtph263pay; + + rtph263pay = GST_RTP_H263_PAY (object); + + switch (prop_id) { + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static void +gst_rtp_h263_pay_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec) { - guint16 test = 0xffff; + GstRtpH263Pay *rtph263pay; + + rtph263pay = GST_RTP_H263_PAY (object); + + switch (prop_id) { + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static GstRtpH263PayPackage * +gst_rtp_h263_pay_package_new_empty () +{ + return (GstRtpH263PayPackage *) g_malloc0 (sizeof (GstRtpH263PayPackage)); +} + +static GstRtpH263PayPackage * +gst_rtp_h263_pay_package_new (guint8 * start, guint8 * end, guint length, + guint8 sbit, guint8 ebit, GstBuffer * outbuf, gboolean marker) +{ + + GstRtpH263PayPackage *package; + + package = gst_rtp_h263_pay_package_new_empty (); + + package->payload_start = start; + package->payload_end = end; + package->payload_len = length; + package->sbit = sbit; + package->ebit = ebit; + package->outbuf = outbuf; + package->marker = marker; + + return package; +} + +static void +gst_rtp_h263_pay_package_destroy (GstRtpH263PayPackage * pack) +{ + if (pack) + g_free (pack); +} + +static GstRtpH263PayBoundry * +gst_rtp_h263_pay_boundry_new (guint8 * start, guint8 * end, guint8 sbit, + guint8 ebit) +{ + GstRtpH263PayBoundry *boundry; + + boundry = (GstRtpH263PayBoundry *) g_malloc0 (sizeof (GstRtpH263PayBoundry)); + + boundry->start = start; + boundry->end = end; + boundry->sbit = sbit; + boundry->ebit = ebit; + + return boundry; +} + +static void +gst_rtp_h263_pay_boundry_destroy (GstRtpH263PayBoundry * boundry) +{ + if (boundry) + g_free (boundry); +} + +static void +gst_rtp_h263_pay_splat_header_A (guint8 * header, + GstRtpH263PayPackage * package, GstRtpH263PayPic * piclayer) +{ + + GstRtpH263PayAHeader *a_header; + + a_header = (GstRtpH263PayAHeader *) header; + + a_header->f = 0; + a_header->p = 0; + a_header->sbit = package->sbit; + a_header->ebit = package->ebit; + a_header->src = GST_H263_PICTURELAYER_PLSRC (piclayer); + a_header->i = GST_H263_PICTURELAYER_PLTYPE (piclayer); + a_header->u = GST_H263_PICTURELAYER_PLUMV (piclayer); + a_header->s = GST_H263_PICTURELAYER_PLSAC (piclayer); + a_header->a = GST_H263_PICTURELAYER_PLAP (piclayer); + a_header->r1 = 0; + a_header->r2 = 0; + a_header->dbq = 0; + a_header->trb = 0; + a_header->tr = 0; + +} + +static void +gst_rtp_h263_pay_splat_header_B (guint8 * header, + GstRtpH263PayPackage * package, GstRtpH263PayPic * piclayer) +{ + + GstRtpH263PayBHeader *b_header; + + b_header = (GstRtpH263PayBHeader *) header; + + b_header->f = 1; + b_header->p = 0; + b_header->sbit = package->sbit; + b_header->ebit = package->ebit; + b_header->src = GST_H263_PICTURELAYER_PLSRC (piclayer); + b_header->quant = package->quant; + b_header->gobn = package->gobn; + b_header->mba1 = package->mba >> 6; + b_header->mba2 = package->mba & 0x003f; + b_header->r = 0; + b_header->i = GST_H263_PICTURELAYER_PLTYPE (piclayer); + b_header->u = GST_H263_PICTURELAYER_PLUMV (piclayer); + b_header->s = GST_H263_PICTURELAYER_PLSAC (piclayer); + b_header->a = GST_H263_PICTURELAYER_PLAP (piclayer); + + b_header->hmv11 = 0; + b_header->hmv12 = 0; + b_header->vmv11 = 0; + b_header->vmv12 = 0; + b_header->hmv21 = 0; + b_header->hmv22 = 0; + b_header->vmv21 = 0; + + if (package->nmvd > 0) { + //mvd[0] + b_header->hmv11 = (package->mvd[0] & 0x7f) >> 3; + b_header->hmv12 = (package->mvd[0] & 0x07); + //mvd[1] + b_header->vmv11 = (package->mvd[1] & 0x07f) >> 2; + b_header->vmv12 = (package->mvd[1] & 0x03); + + if (package->nmvd == 8) { + //mvd[4] + b_header->hmv21 = (package->mvd[4] & 0x7f) >> 1; + b_header->hmv22 = (package->mvd[4] & 0x01); + //mvd[5] + b_header->vmv21 = (package->mvd[5] & 0x7f); + } + } + +} + +static GstRtpH263PayBoundry * +gst_rtp_h263_pay_gobfinder (GstRtpH263Pay * rtph263pay, + GstRtpH263PayBoundry * boundry) +{ + guint8 *current; + guint range; guint i; + GstRtpH263PayBoundry *b; + + current = boundry->end + 1; + range = (rtph263pay->data - current) + rtph263pay->available_data; + + GST_DEBUG + ("Searching for next GOB, data:%p, len:%u, payload_len:%p, current:%p, range:%u", + rtph263pay->data, rtph263pay->available_data, boundry->end + 1, current, + range); /* If we are past the end, stop */ - if (curpos >= len) - return 0; - - /* start at curpos and loop looking for next gob start */ - for (i = curpos; i < len; i++) { - test = (test << 8) | data[i]; - if ((test == 0) && (i > curpos + 4)) { - return i - 3; + if (current >= rtph263pay->data + rtph263pay->available_data) + return NULL; + + for (i = 3; i < range - 3; i++) { + if ((current[i] == 0x0) && + (current[i + 1] == 0x0) && (current[i + 2] >> 7 == 0x1)) { + GST_LOG ("GOB end found at: %p start: %p len: %d", current + i - 1, + boundry->end + 1, current + i - boundry->end + 2); + return gst_rtp_h263_pay_boundry_new (boundry->end + 1, current + i - 1, 0, + 0); + } + } + + GST_DEBUG ("Couldn't find any new GBSC in this frame", range); + + b = gst_rtp_h263_pay_boundry_new (boundry->end + 1, + (guint8 *) (rtph263pay->data + rtph263pay->available_data - 1), 0, 0); + gst_rtp_h263_pay_boundry_destroy (boundry); + + return b; +} + +static GstRtpH263PayGob * +gst_rtp_h263_pay_gob_new (GstRtpH263PayBoundry * boundry, guint gobn) +{ + GstRtpH263PayGob *gob; + + gob = (GstRtpH263PayGob *) g_malloc0 (sizeof (GstRtpH263PayGob)); + + gob->start = boundry->start; + gob->end = boundry->end; + gob->length = boundry->end - boundry->start + 1; + gob->ebit = boundry->ebit; + gob->sbit = boundry->sbit; + gob->gobn = gobn; + gob->quant = 0; + gob->macroblocks = NULL; + gob->nmacroblocs = 0; + + return gob; +} + +static void +gst_rtp_h263_pay_gob_destroy (GstRtpH263PayGob * gob, guint ind) +{ + + if (!gob) + return; + + if (gob->macroblocks) { + + guint i; + + for (i = 0; i < gob->nmacroblocs; i++) { + gst_rtp_h263_pay_mb_destroy (gob->macroblocks[i]); + } + + g_free (gob->macroblocks); + } + + g_free (gob); +} + +/* + * decode MCBPC for I frames and return index in table or -1 if not found + */ +static gint +gst_rtp_h263_pay_decode_mcbpc_I (guint32 value) +{ + + gint i; + guint16 code; + + code = value >> 16; + + GST_DEBUG ("value:0x%08x, code:0x%04x", value, code); + + for (i = 0; i < MCBPC_I_LEN; i++) { + if ((code & mcbpc_I[i][1]) == mcbpc_I[i][0]) { + return i; + } + } + + GST_WARNING ("Couldn't find code, returning -1"); + + return -1; +} + +/* + * decode MCBPC for P frames and return index in table or -1 if not found + */ +static gint +gst_rtp_h263_pay_decode_mcbpc_P (guint32 value) +{ + + gint i; + guint16 code; + + code = value >> 16; + + GST_DEBUG ("value:0x%08x, code:0x%04x", value, code); + + for (i = 0; i < MCBPC_P_LEN; i++) { + if ((code & mcbpc_P[i][1]) == mcbpc_P[i][0]) { + return i; } } - return len; + + GST_WARNING ("Couldn't find code, returning -1"); + + return -1; +} + +/* + * decode CBPY and return index in table or -1 if not found + */ +static gint +gst_rtp_h263_pay_decode_cbpy (guint32 value, guint8 cbpy_table[16][7]) +{ + + gint i; + guint8 code; + + code = value >> 24; + + GST_DEBUG ("value:0x%08x, code:0x%04x", value, code); + + for (i = 0; i < CBPY_LEN; i++) { + if ((code & cbpy_table[i][1]) == cbpy_table[i][0]) { + return i; + } + } + + GST_WARNING ("Couldn't find code, returning -1"); + + return -1; +} + +/* + * decode MVD and return index in table or -1 if not found + */ +static gint +gst_rtp_h263_pay_decode_mvd (guint32 value) +{ + + gint i; + guint16 code; + + code = value >> 16; + + GST_DEBUG ("value:0x%08x, code:0x%04x", value, code); + + for (i = 0; i < MVD_LEN; i++) { + if ((code & mvd[i][1]) == mvd[i][0]) { + return i; + } + } + + GST_WARNING ("Couldn't find code, returning -1"); + + return -1; +} + +/* + * decode TCOEF and return index in table or -1 if not found + */ +static gint +gst_rtp_h263_pay_decode_tcoef (guint32 value) +{ + + gint i; + guint16 code; + + code = value >> 16; + + GST_DEBUG ("value:0x%08x, code:0x%04x", value, code); + + for (i = 0; i < TCOEF_LEN; i++) { + if ((code & tcoef[i][1]) == tcoef[i][0]) { + GST_LOG ("tcoef is %d", i); + return i; + } + } + + GST_WARNING ("Couldn't find code, returning -1"); + + return -1; +} + +/* + * the 32-bit register is like a window that we move right for "move_bits" to get the next v "data" h263 field + * "rest_bits" tells how many bits in the "data" byte address are still not used + */ + +static gint +gst_rtp_h263_pay_move_window_right (GstRtpH263PayContext * context, guint n, + guint rest_bits, guint8 ** orig_data, guint8 ** data_end) +{ + + GST_DEBUG ("Moving window: 0x%08x from: %p for %d bits, rest_bits: %d", + context->window, context->win_end, n, rest_bits); + + if (n == 0) + return rest_bits; + + while (n != 0 || context->win_end == ((*data_end) + 1)) { + //guint8 a = *data; + if (rest_bits == 0) { + if (n > 8) { + context->window = (context->window << 8) | *context->win_end; + n -= 8; + } else { + context->window = + (context->window << n) | (*context->win_end >> (8 - n)); + rest_bits = 8 - n; + if (rest_bits == 0) + context->win_end++; + break; + } + } else { + if (n > rest_bits) { + context->window = + (context-> + window << rest_bits) | (*context->win_end & (((guint) pow (2.0, + (double) rest_bits)) - 1)); + n -= rest_bits; + rest_bits = 0; + } else { + context->window = + (context->window << n) | ((*context->win_end & (((guint) pow (2.0, + (double) rest_bits)) - 1)) >> (rest_bits - n)); + rest_bits -= n; + if (rest_bits == 0) + context->win_end++; + break; + } + } + + context->win_end++; + } + + *orig_data = context->win_end - 4; + + GST_DEBUG + ("Window moved to %p with value: 0x%08x and orig_data: %p rest_bits: %d", + context->win_end, context->window, *orig_data, rest_bits); + return rest_bits; +} + +/* + * Find the start of the next MB (end of the current MB) + * returns the number of excess bits and stores the end of the MB in end + * data must be placed on first MB byte + */ +static GstRtpH263PayMB * +gst_rtp_h263_pay_B_mbfinder (GstRtpH263PayContext * context, + GstRtpH263PayGob * gob, GstRtpH263PayMB * macroblock, guint mba) +{ + + guint mb_type_index; + guint cbpy_type_index; + guint tcoef_type_index; + GstRtpH263PayMB *mac; + + mac = + gst_rtp_h263_pay_mb_new (gst_rtp_h263_pay_boundry_new (macroblock->end, + macroblock->end, 8 - macroblock->ebit, macroblock->ebit), mba); + + if (mac->sbit == 8) { + mac->start++; +// mac->end++; + mac->sbit = 0; + } + + GST_LOG ("current_pos:%p, end:%p, rest_bits:%d, window:%x", + mac->start, mac->end, macroblock->ebit, context->window); + + GST_LOG ("Current pos after COD: %p", mac->end); + + if (context->piclayer->ptype_pictype == 0) { + //We have an I frame + gint i; + guint last; + guint ind; + + //Step 2 decode MCBPC I + mb_type_index = gst_rtp_h263_pay_decode_mcbpc_I (context->window); + + GST_DEBUG ("MCBPC index: %d", mb_type_index); + if (mb_type_index == -1) { + GST_ERROR ("MB index shouldn't be -1 in window: %08x", context->window); + return NULL; + } + + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, mcbpc_I[mb_type_index][2], + mac->ebit, &mac->end, &gob->end); + + mac->mb_type = mcbpc_I[mb_type_index][5]; + + if (mb_type_index == 8) { + GST_LOG ("Stuffing skipping rest of MB header"); + return mac; + } + //Step 3 decode CBPY I + cbpy_type_index = gst_rtp_h263_pay_decode_cbpy (context->window, cbpy_I); + + GST_DEBUG ("CBPY index: %d", cbpy_type_index); + if (cbpy_type_index == -1) { + GST_ERROR ("CBPY index shouldn't be -1 in window: %08x", context->window); + return NULL; + } + + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, cbpy_I[cbpy_type_index][2], + mac->ebit, &mac->end, &gob->end); + + //Step 4 decode rest of MB + //MB type 1 and 4 have DQUANT - we store it for packaging purposes + if (mcbpc_I[mb_type_index][5] == 4) { + GST_DEBUG ("Shifting DQUANT"); + + mac->quant = (context->window >> 30); + + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, 2, mac->ebit, &mac->end, + &gob->end); + } + //Step 5 go trough the blocks - decode DC and TCOEF + i = 0; + last = 0; + ind = 0; + for (i = 0; i < N_BLOCKS; i++) { + + GST_DEBUG ("Decoding INTRADC and TCOEF, i:%d", i); + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, 8, mac->ebit, &mac->end, + &gob->end); + + if (i > 3) { + ind = mcbpc_I[mb_type_index][i - 1]; + } else { + ind = cbpy_I[cbpy_type_index][i + 3]; + } + + if (ind == 1) { + while (last == 0) { + tcoef_type_index = gst_rtp_h263_pay_decode_tcoef (context->window); + + GST_DEBUG ("TCOEF index: %d", tcoef_type_index); + if (tcoef_type_index == -1) { + GST_ERROR ("TCOEF index shouldn't be -1 in window: %08x", + context->window); + return NULL; + } + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, + tcoef[tcoef_type_index][2], mac->ebit, &mac->end, &gob->end); + + last = tcoef[tcoef_type_index][3]; + if (tcoef_type_index == 102) { + if ((context->window & 0x80000000) == 0x80000000) + last = 1; + else + last = 0; + + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, 15, + mac->ebit, &mac->end, &gob->end); + } + } + last = 0; + } + } + + } else { + //We have a P frame + guint i; + guint last; + guint ind; + + //Step 1 check COD + GST_DEBUG ("Checking for COD"); + if ((context->window & 0x80000000) == 0x80000000) { + //The MB is not coded + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, 1, mac->ebit, &mac->end, + &gob->end); + GST_DEBUG ("COOOOOOOOOOOD = 1"); + + return mac; + } else { + //The MB is coded + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, 1, mac->ebit, &mac->end, + &gob->end); + } + + //Step 2 decode MCBPC P + mb_type_index = gst_rtp_h263_pay_decode_mcbpc_P (context->window); + + GST_DEBUG ("MCBPC index: %d", mb_type_index); + if (mb_type_index == -1) { + GST_ERROR ("MB index shouldn't be -1 in window: %08x", context->window); + return NULL; + } + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, mcbpc_P[mb_type_index][2], + mac->ebit, &mac->end, &gob->end); + + mac->mb_type = mcbpc_P[mb_type_index][5]; + + if (mb_type_index == 20) { + GST_LOG ("Stuffing skipping rest of MB header"); + return mac; + } + //Step 3 decode CBPY P + cbpy_type_index = gst_rtp_h263_pay_decode_cbpy (context->window, cbpy_P); + + GST_DEBUG ("CBPY index: %d", cbpy_type_index); + if (cbpy_type_index == -1) { + GST_ERROR ("CBPY index shouldn't be -1 in window: %08x", context->window); + return NULL; + } + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, cbpy_P[cbpy_type_index][2], + mac->ebit, &mac->end, &gob->end); + + //MB type 1 and 4 have DQUANT - we add it to MB object and jump over + if (mcbpc_P[mb_type_index][5] == 4 || mcbpc_P[mb_type_index][5] == 1) { + GST_DEBUG ("Shifting DQUANT"); + + mac->quant = context->window >> 30; + + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, 2, mac->ebit, &mac->end, + &gob->end); + } + //MB types < 3 have MVD1-4 + if (mcbpc_P[mb_type_index][5] < 3) { + + guint nmvd; + gint j; + + nmvd = 2; + if (mcbpc_P[mb_type_index][5] == 2) + nmvd = 8; + + for (j = 0; j < nmvd; j++) { + guint mvd_type; + + mvd_type = gst_rtp_h263_pay_decode_mvd (context->window); + + if (mvd_type == -1) { + GST_ERROR ("MVD1-4 index shouldn't be -1 in window: %08x", + context->window); + return NULL; + } + //set the MB mvd values + mac->mvd[j] = mvd[mvd_type][3]; + + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, mvd[mvd_type][2], + mac->ebit, &mac->end, &gob->end); + } + + + } + //Step 5 go trough the blocks - decode DC and TCOEF + i = 0; + last = 0; + ind = 0; + for (i = 0; i < N_BLOCKS; i++) { + + //if MB type 3 or 4 then INTRADC coef are present in blocks + if (mcbpc_P[mb_type_index][5] > 2) { + GST_DEBUG ("INTRADC coef: %d", i); + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, 8, mac->ebit, + &mac->end, &gob->end); + } else { + GST_DEBUG ("INTRADC coef is not present"); + } + + //check if the block has TCOEF + if (i > 3) { + ind = mcbpc_P[mb_type_index][i - 1]; + } else { + if (mcbpc_P[mb_type_index][5] > 2) { + ind = cbpy_I[cbpy_type_index][i + 3]; + } else { + ind = cbpy_P[cbpy_type_index][i + 3]; + } + } + + if (ind == 1) { + while (last == 0) { + tcoef_type_index = gst_rtp_h263_pay_decode_tcoef (context->window); + + GST_DEBUG ("TCOEF index: %d", tcoef_type_index); + if (tcoef_type_index == -1) { + GST_ERROR ("TCOEF index shouldn't be -1 in window: %08x", + context->window); + return NULL; + } + + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, + tcoef[tcoef_type_index][2], mac->ebit, &mac->end, &gob->end); + + last = tcoef[tcoef_type_index][3]; + if (tcoef_type_index == 102) { + if ((context->window & 0x80000000) == 0x80000000) + last = 1; + else + last = 0; + + mac->ebit = + gst_rtp_h263_pay_move_window_right (context, 15, + mac->ebit, &mac->end, &gob->end); + } + } + last = 0; + } + } + } + + mac->length = mac->end - mac->start + 1; + + return mac; +} + +static GstRtpH263PayMB * +gst_rtp_h263_pay_mb_new (GstRtpH263PayBoundry * boundry, guint mba) +{ + GstRtpH263PayMB *mb; + gint i; + + mb = (GstRtpH263PayMB *) g_malloc0 (sizeof (GstRtpH263PayMB)); + + mb->start = boundry->start; + mb->end = boundry->end; + mb->length = boundry->end - boundry->start + 1; + mb->sbit = boundry->sbit; + mb->ebit = boundry->ebit; + mb->mba = mba; + + for (i = 0; i < 5; i++) + mb->mvd[i] = 0; + + return mb; +} + +static void +gst_rtp_h263_pay_mb_destroy (GstRtpH263PayMB * mb) +{ + if (!mb) + return; + + g_free (mb); } static GstFlowReturn -gst_rtp_h263_pay_flush (GstRtpH263Pay * rtph263pay) +gst_rtp_h263_pay_push (GstRtpH263Pay * rtph263pay, + GstRtpH263PayContext * context, GstRtpH263PayPackage * package) { - guint avail; - GstBuffer *outbuf; - GstFlowReturn ret = 0; - gboolean fragmented; - guint8 *data, *header; - GstH263PictureLayer *piclayer; + + /* + * Splat the payload header values + */ + guint8 *header; guint8 *payload; - guint payload_len, total_len; - guint curpos, nextgobpos; + GstFlowReturn ret; - avail = gst_adapter_available (rtph263pay->adapter); - if (avail == 0) - return GST_FLOW_OK; + header = gst_rtp_buffer_get_payload (package->outbuf); + payload = header + package->mode; + + switch (package->mode) { + case GST_RTP_H263_PAYLOAD_HEADER_MODE_A: + GST_LOG ("Pushing A packet"); + gst_rtp_h263_pay_splat_header_A (header, package, context->piclayer); + break; + case GST_RTP_H263_PAYLOAD_HEADER_MODE_B: + GST_LOG ("Pushing B packet"); + gst_rtp_h263_pay_splat_header_B (header, package, context->piclayer); + break; + case GST_RTP_H263_PAYLOAD_HEADER_MODE_C: + //gst_rtp_h263_pay_splat_header_C(header, package, context->piclayer); + //break; + default: + return GST_FLOW_ERROR; + } - fragmented = FALSE; - /* Get a pointer to all the data for the frame */ - data = (guint8 *) gst_adapter_peek (rtph263pay->adapter, avail); + /* + * Copy the payload data in the buffer + */ + GST_DEBUG ("Copiing memory"); + memcpy (payload, (guint8 *) package->payload_start, package->payload_len); - /* Start at the begining and loop looking for gobs */ - curpos = 0; + /* + * timestamp the buffer + */ + GST_BUFFER_TIMESTAMP (package->outbuf) = rtph263pay->first_ts; - /* Picture header */ - piclayer = (GstH263PictureLayer *) data; + gst_rtp_buffer_set_marker (package->outbuf, package->marker); + if (package->marker) + GST_DEBUG ("Marker set!"); - while ((nextgobpos = gst_rtp_h263_pay_gobfiner (data, avail, curpos)) > 0) { + ret = + gst_basertppayload_push (GST_BASE_RTP_PAYLOAD (rtph263pay), + package->outbuf); + GST_DEBUG ("Package pushed, returning"); - payload_len = nextgobpos - curpos; - total_len = payload_len + GST_RFC2190A_HEADER_LEN; + gst_rtp_h263_pay_package_destroy (package); - outbuf = gst_rtp_buffer_new_allocate (total_len, 0, 0); + return ret; +} - header = gst_rtp_buffer_get_payload (outbuf); - payload = header + GST_RFC2190A_HEADER_LEN; +static GstFlowReturn +gst_rtp_h263_pay_A_fragment_push (GstRtpH263Pay * rtph263pay, + GstRtpH263PayContext * context, guint first, guint last) +{ + + GstRtpH263PayPackage *pack; + + pack = gst_rtp_h263_pay_package_new_empty (); + + pack->payload_start = context->gobs[first]->start; + pack->sbit = context->gobs[first]->sbit; + pack->ebit = context->gobs[last]->ebit; + pack->payload_len = + (context->gobs[last]->end - context->gobs[first]->start) + 1; + pack->marker = FALSE; - /* Build the headers */ - GST_RFC2190A_HEADER_F (header) = 0; - GST_RFC2190A_HEADER_P (header) = 0; - GST_RFC2190A_HEADER_SBIT (header) = 0; - GST_RFC2190A_HEADER_EBIT (header) = 0; +//if (last == format_props[context->piclayer->ptype_srcformat][0] - 1) { + pack->marker = TRUE; +//} - GST_RFC2190A_HEADER_SRC (header) = GST_H263_PICTURELAYER_PLSRC (piclayer); - GST_RFC2190A_HEADER_I (header) = GST_H263_PICTURELAYER_PLTYPE (piclayer); - GST_RFC2190A_HEADER_U (header) = GST_H263_PICTURELAYER_PLUMV (piclayer); - GST_RFC2190A_HEADER_S (header) = GST_H263_PICTURELAYER_PLSAC (piclayer); - GST_RFC2190A_HEADER_A (header) = GST_H263_PICTURELAYER_PLAP (piclayer); - GST_RFC2190A_HEADER_R1 (header) = 0; - GST_RFC2190A_HEADER_R2 (header) = 0; - GST_RFC2190A_HEADER_DBQ (header) = 0; - GST_RFC2190A_HEADER_TRB (header) = 0; - GST_RFC2190A_HEADER_TR (header) = 0; + pack->gobn = context->gobs[first]->gobn; + pack->mode = GST_RTP_H263_PAYLOAD_HEADER_MODE_A; + pack->outbuf = + gst_rtp_buffer_new_allocate (pack->payload_len + pack->mode, 0, 0); - /* last fragment gets the marker bit set */ - gst_rtp_buffer_set_marker (outbuf, nextgobpos < avail ? 0 : 1); + GST_DEBUG ("Sending len:%d data to push function", pack->payload_len); - memcpy (payload, data + curpos, payload_len); + return gst_rtp_h263_pay_push (rtph263pay, context, pack); +} - GST_BUFFER_TIMESTAMP (outbuf) = rtph263pay->first_timestamp; - GST_BUFFER_DURATION (outbuf) = rtph263pay->first_duration; +static GstFlowReturn +gst_rtp_h263_pay_B_fragment_push (GstRtpH263Pay * rtph263pay, + GstRtpH263PayContext * context, GstRtpH263PayGob * gob, guint first, + guint last, GstRtpH263PayBoundry * boundry) +{ - ret = gst_basertppayload_push (GST_BASE_RTP_PAYLOAD (rtph263pay), outbuf); + GstRtpH263PayPackage *pack; + guint mv; - curpos = nextgobpos; + pack = gst_rtp_h263_pay_package_new_empty (); + + pack->payload_start = gob->macroblocks[first]->start; + pack->sbit = gob->macroblocks[first]->sbit; + if (first == 0) { + pack->payload_start = boundry->start; + pack->sbit = boundry->sbit; + pack->quant = gob->quant; + } else { + pack->quant = gob->macroblocks[first]->quant; + } + pack->payload_end = gob->macroblocks[last]->end; + + pack->ebit = gob->macroblocks[last]->ebit; + pack->mba = gob->macroblocks[first]->mba; + pack->gobn = gob->gobn; + pack->mode = GST_RTP_H263_PAYLOAD_HEADER_MODE_B; + pack->nmvd = 0; + + if (gob->macroblocks[first]->mb_type < 3) { + if (gob->macroblocks[first]->mb_type == 2) + pack->nmvd = 8; + else if (gob->macroblocks[first]->mb_type < 2) + pack->nmvd = 2; + + for (mv = 0; mv < pack->nmvd; mv++) + pack->mvd[mv] = gob->macroblocks[first]->mvd[mv]; } - /* Flush the whole packet */ - gst_adapter_flush (rtph263pay->adapter, avail); + pack->marker = FALSE; + if (last == gob->nmacroblocs - 1) { + pack->ebit = 0; + } + + if ((format_props[context->piclayer->ptype_srcformat][0] - 1 == gob->gobn) + && (last == gob->nmacroblocs - 1)) { + pack->marker = TRUE; + } + + pack->payload_len = pack->payload_end - pack->payload_start + 1; + pack->outbuf = + gst_rtp_buffer_new_allocate (pack->payload_len + pack->mode, 0, 0); + + return gst_rtp_h263_pay_push (rtph263pay, context, pack); +} + + +static gboolean +gst_rtp_h263_pay_mode_B_fragment (GstRtpH263Pay * rtph263pay, + GstRtpH263PayContext * context, GstRtpH263PayGob * gob) +{ + + + /*---------- MODE B MODE FRAGMENTATION ----------*/ + GstRtpH263PayMB *a; + GstRtpH263PayMB *mac; + guint max_payload_size; + GstRtpH263PayBoundry *boundry; + guint mb; + //TODO remove m + GstRtpH263PayMB *m; + + guint first = 0; + guint payload_len; + + max_payload_size = + context->mtu - GST_RTP_H263_PAYLOAD_HEADER_MODE_B - GST_RTP_HEADER_LEN; + + boundry = gst_rtp_h263_pay_boundry_new (gob->start, gob->start, gob->sbit, 0); + + gob->macroblocks = + (GstRtpH263PayMB **) g_malloc0 (sizeof (GstRtpH263PayMB *) * + format_props[context->piclayer->ptype_srcformat][1]); + + GST_LOG ("GOB isn't PB frame, applying mode B"); + + //initializing window + context->win_end = boundry->end; + if (gst_rtp_h263_pay_move_window_right (context, 32, boundry->ebit, + &boundry->end, &gob->end) != 0) { + GST_ERROR + ("The rest of the bits should be 0, exiting, because something bad happend"); + gst_adapter_flush (rtph263pay->adapter, rtph263pay->available_data); + goto decode_error; + } + //The first GOB of a frame "has no" actual header - PICTURE header is his header + if (gob->gobn == 0) { + guint shift; + GST_LOG ("Initial GOB"); + shift = 43; + + boundry->ebit = + gst_rtp_h263_pay_move_window_right (context, shift, boundry->ebit, + &boundry->end, &gob->end); + + //We need PQUANT for mode B packages - so we store it + gob->quant = context->window >> 27; + + //if PCM == 1, then PSBI is present - header has 51 bits + //shift for PQUANT (5) and PCM (1) = 6 bits + shift = 6; + if (context->cpm == 1) + shift += 2; + boundry->ebit = + gst_rtp_h263_pay_move_window_right (context, shift, boundry->ebit, + &boundry->end, &gob->end); + + GST_DEBUG ("window: 0x%08x", context->window); + + //Shifting the PEI and PSPARE fields + while ((context->window & 0x80000000) == 0x80000000) { + boundry->ebit = + gst_rtp_h263_pay_move_window_right (context, 9, + boundry->ebit, &boundry->end, &gob->end); + GST_LOG ("window: 0x%x", context->window); + } + + //shift the last PEI field + boundry->ebit = + gst_rtp_h263_pay_move_window_right (context, 1, boundry->ebit, + &boundry->end, &gob->end); + + } else { + //skipping GOBs 24 header bits + 5 GQUANT + guint shift = 24; + + GST_LOG ("INTER GOB"); + + //if CPM == 1, there are 2 more bits in the header - GSBI header is 31 bits long + if (context->cpm == 1) + shift += 2; + + GST_LOG ("window: 0x%x", context->window); + boundry->ebit = + gst_rtp_h263_pay_move_window_right (context, shift, + boundry->ebit, &boundry->end, &gob->end); + + //We need GQUANT for mode B packages - so we store it + gob->quant = context->window >> 27; + + shift = 5; + boundry->ebit = + gst_rtp_h263_pay_move_window_right (context, shift, + boundry->ebit, &boundry->end, &gob->end); + + GST_LOG ("window: 0x%x", context->window); + } + + GST_DEBUG ("GQUANT IS: %08x", gob->quant); + + // We are on MB layer + mb = 0; + + + m = mac = a = gst_rtp_h263_pay_mb_new (boundry, 0); + for (mb = 0; mb < format_props[context->piclayer->ptype_srcformat][1]; mb++) { + + GST_DEBUG ("================ START MB %d =================", mb); + + //Find next macroblock boundaries + if (!(mac = gst_rtp_h263_pay_B_mbfinder (context, gob, mac, mb))) { + + GST_DEBUG ("Error decoding MB - sbit: %d", 8 - m->ebit); + GST_ERROR ("Error decoding in GOB"); + + goto decode_error; + } + //If mb_type == stuffing, don't increment the mb address + if (mac->mb_type == 10) { + mb--; + continue; + } else { + gob->nmacroblocs++; + } + + gob->macroblocks[mb] = mac; + + if (mac->end >= gob->end) { + GST_LOG ("No more MBs in this GOB"); + if (!mac->ebit) { + mac->end--; + } + gob->end = mac->end; + break; + } + m = mac; + GST_DEBUG ("Found MB: mba: %d start: %p end: %p len: %d sbit: %d ebit: %d", + mac->mba, mac->start, mac->end, mac->length, mac->sbit, mac->ebit); + GST_DEBUG ("================ END MB %d =================", mb); + } + + mb = 0; + first = 0; + payload_len = boundry->end - boundry->start + 1; + GST_DEBUG ("------------------------- NEW PACKAGE ----------------------"); + while (mb < gob->nmacroblocs) { + if (payload_len + gob->macroblocks[mb]->length < max_payload_size) { + + //FIXME: payload_len is not the real length -> ignoring sbit/ebit + payload_len += gob->macroblocks[mb]->length; + mb++; + + } else { + //FIXME: we should include the last few bits of the GOB in the package - do we do that now? + //GST_DEBUG ("Pushing GOBS %d to %d because payload size is %d", first, + // first == mb - 1, payload_len); + + if (gst_rtp_h263_pay_B_fragment_push (rtph263pay, context, gob, first, + mb - 1, boundry)) { + GST_ERROR ("Oooops, there was an error sending"); + return FALSE; + } + + payload_len = 0; + first = mb; + GST_DEBUG + ("------------------------- END PACKAGE ----------------------"); + GST_DEBUG + ("------------------------- NEW PACKAGE ----------------------"); + } + } + + /* Push rest */ + GST_DEBUG ("Remainder first: %d, MB: %d", first, mb); + if (payload_len != 0) { + if (gst_rtp_h263_pay_B_fragment_push (rtph263pay, context, gob, first, + mb - 1, boundry)) { + GST_ERROR ("Oooops, there was an error sending!"); + return FALSE; + } + } + + /*---------- END OF MODE B FRAGMENTATION ----------*/ + + return TRUE; + +decode_error: + return FALSE; +} + +static GstFlowReturn +gst_rtp_h263_send_entire_frame (GstRtpH263Pay * rtph263pay, + GstRtpH263PayContext * context) +{ + GstRtpH263PayPackage *pack; + pack = + gst_rtp_h263_pay_package_new (rtph263pay->data, + rtph263pay->data + rtph263pay->available_data, + rtph263pay->available_data, 0, 0, NULL, TRUE); + pack->mode = GST_RTP_H263_PAYLOAD_HEADER_MODE_A; + + GST_DEBUG ("Available data: %d", rtph263pay->available_data); + + pack->outbuf = + gst_rtp_buffer_new_allocate (pack->payload_len + + GST_RTP_H263_PAYLOAD_HEADER_MODE_A, 0, 0); + + return gst_rtp_h263_pay_push (rtph263pay, context, pack); +} + +static GstFlowReturn +gst_rtp_h263_pay_flush (GstRtpH263Pay * rtph263pay) +{ + + /* + * FIXME: GSTUF bits are ignored right now, + * - not using EBIT/SBIT payload header fields in mode A fragmentation - ffmpeg doesn't need them, but others? + */ + + GstFlowReturn ret; + GstRtpH263PayContext *context; + gint i; + + ret = 0; + context = (GstRtpH263PayContext *) g_malloc0 (sizeof (GstRtpH263PayContext)); + context->mtu = + rtph263pay->payload.mtu - (MTU_SECURITY_OFFSET + GST_RTP_HEADER_LEN + + GST_RTP_H263_PAYLOAD_HEADER_MODE_C); + + GST_DEBUG ("MTU: %d", context->mtu); + rtph263pay->available_data = gst_adapter_available (rtph263pay->adapter); + if (rtph263pay->available_data == 0) { + ret = GST_FLOW_RESEND; + goto end; + } + + /* Get a pointer to all the data for the frame */ + rtph263pay->data = + (guint8 *) gst_adapter_peek (rtph263pay->adapter, + rtph263pay->available_data); + + /* Picture header */ + context->piclayer = (GstRtpH263PayPic *) rtph263pay->data; + + if (context->piclayer->ptype_pictype == 0) + GST_DEBUG ("We got an I-frame"); + else + GST_DEBUG ("We got a P-frame"); + + context->cpm = rtph263pay->data[6] >> 7; + + GST_DEBUG ("CPM: %d", context->cpm); + + GST_DEBUG ("Payload length is: %d", rtph263pay->available_data); + + /* + * - MODE A - If normal, I and P frames, -> mode A + * - GOB layer fragmentation + * - MODE B - If normal, I and P frames, but GOBs > mtu + * - MB layer fragmentation + * - MODE C - For P frames with PB option, but GOBs > mtu + * - MB layer fragmentation + */ + if (rtph263pay->available_data + GST_RTP_H263_PAYLOAD_HEADER_MODE_A + + GST_RTP_HEADER_LEN < context->mtu) { + ret = gst_rtp_h263_send_entire_frame (rtph263pay, context); + } else { + + /*---------- FRAGMENTING THE FRAME BECAUSE TOO LARGE TO FIT IN MTU ----------*/ + GstRtpH263PayBoundry *bound; + gint first; + guint payload_len; + + GST_DEBUG ("Frame too large for MTU"); + /* + * Let's go trough all the data and fragment it untill end is reached + */ + + bound = gst_rtp_h263_pay_boundry_new (NULL, rtph263pay->data - 1, 0, 0); + context->gobs = + (GstRtpH263PayGob **) g_malloc0 (format_props[context-> + piclayer->ptype_srcformat][0] * sizeof (GstRtpH263PayGob *)); + + + for (i = 0; i < format_props[context->piclayer->ptype_srcformat][0]; i++) { + GST_DEBUG ("Searching for gob %d", i); + bound = gst_rtp_h263_pay_gobfinder (rtph263pay, bound); + if (!bound) { + GST_WARNING + ("No GOB's were found in data stream! Please enable RTP mode in encoder. Forcing mode A for now."); + ret = gst_rtp_h263_send_entire_frame (rtph263pay, context); + goto end; + } + + context->gobs[i] = gst_rtp_h263_pay_gob_new (bound, i); + //FIXME - encoders may generate an EOS gob that has to be processed + GST_DEBUG + ("Gob values are: gobn: %d, start: %p len: %d ebit: %d sbit: %d", i, + context->gobs[i]->start, context->gobs[i]->length, + context->gobs[i]->ebit, context->gobs[i]->sbit); + } + + gst_rtp_h263_pay_boundry_destroy (bound); + + // Make packages smaller than MTU + // A mode + // - if ( GOB > MTU) -> B mode || C mode + // Push packages + + first = 0; + payload_len = 0; + i = 0; + while (i < format_props[context->piclayer->ptype_srcformat][0]) { + + if (context->gobs[i]->length >= context->mtu) { + if (payload_len == 0) { + + GST_DEBUG ("GOB len > MTU"); +#if 0 + if (rtph263pay->prop_payload_mode) { + payload_len = context->gobs[i]->length; + goto force_a; + } +#endif + if (!context->piclayer->ptype_pbmode) { + GST_DEBUG ("MODE B on GOB %d needed", i); + if (!gst_rtp_h263_pay_mode_B_fragment (rtph263pay, context, + context->gobs[i])) { + GST_ERROR ("There was an error fragmenting in mode B"); + return GST_FLOW_ERROR; + } + } else { + //IMPLEMENT C mode + GST_ERROR ("MODE C on GOB %d needed, but not supported yet", i); + /*if(!gst_rtp_h263_pay_mode_C_fragment(rtph263pay, context, context->gobs[i])) { + ret = GST_FLOW_OK; + GST_ERROR("There was an error fragmenting in mode C"); + goto decode_error; + } */ + goto decode_error; + } + decode_error: + i++; + first = i; + continue; + + } else { + goto payload_a_push; + } + } + + if (payload_len + context->gobs[i]->length < context->mtu) { + GST_DEBUG ("GOB %d fills mtu", i); + payload_len += context->gobs[i]->length; + i++; + if (i == format_props[context->piclayer->ptype_srcformat][0]) { + GST_DEBUG ("LAST GOB %d", i); + goto payload_a_push; + } + + } else { + payload_a_push: + GST_DEBUG ("Pushing GOBS %d to %d because payload size is %d", first, + first == i ? i : i - 1, payload_len); + gst_rtp_h263_pay_A_fragment_push (rtph263pay, context, first, + first == i ? i : i - 1); + payload_len = 0; + first = i; + } + continue; +#if 0 + GST_DEBUG ("Pushing GOBS %d to %d because payload size is %d", first, i, + payload_len); + gst_rtp_h263_pay_A_fragment_push (rtph263pay, context, first, i); + payload_len = 0; + i++; + first = i; +#endif + } + + + }/*---------- END OF FRAGMENTATION ----------*/ + + /* Flush the input buffer data */ + +end: + gst_rtp_h263_pay_context_destroy (context, + context->piclayer->ptype_srcformat); + gst_adapter_flush (rtph263pay->adapter, rtph263pay->available_data); return ret; } @@ -368,19 +1815,21 @@ gst_rtp_h263_pay_flush (GstRtpH263Pay * rtph263pay) static GstFlowReturn gst_rtp_h263_pay_handle_buffer (GstBaseRTPPayload * payload, GstBuffer * buffer) { + GstRtpH263Pay *rtph263pay; GstFlowReturn ret; guint size; + GST_DEBUG ("-------------------- NEW FRAME ---------------"); rtph263pay = GST_RTP_H263_PAY (payload); size = GST_BUFFER_SIZE (buffer); - rtph263pay->first_timestamp = GST_BUFFER_TIMESTAMP (buffer); - rtph263pay->first_duration = GST_BUFFER_DURATION (buffer); + rtph263pay->first_ts = GST_BUFFER_TIMESTAMP (buffer); /* we always encode and flush a full picture */ gst_adapter_push (rtph263pay->adapter, buffer); ret = gst_rtp_h263_pay_flush (rtph263pay); + GST_DEBUG ("-------------------- END FRAME ---------------"); return ret; } |