summaryrefslogtreecommitdiff
path: root/libavcodec/hevc_ps_enc.c
diff options
context:
space:
mode:
authorMichael Niedermayer <michaelni@gmx.at>2015-07-09 11:52:05 +0200
committerMichael Niedermayer <michaelni@gmx.at>2015-07-09 11:53:14 +0200
commit7871eb43616776626ec9f9075b6d0a8cf73d159e (patch)
treef9a88703414cd33e32c5f682974d8b4f96837b92 /libavcodec/hevc_ps_enc.c
parent587980eb7a1f4b0228204aa846b72dc761e49779 (diff)
parent66acb76bb0492b263215ca9b4d927a7be39ace02 (diff)
downloadffmpeg-7871eb43616776626ec9f9075b6d0a8cf73d159e.tar.gz
Merge commit '66acb76bb0492b263215ca9b4d927a7be39ace02'
* commit '66acb76bb0492b263215ca9b4d927a7be39ace02': lavc: add Intel libmfx-based HEVC encoder Conflicts: Changelog configure libavcodec/Makefile libavcodec/allcodecs.c libavcodec/qsv.c libavcodec/qsvenc.c libavcodec/version.h Merged-by: Michael Niedermayer <michaelni@gmx.at>
Diffstat (limited to 'libavcodec/hevc_ps_enc.c')
-rw-r--r--libavcodec/hevc_ps_enc.c116
1 files changed, 116 insertions, 0 deletions
diff --git a/libavcodec/hevc_ps_enc.c b/libavcodec/hevc_ps_enc.c
new file mode 100644
index 0000000000..c05bf63d3b
--- /dev/null
+++ b/libavcodec/hevc_ps_enc.c
@@ -0,0 +1,116 @@
+/*
+ * HEVC Parameter Set encoding
+ *
+ * This file is part of FFmpeg.
+ *
+ * FFmpeg is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * FFmpeg 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
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with FFmpeg; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+
+#include "golomb.h"
+#include "hevc.h"
+#include "put_bits.h"
+
+static void write_ptl_layer(PutBitContext *pb, PTLCommon *ptl)
+{
+ int i;
+
+ put_bits(pb, 2, ptl->profile_space);
+ put_bits(pb, 1, ptl->tier_flag);
+ put_bits(pb, 5, ptl->profile_idc);
+ for (i = 0; i < 32; i++)
+ put_bits(pb, 1, ptl->profile_compatibility_flag[i]);
+ put_bits(pb, 1, ptl->progressive_source_flag);
+ put_bits(pb, 1, ptl->interlaced_source_flag);
+ put_bits(pb, 1, ptl->non_packed_constraint_flag);
+ put_bits(pb, 1, ptl->frame_only_constraint_flag);
+ put_bits32(pb, 0); // reserved
+ put_bits(pb, 12, 0); // reserved
+}
+
+static void write_ptl(PutBitContext *pb, PTL *ptl, int max_num_sub_layers)
+{
+ int i;
+
+ write_ptl_layer(pb, &ptl->general_ptl);
+ put_bits(pb, 8, ptl->general_ptl.level_idc);
+
+ for (i = 0; i < max_num_sub_layers - 1; i++) {
+ put_bits(pb, 1, ptl->sub_layer_profile_present_flag[i]);
+ put_bits(pb, 1, ptl->sub_layer_level_present_flag[i]);
+ }
+
+ if (max_num_sub_layers > 1)
+ for (i = max_num_sub_layers - 1; i < 8; i++)
+ put_bits(pb, 2, 0); // reserved
+
+ for (i = 0; i < max_num_sub_layers - 1; i++) {
+ if (ptl->sub_layer_profile_present_flag[i])
+ write_ptl_layer(pb, &ptl->sub_layer_ptl[i]);
+ if (ptl->sub_layer_level_present_flag[i])
+ put_bits(pb, 8, ptl->sub_layer_ptl[i].level_idc);
+ }
+}
+
+int ff_hevc_encode_nal_vps(HEVCVPS *vps, unsigned int id,
+ uint8_t *buf, int buf_size)
+{
+ PutBitContext pb;
+ int i;
+
+ init_put_bits(&pb, buf, buf_size);
+ put_bits(&pb, 4, id);
+ put_bits(&pb, 2, 3); // reserved
+ put_bits(&pb, 6, vps->vps_max_layers - 1);
+ put_bits(&pb, 3, vps->vps_max_sub_layers - 1);
+ put_bits(&pb, 1, vps->vps_temporal_id_nesting_flag);
+ put_bits(&pb, 16, 0xffff); // reserved
+
+ write_ptl(&pb, &vps->ptl, vps->vps_max_sub_layers);
+
+ put_bits(&pb, 1, vps->vps_sub_layer_ordering_info_present_flag);
+ for (i = vps->vps_sub_layer_ordering_info_present_flag ? 0 : vps->vps_max_layers - 1;
+ i < vps->vps_max_sub_layers; i++) {
+ set_ue_golomb(&pb, vps->vps_max_dec_pic_buffering[i] - 1);
+ set_ue_golomb(&pb, vps->vps_num_reorder_pics[i]);
+ set_ue_golomb(&pb, vps->vps_max_latency_increase[i] + 1);
+ }
+
+ put_bits(&pb, 6, vps->vps_max_layer_id);
+ set_ue_golomb(&pb, vps->vps_num_layer_sets - 1);
+
+ // writing layer_id_included_flag not supported
+ if (vps->vps_num_layer_sets > 1)
+ return AVERROR_PATCHWELCOME;
+
+ put_bits(&pb, 1, vps->vps_timing_info_present_flag);
+ if (vps->vps_timing_info_present_flag) {
+ put_bits32(&pb, vps->vps_num_units_in_tick);
+ put_bits32(&pb, vps->vps_time_scale);
+ put_bits(&pb, 1, vps->vps_poc_proportional_to_timing_flag);
+ if (vps->vps_poc_proportional_to_timing_flag)
+ set_ue_golomb(&pb, vps->vps_num_ticks_poc_diff_one - 1);
+
+ // writing HRD parameters not supported
+ if (vps->vps_num_hrd_parameters)
+ return AVERROR_PATCHWELCOME;
+ }
+
+ put_bits(&pb, 1, 0); // extension flag
+
+ put_bits(&pb, 1, 1); // stop bit
+ avpriv_align_put_bits(&pb);
+
+ return put_bits_count(&pb) / 8;
+}