mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-23 12:43:46 +02:00
hevc_parse: rename into h2645_parse
This code will be shared with h264.
This commit is contained in:
parent
cdb1665f70
commit
fa936a307f
@ -261,10 +261,10 @@ 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_NVENC_ENCODER) += nvenc_hevc.o
|
OBJS-$(CONFIG_HEVC_NVENC_ENCODER) += nvenc_hevc.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
|
||||||
@ -719,7 +719,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 \
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
/*
|
/*
|
||||||
* HEVC common code
|
* H.264/HEVC common parsing code
|
||||||
*
|
*
|
||||||
* This file is part of Libav.
|
* This file is part of Libav.
|
||||||
*
|
*
|
||||||
@ -25,12 +25,12 @@
|
|||||||
#include "libavutil/intreadwrite.h"
|
#include "libavutil/intreadwrite.h"
|
||||||
#include "libavutil/mem.h"
|
#include "libavutil/mem.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;
|
||||||
@ -132,7 +132,7 @@ nsc:
|
|||||||
* @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;
|
||||||
@ -155,14 +155,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) {
|
||||||
@ -192,7 +192,7 @@ int ff_hevc_split_packet(HEVCPacket *pkt, const uint8_t *buf, int length,
|
|||||||
|
|
||||||
if (pkt->nals_allocated < pkt->nb_nals + 1) {
|
if (pkt->nals_allocated < pkt->nb_nals + 1) {
|
||||||
int new_size = pkt->nals_allocated + 1;
|
int new_size = pkt->nals_allocated + 1;
|
||||||
HEVCNAL *tmp = av_realloc_array(pkt->nals, new_size, sizeof(*tmp));
|
H2645NAL *tmp = av_realloc_array(pkt->nals, new_size, sizeof(*tmp));
|
||||||
if (!tmp)
|
if (!tmp)
|
||||||
return AVERROR(ENOMEM);
|
return AVERROR(ENOMEM);
|
||||||
|
|
||||||
@ -203,7 +203,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;
|
||||||
|
|
||||||
@ -211,7 +211,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",
|
64
libavcodec/h2645_parse.h
Normal file
64
libavcodec/h2645_parse.h
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
/*
|
||||||
|
* H.264/HEVC common parsing code
|
||||||
|
*
|
||||||
|
* This file is part of Libav.
|
||||||
|
*
|
||||||
|
* Libav 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.
|
||||||
|
*
|
||||||
|
* Libav 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 Libav; 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;
|
||||||
|
} 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 */
|
@ -2429,7 +2429,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;
|
||||||
@ -2581,8 +2581,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");
|
||||||
|
@ -33,6 +33,7 @@
|
|||||||
#include "bswapdsp.h"
|
#include "bswapdsp.h"
|
||||||
#include "cabac.h"
|
#include "cabac.h"
|
||||||
#include "get_bits.h"
|
#include "get_bits.h"
|
||||||
|
#include "h2645_parse.h"
|
||||||
#include "hevcdsp.h"
|
#include "hevcdsp.h"
|
||||||
#include "internal.h"
|
#include "internal.h"
|
||||||
#include "thread.h"
|
#include "thread.h"
|
||||||
@ -704,29 +705,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;
|
|
||||||
} HEVCNAL;
|
|
||||||
|
|
||||||
/* an input packet split into unescaped NAL units */
|
|
||||||
typedef struct HEVCPacket {
|
|
||||||
HEVCNAL *nals;
|
|
||||||
int nb_nals;
|
|
||||||
int nals_allocated;
|
|
||||||
} HEVCPacket;
|
|
||||||
|
|
||||||
struct HEVCContext;
|
struct HEVCContext;
|
||||||
|
|
||||||
typedef struct HEVCPredContext {
|
typedef struct HEVCPredContext {
|
||||||
@ -852,7 +830,7 @@ typedef struct HEVCContext {
|
|||||||
uint16_t seq_decode;
|
uint16_t seq_decode;
|
||||||
uint16_t seq_output;
|
uint16_t seq_output;
|
||||||
|
|
||||||
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;
|
||||||
|
|
||||||
@ -1023,18 +1001,6 @@ void ff_hevc_pps_free(HEVCPPS **ppps);
|
|||||||
|
|
||||||
void ff_hevc_pred_init(HEVCPredContext *hpc, int bit_depth);
|
void ff_hevc_pred_init(HEVCPredContext *hpc, int bit_depth);
|
||||||
|
|
||||||
/**
|
|
||||||
* 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);
|
||||||
|
|
||||||
|
@ -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
|
||||||
@ -33,13 +34,13 @@
|
|||||||
typedef struct HEVCParserContext {
|
typedef struct HEVCParserContext {
|
||||||
ParseContext pc;
|
ParseContext pc;
|
||||||
|
|
||||||
HEVCPacket pkt;
|
H2645Packet pkt;
|
||||||
HEVCParamSets ps;
|
HEVCParamSets ps;
|
||||||
|
|
||||||
int parsed_extradata;
|
int parsed_extradata;
|
||||||
} HEVCParserContext;
|
} HEVCParserContext;
|
||||||
|
|
||||||
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;
|
||||||
@ -81,12 +82,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(&ctx->pkt, buf, buf_size, avctx, 0, 0);
|
ret = ff_h2645_packet_split(&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) {
|
||||||
|
@ -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;
|
||||||
|
Loading…
Reference in New Issue
Block a user