vpx_decoder.c 5.95 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
2
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
John Koleszar's avatar
John Koleszar committed
3
 *
4
 *  Use of this source code is governed by a BSD-style license
5 6
 *  that can be found in the LICENSE file in the root of the source
 *  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.
John Koleszar's avatar
John Koleszar committed
9 10 11
 */


12
/*!\file
John Koleszar's avatar
John Koleszar committed
13 14 15
 * \brief Provides the high level interface to wrap decoder algorithms.
 *
 */
16
#include <string.h>
17
#include "vpx/internal/vpx_codec_internal.h"
John Koleszar's avatar
John Koleszar committed
18 19 20

#define SAVE_STATUS(ctx,var) (ctx?(ctx->err = var):var)

21 22 23 24
vpx_codec_err_t vpx_codec_dec_init_ver(vpx_codec_ctx_t      *ctx,
                                       vpx_codec_iface_t    *iface,
                                       vpx_codec_dec_cfg_t  *cfg,
                                       vpx_codec_flags_t     flags,
John Koleszar's avatar
John Koleszar committed
25 26 27 28 29 30 31 32 33 34 35
                                       int                   ver) {
  vpx_codec_err_t res;

  if (ver != VPX_DECODER_ABI_VERSION)
    res = VPX_CODEC_ABI_MISMATCH;
  else if (!ctx || !iface)
    res = VPX_CODEC_INVALID_PARAM;
  else if (iface->abi_version != VPX_CODEC_INTERNAL_ABI_VERSION)
    res = VPX_CODEC_ABI_MISMATCH;
  else if ((flags & VPX_CODEC_USE_POSTPROC) && !(iface->caps & VPX_CODEC_CAP_POSTPROC))
    res = VPX_CODEC_INCAPABLE;
John Koleszar's avatar
John Koleszar committed
36 37 38 39 40
  else if ((flags & VPX_CODEC_USE_ERROR_CONCEALMENT) &&
           !(iface->caps & VPX_CODEC_CAP_ERROR_CONCEALMENT))
    res = VPX_CODEC_INCAPABLE;
  else if ((flags & VPX_CODEC_USE_INPUT_FRAGMENTS) &&
           !(iface->caps & VPX_CODEC_CAP_INPUT_FRAGMENTS))
John Koleszar's avatar
John Koleszar committed
41 42 43 44 45 46 47 48 49 50 51
    res = VPX_CODEC_INCAPABLE;
  else if (!(iface->caps & VPX_CODEC_CAP_DECODER))
    res = VPX_CODEC_INCAPABLE;
  else {
    memset(ctx, 0, sizeof(*ctx));
    ctx->iface = iface;
    ctx->name = iface->name;
    ctx->priv = NULL;
    ctx->init_flags = flags;
    ctx->config.dec = cfg;

52 53 54 55
    res = ctx->iface->init(ctx, NULL);
    if (res) {
      ctx->err_detail = ctx->priv ? ctx->priv->err_detail : NULL;
      vpx_codec_destroy(ctx);
56
    }
57 58 59

    if (ctx->priv)
      ctx->priv->iface = ctx->iface;
John Koleszar's avatar
John Koleszar committed
60
  }
61

John Koleszar's avatar
John Koleszar committed
62
  return SAVE_STATUS(ctx, res);
63 64 65
}


John Koleszar's avatar
John Koleszar committed
66
vpx_codec_err_t vpx_codec_peek_stream_info(vpx_codec_iface_t       *iface,
John Koleszar's avatar
John Koleszar committed
67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
                                           const uint8_t         *data,
                                           unsigned int           data_sz,
                                           vpx_codec_stream_info_t *si) {
  vpx_codec_err_t res;

  if (!iface || !data || !data_sz || !si
      || si->sz < sizeof(vpx_codec_stream_info_t))
    res = VPX_CODEC_INVALID_PARAM;
  else {
    /* Set default/unknown values */
    si->w = 0;
    si->h = 0;

    res = iface->dec.peek_si(data, data_sz, si);
  }

  return res;
John Koleszar's avatar
John Koleszar committed
84 85 86 87
}


vpx_codec_err_t vpx_codec_get_stream_info(vpx_codec_ctx_t         *ctx,
John Koleszar's avatar
John Koleszar committed
88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
                                          vpx_codec_stream_info_t *si) {
  vpx_codec_err_t res;

  if (!ctx || !si || si->sz < sizeof(vpx_codec_stream_info_t))
    res = VPX_CODEC_INVALID_PARAM;
  else if (!ctx->iface || !ctx->priv)
    res = VPX_CODEC_ERROR;
  else {
    /* Set default/unknown values */
    si->w = 0;
    si->h = 0;

    res = ctx->iface->dec.get_si(ctx->priv->alg_priv, si);
  }

  return SAVE_STATUS(ctx, res);
John Koleszar's avatar
John Koleszar committed
104 105 106 107 108 109 110
}


