Initial import

4 files changed, 350 insertions, 0 deletions

diff --git a/tools/h264.c b/tools/h264.c
new file mode 100644
index 00000000..65be1180
--- /dev/null
+++ b/tools/h264.c
@@ -0,0 +1,188 @@
+/*
+ * h264.c: H.264 bitstream decoding
+ *
+ * See the main source file 'xineliboutput.c' for copyright information and
+ * how to reach the author.
+ *
+ * $Id: h264.c,v 1.1 2008-02-04 22:43:27 phintuka Exp $
+ *
+ */
+
+
+#ifndef LOGDBG
+# include <vdr/tools.h>
+# include "../logdefs.h"
+#endif
+
+#define NOCACHE 1
+#include "bitstream.h"
+
+#include "mpeg.h"
+#include "h264.h"
+
+
+int h264_parse_sps(const uint8_t *buf, int len, h264_sps_data_t *sps)
+{
+  br_state br = BR_INIT(buf, len);
+  int profile_idc, pic_order_cnt_type;
+  int frame_mbs_only;
+  int i, j;
+
+  profile_idc = br_get_u8(&br);
+  LOGDBG("H.264 SPS: profile_idc %d", profile_idc);
+  /* constraint_set0_flag = br_get_bit(br);    */
+  /* constraint_set1_flag = br_get_bit(br);    */
+  /* constraint_set2_flag = br_get_bit(br);    */
+  /* constraint_set3_flag = br_get_bit(br);    */
+  /* reserved             = br_get_bits(br,4); */
+  /* level_idc            = br_get_u8(br);     */ 
+  br_skip_bits(&br, 16);
+  br_skip_ue_golomb(&br);   /* seq_parameter_set_id */
+  if (profile_idc >= 100) {
+    if (br_get_ue_golomb(&br) == 3) /* chroma_format_idc      */
+      br_skip_bit(&br);     /* residual_colour_transform_flag */
+    br_skip_ue_golomb(&br); /* bit_depth_luma - 8             */
+    br_skip_ue_golomb(&br); /* bit_depth_chroma - 8           */
+    br_skip_bit(&br);       /* transform_bypass               */
+    if (br_get_bit(&br))    /* seq_scaling_matrix_present     */
+      for (i = 0; i < 8; i++)  
+	if (br_get_bit(&br)) { /* seq_scaling_list_present    */
+	  int last = 8, next = 8, size = (i<6) ? 16 : 64;
+	  for (j = 0; j < size; j++) {
+	    if (next)
+	      next = (last + br_get_se_golomb(&br)) & 0xff;
+	    last = next ?: last;
+	  }
+	}
+  }
+
+  br_skip_ue_golomb(&br);      /* log2_max_frame_num - 4 */
+  pic_order_cnt_type = br_get_ue_golomb(&br);
+  if (pic_order_cnt_type == 0)
+    br_skip_ue_golomb(&br);    /* log2_max_poc_lsb - 4 */
+  else if (pic_order_cnt_type == 1) {
+    br_skip_bit(&br);          /* delta_pic_order_always_zero     */
+    br_skip_se_golomb(&br);    /* offset_for_non_ref_pic          */
+    br_skip_se_golomb(&br);    /* offset_for_top_to_bottom_field  */
+    j = br_get_ue_golomb(&br); /* num_ref_frames_in_pic_order_cnt_cycle */
+    for (i = 0; i < j; i++)
+      br_skip_se_golomb(&br);  /* offset_for_ref_frame[i]         */
+  }
+  br_skip_ue_golomb(&br);      /* ref_frames                      */
+  br_skip_bit(&br);            /* gaps_in_frame_num_allowed       */
+  sps->width  /* mbs */ = br_get_ue_golomb(&br) + 1;
+  sps->height /* mbs */ = br_get_ue_golomb(&br) + 1;
+  frame_mbs_only        = br_get_bit(&br);
+  LOGDBG("H.264 SPS: pic_width:  %u mbs", (unsigned) sps->width);
+  LOGDBG("H.264 SPS: pic_height: %u mbs", (unsigned) sps->height);
+  LOGDBG("H.264 SPS: frame only flag: %d", frame_mbs_only);
+  
+  sps->width  *= 16;
+  sps->height *= 16 * (2-frame_mbs_only);
+  
+  if (!frame_mbs_only)
+    if (br_get_bit(&br)) /* mb_adaptive_frame_field_flag */
+      LOGDBG("H.264 SPS: MBAFF");
+  br_skip_bit(&br);      /* direct_8x8_inference_flag    */
+  if (br_get_bit(&br)) { /* frame_cropping_flag */
+    uint32_t crop_left   = br_get_ue_golomb(&br);
+    uint32_t crop_right  = br_get_ue_golomb(&br);
+    uint32_t crop_top    = br_get_ue_golomb(&br);
+    uint32_t crop_bottom = br_get_ue_golomb(&br);
+    LOGDBG("H.264 SPS: cropping %d %d %d %d", 
+	   crop_left, crop_top, crop_right, crop_bottom);
+
+    sps->width -= 2*(crop_left + crop_right);
+    if (frame_mbs_only)
+      sps->height -= 2*(crop_top + crop_bottom); 
+    else
+      sps->height -= 4*(crop_top + crop_bottom);
+  }
+  /* skip VUI parameters */
+
+  LOGDBG("H.264 SPS: -> video size %dx%d ", sps->width, sps->height);
+
+  if(BR_EOF(&br)) {
+    LOGDBG("H.264 SPS: not enough data ?");
+    return 0;
+  }
+  return 1;
+}
+
+static int h264_nal_unescape(uint8_t *dst, const uint8_t *src, int len)
+{
+  int s = 0, d = 0;
+  while (s < len) {
+    if (!src[s] && !src[s+1]) {
+      /* hit 00 00 xx */
+      dst[d] = dst[d+1] = 0;
+      s += 2;
+      d += 2;
+      if (src[s] == 3) {
+	s++; /* 00 00 03 xx --> 00 00 xx */
+	/*LOGDBG("h264_nal_unescape: hit 00 00 03 %02x", src[s]);*/
+	if (s >= len)
+	  return d;
+      } /* else if (src[s] == 0 || src[s] == 1) {
+	LOGDBG("h264_nal_unescape: invalid NAL sequence 00 00 %02x %02x", src[s], src[s+1]);
+	return -1;
+      }*/
+    }
+    dst[d++] = src[s++];
+  }
+  return d;
+}
+
+int h264_get_picture_type(const uint8_t *buf, int len)
+{
+  int i;
+  for (i = 0; i < len-5; i++) {
+    if (buf[i] == 0 && buf[i + 1] == 0 && buf[i + 2] == 1 && buf[i + 3] == 9) {
+      uint8_t type = (buf[i + 4] >> 5);
+      switch (type) {
+        case 0: case 3: case 5: return I_FRAME;
+        case 1: case 4: case 6: return P_FRAME;
+        case 2: case 7:         return B_FRAME;
+        default:;
+      }
+    }
+  }
+  return NO_PICTURE;
+}
+
+int h264_get_video_size(const uint8_t *buf, int len, video_size_t *size)
+{
+  int i;
+
+  /* H.264 detection, search for NAL AUD */
+  if (!(buf[0] == 0 && buf[1] == 0 && buf[2] == 1 && buf[3] == 0x09))
+    return 0;
+
+  /* if I-frame, search for NAL SPS */
+  if (h264_get_picture_type(buf, len) != I_FRAME)
+    return 0;
+
+  /* scan video packet for sequence parameter set */
+  for (i = 5; i < len-4; i++) 
+    if (buf[i] == 0 && buf[i + 1] == 0 && buf[i + 2] == 1 && (buf[i + 3] & 0x1f) == 0x07) {
+
+      h264_sps_data_t sps = {0};
+      uint8_t nal_data[len];
+      int     nal_len;
+
+      LOGDBG("H.264: Found NAL SPS at offset %d/%d", i, len);
+
+      if (0 < (nal_len = h264_nal_unescape(nal_data, buf+i+4, len-i-4))) {
+
+	if (h264_parse_sps(nal_data, nal_len, &sps)) {
+	  size->width  = sps.width;
+	  size->height = sps.height;
+	  return 1;
+	}
+	LOGMSG("h264_get_video_size: not enough data ?");
+      }
+    }
+
+  return 0;
+}
+
diff --git a/tools/h264.h b/tools/h264.h
new file mode 100644
index 00000000..2e14136b
--- /dev/null
+++ b/tools/h264.h
@@ -0,0 +1,55 @@
+/*
+ * h264.h: H.264 bitstream decoding
+ *
+ * See the main source file 'xineliboutput.c' for copyright information and
+ * how to reach the author.
+ *
+ * $Id: h264.h,v 1.1 2008-02-04 22:43:27 phintuka Exp $
+ *
+ */
+
+#ifndef _XINELIBOUTPUT_H264_H_
+#define _XINELIBOUTPUT_H264_H_
+
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+
+#ifndef VIDEO_SIZE_T
+#define VIDEO_SIZE_T
+typedef struct {
+  int width;
+  int height;
+} video_size_t;
+#endif
+
+typedef struct {
+  int width;
+  int height;
+  /* ... */
+} h264_sps_data_t;
+
+/*
+ * input: start of NAL SPS (without 00 00 01 07)
+ */
+int h264_parse_sps(const uint8_t *buf, int len, h264_sps_data_t *sps);
+
+/*
+ * input: start of H.264 video data (not PES)
+ */
+int  h264_get_picture_type(const uint8_t *buf, int len);
+
+/*
+ * input: start of H.264 video data (not PES)
+ */
+int  h264_get_video_size(const uint8_t *buf, int len, video_size_t *size);
+
+
+#ifdef __cplusplus
+} /* extern "C" { */
+#endif
+
+
+#endif /* _XINELIBOUTPUT_H264_H_ */
diff --git a/tools/mpeg.c b/tools/mpeg.c
new file mode 100644
index 00000000..35e95da3
--- /dev/null
+++ b/tools/mpeg.c
@@ -0,0 +1,51 @@
+/*
+ * mpeg.c:
+ *
+ * See the main source file 'xineliboutput.c' for copyright information and
+ * how to reach the author.
+ *
+ * $Id: mpeg.c,v 1.1 2008-02-04 22:42:13 phintuka Exp $
+ *
+ */
+
+#include "mpeg.h"
+
+
+const char * const picture_type_str[] = {
+  "(none)",
+  "I-Frame",
+  "B-Frame",
+  "P-Frame"
+};
+
+
+int mpeg2_get_picture_type(const uint8_t *buf, int len)
+{
+  int i;
+  for (i = 0; i < len-5; i++) {
+    if (buf[i] == 0 && buf[i + 1] == 0 && buf[i + 2] == 1) {
+      switch (buf[i + 3]) {
+        case SC_PICTURE:
+	  return (buf[i + 5] >> 3) & 0x07;
+      }
+    }
+  }
+  return NO_PICTURE;
+}
+
+int mpeg2_get_video_size(const uint8_t *buf, int len, video_size_t *size)
+{
+  int i;
+  for (i = 0; i < len-6; i++) {
+    if (buf[i] == 0 && buf[i + 1] == 0 && buf[i + 2] == 1) {
+      if (buf[i + 3] == SC_SEQUENCE) {
+	int d = (buf[i+4] << 16) | (buf[i+5] << 8) | buf[i+6];
+	size->width  = (d >> 12);
+	size->height = (d & 0xfff);
+	return 1;
+      }
+    }
+  }
+  return 0;
+}
+
diff --git a/tools/mpeg.h b/tools/mpeg.h
new file mode 100644
index 00000000..c752c184
--- /dev/null
+++ b/tools/mpeg.h
@@ -0,0 +1,56 @@
+/*
+ * mpeg.h: MPEG definitions
+ *
+ * See the main source file 'xineliboutput.c' for copyright information and
+ * how to reach the author.
+ *
+ * $Id: mpeg.h,v 1.1 2008-02-04 22:42:13 phintuka Exp $
+ *
+ */
+
+#ifndef XINELIBOUTPUT_MPEG_H_
+#define XINELIBOUTPUT_MPEG_H_
+
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+
+#define SC_PICTURE     0x00  /* picture atart code */
+#define SC_SEQUENCE    0xb3  /* sequence header    */
+
+/* Picture types */
+#define NO_PICTURE  0
+#define I_FRAME     1
+#define P_FRAME     2
+#define B_FRAME     3
+
+
+#ifndef VIDEO_SIZE_T
+#define VIDEO_SIZE_T
+typedef struct {
+  int width;
+  int height;
+} video_size_t;
+#endif
+
+extern const char * const picture_type_str[];
+
+/*
+ * input: start of MPEG video data (not PES)
+ */
+int mpeg2_get_picture_type(const uint8_t *buf, int len);
+
+/*
+ * input: start of MPEG video data (not PES)
+ */
+int mpeg2_get_video_size(const uint8_t *buf, int len, video_size_t *size);
+
+
+#ifdef __cplusplus
+} /* extern "C" { */
+#endif
+
+
+#endif