Added initial AVC parser. (#176)
* * rocDecode/AVC: Added initial AVC parser.
- Added AVC SPS, PPS and slice header parsing functions.
- Merged common parsing code from AVC and HEVC parsers to their super class.
* * rocDecode/AVC: Formating changes. No functional changes.
[ROCm/rocdecode commit: 395a1249c1]
This commit is contained in:
jeffqjiangNew
GitHub
@@ -0,0 +1,153 @@
|
||||
/*
|
||||
Copyright (c) 2023 - 2024 Advanced Micro Devices, Inc. All rights reserved.
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in
|
||||
all copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
THE SOFTWARE.
|
||||
*/
|
||||
#pragma once
|
||||
|
||||
#include "avc_defines.h"
|
||||
#include "roc_video_parser.h"
|
||||
|
||||
class AvcVideoParser : public RocVideoParser {
|
||||
|
||||
public:
|
||||
/*! \brief AvcVideoParser constructor
|
||||
*/
|
||||
AvcVideoParser();
|
||||
|
||||
/*! \brief AvcVideoParser destructor
|
||||
*/
|
||||
virtual ~AvcVideoParser();
|
||||
|
||||
/*! \brief Function to Initialize the parser
|
||||
* \param [in] p_params Input of <tt>RocdecParserParams</tt> with codec type to initialize parser.
|
||||
* \return <tt>rocDecStatus</tt> Returns success on completion, else error code for failure
|
||||
*/
|
||||
virtual rocDecStatus Initialize(RocdecParserParams *p_params);
|
||||
|
||||
/*! \brief Function to Parse video data: Typically called from application when a demuxed picture is ready to be parsed
|
||||
* \param [in] p_data Pointer to picture data of type <tt>RocdecSourceDataPacket</tt>
|
||||
* @return <tt>rocDecStatus</tt> Returns success on completion, else error_code for failure
|
||||
*/
|
||||
virtual rocDecStatus ParseVideoData(RocdecSourceDataPacket *p_data);
|
||||
|
||||
/*! \brief function to uninitialize AVC parser
|
||||
* @return rocDecStatus
|
||||
*/
|
||||
virtual rocDecStatus UnInitialize(); // derived method
|
||||
|
||||
|
||||
protected:
|
||||
/*! \brief Decoded picture buffer
|
||||
*/
|
||||
typedef struct{
|
||||
uint32_t dpb_size; // DPB buffer size in number of frames
|
||||
} DecodedPictureBuffer;
|
||||
|
||||
AvcNalUnitHeader nal_unit_header_;
|
||||
AvcSeqParameterSet sps_list_[AVC_MAX_SPS_NUM];
|
||||
int32_t active_sps_id_;
|
||||
AvcPicParameterSet pps_list_[AVC_MAX_PPS_NUM];
|
||||
int32_t active_pps_id_;
|
||||
|
||||
AvcNalUnitHeader slice_nal_unit_header_;
|
||||
AvcSliceHeader slice_header_0_;
|
||||
|
||||
int prev_has_mmco_5_;
|
||||
int curr_has_mmco_5_;
|
||||
int prev_ref_pic_bottom_field_;
|
||||
int curr_ref_pic_bottom_field_;
|
||||
|
||||
// DPB
|
||||
DecodedPictureBuffer dpb_buffer_;
|
||||
|
||||
/*! \brief Function to parse one picture bit stream received from the demuxer.
|
||||
* \param [in] p_stream A pointer of <tt>uint8_t</tt> for the input stream to be parsed
|
||||
* \param [in] pic_data_size Size of the input stream
|
||||
* \return <tt>ParserResult</tt>
|
||||
*/
|
||||
ParserResult ParsePictureData(const uint8_t *p_stream, uint32_t pic_data_size);
|
||||
|
||||
/*! \brief Function to parse the NAL unit header
|
||||
* \param [in] header_byte The AVC NAL unit header byte
|
||||
* \return <tt>AvcNalUnitHeader</tt> Parsed nal header
|
||||
*/
|
||||
AvcNalUnitHeader ParseNalUnitHeader(uint8_t header_byte);
|
||||
|
||||
/*! \brief Function to parse Sequence Parameter Set
|
||||
* \param [in] p_stream A pointer of <tt>uint8_t</tt> for the input stream to be parsed
|
||||
* \param [in] size Size of the input stream
|
||||
* \return No return value
|
||||
*/
|
||||
void ParseSps(uint8_t *p_stream, size_t size);
|
||||
|
||||
/*! \brief Function to parse Picture Parameter Set
|
||||
* \param [in] p_stream A pointer of <tt>uint8_t</tt> for the input stream to be parsed
|
||||
* \param [in] size Size of the input stream
|
||||
* \return No return value
|
||||
*/
|
||||
void ParsePps(uint8_t *p_stream, size_t stream_size_in_byte);
|
||||
|
||||
/*! \brief Function to parse slice header
|
||||
* \param p_stream The pointer to the input bit stream
|
||||
* \param [in] stream_size_in_byte The byte size of the stream
|
||||
* \param [out] p_slice_header The pointer to the slice header strucutre
|
||||
* \return <tt>ParserResult</tt>
|
||||
*/
|
||||
ParserResult ParseSliceHeader(uint8_t *p_stream, size_t stream_size_in_byte, AvcSliceHeader *p_slice_header);
|
||||
|
||||
/*! \brief Function to parse a scaling list
|
||||
* \param [in] p_stream A pointer of <tt>uint8_t</tt> for the input stream to be parsed
|
||||
* \param [in/out] offset Current bit offset
|
||||
* \param [out] scaling_list Pointer to the output scaling list
|
||||
* \param [in] list_size Scaling list size
|
||||
* \param [out] use_default_scaling_matrix_flag Array of flags that indicate whether to use default values
|
||||
*/
|
||||
void GetScalingList(uint8_t *p_stream, size_t &offset, uint32_t *scaling_list, uint32_t list_size, uint32_t *use_default_scaling_matrix_flag);
|
||||
|
||||
/*! \brief Function to parse vidio usability information (VUI) parameters
|
||||
* \param [in] p_stream The pointer to the input bit stream
|
||||
* \param [in/out] offset Current bit offset
|
||||
* \param [out] p_vui_params The pointer to VUI structure
|
||||
* \return No return value
|
||||
*/
|
||||
void GetVuiParameters(uint8_t *p_stream, size_t &offset, AvcVuiSeqParameters *p_vui_params);
|
||||
|
||||
/*! \brief Function to check if there is more data in RBSP
|
||||
* \param [in] p_stream The pointer to the input bit stream
|
||||
* \param [in] stream_size_in_byte The byte size of the stream
|
||||
* \param [in] bit_offset The current bit offset
|
||||
* \return true/false
|
||||
*/
|
||||
bool MoreRbspData(uint8_t *p_stream, size_t stream_size_in_byte, size_t bit_offset);
|
||||
|
||||
#if DBGINFO
|
||||
/*! \brief Function to log out parsed SPS content for debug.
|
||||
*/
|
||||
void PrintSps(AvcSeqParameterSet *p_sps);
|
||||
|
||||
/*! \brief Function to log out parsed PPS content for debug.
|
||||
*/
|
||||
void PrintPps(AvcPicParameterSet *p_pps);
|
||||
|
||||
/*! \brief Function to log out parsed slice header content for debug.
|
||||
*/
|
||||
void PrintSliceHeader(AvcSliceHeader *p_slice_header);
|
||||
#endif // DBGINFO
|
||||
};
|
||||
Reference in New Issue
Block a user