1 /*-------------------------------------------------------------------------- 2 Copyright (c) 2014-2017, 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 6 met: 7 * Redistributions of source code must retain the above copyright 8 notice, this list of conditions and the following disclaimer. 9 * Redistributions in binary form must reproduce the above 10 copyright notice, this list of conditions and the following 11 disclaimer in the documentation and/or other materials provided 12 with the distribution. 13 * Neither the name of The Linux Foundation nor the names of its 14 contributors may be used to endorse or promote products derived 15 from this software without specific prior written permission. 16 17 THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED 18 WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 19 MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT 20 ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS 21 BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 22 CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 23 SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR 24 BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 25 WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE 26 OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN 27 IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 --------------------------------------------------------------------------*/ 29 #ifndef __OMX_VENC__H 30 #define __OMX_VENC__H 31 32 #define VEN_EXTRADATA_SLICEINFO 0x100 33 #define VEN_EXTRADATA_MBINFO 0x400 34 35 #include <unistd.h> 36 #include "omx_video_base.h" 37 #include "video_encoder_device_v4l2.h" 38 39 #include "swvenc_api.h" 40 #include "swvenc_types.h" 41 42 #include <ui/GraphicBuffer.h> 43 44 extern "C" { 45 OMX_API void * get_omx_component_factory_fn(void); 46 } 47 48 struct swvenc_video_capability { 49 unsigned int min_width; 50 unsigned int max_width; 51 unsigned int min_height; 52 unsigned int max_height; 53 }; 54 55 56 class omx_venc: public omx_video 57 { 58 public: 59 omx_venc(); 60 ~omx_venc(); 61 OMX_ERRORTYPE component_init(OMX_STRING role); 62 OMX_ERRORTYPE set_parameter(OMX_HANDLETYPE hComp, 63 OMX_INDEXTYPE paramIndex, 64 OMX_PTR paramData); 65 OMX_ERRORTYPE set_config(OMX_HANDLETYPE hComp, 66 OMX_INDEXTYPE configIndex, 67 OMX_PTR configData); 68 OMX_ERRORTYPE component_deinit(OMX_HANDLETYPE hComp); 69 bool is_secure_session(); 70 //OMX strucutres 71 OMX_U32 m_nVenc_format; 72 73 SWVENC_HANDLE m_hSwVenc; 74 SWVENC_CODEC m_codec; 75 swvenc_video_capability m_capability; 76 bool m_max_allowed_bitrate_check; 77 bool m_stopped; 78 bool set_format; 79 80 int dev_handle_output_extradata(void *, int); 81 int dev_handle_input_extradata(void *, int, int); 82 bool dev_buffer_ready_to_queue(OMX_BUFFERHEADERTYPE *buffer); 83 bool dev_get_dimensions(OMX_U32 ,OMX_U32 *,OMX_U32 *); 84 void dev_set_extradata_cookie(void *); 85 int dev_set_format(int); 86 87 static SWVENC_STATUS swvenc_empty_buffer_done_cb 88 ( 89 SWVENC_HANDLE swvenc, 90 SWVENC_IPBUFFER *p_ipbuffer, 91 void *p_client 92 ); 93 SWVENC_STATUS swvenc_empty_buffer_done 94 ( 95 SWVENC_IPBUFFER *p_ipbuffer 96 ); 97 static SWVENC_STATUS swvenc_fill_buffer_done_cb 98 ( 99 SWVENC_HANDLE swvenc, 100 SWVENC_OPBUFFER *p_opbuffer, 101 void *p_client 102 ); 103 static SWVENC_STATUS swvenc_handle_event_cb 104 ( 105 SWVENC_HANDLE swvenc, 106 SWVENC_EVENT event, 107 void *p_client 108 ); 109 110 static void init_sw_vendor_extensions(VendorExtensionStore &store); 111 112 private: 113 venc_debug_cap m_debug; 114 bool m_bSeqHdrRequested; 115 116 bool m_bDimensionsNeedFlip; 117 bool m_bIsRotationSupported; 118 bool m_bIsInFrameSizeSet; 119 bool m_bIsOutFrameSizeSet; 120 bool m_bIsInFlipDone; 121 bool m_bIsOutFlipDone; 122 bool m_bUseAVTimerTimestamps; 123 sp<GraphicBuffer> dstBuffer; 124 SWVENC_IPBUFFER *m_pIpbuffers; 125 126 int m_pipe_in; 127 int m_pipe_out; 128 OMX_VIDEO_PARAM_MPEG4TYPE m_sParamMPEG4; 129 OMX_VIDEO_PARAM_H263TYPE m_sParamH263; 130 131 OMX_U32 dev_stop(void); 132 OMX_U32 dev_pause(void); 133 OMX_U32 dev_start(void); 134 OMX_U32 dev_flush(unsigned); 135 OMX_U32 dev_resume(void); 136 OMX_U32 dev_start_done(void); 137 OMX_U32 dev_set_message_thread_id(pthread_t); 138 bool dev_use_buf( unsigned); 139 bool dev_handle_empty_eos_buffer(void); 140 bool dev_free_buf( void *,unsigned); 141 bool dev_empty_buf(void *, void *,unsigned,unsigned); 142 bool dev_fill_buf(void *, void *,unsigned,unsigned); 143 bool dev_get_buf_req(OMX_U32 *,OMX_U32 *,OMX_U32 *,OMX_U32); 144 bool is_streamon_done(OMX_U32 port); 145 bool dev_set_buf_req(OMX_U32 const *,OMX_U32 const *,OMX_U32 const *,OMX_U32); 146 bool dev_get_seq_hdr(void *, unsigned, unsigned *); 147 bool dev_loaded_start(void); 148 bool dev_loaded_stop(void); 149 bool dev_loaded_start_done(void); 150 bool dev_loaded_stop_done(void); 151 bool dev_get_capability_ltrcount(OMX_U32 *, OMX_U32 *, OMX_U32 *); 152 bool dev_get_vui_timing_info(OMX_U32 *); 153 bool dev_get_vqzip_sei_info(OMX_U32 *); 154 bool dev_get_peak_bitrate(OMX_U32 *); 155 bool dev_get_batch_size(OMX_U32 *); dev_get_temporal_layer_caps(OMX_U32 *,OMX_U32 *,OMX_VIDEO_ANDROID_TEMPORALLAYERINGPATTERNTYPE *)156 bool dev_get_temporal_layer_caps(OMX_U32 * /*nMaxLayers*/, 157 OMX_U32 * /*nMaxBLayers*/, 158 OMX_VIDEO_ANDROID_TEMPORALLAYERINGPATTERNTYPE */*SupportedPattern*/) { 159 return false; 160 } 161 OMX_ERRORTYPE dev_get_supported_profile_level(OMX_VIDEO_PARAM_PROFILELEVELTYPE */*profileLevelType*/); 162 bool dev_get_supported_color_format(unsigned index, OMX_U32 *colorFormat); 163 bool dev_is_video_session_supported(OMX_U32 width, OMX_U32 height); 164 bool dev_color_align(OMX_BUFFERHEADERTYPE *buffer, OMX_U32 width, 165 OMX_U32 height); 166 bool dev_get_output_log_flag(); 167 int dev_output_log_buffers(const char *buffer_addr, int buffer_len, uint64_t timestamp); 168 int dev_extradata_log_buffers(char *buffer); 169 bool swvenc_color_align(OMX_BUFFERHEADERTYPE *buffer, OMX_U32 width, 170 OMX_U32 height); 171 OMX_ERRORTYPE swvenc_do_flip_inport(); 172 OMX_ERRORTYPE swvenc_do_flip_outport(); 173 bool swvenc_do_rotate(int, SWVENC_IPBUFFER &, OMX_U32); 174 175 template<typename T> swvenc_delete_pointer(T * & ptr)176 inline void swvenc_delete_pointer(T * &ptr) { 177 if (ptr != nullptr) { 178 delete ptr; 179 ptr = nullptr; 180 } 181 } 182 183 SWVENC_STATUS swvenc_set_rc_mode(OMX_VIDEO_CONTROLRATETYPE eControlRate); 184 SWVENC_STATUS swvenc_set_profile_level(OMX_U32 eProfile,OMX_U32 eLevel); 185 SWVENC_STATUS swvenc_set_intra_refresh(OMX_VIDEO_PARAM_INTRAREFRESHTYPE *IntraRefresh); 186 SWVENC_STATUS swvenc_set_frame_rate(OMX_U32 nFrameRate); 187 SWVENC_STATUS swvenc_set_bit_rate(OMX_U32 nTargetBitrate); 188 SWVENC_STATUS swvenc_set_intra_period(OMX_U32 nPFrame,OMX_U32 nBFrame); 189 SWVENC_STATUS swvenc_set_color_format(OMX_COLOR_FORMATTYPE); 190 SWVENC_STATUS swvenc_get_buffer_req 191 ( 192 OMX_U32 *min_buff_count, 193 OMX_U32 *actual_buff_count, 194 OMX_U32 *buff_size, 195 OMX_U32 *buff_alignment, 196 OMX_U32 port 197 ); 198 int swvenc_input_log_buffers(const char *buffer, int bufferlen); dev_get_hevc_profile(OMX_U32 *)199 bool dev_get_hevc_profile(OMX_U32*) { return false; } 200 201 }; 202 203 #endif //__OMX_VENC__H 204