Arun Menon | 4ab8ae8 | 2013-03-18 13:35:41 -0700 | [diff] [blame] | 1 | /*-------------------------------------------------------------------------- |
| 2 | Copyright (c) 2010-2013, 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 | { |
| 38 | CODEC_TYPE_MPEG4 = 0, |
| 39 | CODEC_TYPE_DIVX = 0, |
| 40 | CODEC_TYPE_H263 = 1, |
| 41 | CODEC_TYPE_H264 = 2, |
| 42 | CODEC_TYPE_VC1 = 3, |
| 43 | CODEC_TYPE_MPEG2 = 4, |
| 44 | #ifdef _MSM8974_ |
| 45 | CODEC_TYPE_VP8 = 5, |
| 46 | CODEC_TYPE_HEVC, |
| 47 | #endif |
| 48 | CODEC_TYPE_MAX |
| 49 | }; |
| 50 | |
| 51 | enum state_start_code_parse |
| 52 | { |
| 53 | A0, |
| 54 | A1, |
| 55 | A2, |
| 56 | A3, |
| 57 | A4, |
| 58 | A5 |
| 59 | }; |
| 60 | |
| 61 | enum state_nal_parse |
| 62 | { |
| 63 | NAL_LENGTH_ACC, |
| 64 | NAL_PARSING |
| 65 | }; |
| 66 | |
| 67 | class frame_parse |
| 68 | { |
| 69 | |
| 70 | public: |
| 71 | H264_Utils *mutils; |
| 72 | int init_start_codes (codec_type codec_type_parse); |
| 73 | int parse_sc_frame (OMX_BUFFERHEADERTYPE *source, |
| 74 | OMX_BUFFERHEADERTYPE *dest , |
| 75 | OMX_U32 *partialframe); |
| 76 | int init_nal_length (unsigned int nal_length); |
| 77 | int parse_h264_nallength (OMX_BUFFERHEADERTYPE *source, |
| 78 | OMX_BUFFERHEADERTYPE *dest , |
| 79 | OMX_U32 *partialframe); |
| 80 | void flush (); |
| 81 | frame_parse (); |
| 82 | ~frame_parse (); |
| 83 | |
| 84 | private: |
| 85 | /*Variables for Start code based Parsing*/ |
| 86 | enum state_start_code_parse parse_state; |
| 87 | unsigned char *start_code; |
| 88 | unsigned char *mask_code; |
| 89 | unsigned char last_byte_h263; |
| 90 | unsigned char last_byte; |
| 91 | bool header_found; |
| 92 | bool skip_frame_boundary; |
| 93 | |
| 94 | /*Variables for NAL Length Parsing*/ |
| 95 | enum state_nal_parse state_nal; |
| 96 | unsigned int nal_length; |
| 97 | unsigned int accum_length; |
| 98 | unsigned int bytes_tobeparsed; |
| 99 | /*Functions to support additional start code parsing*/ |
| 100 | void parse_additional_start_code(OMX_U8 *psource, OMX_U32 *parsed_length); |
| 101 | void check_skip_frame_boundary(OMX_U32 *partial_frame); |
| 102 | void update_skip_frame(); |
| 103 | }; |
| 104 | |
| 105 | #endif /* FRAMEPARSER_H */ |