git.sesse.net Git - vlc/blob - modules/codec/omxil/omxil.c

   1 /*****************************************************************************
   2  * omxil.c: Video decoder module making use of OpenMAX IL components.
   3  *****************************************************************************
   4  * Copyright (C) 2010 VLC authors and VideoLAN
   5  * $Id$
   6  *
   7  * Authors: Gildas Bazin <gbazin@videolan.org>
   8  *
   9  * This program is free software; you can redistribute it and/or modify it
  10  * under the terms of the GNU Lesser General Public License as published by
  11  * the Free Software Foundation; either version 2.1 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  17  * GNU Lesser General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Lesser General Public License
  20  * along with this program; if not, write to the Free Software Foundation,
  21  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27 #ifdef HAVE_CONFIG_H
  28 # include "config.h"
  29 #endif
  30
  31 #include <dlfcn.h>
  32 #if defined(USE_IOMX)
  33 /* On dll_open, just check that the OMX_Init symbol already is loaded */
  34 # define dll_open(name) dlsym(RTLD_DEFAULT, "OMX_Init")
  35 # define dll_close(handle) do { } while (0)
  36 # define dlsym(handle, name) dlsym(RTLD_DEFAULT, "I" name)
  37 #else
  38 # define dll_open(name) dlopen( name, RTLD_NOW )
  39 # define dll_close(handle) dlclose(handle)
  40 #endif
  41
  42 #include <limits.h>
  43
  44 #include <vlc_common.h>
  45 #include <vlc_plugin.h>
  46 #include <vlc_codec.h>
  47 #include <vlc_block_helper.h>
  48 #include <vlc_cpu.h>
  49 #include "../h264_nal.h"
  50
  51 #include "omxil.h"
  52
  53 #ifndef NDEBUG
  54 # define OMXIL_EXTRA_DEBUG
  55 #endif
  56
  57 #define SENTINEL_FLAG 0x10000
  58
  59 /*****************************************************************************
  60  * List of OpenMAX IL core we will try in order
  61  *****************************************************************************/
  62 static const char *ppsz_dll_list[] =
  63 {
  64 #if defined(USE_IOMX)
  65     "libiomx.so", /* Not used when using IOMX, the lib should already be loaded */
  66 #else
  67     "libOMX_Core.so", /* TI OMAP IL core */
  68     "libOmxCore.so", /* Qualcomm IL core */
  69     "libomxil-bellagio.so",  /* Bellagio IL core */
  70 #endif
  71     0
  72 };
  73
  74 /*****************************************************************************
  75  * Global OMX Core instance, shared between module instances
  76  *****************************************************************************/
  77 static vlc_mutex_t omx_core_mutex = VLC_STATIC_MUTEX;
  78 static unsigned int omx_refcount = 0;
  79 static void *dll_handle;
  80 static OMX_ERRORTYPE (*pf_init) (void);
  81 static OMX_ERRORTYPE (*pf_deinit) (void);
  82 static OMX_ERRORTYPE (*pf_get_handle) (OMX_HANDLETYPE *, OMX_STRING,
  83                                        OMX_PTR, OMX_CALLBACKTYPE *);
  84 static OMX_ERRORTYPE (*pf_free_handle) (OMX_HANDLETYPE);
  85 static OMX_ERRORTYPE (*pf_component_enum)(OMX_STRING, OMX_U32, OMX_U32);
  86 static OMX_ERRORTYPE (*pf_get_roles_of_component)(OMX_STRING, OMX_U32 *, OMX_U8 **);
  87
  88 /*****************************************************************************
  89  * Local prototypes
  90  *****************************************************************************/
  91 static int  OpenDecoder( vlc_object_t * );
  92 static int  OpenEncoder( vlc_object_t * );
  93 static int  OpenGeneric( vlc_object_t *, bool b_encode );
  94 static void CloseGeneric( vlc_object_t * );
  95
  96 static picture_t *DecodeVideo( decoder_t *, block_t ** );
  97 static block_t *DecodeAudio ( decoder_t *, block_t ** );
  98 static block_t *EncodeVideo( encoder_t *, picture_t * );
  99
 100 static OMX_ERRORTYPE OmxEventHandler( OMX_HANDLETYPE, OMX_PTR, OMX_EVENTTYPE,
 101                                       OMX_U32, OMX_U32, OMX_PTR );
 102 static OMX_ERRORTYPE OmxEmptyBufferDone( OMX_HANDLETYPE, OMX_PTR,
 103                                          OMX_BUFFERHEADERTYPE * );
 104 static OMX_ERRORTYPE OmxFillBufferDone( OMX_HANDLETYPE, OMX_PTR,
 105                                         OMX_BUFFERHEADERTYPE * );
 106
 107 /*****************************************************************************
 108  * Module descriptor
 109  *****************************************************************************/
 110 vlc_module_begin ()
 111     set_description( N_("Audio/Video decoder (using OpenMAX IL)") )
 112     set_category( CAT_INPUT )
 113     set_subcategory( SUBCAT_INPUT_VCODEC )
 114     set_section( N_("Decoding") , NULL )
 115 #if defined(USE_IOMX)
 116     /* For IOMX, don't enable it automatically via priorities,
 117      * enable it only via the --codec iomx command line parameter when
 118      * wanted. */
 119     set_capability( "decoder", 0 )
 120 #else
 121     set_capability( "decoder", 80 )
 122 #endif
 123     set_callbacks( OpenDecoder, CloseGeneric )
 124
 125     add_submodule ()
 126     set_section( N_("Encoding") , NULL )
 127     set_description( N_("Video encoder (using OpenMAX IL)") )
 128     set_capability( "encoder", 0 )
 129     set_callbacks( OpenEncoder, CloseGeneric )
 130 vlc_module_end ()
 131
 132 /*****************************************************************************
 133  * CreateComponentsList: creates a list of components matching the given role
 134  *****************************************************************************/
 135 static int CreateComponentsList(decoder_t *p_dec, const char *psz_role)
 136 {
 137     decoder_sys_t *p_sys = p_dec->p_sys;
 138     char psz_name[OMX_MAX_STRINGNAME_SIZE];
 139     OMX_ERRORTYPE omx_error;
 140     OMX_U32 roles = 0;
 141     OMX_U8 **ppsz_roles = 0;
 142     unsigned int i, j, len;
 143
 144     if(!psz_role) goto end;
 145     len = strlen(psz_role);
 146
 147     for( i = 0; ; i++ )
 148     {
 149         bool b_found = false;
 150
 151         omx_error = pf_component_enum(psz_name, OMX_MAX_STRINGNAME_SIZE, i);
 152         if(omx_error != OMX_ErrorNone) break;
 153
 154         msg_Dbg(p_dec, "component %s", psz_name);
 155
 156         omx_error = pf_get_roles_of_component(psz_name, &roles, 0);
 157         if(omx_error != OMX_ErrorNone || !roles) continue;
 158
 159         ppsz_roles = malloc(roles * (sizeof(OMX_U8*) + OMX_MAX_STRINGNAME_SIZE));
 160         if(!ppsz_roles) continue;
 161
 162         for( j = 0; j < roles; j++ )
 163             ppsz_roles[j] = ((OMX_U8 *)(&ppsz_roles[roles])) +
 164                 j * OMX_MAX_STRINGNAME_SIZE;
 165
 166         omx_error = pf_get_roles_of_component(psz_name, &roles, ppsz_roles);
 167         if(omx_error != OMX_ErrorNone) roles = 0;
 168
 169         for(j = 0; j < roles; j++)
 170         {
 171             msg_Dbg(p_dec, "  - role: %s", ppsz_roles[j]);
 172             if(!strncmp((char *)ppsz_roles[j], psz_role, len)) b_found = true;
 173         }
 174
 175         free(ppsz_roles);
 176
 177         if(!b_found) continue;
 178
 179         if(p_sys->components >= MAX_COMPONENTS_LIST_SIZE)
 180         {
 181             msg_Dbg(p_dec, "too many matching components");
 182             continue;
 183         }
 184
 185         strncpy(p_sys->ppsz_components[p_sys->components], psz_name,
 186                 OMX_MAX_STRINGNAME_SIZE-1);
 187         p_sys->components++;
 188     }
 189
 190  end:
 191     msg_Dbg(p_dec, "found %i matching components for role %s",
 192             p_sys->components, psz_role);
 193     for( i = 0; i < p_sys->components; i++ )
 194         msg_Dbg(p_dec, "- %s", p_sys->ppsz_components[i]);
 195
 196     return p_sys->components;
 197 }
 198
 199 /*****************************************************************************
 200  * ImplementationSpecificWorkarounds: place-holder for implementation
 201  * specific workarounds
 202  *****************************************************************************/
 203 static OMX_ERRORTYPE ImplementationSpecificWorkarounds(decoder_t *p_dec,
 204     OmxPort *p_port, es_format_t *p_fmt)
 205 {
 206     decoder_sys_t *p_sys = p_dec->p_sys;
 207     OMX_PARAM_PORTDEFINITIONTYPE *def = &p_port->definition;
 208     int i_profile = 0xFFFF, i_level = 0xFFFF;
 209
 210     /* Try to find out the profile of the video */
 211     while(p_fmt->i_cat == VIDEO_ES && def->eDir == OMX_DirInput &&
 212           p_fmt->i_codec == VLC_CODEC_H264)
 213     {
 214         uint8_t *p = (uint8_t*)p_dec->fmt_in.p_extra;
 215         if(!p || !p_dec->fmt_in.p_extra) break;
 216
 217         /* Check the profile / level */
 218         if(p_dec->fmt_in.i_original_fourcc == VLC_FOURCC('a','v','c','1') &&
 219            p[0] == 1)
 220         {
 221             if(p_dec->fmt_in.i_extra < 12) break;
 222             p_sys->i_nal_size_length = 1 + (p[4]&0x03);
 223             if( !(p[5]&0x1f) ) break;
 224             p += 8;
 225         }
 226         else
 227         {
 228             if(p_dec->fmt_in.i_extra < 8) break;
 229             if(!p[0] && !p[1] && !p[2] && p[3] == 1) p += 4;
 230             else if(!p[0] && !p[1] && p[2] == 1) p += 3;
 231             else break;
 232         }
 233
 234         if( ((*p++)&0x1f) != 7) break;
 235
 236         /* Get profile/level out of first SPS */
 237         i_profile = p[0];
 238         i_level = p[2];
 239         break;
 240     }
 241
 242     if(!strcmp(p_sys->psz_component, "OMX.TI.Video.Decoder"))
 243     {
 244         if(p_fmt->i_cat == VIDEO_ES && def->eDir == OMX_DirInput &&
 245            p_fmt->i_codec == VLC_CODEC_H264 &&
 246            (i_profile != 66 || i_level > 30))
 247         {
 248             msg_Dbg(p_dec, "h264 profile/level not supported (0x%x, 0x%x)",
 249                     i_profile, i_level);
 250             return OMX_ErrorNotImplemented;
 251         }
 252
 253         if(p_fmt->i_cat == VIDEO_ES && def->eDir == OMX_DirOutput &&
 254            p_fmt->i_codec == VLC_CODEC_I420)
 255         {
 256             /* I420 xvideo is slow on OMAP */
 257             def->format.video.eColorFormat = OMX_COLOR_FormatCbYCrY;
 258             GetVlcChromaFormat( def->format.video.eColorFormat,
 259                                 &p_fmt->i_codec, 0 );
 260             GetVlcChromaSizes( p_fmt->i_codec,
 261                                def->format.video.nFrameWidth,
 262                                def->format.video.nFrameHeight,
 263                                &p_port->i_frame_size, &p_port->i_frame_stride,
 264                                &p_port->i_frame_stride_chroma_div );
 265             def->format.video.nStride = p_port->i_frame_stride;
 266             def->nBufferSize = p_port->i_frame_size;
 267         }
 268     }
 269     else if(!strcmp(p_sys->psz_component, "OMX.st.video_encoder"))
 270     {
 271         if(p_fmt->i_cat == VIDEO_ES)
 272         {
 273             /* Bellagio's encoder doesn't encode the framerate in Q16 */
 274             def->format.video.xFramerate >>= 16;
 275         }
 276     }
 277 #if 0 /* FIXME: doesn't apply for HP Touchpad */
 278     else if (!strncmp(p_sys->psz_component, "OMX.qcom.video.decoder.",
 279                       strlen("OMX.qcom.video.decoder")))
 280     {
 281         /* qdsp6 refuses buffer size larger than 450K on input port */
 282         if (def->nBufferSize > 450 * 1024)
 283         {
 284             def->nBufferSize = 450 * 1024;
 285             p_port->i_frame_size = def->nBufferSize;
 286         }
 287     }
 288 #endif
 289
 290     return OMX_ErrorNone;
 291 }
 292
 293 /*****************************************************************************
 294  * SetPortDefinition: set definition of the omx port based on the vlc format
 295  *****************************************************************************/
 296 static OMX_ERRORTYPE SetPortDefinition(decoder_t *p_dec, OmxPort *p_port,
 297                                        es_format_t *p_fmt)
 298 {
 299     OMX_PARAM_PORTDEFINITIONTYPE *def = &p_port->definition;
 300     OMX_ERRORTYPE omx_error;
 301
 302     omx_error = OMX_GetParameter(p_port->omx_handle,
 303                                  OMX_IndexParamPortDefinition, def);
 304     CHECK_ERROR(omx_error, "OMX_GetParameter failed (%x : %s)",
 305                 omx_error, ErrorToString(omx_error));
 306
 307     switch(p_fmt->i_cat)
 308     {
 309     case VIDEO_ES:
 310         def->format.video.nFrameWidth = p_fmt->video.i_width;
 311         def->format.video.nFrameHeight = p_fmt->video.i_height;
 312         if(def->format.video.eCompressionFormat == OMX_VIDEO_CodingUnused)
 313             def->format.video.nStride = def->format.video.nFrameWidth;
 314         if( p_fmt->video.i_frame_rate > 0 &&
 315             p_fmt->video.i_frame_rate_base > 0 )
 316             def->format.video.xFramerate = (p_fmt->video.i_frame_rate << 16) /
 317                 p_fmt->video.i_frame_rate_base;
 318
 319         if(def->eDir == OMX_DirInput || p_dec->p_sys->b_enc)
 320         {
 321             if (def->eDir == OMX_DirInput && p_dec->p_sys->b_enc)
 322                 def->nBufferSize = def->format.video.nFrameWidth *
 323                   def->format.video.nFrameHeight * 2;
 324             p_port->i_frame_size = def->nBufferSize;
 325
 326             if(!GetOmxVideoFormat(p_fmt->i_codec,
 327                                   &def->format.video.eCompressionFormat, 0) )
 328             {
 329                 if(!GetOmxChromaFormat(p_fmt->i_codec,
 330                                        &def->format.video.eColorFormat, 0) )
 331                 {
 332                     omx_error = OMX_ErrorNotImplemented;
 333                     CHECK_ERROR(omx_error, "codec %4.4s doesn't match any OMX format",
 334                                 (char *)&p_fmt->i_codec );
 335                 }
 336                 GetVlcChromaSizes( p_fmt->i_codec,
 337                                    def->format.video.nFrameWidth,
 338                                    def->format.video.nFrameHeight,
 339                                    &p_port->i_frame_size, &p_port->i_frame_stride,
 340                                    &p_port->i_frame_stride_chroma_div );
 341                 def->format.video.nStride = p_port->i_frame_stride;
 342                 def->nBufferSize = p_port->i_frame_size;
 343             }
 344         }
 345         else
 346         {
 347             if( !GetVlcChromaFormat( def->format.video.eColorFormat,
 348                                      &p_fmt->i_codec, 0 ) )
 349             {
 350                 omx_error = OMX_ErrorNotImplemented;
 351                 CHECK_ERROR(omx_error, "OMX color format %i not supported",
 352                             (int)def->format.video.eColorFormat );
 353             }
 354             GetVlcChromaSizes( p_fmt->i_codec,
 355                                def->format.video.nFrameWidth,
 356                                def->format.video.nFrameHeight,
 357                                &p_port->i_frame_size, &p_port->i_frame_stride,
 358                                &p_port->i_frame_stride_chroma_div );
 359             def->format.video.nStride = p_port->i_frame_stride;
 360             if (p_port->i_frame_size > def->nBufferSize)
 361                 def->nBufferSize = p_port->i_frame_size;
 362         }
 363         break;
 364
 365     case AUDIO_ES:
 366         p_port->i_frame_size = def->nBufferSize;
 367         if(def->eDir == OMX_DirInput)
 368         {
 369             if(!GetOmxAudioFormat(p_fmt->i_codec,
 370                                   &def->format.audio.eEncoding, 0) )
 371             {
 372                 omx_error = OMX_ErrorNotImplemented;
 373                 CHECK_ERROR(omx_error, "codec %4.4s doesn't match any OMX format",
 374                             (char *)&p_fmt->i_codec );
 375             }
 376         }
 377         else
 378         {
 379             if( !OmxToVlcAudioFormat(def->format.audio.eEncoding,
 380                                    &p_fmt->i_codec, 0 ) )
 381             {
 382                 omx_error = OMX_ErrorNotImplemented;
 383                 CHECK_ERROR(omx_error, "OMX audio encoding %i not supported",
 384                             (int)def->format.audio.eEncoding );
 385             }
 386         }
 387         break;
 388
 389     default: return OMX_ErrorNotImplemented;
 390     }
 391
 392     omx_error = ImplementationSpecificWorkarounds(p_dec, p_port, p_fmt);
 393     CHECK_ERROR(omx_error, "ImplementationSpecificWorkarounds failed (%x : %s)",
 394                 omx_error, ErrorToString(omx_error));
 395
 396     omx_error = OMX_SetParameter(p_port->omx_handle,
 397                                  OMX_IndexParamPortDefinition, def);
 398     CHECK_ERROR(omx_error, "OMX_SetParameter failed (%x : %s)",
 399                 omx_error, ErrorToString(omx_error));
 400
 401     omx_error = OMX_GetParameter(p_port->omx_handle,
 402                                  OMX_IndexParamPortDefinition, def);
 403     CHECK_ERROR(omx_error, "OMX_GetParameter failed (%x : %s)",
 404                 omx_error, ErrorToString(omx_error));
 405
 406     if(p_port->i_frame_size > def->nBufferSize)
 407         def->nBufferSize = p_port->i_frame_size;
 408     p_port->i_frame_size = def->nBufferSize;
 409
 410     /* Deal with audio params */
 411     if(p_fmt->i_cat == AUDIO_ES)
 412     {
 413         omx_error = SetAudioParameters(p_port->omx_handle,
 414                                        &p_port->format_param, def->nPortIndex,
 415                                        def->format.audio.eEncoding,
 416                                        p_fmt->audio.i_channels,
 417                                        p_fmt->audio.i_rate,
 418                                        p_fmt->i_bitrate,
 419                                        p_fmt->audio.i_bitspersample,
 420                                        p_fmt->audio.i_blockalign);
 421         if (def->eDir == OMX_DirInput) {
 422             CHECK_ERROR(omx_error, "SetAudioParameters failed (%x : %s)",
 423                         omx_error, ErrorToString(omx_error));
 424         } else {
 425             msg_Warn(p_dec, "SetAudioParameters failed (%x : %s) on output port",
 426                      omx_error, ErrorToString(omx_error));
 427             omx_error = OMX_ErrorNone;
 428         }
 429     }
 430     if (!strcmp(p_dec->p_sys->psz_component, "OMX.TI.DUCATI1.VIDEO.DECODER") &&
 431                 def->eDir == OMX_DirOutput)
 432     {
 433         /* When setting the output buffer size above, the decoder actually
 434          * sets the buffer size to a lower value than what was chosen. If
 435          * we try to allocate buffers of this size, it fails. Thus, forcibly
 436          * use a larger buffer size. */
 437         def->nBufferSize *= 2;
 438     }
 439
 440     if (def->format.video.eCompressionFormat == OMX_VIDEO_CodingWMV) {
 441         OMX_VIDEO_PARAM_WMVTYPE wmvtype = { 0 };
 442         OMX_INIT_STRUCTURE(wmvtype);
 443         wmvtype.nPortIndex = def->nPortIndex;
 444         switch (p_dec->fmt_in.i_codec) {
 445         case VLC_CODEC_WMV1:
 446             wmvtype.eFormat = OMX_VIDEO_WMVFormat7;
 447             break;
 448         case VLC_CODEC_WMV2:
 449             wmvtype.eFormat = OMX_VIDEO_WMVFormat8;
 450             break;
 451         case VLC_CODEC_WMV3:
 452         case VLC_CODEC_VC1:
 453             wmvtype.eFormat = OMX_VIDEO_WMVFormat9;
 454             break;
 455         }
 456         omx_error = OMX_SetParameter(p_port->omx_handle, OMX_IndexParamVideoWmv, &wmvtype);
 457         CHECK_ERROR(omx_error, "OMX_SetParameter OMX_IndexParamVideoWmv failed (%x : %s)",
 458                     omx_error, ErrorToString(omx_error));
 459     }
 460
 461  error:
 462     return omx_error;
 463 }
 464
 465 /*****************************************************************************
 466  * GetPortDefinition: set vlc format based on the definition of the omx port
 467  *****************************************************************************/
 468 static OMX_ERRORTYPE GetPortDefinition(decoder_t *p_dec, OmxPort *p_port,
 469                                        es_format_t *p_fmt)
 470 {
 471     decoder_sys_t *p_sys = p_dec->p_sys;
 472     OMX_PARAM_PORTDEFINITIONTYPE *def = &p_port->definition;
 473     OMX_ERRORTYPE omx_error;
 474     OMX_CONFIG_RECTTYPE crop_rect;
 475
 476     omx_error = OMX_GetParameter(p_port->omx_handle,
 477                                  OMX_IndexParamPortDefinition, def);
 478     CHECK_ERROR(omx_error, "OMX_GetParameter failed (%x : %s)",
 479                 omx_error, ErrorToString(omx_error));
 480
 481     switch(p_fmt->i_cat)
 482     {
 483     case VIDEO_ES:
 484         p_fmt->video.i_width = def->format.video.nFrameWidth;
 485         p_fmt->video.i_visible_width = def->format.video.nFrameWidth;
 486         p_fmt->video.i_height = def->format.video.nFrameHeight;
 487         p_fmt->video.i_visible_height = def->format.video.nFrameHeight;
 488         p_fmt->video.i_frame_rate = p_dec->fmt_in.video.i_frame_rate;
 489         p_fmt->video.i_frame_rate_base = p_dec->fmt_in.video.i_frame_rate_base;
 490
 491         OMX_INIT_STRUCTURE(crop_rect);
 492         crop_rect.nPortIndex = def->nPortIndex;
 493         omx_error = OMX_GetConfig(p_port->omx_handle, OMX_IndexConfigCommonOutputCrop, &crop_rect);
 494         if (omx_error == OMX_ErrorNone)
 495         {
 496             if (!def->format.video.nSliceHeight)
 497                 def->format.video.nSliceHeight = def->format.video.nFrameHeight;
 498             if (!def->format.video.nStride)
 499                 def->format.video.nStride = def->format.video.nFrameWidth;
 500             p_fmt->video.i_width = crop_rect.nWidth;
 501             p_fmt->video.i_visible_width = crop_rect.nWidth;
 502             p_fmt->video.i_height = crop_rect.nHeight;
 503             p_fmt->video.i_visible_height = crop_rect.nHeight;
 504             if (def->format.video.eColorFormat == OMX_TI_COLOR_FormatYUV420PackedSemiPlanar)
 505                 def->format.video.nSliceHeight -= crop_rect.nTop/2;
 506         }
 507         else
 508         {
 509             /* Don't pass the error back to the caller, this isn't mandatory */
 510             omx_error = OMX_ErrorNone;
 511         }
 512
 513         /* Hack: Nexus One (stock firmware with binary OMX driver blob)
 514          * claims to output 420Planar even though it in in practice is
 515          * NV21. */
 516         if(def->format.video.eColorFormat == OMX_COLOR_FormatYUV420Planar &&
 517            !strncmp(p_sys->psz_component, "OMX.qcom.video.decoder",
 518                     strlen("OMX.qcom.video.decoder")))
 519             def->format.video.eColorFormat = OMX_QCOM_COLOR_FormatYVU420SemiPlanar;
 520
 521         /* Hack: Some Samsung devices (e.g. Galaxy S III) have multiple
 522          * H264 decoders with different quirks (OMX.SEC.avc.dec, OMX.SEC.avcdec
 523          * and OMX.SEC.AVC.Decoder) - the latter is well-behaved while the
 524          * former ones signal padding but in practice doesn't have any padding.
 525          * We can't simply ignore the buggy ones, because some devices only
 526          * have that one (Galaxy S II has only got one named OMX.SEC.avcdec,
 527          * at least in the pre-4.0 firmwares). Thus, we enable this quirk on
 528          * any OMX.SEC. decoder that doesn't contain the string ".Decoder". */
 529         if(!strncmp(p_sys->psz_component, "OMX.SEC.", strlen("OMX.SEC.")) &&
 530            !strstr(p_sys->psz_component, ".Decoder")) {
 531             def->format.video.nSliceHeight = 0;
 532             def->format.video.nStride = p_fmt->video.i_width;
 533         }
 534
 535         if(!GetVlcVideoFormat( def->format.video.eCompressionFormat,
 536                                &p_fmt->i_codec, 0 ) )
 537         {
 538             if( !GetVlcChromaFormat( def->format.video.eColorFormat,
 539                                      &p_fmt->i_codec, 0 ) )
 540             {
 541                 omx_error = OMX_ErrorNotImplemented;
 542                 CHECK_ERROR(omx_error, "OMX color format %i not supported",
 543                             (int)def->format.video.eColorFormat );
 544             }
 545             GetVlcChromaSizes( p_fmt->i_codec,
 546                                def->format.video.nFrameWidth,
 547                                def->format.video.nFrameHeight,
 548                                &p_port->i_frame_size, &p_port->i_frame_stride,
 549                                &p_port->i_frame_stride_chroma_div );
 550         }
 551         if(p_port->i_frame_size > def->nBufferSize)
 552             def->nBufferSize = p_port->i_frame_size;
 553         p_port->i_frame_size = def->nBufferSize;
 554 #if 0
 555         if((int)p_port->i_frame_stride > def->format.video.nStride)
 556             def->format.video.nStride = p_port->i_frame_stride;
 557 #endif
 558         p_port->i_frame_stride = def->format.video.nStride;
 559         break;
 560
 561     case AUDIO_ES:
 562         if( !OmxToVlcAudioFormat( def->format.audio.eEncoding,
 563                                 &p_fmt->i_codec, 0 ) )
 564         {
 565             omx_error = OMX_ErrorNotImplemented;
 566             CHECK_ERROR(omx_error, "OMX audio format %i not supported",
 567                         (int)def->format.audio.eEncoding );
 568         }
 569
 570         omx_error = GetAudioParameters(p_port->omx_handle,
 571                                        &p_port->format_param, def->nPortIndex,
 572                                        def->format.audio.eEncoding,
 573                                        &p_fmt->audio.i_channels,
 574                                        &p_fmt->audio.i_rate,
 575                                        &p_fmt->i_bitrate,
 576                                        &p_fmt->audio.i_bitspersample,
 577                                        &p_fmt->audio.i_blockalign);
 578         CHECK_ERROR(omx_error, "GetAudioParameters failed (%x : %s)",
 579                     omx_error, ErrorToString(omx_error));
 580
 581         if(p_fmt->audio.i_channels < 9)
 582         {
 583             static const int pi_channels_maps[9] =
 584             {
 585                 0, AOUT_CHAN_CENTER, AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
 586                 AOUT_CHAN_CENTER | AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
 587                 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_REARLEFT
 588                 | AOUT_CHAN_REARRIGHT,
 589                 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
 590                 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT,
 591                 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
 592                 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT | AOUT_CHAN_LFE,
 593                 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
 594                 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT | AOUT_CHAN_MIDDLELEFT
 595                 | AOUT_CHAN_MIDDLERIGHT,
 596                 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER | AOUT_CHAN_REARLEFT
 597                 | AOUT_CHAN_REARRIGHT | AOUT_CHAN_MIDDLELEFT | AOUT_CHAN_MIDDLERIGHT
 598                 | AOUT_CHAN_LFE
 599             };
 600             p_fmt->audio.i_physical_channels =
 601                 p_fmt->audio.i_original_channels =
 602                     pi_channels_maps[p_fmt->audio.i_channels];
 603         }
 604
 605         date_Init( &p_dec->p_sys->end_date, p_fmt->audio.i_rate, 1 );
 606
 607         break;
 608
 609     default: return OMX_ErrorNotImplemented;
 610     }
 611
 612  error:
 613     return omx_error;
 614 }
 615
 616 /*****************************************************************************
 617  * DeinitialiseComponent: Deinitialise and unload an OMX component
 618  *****************************************************************************/
 619 static OMX_ERRORTYPE DeinitialiseComponent(decoder_t *p_dec,
 620                                            OMX_HANDLETYPE omx_handle)
 621 {
 622     decoder_sys_t *p_sys = p_dec->p_sys;
 623     OMX_ERRORTYPE omx_error;
 624     OMX_STATETYPE state;
 625     unsigned int i, j;
 626
 627     if(!omx_handle) return OMX_ErrorNone;
 628
 629     omx_error = OMX_GetState(omx_handle, &state);
 630     CHECK_ERROR(omx_error, "OMX_GetState failed (%x)", omx_error );
 631
 632     if(state == OMX_StateExecuting)
 633     {
 634         omx_error = OMX_SendCommand( omx_handle, OMX_CommandStateSet,
 635                                      OMX_StateIdle, 0 );
 636         CHECK_ERROR(omx_error, "OMX_CommandStateSet Idle failed (%x)", omx_error );
 637         while (1) {
 638             OMX_U32 cmd, state;
 639             omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, &cmd, &state, 0);
 640             CHECK_ERROR(omx_error, "Wait for Idle failed (%x)", omx_error );
 641             // The event queue can contain other OMX_EventCmdComplete items,
 642             // such as for OMX_CommandFlush
 643             if (cmd == OMX_CommandStateSet && state == OMX_StateIdle)
 644                 break;
 645         }
 646     }
 647
 648     omx_error = OMX_GetState(omx_handle, &state);
 649     CHECK_ERROR(omx_error, "OMX_GetState failed (%x)", omx_error );
 650
 651     if(state == OMX_StateIdle)
 652     {
 653         omx_error = OMX_SendCommand( omx_handle, OMX_CommandStateSet,
 654                                      OMX_StateLoaded, 0 );
 655         CHECK_ERROR(omx_error, "OMX_CommandStateSet Loaded failed (%x)", omx_error );
 656
 657         for(i = 0; i < p_sys->ports; i++)
 658         {
 659             OmxPort *p_port = &p_sys->p_ports[i];
 660             OMX_BUFFERHEADERTYPE *p_buffer;
 661
 662             for(j = 0; j < p_port->i_buffers; j++)
 663             {
 664                 OMX_FIFO_GET(&p_port->fifo, p_buffer);
 665                 if (p_buffer->nFlags & SENTINEL_FLAG) {
 666                     free(p_buffer);
 667                     j--;
 668                     continue;
 669                 }
 670                 omx_error = OMX_FreeBuffer( omx_handle,
 671                                             p_port->i_port_index, p_buffer );
 672
 673                 if(omx_error != OMX_ErrorNone) break;
 674             }
 675             CHECK_ERROR(omx_error, "OMX_FreeBuffer failed (%x, %i, %i)",
 676                         omx_error, (int)p_port->i_port_index, j );
 677             while (1) {
 678                 OMX_FIFO_PEEK(&p_port->fifo, p_buffer);
 679                 if (!p_buffer) break;
 680
 681                 OMX_FIFO_GET(&p_port->fifo, p_buffer);
 682                 if (p_buffer->nFlags & SENTINEL_FLAG) {
 683                     free(p_buffer);
 684                     continue;
 685                 }
 686                 msg_Warn( p_dec, "Stray buffer left in fifo, %p", p_buffer );
 687             }
 688         }
 689
 690         omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
 691         CHECK_ERROR(omx_error, "Wait for Loaded failed (%x)", omx_error );
 692     }
 693
 694  error:
 695     for(i = 0; i < p_sys->ports; i++)
 696     {
 697         OmxPort *p_port = &p_sys->p_ports[i];
 698         free(p_port->pp_buffers);
 699         p_port->pp_buffers = 0;
 700     }
 701     omx_error = pf_free_handle( omx_handle );
 702     return omx_error;
 703 }
 704
 705 /*****************************************************************************
 706  * InitialiseComponent: Load and initialise an OMX component
 707  *****************************************************************************/
 708 static OMX_ERRORTYPE InitialiseComponent(decoder_t *p_dec,
 709     OMX_STRING psz_component, OMX_HANDLETYPE *p_handle)
 710 {
 711     static OMX_CALLBACKTYPE callbacks =
 712         { OmxEventHandler, OmxEmptyBufferDone, OmxFillBufferDone };
 713     decoder_sys_t *p_sys = p_dec->p_sys;
 714     OMX_HANDLETYPE omx_handle;
 715     OMX_ERRORTYPE omx_error;
 716     unsigned int i;
 717     OMX_U8 psz_role[OMX_MAX_STRINGNAME_SIZE];
 718     OMX_PARAM_COMPONENTROLETYPE role;
 719     OMX_PARAM_PORTDEFINITIONTYPE definition;
 720     OMX_PORT_PARAM_TYPE param;
 721
 722     /* Load component */
 723     omx_error = pf_get_handle( &omx_handle, psz_component, p_dec, &callbacks );
 724     if(omx_error != OMX_ErrorNone)
 725     {
 726         msg_Warn( p_dec, "OMX_GetHandle(%s) failed (%x: %s)", psz_component,
 727                   omx_error, ErrorToString(omx_error) );
 728         return omx_error;
 729     }
 730     strncpy(p_sys->psz_component, psz_component, OMX_MAX_STRINGNAME_SIZE-1);
 731
 732     omx_error = OMX_ComponentRoleEnum(omx_handle, psz_role, 0);
 733     if(omx_error == OMX_ErrorNone)
 734         msg_Dbg(p_dec, "loaded component %s of role %s", psz_component, psz_role);
 735     else
 736         msg_Dbg(p_dec, "loaded component %s", psz_component);
 737     PrintOmx(p_dec, omx_handle, OMX_ALL);
 738
 739     /* Set component role */
 740     OMX_INIT_STRUCTURE(role);
 741     strcpy((char*)role.cRole,
 742            GetOmxRole(p_sys->b_enc ? p_dec->fmt_out.i_codec : p_dec->fmt_in.i_codec,
 743                       p_dec->fmt_in.i_cat, p_sys->b_enc));
 744
 745     omx_error = OMX_SetParameter(omx_handle, OMX_IndexParamStandardComponentRole,
 746                                  &role);
 747     omx_error = OMX_GetParameter(omx_handle, OMX_IndexParamStandardComponentRole,
 748                                  &role);
 749     if(omx_error == OMX_ErrorNone)
 750         msg_Dbg(p_dec, "component standard role set to %s", role.cRole);
 751
 752     /* Find the input / output ports */
 753     OMX_INIT_STRUCTURE(param);
 754     OMX_INIT_STRUCTURE(definition);
 755     omx_error = OMX_GetParameter(omx_handle, p_dec->fmt_in.i_cat == VIDEO_ES ?
 756                                  OMX_IndexParamVideoInit : OMX_IndexParamAudioInit, &param);
 757     if(omx_error != OMX_ErrorNone) {
 758 #ifdef __ANDROID__
 759         param.nPorts = 2;
 760         param.nStartPortNumber = 0;
 761 #else
 762         param.nPorts = 0;
 763 #endif
 764     }
 765
 766     for(i = 0; i < param.nPorts; i++)
 767     {
 768         OmxPort *p_port;
 769
 770         /* Get port definition */
 771         definition.nPortIndex = param.nStartPortNumber + i;
 772         omx_error = OMX_GetParameter(omx_handle, OMX_IndexParamPortDefinition,
 773                                      &definition);
 774         if(omx_error != OMX_ErrorNone) continue;
 775
 776         if(definition.eDir == OMX_DirInput) p_port = &p_sys->in;
 777         else  p_port = &p_sys->out;
 778
 779         p_port->b_valid = true;
 780         p_port->i_port_index = definition.nPortIndex;
 781         p_port->definition = definition;
 782         p_port->omx_handle = omx_handle;
 783     }
 784
 785     if(!p_sys->in.b_valid || !p_sys->out.b_valid)
 786     {
 787         omx_error = OMX_ErrorInvalidComponent;
 788         CHECK_ERROR(omx_error, "couldn't find an input and output port");
 789     }
 790
 791     if(!strncmp(p_sys->psz_component, "OMX.SEC.", 8))
 792     {
 793         OMX_INDEXTYPE index;
 794         omx_error = OMX_GetExtensionIndex(omx_handle, (OMX_STRING) "OMX.SEC.index.ThumbnailMode", &index);
 795         if(omx_error == OMX_ErrorNone)
 796         {
 797             OMX_BOOL enable = OMX_TRUE;
 798             omx_error = OMX_SetConfig(omx_handle, index, &enable);
 799             CHECK_ERROR(omx_error, "Unable to set ThumbnailMode");
 800         } else {
 801             OMX_BOOL enable = OMX_TRUE;
 802             /* Needed on Samsung Galaxy S II */
 803             omx_error = OMX_SetConfig(omx_handle, OMX_IndexVendorSetYUV420pMode, &enable);
 804             if (omx_error == OMX_ErrorNone)
 805                 msg_Dbg(p_dec, "Set OMX_IndexVendorSetYUV420pMode successfully");
 806             else
 807                 msg_Dbg(p_dec, "Unable to set OMX_IndexVendorSetYUV420pMode: %x", omx_error);
 808         }
 809     }
 810
 811     /* Set port definitions */
 812     for(i = 0; i < p_sys->ports; i++)
 813     {
 814         omx_error = SetPortDefinition(p_dec, &p_sys->p_ports[i],
 815                                       p_sys->p_ports[i].p_fmt);
 816         if(omx_error != OMX_ErrorNone) goto error;
 817     }
 818
 819     /* Allocate our array for the omx buffers and enable ports */
 820     for(i = 0; i < p_sys->ports; i++)
 821     {
 822         OmxPort *p_port = &p_sys->p_ports[i];
 823
 824         p_port->pp_buffers =
 825             malloc(p_port->definition.nBufferCountActual *
 826                    sizeof(OMX_BUFFERHEADERTYPE*));
 827         if(!p_port->pp_buffers)
 828         {
 829           omx_error = OMX_ErrorInsufficientResources;
 830           CHECK_ERROR(omx_error, "memory allocation failed");
 831         }
 832         p_port->i_buffers = p_port->definition.nBufferCountActual;
 833
 834         /* Enable port */
 835         if(!p_port->definition.bEnabled)
 836         {
 837             omx_error = OMX_SendCommand( omx_handle, OMX_CommandPortEnable,
 838                                          p_port->i_port_index, NULL);
 839             CHECK_ERROR(omx_error, "OMX_CommandPortEnable on %i failed (%x)",
 840                         (int)p_port->i_port_index, omx_error );
 841             omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
 842             CHECK_ERROR(omx_error, "Wait for PortEnable on %i failed (%x)",
 843                         (int)p_port->i_port_index, omx_error );
 844         }
 845     }
 846
 847     *p_handle = omx_handle;
 848     return OMX_ErrorNone;
 849
 850  error:
 851     DeinitialiseComponent(p_dec, omx_handle);
 852     *p_handle = 0;
 853     return omx_error;
 854 }
 855
 856 /*****************************************************************************
 857  * OpenDecoder: Create the decoder instance
 858  *****************************************************************************/
 859 static int OpenDecoder( vlc_object_t *p_this )
 860 {
 861     decoder_t *p_dec = (decoder_t*)p_this;
 862     int status;
 863
 864     if( 0 || !GetOmxRole(p_dec->fmt_in.i_codec, p_dec->fmt_in.i_cat, false) )
 865         return VLC_EGENERIC;
 866
 867 #ifdef HAVE_MAEMO
 868     if( p_dec->fmt_in.i_cat != VIDEO_ES && !p_dec->b_force)
 869         return VLC_EGENERIC;
 870 #endif
 871
 872     status = OpenGeneric( p_this, false );
 873     if(status != VLC_SUCCESS) return status;
 874
 875     p_dec->pf_decode_video = DecodeVideo;
 876     p_dec->pf_decode_audio = DecodeAudio;
 877
 878     return VLC_SUCCESS;
 879 }
 880
 881 /*****************************************************************************
 882  * OpenEncoder: Create the encoder instance
 883  *****************************************************************************/
 884 static int OpenEncoder( vlc_object_t *p_this )
 885 {
 886     encoder_t *p_enc = (encoder_t*)p_this;
 887     int status;
 888
 889     if( !GetOmxRole(p_enc->fmt_out.i_codec, p_enc->fmt_in.i_cat, true) )
 890         return VLC_EGENERIC;
 891
 892     status = OpenGeneric( p_this, true );
 893     if(status != VLC_SUCCESS) return status;
 894
 895     p_enc->pf_encode_video = EncodeVideo;
 896
 897     return VLC_SUCCESS;
 898 }
 899
 900 /*****************************************************************************
 901  * OpenGeneric: Create the generic decoder/encoder instance
 902  *****************************************************************************/
 903 static int OpenGeneric( vlc_object_t *p_this, bool b_encode )
 904 {
 905     decoder_t *p_dec = (decoder_t*)p_this;
 906     decoder_sys_t *p_sys;
 907     OMX_ERRORTYPE omx_error;
 908     OMX_BUFFERHEADERTYPE *p_header;
 909     unsigned int i, j;
 910
 911     vlc_mutex_lock( &omx_core_mutex );
 912     if( omx_refcount > 0 )
 913         goto loaded;
 914
 915     /* Load the OMX core */
 916     for( i = 0; ppsz_dll_list[i]; i++ )
 917     {
 918         dll_handle = dll_open( ppsz_dll_list[i] );
 919         if( dll_handle ) break;
 920     }
 921     if( !dll_handle )
 922     {
 923         vlc_mutex_unlock( &omx_core_mutex );
 924         return VLC_EGENERIC;
 925     }
 926
 927     pf_init = dlsym( dll_handle, "OMX_Init" );
 928     pf_deinit = dlsym( dll_handle, "OMX_Deinit" );
 929     pf_get_handle = dlsym( dll_handle, "OMX_GetHandle" );
 930     pf_free_handle = dlsym( dll_handle, "OMX_FreeHandle" );
 931     pf_component_enum = dlsym( dll_handle, "OMX_ComponentNameEnum" );
 932     pf_get_roles_of_component = dlsym( dll_handle, "OMX_GetRolesOfComponent" );
 933     if( !pf_init || !pf_deinit || !pf_get_handle || !pf_free_handle ||
 934         !pf_component_enum || !pf_get_roles_of_component )
 935     {
 936         msg_Warn( p_this, "cannot find OMX_* symbols in `%s' (%s)",
 937                   ppsz_dll_list[i], dlerror() );
 938         dll_close(dll_handle);
 939         vlc_mutex_unlock( &omx_core_mutex );
 940         return VLC_EGENERIC;
 941     }
 942
 943 loaded:
 944     /* Allocate the memory needed to store the decoder's structure */
 945     if( ( p_dec->p_sys = p_sys = calloc( 1, sizeof(*p_sys)) ) == NULL )
 946     {
 947         if( omx_refcount == 0 )
 948             dll_close(dll_handle);
 949         vlc_mutex_unlock( &omx_core_mutex );
 950         return VLC_ENOMEM;
 951     }
 952
 953     /* Initialise the thread properties */
 954     if(!b_encode)
 955     {
 956         p_dec->fmt_out.i_cat = p_dec->fmt_in.i_cat;
 957         p_dec->fmt_out.video = p_dec->fmt_in.video;
 958         p_dec->fmt_out.audio = p_dec->fmt_in.audio;
 959         p_dec->fmt_out.i_codec = 0;
 960     }
 961     p_sys->b_enc = b_encode;
 962     p_sys->pp_last_event = &p_sys->p_events;
 963     vlc_mutex_init (&p_sys->mutex);
 964     vlc_cond_init (&p_sys->cond);
 965     vlc_mutex_init (&p_sys->lock);
 966     vlc_mutex_init (&p_sys->in.fifo.lock);
 967     vlc_cond_init (&p_sys->in.fifo.wait);
 968     p_sys->in.fifo.offset = offsetof(OMX_BUFFERHEADERTYPE, pOutputPortPrivate) / sizeof(void *);
 969     p_sys->in.fifo.pp_last = &p_sys->in.fifo.p_first;
 970     p_sys->in.b_direct = false;
 971     p_sys->in.b_flushed = true;
 972     p_sys->in.p_fmt = &p_dec->fmt_in;
 973     vlc_mutex_init (&p_sys->out.fifo.lock);
 974     vlc_cond_init (&p_sys->out.fifo.wait);
 975     p_sys->out.fifo.offset = offsetof(OMX_BUFFERHEADERTYPE, pInputPortPrivate) / sizeof(void *);
 976     p_sys->out.fifo.pp_last = &p_sys->out.fifo.p_first;
 977     p_sys->out.b_direct = true;
 978     p_sys->out.b_flushed = true;
 979     p_sys->out.p_fmt = &p_dec->fmt_out;
 980     p_sys->ports = 2;
 981     p_sys->p_ports = &p_sys->in;
 982     p_sys->b_use_pts = 0;
 983
 984     msg_Dbg(p_dec, "fmt in:%4.4s, out: %4.4s", (char *)&p_dec->fmt_in.i_codec,
 985             (char *)&p_dec->fmt_out.i_codec);
 986
 987     /* Initialise the OMX core */
 988     omx_error = omx_refcount > 0 ? OMX_ErrorNone : pf_init();
 989     omx_refcount++;
 990     if(omx_error != OMX_ErrorNone)
 991     {
 992         msg_Warn( p_this, "OMX_Init failed (%x: %s)", omx_error,
 993                   ErrorToString(omx_error) );
 994         vlc_mutex_unlock( &omx_core_mutex );
 995         CloseGeneric(p_this);
 996         return VLC_EGENERIC;
 997     }
 998     p_sys->b_init = true;
 999     vlc_mutex_unlock( &omx_core_mutex );
1000
1001     /* Enumerate components and build a list of the one we want to try */
1002     if( !CreateComponentsList(p_dec,
1003              GetOmxRole(p_sys->b_enc ? p_dec->fmt_out.i_codec :
1004                         p_dec->fmt_in.i_codec, p_dec->fmt_in.i_cat,
1005                         p_sys->b_enc)) )
1006     {
1007         msg_Warn( p_this, "couldn't find an omx component for codec %4.4s",
1008                   (char *)&p_dec->fmt_in.i_codec );
1009         CloseGeneric(p_this);
1010         return VLC_EGENERIC;
1011     }
1012
1013     /* Try to load and initialise a component */
1014     omx_error = OMX_ErrorUndefined;
1015     for(i = 0; i < p_sys->components; i++)
1016     {
1017 #ifdef __ANDROID__
1018         /* ignore OpenCore software codecs */
1019         if (!strncmp(p_sys->ppsz_components[i], "OMX.PV.", 7))
1020             continue;
1021         /* The same sw codecs, renamed in ICS (perhaps also in honeycomb) */
1022         if (!strncmp(p_sys->ppsz_components[i], "OMX.google.", 11))
1023             continue;
1024         /* This one has been seen on HTC One V - it behaves like it works,
1025          * but FillBufferDone returns buffers filled with 0 bytes. The One V
1026          * has got a working OMX.qcom.video.decoder.avc instead though. */
1027         if (!strncmp(p_sys->ppsz_components[i], "OMX.ARICENT.", 12))
1028             continue;
1029         /* Some nVidia codec with DRM */
1030         if (!strncmp(p_sys->ppsz_components[i], "OMX.Nvidia.h264.decode.secure", 29))
1031             continue;
1032         /* Use VC1 decoder for WMV3 for now */
1033         if (!strcmp(p_sys->ppsz_components[i], "OMX.SEC.WMV.Decoder"))
1034             continue;
1035         /* This decoder does work, but has an insane latency (leading to errors
1036          * about "main audio output playback way too late" and dropped frames).
1037          * At least Samsung Galaxy S III (where this decoder is present) has
1038          * got another one, OMX.SEC.mp3.dec, that works well and has a
1039          * sensible latency. (Also, even if that one isn't found, in general,
1040          * using SW codecs is usually more than fast enough for MP3.) */
1041         if (!strcmp(p_sys->ppsz_components[i], "OMX.SEC.MP3.Decoder"))
1042             continue;
1043 #endif
1044         omx_error = InitialiseComponent(p_dec, p_sys->ppsz_components[i],
1045                                         &p_sys->omx_handle);
1046         if(omx_error == OMX_ErrorNone) break;
1047     }
1048     CHECK_ERROR(omx_error, "no component could be initialised" );
1049
1050     /* Move component to Idle then Executing state */
1051     OMX_SendCommand( p_sys->omx_handle, OMX_CommandStateSet, OMX_StateIdle, 0 );
1052     CHECK_ERROR(omx_error, "OMX_CommandStateSet Idle failed (%x)", omx_error );
1053
1054     /* Allocate omx buffers */
1055     for(i = 0; i < p_sys->ports; i++)
1056     {
1057         OmxPort *p_port = &p_sys->p_ports[i];
1058
1059         for(j = 0; j < p_port->i_buffers; j++)
1060         {
1061 #if 0
1062 #define ALIGN(x,BLOCKLIGN) (((x) + BLOCKLIGN - 1) & ~(BLOCKLIGN - 1))
1063             char *p_buf = malloc(p_port->definition.nBufferSize +
1064                                  p_port->definition.nBufferAlignment);
1065             p_port->pp_buffers[i] = (void *)ALIGN((uintptr_t)p_buf, p_port->definition.nBufferAlignment);
1066 #endif
1067
1068             if(0 && p_port->b_direct)
1069                 omx_error =
1070                     OMX_UseBuffer( p_sys->omx_handle, &p_port->pp_buffers[j],
1071                                    p_port->i_port_index, 0,
1072                                    p_port->definition.nBufferSize, (void*)1);
1073             else
1074                 omx_error =
1075                     OMX_AllocateBuffer( p_sys->omx_handle, &p_port->pp_buffers[j],
1076                                         p_port->i_port_index, 0,
1077                                         p_port->definition.nBufferSize);
1078
1079             if(omx_error != OMX_ErrorNone) break;
1080             OMX_FIFO_PUT(&p_port->fifo, p_port->pp_buffers[j]);
1081         }
1082         p_port->i_buffers = j;
1083         CHECK_ERROR(omx_error, "OMX_UseBuffer failed (%x, %i, %i)",
1084                     omx_error, (int)p_port->i_port_index, j );
1085     }
1086
1087     omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
1088     CHECK_ERROR(omx_error, "Wait for Idle failed (%x)", omx_error );
1089
1090     omx_error = OMX_SendCommand( p_sys->omx_handle, OMX_CommandStateSet,
1091                                  OMX_StateExecuting, 0);
1092     CHECK_ERROR(omx_error, "OMX_CommandStateSet Executing failed (%x)", omx_error );
1093     omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
1094     CHECK_ERROR(omx_error, "Wait for Executing failed (%x)", omx_error );
1095
1096     /* Send codec configuration data */
1097     if( p_dec->fmt_in.i_extra )
1098     {
1099         OMX_FIFO_GET(&p_sys->in.fifo, p_header);
1100         p_header->nFilledLen = p_dec->fmt_in.i_extra;
1101
1102         /* Convert H.264 NAL format to annex b */
1103         if( p_sys->i_nal_size_length && !p_sys->in.b_direct )
1104         {
1105             p_header->nFilledLen = 0;
1106             convert_sps_pps( p_dec, p_dec->fmt_in.p_extra, p_dec->fmt_in.i_extra,
1107                              p_header->pBuffer, p_header->nAllocLen,
1108                              (uint32_t*) &p_header->nFilledLen, NULL );
1109         }
1110         else if(p_sys->in.b_direct)
1111         {
1112             p_header->pOutputPortPrivate = p_header->pBuffer;
1113             p_header->pBuffer = p_dec->fmt_in.p_extra;
1114         }
1115         else if (p_dec->fmt_in.i_codec == VLC_CODEC_WMV3 &&
1116                  p_dec->fmt_in.i_extra >= 4 &&
1117                  p_header->nAllocLen >= 36)
1118         {
1119             int profile;
1120             // According to OMX IL 1.2.0 spec (4.3.33.2), the codec config
1121             // data for VC-1 Main/Simple (aka WMV3) is according to table 265
1122             // in the VC-1 spec. Most of the fields are just set with placeholders
1123             // (like framerate, hrd_buffer/rate).
1124             static const uint8_t wmv3seq[] = {
1125                 0xff, 0xff, 0xff, 0xc5, // numframes=ffffff, marker byte
1126                 0x04, 0x00, 0x00, 0x00, // marker byte
1127                 0x00, 0x00, 0x00, 0x00, // struct C, almost equal to p_extra
1128                 0x00, 0x00, 0x00, 0x00, // struct A, vert size
1129                 0x00, 0x00, 0x00, 0x00, // struct A, horiz size
1130                 0x0c, 0x00, 0x00, 0x00, // marker byte
1131                 0xff, 0xff, 0x00, 0x80, // struct B, level=4, cbr=0, hrd_buffer=ffff
1132                 0xff, 0xff, 0x00, 0x00, // struct B, hrd_rate=ffff
1133                 0xff, 0xff, 0xff, 0xff, // struct B, framerate=ffffffff
1134             };
1135             p_header->nFilledLen = sizeof(wmv3seq);
1136             memcpy(p_header->pBuffer, wmv3seq, p_header->nFilledLen);
1137             // Struct C - almost equal to the extradata
1138             memcpy(&p_header->pBuffer[8], p_dec->fmt_in.p_extra, 4);
1139             // Expand profile from the highest 2 bits to the highest 4 bits
1140             profile = p_header->pBuffer[8] >> 6;
1141             p_header->pBuffer[8] = (p_header->pBuffer[8] & 0x0f) | (profile << 4);
1142             // Fill in the height/width for struct A
1143             SetDWLE(&p_header->pBuffer[12], p_dec->fmt_in.video.i_height);
1144             SetDWLE(&p_header->pBuffer[16], p_dec->fmt_in.video.i_width);
1145         }
1146         else
1147         {
1148             if(p_header->nFilledLen > p_header->nAllocLen)
1149             {
1150                 msg_Dbg(p_dec, "buffer too small (%i,%i)", (int)p_header->nFilledLen,
1151                         (int)p_header->nAllocLen);
1152                 p_header->nFilledLen = p_header->nAllocLen;
1153             }
1154             memcpy(p_header->pBuffer, p_dec->fmt_in.p_extra, p_header->nFilledLen);
1155         }
1156
1157         p_header->nOffset = 0;
1158         p_header->nFlags = OMX_BUFFERFLAG_CODECCONFIG | OMX_BUFFERFLAG_ENDOFFRAME;
1159         msg_Dbg(p_dec, "sending codec config data %p, %p, %i", p_header,
1160                 p_header->pBuffer, (int)p_header->nFilledLen);
1161         OMX_EmptyThisBuffer(p_sys->omx_handle, p_header);
1162     }
1163
1164     /* Get back output port definition */
1165     omx_error = GetPortDefinition(p_dec, &p_sys->out, p_sys->out.p_fmt);
1166     if(omx_error != OMX_ErrorNone) goto error;
1167
1168     PrintOmx(p_dec, p_sys->omx_handle, p_dec->p_sys->in.i_port_index);
1169     PrintOmx(p_dec, p_sys->omx_handle, p_dec->p_sys->out.i_port_index);
1170
1171     if(p_sys->b_error) goto error;
1172
1173     p_dec->b_need_packetized = true;
1174     if (!strcmp(p_sys->psz_component, "OMX.TI.DUCATI1.VIDEO.DECODER"))
1175         p_sys->b_use_pts = 1;
1176
1177     if (!strcmp(p_sys->psz_component, "OMX.STM.Video.Decoder"))
1178         p_sys->b_use_pts = 1;
1179
1180     if (p_sys->b_use_pts)
1181         msg_Dbg( p_dec, "using pts timestamp mode for %s", p_sys->psz_component);
1182
1183     return VLC_SUCCESS;
1184
1185  error:
1186     CloseGeneric(p_this);
1187     return VLC_EGENERIC;
1188 }
1189
1190 /*****************************************************************************
1191  * PortReconfigure
1192  *****************************************************************************/
1193 static OMX_ERRORTYPE PortReconfigure(decoder_t *p_dec, OmxPort *p_port)
1194 {
1195     decoder_sys_t *p_sys = p_dec->p_sys;
1196     OMX_PARAM_PORTDEFINITIONTYPE definition;
1197     OMX_BUFFERHEADERTYPE *p_buffer;
1198     OMX_ERRORTYPE omx_error;
1199     unsigned int i;
1200
1201     /* Sanity checking */
1202     OMX_INIT_STRUCTURE(definition);
1203     definition.nPortIndex = p_port->i_port_index;
1204     omx_error = OMX_GetParameter(p_dec->p_sys->omx_handle, OMX_IndexParamPortDefinition,
1205                                  &definition);
1206     if(omx_error != OMX_ErrorNone || (p_dec->fmt_in.i_cat == VIDEO_ES &&
1207        (!definition.format.video.nFrameWidth ||
1208        !definition.format.video.nFrameHeight)) )
1209         return OMX_ErrorUndefined;
1210
1211     omx_error = OMX_SendCommand( p_sys->omx_handle, OMX_CommandPortDisable,
1212                                  p_port->i_port_index, NULL);
1213     CHECK_ERROR(omx_error, "OMX_CommandPortDisable on %i failed (%x)",
1214                 (int)p_port->i_port_index, omx_error );
1215
1216     for(i = 0; i < p_port->i_buffers; i++)
1217     {
1218         OMX_FIFO_GET(&p_port->fifo, p_buffer);
1219         if (p_buffer->nFlags & SENTINEL_FLAG) {
1220             free(p_buffer);
1221             i--;
1222             continue;
1223         }
1224         omx_error = OMX_FreeBuffer( p_sys->omx_handle,
1225                                     p_port->i_port_index, p_buffer );
1226
1227         if(omx_error != OMX_ErrorNone) break;
1228     }
1229     CHECK_ERROR(omx_error, "OMX_FreeBuffer failed (%x, %i, %i)",
1230                 omx_error, (int)p_port->i_port_index, i );
1231
1232     omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
1233     CHECK_ERROR(omx_error, "Wait for PortDisable failed (%x)", omx_error );
1234
1235     /* Get the new port definition */
1236     omx_error = GetPortDefinition(p_dec, &p_sys->out, p_sys->out.p_fmt);
1237     if(omx_error != OMX_ErrorNone) goto error;
1238
1239     if( p_dec->fmt_in.i_cat != AUDIO_ES )
1240     {
1241         /* Don't explicitly set the new parameters that we got with
1242          * OMX_GetParameter above when using audio codecs.
1243          * That struct hasn't been changed since, so there should be
1244          * no need to set it here, unless some codec expects the
1245          * SetParameter call as a trigger event for some part of
1246          * the reconfiguration.
1247          * This fixes using audio decoders on Samsung Galaxy S II,
1248          *
1249          * Only skipping this for audio codecs, to minimize the
1250          * change for current working configurations for video.
1251          */
1252         omx_error = OMX_SetParameter(p_dec->p_sys->omx_handle, OMX_IndexParamPortDefinition,
1253                                      &definition);
1254         CHECK_ERROR(omx_error, "OMX_SetParameter failed (%x : %s)",
1255                     omx_error, ErrorToString(omx_error));
1256     }
1257
1258     omx_error = OMX_SendCommand( p_sys->omx_handle, OMX_CommandPortEnable,
1259                                  p_port->i_port_index, NULL);
1260     CHECK_ERROR(omx_error, "OMX_CommandPortEnable on %i failed (%x)",
1261                 (int)p_port->i_port_index, omx_error );
1262
1263     if (p_port->definition.nBufferCountActual > p_port->i_buffers) {
1264         free(p_port->pp_buffers);
1265         p_port->pp_buffers = malloc(p_port->definition.nBufferCountActual * sizeof(OMX_BUFFERHEADERTYPE*));
1266         if(!p_port->pp_buffers)
1267         {
1268             omx_error = OMX_ErrorInsufficientResources;
1269             CHECK_ERROR(omx_error, "memory allocation failed");
1270         }
1271     }
1272     p_port->i_buffers = p_port->definition.nBufferCountActual;
1273     for(i = 0; i < p_port->i_buffers; i++)
1274     {
1275         if(0 && p_port->b_direct)
1276             omx_error =
1277                 OMX_UseBuffer( p_sys->omx_handle, &p_port->pp_buffers[i],
1278                                p_port->i_port_index, 0,
1279                                p_port->definition.nBufferSize, (void*)1);
1280         else
1281             omx_error =
1282                 OMX_AllocateBuffer( p_sys->omx_handle, &p_port->pp_buffers[i],
1283                                     p_port->i_port_index, 0,
1284                                     p_port->definition.nBufferSize);
1285
1286         if(omx_error != OMX_ErrorNone) break;
1287         OMX_FIFO_PUT(&p_port->fifo, p_port->pp_buffers[i]);
1288     }
1289     p_port->i_buffers = i;
1290     CHECK_ERROR(omx_error, "OMX_UseBuffer failed (%x, %i, %i)",
1291                 omx_error, (int)p_port->i_port_index, i );
1292
1293     omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
1294     CHECK_ERROR(omx_error, "Wait for PortEnable failed (%x)", omx_error );
1295
1296     PrintOmx(p_dec, p_sys->omx_handle, p_dec->p_sys->in.i_port_index);
1297     PrintOmx(p_dec, p_sys->omx_handle, p_dec->p_sys->out.i_port_index);
1298
1299  error:
1300     return omx_error;
1301 }
1302
1303 /*****************************************************************************
1304  * DecodeVideo: Called to decode one frame
1305  *****************************************************************************/
1306 static picture_t *DecodeVideo( decoder_t *p_dec, block_t **pp_block )
1307 {
1308     decoder_sys_t *p_sys = p_dec->p_sys;
1309     picture_t *p_pic = NULL, *p_next_pic;
1310     OMX_ERRORTYPE omx_error;
1311     unsigned int i;
1312
1313     OMX_BUFFERHEADERTYPE *p_header;
1314     block_t *p_block;
1315
1316     if( !pp_block || !*pp_block )
1317         return NULL;
1318
1319     p_block = *pp_block;
1320
1321     /* Check for errors from codec */
1322     if(p_sys->b_error)
1323     {
1324         msg_Dbg(p_dec, "error during decoding");
1325         block_Release( p_block );
1326         return 0;
1327     }
1328
1329     if( p_block->i_flags & (BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
1330     {
1331         block_Release( p_block );
1332         if(!p_sys->in.b_flushed)
1333         {
1334             msg_Dbg(p_dec, "flushing");
1335             OMX_SendCommand( p_sys->omx_handle, OMX_CommandFlush,
1336                              p_sys->in.definition.nPortIndex, 0 );
1337         }
1338         p_sys->in.b_flushed = true;
1339         return NULL;
1340     }
1341
1342     /* Take care of decoded frames first */
1343     while(!p_pic)
1344     {
1345         OMX_FIFO_PEEK(&p_sys->out.fifo, p_header);
1346         if(!p_header) break; /* No frame available */
1347
1348         if(p_sys->out.b_update_def)
1349         {
1350             omx_error = GetPortDefinition(p_dec, &p_sys->out, p_sys->out.p_fmt);
1351             p_sys->out.b_update_def = 0;
1352         }
1353
1354         if(p_header->nFilledLen)
1355         {
1356             p_pic = p_header->pAppPrivate;
1357             if(!p_pic)
1358             {
1359                 /* We're not in direct rendering mode.
1360                  * Get a new picture and copy the content */
1361                 p_pic = decoder_NewPicture( p_dec );
1362
1363                 if (p_pic)
1364                     CopyOmxPicture(p_sys->out.definition.format.video.eColorFormat,
1365                                    p_pic, p_sys->out.definition.format.video.nSliceHeight,
1366                                    p_sys->out.i_frame_stride,
1367                                    p_header->pBuffer + p_header->nOffset,
1368                                    p_sys->out.i_frame_stride_chroma_div);
1369             }
1370
1371             if (p_pic)
1372                 p_pic->date = p_header->nTimeStamp;
1373             p_header->nFilledLen = 0;
1374             p_header->pAppPrivate = 0;
1375         }
1376
1377         /* Get a new picture */
1378         if(p_sys->in.b_direct && !p_header->pAppPrivate)
1379         {
1380             p_next_pic = decoder_NewPicture( p_dec );
1381             if(!p_next_pic) break;
1382
1383             OMX_FIFO_GET(&p_sys->out.fifo, p_header);
1384             p_header->pAppPrivate = p_next_pic;
1385             p_header->pInputPortPrivate = p_header->pBuffer;
1386             p_header->pBuffer = p_next_pic->p[0].p_pixels;
1387         }
1388         else
1389         {
1390             OMX_FIFO_GET(&p_sys->out.fifo, p_header);
1391         }
1392
1393 #ifdef OMXIL_EXTRA_DEBUG
1394         msg_Dbg( p_dec, "FillThisBuffer %p, %p", p_header, p_header->pBuffer );
1395 #endif
1396         OMX_FillThisBuffer(p_sys->omx_handle, p_header);
1397     }
1398
1399     /* Send the input buffer to the component */
1400     OMX_FIFO_GET(&p_sys->in.fifo, p_header);
1401
1402     if (p_header && p_header->nFlags & SENTINEL_FLAG) {
1403         free(p_header);
1404         goto reconfig;
1405     }
1406
1407     if(p_header)
1408     {
1409         p_header->nFilledLen = p_block->i_buffer;
1410         p_header->nOffset = 0;
1411         p_header->nFlags = OMX_BUFFERFLAG_ENDOFFRAME;
1412         if (p_sys->b_use_pts && p_block->i_pts)
1413             p_header->nTimeStamp = p_block->i_pts;
1414         else
1415             p_header->nTimeStamp = p_block->i_dts;
1416
1417         /* In direct mode we pass the input pointer as is.
1418          * Otherwise we memcopy the data */
1419         if(p_sys->in.b_direct)
1420         {
1421             p_header->pOutputPortPrivate = p_header->pBuffer;
1422             p_header->pBuffer = p_block->p_buffer;
1423             p_header->pAppPrivate = p_block;
1424         }
1425         else
1426         {
1427             if(p_header->nFilledLen > p_header->nAllocLen)
1428             {
1429                 msg_Dbg(p_dec, "buffer too small (%i,%i)",
1430                         (int)p_header->nFilledLen, (int)p_header->nAllocLen);
1431                 p_header->nFilledLen = p_header->nAllocLen;
1432             }
1433             memcpy(p_header->pBuffer, p_block->p_buffer, p_header->nFilledLen );
1434             block_Release(p_block);
1435         }
1436
1437         /* Convert H.264 NAL format to annex b. Doesn't do anything if
1438          * i_nal_size_length == 0, which is the case for codecs other
1439          * than H.264 */
1440         convert_h264_to_annexb( p_header->pBuffer, p_header->nFilledLen,
1441                                 p_sys->i_nal_size_length );
1442 #ifdef OMXIL_EXTRA_DEBUG
1443         msg_Dbg( p_dec, "EmptyThisBuffer %p, %p, %i", p_header, p_header->pBuffer,
1444                  (int)p_header->nFilledLen );
1445 #endif
1446         OMX_EmptyThisBuffer(p_sys->omx_handle, p_header);
1447         p_sys->in.b_flushed = false;
1448         *pp_block = NULL; /* Avoid being fed the same packet again */
1449     }
1450
1451 reconfig:
1452     /* Handle the PortSettingsChanged events */
1453     for(i = 0; i < p_sys->ports; i++)
1454     {
1455         OmxPort *p_port = &p_sys->p_ports[i];
1456         if(p_port->b_reconfigure)
1457         {
1458             omx_error = PortReconfigure(p_dec, p_port);
1459             p_port->b_reconfigure = 0;
1460         }
1461         if(p_port->b_update_def)
1462         {
1463             omx_error = GetPortDefinition(p_dec, p_port, p_port->p_fmt);
1464             p_port->b_update_def = 0;
1465         }
1466     }
1467
1468     return p_pic;
1469 }
1470
1471 /*****************************************************************************
1472  * DecodeAudio: Called to decode one frame
1473  *****************************************************************************/
1474 block_t *DecodeAudio ( decoder_t *p_dec, block_t **pp_block )
1475 {
1476     decoder_sys_t *p_sys = p_dec->p_sys;
1477     block_t *p_buffer = NULL;
1478     OMX_BUFFERHEADERTYPE *p_header;
1479     OMX_ERRORTYPE omx_error;
1480     block_t *p_block;
1481     unsigned int i;
1482
1483     if( !pp_block || !*pp_block ) return NULL;
1484
1485     p_block = *pp_block;
1486
1487     /* Check for errors from codec */
1488     if(p_sys->b_error)
1489     {
1490         msg_Dbg(p_dec, "error during decoding");
1491         block_Release( p_block );
1492         return 0;
1493     }
1494
1495     if( p_block->i_flags & (BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
1496     {
1497         block_Release( p_block );
1498         date_Set( &p_sys->end_date, 0 );
1499         if(!p_sys->in.b_flushed)
1500         {
1501             msg_Dbg(p_dec, "flushing");
1502             OMX_SendCommand( p_sys->omx_handle, OMX_CommandFlush,
1503                              p_sys->in.definition.nPortIndex, 0 );
1504         }
1505         p_sys->in.b_flushed = true;
1506         return NULL;
1507     }
1508
1509     if( !date_Get( &p_sys->end_date ) )
1510     {
1511         if( !p_block->i_pts )
1512         {
1513             /* We've just started the stream, wait for the first PTS. */
1514             block_Release( p_block );
1515             return NULL;
1516         }
1517         date_Set( &p_sys->end_date, p_block->i_pts );
1518     }
1519
1520     /* Take care of decoded frames first */
1521     while(!p_buffer)
1522     {
1523         unsigned int i_samples = 0;
1524
1525         OMX_FIFO_PEEK(&p_sys->out.fifo, p_header);
1526         if(!p_header) break; /* No frame available */
1527
1528         if (p_sys->out.p_fmt->audio.i_channels)
1529             i_samples = p_header->nFilledLen / p_sys->out.p_fmt->audio.i_channels / 2;
1530         if(i_samples)
1531         {
1532             p_buffer = decoder_NewAudioBuffer( p_dec, i_samples );
1533             if( !p_buffer ) break; /* No audio buffer available */
1534
1535             memcpy( p_buffer->p_buffer, p_header->pBuffer, p_buffer->i_buffer );
1536             p_header->nFilledLen = 0;
1537
1538             if( p_header->nTimeStamp != 0 &&
1539                 p_header->nTimeStamp != date_Get( &p_sys->end_date ) )
1540                 date_Set( &p_sys->end_date, p_header->nTimeStamp );
1541
1542             p_buffer->i_pts = date_Get( &p_sys->end_date );
1543             p_buffer->i_length = date_Increment( &p_sys->end_date, i_samples ) -
1544                 p_buffer->i_pts;
1545         }
1546
1547 #ifdef OMXIL_EXTRA_DEBUG
1548         msg_Dbg( p_dec, "FillThisBuffer %p, %p", p_header, p_header->pBuffer );
1549 #endif
1550         OMX_FIFO_GET(&p_sys->out.fifo, p_header);
1551         OMX_FillThisBuffer(p_sys->omx_handle, p_header);
1552     }
1553
1554
1555     /* Send the input buffer to the component */
1556     OMX_FIFO_GET(&p_sys->in.fifo, p_header);
1557
1558     if (p_header && p_header->nFlags & SENTINEL_FLAG) {
1559         free(p_header);
1560         goto reconfig;
1561     }
1562
1563     if(p_header)
1564     {
1565         p_header->nFilledLen = p_block->i_buffer;
1566         p_header->nOffset = 0;
1567         p_header->nFlags = OMX_BUFFERFLAG_ENDOFFRAME;
1568         p_header->nTimeStamp = p_block->i_dts;
1569
1570         /* In direct mode we pass the input pointer as is.
1571          * Otherwise we memcopy the data */
1572         if(p_sys->in.b_direct)
1573         {
1574             p_header->pOutputPortPrivate = p_header->pBuffer;
1575             p_header->pBuffer = p_block->p_buffer;
1576             p_header->pAppPrivate = p_block;
1577         }
1578         else
1579         {
1580             if(p_header->nFilledLen > p_header->nAllocLen)
1581             {
1582                 msg_Dbg(p_dec, "buffer too small (%i,%i)",
1583                         (int)p_header->nFilledLen, (int)p_header->nAllocLen);
1584                 p_header->nFilledLen = p_header->nAllocLen;
1585             }
1586             memcpy(p_header->pBuffer, p_block->p_buffer, p_header->nFilledLen );
1587             block_Release(p_block);
1588         }
1589
1590 #ifdef OMXIL_EXTRA_DEBUG
1591         msg_Dbg( p_dec, "EmptyThisBuffer %p, %p, %i", p_header, p_header->pBuffer,
1592                  (int)p_header->nFilledLen );
1593 #endif
1594         OMX_EmptyThisBuffer(p_sys->omx_handle, p_header);
1595         p_sys->in.b_flushed = false;
1596         *pp_block = NULL; /* Avoid being fed the same packet again */
1597     }
1598
1599 reconfig:
1600     /* Handle the PortSettingsChanged events */
1601     for(i = 0; i < p_sys->ports; i++)
1602     {
1603         OmxPort *p_port = &p_sys->p_ports[i];
1604         if(!p_port->b_reconfigure) continue;
1605         p_port->b_reconfigure = 0;
1606         omx_error = PortReconfigure(p_dec, p_port);
1607     }
1608
1609     return p_buffer;
1610 }
1611
1612 /*****************************************************************************
1613  * EncodeVideo: Called to encode one frame
1614  *****************************************************************************/
1615 static block_t *EncodeVideo( encoder_t *p_enc, picture_t *p_pic )
1616 {
1617     decoder_t *p_dec = ( decoder_t *)p_enc;
1618     decoder_sys_t *p_sys = p_dec->p_sys;
1619     OMX_ERRORTYPE omx_error;
1620     unsigned int i;
1621
1622     OMX_BUFFERHEADERTYPE *p_header;
1623     block_t *p_block = 0;
1624
1625     if( !p_pic ) return NULL;
1626
1627     /* Check for errors from codec */
1628     if(p_sys->b_error)
1629     {
1630         msg_Dbg(p_dec, "error during encoding");
1631         return NULL;
1632     }
1633
1634     /* Send the input buffer to the component */
1635     OMX_FIFO_GET(&p_sys->in.fifo, p_header);
1636     if(p_header)
1637     {
1638         /* In direct mode we pass the input pointer as is.
1639          * Otherwise we memcopy the data */
1640         if(p_sys->in.b_direct)
1641         {
1642             p_header->pOutputPortPrivate = p_header->pBuffer;
1643             p_header->pBuffer = p_pic->p[0].p_pixels;
1644         }
1645         else
1646         {
1647             CopyVlcPicture(p_dec, p_header, p_pic);
1648         }
1649
1650         p_header->nFilledLen = p_sys->in.i_frame_size;
1651         p_header->nOffset = 0;
1652         p_header->nFlags = OMX_BUFFERFLAG_ENDOFFRAME;
1653         p_header->nTimeStamp = p_pic->date;
1654 #ifdef OMXIL_EXTRA_DEBUG
1655         msg_Dbg( p_dec, "EmptyThisBuffer %p, %p, %i", p_header, p_header->pBuffer,
1656                  (int)p_header->nFilledLen );
1657 #endif
1658         OMX_EmptyThisBuffer(p_sys->omx_handle, p_header);
1659         p_sys->in.b_flushed = false;
1660     }
1661
1662     /* Handle the PortSettingsChanged events */
1663     for(i = 0; i < p_sys->ports; i++)
1664     {
1665         OmxPort *p_port = &p_sys->p_ports[i];
1666         if(!p_port->b_reconfigure) continue;
1667         p_port->b_reconfigure = 0;
1668         omx_error = PortReconfigure(p_dec, p_port);
1669     }
1670
1671     /* Wait for the decoded frame */
1672     while(!p_block)
1673     {
1674         OMX_FIFO_GET(&p_sys->out.fifo, p_header);
1675
1676         if(p_header->nFilledLen)
1677         {
1678             if(p_header->nFlags & OMX_BUFFERFLAG_CODECCONFIG)
1679             {
1680                 /* TODO: need to store codec config */
1681                 msg_Dbg(p_dec, "received codec config %i", (int)p_header->nFilledLen);
1682             }
1683
1684             p_block = p_header->pAppPrivate;
1685             if(!p_block)
1686             {
1687                 /* We're not in direct rendering mode.
1688                  * Get a new block and copy the content */
1689                 p_block = block_Alloc( p_header->nFilledLen );
1690                 memcpy(p_block->p_buffer, p_header->pBuffer, p_header->nFilledLen );
1691             }
1692
1693             p_block->i_buffer = p_header->nFilledLen;
1694             p_block->i_pts = p_block->i_dts = p_header->nTimeStamp;
1695             p_header->nFilledLen = 0;
1696             p_header->pAppPrivate = 0;
1697         }
1698
1699 #ifdef OMXIL_EXTRA_DEBUG
1700         msg_Dbg( p_dec, "FillThisBuffer %p, %p", p_header, p_header->pBuffer );
1701 #endif
1702         OMX_FillThisBuffer(p_sys->omx_handle, p_header);
1703     }
1704
1705     msg_Dbg(p_dec, "done");
1706     return p_block;
1707 }
1708
1709 /*****************************************************************************
1710  * CloseGeneric: omxil decoder destruction
1711  *****************************************************************************/
1712 static void CloseGeneric( vlc_object_t *p_this )
1713 {
1714     decoder_t *p_dec = (decoder_t *)p_this;
1715     decoder_sys_t *p_sys = p_dec->p_sys;
1716
1717     if(p_sys->omx_handle) DeinitialiseComponent(p_dec, p_sys->omx_handle);
1718     vlc_mutex_lock( &omx_core_mutex );
1719     omx_refcount--;
1720     if( omx_refcount == 0 )
1721     {
1722         if( p_sys->b_init ) pf_deinit();
1723         dll_close( dll_handle );
1724     }
1725     vlc_mutex_unlock( &omx_core_mutex );
1726
1727     vlc_mutex_destroy (&p_sys->mutex);
1728     vlc_cond_destroy (&p_sys->cond);
1729     vlc_mutex_destroy (&p_sys->in.fifo.lock);
1730     vlc_cond_destroy (&p_sys->in.fifo.wait);
1731     vlc_mutex_destroy (&p_sys->out.fifo.lock);
1732     vlc_cond_destroy (&p_sys->out.fifo.wait);
1733
1734     free( p_sys );
1735 }
1736
1737 /*****************************************************************************
1738  * OmxEventHandler:
1739  *****************************************************************************/
1740 static OMX_ERRORTYPE OmxEventHandler( OMX_HANDLETYPE omx_handle,
1741     OMX_PTR app_data, OMX_EVENTTYPE event, OMX_U32 data_1,
1742     OMX_U32 data_2, OMX_PTR event_data )
1743 {
1744     decoder_t *p_dec = (decoder_t *)app_data;
1745     decoder_sys_t *p_sys = p_dec->p_sys;
1746     unsigned int i;
1747     (void)omx_handle;
1748
1749     switch (event)
1750     {
1751     case OMX_EventCmdComplete:
1752         switch ((OMX_STATETYPE)data_1)
1753         {
1754         case OMX_CommandStateSet:
1755             msg_Dbg( p_dec, "OmxEventHandler (%s, %s, %s)", EventToString(event),
1756                      CommandToString(data_1), StateToString(data_2) );
1757             break;
1758
1759         default:
1760             msg_Dbg( p_dec, "OmxEventHandler (%s, %s, %u)", EventToString(event),
1761                      CommandToString(data_1), (unsigned int)data_2 );
1762             break;
1763         }
1764         break;
1765
1766     case OMX_EventError:
1767         msg_Dbg( p_dec, "OmxEventHandler (%s, %s, %u, %s)", EventToString(event),
1768                  ErrorToString((OMX_ERRORTYPE)data_1), (unsigned int)data_2,
1769                  (const char *)event_data);
1770         //p_sys->b_error = true;
1771         break;
1772
1773     case OMX_EventPortSettingsChanged:
1774         msg_Dbg( p_dec, "OmxEventHandler (%s, %u, %u)", EventToString(event),
1775                  (unsigned int)data_1, (unsigned int)data_2 );
1776         if( data_2 == 0 || data_2 == OMX_IndexParamPortDefinition )
1777         {
1778             OMX_BUFFERHEADERTYPE *sentinel;
1779             for(i = 0; i < p_sys->ports; i++)
1780                 if(p_sys->p_ports[i].definition.eDir == OMX_DirOutput)
1781                     p_sys->p_ports[i].b_reconfigure = true;
1782             sentinel = calloc(1, sizeof(*sentinel));
1783             if (sentinel) {
1784                 sentinel->nFlags = SENTINEL_FLAG;
1785                 OMX_FIFO_PUT(&p_sys->in.fifo, sentinel);
1786             }
1787         }
1788         else if( data_2 == OMX_IndexConfigCommonOutputCrop )
1789         {
1790             for(i = 0; i < p_sys->ports; i++)
1791                 if(p_sys->p_ports[i].definition.nPortIndex == data_1)
1792                     p_sys->p_ports[i].b_update_def = true;
1793         }
1794         else
1795         {
1796             msg_Dbg( p_dec, "Unhandled setting change %x", (unsigned int)data_2 );
1797         }
1798         break;
1799
1800     default:
1801         msg_Dbg( p_dec, "OmxEventHandler (%s, %u, %u)", EventToString(event),
1802                  (unsigned int)data_1, (unsigned int)data_2 );
1803         break;
1804     }
1805
1806     PostOmxEvent(p_dec, event, data_1, data_2, event_data);
1807     return OMX_ErrorNone;
1808 }
1809
1810 static OMX_ERRORTYPE OmxEmptyBufferDone( OMX_HANDLETYPE omx_handle,
1811     OMX_PTR app_data, OMX_BUFFERHEADERTYPE *omx_header )
1812 {
1813     decoder_t *p_dec = (decoder_t *)app_data;
1814     decoder_sys_t *p_sys = p_dec->p_sys;
1815     (void)omx_handle;
1816
1817 #ifdef OMXIL_EXTRA_DEBUG
1818     msg_Dbg( p_dec, "OmxEmptyBufferDone %p, %p", omx_header, omx_header->pBuffer );
1819 #endif
1820
1821     if(omx_header->pAppPrivate || omx_header->pOutputPortPrivate)
1822     {
1823         block_t *p_block = (block_t *)omx_header->pAppPrivate;
1824         omx_header->pBuffer = omx_header->pOutputPortPrivate;
1825         if(p_block) block_Release(p_block);
1826         omx_header->pAppPrivate = 0;
1827     }
1828     OMX_FIFO_PUT(&p_sys->in.fifo, omx_header);
1829
1830     return OMX_ErrorNone;
1831 }
1832
1833 static OMX_ERRORTYPE OmxFillBufferDone( OMX_HANDLETYPE omx_handle,
1834     OMX_PTR app_data, OMX_BUFFERHEADERTYPE *omx_header )
1835 {
1836     decoder_t *p_dec = (decoder_t *)app_data;
1837     decoder_sys_t *p_sys = p_dec->p_sys;
1838     (void)omx_handle;
1839
1840 #ifdef OMXIL_EXTRA_DEBUG
1841     msg_Dbg( p_dec, "OmxFillBufferDone %p, %p, %i", omx_header, omx_header->pBuffer,
1842              (int)omx_header->nFilledLen );
1843 #endif
1844
1845     if(omx_header->pInputPortPrivate)
1846     {
1847         omx_header->pBuffer = omx_header->pInputPortPrivate;
1848     }
1849     OMX_FIFO_PUT(&p_sys->out.fifo, omx_header);
1850
1851     return OMX_ErrorNone;
1852 }