Merge commit 'fa936a307f5cddfc2664600157a8207ca8080af6'

* commit 'fa936a307f5cddfc2664600157a8207ca8080af6':
  hevc_parse: rename into h2645_parse

Merged-by: Derek Buitenhuis <derek.buitenhuis@gmail.com>
This commit is contained in:
Derek Buitenhuis 2016-04-26 13:38:45 +01:00
commit 3c4ca4c5d7
7 changed files with 99 additions and 66 deletions

View File

@ -318,9 +318,9 @@ OBJS-$(CONFIG_HAP_DECODER) += hapdec.o hap.o
OBJS-$(CONFIG_HAP_ENCODER) += hapenc.o hap.o OBJS-$(CONFIG_HAP_ENCODER) += hapenc.o hap.o
OBJS-$(CONFIG_HEVC_DECODER) += hevc.o hevc_mvs.o hevc_ps.o hevc_sei.o \ OBJS-$(CONFIG_HEVC_DECODER) += hevc.o hevc_mvs.o hevc_ps.o hevc_sei.o \
hevc_cabac.o hevc_refs.o hevcpred.o \ hevc_cabac.o hevc_refs.o hevcpred.o \
hevcdsp.o hevc_filter.o hevc_parse.o hevc_data.o hevcdsp.o hevc_filter.o h2645_parse.o hevc_data.o
OBJS-$(CONFIG_HEVC_QSV_DECODER) += qsvdec_h2645.o OBJS-$(CONFIG_HEVC_QSV_DECODER) += qsvdec_h2645.o
OBJS-$(CONFIG_HEVC_QSV_ENCODER) += qsvenc_hevc.o hevc_ps_enc.o hevc_parse.o OBJS-$(CONFIG_HEVC_QSV_ENCODER) += qsvenc_hevc.o hevc_ps_enc.o h2645_parse.o
OBJS-$(CONFIG_HNM4_VIDEO_DECODER) += hnm4video.o OBJS-$(CONFIG_HNM4_VIDEO_DECODER) += hnm4video.o
OBJS-$(CONFIG_HQ_HQA_DECODER) += hq_hqa.o hq_hqadata.o hq_hqadsp.o \ OBJS-$(CONFIG_HQ_HQA_DECODER) += hq_hqa.o hq_hqadata.o hq_hqadsp.o \
canopus.o canopus.o
@ -901,7 +901,7 @@ OBJS-$(CONFIG_GSM_PARSER) += gsm_parser.o
OBJS-$(CONFIG_H261_PARSER) += h261_parser.o OBJS-$(CONFIG_H261_PARSER) += h261_parser.o
OBJS-$(CONFIG_H263_PARSER) += h263_parser.o OBJS-$(CONFIG_H263_PARSER) += h263_parser.o
OBJS-$(CONFIG_H264_PARSER) += h264_parser.o OBJS-$(CONFIG_H264_PARSER) += h264_parser.o
OBJS-$(CONFIG_HEVC_PARSER) += hevc_parser.o hevc_parse.o hevc_ps.o hevc_data.o OBJS-$(CONFIG_HEVC_PARSER) += hevc_parser.o h2645_parse.o hevc_ps.o hevc_data.o
OBJS-$(CONFIG_MJPEG_PARSER) += mjpeg_parser.o OBJS-$(CONFIG_MJPEG_PARSER) += mjpeg_parser.o
OBJS-$(CONFIG_MLP_PARSER) += mlp_parser.o mlp.o OBJS-$(CONFIG_MLP_PARSER) += mlp_parser.o mlp.o
OBJS-$(CONFIG_MPEG4VIDEO_PARSER) += mpeg4video_parser.o h263.o \ OBJS-$(CONFIG_MPEG4VIDEO_PARSER) += mpeg4video_parser.o h263.o \

View File

