Home | History | Annotate | Download | only in src
      1 /*
      2  *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
      3  *
      4  *  Use of this source code is governed by a BSD-style license
      5  *  that can be found in the LICENSE file in the root of the source
      6  *  tree. An additional intellectual property rights grant can be found
      7  *  in the file PATENTS.  All contributing project authors may
      8  *  be found in the AUTHORS file in the root of the source tree.
      9  */
     10 
     11 /*!\file
     12  * \brief Provides the high level interface to wrap decoder algorithms.
     13  *
     14  */
     15 #include <string.h>
     16 #include "vpx/internal/vpx_codec_internal.h"
     17 
     18 #define SAVE_STATUS(ctx, var) (ctx ? (ctx->err = var) : var)
     19 
     20 static vpx_codec_alg_priv_t *get_alg_priv(vpx_codec_ctx_t *ctx) {
     21   return (vpx_codec_alg_priv_t *)ctx->priv;
     22 }
     23 
     24 vpx_codec_err_t vpx_codec_dec_init_ver(vpx_codec_ctx_t *ctx,
     25                                        vpx_codec_iface_t *iface,
     26                                        const vpx_codec_dec_cfg_t *cfg,
     27                                        vpx_codec_flags_t flags, int ver) {
     28   vpx_codec_err_t res;
     29 
     30   if (ver != VPX_DECODER_ABI_VERSION)
     31     res = VPX_CODEC_ABI_MISMATCH;
     32   else if (!ctx || !iface)
     33     res = VPX_CODEC_INVALID_PARAM;
     34   else if (iface->abi_version != VPX_CODEC_INTERNAL_ABI_VERSION)
     35     res = VPX_CODEC_ABI_MISMATCH;
     36   else if ((flags & VPX_CODEC_USE_POSTPROC) &&
     37            !(iface->caps & VPX_CODEC_CAP_POSTPROC))
     38     res = VPX_CODEC_INCAPABLE;
     39   else if ((flags & VPX_CODEC_USE_ERROR_CONCEALMENT) &&
     40            !(iface->caps & VPX_CODEC_CAP_ERROR_CONCEALMENT))
     41     res = VPX_CODEC_INCAPABLE;
     42   else if ((flags & VPX_CODEC_USE_INPUT_FRAGMENTS) &&
     43            !(iface->caps & VPX_CODEC_CAP_INPUT_FRAGMENTS))
     44     res = VPX_CODEC_INCAPABLE;
     45   else if (!(iface->caps & VPX_CODEC_CAP_DECODER))
     46     res = VPX_CODEC_INCAPABLE;
     47   else {
     48     memset(ctx, 0, sizeof(*ctx));
     49     ctx->iface = iface;
     50     ctx->name = iface->name;
     51     ctx->priv = NULL;
     52     ctx->init_flags = flags;
     53     ctx->config.dec = cfg;
     54 
     55     res = ctx->iface->init(ctx, NULL);
     56     if (res) {
     57       ctx->err_detail = ctx->priv ? ctx->priv->err_detail : NULL;
     58       vpx_codec_destroy(ctx);
     59     }
     60   }
     61 
     62   return SAVE_STATUS(ctx, res);
     63 }
     64 
     65 vpx_codec_err_t vpx_codec_peek_stream_info(vpx_codec_iface_t *iface,
     66                                            const uint8_t *data,
     67                                            unsigned int data_sz,
     68                                            vpx_codec_stream_info_t *si) {
     69   vpx_codec_err_t res;
     70 
     71   if (!iface || !data || !data_sz || !si ||
     72       si->sz < sizeof(vpx_codec_stream_info_t))
     73     res = VPX_CODEC_INVALID_PARAM;
     74   else {
     75     /* Set default/unknown values */
     76     si->w = 0;
     77     si->h = 0;
     78 
     79     res = iface->dec.peek_si(data, data_sz, si);
     80   }
     81 
     82   return res;
     83 }
     84 
     85 vpx_codec_err_t vpx_codec_get_stream_info(vpx_codec_ctx_t *ctx,
     86                                           vpx_codec_stream_info_t *si) {
     87   vpx_codec_err_t res;
     88 
     89   if (!ctx || !si || si->sz < sizeof(vpx_codec_stream_info_t))
     90     res = VPX_CODEC_INVALID_PARAM;
     91   else if (!ctx->iface || !ctx->priv)
     92     res = VPX_CODEC_ERROR;
     93   else {
     94     /* Set default/unknown values */
     95     si->w = 0;
     96     si->h = 0;
     97 
     98     res = ctx->iface->dec.get_si(get_alg_priv(ctx), si);
     99   }
    100 
    101   return SAVE_STATUS(ctx, res);
    102 }
    103 
    104 vpx_codec_err_t vpx_codec_decode(vpx_codec_ctx_t *ctx, const uint8_t *data,
    105                                  unsigned int data_sz, void *user_priv,
    106                                  long deadline) {
    107   vpx_codec_err_t res;
    108 
    109   /* Sanity checks */
    110   /* NULL data ptr allowed if data_sz is 0 too */
    111   if (!ctx || (!data && data_sz) || (data && !data_sz))
    112     res = VPX_CODEC_INVALID_PARAM;
    113   else if (!ctx->iface || !ctx->priv)
    114     res = VPX_CODEC_ERROR;
    115   else {
    116     res = ctx->iface->dec.decode(get_alg_priv(ctx), data, data_sz, user_priv,
    117                                  deadline);
    118   }
    119 
    120   return SAVE_STATUS(ctx, res);
    121 }
    122 
    123 vpx_image_t *vpx_codec_get_frame(vpx_codec_ctx_t *ctx, vpx_codec_iter_t *iter) {
    124   vpx_image_t *img;
    125 
    126   if (!ctx || !iter || !ctx->iface || !ctx->priv)
    127     img = NULL;
    128   else
    129     img = ctx->iface->dec.get_frame(get_alg_priv(ctx), iter);
    130 
    131   return img;
    132 }
    133 
    134 vpx_codec_err_t vpx_codec_register_put_frame_cb(vpx_codec_ctx_t *ctx,
    135                                                 vpx_codec_put_frame_cb_fn_t cb,
    136                                                 void *user_priv) {
    137   vpx_codec_err_t res;
    138 
    139   if (!ctx || !cb)
    140     res = VPX_CODEC_INVALID_PARAM;
    141   else if (!ctx->iface || !ctx->priv ||
    142            !(ctx->iface->caps & VPX_CODEC_CAP_PUT_FRAME))
    143     res = VPX_CODEC_ERROR;
    144   else {
    145     ctx->priv->dec.put_frame_cb.u.put_frame = cb;
    146     ctx->priv->dec.put_frame_cb.user_priv = user_priv;
    147     res = VPX_CODEC_OK;
    148   }
    149 
    150   return SAVE_STATUS(ctx, res);
    151 }
    152 
    153 vpx_codec_err_t vpx_codec_register_put_slice_cb(vpx_codec_ctx_t *ctx,
    154                                                 vpx_codec_put_slice_cb_fn_t cb,
    155                                                 void *user_priv) {
    156   vpx_codec_err_t res;
    157 
    158   if (!ctx || !cb)
    159     res = VPX_CODEC_INVALID_PARAM;
    160   else if (!ctx->iface || !ctx->priv ||
    161            !(ctx->iface->caps & VPX_CODEC_CAP_PUT_SLICE))
    162     res = VPX_CODEC_ERROR;
    163   else {
    164     ctx->priv->dec.put_slice_cb.u.put_slice = cb;
    165     ctx->priv->dec.put_slice_cb.user_priv = user_priv;
    166     res = VPX_CODEC_OK;
    167   }
    168 
    169   return SAVE_STATUS(ctx, res);
    170 }
    171 
    172 vpx_codec_err_t vpx_codec_set_frame_buffer_functions(
    173     vpx_codec_ctx_t *ctx, vpx_get_frame_buffer_cb_fn_t cb_get,
    174     vpx_release_frame_buffer_cb_fn_t cb_release, void *cb_priv) {
    175   vpx_codec_err_t res;
    176 
    177   if (!ctx || !cb_get || !cb_release) {
    178     res = VPX_CODEC_INVALID_PARAM;
    179   } else if (!ctx->iface || !ctx->priv ||
    180              !(ctx->iface->caps & VPX_CODEC_CAP_EXTERNAL_FRAME_BUFFER)) {
    181     res = VPX_CODEC_ERROR;
    182   } else {
    183     res = ctx->iface->dec.set_fb_fn(get_alg_priv(ctx), cb_get, cb_release,
    184                                     cb_priv);
    185   }
    186 
    187   return SAVE_STATUS(ctx, res);
    188 }
    189