1 /*-------------------------------------------------------------------------- 2 Copyright (c) 2010-2015, The Linux Foundation. All rights reserved. 3 4 Redistribution and use in source and binary forms, with or without 5 modification, are permitted provided that the following conditions are met: 6 * Redistributions of source code must retain the above copyright 7 notice, this list of conditions and the following disclaimer. 8 * Redistributions in binary form must reproduce the above copyright 9 notice, this list of conditions and the following disclaimer in the 10 documentation and/or other materials provided with the distribution. 11 * Neither the name of The Linux Foundation nor 12 the names of its contributors may be used to endorse or promote 13 products derived from this software without specific prior written 14 permission. 15 16 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 17 AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18 IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 19 NON-INFRINGEMENT ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR 20 CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 21 EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 22 PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 23 OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 24 WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 25 OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 26 ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 --------------------------------------------------------------------------*/ 28 #ifndef FRAMEPARSER_H 29 #define FRAMEPARSER_H 30 31 #include "OMX_Core.h" 32 #include "OMX_QCOMExtns.h" 33 #include "h264_utils.h" 34 //#include <stdlib.h> 35 36 enum codec_type { 37 CODEC_TYPE_MPEG4 = 0, 38 CODEC_TYPE_DIVX = 0, 39 CODEC_TYPE_H263 = 1, 40 CODEC_TYPE_H264 = 2, 41 CODEC_TYPE_VC1 = 3, 42 CODEC_TYPE_MPEG2 = 4, 43 #ifdef _MSM8974_ 44 CODEC_TYPE_VP8 = 5, 45 CODEC_TYPE_VP9 = 5, 46 CODEC_TYPE_HEVC, 47 #endif 48 CODEC_TYPE_MAX 49 }; 50 51 enum state_start_code_parse { 52 A0, 53 A1, 54 A2, 55 A3, 56 A4, 57 A5 58 }; 59 60 enum state_nal_parse { 61 NAL_LENGTH_ACC, 62 NAL_PARSING 63 }; 64 65 class frame_parse 66 { 67 68 public: 69 H264_Utils *mutils; 70 int init_start_codes (codec_type codec_type_parse); 71 int parse_sc_frame (OMX_BUFFERHEADERTYPE *source, 72 OMX_BUFFERHEADERTYPE *dest , 73 OMX_U32 *partialframe); 74 int init_nal_length (unsigned int nal_length); 75 int parse_h264_nallength (OMX_BUFFERHEADERTYPE *source, 76 OMX_BUFFERHEADERTYPE *dest , 77 OMX_U32 *partialframe); 78 void flush (); 79 frame_parse (); 80 ~frame_parse (); 81 82 private: 83 /*Variables for Start code based Parsing*/ 84 enum state_start_code_parse parse_state; 85 unsigned char *start_code; 86 unsigned char *mask_code; 87 unsigned char last_byte_h263; 88 unsigned char last_byte; 89 bool header_found; 90 bool skip_frame_boundary; 91 92 /*Variables for NAL Length Parsing*/ 93 enum state_nal_parse state_nal; 94 unsigned int nal_length; 95 unsigned int accum_length; 96 unsigned int bytes_tobeparsed; 97 /*Functions to support additional start code parsing*/ 98 void parse_additional_start_code(OMX_U8 *psource, OMX_U32 *parsed_length); 99 void check_skip_frame_boundary(OMX_U32 *partial_frame); 100 void update_skip_frame(); 101 }; 102 103 #endif /* FRAMEPARSER_H */ 104