@ -1,5 +1,5 @@
/* /*
* HEVC common code * H.264/HEVC common parsing code
* *
* This file is part of FFmpeg. * This file is part of FFmpeg.
* *
@ -26,11 +26,12 @@
#include "libavutil/mem.h" #include "libavutil/mem.h"
#include "hevc.h" #include "hevc.h"
#include "h2645_parse.h"
/* FIXME: This is adapted from ff_h264_decode_nal, avoiding duplication /* FIXME: This is adapted from ff_h264_decode_nal, avoiding duplication
* between these functions would be nice. */ * between these functions would be nice. */
int ff_hevc_extract_rbsp(const uint8_t *src, int length, int ff_h2645_extract_rbsp(const uint8_t *src, int length,
HEVCNAL *nal) H2645NAL *nal)
{ {
int i, si, di; int i, si, di;
uint8_t *dst; uint8_t *dst;
@ -181,7 +182,7 @@ static const char *nal_unit_name(int nal_type)
* @return AVERROR_INVALIDDATA if the packet is not a valid NAL unit, * @return AVERROR_INVALIDDATA if the packet is not a valid NAL unit,
* 0 if the unit should be skipped, 1 otherwise * 0 if the unit should be skipped, 1 otherwise
*/ */
static int hls_nal_unit(HEVCNAL *nal, AVCodecContext *avctx) static int hevc_parse_nal_header(H2645NAL *nal, AVCodecContext *avctx)
{ {
GetBitContext *gb = &nal->gb; GetBitContext *gb = &nal->gb;
int nuh_layer_id; int nuh_layer_id;
@ -204,14 +205,14 @@ static int hls_nal_unit(HEVCNAL *nal, AVCodecContext *avctx)
} }
int ff_hevc_split_packet(HEVCPacket *pkt, const uint8_t *buf, int length, int ff_h2645_packet_split(H2645Packet *pkt, const uint8_t *buf, int length,
AVCodecContext *avctx, int is_nalff, int nal_length_size) AVCodecContext *avctx, int is_nalff, int nal_length_size)
{ {
int consumed, ret = 0; int consumed, ret = 0;
pkt->nb_nals = 0; pkt->nb_nals = 0;
while (length >= 4) { while (length >= 4) {
HEVCNAL *nal; H2645NAL *nal;
int extract_length = 0; int extract_length = 0;
if (is_nalff) { if (is_nalff) {
@ -268,7 +269,7 @@ int ff_hevc_split_packet(HEVCPacket *pkt, const uint8_t *buf, int length,
} }
nal = &pkt->nals[pkt->nb_nals]; nal = &pkt->nals[pkt->nb_nals];
consumed = ff_hevc_extract_rbsp(buf, extract_length, nal); consumed = ff_h2645_extract_rbsp(buf, extract_length, nal);
if (consumed < 0) if (consumed < 0)
return consumed; return consumed;
@ -278,7 +279,7 @@ int ff_hevc_split_packet(HEVCPacket *pkt, const uint8_t *buf, int length,
if (ret < 0) if (ret < 0)
return ret; return ret;
ret = hls_nal_unit(nal, avctx); ret = hevc_parse_nal_header(nal, avctx);
if (ret <= 0) { if (ret <= 0) {
if (ret < 0) { if (ret < 0) {
av_log(avctx, AV_LOG_ERROR, "Invalid NAL unit %d, skipping.\n", av_log(avctx, AV_LOG_ERROR, "Invalid NAL unit %d, skipping.\n",

68
libavcodec/h2645_parse.h Normal file
View File

@ -0,0 +1,68 @@
/*
* H.264/HEVC common parsing code
*
* 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
*/
#ifndef AVCODEC_H2645_PARSE_H
#define AVCODEC_H2645_PARSE_H
#include <stdint.h>
#include "avcodec.h"
#include "get_bits.h"
typedef struct H2645NAL {
uint8_t *rbsp_buffer;
int rbsp_buffer_size;
int size;
const uint8_t *data;
int raw_size;
const uint8_t *raw_data;
GetBitContext gb;
int type;
int temporal_id;
int skipped_bytes;
int skipped_bytes_pos_size;
int *skipped_bytes_pos;
} H2645NAL;
/* an input packet split into unescaped NAL units */
typedef struct H2645Packet {
H2645NAL *nals;
int nb_nals;
int nals_allocated;
} H2645Packet;
/**
* Extract the raw (unescaped) bitstream.
*/
int ff_h2645_extract_rbsp(const uint8_t *src, int length,
H2645NAL *nal);
/**
* Split an input packet into NAL units.
*/
int ff_h2645_packet_split(H2645Packet *pkt, const uint8_t *buf, int length,
AVCodecContext *avctx, int is_nalff, int nal_length_size);
#endif /* AVCODEC_H2645_PARSE_H */

View File

@ -2448,7 +2448,7 @@ static int hls_decode_entry_wpp(AVCodecContext *avctxt, void *input_ctb_row, int
return 0; return 0;
} }
static int hls_slice_data_wpp(HEVCContext *s, const HEVCNAL *nal) static int hls_slice_data_wpp(HEVCContext *s, const H2645NAL *nal)
{ {
const uint8_t *data = nal->data; const uint8_t *data = nal->data;
int length = nal->size; int length = nal->size;
@ -2714,7 +2714,7 @@ fail:
return ret; return ret;
} }
static int decode_nal_unit(HEVCContext *s, const HEVCNAL *nal) static int decode_nal_unit(HEVCContext *s, const H2645NAL *nal)
{ {
HEVCLocalContext *lc = s->HEVClc; HEVCLocalContext *lc = s->HEVClc;
GetBitContext *gb = &lc->gb; GetBitContext *gb = &lc->gb;
@ -2866,8 +2866,8 @@ static int decode_nal_units(HEVCContext *s, const uint8_t *buf, int length)
/* split the input packet into NAL units, so we know the upper bound on the /* split the input packet into NAL units, so we know the upper bound on the
* number of slices in the frame */ * number of slices in the frame */
ret = ff_hevc_split_packet(&s->pkt, buf, length, s->avctx, s->is_nalff, ret = ff_h2645_packet_split(&s->pkt, buf, length, s->avctx, s->is_nalff,
s->nal_length_size); s->nal_length_size);
if (ret < 0) { if (ret < 0) {
av_log(s->avctx, AV_LOG_ERROR, av_log(s->avctx, AV_LOG_ERROR,
"Error splitting the input into NAL units.\n"); "Error splitting the input into NAL units.\n");

View File

@ -31,6 +31,7 @@
#include "cabac.h" #include "cabac.h"
#include "get_bits.h" #include "get_bits.h"
#include "hevcpred.h" #include "hevcpred.h"
#include "h2645_parse.h"
#include "hevcdsp.h" #include "hevcdsp.h"
#include "internal.h" #include "internal.h"
#include "thread.h" #include "thread.h"
@ -746,33 +747,6 @@ typedef struct HEVCFrame {
uint8_t flags; uint8_t flags;
} HEVCFrame; } HEVCFrame;
typedef struct HEVCNAL {
uint8_t *rbsp_buffer;
int rbsp_buffer_size;
int size;
const uint8_t *data;
int raw_size;
const uint8_t *raw_data;
GetBitContext gb;
enum NALUnitType type;
int temporal_id;
int skipped_bytes;
int skipped_bytes_pos_size;
int *skipped_bytes_pos;
} HEVCNAL;
/* an input packet split into unescaped NAL units */
typedef struct HEVCPacket {
HEVCNAL *nals;
int nb_nals;
int nals_allocated;
} HEVCPacket;
typedef struct HEVCLocalContext { typedef struct HEVCLocalContext {
uint8_t cabac_state[HEVC_CONTEXTS]; uint8_t cabac_state[HEVC_CONTEXTS];
@ -906,7 +880,7 @@ typedef struct HEVCContext {
const uint8_t *data; const uint8_t *data;
HEVCPacket pkt; H2645Packet pkt;
// type of the first VCL NAL of the current frame // type of the first VCL NAL of the current frame
enum NALUnitType first_nal_type; enum NALUnitType first_nal_type;
@ -1077,18 +1051,6 @@ void ff_hevc_hls_residual_coding(HEVCContext *s, int x0, int y0,
void ff_hevc_hls_mvd_coding(HEVCContext *s, int x0, int y0, int log2_cb_size); void ff_hevc_hls_mvd_coding(HEVCContext *s, int x0, int y0, int log2_cb_size);
/**
* Extract the raw (unescaped) HEVC bitstream.
*/
int ff_hevc_extract_rbsp(const uint8_t *src, int length,
HEVCNAL *nal);
/**
* Split an input packet into NAL units.
*/
int ff_hevc_split_packet(HEVCPacket *pkt, const uint8_t *buf, int length,
AVCodecContext *avctx, int is_nalff, int nal_length_size);
int ff_hevc_encode_nal_vps(HEVCVPS *vps, unsigned int id, int ff_hevc_encode_nal_vps(HEVCVPS *vps, unsigned int id,
uint8_t *buf, int buf_size); uint8_t *buf, int buf_size);

View File

@ -24,6 +24,7 @@
#include "golomb.h" #include "golomb.h"
#include "hevc.h" #include "hevc.h"
#include "h2645_parse.h"
#include "parser.h" #include "parser.h"
#define START_CODE 0x000001 ///< start_code_prefix_one_3bytes #define START_CODE 0x000001 ///< start_code_prefix_one_3bytes
@ -35,7 +36,7 @@
typedef struct HEVCParserContext { typedef struct HEVCParserContext {
ParseContext pc; ParseContext pc;
HEVCPacket pkt; H2645Packet pkt;
HEVCParamSets ps; HEVCParamSets ps;
int parsed_extradata; int parsed_extradata;
@ -46,7 +47,7 @@ typedef struct HEVCParserContext {
} HEVCParserContext; } HEVCParserContext;
#if !ADVANCED_PARSER #if !ADVANCED_PARSER
static int hevc_parse_slice_header(AVCodecParserContext *s, HEVCNAL *nal, static int hevc_parse_slice_header(AVCodecParserContext *s, H2645NAL *nal,
AVCodecContext *avctx) AVCodecContext *avctx)
{ {
HEVCParserContext *ctx = s->priv_data; HEVCParserContext *ctx = s->priv_data;
@ -88,12 +89,12 @@ static int parse_nal_units(AVCodecParserContext *s, const uint8_t *buf,
HEVCParserContext *ctx = s->priv_data; HEVCParserContext *ctx = s->priv_data;
int ret, i; int ret, i;
ret = ff_hevc_split_packet(NULL, &ctx->pkt, buf, buf_size, avctx, 0, 0); ret = ff_h2645_split_packet(&ctx->pkt, buf, buf_size, avctx, 0, 0);
if (ret < 0) if (ret < 0)
return ret; return ret;
for (i = 0; i < ctx->pkt.nb_nals; i++) { for (i = 0; i < ctx->pkt.nb_nals; i++) {
HEVCNAL *nal = &ctx->pkt.nals[i]; H2645NAL *nal = &ctx->pkt.nals[i];
/* ignore everything except parameter sets and VCL NALUs */ /* ignore everything except parameter sets and VCL NALUs */
switch (nal->type) { switch (nal->type) {
@ -189,10 +190,10 @@ static inline int parse_nal_units(AVCodecParserContext *s, const uint8_t *buf,
GetBitContext *gb; GetBitContext *gb;
SliceHeader *sh = &h->sh; SliceHeader *sh = &h->sh;
HEVCParamSets *ps = &h->ps; HEVCParamSets *ps = &h->ps;
HEVCPacket *pkt = &ctx->pkt; H2645Packet *pkt = &ctx->pkt;
const uint8_t *buf_end = buf + buf_size; const uint8_t *buf_end = buf + buf_size;
int state = -1, i; int state = -1, i;
HEVCNAL *nal; H2645NAL *nal;
int is_global = buf == avctx->extradata; int is_global = buf == avctx->extradata;
if (!h->HEVClc) if (!h->HEVClc)
@ -213,7 +214,7 @@ static inline int parse_nal_units(AVCodecParserContext *s, const uint8_t *buf,
return 0; return 0;
if (pkt->nals_allocated < 1) { if (pkt->nals_allocated < 1) {
HEVCNAL *tmp = av_realloc_array(pkt->nals, 1, sizeof(*tmp)); H2645NAL *tmp = av_realloc_array(pkt->nals, 1, sizeof(*tmp));
if (!tmp) if (!tmp)
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
pkt->nals = tmp; pkt->nals = tmp;
@ -239,7 +240,7 @@ static inline int parse_nal_units(AVCodecParserContext *s, const uint8_t *buf,
src_length = 20; src_length = 20;
} }
consumed = ff_hevc_extract_rbsp(buf, src_length, nal); consumed = ff_h2645_extract_rbsp(buf, src_length, nal);
if (consumed < 0) if (consumed < 0)
return consumed; return consumed;

View File

@ -31,6 +31,7 @@
#include "bytestream.h" #include "bytestream.h"
#include "get_bits.h" #include "get_bits.h"
#include "hevc.h" #include "hevc.h"
#include "h2645_parse.h"
#include "internal.h" #include "internal.h"
#include "qsv.h" #include "qsv.h"
#include "qsv_internal.h" #include "qsv_internal.h"
@ -54,7 +55,7 @@ static int generate_fake_vps(QSVEncContext *q, AVCodecContext *avctx)
PutByteContext pbc; PutByteContext pbc;
GetBitContext gb; GetBitContext gb;
HEVCNAL sps_nal = { NULL }; H2645NAL sps_nal = { NULL };
HEVCSPS sps = { 0 }; HEVCSPS sps = { 0 };
HEVCVPS vps = { 0 }; HEVCVPS vps = { 0 };
uint8_t vps_buf[128], vps_rbsp_buf[128]; uint8_t vps_buf[128], vps_rbsp_buf[128];
@ -68,7 +69,7 @@ static int generate_fake_vps(QSVEncContext *q, AVCodecContext *avctx)
} }
/* parse the SPS */ /* parse the SPS */
ret = ff_hevc_extract_rbsp(avctx->extradata + 4, avctx->extradata_size - 4, &sps_nal); ret = ff_h2645_extract_rbsp(avctx->extradata + 4, avctx->extradata_size - 4, &sps_nal);
if (ret < 0) { if (ret < 0) {
av_log(avctx, AV_LOG_ERROR, "Error unescaping the SPS buffer\n"); av_log(avctx, AV_LOG_ERROR, "Error unescaping the SPS buffer\n");
return ret; return ret;