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(!strcmp((char *)ppsz_roles[j], psz_role)) 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  error:
 441     return omx_error;
 442 }
 443
 444 /*****************************************************************************
 445  * GetPortDefinition: set vlc format based on the definition of the omx port
 446  *****************************************************************************/
 447 static OMX_ERRORTYPE GetPortDefinition(decoder_t *p_dec, OmxPort *p_port,
 448                                        es_format_t *p_fmt)
 449 {
 450     decoder_sys_t *p_sys = p_dec->p_sys;
 451     OMX_PARAM_PORTDEFINITIONTYPE *def = &p_port->definition;
 452     OMX_ERRORTYPE omx_error;
 453     OMX_CONFIG_RECTTYPE crop_rect;
 454
 455     omx_error = OMX_GetParameter(p_port->omx_handle,
 456                                  OMX_IndexParamPortDefinition, def);
 457     CHECK_ERROR(omx_error, "OMX_GetParameter failed (%x : %s)",
 458                 omx_error, ErrorToString(omx_error));
 459
 460     switch(p_fmt->i_cat)
 461     {
 462     case VIDEO_ES:
 463         p_fmt->video.i_width = def->format.video.nFrameWidth;
 464         p_fmt->video.i_visible_width = def->format.video.nFrameWidth;
 465         p_fmt->video.i_height = def->format.video.nFrameHeight;
 466         p_fmt->video.i_visible_height = def->format.video.nFrameHeight;
 467         p_fmt->video.i_frame_rate = p_dec->fmt_in.video.i_frame_rate;
 468         p_fmt->video.i_frame_rate_base = p_dec->fmt_in.video.i_frame_rate_base;
 469
 470         OMX_INIT_STRUCTURE(crop_rect);
 471         crop_rect.nPortIndex = def->nPortIndex;
 472         omx_error = OMX_GetConfig(p_port->omx_handle, OMX_IndexConfigCommonOutputCrop, &crop_rect);
 473         if (omx_error == OMX_ErrorNone)
 474         {
 475             if (!def->format.video.nSliceHeight)
 476                 def->format.video.nSliceHeight = def->format.video.nFrameHeight;
 477             if (!def->format.video.nStride)
 478                 def->format.video.nStride = def->format.video.nFrameWidth;
 479             p_fmt->video.i_width = crop_rect.nWidth;
 480             p_fmt->video.i_visible_width = crop_rect.nWidth;
 481             p_fmt->video.i_height = crop_rect.nHeight;
 482             p_fmt->video.i_visible_height = crop_rect.nHeight;
 483             if (def->format.video.eColorFormat == OMX_TI_COLOR_FormatYUV420PackedSemiPlanar)
 484                 def->format.video.nSliceHeight -= crop_rect.nTop/2;
 485         }
 486         else
 487         {
 488             /* Don't pass the error back to the caller, this isn't mandatory */
 489             omx_error = OMX_ErrorNone;
 490         }
 491
 492         /* Hack: Nexus One (stock firmware with binary OMX driver blob)
 493          * claims to output 420Planar even though it in in practice is
 494          * NV21. */
 495         if(def->format.video.eColorFormat == OMX_COLOR_FormatYUV420Planar &&
 496            !strncmp(p_sys->psz_component, "OMX.qcom.video.decoder",
 497                     strlen("OMX.qcom.video.decoder")))
 498             def->format.video.eColorFormat = OMX_QCOM_COLOR_FormatYVU420SemiPlanar;
 499
 500         if (IgnoreOmxDecoderPadding(p_sys->psz_component)) {
 501             def->format.video.nSliceHeight = 0;
 502             def->format.video.nStride = p_fmt->video.i_width;
 503         }
 504
 505         if(!GetVlcVideoFormat( def->format.video.eCompressionFormat,
 506                                &p_fmt->i_codec, 0 ) )
 507         {
 508             if( !GetVlcChromaFormat( def->format.video.eColorFormat,
 509                                      &p_fmt->i_codec, 0 ) )
 510             {
 511                 omx_error = OMX_ErrorNotImplemented;
 512                 CHECK_ERROR(omx_error, "OMX color format %i not supported",
 513                             (int)def->format.video.eColorFormat );
 514             }
 515             GetVlcChromaSizes( p_fmt->i_codec,
 516                                def->format.video.nFrameWidth,
 517                                def->format.video.nFrameHeight,
 518                                &p_port->i_frame_size, &p_port->i_frame_stride,
 519                                &p_port->i_frame_stride_chroma_div );
 520         }
 521         if(p_port->i_frame_size > def->nBufferSize)
 522             def->nBufferSize = p_port->i_frame_size;
 523         p_port->i_frame_size = def->nBufferSize;
 524 #if 0
 525         if((int)p_port->i_frame_stride > def->format.video.nStride)
 526             def->format.video.nStride = p_port->i_frame_stride;
 527 #endif
 528         p_port->i_frame_stride = def->format.video.nStride;
 529         break;
 530
 531     case AUDIO_ES:
 532         if( !OmxToVlcAudioFormat( def->format.audio.eEncoding,
 533                                 &p_fmt->i_codec, 0 ) )
 534         {
 535             omx_error = OMX_ErrorNotImplemented;
 536             CHECK_ERROR(omx_error, "OMX audio format %i not supported",
 537                         (int)def->format.audio.eEncoding );
 538         }
 539
 540         omx_error = GetAudioParameters(p_port->omx_handle,
 541                                        &p_port->format_param, def->nPortIndex,
 542                                        def->format.audio.eEncoding,
 543                                        &p_fmt->audio.i_channels,
 544                                        &p_fmt->audio.i_rate,
 545                                        &p_fmt->i_bitrate,
 546                                        &p_fmt->audio.i_bitspersample,
 547                                        &p_fmt->audio.i_blockalign);
 548         CHECK_ERROR(omx_error, "GetAudioParameters failed (%x : %s)",
 549                     omx_error, ErrorToString(omx_error));
 550
 551         if(p_fmt->audio.i_channels < 9)
 552         {
 553             static const int pi_channels_maps[9] =
 554             {
 555                 0, AOUT_CHAN_CENTER, AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
 556                 AOUT_CHAN_CENTER | AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
 557                 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_REARLEFT
 558                 | AOUT_CHAN_REARRIGHT,
 559                 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
 560                 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT,
 561                 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
 562                 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT | AOUT_CHAN_LFE,
 563                 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
 564                 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT | AOUT_CHAN_MIDDLELEFT
 565                 | AOUT_CHAN_MIDDLERIGHT,
 566                 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER | AOUT_CHAN_REARLEFT
 567                 | AOUT_CHAN_REARRIGHT | AOUT_CHAN_MIDDLELEFT | AOUT_CHAN_MIDDLERIGHT
 568                 | AOUT_CHAN_LFE
 569             };
 570             p_fmt->audio.i_physical_channels =
 571                 p_fmt->audio.i_original_channels =
 572                     pi_channels_maps[p_fmt->audio.i_channels];
 573         }
 574
 575         date_Init( &p_dec->p_sys->end_date, p_fmt->audio.i_rate, 1 );
 576
 577         break;
 578
 579     default: return OMX_ErrorNotImplemented;
 580     }
 581
 582  error:
 583     return omx_error;
 584 }
 585
 586 /*****************************************************************************
 587  * DeinitialiseComponent: Deinitialise and unload an OMX component
 588  *****************************************************************************/
 589 static OMX_ERRORTYPE DeinitialiseComponent(decoder_t *p_dec,
 590                                            OMX_HANDLETYPE omx_handle)
 591 {
 592     decoder_sys_t *p_sys = p_dec->p_sys;
 593     OMX_ERRORTYPE omx_error;
 594     OMX_STATETYPE state;
 595     unsigned int i, j;
 596
 597     if(!omx_handle) return OMX_ErrorNone;
 598
 599     omx_error = OMX_GetState(omx_handle, &state);
 600     CHECK_ERROR(omx_error, "OMX_GetState failed (%x)", omx_error );
 601
 602     if(state == OMX_StateExecuting)
 603     {
 604         omx_error = OMX_SendCommand( omx_handle, OMX_CommandStateSet,
 605                                      OMX_StateIdle, 0 );
 606         CHECK_ERROR(omx_error, "OMX_CommandStateSet Idle failed (%x)", omx_error );
 607         while (1) {
 608             OMX_U32 cmd, state;
 609             omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, &cmd, &state, 0);
 610             CHECK_ERROR(omx_error, "Wait for Idle failed (%x)", omx_error );
 611             // The event queue can contain other OMX_EventCmdComplete items,
 612             // such as for OMX_CommandFlush
 613             if (cmd == OMX_CommandStateSet && state == OMX_StateIdle)
 614                 break;
 615         }
 616     }
 617
 618     omx_error = OMX_GetState(omx_handle, &state);
 619     CHECK_ERROR(omx_error, "OMX_GetState failed (%x)", omx_error );
 620
 621     if(state == OMX_StateIdle)
 622     {
 623         omx_error = OMX_SendCommand( omx_handle, OMX_CommandStateSet,
 624                                      OMX_StateLoaded, 0 );
 625         CHECK_ERROR(omx_error, "OMX_CommandStateSet Loaded failed (%x)", omx_error );
 626
 627         for(i = 0; i < p_sys->ports; i++)
 628         {
 629             OmxPort *p_port = &p_sys->p_ports[i];
 630             OMX_BUFFERHEADERTYPE *p_buffer;
 631
 632             for(j = 0; j < p_port->i_buffers; j++)
 633             {
 634                 OMX_FIFO_GET(&p_port->fifo, p_buffer);
 635                 if (p_buffer->nFlags & SENTINEL_FLAG) {
 636                     free(p_buffer);
 637                     j--;
 638                     continue;
 639                 }
 640                 omx_error = OMX_FreeBuffer( omx_handle,
 641                                             p_port->i_port_index, p_buffer );
 642
 643                 if(omx_error != OMX_ErrorNone) break;
 644             }
 645             CHECK_ERROR(omx_error, "OMX_FreeBuffer failed (%x, %i, %i)",
 646                         omx_error, (int)p_port->i_port_index, j );
 647             while (1) {
 648                 OMX_FIFO_PEEK(&p_port->fifo, p_buffer);
 649                 if (!p_buffer) break;
 650
 651                 OMX_FIFO_GET(&p_port->fifo, p_buffer);
 652                 if (p_buffer->nFlags & SENTINEL_FLAG) {
 653                     free(p_buffer);
 654                     continue;
 655                 }
 656                 msg_Warn( p_dec, "Stray buffer left in fifo, %p", p_buffer );
 657             }
 658         }
 659
 660         omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
 661         CHECK_ERROR(omx_error, "Wait for Loaded failed (%x)", omx_error );
 662     }
 663
 664  error:
 665     for(i = 0; i < p_sys->ports; i++)
 666     {
 667         OmxPort *p_port = &p_sys->p_ports[i];
 668         free(p_port->pp_buffers);
 669         p_port->pp_buffers = 0;
 670     }
 671     omx_error = pf_free_handle( omx_handle );
 672     return omx_error;
 673 }
 674
 675 /*****************************************************************************
 676  * InitialiseComponent: Load and initialise an OMX component
 677  *****************************************************************************/
 678 static OMX_ERRORTYPE InitialiseComponent(decoder_t *p_dec,
 679     OMX_STRING psz_component, OMX_HANDLETYPE *p_handle)
 680 {
 681     static OMX_CALLBACKTYPE callbacks =
 682         { OmxEventHandler, OmxEmptyBufferDone, OmxFillBufferDone };
 683     decoder_sys_t *p_sys = p_dec->p_sys;
 684     OMX_HANDLETYPE omx_handle;
 685     OMX_ERRORTYPE omx_error;
 686     unsigned int i;
 687     OMX_U8 psz_role[OMX_MAX_STRINGNAME_SIZE];
 688     OMX_PARAM_COMPONENTROLETYPE role;
 689     OMX_PARAM_PORTDEFINITIONTYPE definition;
 690     OMX_PORT_PARAM_TYPE param;
 691
 692     /* Load component */
 693     omx_error = pf_get_handle( &omx_handle, psz_component, p_dec, &callbacks );
 694     if(omx_error != OMX_ErrorNone)
 695     {
 696         msg_Warn( p_dec, "OMX_GetHandle(%s) failed (%x: %s)", psz_component,
 697                   omx_error, ErrorToString(omx_error) );
 698         return omx_error;
 699     }
 700     strncpy(p_sys->psz_component, psz_component, OMX_MAX_STRINGNAME_SIZE-1);
 701
 702     omx_error = OMX_ComponentRoleEnum(omx_handle, psz_role, 0);
 703     if(omx_error == OMX_ErrorNone)
 704         msg_Dbg(p_dec, "loaded component %s of role %s", psz_component, psz_role);
 705     else
 706         msg_Dbg(p_dec, "loaded component %s", psz_component);
 707     PrintOmx(p_dec, omx_handle, OMX_ALL);
 708
 709     /* Set component role */
 710     OMX_INIT_STRUCTURE(role);
 711     strcpy((char*)role.cRole,
 712            GetOmxRole(p_sys->b_enc ? p_dec->fmt_out.i_codec : p_dec->fmt_in.i_codec,
 713                       p_dec->fmt_in.i_cat, p_sys->b_enc));
 714
 715     omx_error = OMX_SetParameter(omx_handle, OMX_IndexParamStandardComponentRole,
 716                                  &role);
 717     omx_error = OMX_GetParameter(omx_handle, OMX_IndexParamStandardComponentRole,
 718                                  &role);
 719     if(omx_error == OMX_ErrorNone)
 720         msg_Dbg(p_dec, "component standard role set to %s", role.cRole);
 721
 722     /* Find the input / output ports */
 723     OMX_INIT_STRUCTURE(param);
 724     OMX_INIT_STRUCTURE(definition);
 725     omx_error = OMX_GetParameter(omx_handle, p_dec->fmt_in.i_cat == VIDEO_ES ?
 726                                  OMX_IndexParamVideoInit : OMX_IndexParamAudioInit, &param);
 727     if(omx_error != OMX_ErrorNone) {
 728 #ifdef __ANDROID__
 729         param.nPorts = 2;
 730         param.nStartPortNumber = 0;
 731 #else
 732         param.nPorts = 0;
 733 #endif
 734     }
 735
 736     for(i = 0; i < param.nPorts; i++)
 737     {
 738         OmxPort *p_port;
 739
 740         /* Get port definition */
 741         definition.nPortIndex = param.nStartPortNumber + i;
 742         omx_error = OMX_GetParameter(omx_handle, OMX_IndexParamPortDefinition,
 743                                      &definition);
 744         if(omx_error != OMX_ErrorNone) continue;
 745
 746         if(definition.eDir == OMX_DirInput) p_port = &p_sys->in;
 747         else  p_port = &p_sys->out;
 748
 749         p_port->b_valid = true;
 750         p_port->i_port_index = definition.nPortIndex;
 751         p_port->definition = definition;
 752         p_port->omx_handle = omx_handle;
 753     }
 754
 755     if(!p_sys->in.b_valid || !p_sys->out.b_valid)
 756     {
 757         omx_error = OMX_ErrorInvalidComponent;
 758         CHECK_ERROR(omx_error, "couldn't find an input and output port");
 759     }
 760
 761     if(!strncmp(p_sys->psz_component, "OMX.SEC.", 8))
 762     {
 763         OMX_INDEXTYPE index;
 764         omx_error = OMX_GetExtensionIndex(omx_handle, (OMX_STRING) "OMX.SEC.index.ThumbnailMode", &index);
 765         if(omx_error == OMX_ErrorNone)
 766         {
 767             OMX_BOOL enable = OMX_TRUE;
 768             omx_error = OMX_SetConfig(omx_handle, index, &enable);
 769             CHECK_ERROR(omx_error, "Unable to set ThumbnailMode");
 770         } else {
 771             OMX_BOOL enable = OMX_TRUE;
 772             /* Needed on Samsung Galaxy S II */
 773             omx_error = OMX_SetConfig(omx_handle, OMX_IndexVendorSetYUV420pMode, &enable);
 774             if (omx_error == OMX_ErrorNone)
 775                 msg_Dbg(p_dec, "Set OMX_IndexVendorSetYUV420pMode successfully");
 776             else
 777                 msg_Dbg(p_dec, "Unable to set OMX_IndexVendorSetYUV420pMode: %x", omx_error);
 778         }
 779     }
 780
 781     /* Set port definitions */
 782     for(i = 0; i < p_sys->ports; i++)
 783     {
 784         omx_error = SetPortDefinition(p_dec, &p_sys->p_ports[i],
 785                                       p_sys->p_ports[i].p_fmt);
 786         if(omx_error != OMX_ErrorNone) goto error;
 787     }
 788
 789     /* Allocate our array for the omx buffers and enable ports */
 790     for(i = 0; i < p_sys->ports; i++)
 791     {
 792         OmxPort *p_port = &p_sys->p_ports[i];
 793
 794         p_port->pp_buffers =
 795             malloc(p_port->definition.nBufferCountActual *
 796                    sizeof(OMX_BUFFERHEADERTYPE*));
 797         if(!p_port->pp_buffers)
 798         {
 799           omx_error = OMX_ErrorInsufficientResources;
 800           CHECK_ERROR(omx_error, "memory allocation failed");
 801         }
 802         p_port->i_buffers = p_port->definition.nBufferCountActual;
 803
 804         /* Enable port */
 805         if(!p_port->definition.bEnabled)
 806         {
 807             omx_error = OMX_SendCommand( omx_handle, OMX_CommandPortEnable,
 808                                          p_port->i_port_index, NULL);
 809             CHECK_ERROR(omx_error, "OMX_CommandPortEnable on %i failed (%x)",
 810                         (int)p_port->i_port_index, omx_error );
 811             omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
 812             CHECK_ERROR(omx_error, "Wait for PortEnable on %i failed (%x)",
 813                         (int)p_port->i_port_index, omx_error );
 814         }
 815     }
 816
 817     *p_handle = omx_handle;
 818     return OMX_ErrorNone;
 819
 820  error:
 821     DeinitialiseComponent(p_dec, omx_handle);
 822     *p_handle = 0;
 823     return omx_error;
 824 }
 825
 826 /*****************************************************************************
 827  * OpenDecoder: Create the decoder instance
 828  *****************************************************************************/
 829 static int OpenDecoder( vlc_object_t *p_this )
 830 {
 831     decoder_t *p_dec = (decoder_t*)p_this;
 832     int status;
 833
 834     if( 0 || !GetOmxRole(p_dec->fmt_in.i_codec, p_dec->fmt_in.i_cat, false) )
 835         return VLC_EGENERIC;
 836
 837 #ifdef HAVE_MAEMO
 838     if( p_dec->fmt_in.i_cat != VIDEO_ES && !p_dec->b_force)
 839         return VLC_EGENERIC;
 840 #endif
 841
 842     status = OpenGeneric( p_this, false );
 843     if(status != VLC_SUCCESS) return status;
 844
 845     p_dec->pf_decode_video = DecodeVideo;
 846     p_dec->pf_decode_audio = DecodeAudio;
 847
 848     return VLC_SUCCESS;
 849 }
 850
 851 /*****************************************************************************
 852  * OpenEncoder: Create the encoder instance
 853  *****************************************************************************/
 854 static int OpenEncoder( vlc_object_t *p_this )
 855 {
 856     encoder_t *p_enc = (encoder_t*)p_this;
 857     int status;
 858
 859     if( !GetOmxRole(p_enc->fmt_out.i_codec, p_enc->fmt_in.i_cat, true) )
 860         return VLC_EGENERIC;
 861
 862     status = OpenGeneric( p_this, true );
 863     if(status != VLC_SUCCESS) return status;
 864
 865     p_enc->pf_encode_video = EncodeVideo;
 866
 867     return VLC_SUCCESS;
 868 }
 869
 870 /*****************************************************************************
 871  * OpenGeneric: Create the generic decoder/encoder instance
 872  *****************************************************************************/
 873 static int OpenGeneric( vlc_object_t *p_this, bool b_encode )
 874 {
 875     decoder_t *p_dec = (decoder_t*)p_this;
 876     decoder_sys_t *p_sys;
 877     OMX_ERRORTYPE omx_error;
 878     OMX_BUFFERHEADERTYPE *p_header;
 879     unsigned int i, j;
 880
 881     vlc_mutex_lock( &omx_core_mutex );
 882     if( omx_refcount > 0 )
 883         goto loaded;
 884
 885     /* Load the OMX core */
 886     for( i = 0; ppsz_dll_list[i]; i++ )
 887     {
 888         dll_handle = dll_open( ppsz_dll_list[i] );
 889         if( dll_handle ) break;
 890     }
 891     if( !dll_handle )
 892     {
 893         vlc_mutex_unlock( &omx_core_mutex );
 894         return VLC_EGENERIC;
 895     }
 896
 897     pf_init = dlsym( dll_handle, "OMX_Init" );
 898     pf_deinit = dlsym( dll_handle, "OMX_Deinit" );
 899     pf_get_handle = dlsym( dll_handle, "OMX_GetHandle" );
 900     pf_free_handle = dlsym( dll_handle, "OMX_FreeHandle" );
 901     pf_component_enum = dlsym( dll_handle, "OMX_ComponentNameEnum" );
 902     pf_get_roles_of_component = dlsym( dll_handle, "OMX_GetRolesOfComponent" );
 903     if( !pf_init || !pf_deinit || !pf_get_handle || !pf_free_handle ||
 904         !pf_component_enum || !pf_get_roles_of_component )
 905     {
 906         msg_Warn( p_this, "cannot find OMX_* symbols in `%s' (%s)",
 907                   ppsz_dll_list[i], dlerror() );
 908         dll_close(dll_handle);
 909         vlc_mutex_unlock( &omx_core_mutex );
 910         return VLC_EGENERIC;
 911     }
 912
 913 loaded:
 914     /* Allocate the memory needed to store the decoder's structure */
 915     if( ( p_dec->p_sys = p_sys = calloc( 1, sizeof(*p_sys)) ) == NULL )
 916     {
 917         if( omx_refcount == 0 )
 918             dll_close(dll_handle);
 919         vlc_mutex_unlock( &omx_core_mutex );
 920         return VLC_ENOMEM;
 921     }
 922
 923     /* Initialise the thread properties */
 924     if(!b_encode)
 925     {
 926         p_dec->fmt_out.i_cat = p_dec->fmt_in.i_cat;
 927         p_dec->fmt_out.video = p_dec->fmt_in.video;
 928         p_dec->fmt_out.audio = p_dec->fmt_in.audio;
 929         p_dec->fmt_out.i_codec = 0;
 930     }
 931     p_sys->b_enc = b_encode;
 932     p_sys->pp_last_event = &p_sys->p_events;
 933     vlc_mutex_init (&p_sys->mutex);
 934     vlc_cond_init (&p_sys->cond);
 935     vlc_mutex_init (&p_sys->lock);
 936     vlc_mutex_init (&p_sys->in.fifo.lock);
 937     vlc_cond_init (&p_sys->in.fifo.wait);
 938     p_sys->in.fifo.offset = offsetof(OMX_BUFFERHEADERTYPE, pOutputPortPrivate) / sizeof(void *);
 939     p_sys->in.fifo.pp_last = &p_sys->in.fifo.p_first;
 940     p_sys->in.b_direct = false;
 941     p_sys->in.b_flushed = true;
 942     p_sys->in.p_fmt = &p_dec->fmt_in;
 943     vlc_mutex_init (&p_sys->out.fifo.lock);
 944     vlc_cond_init (&p_sys->out.fifo.wait);
 945     p_sys->out.fifo.offset = offsetof(OMX_BUFFERHEADERTYPE, pInputPortPrivate) / sizeof(void *);
 946     p_sys->out.fifo.pp_last = &p_sys->out.fifo.p_first;
 947     p_sys->out.b_direct = true;
 948     p_sys->out.b_flushed = true;
 949     p_sys->out.p_fmt = &p_dec->fmt_out;
 950     p_sys->ports = 2;
 951     p_sys->p_ports = &p_sys->in;
 952     p_sys->b_use_pts = 0;
 953
 954     msg_Dbg(p_dec, "fmt in:%4.4s, out: %4.4s", (char *)&p_dec->fmt_in.i_codec,
 955             (char *)&p_dec->fmt_out.i_codec);
 956
 957     /* Initialise the OMX core */
 958     omx_error = omx_refcount > 0 ? OMX_ErrorNone : pf_init();
 959     omx_refcount++;
 960     if(omx_error != OMX_ErrorNone)
 961     {
 962         msg_Warn( p_this, "OMX_Init failed (%x: %s)", omx_error,
 963                   ErrorToString(omx_error) );
 964         vlc_mutex_unlock( &omx_core_mutex );
 965         CloseGeneric(p_this);
 966         return VLC_EGENERIC;
 967     }
 968     p_sys->b_init = true;
 969     vlc_mutex_unlock( &omx_core_mutex );
 970
 971     /* Enumerate components and build a list of the one we want to try */
 972     if( !CreateComponentsList(p_dec,
 973              GetOmxRole(p_sys->b_enc ? p_dec->fmt_out.i_codec :
 974                         p_dec->fmt_in.i_codec, p_dec->fmt_in.i_cat,
 975                         p_sys->b_enc)) )
 976     {
 977         msg_Warn( p_this, "couldn't find an omx component for codec %4.4s",
 978                   (char *)&p_dec->fmt_in.i_codec );
 979         CloseGeneric(p_this);
 980         return VLC_EGENERIC;
 981     }
 982
 983     /* Try to load and initialise a component */
 984     omx_error = OMX_ErrorUndefined;
 985     for(i = 0; i < p_sys->components; i++)
 986     {
 987 #ifdef __ANDROID__
 988         /* ignore OpenCore software codecs */
 989         if (!strncmp(p_sys->ppsz_components[i], "OMX.PV.", 7))
 990             continue;
 991         /* The same sw codecs, renamed in ICS (perhaps also in honeycomb) */
 992         if (!strncmp(p_sys->ppsz_components[i], "OMX.google.", 11))
 993             continue;
 994         /* This one has been seen on HTC One V - it behaves like it works,
 995          * but FillBufferDone returns buffers filled with 0 bytes. The One V
 996          * has got a working OMX.qcom.video.decoder.avc instead though. */
 997         if (!strncmp(p_sys->ppsz_components[i], "OMX.ARICENT.", 12))
 998             continue;
 999         /* Some nVidia codec with DRM */
1000         if (!strncmp(p_sys->ppsz_components[i], "OMX.Nvidia.h264.decode.secure", 29))
1001             continue;
1002         /* Use VC1 decoder for WMV3 for now */
1003         if (!strcmp(p_sys->ppsz_components[i], "OMX.SEC.WMV.Decoder"))
1004             continue;
1005         /* This decoder does work, but has an insane latency (leading to errors
1006          * about "main audio output playback way too late" and dropped frames).
1007          * At least Samsung Galaxy S III (where this decoder is present) has
1008          * got another one, OMX.SEC.mp3.dec, that works well and has a
1009          * sensible latency. (Also, even if that one isn't found, in general,
1010          * using SW codecs is usually more than fast enough for MP3.) */
1011         if (!strcmp(p_sys->ppsz_components[i], "OMX.SEC.MP3.Decoder"))
1012             continue;
1013 #endif
1014         omx_error = InitialiseComponent(p_dec, p_sys->ppsz_components[i],
1015                                         &p_sys->omx_handle);
1016         if(omx_error == OMX_ErrorNone) break;
1017     }
1018     CHECK_ERROR(omx_error, "no component could be initialised" );
1019
1020     /* Move component to Idle then Executing state */
1021     OMX_SendCommand( p_sys->omx_handle, OMX_CommandStateSet, OMX_StateIdle, 0 );
1022     CHECK_ERROR(omx_error, "OMX_CommandStateSet Idle failed (%x)", omx_error );
1023
1024     /* Allocate omx buffers */
1025     for(i = 0; i < p_sys->ports; i++)
1026     {
1027         OmxPort *p_port = &p_sys->p_ports[i];
1028
1029         for(j = 0; j < p_port->i_buffers; j++)
1030         {
1031 #if 0
1032 #define ALIGN(x,BLOCKLIGN) (((x) + BLOCKLIGN - 1) & ~(BLOCKLIGN - 1))
1033             char *p_buf = malloc(p_port->definition.nBufferSize +
1034                                  p_port->definition.nBufferAlignment);
1035             p_port->pp_buffers[i] = (void *)ALIGN((uintptr_t)p_buf, p_port->definition.nBufferAlignment);
1036 #endif
1037
1038             if(0 && p_port->b_direct)
1039                 omx_error =
1040                     OMX_UseBuffer( p_sys->omx_handle, &p_port->pp_buffers[j],
1041                                    p_port->i_port_index, 0,
1042                                    p_port->definition.nBufferSize, (void*)1);
1043             else
1044                 omx_error =
1045                     OMX_AllocateBuffer( p_sys->omx_handle, &p_port->pp_buffers[j],
1046                                         p_port->i_port_index, 0,
1047                                         p_port->definition.nBufferSize);
1048
1049             if(omx_error != OMX_ErrorNone) break;
1050             OMX_FIFO_PUT(&p_port->fifo, p_port->pp_buffers[j]);
1051         }
1052         p_port->i_buffers = j;
1053         CHECK_ERROR(omx_error, "OMX_UseBuffer failed (%x, %i, %i)",
1054                     omx_error, (int)p_port->i_port_index, j );
1055     }
1056
1057     omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
1058     CHECK_ERROR(omx_error, "Wait for Idle failed (%x)", omx_error );
1059
1060     omx_error = OMX_SendCommand( p_sys->omx_handle, OMX_CommandStateSet,
1061                                  OMX_StateExecuting, 0);
1062     CHECK_ERROR(omx_error, "OMX_CommandStateSet Executing failed (%x)", omx_error );
1063     omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
1064     CHECK_ERROR(omx_error, "Wait for Executing failed (%x)", omx_error );
1065
1066     /* Send codec configuration data */
1067     if( p_dec->fmt_in.i_extra )
1068     {
1069         OMX_FIFO_GET(&p_sys->in.fifo, p_header);
1070         p_header->nFilledLen = p_dec->fmt_in.i_extra;
1071
1072         /* Convert H.264 NAL format to annex b */
1073         if( p_sys->i_nal_size_length && !p_sys->in.b_direct )
1074         {
1075             p_header->nFilledLen = 0;
1076             convert_sps_pps( p_dec, p_dec->fmt_in.p_extra, p_dec->fmt_in.i_extra,
1077                              p_header->pBuffer, p_header->nAllocLen,
1078                              (uint32_t*) &p_header->nFilledLen, NULL );
1079         }
1080         else if(p_sys->in.b_direct)
1081         {
1082             p_header->pOutputPortPrivate = p_header->pBuffer;
1083             p_header->pBuffer = p_dec->fmt_in.p_extra;
1084         }
1085         else if (p_dec->fmt_in.i_codec == VLC_CODEC_WMV3 &&
1086                  p_dec->fmt_in.i_extra >= 4 &&
1087                  p_header->nAllocLen >= 36)
1088         {
1089             int profile;
1090             // According to OMX IL 1.2.0 spec (4.3.33.2), the codec config
1091             // data for VC-1 Main/Simple (aka WMV3) is according to table 265
1092             // in the VC-1 spec. Most of the fields are just set with placeholders
1093             // (like framerate, hrd_buffer/rate).
1094             static const uint8_t wmv3seq[] = {
1095                 0xff, 0xff, 0xff, 0xc5, // numframes=ffffff, marker byte
1096                 0x04, 0x00, 0x00, 0x00, // marker byte
1097                 0x00, 0x00, 0x00, 0x00, // struct C, almost equal to p_extra
1098                 0x00, 0x00, 0x00, 0x00, // struct A, vert size
1099                 0x00, 0x00, 0x00, 0x00, // struct A, horiz size
1100                 0x0c, 0x00, 0x00, 0x00, // marker byte
1101                 0xff, 0xff, 0x00, 0x80, // struct B, level=4, cbr=0, hrd_buffer=ffff
1102                 0xff, 0xff, 0x00, 0x00, // struct B, hrd_rate=ffff
1103                 0xff, 0xff, 0xff, 0xff, // struct B, framerate=ffffffff
1104             };
1105             p_header->nFilledLen = sizeof(wmv3seq);
1106             memcpy(p_header->pBuffer, wmv3seq, p_header->nFilledLen);
1107             // Struct C - almost equal to the extradata
1108             memcpy(&p_header->pBuffer[8], p_dec->fmt_in.p_extra, 4);
1109             // Expand profile from the highest 2 bits to the highest 4 bits
1110             profile = p_header->pBuffer[8] >> 6;
1111             p_header->pBuffer[8] = (p_header->pBuffer[8] & 0x0f) | (profile << 4);
1112             // Fill in the height/width for struct A
1113             SetDWLE(&p_header->pBuffer[12], p_dec->fmt_in.video.i_height);
1114             SetDWLE(&p_header->pBuffer[16], p_dec->fmt_in.video.i_width);
1115         }
1116         else
1117         {
1118             if(p_header->nFilledLen > p_header->nAllocLen)
1119             {
1120                 msg_Dbg(p_dec, "buffer too small (%i,%i)", (int)p_header->nFilledLen,
1121                         (int)p_header->nAllocLen);
1122                 p_header->nFilledLen = p_header->nAllocLen;
1123             }
1124             memcpy(p_header->pBuffer, p_dec->fmt_in.p_extra, p_header->nFilledLen);
1125         }
1126
1127         p_header->nOffset = 0;
1128         p_header->nFlags = OMX_BUFFERFLAG_CODECCONFIG | OMX_BUFFERFLAG_ENDOFFRAME;
1129         msg_Dbg(p_dec, "sending codec config data %p, %p, %i", p_header,
1130                 p_header->pBuffer, (int)p_header->nFilledLen);
1131         OMX_EmptyThisBuffer(p_sys->omx_handle, p_header);
1132     }
1133
1134     /* Get back output port definition */
1135     omx_error = GetPortDefinition(p_dec, &p_sys->out, p_sys->out.p_fmt);
1136     if(omx_error != OMX_ErrorNone) goto error;
1137
1138     PrintOmx(p_dec, p_sys->omx_handle, p_dec->p_sys->in.i_port_index);
1139     PrintOmx(p_dec, p_sys->omx_handle, p_dec->p_sys->out.i_port_index);
1140
1141     if(p_sys->b_error) goto error;
1142
1143     p_dec->b_need_packetized = true;
1144     if (!strcmp(p_sys->psz_component, "OMX.TI.DUCATI1.VIDEO.DECODER"))
1145         p_sys->b_use_pts = 1;
1146
1147     if (!strcmp(p_sys->psz_component, "OMX.STM.Video.Decoder"))
1148         p_sys->b_use_pts = 1;
1149
1150     if (p_sys->b_use_pts)
1151         msg_Dbg( p_dec, "using pts timestamp mode for %s", p_sys->psz_component);
1152
1153     return VLC_SUCCESS;
1154
1155  error:
1156     CloseGeneric(p_this);
1157     return VLC_EGENERIC;
1158 }
1159
1160 /*****************************************************************************
1161  * PortReconfigure
1162  *****************************************************************************/
1163 static OMX_ERRORTYPE PortReconfigure(decoder_t *p_dec, OmxPort *p_port)
1164 {
1165     decoder_sys_t *p_sys = p_dec->p_sys;
1166     OMX_PARAM_PORTDEFINITIONTYPE definition;
1167     OMX_BUFFERHEADERTYPE *p_buffer;
1168     OMX_ERRORTYPE omx_error;
1169     unsigned int i;
1170
1171     /* Sanity checking */
1172     OMX_INIT_STRUCTURE(definition);
1173     definition.nPortIndex = p_port->i_port_index;
1174     omx_error = OMX_GetParameter(p_dec->p_sys->omx_handle, OMX_IndexParamPortDefinition,
1175                                  &definition);
1176     if(omx_error != OMX_ErrorNone || (p_dec->fmt_in.i_cat == VIDEO_ES &&
1177        (!definition.format.video.nFrameWidth ||
1178        !definition.format.video.nFrameHeight)) )
1179         return OMX_ErrorUndefined;
1180
1181     omx_error = OMX_SendCommand( p_sys->omx_handle, OMX_CommandPortDisable,
1182                                  p_port->i_port_index, NULL);
1183     CHECK_ERROR(omx_error, "OMX_CommandPortDisable on %i failed (%x)",
1184                 (int)p_port->i_port_index, omx_error );
1185
1186     for(i = 0; i < p_port->i_buffers; i++)
1187     {
1188         OMX_FIFO_GET(&p_port->fifo, p_buffer);
1189         if (p_buffer->nFlags & SENTINEL_FLAG) {
1190             free(p_buffer);
1191             i--;
1192             continue;
1193         }
1194         omx_error = OMX_FreeBuffer( p_sys->omx_handle,
1195                                     p_port->i_port_index, p_buffer );
1196
1197         if(omx_error != OMX_ErrorNone) break;
1198     }
1199     CHECK_ERROR(omx_error, "OMX_FreeBuffer failed (%x, %i, %i)",
1200                 omx_error, (int)p_port->i_port_index, i );
1201
1202     omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
1203     CHECK_ERROR(omx_error, "Wait for PortDisable failed (%x)", omx_error );
1204
1205     /* Get the new port definition */
1206     omx_error = GetPortDefinition(p_dec, &p_sys->out, p_sys->out.p_fmt);
1207     if(omx_error != OMX_ErrorNone) goto error;
1208
1209     if( p_dec->fmt_in.i_cat != AUDIO_ES )
1210     {
1211         /* Don't explicitly set the new parameters that we got with
1212          * OMX_GetParameter above when using audio codecs.
1213          * That struct hasn't been changed since, so there should be
1214          * no need to set it here, unless some codec expects the
1215          * SetParameter call as a trigger event for some part of
1216          * the reconfiguration.
1217          * This fixes using audio decoders on Samsung Galaxy S II,
1218          *
1219          * Only skipping this for audio codecs, to minimize the
1220          * change for current working configurations for video.
1221          */
1222         omx_error = OMX_SetParameter(p_dec->p_sys->omx_handle, OMX_IndexParamPortDefinition,
1223                                      &definition);
1224         CHECK_ERROR(omx_error, "OMX_SetParameter failed (%x : %s)",
1225                     omx_error, ErrorToString(omx_error));
1226     }
1227
1228     omx_error = OMX_SendCommand( p_sys->omx_handle, OMX_CommandPortEnable,
1229                                  p_port->i_port_index, NULL);
1230     CHECK_ERROR(omx_error, "OMX_CommandPortEnable on %i failed (%x)",
1231                 (int)p_port->i_port_index, omx_error );
1232
1233     if (p_port->definition.nBufferCountActual > p_port->i_buffers) {
1234         free(p_port->pp_buffers);
1235         p_port->pp_buffers = malloc(p_port->definition.nBufferCountActual * sizeof(OMX_BUFFERHEADERTYPE*));
1236         if(!p_port->pp_buffers)
1237         {
1238             omx_error = OMX_ErrorInsufficientResources;
1239             CHECK_ERROR(omx_error, "memory allocation failed");
1240         }
1241     }
1242     p_port->i_buffers = p_port->definition.nBufferCountActual;
1243     for(i = 0; i < p_port->i_buffers; i++)
1244     {
1245         if(0 && p_port->b_direct)
1246             omx_error =
1247                 OMX_UseBuffer( p_sys->omx_handle, &p_port->pp_buffers[i],
1248                                p_port->i_port_index, 0,
1249                                p_port->definition.nBufferSize, (void*)1);
1250         else
1251             omx_error =
1252                 OMX_AllocateBuffer( p_sys->omx_handle, &p_port->pp_buffers[i],
1253                                     p_port->i_port_index, 0,
1254                                     p_port->definition.nBufferSize);
1255
1256         if(omx_error != OMX_ErrorNone) break;
1257         OMX_FIFO_PUT(&p_port->fifo, p_port->pp_buffers[i]);
1258     }
1259     p_port->i_buffers = i;
1260     CHECK_ERROR(omx_error, "OMX_UseBuffer failed (%x, %i, %i)",
1261                 omx_error, (int)p_port->i_port_index, i );
1262
1263     omx_error = WaitForSpecificOmxEvent(p_dec, OMX_EventCmdComplete, 0, 0, 0);
1264     CHECK_ERROR(omx_error, "Wait for PortEnable failed (%x)", omx_error );
1265
1266     PrintOmx(p_dec, p_sys->omx_handle, p_dec->p_sys->in.i_port_index);
1267     PrintOmx(p_dec, p_sys->omx_handle, p_dec->p_sys->out.i_port_index);
1268
1269  error:
1270     return omx_error;
1271 }
1272
1273 /*****************************************************************************
1274  * DecodeVideo: Called to decode one frame
1275  *****************************************************************************/
1276 static picture_t *DecodeVideo( decoder_t *p_dec, block_t **pp_block )
1277 {
1278     decoder_sys_t *p_sys = p_dec->p_sys;
1279     picture_t *p_pic = NULL, *p_next_pic;
1280     OMX_ERRORTYPE omx_error;
1281     unsigned int i;
1282
1283     OMX_BUFFERHEADERTYPE *p_header;
1284     block_t *p_block;
1285
1286     if( !pp_block || !*pp_block )
1287         return NULL;
1288
1289     p_block = *pp_block;
1290
1291     /* Check for errors from codec */
1292     if(p_sys->b_error)
1293     {
1294         msg_Dbg(p_dec, "error during decoding");
1295         block_Release( p_block );
1296         return 0;
1297     }
1298
1299     if( p_block->i_flags & (BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
1300     {
1301         block_Release( p_block );
1302         if(!p_sys->in.b_flushed)
1303         {
1304             msg_Dbg(p_dec, "flushing");
1305             OMX_SendCommand( p_sys->omx_handle, OMX_CommandFlush,
1306                              p_sys->in.definition.nPortIndex, 0 );
1307         }
1308         p_sys->in.b_flushed = true;
1309         return NULL;
1310     }
1311
1312     /* Take care of decoded frames first */
1313     while(!p_pic)
1314     {
1315         OMX_FIFO_PEEK(&p_sys->out.fifo, p_header);
1316         if(!p_header) break; /* No frame available */
1317
1318         if(p_sys->out.b_update_def)
1319         {
1320             omx_error = GetPortDefinition(p_dec, &p_sys->out, p_sys->out.p_fmt);
1321             p_sys->out.b_update_def = 0;
1322         }
1323
1324         if(p_header->nFilledLen)
1325         {
1326             p_pic = p_header->pAppPrivate;
1327             if(!p_pic)
1328             {
1329                 /* We're not in direct rendering mode.
1330                  * Get a new picture and copy the content */
1331                 p_pic = decoder_NewPicture( p_dec );
1332
1333                 if (p_pic)
1334                     CopyOmxPicture(p_sys->out.definition.format.video.eColorFormat,
1335                                    p_pic, p_sys->out.definition.format.video.nSliceHeight,
1336                                    p_sys->out.i_frame_stride,
1337                                    p_header->pBuffer + p_header->nOffset,
1338                                    p_sys->out.i_frame_stride_chroma_div);
1339             }
1340
1341             if (p_pic)
1342                 p_pic->date = p_header->nTimeStamp;
1343             p_header->nFilledLen = 0;
1344             p_header->pAppPrivate = 0;
1345         }
1346
1347         /* Get a new picture */
1348         if(p_sys->in.b_direct && !p_header->pAppPrivate)
1349         {
1350             p_next_pic = decoder_NewPicture( p_dec );
1351             if(!p_next_pic) break;
1352
1353             OMX_FIFO_GET(&p_sys->out.fifo, p_header);
1354             p_header->pAppPrivate = p_next_pic;
1355             p_header->pInputPortPrivate = p_header->pBuffer;
1356             p_header->pBuffer = p_next_pic->p[0].p_pixels;
1357         }
1358         else
1359         {
1360             OMX_FIFO_GET(&p_sys->out.fifo, p_header);
1361         }
1362
1363 #ifdef OMXIL_EXTRA_DEBUG
1364         msg_Dbg( p_dec, "FillThisBuffer %p, %p", p_header, p_header->pBuffer );
1365 #endif
1366         OMX_FillThisBuffer(p_sys->omx_handle, p_header);
1367     }
1368
1369     /* Send the input buffer to the component */
1370     OMX_FIFO_GET(&p_sys->in.fifo, p_header);
1371
1372     if (p_header && p_header->nFlags & SENTINEL_FLAG) {
1373         free(p_header);
1374         goto reconfig;
1375     }
1376
1377     if(p_header)
1378     {
1379         p_header->nFilledLen = p_block->i_buffer;
1380         p_header->nOffset = 0;
1381         p_header->nFlags = OMX_BUFFERFLAG_ENDOFFRAME;
1382         if (p_sys->b_use_pts && p_block->i_pts)
1383             p_header->nTimeStamp = p_block->i_pts;
1384         else
1385             p_header->nTimeStamp = p_block->i_dts;
1386
1387         /* In direct mode we pass the input pointer as is.
1388          * Otherwise we memcopy the data */
1389         if(p_sys->in.b_direct)
1390         {
1391             p_header->pOutputPortPrivate = p_header->pBuffer;
1392             p_header->pBuffer = p_block->p_buffer;
1393             p_header->pAppPrivate = p_block;
1394         }
1395         else
1396         {
1397             if(p_header->nFilledLen > p_header->nAllocLen)
1398             {
1399                 msg_Dbg(p_dec, "buffer too small (%i,%i)",
1400                         (int)p_header->nFilledLen, (int)p_header->nAllocLen);
1401                 p_header->nFilledLen = p_header->nAllocLen;
1402             }
1403             memcpy(p_header->pBuffer, p_block->p_buffer, p_header->nFilledLen );
1404             block_Release(p_block);
1405         }
1406
1407         /* Convert H.264 NAL format to annex b. Doesn't do anything if
1408          * i_nal_size_length == 0, which is the case for codecs other
1409          * than H.264 */
1410         convert_h264_to_annexb( p_header->pBuffer, p_header->nFilledLen,
1411                                 p_sys->i_nal_size_length );
1412 #ifdef OMXIL_EXTRA_DEBUG
1413         msg_Dbg( p_dec, "EmptyThisBuffer %p, %p, %i", p_header, p_header->pBuffer,
1414                  (int)p_header->nFilledLen );
1415 #endif
1416         OMX_EmptyThisBuffer(p_sys->omx_handle, p_header);
1417         p_sys->in.b_flushed = false;
1418         *pp_block = NULL; /* Avoid being fed the same packet again */
1419     }
1420
1421 reconfig:
1422     /* Handle the PortSettingsChanged events */
1423     for(i = 0; i < p_sys->ports; i++)
1424     {
1425         OmxPort *p_port = &p_sys->p_ports[i];
1426         if(p_port->b_reconfigure)
1427         {
1428             omx_error = PortReconfigure(p_dec, p_port);
1429             p_port->b_reconfigure = 0;
1430         }
1431         if(p_port->b_update_def)
1432         {
1433             omx_error = GetPortDefinition(p_dec, p_port, p_port->p_fmt);
1434             p_port->b_update_def = 0;
1435         }
1436     }
1437
1438     return p_pic;
1439 }
1440
1441 /*****************************************************************************
1442  * DecodeAudio: Called to decode one frame
1443  *****************************************************************************/
1444 block_t *DecodeAudio ( decoder_t *p_dec, block_t **pp_block )
1445 {
1446     decoder_sys_t *p_sys = p_dec->p_sys;
1447     block_t *p_buffer = NULL;
1448     OMX_BUFFERHEADERTYPE *p_header;
1449     OMX_ERRORTYPE omx_error;
1450     block_t *p_block;
1451     unsigned int i;
1452
1453     if( !pp_block || !*pp_block ) return NULL;
1454
1455     p_block = *pp_block;
1456
1457     /* Check for errors from codec */
1458     if(p_sys->b_error)
1459     {
1460         msg_Dbg(p_dec, "error during decoding");
1461         block_Release( p_block );
1462         return 0;
1463     }
1464
1465     if( p_block->i_flags & (BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
1466     {
1467         block_Release( p_block );
1468         date_Set( &p_sys->end_date, 0 );
1469         if(!p_sys->in.b_flushed)
1470         {
1471             msg_Dbg(p_dec, "flushing");
1472             OMX_SendCommand( p_sys->omx_handle, OMX_CommandFlush,
1473                              p_sys->in.definition.nPortIndex, 0 );
1474         }
1475         p_sys->in.b_flushed = true;
1476         return NULL;
1477     }
1478
1479     if( !date_Get( &p_sys->end_date ) )
1480     {
1481         if( !p_block->i_pts )
1482         {
1483             /* We've just started the stream, wait for the first PTS. */
1484             block_Release( p_block );
1485             return NULL;
1486         }
1487         date_Set( &p_sys->end_date, p_block->i_pts );
1488     }
1489
1490     /* Take care of decoded frames first */
1491     while(!p_buffer)
1492     {
1493         unsigned int i_samples = 0;
1494
1495         OMX_FIFO_PEEK(&p_sys->out.fifo, p_header);
1496         if(!p_header) break; /* No frame available */
1497
1498         if (p_sys->out.p_fmt->audio.i_channels)
1499             i_samples = p_header->nFilledLen / p_sys->out.p_fmt->audio.i_channels / 2;
1500         if(i_samples)
1501         {
1502             p_buffer = decoder_NewAudioBuffer( p_dec, i_samples );
1503             if( !p_buffer ) break; /* No audio buffer available */
1504
1505             memcpy( p_buffer->p_buffer, p_header->pBuffer, p_buffer->i_buffer );
1506             p_header->nFilledLen = 0;
1507
1508             if( p_header->nTimeStamp != 0 &&
1509                 p_header->nTimeStamp != date_Get( &p_sys->end_date ) )
1510                 date_Set( &p_sys->end_date, p_header->nTimeStamp );
1511
1512             p_buffer->i_pts = date_Get( &p_sys->end_date );
1513             p_buffer->i_length = date_Increment( &p_sys->end_date, i_samples ) -
1514                 p_buffer->i_pts;
1515         }
1516
1517 #ifdef OMXIL_EXTRA_DEBUG
1518         msg_Dbg( p_dec, "FillThisBuffer %p, %p", p_header, p_header->pBuffer );
1519 #endif
1520         OMX_FIFO_GET(&p_sys->out.fifo, p_header);
1521         OMX_FillThisBuffer(p_sys->omx_handle, p_header);
1522     }
1523
1524
1525     /* Send the input buffer to the component */
1526     OMX_FIFO_GET(&p_sys->in.fifo, p_header);
1527
1528     if (p_header && p_header->nFlags & SENTINEL_FLAG) {
1529         free(p_header);
1530         goto reconfig;
1531     }
1532
1533     if(p_header)
1534     {
1535         p_header->nFilledLen = p_block->i_buffer;
1536         p_header->nOffset = 0;
1537         p_header->nFlags = OMX_BUFFERFLAG_ENDOFFRAME;
1538         p_header->nTimeStamp = p_block->i_dts;
1539
1540         /* In direct mode we pass the input pointer as is.
1541          * Otherwise we memcopy the data */
1542         if(p_sys->in.b_direct)
1543         {
1544             p_header->pOutputPortPrivate = p_header->pBuffer;
1545             p_header->pBuffer = p_block->p_buffer;
1546             p_header->pAppPrivate = p_block;
1547         }
1548         else
1549         {
1550             if(p_header->nFilledLen > p_header->nAllocLen)
1551             {
1552                 msg_Dbg(p_dec, "buffer too small (%i,%i)",
1553                         (int)p_header->nFilledLen, (int)p_header->nAllocLen);
1554                 p_header->nFilledLen = p_header->nAllocLen;
1555             }
1556             memcpy(p_header->pBuffer, p_block->p_buffer, p_header->nFilledLen );
1557             block_Release(p_block);
1558         }
1559
1560 #ifdef OMXIL_EXTRA_DEBUG
1561         msg_Dbg( p_dec, "EmptyThisBuffer %p, %p, %i", p_header, p_header->pBuffer,
1562                  (int)p_header->nFilledLen );
1563 #endif
1564         OMX_EmptyThisBuffer(p_sys->omx_handle, p_header);
1565         p_sys->in.b_flushed = false;
1566         *pp_block = NULL; /* Avoid being fed the same packet again */
1567     }
1568
1569 reconfig:
1570     /* Handle the PortSettingsChanged events */
1571     for(i = 0; i < p_sys->ports; i++)
1572     {
1573         OmxPort *p_port = &p_sys->p_ports[i];
1574         if(!p_port->b_reconfigure) continue;
1575         p_port->b_reconfigure = 0;
1576         omx_error = PortReconfigure(p_dec, p_port);
1577     }
1578
1579     return p_buffer;
1580 }
1581
1582 /*****************************************************************************
1583  * EncodeVideo: Called to encode one frame
1584  *****************************************************************************/
1585 static block_t *EncodeVideo( encoder_t *p_enc, picture_t *p_pic )
1586 {
1587     decoder_t *p_dec = ( decoder_t *)p_enc;
1588     decoder_sys_t *p_sys = p_dec->p_sys;
1589     OMX_ERRORTYPE omx_error;
1590     unsigned int i;
1591
1592     OMX_BUFFERHEADERTYPE *p_header;
1593     block_t *p_block = 0;
1594
1595     if( !p_pic ) return NULL;
1596
1597     /* Check for errors from codec */
1598     if(p_sys->b_error)
1599     {
1600         msg_Dbg(p_dec, "error during encoding");
1601         return NULL;
1602     }
1603
1604     /* Send the input buffer to the component */
1605     OMX_FIFO_GET(&p_sys->in.fifo, p_header);
1606     if(p_header)
1607     {
1608         /* In direct mode we pass the input pointer as is.
1609          * Otherwise we memcopy the data */
1610         if(p_sys->in.b_direct)
1611         {
1612             p_header->pOutputPortPrivate = p_header->pBuffer;
1613             p_header->pBuffer = p_pic->p[0].p_pixels;
1614         }
1615         else
1616         {
1617             CopyVlcPicture(p_dec, p_header, p_pic);
1618         }
1619
1620         p_header->nFilledLen = p_sys->in.i_frame_size;
1621         p_header->nOffset = 0;
1622         p_header->nFlags = OMX_BUFFERFLAG_ENDOFFRAME;
1623         p_header->nTimeStamp = p_pic->date;
1624 #ifdef OMXIL_EXTRA_DEBUG
1625         msg_Dbg( p_dec, "EmptyThisBuffer %p, %p, %i", p_header, p_header->pBuffer,
1626                  (int)p_header->nFilledLen );
1627 #endif
1628         OMX_EmptyThisBuffer(p_sys->omx_handle, p_header);
1629         p_sys->in.b_flushed = false;
1630     }
1631
1632     /* Handle the PortSettingsChanged events */
1633     for(i = 0; i < p_sys->ports; i++)
1634     {
1635         OmxPort *p_port = &p_sys->p_ports[i];
1636         if(!p_port->b_reconfigure) continue;
1637         p_port->b_reconfigure = 0;
1638         omx_error = PortReconfigure(p_dec, p_port);
1639     }
1640
1641     /* Wait for the decoded frame */
1642     while(!p_block)
1643     {
1644         OMX_FIFO_GET(&p_sys->out.fifo, p_header);
1645
1646         if(p_header->nFilledLen)
1647         {
1648             if(p_header->nFlags & OMX_BUFFERFLAG_CODECCONFIG)
1649             {
1650                 /* TODO: need to store codec config */
1651                 msg_Dbg(p_dec, "received codec config %i", (int)p_header->nFilledLen);
1652             }
1653
1654             p_block = p_header->pAppPrivate;
1655             if(!p_block)
1656             {
1657                 /* We're not in direct rendering mode.
1658                  * Get a new block and copy the content */
1659                 p_block = block_Alloc( p_header->nFilledLen );
1660                 memcpy(p_block->p_buffer, p_header->pBuffer, p_header->nFilledLen );
1661             }
1662
1663             p_block->i_buffer = p_header->nFilledLen;
1664             p_block->i_pts = p_block->i_dts = p_header->nTimeStamp;
1665             p_header->nFilledLen = 0;
1666             p_header->pAppPrivate = 0;
1667         }
1668
1669 #ifdef OMXIL_EXTRA_DEBUG
1670         msg_Dbg( p_dec, "FillThisBuffer %p, %p", p_header, p_header->pBuffer );
1671 #endif
1672         OMX_FillThisBuffer(p_sys->omx_handle, p_header);
1673     }
1674
1675     msg_Dbg(p_dec, "done");
1676     return p_block;
1677 }
1678
1679 /*****************************************************************************
1680  * CloseGeneric: omxil decoder destruction
1681  *****************************************************************************/
1682 static void CloseGeneric( vlc_object_t *p_this )
1683 {
1684     decoder_t *p_dec = (decoder_t *)p_this;
1685     decoder_sys_t *p_sys = p_dec->p_sys;
1686
1687     if(p_sys->omx_handle) DeinitialiseComponent(p_dec, p_sys->omx_handle);
1688     vlc_mutex_lock( &omx_core_mutex );
1689     omx_refcount--;
1690     if( omx_refcount == 0 )
1691     {
1692         if( p_sys->b_init ) pf_deinit();
1693         dll_close( dll_handle );
1694     }
1695     vlc_mutex_unlock( &omx_core_mutex );
1696
1697     vlc_mutex_destroy (&p_sys->mutex);
1698     vlc_cond_destroy (&p_sys->cond);
1699     vlc_mutex_destroy (&p_sys->in.fifo.lock);
1700     vlc_cond_destroy (&p_sys->in.fifo.wait);
1701     vlc_mutex_destroy (&p_sys->out.fifo.lock);
1702     vlc_cond_destroy (&p_sys->out.fifo.wait);
1703
1704     free( p_sys );
1705 }
1706
1707 /*****************************************************************************
1708  * OmxEventHandler:
1709  *****************************************************************************/
1710 static OMX_ERRORTYPE OmxEventHandler( OMX_HANDLETYPE omx_handle,
1711     OMX_PTR app_data, OMX_EVENTTYPE event, OMX_U32 data_1,
1712     OMX_U32 data_2, OMX_PTR event_data )
1713 {
1714     decoder_t *p_dec = (decoder_t *)app_data;
1715     decoder_sys_t *p_sys = p_dec->p_sys;
1716     unsigned int i;
1717     (void)omx_handle;
1718
1719     switch (event)
1720     {
1721     case OMX_EventCmdComplete:
1722         switch ((OMX_STATETYPE)data_1)
1723         {
1724         case OMX_CommandStateSet:
1725             msg_Dbg( p_dec, "OmxEventHandler (%s, %s, %s)", EventToString(event),
1726                      CommandToString(data_1), StateToString(data_2) );
1727             break;
1728
1729         default:
1730             msg_Dbg( p_dec, "OmxEventHandler (%s, %s, %u)", EventToString(event),
1731                      CommandToString(data_1), (unsigned int)data_2 );
1732             break;
1733         }
1734         break;
1735
1736     case OMX_EventError:
1737         msg_Dbg( p_dec, "OmxEventHandler (%s, %s, %u, %s)", EventToString(event),
1738                  ErrorToString((OMX_ERRORTYPE)data_1), (unsigned int)data_2,
1739                  (const char *)event_data);
1740         //p_sys->b_error = true;
1741         break;
1742
1743     case OMX_EventPortSettingsChanged:
1744         msg_Dbg( p_dec, "OmxEventHandler (%s, %u, %u)", EventToString(event),
1745                  (unsigned int)data_1, (unsigned int)data_2 );
1746         if( data_2 == 0 || data_2 == OMX_IndexParamPortDefinition )
1747         {
1748             OMX_BUFFERHEADERTYPE *sentinel;
1749             for(i = 0; i < p_sys->ports; i++)
1750                 if(p_sys->p_ports[i].definition.eDir == OMX_DirOutput)
1751                     p_sys->p_ports[i].b_reconfigure = true;
1752             sentinel = calloc(1, sizeof(*sentinel));
1753             if (sentinel) {
1754                 sentinel->nFlags = SENTINEL_FLAG;
1755                 OMX_FIFO_PUT(&p_sys->in.fifo, sentinel);
1756             }
1757         }
1758         else if( data_2 == OMX_IndexConfigCommonOutputCrop )
1759         {
1760             for(i = 0; i < p_sys->ports; i++)
1761                 if(p_sys->p_ports[i].definition.nPortIndex == data_1)
1762                     p_sys->p_ports[i].b_update_def = true;
1763         }
1764         else
1765         {
1766             msg_Dbg( p_dec, "Unhandled setting change %x", (unsigned int)data_2 );
1767         }
1768         break;
1769
1770     default:
1771         msg_Dbg( p_dec, "OmxEventHandler (%s, %u, %u)", EventToString(event),
1772                  (unsigned int)data_1, (unsigned int)data_2 );
1773         break;
1774     }
1775
1776     PostOmxEvent(p_dec, event, data_1, data_2, event_data);
1777     return OMX_ErrorNone;
1778 }
1779
1780 static OMX_ERRORTYPE OmxEmptyBufferDone( OMX_HANDLETYPE omx_handle,
1781     OMX_PTR app_data, OMX_BUFFERHEADERTYPE *omx_header )
1782 {
1783     decoder_t *p_dec = (decoder_t *)app_data;
1784     decoder_sys_t *p_sys = p_dec->p_sys;
1785     (void)omx_handle;
1786
1787 #ifdef OMXIL_EXTRA_DEBUG
1788     msg_Dbg( p_dec, "OmxEmptyBufferDone %p, %p", omx_header, omx_header->pBuffer );
1789 #endif
1790
1791     if(omx_header->pAppPrivate || omx_header->pOutputPortPrivate)
1792     {
1793         block_t *p_block = (block_t *)omx_header->pAppPrivate;
1794         omx_header->pBuffer = omx_header->pOutputPortPrivate;
1795         if(p_block) block_Release(p_block);
1796         omx_header->pAppPrivate = 0;
1797     }
1798     OMX_FIFO_PUT(&p_sys->in.fifo, omx_header);
1799
1800     return OMX_ErrorNone;
1801 }
1802
1803 static OMX_ERRORTYPE OmxFillBufferDone( OMX_HANDLETYPE omx_handle,
1804     OMX_PTR app_data, OMX_BUFFERHEADERTYPE *omx_header )
1805 {
1806     decoder_t *p_dec = (decoder_t *)app_data;
1807     decoder_sys_t *p_sys = p_dec->p_sys;
1808     (void)omx_handle;
1809
1810 #ifdef OMXIL_EXTRA_DEBUG
1811     msg_Dbg( p_dec, "OmxFillBufferDone %p, %p, %i", omx_header, omx_header->pBuffer,
1812              (int)omx_header->nFilledLen );
1813 #endif
1814
1815     if(omx_header->pInputPortPrivate)
1816     {
1817         omx_header->pBuffer = omx_header->pInputPortPrivate;
1818     }
1819     OMX_FIFO_PUT(&p_sys->out.fifo, omx_header);
1820
1821     return OMX_ErrorNone;
1822 }