Home | History | Annotate | Download | only in inc
      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 extern "C" {
     43     OMX_API void * get_omx_component_factory_fn(void);
     44 }
     45 
     46 struct swvenc_video_capability {
     47     unsigned int min_width;
     48     unsigned int max_width;
     49     unsigned int min_height;
     50     unsigned int max_height;
     51 };
     52 
     53 
     54 class omx_venc: public omx_video
     55 {
     56     public:
     57         omx_venc();
     58         ~omx_venc();
     59         OMX_ERRORTYPE component_init(OMX_STRING role);
     60         OMX_ERRORTYPE set_parameter(OMX_HANDLETYPE hComp,
     61                 OMX_INDEXTYPE  paramIndex,
     62                 OMX_PTR        paramData);
     63         OMX_ERRORTYPE set_config(OMX_HANDLETYPE hComp,
     64                 OMX_INDEXTYPE  configIndex,
     65                 OMX_PTR        configData);
     66         OMX_ERRORTYPE component_deinit(OMX_HANDLETYPE hComp);
     67         bool is_secure_session();
     68         //OMX strucutres
     69         OMX_U32 m_nVenc_format;
     70 
     71         SWVENC_HANDLE m_hSwVenc;
     72         SWVENC_CODEC  m_codec;
     73         swvenc_video_capability m_capability;
     74         bool m_max_allowed_bitrate_check;
     75         bool m_stopped;
     76         bool set_format;
     77 
     78         int dev_handle_output_extradata(void *, int);
     79         int dev_handle_input_extradata(void *, int, int);
     80         bool dev_buffer_ready_to_queue(OMX_BUFFERHEADERTYPE *buffer);
     81         void dev_set_extradata_cookie(void *);
     82         int dev_set_format(int);
     83 
     84         static SWVENC_STATUS swvenc_empty_buffer_done_cb
     85         (
     86           SWVENC_HANDLE    swvenc,
     87           SWVENC_IPBUFFER *p_ipbuffer,
     88           void            *p_client
     89         );
     90         SWVENC_STATUS swvenc_empty_buffer_done
     91         (
     92           SWVENC_IPBUFFER *p_ipbuffer
     93         );
     94         static SWVENC_STATUS swvenc_fill_buffer_done_cb
     95         (
     96             SWVENC_HANDLE    swvenc,
     97             SWVENC_OPBUFFER *p_opbuffer,
     98             void            *p_client
     99         );
    100         static SWVENC_STATUS swvenc_handle_event_cb
    101         (
    102             SWVENC_HANDLE swvenc,
    103             SWVENC_EVENT  event,
    104             void         *p_client
    105         );
    106 
    107     private:
    108         venc_debug_cap m_debug;
    109         bool m_bSeqHdrRequested;
    110 
    111         int  m_pipe_in;
    112         int  m_pipe_out;
    113         OMX_VIDEO_PARAM_MPEG4TYPE m_sParamMPEG4;
    114         OMX_VIDEO_PARAM_H263TYPE m_sParamH263;
    115 
    116         OMX_U32 dev_stop(void);
    117         OMX_U32 dev_pause(void);
    118         OMX_U32 dev_start(void);
    119         OMX_U32 dev_flush(unsigned);
    120         OMX_U32 dev_resume(void);
    121         OMX_U32 dev_start_done(void);
    122         OMX_U32 dev_set_message_thread_id(pthread_t);
    123         bool dev_use_buf( unsigned);
    124         bool dev_free_buf( void *,unsigned);
    125         bool dev_empty_buf(void *, void *,unsigned,unsigned);
    126         bool dev_fill_buf(void *, void *,unsigned,unsigned);
    127         bool dev_get_buf_req(OMX_U32 *,OMX_U32 *,OMX_U32 *,OMX_U32);
    128         bool dev_set_buf_req(OMX_U32 const *,OMX_U32 const *,OMX_U32 const *,OMX_U32);
    129         bool dev_get_seq_hdr(void *, unsigned, unsigned *);
    130         bool dev_loaded_start(void);
    131         bool dev_loaded_stop(void);
    132         bool dev_loaded_start_done(void);
    133         bool dev_loaded_stop_done(void);
    134         bool dev_get_capability_ltrcount(OMX_U32 *, OMX_U32 *, OMX_U32 *);
    135         bool dev_get_vui_timing_info(OMX_U32 *);
    136         bool dev_get_vqzip_sei_info(OMX_U32 *);
    137         bool dev_get_peak_bitrate(OMX_U32 *);
    138         bool dev_get_batch_size(OMX_U32 *);
    139         bool dev_get_temporal_layer_caps(OMX_U32 * /*nMaxLayers*/,
    140                     OMX_U32 * /*nMaxBLayers*/,
    141                     OMX_VIDEO_ANDROID_TEMPORALLAYERINGPATTERNTYPE */*SupportedPattern*/) {
    142             return false;
    143         }
    144         bool dev_is_video_session_supported(OMX_U32 width, OMX_U32 height);
    145         bool dev_color_align(OMX_BUFFERHEADERTYPE *buffer, OMX_U32 width,
    146                         OMX_U32 height);
    147         bool dev_get_output_log_flag();
    148         int dev_output_log_buffers(const char *buffer_addr, int buffer_len);
    149         int dev_extradata_log_buffers(char *buffer);
    150         bool swvenc_color_align(OMX_BUFFERHEADERTYPE *buffer, OMX_U32 width,
    151                                 OMX_U32 height);
    152 
    153         SWVENC_STATUS swvenc_set_rc_mode(OMX_VIDEO_CONTROLRATETYPE eControlRate);
    154         SWVENC_STATUS swvenc_set_profile_level(OMX_U32 eProfile,OMX_U32 eLevel);
    155         SWVENC_STATUS swvenc_set_intra_refresh(OMX_VIDEO_PARAM_INTRAREFRESHTYPE *IntraRefresh);
    156         SWVENC_STATUS swvenc_set_frame_rate(OMX_U32 nFrameRate);
    157         SWVENC_STATUS swvenc_set_bit_rate(OMX_U32 nTargetBitrate);
    158         SWVENC_STATUS swvenc_set_intra_period(OMX_U32 nPFrame,OMX_U32 nBFrame);
    159         SWVENC_STATUS swvenc_set_color_format(OMX_COLOR_FORMATTYPE);
    160         SWVENC_STATUS swvenc_get_buffer_req
    161         (
    162            OMX_U32 *min_buff_count,
    163            OMX_U32 *actual_buff_count,
    164            OMX_U32 *buff_size,
    165            OMX_U32 *buff_alignment,
    166            OMX_U32 port
    167         );
    168         int swvenc_input_log_buffers(const char *buffer, int bufferlen);
    169 
    170 };
    171 
    172 #endif //__OMX_VENC__H
    173