1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
/*
* Copyright (C) 2010-2018 Team Kodi
* This file is part of Kodi - https://kodi.tv
*
* SPDX-License-Identifier: GPL-2.0-or-later
* See LICENSES/README.md for more information.
*/
#pragma once
#include <stdint.h>
extern "C" {
#include <libavutil/avutil.h>
#include <libavformat/avformat.h>
#include <libavfilter/avfilter.h>
#include <libavcodec/avcodec.h>
}
typedef struct
{
const uint8_t *data;
const uint8_t *end;
int head;
uint64_t cache;
} nal_bitstream;
typedef struct mpeg2_sequence
{
uint32_t width;
uint32_t height;
uint32_t fps_rate;
uint32_t fps_scale;
float ratio;
uint32_t ratio_info;
} mpeg2_sequence;
typedef struct
{
int profile_idc;
int level_idc;
int sps_id;
int chroma_format_idc;
int separate_colour_plane_flag;
int bit_depth_luma_minus8;
int bit_depth_chroma_minus8;
int qpprime_y_zero_transform_bypass_flag;
int seq_scaling_matrix_present_flag;
int log2_max_frame_num_minus4;
int pic_order_cnt_type;
int log2_max_pic_order_cnt_lsb_minus4;
int max_num_ref_frames;
int gaps_in_frame_num_value_allowed_flag;
int pic_width_in_mbs_minus1;
int pic_height_in_map_units_minus1;
int frame_mbs_only_flag;
int mb_adaptive_frame_field_flag;
int direct_8x8_inference_flag;
int frame_cropping_flag;
int frame_crop_left_offset;
int frame_crop_right_offset;
int frame_crop_top_offset;
int frame_crop_bottom_offset;
} sps_info_struct;
class CBitstreamParser
{
public:
CBitstreamParser();
~CBitstreamParser() = default;
static bool Open(){ return true; };
static void Close();
static bool CanStartDecode(const uint8_t *buf, int buf_size);
};
class CBitstreamConverter
{
public:
CBitstreamConverter();
~CBitstreamConverter();
bool Open(enum AVCodecID codec, uint8_t *in_extradata, int in_extrasize, bool to_annexb);
void Close(void);
bool NeedConvert(void) const { return m_convert_bitstream; };
bool Convert(uint8_t *pData, int iSize);
uint8_t* GetConvertBuffer(void) const;
int GetConvertSize() const;
uint8_t* GetExtraData(void) const;
int GetExtraSize() const;
void ResetStartDecode(void);
bool CanStartDecode() const;
static bool mpeg2_sequence_header(const uint8_t *data, const uint32_t size, mpeg2_sequence *sequence);
protected:
static int avc_parse_nal_units(AVIOContext *pb, const uint8_t *buf_in, int size);
static int avc_parse_nal_units_buf(const uint8_t *buf_in, uint8_t **buf, int *size);
int isom_write_avcc(AVIOContext *pb, const uint8_t *data, int len);
// bitstream to bytestream (Annex B) conversion support.
bool IsIDR(uint8_t unit_type);
bool IsSlice(uint8_t unit_type);
bool BitstreamConvertInitAVC(void *in_extradata, int in_extrasize);
bool BitstreamConvertInitHEVC(void *in_extradata, int in_extrasize);
bool BitstreamConvert(uint8_t* pData, int iSize, uint8_t **poutbuf, int *poutbuf_size);
static void BitstreamAllocAndCopy(uint8_t **poutbuf, int *poutbuf_size,
const uint8_t *sps_pps, uint32_t sps_pps_size, const uint8_t *in, uint32_t in_size);
typedef struct omx_bitstream_ctx {
uint8_t length_size;
uint8_t first_idr;
uint8_t idr_sps_pps_seen;
uint8_t *sps_pps_data;
uint32_t size;
} omx_bitstream_ctx;
uint8_t *m_convertBuffer;
int m_convertSize;
uint8_t *m_inputBuffer;
int m_inputSize;
uint32_t m_sps_pps_size;
omx_bitstream_ctx m_sps_pps_context;
bool m_convert_bitstream;
bool m_to_annexb;
uint8_t *m_extradata;
int m_extrasize;
bool m_convert_3byteTo4byteNALSize;
bool m_convert_bytestream;
AVCodecID m_codec;
bool m_start_decode;
};
|