vpx_codec_err_t vpx_codec_decode(vpx_codec_ctx_t    *ctx,
                                 const uint8_t        *data,
                                 unsigned int    data_sz,
                                 void       *user_priv,
John Koleszar's avatar
John Koleszar committed
111 112 113 114 115 116 117 118 119 120
                                 long        deadline) {
  vpx_codec_err_t res;

  /* Sanity checks */
  /* NULL data ptr allowed if data_sz is 0 too */
  if (!ctx || (!data && data_sz))
    res = VPX_CODEC_INVALID_PARAM;
  else if (!ctx->iface || !ctx->priv)
    res = VPX_CODEC_ERROR;
  else {
121 122
    res = ctx->iface->dec.decode(ctx->priv->alg_priv, data, data_sz,
                                 user_priv, deadline);
John Koleszar's avatar
John Koleszar committed
123 124 125
  }

  return SAVE_STATUS(ctx, res);
John Koleszar's avatar
John Koleszar committed
126 127 128
}

vpx_image_t *vpx_codec_get_frame(vpx_codec_ctx_t  *ctx,
John Koleszar's avatar
John Koleszar committed
129 130
                                 vpx_codec_iter_t *iter) {
  vpx_image_t *img;
John Koleszar's avatar
John Koleszar committed
131

John Koleszar's avatar
John Koleszar committed
132 133 134 135
  if (!ctx || !iter || !ctx->iface || !ctx->priv)
    img = NULL;
  else
    img = ctx->iface->dec.get_frame(ctx->priv->alg_priv, iter);
John Koleszar's avatar
John Koleszar committed
136

John Koleszar's avatar
John Koleszar committed
137
  return img;
John Koleszar's avatar
John Koleszar committed
138 139 140 141
}


vpx_codec_err_t vpx_codec_register_put_frame_cb(vpx_codec_ctx_t             *ctx,
John Koleszar's avatar
John Koleszar committed
142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157
                                                vpx_codec_put_frame_cb_fn_t  cb,
                                                void                      *user_priv) {
  vpx_codec_err_t res;

  if (!ctx || !cb)
    res = VPX_CODEC_INVALID_PARAM;
  else if (!ctx->iface || !ctx->priv
           || !(ctx->iface->caps & VPX_CODEC_CAP_PUT_FRAME))
    res = VPX_CODEC_ERROR;
  else {
    ctx->priv->dec.put_frame_cb.u.put_frame = cb;
    ctx->priv->dec.put_frame_cb.user_priv = user_priv;
    res = VPX_CODEC_OK;
  }

  return SAVE_STATUS(ctx, res);
John Koleszar's avatar
John Koleszar committed
158 159 160 161
}


vpx_codec_err_t vpx_codec_register_put_slice_cb(vpx_codec_ctx_t             *ctx,
John Koleszar's avatar
John Koleszar committed
162 163 164 165 166 167 168
                                                vpx_codec_put_slice_cb_fn_t  cb,
                                                void                      *user_priv) {
  vpx_codec_err_t res;

  if (!ctx || !cb)
    res = VPX_CODEC_INVALID_PARAM;
  else if (!ctx->iface || !ctx->priv
169
           || !(ctx->iface->caps & VPX_CODEC_CAP_PUT_SLICE))
John Koleszar's avatar
John Koleszar committed
170 171 172 173 174 175 176 177
    res = VPX_CODEC_ERROR;
  else {
    ctx->priv->dec.put_slice_cb.u.put_slice = cb;
    ctx->priv->dec.put_slice_cb.user_priv = user_priv;
    res = VPX_CODEC_OK;
  }

  return SAVE_STATUS(ctx, res);
John Koleszar's avatar
John Koleszar committed
178 179
}

180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196
vpx_codec_err_t vpx_codec_set_frame_buffer_functions(
    vpx_codec_ctx_t *ctx, vpx_get_frame_buffer_cb_fn_t cb_get,
    vpx_release_frame_buffer_cb_fn_t cb_release, void *cb_priv) {
  vpx_codec_err_t res;

  if (!ctx || !cb_get || !cb_release) {
    res = VPX_CODEC_INVALID_PARAM;
  } else if (!ctx->iface || !ctx->priv ||
             !(ctx->iface->caps & VPX_CODEC_CAP_EXTERNAL_FRAME_BUFFER)) {
    res = VPX_CODEC_ERROR;
  } else {
    res = ctx->iface->dec.set_fb_fn(ctx->priv->alg_priv, cb_get, cb_release,
                                    cb_priv);
  }

  return SAVE_STATUS(ctx, res);
}