You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
3.3 KiB
96 lines
3.3 KiB
4 months ago
|
/*--------------------------------------------------------------------------
|
||
|
Copyright (c) 2010-2015, 2018, The Linux Foundation. All rights reserved.
|
||
|
|
||
|
Redistribution and use in source and binary forms, with or without
|
||
|
modification, are permitted provided that the following conditions are met:
|
||
|
* Redistributions of source code must retain the above copyright
|
||
|
notice, this list of conditions and the following disclaimer.
|
||
|
* Redistributions in binary form must reproduce the above copyright
|
||
|
notice, this list of conditions and the following disclaimer in the
|
||
|
documentation and/or other materials provided with the distribution.
|
||
|
* Neither the name of The Linux Foundation nor
|
||
|
the names of its contributors may be used to endorse or promote
|
||
|
products derived from this software without specific prior written
|
||
|
permission.
|
||
|
|
||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||
|
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||
|
IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
||
|
NON-INFRINGEMENT ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
|
||
|
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||
|
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||
|
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
|
||
|
OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
|
||
|
WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
|
||
|
OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
|
||
|
ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
--------------------------------------------------------------------------*/
|
||
|
#ifndef FRAMEPARSER_H
|
||
|
#define FRAMEPARSER_H
|
||
|
|
||
|
#include "OMX_Core.h"
|
||
|
#include "OMX_QCOMExtns.h"
|
||
|
#include "h264_utils.h"
|
||
|
|
||
|
enum codec_type {
|
||
|
CODEC_TYPE_H264 = 1,
|
||
|
CODEC_TYPE_MPEG2 = 2,
|
||
|
CODEC_TYPE_VP8 = 3,
|
||
|
CODEC_TYPE_VP9 = 3,
|
||
|
CODEC_TYPE_HEVC,
|
||
|
CODEC_TYPE_MAX
|
||
|
};
|
||
|
|
||
|
enum state_start_code_parse {
|
||
|
A0,
|
||
|
A1,
|
||
|
A2,
|
||
|
A3,
|
||
|
A4,
|
||
|
A5
|
||
|
};
|
||
|
|
||
|
enum state_nal_parse {
|
||
|
NAL_LENGTH_ACC,
|
||
|
NAL_PARSING
|
||
|
};
|
||
|
|
||
|
class frame_parse
|
||
|
{
|
||
|
|
||
|
public:
|
||
|
H264_Utils *mutils;
|
||
|
int init_start_codes (codec_type codec_type_parse);
|
||
|
int parse_sc_frame (OMX_BUFFERHEADERTYPE *source,
|
||
|
OMX_BUFFERHEADERTYPE *dest ,
|
||
|
OMX_U32 *partialframe);
|
||
|
int init_nal_length (unsigned int nal_length);
|
||
|
int parse_h264_nallength (OMX_BUFFERHEADERTYPE *source,
|
||
|
OMX_BUFFERHEADERTYPE *dest ,
|
||
|
OMX_U32 *partialframe);
|
||
|
void flush ();
|
||
|
frame_parse ();
|
||
|
~frame_parse ();
|
||
|
|
||
|
private:
|
||
|
/*Variables for Start code based Parsing*/
|
||
|
enum state_start_code_parse parse_state;
|
||
|
unsigned char *start_code;
|
||
|
unsigned char *mask_code;
|
||
|
unsigned char last_byte;
|
||
|
bool header_found;
|
||
|
bool skip_frame_boundary;
|
||
|
|
||
|
/*Variables for NAL Length Parsing*/
|
||
|
enum state_nal_parse state_nal;
|
||
|
unsigned int nal_length;
|
||
|
unsigned int accum_length;
|
||
|
unsigned int bytes_tobeparsed;
|
||
|
/*Functions to support additional start code parsing*/
|
||
|
void parse_additional_start_code(OMX_U8 *psource, OMX_U32 *parsed_length);
|
||
|
void check_skip_frame_boundary(OMX_U32 *partial_frame);
|
||
|
void update_skip_frame();
|
||
|
};
|
||
|
|
||
|
#endif /* FRAMEPARSER_H */
|